add a separate test_size_tuple

Author: Vahid Tavanashad

dpnp/tests/test_manipulation.py

@@ -74,26 +74,32 @@ def test_ndim():
     assert dpnp.ndim(ia) == exp
 
 
-# TODO: include commented code in the test when numpy-2.4 is released
-# @testing.with_requires("numpy>=2.4")
-def test_size():
-    a = [[1, 2, 3], [4, 5, 6]]
-    ia = dpnp.array(a)
+class TestSize:
+    def test_size(self):
+        a = [[1, 2, 3], [4, 5, 6]]
+        ia = dpnp.array(a)
 
-    exp = numpy.size(a)
-    assert ia.size == exp
-    assert dpnp.size(a) == exp
-    assert dpnp.size(ia) == exp
+        exp = numpy.size(a)
+        assert ia.size == exp
+        assert dpnp.size(a) == exp
+        assert dpnp.size(ia) == exp
 
-    exp = numpy.size(a, 0)
-    assert dpnp.size(a, 0) == exp
-    assert dpnp.size(ia, 0) == exp
+        exp = numpy.size(a, 0)
+        assert dpnp.size(a, 0) == exp
+        assert dpnp.size(ia, 0) == exp
+
+        assert dpnp.size(ia, 1) == numpy.size(a, 1)
+
+    # TODO: include commented code in the test when numpy-2.4 is released
+    # @testing.with_requires("numpy>=2.4")
+    def test_size_tuple(self):
+        a = [[1, 2, 3], [4, 5, 6]]
+        ia = dpnp.array(a)
 
-    assert dpnp.size(ia, 1) == numpy.size(a, 1)
-    assert dpnp.size(ia, ()) == 1  # numpy.size(a, ())
-    assert dpnp.size(ia, (0,)) == 2  # numpy.size(a, (0,))
-    assert dpnp.size(ia, (1,)) == 3  # numpy.size(a, (1,))
-    assert dpnp.size(ia, (0, 1)) == 6  # numpy.size(a, (0, 1))
+        assert dpnp.size(ia, ()) == 1  # numpy.size(a, ())
+        assert dpnp.size(ia, (0,)) == 2  # numpy.size(a, (0,))
+        assert dpnp.size(ia, (1,)) == 3  # numpy.size(a, (1,))
+        assert dpnp.size(ia, (0, 1)) == 6  # numpy.size(a, (0, 1))
 
 
 class TestAppend: