@@ -57,7 +57,7 @@ def func_numba(values, index):
5757 func_numba = numba .jit (func_numba )
5858
5959 data = DataFrame (
60- {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ],
60+ {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ]
6161 )
6262 engine_kwargs = {"nogil" : nogil , "parallel" : parallel , "nopython" : nopython }
6363 grouped = data .groupby (0 )
@@ -90,7 +90,7 @@ def func_2(values, index):
9090 func_2 = numba .jit (func_2 )
9191
9292 data = DataFrame (
93- {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ],
93+ {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ]
9494 )
9595 engine_kwargs = {"nogil" : nogil , "parallel" : parallel , "nopython" : nopython }
9696 grouped = data .groupby (0 )
@@ -121,7 +121,7 @@ def func_1(values, index):
121121 return np .mean (values ) - 3.4
122122
123123 data = DataFrame (
124- {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ],
124+ {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ]
125125 )
126126 grouped = data .groupby (0 )
127127 expected = grouped .agg (func_1 , engine = "numba" )
@@ -142,7 +142,7 @@ def func_1(values, index):
142142)
143143def test_multifunc_notimplimented (agg_func ):
144144 data = DataFrame (
145- {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ],
145+ {0 : ["a" , "a" , "b" , "b" , "a" ], 1 : [1.0 , 2.0 , 3.0 , 4.0 , 5.0 ]}, columns = [0 , 1 ]
146146 )
147147 grouped = data .groupby (0 )
148148 with pytest .raises (NotImplementedError , match = "Numba engine can" ):
0 commit comments