explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-25 17:36:30 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix displaCy test

Browse Source
This commit is contained in:
ines 2017-11-22 05:04:39 +01:00
parent 42ceece110
commit c90fe92e15
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/tests/test_misc.py
View File

@ -49,8 +49,10 @@ def test_displacy_parse_deps(en_vocab):
words = ["This", "is", "a", "sentence"] words = ["This", "is", "a", "sentence"]
heads = [1, 0, 1, -2] heads = [1, 0, 1, -2]
pos = ['DET', 'VERB', 'DET', 'NOUN'] pos = ['DET', 'VERB', 'DET', 'NOUN']
tags = ['DT', 'VBZ', 'DT', 'NN']
deps = ['nsubj', 'ROOT', 'det', 'attr'] deps = ['nsubj', 'ROOT', 'det', 'attr']
doc = get_doc(en_vocab, words=words, heads=heads, pos=pos, deps=deps) doc = get_doc(en_vocab, words=words, heads=heads, pos=pos, tags=tags,
deps=deps)
deps = parse_deps(doc) deps = parse_deps(doc)
assert isinstance(deps, dict) assert isinstance(deps, dict)
assert deps['words'] == [{'text': 'This', 'tag': 'DET'}, assert deps['words'] == [{'text': 'This', 'tag': 'DET'},