test: test param helpers

FlickerSoul · Oct 29, 2023 · 269540a · 269540a
1 parent 17b6ad3
commit 269540a
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/tests/test_test_param.py b/tests/test_test_param.py
@@ -19,3 +19,21 @@ def square(a: int) -> int:
         @problem()
         def square(a: int) -> int:
             return a**2
+
+
+def test_test_params_params() -> None:
+    @test_cases.params([1], [2], [3])
+    @problem()
+    def square(a: int) -> int:
+        return a**2
+
+    assert len(square.test_cases) == 3
+
+
+def test_test_params_param_iter() -> None:
+    @test_cases.param_iter([i] for i in range(3))
+    @problem()
+    def square(a: int) -> int:
+        return a**2
+
+    assert len(square.test_cases) == 3