Add support to specify test cases with different built-in Python types.

fkdosilovic · fkdosilovic · commit c3aa10d6f0fb · 2024-11-25T18:28:20.000+01:00
diff --git a/examples/test_cases_01.py b/examples/test_cases_01.py
@@ -0,0 +1,16 @@
+import judge0
+
+
+submissions = judge0.run(
+    source_code="print(f'hello {input()}')",
+    test_cases=[
+        ("tuple", "hello tuple"),
+        {"input": "dict", "expected_output": "hello dict"},
+        ["list", "hello list"],
+    ],
+)
+
+submissions = judge0.run(submissions=submissions)
+
+for submission in submissions:
+    print(submission.status)
diff --git a/src/judge0/api.py b/src/judge0/api.py
@@ -165,10 +165,12 @@ def create_submissions_from_test_cases(
     else:
         submissions_list = submissions
 
-    if isinstance(test_cases, list):
-        test_cases_list = test_cases
-    else:
+    if isinstance(test_cases, TestCase) or test_cases is None:
         test_cases_list = [test_cases]
+    else:
+        test_cases_list = test_cases
+
+    test_cases_list = [TestCase.from_record(tc) for tc in test_cases_list]
 
     all_submissions = []
     for submission in submissions_list:
diff --git a/src/judge0/base_types.py b/src/judge0/base_types.py
@@ -4,7 +4,16 @@
 from typing import Optional, Union
 
 
-TestCases = Union[list["TestCase"], tuple["TestCase"]]
+TestCases = Union[
+    list["TestCase"],
+    tuple["TestCase"],
+    list[dict],
+    tuple[dict],
+    list[list],
+    list[tuple],
+    tuple[list],
+    tuple[tuple],
+]
 
 
 @dataclass(frozen=True)
@@ -15,6 +24,26 @@ class TestCase:
     input: Optional[str] = None
     expected_output: Optional[str] = None
 
+    @staticmethod
+    def from_record(
+        test_case: Optional[Union[tuple, list, dict, "TestCase"]] = None
+    ) -> "TestCase":
+        if isinstance(test_case, (tuple, list)):
+            test_case = {
+                field: value
+                for field, value in zip(("input", "expected_output"), test_case)
+            }
+        if isinstance(test_case, dict):
+            return TestCase(
+                input=test_case.get("input", None),
+                expected_output=test_case.get("expected_output", None),
+            )
+        if isinstance(test_case, TestCase) or test_case is None:
+            return test_case
+        raise ValueError(
+            f"Cannot create TestCase object from object of type {type(test_case)}."
+        )
+
 
 class Encodeable(ABC):
     @abstractmethod
