Skip to content

Commit 940e263

Browse files
committed
test: disable sliding simop tests on CuPy
1 parent 994ff1c commit 940e263

1 file changed

Lines changed: 15 additions & 13 deletions

File tree

pytests/test_sliding.py

Lines changed: 15 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
backend = "numpy"
1313
import pytest
1414

15-
import pylops
1615
from pylops.basicoperators import Identity, MatrixMult
1716
from pylops.signalprocessing import Sliding1D, Sliding2D, Sliding3D
1817
from pylops.signalprocessing.sliding1d import sliding1d_design
@@ -134,11 +133,13 @@ def test_Sliding1D(par):
134133
x = np.ones(par["ny"] * nwins)
135134
y = Slid * x.ravel()
136135

137-
with pylops.enabled_ndarray_multiplication():
138-
xinv = Slid / y
136+
xinv = Slid / y
139137
assert_array_almost_equal(x.ravel(), xinv)
140138

141139

140+
@pytest.mark.skipif(
141+
int(os.environ.get("TEST_CUPY_PYLOPS", 0)) == 1, reason="Not CuPy enabled"
142+
)
142143
@pytest.mark.parametrize("par", [(par1), (par2), (par3), (par4)])
143144
def test_Sliding1D_simOp(par):
144145
"""Dot-test and inverse for Sliding1D operator with
@@ -163,8 +164,7 @@ def test_Sliding1D_simOp(par):
163164
x = np.ones(par["nwiny"] * nwins)
164165
y = Slid * x.ravel()
165166

166-
with pylops.enabled_ndarray_multiplication():
167-
xinv = Slid / y
167+
xinv = Slid / y
168168
assert_array_almost_equal(x.ravel(), xinv)
169169

170170

@@ -191,11 +191,13 @@ def test_Sliding2D(par):
191191
x = np.ones((par["ny"] * nwins, par["nt"]))
192192
y = Slid * x.ravel()
193193

194-
with pylops.enabled_ndarray_multiplication():
195-
xinv = Slid / y
194+
xinv = Slid / y
196195
assert_array_almost_equal(x.ravel(), xinv)
197196

198197

198+
@pytest.mark.skipif(
199+
int(os.environ.get("TEST_CUPY_PYLOPS", 0)) == 1, reason="Not CuPy enabled"
200+
)
199201
@pytest.mark.parametrize("par", [(par1), (par2), (par3), (par4)])
200202
def test_Sliding2D_simOp(par):
201203
"""Dot-test and inverse for Sliding2D operator with
@@ -219,8 +221,7 @@ def test_Sliding2D_simOp(par):
219221
x = np.ones((nwins, par["nwiny"], par["nt"]))
220222
y = Slid * x.ravel()
221223

222-
with pylops.enabled_ndarray_multiplication():
223-
xinv = Slid / y
224+
xinv = Slid / y
224225
assert_array_almost_equal(x.ravel(), xinv)
225226

226227

@@ -259,11 +260,13 @@ def test_Sliding3D(par):
259260
x = np.ones((par["ny"] * par["nx"] * nwins[0] * nwins[1], par["nt"]))
260261
y = Slid * x.ravel()
261262

262-
with pylops.enabled_ndarray_multiplication():
263-
xinv = Slid / y
263+
xinv = Slid / y
264264
assert_array_almost_equal(x.ravel(), xinv)
265265

266266

267+
@pytest.mark.skipif(
268+
int(os.environ.get("TEST_CUPY_PYLOPS", 0)) == 1, reason="Not CuPy enabled"
269+
)
267270
@pytest.mark.parametrize("par", [(par1), (par2), (par3), (par4)])
268271
def test_Sliding3D_simOp(par):
269272
"""Dot-test and inverse for Sliding3D operator with
@@ -296,6 +299,5 @@ def test_Sliding3D_simOp(par):
296299
x = np.ones((par["nwiny"] * par["nwinx"] * nwins[0] * nwins[1], par["nt"]))
297300
y = Slid * x.ravel()
298301

299-
with pylops.enabled_ndarray_multiplication():
300-
xinv = Slid / y
302+
xinv = Slid / y
301303
assert_array_almost_equal(x.ravel(), xinv)

0 commit comments

Comments
 (0)