* Fix partial actions test

This commit is contained in:
Matthew Honnibal 2015-08-08 23:38:19 +02:00
parent 01be34d55a
commit c044c0e18e

View File

@ -4,5 +4,9 @@ import pytest
def test_initial(EN): def test_initial(EN):
doc = EN.tokenizer(u'I ate the pizza with anchovies.') doc = EN.tokenizer(u'I ate the pizza with anchovies.')
EN.tagger(doc) EN.tagger(doc)
EN.parser(doc, initial_actions=['L-nsubj', 'S', 'L-det']) next_actions = EN.parser.partial(doc, actions=['L-nsubj', 'S', 'L-det'])
assert doc[0].head.i == 1 assert doc[0].head.i == 1
assert doc[1].head.i == 1
assert doc[2].head.i == 2
assert doc[3].head.i == 2
assert doc