B @`a @s:ddlZddlZddlmZmZddlmZddl 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}ttjdd |dddgdS)NfoobarZbazZquxonetwothreerfirstsecond)levelscodesnames ABC)indexcolumns)rrnprandomrandn)rrA/tmp/pip-unpacked-wheel-q9tj5l6a/pandas/tests/groupby/conftest.pymframe s . rc CsBtddddddddgddddddddgtjdtjddS)Nrrrr r )rrrD)rrrrrrrrdfs  r!cCstS)N)tmZmakeTimeSeriesrrrrtssr#cCstS)N)r"ZgetTimeSeriesDatarrrrtsd$sr$cCst|S)N)r)r$rrrtsframe)sr%c CsLtddddddddgddddddddgtjdtjtjdddd S) Nrrrr r rZfloat32)Zdtype)rrrr )rrrrarrayrrrrdf_mixed_floats.s  r'cCsptdddddddddddg dddddddddddg dddddddddddg tjdtjdtjddS) Nrrrr ZdullZshiny )rrrr EF)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_functsr0cCs|jS)z5yields both aggregation and transformation functions.)r-)r.rrr groupby_funczsr1TFcCs|jS)z'parallel keyword argument for numba.jit)r-)r.rrrparallelsr2cCs|jS)z$nogil keyword argument for numba.jit)r-)r.rrrnogilsr3cCs|jS)z'nopython keyword argument for numba.jit)r-)r.rrrnopythonsr4)ZnumpyrZpytestZpandasrrZpandas._testingZ_testingr"Zpandas.core.groupby.baserrZfixturerr!r#r$r%r'r+sortedr/r0r1r2r3r4rrrrs"    2