U `{j @sddlZddlZddlmZddlmZddlm Z m Z ej ddgdddZ ej ddgdd d Z ej ddgdd d Zej d dZej ddZej ddZej ddZej ddZej ddZej ddZe ddZe ddZej ee ddd Zej ee dd!d"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-difd0ifd1ifd2ifgd+d3d4d5d6d0d1d2gd7d8d9ZdS):N DataFrame)reduction_kernelstransformation_kernelsTF)paramscCs|jSNparamrequestr \/var/www/html/bwave/python/venv/lib/python3.8/site-packages/pandas/tests/groupby/conftest.pysort srcCs|jSrrr r r r as_indexsrcCs|jSrrr r r r dropnasrcCs|Srr )Z multiindex_dataframe_random_datar r r mframesrc CsBtddddddddgddddddddgtjdtjddS)NfoobaronetwothreeABCDrnprandomrandnr r r r df s  r!cCstSr)tmZmakeTimeSeriesr r r r ts,sr#cCstSr)r"ZgetTimeSeriesDatar r r r tsd1sr$cCst|Srr)r$r r r tsframe6sr%c CsLtddddddddgddddddddgtjdtjtjdddd S) NrrrrrrZfloat32)Zdtyper)rrrr arrayr r r r df_mixed_floats;s r'cCsptdddddddddddg dddddddddddg dddddddddddg tjdtjdtjddS) NrrrrZdullZshiny )rrrrEFrr r r r three_groupGsT   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_dfys r;cCs|jdddS)Nr7F)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_funcsr@cCs|jS)z'parallel keyword argument for numba.jitrr r r r parallelsrAcCs|jS)z$nogil keyword argument for numba.jitrr r r r nogilsrBcCs|jS)z'nopython keyword argument for numba.jitrr r r r nopythonsrCZmeanvarZddofr-ZstdsumminmaxZvar_1Zvar_0Zstd_1Zstd_0)rZidscCs|jS)z(reductions supported with engine='numba'rr r r r numba_supported_reductionssrH) numpyrZpytestZpandasrZpandas._testingZ_testingr"Zpandas.core.groupby.baserrZfixturerrrrr!r#r$r%r'r+r;r=sortedr>r?r@rArBrCrHr r r r sj            1