o d @s ddlZddlZddlmZddlmZddlm 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/difd2ifd3ifd4ifggd5d6d7d8Z dS)9N DataFrame)reduction_kernelstransformation_kernelsTF)paramscC|jSNparamrequestr Q/app/.heroku/python/lib/python3.10/site-packages/pandas/tests/groupby/conftest.pysort rcCrrr r r r ras_indexrrcCrrr r r r rdropnarrcCrrr r r r robservedrrcCs|Srr )Z multiindex_dataframe_random_datar r rmframe srcCs*tgdgdtjdtjddS)NfoobarrrrrrronertwothreerrrrABCDrnprandomrandnr r r rdf%s  r&cCtSr)tmZmakeTimeSeriesr r r rts1r)cCr'r)r(ZgetTimeSeriesDatar r r rtsd6r*r+cCst|Srr)r+r r rtsframe;r*r,cCs4tgdgdtjdtjtjddddS)NrrrZfloat32)Zdtyper)rr#r$r%arrayr r r rdf_mixed_floats@s r.c Cs:tgdgdgdtjdtjdtjddS)N) rrrrrrrrrrr) rrrrrrrrrrr) dullr/shinyr/r/r0r0r/r0r0r0 )rrr r!EFr"r r r r three_groupLs   r4c CsNgdgdgdgdgdgdgdgdg}t|gd d }|d S) N)raZa0_at_0)bZb0_at_1)r5Za1_at_2)r7Zb1_at_3)cZc0_at_4)r5Za2_at_5)r5Za3_at_6)r5Za4_at_7)IndexGroupValue)columnsr?)rZ set_index)datar&r r r slice_test_df~s rDcCs|jdddS)Nr@F)r)groupby)rDr r rslice_test_groupedsrFcCr)zT yields the string names of all groupby reduction functions, one at a time. r r r r rreduction_funcsrGcCr)z@yields the string names of all groupby transformation functions.r r r r rtransformation_funcrHcCr)z5yields both aggregation and transformation functions.r r r r r groupby_funcrIrJcCr)z'parallel keyword argument for numba.jitr r r r rparallelrIrKcCr)z$nogil keyword argument for numba.jitr r r r rnogilrIrLcCr)z'nopython keyword argument for numba.jitr r r r rnopythonrIrMmeanvarZddofr6Zstdsumminmax)rNZvar_1Zvar_0Zstd_1Zstd_0rPrQrR)rZidscCr)z(reductions supported with engine='numba'r r r r rnumba_supported_reductionssrS)!numpyr#ZpytestZpandasrZpandas._testingZ_testingr(Zpandas.core.groupby.baserrZfixturerrrrrr&r)r+r,r.r4rDrFsortedrGrHrJrKrLrMrSr r r rsp             1