feat(nlp-service): spaCy model loading with get_nlp/load_all_models

This commit is contained in:
Marcel
2026-06-07 10:17:07 +02:00
committed by marcel
parent 8e63867ad8
commit 482a1c2863
2 changed files with 74 additions and 0 deletions

View File

@@ -31,3 +31,44 @@ def test_parse_response_serializes_nulls():
assert data["dateFrom"] is None
assert data["dateTo"] == "1920-12-31"
assert data["personRole"] == "sender"
# ── Model loading ────────────────────────────────────────────────────────────
@pytest.fixture(scope="session")
def nlp_de():
from extractor import get_nlp
return get_nlp("de")
@pytest.fixture(scope="session")
def nlp_en():
from extractor import get_nlp
return get_nlp("en")
@pytest.fixture(scope="session")
def nlp_es():
from extractor import get_nlp
return get_nlp("es")
def test_get_nlp_de_loads(nlp_de):
doc = nlp_de("Test")
assert doc is not None
def test_get_nlp_en_loads(nlp_en):
doc = nlp_en("Test")
assert doc is not None
def test_get_nlp_es_loads(nlp_es):
doc = nlp_es("Prueba")
assert doc is not None
def test_get_nlp_unknown_lang_raises():
from extractor import get_nlp
with pytest.raises(ValueError, match="Unsupported language"):
get_nlp("fr")