U b@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 A/tmp/pip-unpacked-wheel-ck39h295/pandas/tests/groupby/conftest.pysort srcCs|jSrrr r r r as_indexsrcCs|Srr )Z multiindex_dataframe_random_datar r r mframesrc CsBtddddddddgddddddddgtjdtjddS)NfoobaronetwothreeABCDrnprandomrandnr 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$c CsLtddddddddgddddddddgtjdtjtjdddd S) NrrrrrrZfloat32)Zdtyper)rrrrarrayr r r r df_mixed_floats6s r&cCsptdddddddddddg dddddddddddg dddddddddddg tjdtjdtjddS) NrrrrZdullZshiny )rrrrEFrr r r r three_groupBsT   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_7ZIndexGroupValue)columns)rZ set_index)datar r r r slice_test_dfts r9cCs|jdddS)Nr5F)r)groupby)r9r 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 parallelsr?cCs|jS)z$nogil keyword argument for numba.jitrr r r r nogilsr@cCs|jS)z'nopython keyword argument for numba.jitrr r r r nopythonsrAZmeanvarZddofr,ZstdsumcCs|jS)z(reductions supported with engine='numba'rr r r r numba_supported_reductionss rD)ZnumpyrZpytestZpandasrZpandas._testingZ_testingr!Zpandas.core.groupby.baserrZfixturerrrr r"r#r$r&r*r9r;sortedr<r=r>r?r@rArDr r r r s`           1