Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

activate parse_tree test, use from_array, test for root correctness

This commit is contained in:
kengz 2016-10-16 15:12:08 -04:00
parent 17b7832419
commit fb92e2d061
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
spacy/tests/tokens/test_tokens_api.py
View File

@ -208,9 +208,14 @@ def test_has_vector(EN):
assert doc.has_vector assert doc.has_vector
@pytest.mark.models
def test_parse_tree(EN): def test_parse_tree(EN):
trees = EN.parse_tree(u'''Bob brought Alice the pizza.''') text = 'I like New York in Autumn.'
EN = English(parser=False)
doc = EN(text, tag=True)
doc.from_array([HEAD], numpy.asarray([[1, 0, 1, -2, -3, -1, -5]], dtype='int32').T)
# full method parse_tree(text) is a trivial composition
trees = EN.parse_tree_(doc)
assert len(trees) > 0 assert len(trees) > 0
tree = trees[0] tree = trees[0]
assert all(k in list(tree.keys()) for k in ['word', 'lemma', 'NE', 'POS_fine', 'POS_coarse', 'arc', 'modifiers']) assert all(k in list(tree.keys()) for k in ['word', 'lemma', 'NE', 'POS_fine', 'POS_coarse', 'arc', 'modifiers'])
assert tree['word'] == 'like' # check root is correct