@@ -3066,8 +3066,8 @@ def test_merge_for_suffix_collisions():
3066
3066
# GH#61402
3067
3067
# Case 1: suffixes=("_dup", "") test collision
3068
3068
df1 = pd .DataFrame ({"col1" : [1 ], "col2" : [2 ]})
3069
- df2 = pd .DataFrame ({"col1" : [1 ], "col2" : [2 ], "col2_dup" : [3 ]})
3070
- with pytest .raises (MergeError , match = "duplicate columns" ):
3069
+ df2 = pd .DataFrame ({"col1" : [1 ], "col2" : [2 ], "col2_dup" : [3 ]})
3070
+ with pytest .raises (MergeError , match = "duplicate columns" ):
3071
3071
pd .merge (df1 , df2 , on = "col1" , suffixes = ("_dup" , "" ))
3072
3072
3073
3073
# Case 2: suffixes=("", "_dup") test collision
@@ -3076,22 +3076,21 @@ def test_merge_for_suffix_collisions():
3076
3076
with pytest .raises (MergeError , match = "duplicate columns" ):
3077
3077
pd .merge (df1 , df2 , on = "col1" , suffixes = ("" , "_dup" ))
3078
3078
3079
- # Case 3: suffixes=("_dup" != "__dup"), test similar
3079
+ # Case 3: suffixes=("_dup" != "__dup"), test similar
3080
3080
# non-collision
3081
3081
df3 = pd .DataFrame ({"col1" : [1 ], "col2__dup" : [4 ]})
3082
3082
df4 = pd .DataFrame ({"col1" : [1 ], "col2" : [2 ]})
3083
- merged = df3 .merge (df4 , on = ' col1' )
3083
+ merged = df3 .merge (df4 , on = " col1" )
3084
3084
expected = pd .DataFrame ([{"col1" : 1 , "col2__dup" : 4 , "col2" : 2 }])
3085
- pd .testing .assert_frame_equal (merged , expected )
3086
-
3085
+ pd .testing .assert_frame_equal (merged , expected )
3087
3086
3088
- # Case 4: suffixes=("", ""), auto-suffixes create
3087
+ # Case 4: suffixes=("", ""), auto-suffixes create
3089
3088
# collision due to co2_x existing
3090
3089
df5 = pd .DataFrame ({"col1" : [1 ], "col2" : [2 ]})
3091
- df6 = pd .DataFrame ({"col1" : [1 ], "col2" : [2 ], "col2_x" : [3 ]})
3090
+ df6 = pd .DataFrame ({"col1" : [1 ], "col2" : [2 ], "col2_x" : [3 ]})
3092
3091
with pytest .raises (MergeError , match = "duplicate columns" ):
3093
3092
pd .merge (df5 , df6 , on = "col1" )
3094
-
3093
+
3095
3094
# Case 5: test special characters in suffixes
3096
3095
df7 = pd .DataFrame ({"col1" : [1 ], "metric" : [10 ]})
3097
3096
df8 = pd .DataFrame ({"col1" : [1 ], "metric" : [10 ], "metric#$%^" : [20 ]})
@@ -3100,13 +3099,13 @@ def test_merge_for_suffix_collisions():
3100
3099
3101
3100
# Case 6: check if escape in column name causes conflict
3102
3101
df7 = pd .DataFrame ({"col1" : [1 ], "metric" : [10 ]})
3103
- df8 = pd .DataFrame ({"col1" : [1 ], "metric" : [10 ], "metric\#$%^" : [20 ]})
3102
+ df8 = pd .DataFrame ({"col1" : [1 ], "metric" : [10 ], r "metric\#$%^" : [20 ]})
3104
3103
with pytest .raises (MergeError , match = "duplicate columns" ):
3105
- pd .merge (df7 , df8 , on = "col1" , suffixes = ("\#$%^" , "" ))
3104
+ pd .merge (df7 , df8 , on = "col1" , suffixes = (r "\#$%^" , "" ))
3106
3105
3107
3106
# Case 7: suffix is a special character string
3108
3107
df9 = pd .DataFrame ({"col1" : [1 ], "col2" : [2 ], "col2" : [3 ]})
3109
3108
df10 = pd .DataFrame ({"col1" : [1 ], "col2" : [2 ]})
3110
- merged = df9 .merge (df10 , on = ' col1' )
3109
+ merged = df9 .merge (df10 , on = " col1" )
3111
3110
expected = pd .DataFrame ([{"col1" : 1 , "col2_x" : 3 , "col2_y" : 2 }])
3112
- pd .testing .assert_frame_equal (merged , expected )
3111
+ pd .testing .assert_frame_equal (merged , expected )
0 commit comments