From 8e60e4a9c92da29d1cc031ad1303b9fd98484fb2 Mon Sep 17 00:00:00 2001 From: Olivier DEBAUCHE Date: Fri, 26 Jan 2024 16:13:41 +0100 Subject: [PATCH] Update test.py --- test.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/test.py b/test.py index 292a9aa..2f53dc9 100644 --- a/test.py +++ b/test.py @@ -37,6 +37,12 @@ def check_model_availability(repo, filename): checks4 = list(flatten_model_data(model_data4)) model_data5 = load_model_data(test_dir.parent / "serge-models/generic/models.json") checks5 = list(flatten_model_data(model_data5)) +model_data6 = load_model_data(test_dir.parent / "serge-models/german/models.json") +checks6 = list(flatten_model_data(model_data6)) +model_data7 = load_model_data(test_dir.parent / "serge-models/german/models.json") +checks7 = list(flatten_model_data(model_data7)) +model_data8 = load_model_data(test_dir.parent / "serge-models/german/models.json") +checks8 = list(flatten_model_data(model_data8)) @pytest.mark.parametrize("repo,filename", checks1) def test_model_available1(repo, filename): @@ -57,3 +63,16 @@ def test_model_available4(repo, filename): @pytest.mark.parametrize("repo,filename", checks5) def test_model_available5(repo, filename): assert check_model_availability(repo, filename), f"Model {repo}/{filename} not available" + +@pytest.mark.parametrize("repo,filename", checks6) +def test_model_available6(repo, filename): + assert check_model_availability(repo, filename), f"Model {repo}/{filename} not available" + +@pytest.mark.parametrize("repo,filename", checks7) +def test_model_available7(repo, filename): + assert check_model_availability(repo, filename), f"Model {repo}/{filename} not available" + +@pytest.mark.parametrize("repo,filename", checks8) +def test_model_available8(repo, filename): + assert check_model_availability(repo, filename), f"Model {repo}/{filename} not available" +