Skip to content

Commit 3e6504c

Browse files
committed
Fixes descending logic for test_argsort
1 parent cad86ef commit 3e6504c

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

array_api_tests/test_sorting_functions.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,12 +62,14 @@ def test_argsort(x, data):
6262
scalar_type = dh.get_scalar_type(x.dtype)
6363
for indices in sh.axes_ndindex(x.shape, axes):
6464
elements = [scalar_type(x[idx]) for idx in indices]
65-
orders = sorted(range(len(elements)), key=elements.__getitem__)
66-
if kw.get("descending", False):
67-
orders = reversed(orders)
65+
orders = sorted(
66+
range(len(elements)),
67+
key=elements.__getitem__,
68+
reverse=kw.get("descending", False),
69+
)
6870
if kw.get("stable", True):
6971
for idx, o in zip(indices, orders):
70-
ph.assert_scalar_equals("argsort", int, idx, int(out[idx]), o)
72+
ph.assert_scalar_equals("argsort", int, idx, int(out[idx]), o, **kw)
7173
else:
7274
idx_elements = dict(zip(indices, elements))
7375
idx_orders = dict(zip(indices, orders))

0 commit comments

Comments
 (0)