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

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

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 spacy.attrs import HEAD
import pytest
import numpy
@pytest.mark.models
def test_merge_tokens(EN):
tokens = EN(u'Los Angeles start.')
tokens.from_array([HEAD], numpy.asarray([[1, 1, 0, -1]], dtype='int32').T)
assert len(tokens) == 4
assert tokens[0].head.orth_ == 'Angeles'
assert tokens[1].head.orth_ == 'start'