test: add for allow duplicate labels false

Author: kubaszkliniarz

pandas/tests/frame/methods/test_rename_axis.py

@@ -28,6 +28,20 @@ def test_rename_axis_inplace(self, float_frame):
         assert no_return is None
         tm.assert_frame_equal(result, expected)
 
+    def test_rename_axis_with_allows_duplicate_labels_false(self):
+        # GH#44958
+        # Test that rename_axis() works correctly when allows_duplicate_labels=False
+        df = DataFrame([[1, 2], [3, 4]], columns=["a", "b"]).set_flags(
+            allows_duplicate_labels=False
+        )
+
+        # Test renaming index (axis=0) with allows_duplicate_labels=False
+        result = df.rename_axis("idx", axis=0)
+        expected = DataFrame(
+            [[1, 2], [3, 4]], index=Index([0, 1], name="idx"), columns=["a", "b"]
+        ).set_flags(allows_duplicate_labels=False)
+        tm.assert_frame_equal(result, expected)
+
     def test_rename_axis_raises(self):
         # GH#17833
         df = DataFrame({"A": [1, 2], "B": [1, 2]})

pandas/tests/frame/methods/test_set_axis.py

@@ -112,6 +112,20 @@ def obj(self):
         )
         return df
 
+    def test_set_axis_with_allows_duplicate_labels_false(self):
+        # GH#44958
+        # Test that set_axis() works correctly when allows_duplicate_labels=False
+        df = DataFrame([[1, 2], [3, 4]], columns=["a", "b"]).set_flags(
+            allows_duplicate_labels=False
+        )
+
+        # Test setting index with allows_duplicate_labels=False
+        result = df.set_axis(labels=["x", "y"], axis=0)
+        expected = DataFrame(
+            [[1, 2], [3, 4]], index=["x", "y"], columns=["a", "b"]
+        ).set_flags(allows_duplicate_labels=False)
+        tm.assert_frame_equal(result, expected)
+
 
 class TestSeriesSetAxis(SharedSetAxisTests):
     @pytest.fixture