Skip to content

Commit e97be6f

Browse files
authored
COMPAT: fixup decimal extension for indexing compat (#19882)
1 parent feedf66 commit e97be6f

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

pandas/tests/extension/base/getitem.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@ def test_iloc_series(self, data):
1818
self.assert_series_equal(result, expected)
1919

2020
def test_iloc_frame(self, data):
21-
df = pd.DataFrame({"A": data, 'B': np.arange(len(data))})
21+
df = pd.DataFrame({"A": data, 'B':
22+
np.arange(len(data), dtype='int64')})
2223
expected = pd.DataFrame({"A": data[:4]})
2324

2425
# slice -> frame
@@ -49,7 +50,8 @@ def test_loc_series(self, data):
4950
self.assert_series_equal(result, expected)
5051

5152
def test_loc_frame(self, data):
52-
df = pd.DataFrame({"A": data, 'B': np.arange(len(data))})
53+
df = pd.DataFrame({"A": data,
54+
'B': np.arange(len(data), dtype='int64')})
5355
expected = pd.DataFrame({"A": data[:4]})
5456

5557
# slice -> frame

pandas/tests/extension/decimal/array.py

+2
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
import pandas as pd
99
from pandas.core.arrays import ExtensionArray
1010
from pandas.core.dtypes.base import ExtensionDtype
11+
from pandas.core.dtypes.common import _ensure_platform_int
1112

1213

1314
class DecimalDtype(ExtensionDtype):
@@ -68,6 +69,7 @@ def isna(self):
6869
def take(self, indexer, allow_fill=True, fill_value=None):
6970
mask = indexer == -1
7071

72+
indexer = _ensure_platform_int(indexer)
7173
out = self.values.take(indexer)
7274
out[mask] = self._na_value
7375

0 commit comments

Comments
 (0)