spaCy/tests/parser/test_initial_actions_parse.py

13 lines
320 B
Python
Raw Normal View History

import pytest
def test_initial(EN):
doc = EN.tokenizer(u'I ate the pizza with anchovies.')
EN.tagger(doc)
2015-08-09 00:41:36 +03:00
next_actions = EN.parser.partial(doc, ['L-nsubj', 'S', 'L-det'])
assert doc[0].head.i == 1
2015-08-09 00:38:19 +03:00
assert doc[1].head.i == 1
2015-08-09 00:45:36 +03:00
assert doc[2].head.i == 3
assert doc[3].head.i == 3
2015-08-09 00:38:19 +03:00
assert doc