B 2cY@sNddlZddlZddlmZmZddlmZddl m Z m Z ej ddgdddZ ej d d Zej d d Zej d dZej ddZej ddZej ddZej ddZej ee dddZej ee dddZej ee ee dddZej ddgdddZej dgddd Zej dgdd!d"ZdS)#N) DataFrame MultiIndex)reduction_kernelstransformation_kernelsTF)paramscCs|jS)N)param)requestr A/tmp/pip-install-v41xt_bg/pandas/pandas/tests/groupby/conftest.pyas_indexsr c Csntddddgdddggdddd d d d d d d g dd d dd d d dd d g gd d gd}ttjdd |dddgdS)NfoobarZbazZquxonetwothreerfirstsecond)Zlevelscodesnames ABC)indexcolumns)rrnprandomrandn)rr r r mframes . r!c CsBtddddddddgddddddddgtjdtjddS)Nr r rrr)rrrD)rrrr r r r r dfs  r$cCstS)N)tmZmakeTimeSeriesr r r r ts*sr&cCstS)N)r%ZgetTimeSeriesDatar r r r tsd/sr'cCst|S)N)r)r'r r r tsframe4sr(c CsLtddddddddgddddddddgtjdtjtjdddd S) Nr r rrrr"Zfloat32)Zdtype)rrrr#)rrrr arrayr r r r df_mixed_floats9s  r*cCsptdddddddddddg dddddddddddg dddddddddddg tjdtjdtjddS) Nr r rrZdullZshiny )rrrr#EF)rrrr r r r r three_groupEsJ  r.cCs|jS)zT yields the string names of all groupby reduction functions, one at a time. )r)rr r r reduction_funcwsr/cCs|jS)z@yields the string names of all groupby transformation functions.)r)rr r r transformation_funcsr0cCs|jS)z5yields both aggregation and transformation functions.)r)rr r r groupby_funcsr1cCs|jS)z'parallel keyword argument for numba.jit)r)rr r r parallelsr2cCs|jS)z$nogil keyword argument for numba.jit)r)rr r r nogilsr3cCs|jS)z'nopython keyword argument for numba.jit)r)rr r r nopythonsr4)ZnumpyrZpytestZpandasrrZpandas._testingZ_testingr%Zpandas.core.groupby.baserrZfixturer r!r$r&r'r(r*r.sortedr/r0r1r2r3r4r r r r s$    2