diff options
Diffstat (limited to 'stanza/tests/pipeline/test_pipeline_ner_processor.py')
-rw-r--r-- | stanza/tests/pipeline/test_pipeline_ner_processor.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/stanza/tests/pipeline/test_pipeline_ner_processor.py b/stanza/tests/pipeline/test_pipeline_ner_processor.py index a2d7768e..c67089cf 100644 --- a/stanza/tests/pipeline/test_pipeline_ner_processor.py +++ b/stanza/tests/pipeline/test_pipeline_ner_processor.py @@ -142,3 +142,6 @@ class TestMultiNERProcessor: multi_ner = [[token.multi_ner for token in sentence.tokens] for sentence in doc.sentences] assert multi_ner == EXPECTED_MULTI_NER + def test_known_tags(self, pipeline): + assert pipeline.processors["ner"].get_known_tags() == ["DISEASE"] + assert len(pipeline.processors["ner"].get_known_tags(1)) == 18 |