add new tests

Author: Vahid Tavanashad

tests/test_sort.py

@@ -4,6 +4,7 @@
 from numpy.testing import assert_array_equal, assert_equal, assert_raises
 
 import dpnp
+from tests.third_party.cupy import testing
 
 from .helper import (
     assert_dtype_allclose,
@@ -61,14 +62,26 @@ def test_argsort_ndarray(self, dtype, axis):
         expected = np_array.argsort(axis=axis)
         assert_dtype_allclose(result, expected)
 
-    def test_argsort_stable(self):
+    @pytest.mark.parametrize("kind", [None, "stable"])
+    def test_sort_kind(self, kind):
         np_array = numpy.repeat(numpy.arange(10), 10)
         dp_array = dpnp.array(np_array)
 
-        result = dpnp.argsort(dp_array, kind="stable")
+        result = dpnp.argsort(dp_array, kind=kind)
         expected = numpy.argsort(np_array, kind="stable")
         assert_dtype_allclose(result, expected)
 
+    # `stable` keyword is supported in numpy 2.0 and above
+    @testing.with_requires("numpy>=2.0")
+    @pytest.mark.parametrize("stable", [None, False, True])
+    def test_sort_stable(self, stable):
+        np_array = numpy.repeat(numpy.arange(10), 10)
+        dp_array = dpnp.array(np_array)
+
+        result = dpnp.argsort(dp_array, stable="stable")
+        expected = numpy.argsort(np_array, stable=True)
+        assert_dtype_allclose(result, expected)
+
     def test_argsort_zero_dim(self):
         np_array = numpy.array(2.5)
         dp_array = dpnp.array(np_array)
@@ -295,14 +308,26 @@ def test_sort_ndarray(self, dtype, axis):
         np_array.sort(axis=axis)
         assert_dtype_allclose(dp_array, np_array)
 
-    def test_sort_stable(self):
+    @pytest.mark.parametrize("kind", [None, "stable"])
+    def test_sort_kind(self, kind):
         np_array = numpy.repeat(numpy.arange(10), 10)
         dp_array = dpnp.array(np_array)
 
-        result = dpnp.sort(dp_array, kind="stable")
+        result = dpnp.sort(dp_array, kind=kind)
         expected = numpy.sort(np_array, kind="stable")
         assert_dtype_allclose(result, expected)
 
+    # `stable` keyword is supported in numpy 2.0 and above
+    @testing.with_requires("numpy>=2.0")
+    @pytest.mark.parametrize("stable", [None, False, True])
+    def test_sort_stable(self, stable):
+        np_array = numpy.repeat(numpy.arange(10), 10)
+        dp_array = dpnp.array(np_array)
+
+        result = dpnp.sort(dp_array, stable="stable")
+        expected = numpy.sort(np_array, stable=True)
+        assert_dtype_allclose(result, expected)
+
     def test_sort_ndarray_axis_none(self):
         a = numpy.random.uniform(-10, 10, 12)
         dp_array = dpnp.array(a).reshape(6, 2)