Update test for dpnp.rot90

IntelPython · Jan 20, 2025 · 61d61b4 · 61d61b4
1 parent a95e8f6
commit 61d61b4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/dpnp/tests/test_manipulation.py b/dpnp/tests/test_manipulation.py
@@ -15,6 +15,7 @@
 from .helper import (
     assert_dtype_allclose,
     get_all_dtypes,
+    get_array,
     get_complex_dtypes,
     get_float_complex_dtypes,
     get_float_dtypes,
@@ -1232,7 +1233,10 @@ def test_axes(self):
     def test_axes_type(self, axes):
         a = numpy.ones((50, 40, 3))
         ia = dpnp.array(a)
-        assert_equal(dpnp.rot90(ia, axes=axes), numpy.rot90(a, axes=axes))
+        assert_equal(
+            dpnp.rot90(ia, axes=axes),
+            numpy.rot90(a, axes=get_array(numpy, axes)),
+        )
 
     def test_rotation_axes(self):
         a = numpy.arange(8).reshape((2, 2, 2))