diff --git a/tests/test_sliced_iteration.py b/tests/test_sliced_iteration.py index 0058f4f904b35d94fe71a3f923c21dc2c60cecb9..cc7b87a48793b6684346a103ac6b1820d9a0b9df 100644 --- a/tests/test_sliced_iteration.py +++ b/tests/test_sliced_iteration.py @@ -16,6 +16,9 @@ from pystencils.simp import sympy_cse_on_assignment_list @pytest.mark.parametrize("target", [Target.CPU, Target.GPU]) def test_sliced_iteration(target): + if target == Target.GPU: + pytest.importorskip("cupy") + size = (4, 4) dh = create_data_handling(size, default_target=target, default_ghost_layers=0)