Add test

spacy/tests/pipeline/test_initialize.py

@@ -0,0 +1,42 @@
+import pytest
+from spacy.language import Language
+from spacy.lang.en import English
+from spacy.training import Example
+from thinc.api import ConfigValidationError
+from pydantic import StrictBool
+
+
+def test_initialize_arguments():
+    name = "test_initialize_arguments"
+
+    class Component:
+        def __init__(self):
+            ...
+
+        def initialize(
+            self, get_examples, nlp, custom1: str, custom2: StrictBool = False
+        ):
+            ...
+
+    Language.factory(name, func=lambda nlp, name: Component())
+
+    nlp = English()
+    example = Example.from_dict(nlp("x"), {})
+    get_examples = lambda: [example]
+    nlp.add_pipe(name)
+    # The settings here will typically come from the [initialize] block
+    with pytest.raises(ConfigValidationError) as e:
+        # Empty settings, no required custom1 argument
+        nlp.initialize(get_examples, settings={"components": {name: {}}})
+    errors = e.value.errors
+    assert len(errors) == 1
+    assert errors[0]["loc"] == ("custom1",)
+    assert errors[0]["type"] == "value_error.missing"
+    with pytest.raises(ConfigValidationError) as e:
+        # Wrong type
+        settings = {"components": {name: {"custom1": "x", "custom2": 1}}}
+        nlp.initialize(get_examples, settings=settings)
+    errors = e.value.errors
+    assert len(errors) == 1
+    assert errors[0]["loc"] == ("custom2",)
+    assert errors[0]["type"] == "value_error.strictbool"