Skip to content

Commit

Permalink
Update test.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Smartappli authored Jan 26, 2024
1 parent aed9b44 commit 8e60e4a
Showing 1 changed file with 19 additions and 0 deletions.
19 changes: 19 additions & 0 deletions test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand All @@ -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"

0 comments on commit 8e60e4a

Please sign in to comment.