2016-04-13 16:28:28 +03:00
|
|
|
from __future__ import unicode_literals
|
|
|
|
|
2015-04-28 21:46:29 +03:00
|
|
|
import pytest
|
|
|
|
|
|
|
|
|
2015-07-23 10:26:43 +03:00
|
|
|
@pytest.mark.models
|
2015-06-07 19:02:24 +03:00
|
|
|
def test_root(EN):
|
|
|
|
tokens = EN(u"i don't have other assistance")
|
2015-04-28 21:46:29 +03:00
|
|
|
for t in tokens:
|
|
|
|
assert t.dep != 0, t.orth_
|
2016-04-13 16:28:28 +03:00
|
|
|
|
|
|
|
|
|
|
|
@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
|
2016-04-25 13:01:19 +03:00
|
|
|
|
|
|
|
|
|
|
|
def apply_transition_sequence(model, doc, sequence):
|
|
|
|
with model.parser.step_through(doc) as stepwise:
|
|
|
|
for transition in sequence:
|
|
|
|
stepwise.transition(transition)
|