Skip to content

Commit 50423ab

Browse files
committed
build: add skip for numpy>=v2.0.0 in all dtcwt tests
1 parent 22c3eca commit 50423ab

1 file changed

Lines changed: 3 additions & 6 deletions

File tree

pytests/test_dtcwt.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -40,11 +40,10 @@ def test_dtcwt1D_input1D(par):
4040
@pytest.mark.skipif(
4141
int(os.environ.get("TEST_CUPY_PYLOPS", 0)) == 1, reason="Not CuPy enabled"
4242
)
43+
@pytest.mark.skipif(int(np_version[0]) >= 2, reason="dtcwt does not support numpy v2")
4344
@pytest.mark.parametrize("par", [(par1), (par2)])
4445
def test_dtcwt1D_input2D(par):
4546
"""Test for DTCWT with 2D input (forward-inverse pair)"""
46-
if int(np_version[0]) >= 2:
47-
return
4847

4948
t = sequential_array(
5049
(
@@ -64,11 +63,10 @@ def test_dtcwt1D_input2D(par):
6463
@pytest.mark.skipif(
6564
int(os.environ.get("TEST_CUPY_PYLOPS", 0)) == 1, reason="Not CuPy enabled"
6665
)
66+
@pytest.mark.skipif(int(np_version[0]) >= 2, reason="dtcwt does not support numpy v2")
6767
@pytest.mark.parametrize("par", [(par1), (par2)])
6868
def test_dtcwt1D_input3D(par):
6969
"""Test for DTCWT with 3D input (forward-inverse pair)"""
70-
if int(np_version[0]) >= 2:
71-
return
7270

7371
t = sequential_array((par["ny"], par["ny"], par["ny"]))
7472

@@ -83,11 +81,10 @@ def test_dtcwt1D_input3D(par):
8381
@pytest.mark.skipif(
8482
int(os.environ.get("TEST_CUPY_PYLOPS", 0)) == 1, reason="Not CuPy enabled"
8583
)
84+
@pytest.mark.skipif(int(np_version[0]) >= 2, reason="dtcwt does not support numpy v2")
8685
@pytest.mark.parametrize("par", [(par1), (par2)])
8786
def test_dtcwt1D_birot(par):
8887
"""Test for DTCWT birot (forward-inverse pair)"""
89-
if int(np_version[0]) >= 2:
90-
return
9188

9289
birots = ["antonini", "legall", "near_sym_a", "near_sym_b"]
9390

0 commit comments

Comments
 (0)