o TieR@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 `/var/www/html/evchargy.com/venv/lib/python3.10/site-packages/pandas/tests/groupby/test_raises.pysr) rrrr r r r rrrrrr r r r rrrrrrrr)paramscC|jSNparamrequestr r rbysrTFcCrrrrr 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 categoriesordered)rr"rr$r r rdf_with_cat_colSs r3cCs|dur+|dkrt|||dS|dkr |j|g|RdS|j|g|RdStj||d<|dkr?t|||n|dkrM|j|g|Rn|j|g|RWddSWddSWddS1sqwYdS)Nmethodaggmatch)getattrr5 transformpytestraises)klassmsghowgb groupby_funcargsr r r_call_and_checkds""rBr>)r4r5r9c Cs|}t||}|j|d}|r |d}|dkr t|drJdSidddddddtdfd dd dd ttfd fd ttfdfdttfdfdttfdfdtdfddddddddddddidddttdfdttdfd dd!dd"dd#tdfd$ttd%fd&td'fd(dd)td*fd+dd,dd-td*fd.td*fd/dd0ttd1f|\}} t|| ||||dS)2Nrr!corrwithallNanybfillzCould not convertcountcumcountcummaxzH(function|cummax) is not (implemented|supported) for (this|object) dtypecumminzH(function|cummin) is not (implemented|supported) for (this|object) dtypecumprodzI(function|cumprod) is not (implemented|supported) for (this|object) dtypecumsumzH(function|cumsum) is not (implemented|supported) for (this|object) dtypediffunsupported operand typeffillfillnafirstidxmaxidxminlastmaxmean-agg function failed [how->mean,dtype->object]medianz/agg function failed [how->median,dtype->object]minngroupnunique pct_changeprodz-agg function failed [how->prod,dtype->object]quantilez,cannot be performed against 'object' dtypes!ranksemz!could not convert string to floatshiftsizeskewstdsumvarz,agg function failed [how->var,dtype->object]) rgroupbyhasattr TypeErrorNotImplementedErrorreescape ValueErrorrB) r>rrr@r&r%rAr?r<r=r r rtest_groupby_raises_stringvs   #'()*+/01234567;=rqr5r9cCf|}|j|d}|r|d}dd}tjtddt|||WddS1s,wYdS)NrCr!cStdNTest error messagerlr r r rfuncz,test_groupby_raises_string_udf..funcrur6rjr:r;rlr8)r>rrr&r%r?rwr r rtest_groupby_raises_string_udf "rzgroupby_func_npc Cs|}|j|d}|r|d}tjdtjttdfi|\}}|r$d} nd} tjt | dt |||||dWddS1sBwYdS) NrCr!rFrZusing SeriesGroupBy.[sum|mean]!using DataFrameGroupBy.[sum|mean]r6r ) rjnprhrYrlrnrotmassert_produces_warning FutureWarningrB) r>rrr|r&r%r?r<r=warn_msgr r rtest_groupby_raises_string_nps$ "rc Cs|}t||}|j|d}|r |d}|dkr t|drJdSidddddddtdfd dd dd dd dd tdfdtdfddddddddddddddidddddddddddddtdfd td!fd"dd#dd$dd%dd&dd'td(d)d*gfd+dd,td-fd.td/f|\}} d} d0|d1} |d2vrt} tj| | d3t|| ||||WddS1swYdS)4NrCr!rDrErFrHrIz+cannot perform __mul__ with this index typerJrKrLrMrNz3datetime64 type does not support cumprod operationsrOz2datetime64 type does not support cumsum operationsrPrRrSrTrUrVrWrXrYr[r\r]r^r_z/cannot perform __truediv__ with this index typer`z%datetime64 type does not support prodrarbrcrdrerf|z1dtype datetime64\[ns\] does not support reductionz0datetime64 type does not support skew operationsrgrh/datetime64 type does not support sum operationsriz/datetime64 type does not support var operations'z&' with datetime64 dtypes is deprecated)rHrEr6) rrjrkrljoinrrrrB) r>rrr@r+r%rAr?r<r=warnrr r rtest_groupby_raises_datetimes       ()*+- "rcCrr)NrCr!cSrsrtrvr r r rrw=rxz.test_groupby_raises_datetime_udf..funcrur6ry)r>rrr+r%r?rwr r r test_groupby_raises_datetime_udf5r{rc Cs|}|j|d}|r|d}tjtdftjdi|\}}|r!d} nd} tjt| dt|||||dWddS1s?wYdS) NrCr!rrFr}r~r6r rjrrhrlrYrrrrB) r>rrr|r+r%r?r<r=rr r rtest_groupby_raises_datetime_npDs  "rrw)r`rNrfricCs&|}|jdd}ttdd||gdS)NrrCz/timedelta64 type does not support .* operationsr4)rjrBrl)rwr.r%r?r r rtest_groupby_raises_timedelta]s rc Cs|}t||}|j|d}|r |d}|dkr t|drJdSidddddddtdfd dd dd ttfd fd ttfdfdttfdfdttfdfdtdfddd|s`tdfndddddddddidddtddd gfd!tdd"d#gfd$dd%dd&dd'td(fd)td*fd+td,fd-dd.tdd/d0gfd1dd2dd3tdd4d5gfd6tdd7d8gfd9td:fd;tddNrCr!rDrErFrHrI=unsupported operand type\(s\) for \*: 'Categorical' and 'int'rJrKrLzz(category type does not support cummax operations|category dtype not supported|cummax is not supported for category dtype)rMzz(category type does not support cummin operations|category dtype not supported|cummin is not supported for category dtype)rNz|(category type does not support cumprod operations|category dtype not supported|cumprod is not supported for category dtype)rOzz(category type does not support cumsum operations|category dtype not supported|cumsum is not supported for category dtype)rPzDunsupported operand type\(s\) for -: 'Categorical' and 'Categorical'rRrSSCannot setitem on a Categorical with a new category \(0\), set the categories firstrTrUrVrWrXrYrz2'Categorical' .* does not support reduction 'mean'2category dtype does not support aggregation 'mean'r[z4'Categorical' .* does not support reduction 'median'4category dtype does not support aggregation 'median'r\r]r^r_zDunsupported operand type\(s\) for /: 'Categorical' and 'Categorical'r`.category type does not support prod operationsrazNo matching signature foundrbrc1'Categorical' .* does not support reduction 'sem'1category dtype does not support aggregation 'sem'rdrerf0dtype category does not support reduction 'skew'.category type does not support skew operationsrg1'Categorical' .* does not support reduction 'std'1category dtype does not support aggregation 'std'rh-category type does not support sum operationsri1'Categorical' .* does not support reduction 'var'1category dtype does not support aggregation 'var')rrjrkrlrmrrB) r>rrr@using_copy_on_writer3r%rAr?r<r=r r rtest_groupby_raises_categoryls*    "&'./0123<EFGHLMNOXYZclmvxrcCrr)NrCr!cSrsrtrvr r r rrwrxz.test_groupby_raises_category_udf..funcrur6ry)r>rrr3r%r?rwr r r test_groupby_raises_category_udfs "rc Cs|}|j|d}|r|d}tjtdftjtdfi|\}}|r#d} nd} tjt| dt|||||dWddS1sAwYdS) NrCr!rrr}r~r6r r) r>rrr|r3r%r?r<r=rr r rtest_groupby_raises_category_nps$  "rc Cs|}tgdgddd|d<t||}|j||d} |r.| d} |dkr.t| dr,JdStd d | jD} id d d d dd dtdfdd dd dttfdfdttfdfdttfdfdttfdfdtdfdd d|sztdfnd dd d | rt d!fnd d"| rt d#fnd d$d id%d d&td'fd(td)fd*d d+d d,d d-tdfd.td/fd0td1fd2d d3td4 d5d6gfd7d d8d d9td4 d:d;gfdgfd?td@fdAtd4 dBdCgf|\} } t | | || ||dS)DNr/rTr0r)robservedr!rDcss|]}|jVqdSr)empty).0groupr r r >sz;test_groupby_raises_category_on_category..rErFrHrIrrJrKrLzz(cummax is not supported for category dtype|category dtype not supported|category type does not support cummax operations)rMzz(cummin is not supported for category dtype|category dtype not supported|category type does not support cummin operations)rNz|(cumprod is not supported for category dtype|category dtype not supported|category type does not support cumprod operations)rOzz(cumsum is not supported for category dtype|category dtype not supported|category type does not support cumsum operations)rPrQrRrSrrTrUz*attempt to get argmax of an empty sequencerVz*attempt to get argmin of an empty sequencerWrXrYrr[rr\r]r^r_r`rrarGrbrcrrrrdrerfrrrgrrrhrrirr) rrrjrkrHgroupsvaluesrlrmrprrB) r>rrr@rrr3r%rAr?Z empty_groupsr<r=r r r(test_groupby_raises_category_on_category#s    "#$+, / 23456789:;<=FGHQZ[dfrcCstdgdgdgd}d}tjt|d|jddd}Wdn1s&wYtjtd d |d WddS1sBwYdS) Nrr r)rrr z+DataFrame.groupby with axis=1 is deprecatedr6r)Zaxisz'Cannot subset columns when using axis=1r)rrrrrjr:r;rp)r%r=r?r r r%test_subsetting_columns_axis_1_raisess "r)+r*rnnumpyrr:ZpandasrrrrZpandas._testingZ_testingrZpandas.tests.groupbyrZfixturearraydictzipr"rrr&r+r.r3rBmarkZ parametrizerqrzrhrYrrrrrrrrrrr r r rsx           N   D