spaCy/tests/parser/test_initial_actions_parse.py
2015-08-08 23:41:36 +02:00

13 lines
320 B
Python

import pytest
def test_initial(EN):
doc = EN.tokenizer(u'I ate the pizza with anchovies.')
EN.tagger(doc)
next_actions = EN.parser.partial(doc, ['L-nsubj', 'S', 'L-det'])
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