From 9d90ab196d3e5410de2e531b345a79d4cbf20019 Mon Sep 17 00:00:00 2001 From: Olivier DEBAUCHE Date: Fri, 26 Jan 2024 15:55:58 +0100 Subject: [PATCH] Update test.py --- test.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/test.py b/test.py index 09f6fdb..c70180d 100644 --- a/test.py +++ b/test.py @@ -31,11 +31,17 @@ def check_model_availability(repo, filename): checks1 = list(flatten_model_data(model_data1)) model_data2 = load_model_data(test_dir.parent / "serge-models/coder/models.json") 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)) @pytest.mark.parametrize("repo,filename", checks1) -def test_model_available(repo, filename): +def test_model_available1(repo, filename): assert check_model_availability(repo, filename), f"Model {repo}/{filename} not available" @pytest.mark.parametrize("repo,filename", checks2) -def test_model_available(repo, filename): +def test_model_available2(repo, filename): + assert check_model_availability(repo, filename), f"Model {repo}/{filename} not available" + +@pytest.mark.parametrize("repo,filename", checks3) +def test_model_available3(repo, filename): assert check_model_availability(repo, filename), f"Model {repo}/{filename} not available"