Update manipulation_tests/test_shape.py

Author: antonwolfy

dpnp/tests/third_party/cupy/manipulation_tests/test_shape.py

@@ -8,6 +8,7 @@
 
 @pytest.mark.parametrize("shape", [(2, 3), (), (4,)])
 class TestShape:
+
     def test_shape(self, shape):
         for xp in (numpy, cupy):
             a = testing.shaped_arange(shape, xp)
@@ -20,10 +21,13 @@ def test_shape_list(self, shape):
 
 
 class TestReshape:
-    def test_reshape_shapes(self):
+
+    def test_reshape_strides(self):
         def func(xp):
             a = testing.shaped_arange((1, 1, 1, 2, 2), xp)
-            return a.shape
+            if xp is cupy:
+                return tuple(el * a.itemsize for el in a.strides)
+            return a.strides
 
         assert func(numpy) == func(cupy)
 
@@ -98,15 +102,21 @@ def test_reshape_zerosize_invalid_unknown(self):
     def test_reshape_zerosize(self, xp):
         a = xp.zeros((0,))
         b = a.reshape((0,))
-        # assert b.base is a
+        if xp is cupy:
+            assert a.get_array()._pointer == b.get_array()._pointer
+        else:
+            assert b.base is a
         return b
 
     @testing.for_orders("CFA")
     @testing.numpy_cupy_array_equal(type_check=has_support_aspect64())
     def test_reshape_zerosize2(self, xp, order):
         a = xp.zeros((2, 0, 3))
         b = a.reshape((5, 0, 4), order=order)
-        # assert b.base is a
+        if xp is cupy:
+            assert a.get_array()._pointer == b.get_array()._pointer
+        else:
+            assert b.base is a
         return b
 
     @testing.for_orders("CFA")
@@ -141,6 +151,7 @@ def test_ndim_limit2(self, dtype, order):
 
 
 class TestRavel:
+
     @testing.for_orders("CFA")
     # order = 'K' is not supported currently
     @testing.numpy_cupy_array_equal()
@@ -233,6 +244,7 @@ def test_external_ravel(self, xp):
     ],
 )
 class TestReshapeOrder:
+
     def test_reshape_contiguity(self, order_init, order_reshape, shape_in_out):
         shape_init, shape_final = shape_in_out
 
@@ -247,4 +259,5 @@ def test_reshape_contiguity(self, order_init, order_reshape, shape_in_out):
         assert b_cupy.flags.f_contiguous == b_numpy.flags.f_contiguous
         assert b_cupy.flags.c_contiguous == b_numpy.flags.c_contiguous
 
+        # testing.assert_array_equal(b_cupy.strides, b_numpy.strides)
         testing.assert_array_equal(b_cupy, b_numpy)