a oEb@sddlZddlZddlmZddlmZddlm Z m Z ej ddgdddZ ej ddgdd d Z ej d d Zej d dZej ddZej ddZej ddZej ddZej ddZe ddZe ddZej ee dddZej ee ddd Zej ee ee dd!d"Zej ddgdd#d$Zej dgdd%d&Zej dgdd'd(Zej d)ifd*d+d,ifd*d+difd-d+d,ifd-d+difd.ifgdd/d0ZdS)1N DataFrame)reduction_kernelstransformation_kernelsTF)paramscCs|jSNparamrequestr ]/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/pandas/tests/groupby/conftest.pysort srcCs|jSrrr r r r as_indexsrcCs|Srr )Z multiindex_dataframe_random_datar r r mframesrcCs*tgdgdtjdtjddS)NfoobarrrrrrronertwothreerrrrABCDrnprandomrandnr r r r dfs  r"cCstSr)tmZmakeTimeSeriesr r r r ts'sr$cCstSr)r#ZgetTimeSeriesDatar r r r tsd,sr%cCst|Srr)r%r r r tsframe1sr&cCs4tgdgdtjdtjtjddddS)NrrrZfloat32)Zdtyper)rrr r!arrayr r r r df_mixed_floats6s r(c Cs:tgdgdgdtjdtjdtjddS)N) rrrrrrrrrrr) rrrrrrrrrrr) dullr)shinyr)r)r*r*r)r*r*r* )rrrrEFrr r r r three_groupBs     r.c CsNgdgdgdgdgdgdgdgdg}t|gd d }|d S) N)raZa0_at_0)bZb0_at_1)r/Za1_at_2)r1Zb1_at_3)cZc0_at_4)r/Za2_at_5)r/Za3_at_6)r/Za4_at_7)IndexGroupValue)columnsr9)rZ set_index)datar"r r r slice_test_dfts r>cCs|jdddS)Nr:F)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_funcsrAcCs|jS)z@yields the string names of all groupby transformation functions.rr r r r transformation_funcsrBcCs|jS)z5yields both aggregation and transformation functions.rr r r r groupby_funcsrCcCs|jS)z'parallel keyword argument for numba.jitrr r r r parallelsrDcCs|jS)z$nogil keyword argument for numba.jitrr r r r nogilsrEcCs|jS)z'nopython keyword argument for numba.jitrr r r r nopythonsrFmeanvarZddofr0ZstdsumcCs|jS)z(reductions supported with engine='numba'rr r r r numba_supported_reductionss rJ)numpyrZpytestZpandasrZpandas._testingZ_testingr#Zpandas.core.groupby.baserrZfixturerrrr"r$r%r&r(r.r>r@sortedrArBrCrDrErFrJr r r r s`           1