def test_loading_no_warning(): syn = EntitySynonymMapper(synonyms=None) syn.persist("test") meta = Metadata({"test": 1}, "test") with pytest.warns(None) as warn: syn.load("test", meta) assert len(warn) == 0
def test_entity_synonyms(): entities = [{ "entity": "test", "value": "chines", "start": 0, "end": 6 }, { "entity": "test", "value": "chinese", "start": 0, "end": 6 }, { "entity": "test", "value": "china", "start": 0, "end": 6 }] ent_synonyms = {"chines": "chinese", "NYC": "New York City"} EntitySynonymMapper(synonyms=ent_synonyms).replace_synonyms(entities) assert len(entities) == 3 assert entities[0]["value"] == "chinese" assert entities[1]["value"] == "chinese" assert entities[2]["value"] == "china"