Commit 1385907

committed

TST: add regression test for interpolate(method='time') with Int64/Float64 dtypes (#40252)

1 parent 95d2ce0 commit 1385907Copy full SHA for 1385907

File tree

+15

-0

lines changed

+15

-0

lines changed

Lines changed: 15 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -440,3 +440,18 @@ def test_interpolate_arrow(self, dtype):`
`440`	`440`	`result = df.interpolate(limit=2)`
`441`	`441`	`expected = DataFrame({"a": [1, 1.5, 2.0, None, 3]}, dtype="float64[pyarrow]")`
`442`	`442`	`tm.assert_frame_equal(result, expected)`
	`443`	`+`
	`444`	`+ def test_interpolate_time_nullable_int_float(self):`
	`445`	`+ # GH#40252 regression test`
	`446`	`+ idx = date_range("1970年01月02日", periods=3, freq="D")`
	`447`	`+`
	`448`	`+ # Nullable Int64`
	`449`	`+ df_int = DataFrame({"a": [1, None, 2]}, index=idx, dtype="Int64")`
	`450`	`+ result_int = df_int.interpolate(method="time")`
	`451`	`+ expected = DataFrame({"a": [1.0, 1.5, 2.0]}, index=idx, dtype="Float64")`
	`452`	`+ tm.assert_frame_equal(result_int, expected)`
	`453`	`+`
	`454`	`+ # Nullable Float64`
	`455`	`+ df_float = df_int.astype("Float64")`
	`456`	`+ result_float = df_float.interpolate(method="time")`
	`457`	`+ tm.assert_frame_equal(result_float, expected)`

Comments

(0)