Skip to content

Commit 61d61b4

Browse files
committed
Update test for dpnp.rot90
1 parent a95e8f6 commit 61d61b4

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

dpnp/tests/test_manipulation.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from .helper import (
1616
assert_dtype_allclose,
1717
get_all_dtypes,
18+
get_array,
1819
get_complex_dtypes,
1920
get_float_complex_dtypes,
2021
get_float_dtypes,
@@ -1232,7 +1233,10 @@ def test_axes(self):
12321233
def test_axes_type(self, axes):
12331234
a = numpy.ones((50, 40, 3))
12341235
ia = dpnp.array(a)
1235-
assert_equal(dpnp.rot90(ia, axes=axes), numpy.rot90(a, axes=axes))
1236+
assert_equal(
1237+
dpnp.rot90(ia, axes=axes),
1238+
numpy.rot90(a, axes=get_array(numpy, axes)),
1239+
)
12361240

12371241
def test_rotation_axes(self):
12381242
a = numpy.arange(8).reshape((2, 2, 2))

0 commit comments

Comments
 (0)