mop-tom/lang-main
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects 3 Releases Wiki Activity

updated tests and config nox

Browse Source
This commit is contained in:
Florian Förster
2025-01-23 11:57:52 +01:00
parent 4aec6aa6c4
commit add8da4c5c
7 changed files with 114 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_config.py
View File

@@ -52,7 +52,7 @@ def test_load_cfg_func(monkeypatch, tmp_path):
stop_folder_name=stop_folder,
lookup_cwd=False,
)
assert loaded_cfg['paths']['models'] == '../lang-models'
assert loaded_cfg['paths']['models'] == './lang-models'
loaded_cfg = config.load_cfg(
starting_path=pkg_dir,
@@ -60,4 +60,4 @@ def test_load_cfg_func(monkeypatch, tmp_path):
stop_folder_name=stop_folder,
lookup_cwd=True,
)
assert loaded_cfg['paths']['models'] == '../lang-models'
assert loaded_cfg['paths']['models'] == './lang-models'

15
tests/test_model_loader.py
View File

@@ -26,7 +26,7 @@ from lang_main.types import LanguageModels
'model_name',
[
STFRModelTypes.ALL_MINI_LM_L6_V2,
STFRModelTypes.ALL_MPNET_BASE_V2,
STFRModelTypes.ALL_MINI_LM_L12_V2,
],
)
@pytest.mark.mload
@@ -58,10 +58,9 @@ def test_preprocess_STFR_model_name() -> None:
model_name_exist = STFRModelTypes.E5_BASE_STS_EN_DE
backend_exist = STFRBackends.ONNX
with pytest.raises(FileNotFoundError):
_ = model_loader._preprocess_STFR_model_name(
model_name=model_name_exist, backend=backend_exist, force_download=False
)
_ = model_loader._preprocess_STFR_model_name(
model_name=model_name_exist, backend=backend_exist, force_download=False
)
@pytest.mark.parametrize(
@@ -75,7 +74,7 @@ def test_preprocess_STFR_model_name() -> None:
'model_name',
[
STFRModelTypes.ALL_MINI_LM_L6_V2,
STFRModelTypes.ALL_MPNET_BASE_V2,
STFRModelTypes.ALL_MINI_LM_L12_V2,
],
)
@pytest.mark.mload
@@ -94,9 +93,7 @@ def test_load_sentence_transformer_onnx(model_name, similarity_func) -> None:
@pytest.mark.parametrize(
'model_name',
[
SpacyModelTypes.DE_CORE_NEWS_SM,
SpacyModelTypes.DE_CORE_NEWS_MD,
SpacyModelTypes.DE_CORE_NEWS_LG,
SpacyModelTypes.DE_DEP_NEWS_TRF,
],
)
@@ -111,7 +108,7 @@ def test_load_spacy_model(model_name):
def test_load_spacy_model_fail():
model_name = 'not_existing'
with pytest.raises(LanguageModelNotFoundError):
model = model_loader.load_spacy(model_name)
_ = model_loader.load_spacy(model_name)
@pytest.mark.mload