diff --git a/spacy/tests/test_misc.py b/spacy/tests/test_misc.py index 66ecd8a8e..edc303b4f 100644 --- a/spacy/tests/test_misc.py +++ b/spacy/tests/test_misc.py @@ -48,15 +48,15 @@ def test_displacy_parse_deps(en_vocab): """Test that deps and tags on a Doc are converted into displaCy's format.""" words = ["This", "is", "a", "sentence"] heads = [1, 0, 1, -2] - tags = ['DT', 'VBZ', 'DT', 'NN'] + pos = ['DET', 'VERB', 'DET', 'NOUN'] deps = ['nsubj', 'ROOT', 'det', 'attr'] - doc = get_doc(en_vocab, words=words, heads=heads, tags=tags, deps=deps) + doc = get_doc(en_vocab, words=words, heads=heads, pos=pos, deps=deps) deps = parse_deps(doc) assert isinstance(deps, dict) - assert deps['words'] == [{'text': 'This', 'tag': 'DT'}, - {'text': 'is', 'tag': 'VBZ'}, - {'text': 'a', 'tag': 'DT'}, - {'text': 'sentence', 'tag': 'NN'}] + assert deps['words'] == [{'text': 'This', 'tag': 'DET'}, + {'text': 'is', 'tag': 'VERB'}, + {'text': 'a', 'tag': 'DET'}, + {'text': 'sentence', 'tag': 'NOUN'}] assert deps['arcs'] == [{'start': 0, 'end': 1, 'label': 'nsubj', 'dir': 'left'}, {'start': 2, 'end': 3, 'label': 'det', 'dir': 'left'}, {'start': 1, 'end': 3, 'label': 'attr', 'dir': 'right'}]