diff --git a/tests/test_ml_artifacts.py b/tests/test_ml_artifacts.py index 853d0dc..0631485 100644 --- a/tests/test_ml_artifacts.py +++ b/tests/test_ml_artifacts.py @@ -23,15 +23,17 @@ def _valid_artifact(): } -def test_pre_purge_repository_artifact_is_rejected_with_actionable_status(): +def test_repository_artifact_has_current_schema_and_purged_provenance(): artifact_path = REPO_ROOT / "config" / "ml_weights.json" artifact = json.loads(artifact_path.read_text()) status = artifacts.validate_ml_artifact(artifact) - assert status["valid"] is False - assert status["schema_version"] is None - assert "artifact_schema_version" in status["errors"] + assert status["valid"] is True + assert status["schema_version"] == artifacts.ML_ARTIFACT_SCHEMA_VERSION + assert status["score_version"] == artifacts.SCORE_VERSION + assert status["has_oos_fold_weights"] is True + assert status["errors"] == [] def test_valid_artifact_requires_schema_score_version_and_purged_provenance():