diff --git a/dpnp/tests/third_party/cupy/core_tests/test_ndarray_adv_indexing.py b/dpnp/tests/third_party/cupy/core_tests/test_ndarray_adv_indexing.py index 68a4a68282de..6a635ee53b2c 100644 --- a/dpnp/tests/third_party/cupy/core_tests/test_ndarray_adv_indexing.py +++ b/dpnp/tests/third_party/cupy/core_tests/test_ndarray_adv_indexing.py @@ -751,7 +751,7 @@ class TestArrayAdvancedIndexingSetitemScalarValueIndexError: def test_adv_setitem(self): for xp in (numpy, cupy): a = xp.zeros(self.shape) - with pytest.raises(IndexError): + with pytest.raises((IndexError, ValueError)): a[self.indexes] = self.value diff --git a/dpnp/tests/third_party/cupy/testing/_loops.py b/dpnp/tests/third_party/cupy/testing/_loops.py index f4d7f7a374da..f57e555ea363 100644 --- a/dpnp/tests/third_party/cupy/testing/_loops.py +++ b/dpnp/tests/third_party/cupy/testing/_loops.py @@ -408,9 +408,7 @@ def test_func(*args, **kw): if cupy_r.shape == (): skip = (mask == 0).all() else: - # mask is numpy.ndarray here which is not supported now - # TODO remove asarray() once dpctl-2053 is addressed - cupy_r = cupy_r[cupy.asarray(mask)].asnumpy() + cupy_r = cupy_r[mask].asnumpy() numpy_r = numpy_r[mask] if not skip: