@@ -56,7 +56,7 @@ def func(values, index):
5656 func = numba .jit (func )
5757
5858 data = DataFrame (
59- {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ],
59+ {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ]
6060 )
6161 engine_kwargs = {"nogil" : nogil , "parallel" : parallel , "nopython" : nopython }
6262 grouped = data .groupby (0 )
@@ -89,7 +89,7 @@ def func_2(values, index):
8989 func_2 = numba .jit (func_2 )
9090
9191 data = DataFrame (
92- {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ],
92+ {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ]
9393 )
9494 engine_kwargs = {"nogil" : nogil , "parallel" : parallel , "nopython" : nopython }
9595 grouped = data .groupby (0 )
@@ -120,7 +120,7 @@ def func_1(values, index):
120120 return values + 1
121121
122122 data = DataFrame (
123- {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ],
123+ {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ]
124124 )
125125 grouped = data .groupby (0 )
126126 expected = grouped .transform (func_1 , engine = "numba" )
0 commit comments