Skip to content

Commit 72138f7

Browse files
Fix nightly with pd.DataFrame.rename calling for MultiIndex.fillna (#1449)
* Fix nightly with pd.DataFrame.rename calling for MultiIndex.fillna * Run pre-commit
1 parent 5e16b86 commit 72138f7

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
@@ -2550,11 +2550,16 @@ def test_types_rename() -> None:
25502550
# Apparently all of these calls are accepted by pandas
25512551
check(assert_type(df.rename(columns={None: "b"}), pd.DataFrame), pd.DataFrame)
25522552
check(assert_type(df.rename(columns={"": "b"}), pd.DataFrame), pd.DataFrame)
2553-
check(assert_type(df.rename(columns={(2, 1): "b"}), pd.DataFrame), pd.DataFrame)
25542553
check(
25552554
assert_type(df.rename(columns=lambda s: s.upper()), pd.DataFrame), pd.DataFrame
25562555
)
25572556

2557+
df_multiindex = pd.DataFrame(columns=[("a", 1), ("a", 2)])
2558+
check(
2559+
assert_type(df_multiindex.rename(columns={(1, 2): ("b", "a")}), pd.DataFrame),
2560+
pd.DataFrame,
2561+
)
2562+
25582563

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

0 commit comments

Comments
 (0)