* Fix test

This commit is contained in:
Matthew Honnibal 2015-10-13 15:58:46 +11:00
parent c70eb776ae
commit 883ff1f59e

View File

@ -67,8 +67,8 @@ def test_count_by(nlp):
from spacy import attrs from spacy import attrs
tokens = nlp('apple apple orange banana') tokens = nlp('apple apple orange banana')
assert tokens.count_by(attrs.ORTH) == {3699: 2, 3750: 1, 5965: 1} assert tokens.count_by(attrs.ORTH) == {3699: 2, 3750: 1, 5965: 1}
assert repr(tokens.to_array([attrs.ORTH])) == repr(numpy.array([[2529], assert repr(tokens.to_array([attrs.ORTH])) == repr(numpy.array([[3699],
[2529], [3699],
[3750], [3750],
[5965]], dtype=numpy.int32)) [5965]], dtype=numpy.int32))