spaCy/spacy/tests/parser/test_parse.py

22 lines
434 B
Python

from __future__ import unicode_literals
import pytest
@pytest.mark.models
def test_root(EN):
tokens = EN(u"i don't have other assistance")
for t in tokens:
assert t.dep != 0, t.orth_
@pytest.mark.models
def test_one_word_sentence(EN):
# one word sentence
doc = EN.tokenizer.tokens_from_list(['Hello'])
EN.tagger(doc)
assert len(doc) == 1
with EN.parser.step_through(doc) as _:
pass
assert doc[0].dep != 0