From 269540a588fd4c1bede905445786726d486a1b67 Mon Sep 17 00:00:00 2001 From: Heyuan Zeng Date: Sun, 29 Oct 2023 01:23:33 +0100 Subject: [PATCH] test: test param helpers --- tests/test_test_param.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/tests/test_test_param.py b/tests/test_test_param.py index 94a7e73..08d3a17 100644 --- 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