a  deR@sddlZddlZddlZddlZddlmZmZmZm Z ddl m Z ddl mZejddgddgeddddgd egd eeed gd e gd e gd e gd gg d ddZejddgd ddZejddZejddZejddZejddZddZejdgdd d!Zejdd"d#gd$d%Zejdd"d#gejd&ejej gd'd(Z!ejdgdd)d*Z"ejdd"d#gd+d,Z#ejdd"d#gejd&ejej gd-d.Z$ejd/gd0d1d2Z%ejdgdd3d4Z&ejdd"d#gd5d6Z'ejdd"d#gejd&ejej gd7d8Z(ejdgdd9d:Z)d;d<Z*dS)=N) Categorical DataFrameGrouperSeries)get_groupby_method_argsab)keycCs|dS)Nxr r i/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/pandas/tests/groupby/test_raises.pyr) rrrr r r r rrrrrr r r r rrrrrrrr)paramscCs|jSNparamrequestr r rbysrTFcCs|jSrrrr r rgroupby_series'srcCs&tgdgdtdtdd}|S)NrrrZ xyzwtyuiorrcd)rrangelistdfr r rdf_with_string_col,sr&c Cs4tgdgdtdtddddddd d }|S) Nrrrir i`=r)rr"datetimer$r r rdf_with_datetime_col9sr+cCs*tgdgdtdtjddd}|S)Nrrrr)daysr)rr"r* timedeltar$r r rdf_with_timedelta_colFs r.c Cs4tgdgdtdtgdgdddd}|S)Nrrr rrrrrrrrr rT categoriesZordered)rr"rr$r r rdf_with_cat_colSs r2cCs|durP|dkr t|||q|dkr<|j|g|Rq|j|g|Rnttj||dV|dkrxt|||n.|dkr|j|g|Rn|j|g|RWdn1s0YdS)Nmethodaggmatch)getattrr4 transformpytestraises)klassmsghowgb groupby_funcargsr r r_call_and_checkdsrAr=)r3r4r8c %Cs |}t||}|j|d}|r@|d}|dkr@t|drmean,dtype->object]z/agg function failed [how->median,dtype->object]z-agg function failed [how->prod,dtype->object]z,cannot be performed against 'object' dtypes!z!could not convert string to floatz,agg function failed [how->var,dtype->object]"allanyZbfillrCcountZcumcountZcummaxZcummincumprodZcumsumdiffZffillZfillnafirstZidxmaxZidxminlastmaxmeanZmedianminZngroupZnuniqueZ pct_changeprodZquantileZrankZsemshiftsizeskewZstdsumvar) rgroupbyhasattr TypeErrorNotImplementedErrorreescape ValueErrorrA) r=rrr?r&r%r@r>r;r<r r rtest_groupby_raises_stringvs|  ;=r`r4r8cCsd|}|j|d}|r|d}dd}tjtddt|||Wdn1sV0YdS)NrBr!cSs tddSNTest error messager[r r r rfuncsz,test_groupby_raises_string_udf..funcrbr5rYr9r:r[r7)r=rrr&r%r>rdr r rtest_groupby_raises_string_udfs rfgroupby_func_npc Cs|}|j|d}|r|d}tjdtjttdfi|\}}|rHd} nd} tjt | d"t |||||dWdn1s0YdS) NrBr!rDrGusing SeriesGroupBy.[sum|mean]!using DataFrameGroupBy.[sum|mean]r5r ) rYnprWrQr[r]r^tmassert_produces_warning FutureWarningrA) r=rrrgr&r%r>r;r<warn_msgr r rtest_groupby_raises_string_nps" roc #Cs|}t||}|j|d}|r@|d}|dkr@t|drr;r<warnrnr r rtest_groupby_raises_datetimesr   +- rucCsd|}|j|d}|r|d}dd}tjtddt|||Wdn1sV0YdS)NrBr!cSs tddSrarcr r r rrd=sz.test_groupby_raises_datetime_udf..funcrbr5re)r=rrr+r%r>rdr r r test_groupby_raises_datetime_udf5s rvc Cs|}|j|d}|r|d}tjtdftjdi|\}}|rBd} nd} tjt| d"t|||||dWdn1s|0YdS) NrBr!rqrDrhrir5r rYrjrWr[rQrkrlrmrA) r=rrrgr+r%r>r;r<rnr r rtest_groupby_raises_datetime_npDs  rxrd)rSrLrVrXcCs&|}|jdd}ttdd||gdS)NrrBz/timedelta64 type does not support .* operationsr3)rYrAr[)rdr.r%r>r r rtest_groupby_raises_timedelta]s ryc &Cs@|}t||}|j|d}|r@|d}|dkr@t|drr;r<r r rtest_groupby_raises_categoryls       vxrcCsd|}|j|d}|r|d}dd}tjtddt|||Wdn1sV0YdS)NrBr!cSs tddSrarcr r r rrdsz.test_groupby_raises_category_udf..funcrbr5re)r=rrr2r%r>rdr r r test_groupby_raises_category_udfs rc Cs|}|j|d}|r|d}tjtdftjtdfi|\}}|rFd} nd} tjt| d"t|||||dWdn1s0YdS) NrBr!rr|rhrir5r rw) r=rrrgr2r%r>r;r<rnr r rtest_groupby_raises_category_nps"  rc &Csx|}tgdgddd|d<t||}|j||d} |r\| d} |dkr\t| drXJdStd d | jD} d d d td fd d ttfd fttfdfttfdfttfdftdfd |stdfnd d | rt dfnd | rt dfnd d d tdftdfd d d tdftdftdfd td ddgfd d td ddgftd ddgftd ftd d!d"gfd#"|\} } t | | || ||dS)$Nr/rTr0r)robservedr!rCcss|] }|jVqdSr)empty).0groupr r r >rz;test_groupby_raises_category_on_category..rDrzzz(cummax is not supported for category dtype|category dtype not supported|category type does not support cummax operations)zz(cummin is not supported for category dtype|category dtype not supported|category type does not support cummin operations)z|(cumprod is not supported for category dtype|category dtype not supported|category type does not support cumprod operations)zz(cumsum is not supported for category dtype|category dtype not supported|category type does not support cumsum operations)rFr{z*attempt to get argmax of an empty sequencez*attempt to get argmin of an empty sequencer|r}r~rErprrrrrrrrrrH) rrrYrZrJgroupsvaluesr[r\r_rsrA) r=rrr?rrr2r%r@r>Z empty_groupsr;r<r r r(test_groupby_raises_category_on_category#s        dfrcCstdgdgdgd}d}tjt|d|jddd}Wdn1sL0Ytjtd d|d Wdn1s0YdS) Nrr r)rrr z+DataFrame.groupby with axis=1 is deprecatedr5r)Zaxisz'Cannot subset columns when using axis=1r)rrkrlrmrYr9r:r_)r%r<r>r r r%test_subsetting_columns_axis_1_raisess ,r)+r*r]numpyrjr9ZpandasrrrrZpandas._testingZ_testingrkZpandas.tests.groupbyrZfixturearraydictzipr"rrr&r+r.r2rAmarkZ parametrizer`rfrWrQrorurvrxryrrrrrr r r rsv           N   D