def test_analysis_validate_attrs_remove_pipe(): """Test that attributes are validated correctly on remove.""" @Language.component("pipe_analysis_c6", assigns=["token.tag"]) def c1(doc): return doc @Language.component("pipe_analysis_c7", requires=["token.pos"]) def c2(doc): return doc nlp = Language() nlp.add_pipe("pipe_analysis_c6") nlp.add_pipe("pipe_analysis_c7") problems = nlp.analyze_pipes()["problems"] assert problems["pipe_analysis_c7"] == ["token.pos"] nlp.remove_pipe("pipe_analysis_c7") problems = nlp.analyze_pipes()["problems"] assert all(p == [] for p in problems.values())
def test_component_decorator_assigns(): @Language.component("c1", assigns=["token.tag", "doc.tensor"]) def test_component1(doc): return doc @Language.component("c2", requires=["token.tag", "token.pos"], assigns=["token.lemma", "doc.tensor"]) def test_component2(doc): return doc @Language.component("c3", requires=["token.lemma"], assigns=["token._.custom_lemma"]) def test_component3(doc): return doc assert Language.has_factory("c1") assert Language.has_factory("c2") assert Language.has_factory("c3") nlp = Language() nlp.add_pipe("c1") nlp.add_pipe("c2") problems = nlp.analyze_pipes()["problems"] assert problems["c2"] == ["token.pos"] nlp.add_pipe("c3") assert get_attr_info(nlp, "doc.tensor")["assigns"] == ["c1", "c2"] nlp.add_pipe("c1", name="c4") test_component4_meta = nlp.get_pipe_meta("c1") assert test_component4_meta.factory == "c1" assert nlp.pipe_names == ["c1", "c2", "c3", "c4"] assert not Language.has_factory("c4") assert nlp.pipe_factories["c1"] == "c1" assert nlp.pipe_factories["c4"] == "c1" assert get_attr_info(nlp, "doc.tensor")["assigns"] == ["c1", "c2", "c4"] assert get_attr_info(nlp, "token.pos")["requires"] == ["c2"] assert nlp("hello world")