Fix a bug in test_search.py::TestWhere::test_ndim

IntelPython · Jan 20, 2025 · d449db8 · d449db8
1 parent f77d8de
commit d449db8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dpnp/tests/test_search.py b/dpnp/tests/test_search.py
@@ -272,7 +272,7 @@ def test_ndim(self):
         assert_array_equal(np_res, dpnp_res)
 
         np_res = numpy.where(c, a.T, b.T)
-        dpnp_res = numpy.where(ic, ia.T, ib.T)
+        dpnp_res = dpnp.where(ic, ia.T, ib.T)
         assert_array_equal(np_res, dpnp_res)
 
     def test_dtype_mix(self):