Skip to content

Commit a64162f

Browse files
TST: GH24798 df.replace() with duplicate columns (#34302)
1 parent 038273f commit a64162f

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pandas/tests/frame/methods/test_replace.py

+15
Original file line numberDiff line numberDiff line change
@@ -1388,3 +1388,18 @@ def test_replace_no_replacement_dtypes(self, dtype, value):
13881388
df = pd.DataFrame(np.eye(2), dtype=dtype)
13891389
result = df.replace(to_replace=[None, -np.inf, np.inf], value=value)
13901390
tm.assert_frame_equal(result, df)
1391+
1392+
@pytest.mark.parametrize("replacement", [np.nan, 5])
1393+
def test_replace_with_duplicate_columns(self, replacement):
1394+
# GH 24798
1395+
result = pd.DataFrame({"A": [1, 2, 3], "A1": [4, 5, 6], "B": [7, 8, 9]})
1396+
result.columns = list("AAB")
1397+
1398+
expected = pd.DataFrame(
1399+
{"A": [1, 2, 3], "A1": [4, 5, 6], "B": [replacement, 8, 9]}
1400+
)
1401+
expected.columns = list("AAB")
1402+
1403+
result["B"] = result["B"].replace(7, replacement)
1404+
1405+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)