add support for more option for kind keyword arguments

Author: Vahid Tavanashad

dpnp/dpnp_iface_sorting.py

@@ -66,9 +66,10 @@ def _wrap_sort_argsort(
         raise NotImplementedError(
             "order keyword argument is only supported with its default value."
         )
-    if kind is not None and kind != "stable":
-        raise NotImplementedError(
-            "kind keyword argument can only be None or 'stable'."
+    if kind is not None and stable is not None:
+        raise ValueError(
+            "`kind` and `stable` parameters can't be provided at the same time."
+            " Use only one of them."
         )
 
     usm_a = dpnp.get_usm_ndarray(a)
@@ -77,11 +78,11 @@ def _wrap_sort_argsort(
         axis = -1
 
     axis = normalize_axis_index(axis, ndim=usm_a.ndim)
-    usm_res = _sorting_fn(usm_a, axis=axis, stable=stable)
+    usm_res = _sorting_fn(usm_a, axis=axis, stable=stable, kind=kind)
     return dpnp_array._create_from_usm_ndarray(usm_res)
 
 
-def argsort(a, axis=-1, kind=None, order=None, *, stable=True):
+def argsort(a, axis=-1, kind=None, order=None, *, stable=None):
     """
     Returns the indices that would sort an array.
 
@@ -94,9 +95,9 @@ def argsort(a, axis=-1, kind=None, order=None, *, stable=True):
     axis : {None, int}, optional
         Axis along which to sort. If ``None``, the array is flattened before
         sorting. The default is ``-1``, which sorts along the last axis.
-    kind : {None, "stable"}, optional
+    kind : {None, "stable", "mergesort", "radixsort"}, optional
         Sorting algorithm. Default is ``None``, which is equivalent to
-         ``"stable"``. Unlike NumPy, no other option is accepted here.
+         ``"stable"``.
     stable : {None, bool}, optional
         Sort stability. If ``True``, the returned array will maintain
         the relative order of ``a`` values which compare as equal.
@@ -121,8 +122,9 @@ def argsort(a, axis=-1, kind=None, order=None, *, stable=True):
     Limitations
     -----------
     Parameters `order` is only supported with its default value.
-    Parameter `kind` can only be ``None`` or ``"stable"`` which are equivalent.
     Otherwise ``NotImplementedError`` exception will be raised.
+    Sorting algorithms ``"quicksort"`` and ``"heapsort"`` are not supported.
+
 
     See Also
     --------
@@ -203,7 +205,7 @@ def partition(x1, kth, axis=-1, kind="introselect", order=None):
     return call_origin(numpy.partition, x1, kth, axis, kind, order)
 
 
-def sort(a, axis=-1, kind=None, order=None, *, stable=True):
+def sort(a, axis=-1, kind=None, order=None, *, stable=None):
     """
     Return a sorted copy of an array.
 
@@ -216,9 +218,9 @@ def sort(a, axis=-1, kind=None, order=None, *, stable=True):
     axis : {None, int}, optional
         Axis along which to sort. If ``None``, the array is flattened before
         sorting. The default is ``-1``, which sorts along the last axis.
-    kind : {None, "stable"}, optional
+    kind : {None, "stable", "mergesort", "radixsort"}, optional
         Sorting algorithm. Default is ``None``, which is equivalent to
-        ``"stable"``. Unlike NumPy, no other option is accepted here.
+        ``"stable"``.
     stable : {None, bool}, optional
         Sort stability. If ``True``, the returned array will maintain
         the relative order of ``a`` values which compare as equal.
@@ -239,8 +241,8 @@ def sort(a, axis=-1, kind=None, order=None, *, stable=True):
     Limitations
     -----------
     Parameters `order` is only supported with its default value.
-    Parameter `kind` can only be ``None`` or ``"stable"`` which are equivalent.
     Otherwise ``NotImplementedError`` exception will be raised.
+    Sorting algorithms ``"quicksort"`` and ``"heapsort"`` are not supported.
 
     See Also
     --------

tests/test_sort.py

@@ -62,7 +62,7 @@ def test_argsort_ndarray(self, dtype, axis):
         expected = np_array.argsort(axis=axis)
         assert_dtype_allclose(result, expected)
 
-    @pytest.mark.parametrize("kind", [None, "stable"])
+    @pytest.mark.parametrize("kind", [None, "stable", "mergesort", "radixsort"])
     def test_sort_kind(self, kind):
         np_array = numpy.repeat(numpy.arange(10), 10)
         dp_array = dpnp.array(np_array)
@@ -308,7 +308,7 @@ def test_sort_ndarray(self, dtype, axis):
         np_array.sort(axis=axis)
         assert_dtype_allclose(dp_array, np_array)
 
-    @pytest.mark.parametrize("kind", [None, "stable"])
+    @pytest.mark.parametrize("kind", [None, "stable", "mergesort", "radixsort"])
     def test_sort_kind(self, kind):
         np_array = numpy.repeat(numpy.arange(10), 10)
         dp_array = dpnp.array(np_array)
@@ -347,15 +347,20 @@ def test_sort_zero_dim(self):
         expected = numpy.sort(np_array, axis=None)
         assert_dtype_allclose(result, expected)
 
-    def test_sort_notimplemented(self):
+    def test_sort_error(self):
         dp_array = dpnp.arange(10)
 
-        with pytest.raises(NotImplementedError):
+        # quicksort is currently not supported
+        with pytest.raises(ValueError):
             dpnp.sort(dp_array, kind="quicksort")
 
         with pytest.raises(NotImplementedError):
             dpnp.sort(dp_array, order=["age"])
 
+        # both kind and stable are given
+        with pytest.raises(ValueError):
+            dpnp.sort(dp_array, kind="mergesort", stable=True)
+
 
 class TestSortComplex:
     @pytest.mark.parametrize(