1212 backend = "numpy"
1313import pytest
1414
15- import pylops
1615from pylops .basicoperators import Identity , MatrixMult
1716from pylops .signalprocessing import Sliding1D , Sliding2D , Sliding3D
1817from 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 )])
143144def 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 )])
200202def 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 )])
268271def 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