Fix test, after IOB tweak.

This commit is contained in:
Matthew Honnibal 2016-10-26 17:22:03 +02:00
parent 271a120d30
commit 6c47048912
1 changed files with 1 additions and 1 deletions

View File

@ -108,7 +108,7 @@ def test_set_ents(EN):
assert len(tokens.ents) == 0
tokens.ents = [(EN.vocab.strings['PRODUCT'], 2, 4)]
assert len(list(tokens.ents)) == 1
assert [t.ent_iob for t in tokens] == [2, 2, 3, 1, 2, 2, 2, 2]
assert [t.ent_iob for t in tokens] == [0, 0, 3, 1, 0, 0, 0, 0]
ent = tokens.ents[0]
assert ent.label_ == 'PRODUCT'
assert ent.start == 2