* Add test for parse tree navigation

This commit is contained in:
Matthew Honnibal 2015-01-30 18:02:58 +11:00
parent ef2493a3bd
commit 0b53fd7daa
1 changed files with 28 additions and 0 deletions

View File

@ -0,0 +1,28 @@
from __future__ import unicode_literals
from os import path
import codecs
from spacy.en import English
import pytest
@pytest.fixture
def sun_text():
with codecs.open(path.join(path.dirname(__file__), 'sun.txt'), 'r', 'utf8') as file_:
text = file_.read()
return text
@pytest.fixture
def nlp():
return English()
def test_consistency(nlp, sun_text):
tokens = nlp(sun_text)
for head in tokens:
for child in head.lefts:
assert child.head is head
for child in head.rights:
assert child.head is head