Skip to content

Commit ff8d694

Browse files
TST: df.set_axis() and rename_axis() not failing when allows_duplicate_labels=False (#63135)
1 parent 9048936 commit ff8d694

File tree

2 files changed

+22
-0
lines changed

2 files changed

+22
-0
lines changed

pandas/tests/frame/methods/test_rename_axis.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,18 @@ def test_rename_axis_inplace(self, float_frame):
2828
assert no_return is None
2929
tm.assert_frame_equal(result, expected)
3030

31+
def test_rename_axis_with_allows_duplicate_labels_false(self):
32+
# GH#44958
33+
df = DataFrame([[1, 2], [3, 4]], columns=["a", "b"]).set_flags(
34+
allows_duplicate_labels=False
35+
)
36+
37+
result = df.rename_axis("idx", axis=0)
38+
expected = DataFrame(
39+
[[1, 2], [3, 4]], index=Index([0, 1], name="idx"), columns=["a", "b"]
40+
)
41+
tm.assert_frame_equal(result, expected, check_flags=False)
42+
3143
def test_rename_axis_raises(self):
3244
# GH#17833
3345
df = DataFrame({"A": [1, 2], "B": [1, 2]})

pandas/tests/frame/methods/test_set_axis.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,16 @@ def obj(self):
112112
)
113113
return df
114114

115+
def test_set_axis_with_allows_duplicate_labels_false(self):
116+
# GH#44958
117+
df = DataFrame([[1, 2], [3, 4]], columns=["a", "b"]).set_flags(
118+
allows_duplicate_labels=False
119+
)
120+
121+
result = df.set_axis(labels=["x", "y"], axis=0)
122+
expected = DataFrame([[1, 2], [3, 4]], index=["x", "y"], columns=["a", "b"])
123+
tm.assert_frame_equal(result, expected, check_flags=False)
124+
115125

116126
class TestSeriesSetAxis(SharedSetAxisTests):
117127
@pytest.fixture

0 commit comments

Comments
 (0)