File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change 88import pandas as pd
99from pandas .core .arrays import ExtensionArray
1010from pandas .core .dtypes .base import ExtensionDtype
11+ from pandas .core .dtypes .common import _ensure_platform_int
1112
1213
1314class 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
You can’t perform that action at this time.
0 commit comments