@@ -813,18 +813,18 @@ def test_interp_inplace(self):
813
813
814
814
result = df .copy ()
815
815
result ['a' ].interpolate (inplace = True , downcast = 'infer' )
816
- assert_frame_equal (result , expected .astype ('int ' ))
816
+ assert_frame_equal (result , expected .astype ('int64 ' ))
817
817
818
818
def test_interp_ignore_all_good (self ):
819
819
# GH
820
820
df = DataFrame ({'A' : [1 , 2 , np .nan , 4 ],
821
821
'B' : [1 , 2 , 3 , 4 ],
822
822
'C' : [1. , 2. , np .nan , 4. ],
823
823
'D' : [1. , 2. , 3. , 4. ]})
824
- expected = DataFrame ({'A' : np .array ([1 , 2 , 3 , 4 ], dtype = 'float ' ),
825
- 'B' : np .array ([1 , 2 , 3 , 4 ], dtype = 'int ' ),
826
- 'C' : np .array ([1. , 2. , 3 , 4. ], dtype = 'float ' ),
827
- 'D' : np .array ([1. , 2. , 3. , 4. ], dtype = 'float ' )})
824
+ expected = DataFrame ({'A' : np .array ([1 , 2 , 3 , 4 ], dtype = 'float64 ' ),
825
+ 'B' : np .array ([1 , 2 , 3 , 4 ], dtype = 'int64 ' ),
826
+ 'C' : np .array ([1. , 2. , 3 , 4. ], dtype = 'float64 ' ),
827
+ 'D' : np .array ([1. , 2. , 3. , 4. ], dtype = 'float64 ' )})
828
828
829
829
result = df .interpolate (downcast = None )
830
830
assert_frame_equal (result , expected )
0 commit comments