From 6c47048912df028bc51288f7ebed6f74fdd9cef3 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Wed, 26 Oct 2016 17:22:03 +0200 Subject: [PATCH] Fix test, after IOB tweak. --- spacy/tests/tokens/test_tokens_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/tests/tokens/test_tokens_api.py b/spacy/tests/tokens/test_tokens_api.py index 62c3a660b..47ad8545f 100644 --- a/spacy/tests/tokens/test_tokens_api.py +++ b/spacy/tests/tokens/test_tokens_api.py @@ -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