diff --git a/tests/test_1d/test_backward_1d.py b/tests/test_1d/test_backward_1d.py index fefc70f..988c578 100644 --- a/tests/test_1d/test_backward_1d.py +++ b/tests/test_1d/test_backward_1d.py @@ -212,7 +212,7 @@ def test_t2_backward_CPU_points( @pytest.mark.parametrize("modifier", length_modifiers) @pytest.mark.parametrize("fftshift", [False, True]) @pytest.mark.parametrize("isign", [-1, 1]) -def test_t2_backward_CPU_values( +def test_t2_backward_cuda_values( N: int, modifier: int, fftshift: bool, isign: int ) -> None: check_t2_backward(N, modifier, fftshift, isign, "cuda", False) @@ -221,7 +221,7 @@ def test_t2_backward_CPU_values( @pytest.mark.parametrize("modifier", length_modifiers) @pytest.mark.parametrize("fftshift", [False, True]) @pytest.mark.parametrize("isign", [-1, 1]) -def test_t2_backward_CPU_points( +def test_t2_backward_cuda_points( N: int, modifier: int, fftshift: bool, isign: int ) -> None: check_t2_backward(N, modifier, fftshift, isign, "cuda", True) diff --git a/tests/test_2d/test_backward_2d.py b/tests/test_2d/test_backward_2d.py index 7650769..88a51bf 100644 --- a/tests/test_2d/test_backward_2d.py +++ b/tests/test_2d/test_backward_2d.py @@ -254,7 +254,7 @@ def test_t2_backward_CPU_points( @pytest.mark.parametrize("modifier", length_modifiers) @pytest.mark.parametrize("fftshift", [False, True]) @pytest.mark.parametrize("isign", [-1, 1]) -def test_t2_backward_CPU_values( +def test_t2_backward_cuda_values( N: int, modifier: int, fftshift: bool, isign: int ) -> None: check_t2_backward(N, modifier, fftshift, isign, "cuda", False) @@ -263,7 +263,7 @@ def test_t2_backward_CPU_values( @pytest.mark.parametrize("modifier", length_modifiers) @pytest.mark.parametrize("fftshift", [False, True]) @pytest.mark.parametrize("isign", [-1, 1]) -def test_t2_backward_CPU_points( +def test_t2_backward_cuda_points( N: int, modifier: int, fftshift: bool, isign: int ) -> None: check_t2_backward(N, modifier, fftshift, isign, "cuda", True) diff --git a/tests/test_3d/test_backward_3d.py b/tests/test_3d/test_backward_3d.py index 5377c25..3c63014 100644 --- a/tests/test_3d/test_backward_3d.py +++ b/tests/test_3d/test_backward_3d.py @@ -292,7 +292,7 @@ def test_t2_backward_CPU_points( @pytest.mark.parametrize("modifier", length_modifiers) @pytest.mark.parametrize("fftshift", [False, True]) @pytest.mark.parametrize("isign", [-1, 1]) -def test_t2_backward_CPU_values( +def test_t2_backward_cuda_values( N: int, modifier: int, fftshift: bool, isign: int ) -> None: check_t2_backward(N, modifier, fftshift, isign, "cuda", False) @@ -301,7 +301,7 @@ def test_t2_backward_CPU_values( @pytest.mark.parametrize("modifier", length_modifiers) @pytest.mark.parametrize("fftshift", [False, True]) @pytest.mark.parametrize("isign", [-1, 1]) -def test_t2_backward_CPU_points( +def test_t2_backward_cuda_points( N: int, modifier: int, fftshift: bool, isign: int ) -> None: check_t2_backward(N, modifier, fftshift, isign, "cuda", True)