Update test.py

Smartappli · Jan 26, 2024 · a606606 · a606606
1 parent e6c6bd7
commit a606606
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test.py b/test.py
@@ -33,6 +33,8 @@ def check_model_availability(repo, filename):
 checks2 = list(flatten_model_data(model_data2))
 model_data3 = load_model_data(test_dir.parent / "serge-models/finance/models.json")
 checks3 = list(flatten_model_data(model_data3))
+model_data4 = load_model_data(test_dir.parent / "serge-models/french/models.json")
+checks4 = list(flatten_model_data(model_data4))
 
 @pytest.mark.parametrize("repo,filename", checks1)
 def test_model_available1(repo, filename):
@@ -45,3 +47,7 @@ def test_model_available2(repo, filename):
 @pytest.mark.parametrize("repo,filename", checks3)
 def test_model_available3(repo, filename):
     assert check_model_availability(repo, filename), f"Model {repo}/{filename} not available"
+
+@pytest.mark.parametrize("repo,filename", checks4)
+def test_model_available3(repo, filename):
+    assert check_model_availability(repo, filename), f"Model {repo}/{filename} not available"