@@ -100,9 +100,9 @@ def test_top_k_1d_largest(dtype, n, skip_known_failues_on_cpu):
100
100
assert s .values .shape == (k ,)
101
101
assert s .values .dtype == inp .dtype
102
102
assert s .indices .shape == (k ,)
103
- assert dpt .all (s .indices == expected_inds )
104
103
assert dpt .all (s .values == dpt .ones (k , dtype = dtype )), s .values
105
104
assert dpt .all (s .values == inp [s .indices ]), s .indices
105
+ assert dpt .all (s .indices == expected_inds ), (s .indices , expected_inds )
106
106
107
107
108
108
def _expected_smallest_inds (inp , n , shift , k ):
@@ -173,9 +173,9 @@ def test_top_k_1d_smallest(dtype, n, skip_known_failues_on_cpu):
173
173
assert s .values .shape == (k ,)
174
174
assert s .values .dtype == inp .dtype
175
175
assert s .indices .shape == (k ,)
176
- assert dpt .all (s .indices == expected_inds )
177
176
assert dpt .all (s .values == dpt .zeros (k , dtype = dtype )), s .values
178
177
assert dpt .all (s .values == inp [s .indices ]), s .indices
178
+ assert dpt .all (s .indices == expected_inds ), (s .indices , expected_inds )
179
179
180
180
181
181
@pytest .mark .parametrize (
0 commit comments