feat: configurable e2e test cases

Author: anna-singleton-resolver

tests/functional/e2e/test_classify_single.py

@@ -11,10 +11,10 @@
 from resolver_athena_client.client.models import ImageData
 from tests.functional.e2e.testcases.parser import (
     AthenaTestCase,
-    load_test_cases,
+    load_test_cases_by_env,
 )
 
-TEST_CASES = load_test_cases("integrator_sample")
+TEST_CASES = load_test_cases_by_env()
 
 FP_ERROR_TOLERANCE = 1e-4
 

tests/functional/e2e/testcases/parser.py

@@ -1,4 +1,5 @@
 import json
+import os
 from pathlib import Path
 
 # Path to the shared testcases directory in athena-protobufs
@@ -23,6 +24,12 @@ def __init__(
         self.classification_labels: list[str] = classification_labels
 
 
+def load_test_cases_by_env() -> list[AthenaTestCase]:
+    return load_test_cases(
+        os.getenv("ATHENA_E2E_TESTCASE_DIR", "integrator_sample")
+    )
+
+
 def load_test_cases(dirname: str = "benign_model") -> list[AthenaTestCase]:
     with Path.open(
         Path(TESTCASES_DIR / dirname / "expected_outputs.json"),