Skip to content

Commit d1c651d

Browse files
committed
Small refactor in merge tests
1 parent 8fa3cdb commit d1c651d

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

pandas/tests/reshape/merge/test_merge.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2382,11 +2382,14 @@ def test_merge_suffix_with_force_simple(force_suffixes):
23822382
})
23832383

23842384
if force_suffixes:
2385-
expected = DataFrame([[2, 2, "B", "D"], [3, 3, "C", "E"]], columns=["ID_left", "Value_left", "ID_right", "Value_right"])
2385+
expected = DataFrame([[2, 2, "B", "D"], [3, 3, "C", "E"]],
2386+
columns=["ID_left", "Value_left", "ID_right", "Value_right"])
23862387
else:
2387-
expected = DataFrame([[2, "B", "D"], [3, "C", "E"]], columns=["ID", "Value_left", "Value_right"])
2388+
expected = DataFrame([[2, "B", "D"], [3, "C", "E"]],
2389+
columns=["ID", "Value_left", "Value_right"])
23882390

2389-
result = merge(df1, df2, on="ID", suffixes=("_left", "_right"), force_suffixes=force_suffixes)
2391+
result = merge(df1, df2, on="ID", suffixes=("_left", "_right"),
2392+
force_suffixes=force_suffixes)
23902393
tm.assert_frame_equal(result, expected)
23912394

23922395
@pytest.mark.parametrize("force_suffixes", [False, True])

0 commit comments

Comments
 (0)