Skip to content

Commit 7f670c1

Browse files
authored
TST: Add test for insert/delete with MultiIndex datetime (#62238)
1 parent 2891172 commit 7f670c1

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

pandas/tests/frame/indexing/test_insert.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,9 @@
1010
from pandas import (
1111
DataFrame,
1212
Index,
13+
MultiIndex,
14+
NaT,
15+
Timestamp,
1316
)
1417
import pandas._testing as tm
1518

@@ -94,3 +97,23 @@ def test_insert_int64_loc(self):
9497
df = DataFrame({"a": [1, 2]})
9598
df.insert(np.int64(0), "b", 0)
9699
tm.assert_frame_equal(df, DataFrame({"b": [0, 0], "a": [1, 2]}))
100+
101+
def test_insert_delete_mixed_multiindex_columns(self):
102+
# GH#56853
103+
104+
df = DataFrame({("A", Timestamp("2024-01-01")): [0]})
105+
df.insert(1, "B", [1])
106+
107+
expected = DataFrame(
108+
[[0, 1]],
109+
columns=MultiIndex.from_tuples(
110+
[("A", Timestamp("2024-01-01")), ("B", NaT)]
111+
),
112+
)
113+
tm.assert_frame_equal(df, expected)
114+
115+
# Should not raise RecursionError (this was the original bug)
116+
del df["B"]
117+
118+
expected = DataFrame({("A", Timestamp("2024-01-01")): [0]})
119+
tm.assert_frame_equal(df, expected)

0 commit comments

Comments
 (0)