o թZh@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|jSNparamrequestrT/var/www/html/lang_env/lib/python3.10/site-packages/pandas/tests/groupby/conftest.pysortrcCr r r r rrras_indexrrcCr r r r rrrdropnarrcCr r r r rrrobservedrrcCs6tgdgdtjddtjdddS)N)foobarrrrrrr)onertwothreerrrr)ABCDrnprandom default_rngstandard_normalrrrrdf$sr&cCs$ttjddtdddddS)Nr 2000-01-01rZperiodsfreq)index)rr"r#r$r%rrrrrts0s r,cCs2ttjddttdtdtddddd S) Nr)r'ZABCD)Zdtyper(r'rr))columnsr+) rr"r#r$r%rlistobjectrrrrrtsframe8s  r1c CsLtgdgdgdtjddtjddtjdddS)N) rrrrrrrrrrr) rrrrrrrrrrr) dullr2shinyr2r2r3r3r2r3r3r3r )rrrr EFr!rrrr three_groupAs   r7c CsNgdgdgdgdgdgdgdgdg}t|gd d }|d S) N)raZa0_at_0)bZb0_at_1)rr8Za1_at_2)r:Zb1_at_3)r-cZc0_at_4)r8Za2_at_5)r8Za3_at_6)r8Za4_at_7)rGroupValue)r.r)rZ set_index)datar&rrr slice_test_dfss rCcCs|jdddS)Nr@F)r)groupby)rCrrrslice_test_groupedsrEcCr )zT yields the string names of all groupby reduction functions, one at a time. r r rrrreduction_funcsrFcCr )z@yields the string names of all groupby transformation functions.r r rrrtransformation_funcrGcCr )z5yields both aggregation and transformation functions.r r rrr groupby_funcrHrIcCr )z'parallel keyword argument for numba.jitr r rrrparallelrHrJcCr )z$nogil keyword argument for numba.jitr r rrrnogilrHrKcCr )z'nopython keyword argument for numba.jitr r rrrnopythonrHrLmeanvarZddofr9ZstdsumminmaxZ min_countr) rMZvar_1Zvar_0Zstd_1Zstd_0rOrPrQz sum-min_countz min-min_countz max-min_count)rZidscCr )z(reductions supported with engine='numba'r r rrrnumba_supported_reductionssrR)numpyr"ZpytestZpandasrrrrZpandas.core.groupby.baserrZfixturerrrrr&r,r1r7rCrEsortedrFrGrIrJrKrLrRrrrrsh        1