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():
12431243 result = libalgos .groupsort_indexer (a , 1000 )[0 ]
12441244
12451245 # need to use a stable sort
1246+ # np.argsort returns int, groupsort_indexer
1247+ # always returns int64
12461248 expected = np .argsort (a , kind = 'mergesort' )
1249+ expected = expected .astype (np .int64 )
1250+
12471251 tm .assert_numpy_array_equal (result , expected )
12481252
12491253 # compare with lexsort
1254+ # np.lexsort returns int, groupsort_indexer
1255+ # always returns int64
12501256 key = a * 1000 + b
12511257 result = libalgos .groupsort_indexer (key , 1000000 )[0 ]
12521258 expected = np .lexsort ((b , a ))
1259+ expected = expected .astype (np .int64 )
1260+
12531261 tm .assert_numpy_array_equal (result , expected )
12541262
12551263
You can’t perform that action at this time.
0 commit comments