We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b4375bd commit 27bbea7Copy full SHA for 27bbea7
pandas/tests/test_algos.py
@@ -1243,13 +1243,21 @@ def test_groupsort_indexer():
1243
result = libalgos.groupsort_indexer(a, 1000)[0]
1244
1245
# need to use a stable sort
1246
+ # np.argsort returns int, groupsort_indexer
1247
+ # always returns int64
1248
expected = np.argsort(a, kind='mergesort')
1249
+ expected = expected.astype(np.int64)
1250
+
1251
tm.assert_numpy_array_equal(result, expected)
1252
1253
# compare with lexsort
1254
+ # np.lexsort returns int, groupsort_indexer
1255
1256
key = a * 1000 + b
1257
result = libalgos.groupsort_indexer(key, 1000000)[0]
1258
expected = np.lexsort((b, a))
1259
1260
1261
1262
1263
0 commit comments