explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-27 09:44:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Set heads for test_merge_tokens, to make the test run without models

Browse Source
This commit is contained in:
Matthew Honnibal 2016-01-18 17:00:11 +01:00
parent c107da9738
commit cc4c335e14
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/tests/spans/test_merge.py
View File

@ -1,9 +1,12 @@
from __future__ import unicode_literals from __future__ import unicode_literals
from spacy.attrs import HEAD
import pytest import pytest
import numpy
@pytest.mark.models
def test_merge_tokens(EN): def test_merge_tokens(EN):
tokens = EN(u'Los Angeles start.') tokens = EN(u'Los Angeles start.')
tokens.from_array([HEAD], numpy.asarray([[1, 1, 0, -1]], dtype='int32').T)
assert len(tokens) == 4 assert len(tokens) == 4
assert tokens[0].head.orth_ == 'Angeles' assert tokens[0].head.orth_ == 'Angeles'
assert tokens[1].head.orth_ == 'start' assert tokens[1].head.orth_ == 'start'