Skip to content

Commit 585a220

Browse files
TST: DataFrame.interpolate axis name for axis argument (pandas-dev#34132)
1 parent 8224843 commit 585a220

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

pandas/tests/frame/methods/test_interpolate.py

+12
Original file line numberDiff line numberDiff line change
@@ -284,3 +284,15 @@ def test_interp_time_inplace_axis(self, axis):
284284
result = expected.interpolate(axis=0, method="time")
285285
expected.interpolate(axis=0, method="time", inplace=True)
286286
tm.assert_frame_equal(result, expected)
287+
288+
@pytest.mark.parametrize("axis_name, axis_number", [("index", 0), ("columns", 1)])
289+
def test_interp_string_axis(self, axis_name, axis_number):
290+
# https://github.com/pandas-dev/pandas/issues/25190
291+
x = np.linspace(0, 100, 1000)
292+
y = np.sin(x)
293+
df = DataFrame(
294+
data=np.tile(y, (10, 1)), index=np.arange(10), columns=x
295+
).reindex(columns=x * 1.005)
296+
result = df.interpolate(method="linear", axis=axis_name)
297+
expected = df.interpolate(method="linear", axis=axis_number)
298+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)