@@ -56,7 +56,7 @@ def func(values, index):
56
56
func = numba .jit (func )
57
57
58
58
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 ]
60
60
)
61
61
engine_kwargs = {"nogil" : nogil , "parallel" : parallel , "nopython" : nopython }
62
62
grouped = data .groupby (0 )
@@ -89,7 +89,7 @@ def func_2(values, index):
89
89
func_2 = numba .jit (func_2 )
90
90
91
91
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 ]
93
93
)
94
94
engine_kwargs = {"nogil" : nogil , "parallel" : parallel , "nopython" : nopython }
95
95
grouped = data .groupby (0 )
@@ -120,7 +120,7 @@ def func_1(values, index):
120
120
return values + 1
121
121
122
122
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 ]
124
124
)
125
125
grouped = data .groupby (0 )
126
126
expected = grouped .transform (func_1 , engine = "numba" )
0 commit comments