Skip to content

Commit c0fa1e4

Browse files
committed
Merge branch 'main' of github.com:pandas-dev/pandas-stubs into feature/python314
2 parents 67ca191 + 72138f7 commit c0fa1e4

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/test_frame.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2553,11 +2553,16 @@ def test_types_rename() -> None:
25532553
# Apparently all of these calls are accepted by pandas
25542554
check(assert_type(df.rename(columns={None: "b"}), pd.DataFrame), pd.DataFrame)
25552555
check(assert_type(df.rename(columns={"": "b"}), pd.DataFrame), pd.DataFrame)
2556-
check(assert_type(df.rename(columns={(2, 1): "b"}), pd.DataFrame), pd.DataFrame)
25572556
check(
25582557
assert_type(df.rename(columns=lambda s: s.upper()), pd.DataFrame), pd.DataFrame
25592558
)
25602559

2560+
df_multiindex = pd.DataFrame(columns=[("a", 1), ("a", 2)])
2561+
check(
2562+
assert_type(df_multiindex.rename(columns={(1, 2): ("b", "a")}), pd.DataFrame),
2563+
pd.DataFrame,
2564+
)
2565+
25612566

25622567
def test_types_rename_axis() -> None:
25632568
df = pd.DataFrame({"col_name": [1, 2, 3]})

0 commit comments

Comments
 (0)