diff --git a/tests/test_jsons.py b/tests/test_jsons.py index a4be928..ca4d0a0 100644 --- a/tests/test_jsons.py +++ b/tests/test_jsons.py @@ -4,16 +4,10 @@ from pydantic import ValidationError import pytest -from aiod_registry import ModelManifest +from aiod_registry import ModelManifest, get_manifest_paths -def get_jsons(): - json_dir = Path(__file__).parent.parent / "aiod_registry" / "manifests" - print(json_dir) - return json_dir.glob("*.json") - - -@pytest.mark.parametrize("json_path", get_jsons()) +@pytest.mark.parametrize("json_path", get_manifest_paths()) def test_manifest(json_path): with open(json_path, "r") as f: json_manifest = json.load(f)