Skip to content

Commit 8da6027

Browse files
Revert deleted fixtures for TestChoose
1 parent 73f2ca0 commit 8da6027

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

dpnp/tests/third_party/cupy/indexing_tests/test_indexing.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -208,34 +208,39 @@ def test_choose(self, xp, dtype):
208208
c = testing.shaped_arange((3, 4), xp, dtype)
209209
return a.choose(c)
210210

211+
@pytest.mark.usefixtures("allow_fall_back_on_numpy")
211212
@testing.for_all_dtypes()
212213
@testing.numpy_cupy_array_equal()
213214
def test_choose_broadcast(self, xp, dtype):
214215
a = xp.array([[1, 0, 1], [0, 1, 0], [1, 0, 1]])
215216
c = xp.array([-10, 10]).astype(dtype)
216217
return a.choose(c)
217218

219+
@pytest.mark.usefixtures("allow_fall_back_on_numpy")
218220
@testing.for_all_dtypes()
219221
@testing.numpy_cupy_array_equal()
220222
def test_choose_broadcast2(self, xp, dtype):
221223
a = xp.array([0, 1])
222224
c = testing.shaped_arange((3, 5, 2), xp, dtype)
223225
return a.choose(c)
224226

227+
@pytest.mark.usefixtures("allow_fall_back_on_numpy")
225228
@testing.for_all_dtypes()
226229
@testing.numpy_cupy_array_equal()
227230
def test_choose_wrap(self, xp, dtype):
228231
a = xp.array([0, 3, -1, 5])
229232
c = testing.shaped_arange((3, 4), xp, dtype)
230233
return a.choose(c, mode="wrap")
231234

235+
@pytest.mark.usefixtures("allow_fall_back_on_numpy")
232236
@testing.for_all_dtypes()
233237
@testing.numpy_cupy_array_equal()
234238
def test_choose_clip(self, xp, dtype):
235239
a = xp.array([0, 3, -1, 5])
236240
c = testing.shaped_arange((3, 4), xp, dtype)
237241
return a.choose(c, mode="clip")
238242

243+
@pytest.mark.usefixtures("allow_fall_back_on_numpy")
239244
@testing.with_requires("numpy>=1.19")
240245
def test_unknown_clip(self):
241246
for xp in (numpy, cupy):
@@ -244,12 +249,14 @@ def test_unknown_clip(self):
244249
with pytest.raises(ValueError):
245250
a.choose(c, mode="unknown")
246251

252+
@pytest.mark.usefixtures("allow_fall_back_on_numpy")
247253
def test_raise(self):
248254
a = cupy.array([2])
249255
c = cupy.array([[0, 1]])
250256
with self.assertRaises(ValueError):
251257
a.choose(c)
252258

259+
@pytest.mark.usefixtures("allow_fall_back_on_numpy")
253260
@testing.for_all_dtypes()
254261
def test_choose_broadcast_fail(self, dtype):
255262
for xp in (numpy, cupy):

0 commit comments

Comments
 (0)