Skip to content

Commit 22da747

Browse files
committed
update test_sort.py
1 parent e9a25dd commit 22da747

File tree

2 files changed

+110
-97
lines changed

2 files changed

+110
-97
lines changed

dpnp/dpnp_iface_sorting.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -64,13 +64,18 @@ def _wrap_sort_argsort(
6464

6565
if order is not None:
6666
raise NotImplementedError(
67-
"order keyword argument is only supported with its default value."
68-
)
69-
if kind is not None and stable is not None:
70-
raise ValueError(
71-
"`kind` and `stable` parameters can't be provided at the same time."
72-
" Use only one of them."
67+
"`order` keyword argument is only supported with its default value."
7368
)
69+
if stable is not None:
70+
if stable not in [True, False]:
71+
raise ValueError(
72+
"`stable` parameter should be None, True, or False."
73+
)
74+
if kind is not None:
75+
raise ValueError(
76+
"`kind` and `stable` parameters can't be provided at"
77+
" the same time. Use only one of them."
78+
)
7479

7580
usm_a = dpnp.get_usm_ndarray(a)
7681
if axis is None:

0 commit comments

Comments
 (0)