explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-11 08:42:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Test labels are added implicitly

Browse Source
This commit is contained in:
Matthew Honnibal 2021-01-27 12:52:29 +11:00
parent e7db07a0b9
commit 68b1c2984d
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
spacy/tests/parser/test_add_label.py
View File

@ -1,6 +1,7 @@
import pytest import pytest
from thinc.api import Adam, fix_random_seed from thinc.api import Adam, fix_random_seed
from spacy import registry from spacy import registry
from spacy.language import Language
from spacy.attrs import NORM from spacy.attrs import NORM
from spacy.vocab import Vocab from spacy.vocab import Vocab
from spacy.training import Example from spacy.training import Example
@ -123,3 +124,28 @@ def test_add_label_get_label(pipe_cls, n_moves, model_config):
assert len(pipe.move_names) == len(labels) * n_moves assert len(pipe.move_names) == len(labels) * n_moves
pipe_labels = sorted(list(pipe.labels)) pipe_labels = sorted(list(pipe.labels))
assert pipe_labels == labels assert pipe_labels == labels
def test_ner_labels_added_implicitly_on_predict():
nlp = Language()
ner = nlp.add_pipe("ner")
for label in ["A", "B", "C"]:
ner.add_label(label)
nlp.initialize()
doc = Doc(nlp.vocab, words=["hello", "world"], ents=["B-D", "O"])
ner(doc)
assert [t.ent_type_ for t in doc] == ["D", ""]
assert "D" in ner.labels
def test_ner_labels_added_implicitly_on_update():
nlp = Language()
ner = nlp.add_pipe("ner")
for label in ["A", "B", "C"]:
ner.add_label(label)
nlp.initialize()
doc = Doc(nlp.vocab, words=["hello", "world"], ents=["B-D", "O"])
example = Example(nlp.make_doc(doc.text), doc)
assert "D" not in ner.labels
nlp.update([example])
assert "D" in ner.labels