3 qda @s8ddlZddlZddlmZmZddljZddl m Z m Z ej ddZ ej ddZej dd Zej d d Zej d d Zej ddZej ddZej ee dddZej ee dddZej ee ee dddZej ddgdddZej ddgdddZej ddgddd ZdS)!N) DataFrame MultiIndex)reduction_kernelstransformation_kernelsc Csntddddgdddggdddd d d d d d d g dd d dd d d dd d g gd d gd}ttjjdd |dddgdS)NfoobarZbazZquxonetwothreerfirstsecond)Zlevelscodesnames ABC)indexcolumns)rrnprandomrandn)rr?/tmp/pip-build-7vycvbft/pandas/pandas/tests/groupby/conftest.pymframe s . rc CsBtddddddddgddddddddgtjjdtjjddS)Nrrrr r )rrrD)rrrrrrrrdfs  r cCstjS)N)tmZmakeTimeSeriesrrrrtssr"cCstjS)N)r!ZgetTimeSeriesDatarrrrtsd$sr#cCst|S)N)r)r#rrrtsframe)sr$c CsLtddddddddgddddddddgtjjdtjtjjdddd S) Nrrrr r rZfloat32)Zdtype)rrrr)rrrrarrayrrrrdf_mixed_floats.s  r&cCsptdddddddddddg dddddddddddg dddddddddddg tjjdtjjdtjjddS) Nrrrr ZdullZshiny )rrrrEF)rrrrrrrr three_group:sJ  r*)paramscCs|jS)zT yields the string names of all groupby reduction functions, one at a time. )param)requestrrrreduction_funclsr.cCs|jS)z@yields the string names of all groupby transformation functions.)r,)r-rrrtransformation_functsr/cCs|jS)z5yields both aggregation and transformation functions.)r,)r-rrr groupby_funczsr0TFcCs|jS)z'parallel keyword argument for numba.jit)r,)r-rrrparallelsr1cCs|jS)z$nogil keyword argument for numba.jit)r,)r-rrrnogilsr2cCs|jS)z'nopython keyword argument for numba.jit)r,)r-rrrnopythonsr3)numpyrZpytestZpandasrrZpandas._testingZ_testingr!Zpandas.core.groupby.baserrZfixturerr r"r#r$r&r*sortedr.r/r0r1r2r3rrrrs"    2