U f/eY@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|jSNparamrequestr A/tmp/pip-unpacked-wheel-tiezk1ph/pandas/tests/groupby/conftest.pyas_indexsrc 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)r r r r mframes .r%c CsBtddddddddgddddddddgtjdtjddS)NrrrrrrrrDrr"r#r$r r r r dfs  r*cCstSr)tmZmakeTimeSeriesr r r r ts*sr,cCstSr)r+ZgetTimeSeriesDatar r r r tsd/sr-cCst|Sr)r)r-r r r tsframe4sr.c CsLtddddddddgddddddddgtjdtjtjdddd S) Nrrrrrr&Zfloat32)Zdtyper')rr"r#r$arrayr r r r df_mixed_floats9s r0cCsptdddddddddddg dddddddddddg dddddddddddg tjdtjdtjddS) NrrrrZdullZshiny )rrrr(EFr)r r r r three_groupEsT   r4cCs|jS)zT yields the string names of all groupby reduction functions, one at a time. rr r r r reduction_funcwsr5cCs|jS)z@yields the string names of all groupby transformation functions.rr r r r transformation_funcsr6cCs|jS)z5yields both aggregation and transformation functions.rr r r r groupby_funcsr7cCs|jS)z'parallel keyword argument for numba.jitrr r r r parallelsr8cCs|jS)z$nogil keyword argument for numba.jitrr r r r nogilsr9cCs|jS)z'nopython keyword argument for numba.jitrr r r r nopythonsr:)Znumpyr"ZpytestZpandasrrZpandas._testingZ_testingr+Zpandas.core.groupby.baserrZfixturerr%r*r,r-r.r0r4sortedr5r6r7r8r9r:r r r r s@         1