fix platform dtype

pandas-dev · Apr 24, 2021 · d8b1052 · d8b1052
1 parent 929cc37
commit d8b1052
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pandas/tests/test_algos.py b/pandas/tests/test_algos.py
@@ -98,14 +98,14 @@ def test_basic(self):
         codes, uniques = algos.factorize(list(reversed(range(5))))
         exp = np.array([0, 1, 2, 3, 4], dtype=np.intp)
         tm.assert_numpy_array_equal(codes, exp)
-        exp = np.array([4, 3, 2, 1, 0], dtype=np.int32)
+        exp = np.array([4, 3, 2, 1, 0], dtype=np.intp)
         tm.assert_numpy_array_equal(uniques, exp)
 
         codes, uniques = algos.factorize(list(reversed(range(5))), sort=True)
 
         exp = np.array([4, 3, 2, 1, 0], dtype=np.intp)
         tm.assert_numpy_array_equal(codes, exp)
-        exp = np.array([0, 1, 2, 3, 4], dtype=np.int32)
+        exp = np.array([0, 1, 2, 3, 4], dtype=np.intp)
         tm.assert_numpy_array_equal(uniques, exp)
 
         codes, uniques = algos.factorize(list(reversed(np.arange(5.0))))