File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change @@ -1243,13 +1243,21 @@ def test_groupsort_indexer():
1243
1243
result = libalgos .groupsort_indexer (a , 1000 )[0 ]
1244
1244
1245
1245
# need to use a stable sort
1246
+ # np.argsort returns int, groupsort_indexer
1247
+ # always returns int64
1246
1248
expected = np .argsort (a , kind = 'mergesort' )
1249
+ expected = expected .astype (np .int64 )
1250
+
1247
1251
tm .assert_numpy_array_equal (result , expected )
1248
1252
1249
1253
# compare with lexsort
1254
+ # np.lexsort returns int, groupsort_indexer
1255
+ # always returns int64
1250
1256
key = a * 1000 + b
1251
1257
result = libalgos .groupsort_indexer (key , 1000000 )[0 ]
1252
1258
expected = np .lexsort ((b , a ))
1259
+ expected = expected .astype (np .int64 )
1260
+
1253
1261
tm .assert_numpy_array_equal (result , expected )
1254
1262
1255
1263
You can’t perform that action at this time.
0 commit comments