U Zj @sddlZddlZddlmZddlmZddlm 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/difd2ifd3ifd4ifgd-d5d6d7d8d2d3d4gd9d:d;Z dS)<N DataFrame)reduction_kernelstransformation_kernelsTF)paramscCs|jSNparamrequestr `/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/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 src CsBtddddddddgddddddddgtjdtjddS)NfoobaronetwothreeABCDrnprandomrandnr 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&c CsLtddddddddgddddddddgtjdtjtjdddd S) NrrrrrrZfloat32)Zdtyper)rrr r!arrayr r r r df_mixed_floats@s r(cCsptdddddddddddg dddddddddddg dddddddddddg tjdtjdtjddS) NrrrrZdullZshiny )rrrrEFrr r r r three_groupLsT   r,c Cs`dddgdddgdddgd dd gd d d gdddgdddgdddgg}t|dddgd}|dS)NraZa0_at_0bZb0_at_1Za1_at_2Zb1_at_3cZc0_at_4Za2_at_5Za3_at_6Za4_at_7IndexGroupValue)columns)rZ set_index)datar"r r r slice_test_df~s r<cCs|jdddS)Nr8F)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_funcsr?cCs|jS)z@yields the string names of all groupby transformation functions.rr r r r transformation_funcsr@cCs|jS)z5yields both aggregation and transformation functions.rr r r r groupby_funcsrAcCs|jS)z'parallel keyword argument for numba.jitrr r r r parallelsrBcCs|jS)z$nogil keyword argument for numba.jitrr r r r nogilsrCcCs|jS)z'nopython keyword argument for numba.jitrr r r r nopythonsrDZmeanvarZddofr.ZstdsumminmaxZvar_1Zvar_0Zstd_1Zstd_0)rZidscCs|jS)z(reductions supported with engine='numba'rr r r r numba_supported_reductionssrI)!numpyrZpytestZpandasrZpandas._testingZ_testingr#Zpandas.core.groupby.baserrZfixturerrrrrr"r$r%r&r(r,r<r>sortedr?r@rArBrCrDrIr r r r sn             1