U sVc @sddlZddlZddlmZddlmZddlm Z m Z ej ddgdddZ ej ddgdd d Z ej ddgdd d Zej ddgdd dZej ddZej ddZej ddZej ddZej ddZej ddZej ddZe ddZe dd Zej ee dd!d"Zej ee dd#d$Zej ee ee dd%d&Zej ddgdd'd(Zej dgdd)d*Zej dgdd+d,Zej d-ifd.d/d0ifd.d/difd1d/d0ifd1d/difd2ifd3ifd4ifgd-d5d6d7d8d2d3d4gd9d:d;Z dS)<N DataFrame)reduction_kernelstransformation_kernelsTF)paramscCs|jSNparamrequestr A/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/groupby/conftest.pysort srcCs|jSrrr r r r as_indexsrcCs|jSrrr r r r dropnasrcCs|jSrrr r r r observedsrcCs|Srr )Z multiindex_dataframe_random_datar r r mframe src CsBtddddddddgddddddddgtjdtjddS)NfoobaronetwothreeABCDrnprandomrandnr r r r df%s  r"cCstSr)tmZmakeTimeSeriesr r r r ts1sr$cCstSr)r#ZgetTimeSeriesDatar r r r tsd6sr%cCst|Srr)r%r r r tsframe;sr&c CsLtddddddddgddddddddgtjdtjtjdddd S) NrrrrrrZfloat32)Zdtyper)rrr r!arrayr r r r df_mixed_floats@s r(cCsptdddddddddddg dddddddddddg dddddddddddg tjdtjdtjddS) NrrrrZdullZshiny )rrrrEFrr r r r three_groupLsT   r,c Cs`dddgdddgdddgd dd gd d d gdddgdddgdddgg}t|dddgd}|dS)NraZa0_at_0bZb0_at_1Za1_at_2Zb1_at_3cZc0_at_4Za2_at_5Za3_at_6Za4_at_7ZIndexGroupValue)columns)rZ set_index)datar"r r r slice_test_df~s r;cCs|jdddS)Nr7F)r)groupby)r;r r r slice_test_groupedsr=cCs|jS)zT yields the string names of all groupby reduction functions, one at a time. rr r r r reduction_funcsr>cCs|jS)z@yields the string names of all groupby transformation functions.rr r r r transformation_funcsr?cCs|jS)z5yields both aggregation and transformation functions.rr r r r groupby_funcsr@cCs|jS)z'parallel keyword argument for numba.jitrr r r r parallelsrAcCs|jS)z$nogil keyword argument for numba.jitrr r r r nogilsrBcCs|jS)z'nopython keyword argument for numba.jitrr r r r nopythonsrCZmeanvarZddofr.ZstdsumminmaxZvar_1Zvar_0Zstd_1Zstd_0)ridscCs|jS)z(reductions supported with engine='numba'rr r r r numba_supported_reductionssrI)!ZnumpyrZpytestZpandasrZpandas._testingZ_testingr#Zpandas.core.groupby.baserrZfixturerrrrrr"r$r%r&r(r,r;r=sortedr>r?r@rArBrCrIr r r r sn             1