diff --git a/tests/test_frame.py b/tests/test_frame.py index 5ebed2bc2..6bbe7e3b7 100644 --- a/tests/test_frame.py +++ b/tests/test_frame.py @@ -2550,11 +2550,16 @@ def test_types_rename() -> None: # Apparently all of these calls are accepted by pandas check(assert_type(df.rename(columns={None: "b"}), pd.DataFrame), pd.DataFrame) check(assert_type(df.rename(columns={"": "b"}), pd.DataFrame), pd.DataFrame) - check(assert_type(df.rename(columns={(2, 1): "b"}), pd.DataFrame), pd.DataFrame) check( assert_type(df.rename(columns=lambda s: s.upper()), pd.DataFrame), pd.DataFrame ) + df_multiindex = pd.DataFrame(columns=[("a", 1), ("a", 2)]) + check( + assert_type(df_multiindex.rename(columns={(1, 2): ("b", "a")}), pd.DataFrame), + pd.DataFrame, + ) + def test_types_rename_axis() -> None: df = pd.DataFrame({"col_name": [1, 2, 3]})