diff --git a/pandas/tests/extension/base/methods.py b/pandas/tests/extension/base/methods.py index 0e6d4892f4239..9b154a8afeabc 100644 --- a/pandas/tests/extension/base/methods.py +++ b/pandas/tests/extension/base/methods.py @@ -50,6 +50,9 @@ def test_argsort(self, data_for_sorting): def test_argsort_missing_array(self, data_missing_for_sorting): result = data_missing_for_sorting.argsort() expected = np.array([2, 0, 1], dtype=np.dtype("int")) + # we don't care whether it's int32 or int64 + result = result.astype("int64", casting="safe") + expected = expected.astype("int64", casting="safe") tm.assert_numpy_array_equal(result, expected) def test_argsort_missing(self, data_missing_for_sorting):