Skip to content

Commit d4f4b85

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

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pandas/tests/frame/methods/test_interpolate.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -440,3 +440,18 @@ 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+
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)

0 commit comments

Comments
 (0)