File tree 2 files changed +6
-2
lines changed
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):
18
18
self .assert_series_equal (result , expected )
19
19
20
20
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' )})
22
23
expected = pd .DataFrame ({"A" : data [:4 ]})
23
24
24
25
# slice -> frame
@@ -49,7 +50,8 @@ def test_loc_series(self, data):
49
50
self .assert_series_equal (result , expected )
50
51
51
52
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' )})
53
55
expected = pd .DataFrame ({"A" : data [:4 ]})
54
56
55
57
# slice -> frame
Original file line number Diff line number Diff line change 8
8
import pandas as pd
9
9
from pandas .core .arrays import ExtensionArray
10
10
from pandas .core .dtypes .base import ExtensionDtype
11
+ from pandas .core .dtypes .common import _ensure_platform_int
11
12
12
13
13
14
class DecimalDtype (ExtensionDtype ):
@@ -68,6 +69,7 @@ def isna(self):
68
69
def take (self , indexer , allow_fill = True , fill_value = None ):
69
70
mask = indexer == - 1
70
71
72
+ indexer = _ensure_platform_int (indexer )
71
73
out = self .values .take (indexer )
72
74
out [mask ] = self ._na_value
73
75
You can’t perform that action at this time.
0 commit comments