From c90fe92e15165abfe5fb828ab9e74e9b9659a9f2 Mon Sep 17 00:00:00 2001 From: ines Date: Wed, 22 Nov 2017 05:04:39 +0100 Subject: [PATCH] Fix displaCy test --- spacy/tests/test_misc.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/spacy/tests/test_misc.py b/spacy/tests/test_misc.py index edc303b4f..69a6fd38e 100644 --- a/spacy/tests/test_misc.py +++ b/spacy/tests/test_misc.py @@ -49,8 +49,10 @@ def test_displacy_parse_deps(en_vocab): words = ["This", "is", "a", "sentence"] heads = [1, 0, 1, -2] pos = ['DET', 'VERB', 'DET', 'NOUN'] + tags = ['DT', 'VBZ', 'DT', 'NN'] 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) assert isinstance(deps, dict) assert deps['words'] == [{'text': 'This', 'tag': 'DET'},