Skip to content

Commit 5a006d3

Browse files
jrebackTomAugspurger
authored andcommitted
COMPAT: compare platform return on 32-bit (#18090)
xref #18047 (cherry picked from commit 27bbea7)
1 parent 27794fa commit 5a006d3

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

pandas/tests/test_algos.py

+8
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)