Skip to content

Commit ea515bf

Browse files
committed
TST: add regression test for interpolate(method='time') with Int64/Float64 dtypes (#40252)
1 parent 1028791 commit ea515bf

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

pandas/tests/frame/methods/test_interpolate.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -440,3 +440,19 @@ def test_interpolate_arrow(self, dtype):
440440
result = df.interpolate(limit=2)
441441
expected = DataFrame({"a": [1, 1.5, 2.0, None, 3]}, dtype="float64[pyarrow]")
442442
tm.assert_frame_equal(result, expected)
443+
444+
@pytest.mark.parametrize(
445+
"dtype",
446+
[
447+
"Int64",
448+
"Float64",
449+
],
450+
)
451+
def test_interpolate_time_nullable_int_float(self, dtype):
452+
# GH#40252
453+
idx = date_range("1970-01-02", periods=3, freq="D")
454+
455+
df = DataFrame({"a": [1, None, 2]}, index=idx, dtype=dtype)
456+
result = df.interpolate(method="time")
457+
expected = DataFrame({"a": [1.0, 1.5, 2.0]}, index=idx, dtype="Float64")
458+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)