mirror of
https://github.com/explosion/spaCy.git
synced 2025-01-13 10:46:29 +03:00
Fix broken test, re Issue #461
This commit is contained in:
parent
4de30a8e38
commit
f97548c6f1
|
@ -108,7 +108,7 @@ def test_set_ents(EN):
|
||||||
assert len(tokens.ents) == 0
|
assert len(tokens.ents) == 0
|
||||||
tokens.ents = [(EN.vocab.strings['PRODUCT'], 2, 4)]
|
tokens.ents = [(EN.vocab.strings['PRODUCT'], 2, 4)]
|
||||||
assert len(list(tokens.ents)) == 1
|
assert len(list(tokens.ents)) == 1
|
||||||
assert [t.ent_iob for t in tokens] == [0, 0, 3, 1, 0, 0, 0, 0]
|
assert [t.ent_iob for t in tokens] == [2, 2, 3, 1, 2, 2, 2, 2]
|
||||||
ent = tokens.ents[0]
|
ent = tokens.ents[0]
|
||||||
assert ent.label_ == 'PRODUCT'
|
assert ent.label_ == 'PRODUCT'
|
||||||
assert ent.start == 2
|
assert ent.start == 2
|
||||||
|
|
Loading…
Reference in New Issue
Block a user