Add tests for TestCase.from_record method. Switch the TestCase.from_record method from static to class method.

fkdosilovic · fkdosilovic · commit 2861d3335fd7 · 2024-12-17T21:23:44.000+01:00
diff --git a/src/judge0/base_types.py b/src/judge0/base_types.py
@@ -1,3 +1,5 @@
+import copy
+
 from dataclasses import dataclass
 from enum import IntEnum
 from typing import Optional, Protocol, runtime_checkable, Sequence, Union
@@ -15,21 +17,23 @@ class TestCase:
     input: Optional[str] = None
     expected_output: Optional[str] = None
 
-    @staticmethod
-    def from_record(test_case: Optional[TestCaseType] = None) -> "TestCase":
+    @classmethod
+    def from_record(cls, test_case: TestCaseType) -> "TestCase":
         """Create a TestCase from built-in types."""
         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(
+            return cls(
                 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
+        if isinstance(test_case, cls):
+            return copy.deepcopy(test_case)
+        if test_case is None:
+            return cls()
         raise ValueError(
             f"Cannot create TestCase object from object of type {type(test_case)}."
         )
diff --git a/tests/test_api_test_cases.py b/tests/test_api_test_cases.py
@@ -6,6 +6,51 @@
 from judge0.api import create_submissions_from_test_cases
 
 
+@pytest.mark.parametrize(
+    "test_case,expected_output",
+    [
+        [
+            TestCase(input="input_1", expected_output="output_1"),
+            TestCase(input="input_1", expected_output="output_1"),
+        ],
+        [
+            tuple([]),
+            TestCase(input=None, expected_output=None),
+        ],
+        [
+            ("input_tuple",),
+            TestCase(input="input_tuple", expected_output=None),
+        ],
+        [
+            ("input_tuple", "output_tuple"),
+            TestCase(input="input_tuple", expected_output="output_tuple"),
+        ],
+        [
+            [],
+            TestCase(input=None, expected_output=None),
+        ],
+        [
+            ["input_list"],
+            TestCase(input="input_list", expected_output=None),
+        ],
+        [
+            ["input_list", "output_list"],
+            TestCase(input="input_list", expected_output="output_list"),
+        ],
+        [
+            {"input": "input_dict", "expected_output": "output_dict"},
+            TestCase(input="input_dict", expected_output="output_dict"),
+        ],
+        [
+            None,
+            TestCase(),
+        ],
+    ],
+)
+def test_test_case_from_record(test_case, expected_output):
+    assert TestCase.from_record(test_case) == expected_output
+
+
 @pytest.mark.parametrize(
     "submissions,test_cases,expected_type",
     [
@@ -19,7 +64,7 @@ def test_create_submissions_from_test_cases_return_type(
     submissions, test_cases, expected_type
 ):
     output = create_submissions_from_test_cases(submissions, test_cases)
-    assert type(output) == expected_type
+    assert type(output) is expected_type
 
 
 @pytest.mark.parametrize(
