mirror of https://github.com/explosion/spaCy.git
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
|
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"
|