TST split test parameters using mark.parametrize instead of within th…

…e test
flatironinstitute · Oct 23, 2023 · ff2d6cf · ff2d6cf
1 parent 340dec8
commit ff2d6cf
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 12 deletions.
diff --git a/tests/test_t1_forward.py b/tests/test_t1_forward.py
@@ -61,13 +61,11 @@ def check_t1_forward(N: int, dim: int, device: str) -> None:
 ]
 
 
-@pytest.mark.parametrize("N_and_dim", Ns_and_dims)
-def test_t1_forward_CPU(N_and_dim) -> None:
-    N, dim = N_and_dim
+@pytest.mark.parametrize("N, dim", Ns_and_dims)
+def test_t1_forward_CPU(N, dim) -> None:
     check_t1_forward(N, dim, "cpu")
 
 
-@pytest.mark.parametrize("N_and_dim", Ns_and_dims)
-def test_t1_forward_cuda(N_and_dim) -> None:
-    N, dim = N_and_dim
+@pytest.mark.parametrize("N, dim", Ns_and_dims)
+def test_t1_forward_cuda(N, dim) -> None:
     check_t1_forward(N, dim, "cuda")
diff --git a/tests/test_t2_forward.py b/tests/test_t2_forward.py
@@ -60,13 +60,11 @@ def check_t2_forward(N: int, dim: int, device: str, fftshift=False) -> None:
 ]
 
 
-@pytest.mark.parametrize("N_and_dim", Ns_and_dims)
-def test_t2_forward_CPU(N_and_dim) -> None:
-    N, dim = N_and_dim
+@pytest.mark.parametrize("N, dim", Ns_and_dims)
+def test_t2_forward_CPU(N, dim) -> None:
     check_t2_forward(N, dim, "cpu")
 
 
-@pytest.mark.parametrize("N_and_dim", Ns_and_dims)
-def test_t2_forward_cuda(N_and_dim) -> None:
-    N, dim = N_and_dim
+@pytest.mark.parametrize("N, dim", Ns_and_dims)
+def test_t2_forward_cuda(N, dim) -> None:
     check_t2_forward(N, dim, "cuda")