o Eh@sddlZddlZddlmZmZmZmZddlm Z m Z ej ddgdddZ ej ddgdd d Z ej ddgdd d Zej ddgdd dZej ddZej ddZej ddZej ddZe ddZe ddZej ee dddZej ee dddZej ee ee ddd Zej ddgdd!d"Zej dgdd#d$Zej dgdd%d&Zej d'ifd(d)d*ifd(d)difd+d)d*ifd+d)difd,ifd-ifd.ifd,d/d0ifd-d/d0ifd.d/d0ifg gd1d2d3d4ZdS)5N) DataFrameIndexSeries date_range)reduction_kernelstransformation_kernelsTF)paramscC|jSNZparamZrequestr =lib/python3.10/site-packages/pandas/tests/groupby/conftest.pysortrcCr r r r r r ras_indexrrcCr r r r r r rdropnarrcCr r r r r r robservedrrcCs6tgdgdtjddtjdddS)N)foobarrrrrrr)onertwothreerrrr)ABCDrnprandom default_rngstandard_normalr r r rdf$sr$cCs$ttjddtdddddS)Nr 2000-01-01rZperiodsZfreq)index)rr r!r"r#rr r r rts0s r)cCs2ttjddttdtdtddddd S) Nr)r%ZABCD)Zdtyper&r%rr')columnsr() rr r!r"r#rlistobjectrr r r rtsframe8s  r.c CsLtgdgdgdtjddtjddtjdddS)N) rrrrrrrrrrr) rrrrrrrrrrr) dullr/shinyr/r/r0r0r/r0r0r0r )rrrrEFrr r r r three_groupAs   r4c CsNgdgdgdgdgdgdgdgdg}t|gd d }|d S) N)raZa0_at_0)bZb0_at_1)rr5Za1_at_2)r7Zb1_at_3)r*cZc0_at_4)r5Za2_at_5)r5Za3_at_6)r5Za4_at_7)rGroupZValue)r+r)rZ set_index)datar$r r r slice_test_dfss r?cCs|jdddS)Nr=F)r)groupby)r?r r rslice_test_groupedsrAcCr )zT yields the string names of all groupby reduction functions, one at a time. r r r r rreduction_funcsrBcCr )z@yields the string names of all groupby transformation functions.r r r r rtransformation_funcrCcCr )z5yields both aggregation and transformation functions.r r r r r groupby_funcrDrEcCr )z'parallel keyword argument for numba.jitr r r r rparallelrDrFcCr )z$nogil keyword argument for numba.jitr r r r rnogilrDrGcCr )z'nopython keyword argument for numba.jitr r r r rnopythonrDrHmeanvarZddofr6ZstdsumminmaxZ min_countr) rIZvar_1Zvar_0Zstd_1Zstd_0rKrLrMz sum-min_countz min-min_countz max-min_count)rZidscCr )z(reductions supported with engine='numba'r r r r rnumba_supported_reductionssrN)Znumpyr ZpytestZpandasrrrrZpandas.core.groupby.baserrZfixturerrrrr$r)r.r4r?rAsortedrBrCrErFrGrHrNr r r rsh        1