@@ -1070,6 +1070,26 @@ def test_apply(self, request, data, func, axis):
1070
1070
modin_result = modin_df .apply (func , axis )
1071
1071
df_equals (modin_result , pandas_result )
1072
1072
1073
+ @pytest .mark .parametrize ("data" , test_data_values , ids = test_data_keys )
1074
+ @pytest .mark .parametrize ("axis" , axis_values , ids = axis_keys )
1075
+ def test_apply_args (self , data , axis ):
1076
+ modin_df = pd .DataFrame (data )
1077
+ pandas_df = pandas .DataFrame (data )
1078
+
1079
+ def apply_func (series , y ):
1080
+ try :
1081
+ return series + y
1082
+ except TypeError :
1083
+ return series .map (str ) + str (y )
1084
+
1085
+ modin_result = modin_df .apply (apply_func , axis = axis , args = (1 ,))
1086
+ pandas_result = pandas_df .apply (apply_func , axis = axis , args = (1 ,))
1087
+ df_equals (modin_result , pandas_result )
1088
+
1089
+ modin_result = modin_df .apply (apply_func , axis = axis , args = ("_A" ,))
1090
+ pandas_result = pandas_df .apply (apply_func , axis = axis , args = ("_A" ,))
1091
+ df_equals (modin_result , pandas_result )
1092
+
1073
1093
def test_apply_metadata (self ):
1074
1094
def add (a , b , c ):
1075
1095
return a + b + c
0 commit comments