a j=ic @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)paramscCs|jSNparamrequestr n/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/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 srcCs*tgdgdtjdtjddS)NfoobarrrrrrronertwothreerrrrABCDrnprandomrandnr 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(cCs4tgdgdtjdtjtjddddS)Nrrrfloat32)Zdtyper)rr!r"r#arrayr r r r df_mixed_floats@s r+c Cs:tgdgdgdtjdtjdtjddS)N) rrrrrrrrrrr) rrrrrrrrrrr) dullr,shinyr,r,r-r-r,r-r-r- )rrrrEFr r r r r three_groupLs     r1c CsNgdgdgdgdgdgdgdgdg}t|gd d }|d S) N)raZa0_at_0)bZb0_at_1)r2Za1_at_2)r4Zb1_at_3)cZc0_at_4)r2Za2_at_5)r2Za3_at_6)r2Za4_at_7)IndexGroupValue)columnsr<)rZ set_index)datar$r r r slice_test_df~s rAcCs|jdddS)Nr=F)r)groupby)rAr r r slice_test_groupedsrCcCs|jS)zT yields the string names of all groupby reduction functions, one at a time. rr r r r reduction_funcsrDcCs|jS)z@yields the string names of all groupby transformation functions.rr r r r transformation_funcsrEcCs|jS)z5yields both aggregation and transformation functions.rr r r r groupby_funcsrFcCs|jS)z'parallel keyword argument for numba.jitrr r r r parallelsrGcCs|jS)z$nogil keyword argument for numba.jitrr r r r nogilsrHcCs|jS)z'nopython keyword argument for numba.jitrr r r r nopythonsrImeanvarZddofr3Zstdsumminmax)rJZvar_1Zvar_0Zstd_1Zstd_0rLrMrN)rZidscCs|jS)z(reductions supported with engine='numba'rr r r r numba_supported_reductionssrO)!numpyr!ZpytestZpandasrZpandas._testingZ_testingr%Zpandas.core.groupby.baserrZfixturerrrrrr$r&r'r(r+r1rArCsortedrDrErFrGrHrIrOr r r r sn             1