* Add tests for new subtree method

This commit is contained in:
Matthew Honnibal 2015-03-03 05:41:00 -05:00
parent 053814ffc8
commit a61dacb4e5
1 changed files with 19 additions and 0 deletions

19
tests/test_subtree.py Normal file
View File

@ -0,0 +1,19 @@
from __future__ import unicode_literals
from spacy.en import English
EN = English()
def test_subtrees():
sent = EN('The four wheels on the bus turned quickly')
wheels = sent[2]
bus = sent[5]
assert len(list(wheels.lefts)) == 2
assert len(list(wheels.rights)) == 1
assert len(list(wheels.children)) == 3
assert len(list(bus.lefts)) == 1
assert len(list(bus.rights)) == 0
assert len(list(bus.children)) == 1
assert len(list(wheels.subtree)) == 6