zhh\xddlZddlZddlZddlZddlmZmZmZm Z ddl m Z ddl mZejddgddgeddgdejgdeeed gde gd e gd e gd gg d Zejddg dZejdZejdZejdZejdZd*dZejdgddZejdddgdZejdddgejdejej gdZ!ejdgddZ"ejdddgd Z#ejdddgejdejej gd!Z$ejd"gd#d$Z%ejdgdd%Z&ejdddgd&Z'ejdddgejdejej gd'Z(ejdgdd(Z)d)Z*dS)+N) Categorical DataFrameGrouperSeries)get_groupby_method_argsab)keyc |dzS)Nxs W/var/www/histauto/venv/lib/python3.11/site-packages/pandas/tests/groupby/test_raises.pyrs !a%) rrrr r r r rrrrrr r r r rrrrrrrr)paramsc|jSNparamrequests rbyr s  =rTFc|jSrrrs rgroupby_seriesr"'s =rcjtgdgdtdtdd}|S)Nrrr xyzwtyuiorr cd)rrangelistdfs rdf_with_string_colr,,sH ,,,,,,qk""     B Irctgdgdtdtjddddddd d }|S) Nrrrir i`=r%)rr(datetimer*s rdf_with_datetime_colr29sW ,,,,,,q"4Ar2r6BB     B Ircvtgdgdtdtjdd}|S)Nrrrr)daysr%)rr(r1 timedeltar*s rdf_with_timedelta_colr6FsN ,,,,,,q#+++     B Irc xtgdgdtdtgdgddd}|S)Nrrr rrrrr r r r r&r%T categoriesordered)rr(rr*s rdf_with_cat_colr<Ssd ,,,,,,q===///   B Irc|dkrdnt}tj||5|>|dkrt|||n|dkr|j|g|Rnz|j|g|Rnkt j||5|dkrt|||n#|dkr|j|g|Rn|j|g|Rdddn #1swxYwYddddS#1swxYwYdS)Nr=matchmethodagg) FutureWarningtmassert_produces_warninggetattrrB transformpytestraises)klassmsghowgb groupby_funcargswarn_msg warn_klasss r_call_and_checkrRds!R]J #Jh ? ? ?66 =h)L))4000|+d+++++ \1D11111uC000 6 6(??-GB --t444E\\BF</$///// BL55555  6 6 6 6 6 6 6 6 6 6 6 6 6 6 6666666666666666666s6AC=>C; CC CC CC#&C#rL)rArBrGc b|}t||}||}|r"|d}|dkrt|drJdSidddddddtdfd dd dd ttfd fd ttfdfdttfdfdttfdfdtdfddddddddddddidddtt jdfdtt jdfd dd!dd"dd#tdfd$tt jd%fd&td'fd(dd)td*fd+dd,dd-td*fd.td*fd/dd0tt jd1f|\} } |rw|d2vrd3|d4} |d5vrt} na|d#kr|djj d6krd7} nB|dkr|djj d6krd8} n#|d9vr| d:d;} n|dkrd<} |dkr |rd=nd>} | d?} nd@} t| | ||||| dS)ANr r'corrwithallNr=anybfillzCould 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 typeffillfillnafirstidxmaxidxminlastmaxmeanz-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]quantilez4dtype 'object' does not support operation 'quantile'ranksemz!could not convert string to floatshiftsizeskewstdsumvarz%agg function failed [how->var,dtype->) rorirjr_r^rvrrrxrurpz(dtype 'str' does not support operation '')rrrvrupyarrowzBoperation 'truediv' not supported for dtype 'str' with dtype 'str'z>operation 'sub' not supported for dtype 'str' with dtype 'str')r]r\objectstrz1Cannot perform reduction 'mean' with string dtyperrGroupBy.fillna is deprecatedr=) rgroupbyhasattr TypeErrorNotImplementedErrorreescape ValueErrordtypestoragereplacerR) rLr r"rNr,using_infer_stringr+rOrMrJrKkindrPs rtest_groupby_raises_stringrxs B "< 4 4D r  B W : % %r:.. . . . F; z; z; ; Y 34 ;  ; J ;  ) , V ;  ) , V ;  ) , W ;&  ) , V ';. 67/;0 1;2 *3;4 5;6 *7;8 *9;:  ;;;< z=;>   IE F F ?;F   IG H H G;N zO;P *Q;R :S;T y"<=U;V   IE F F W;^ Y VW_;`  a;b  ?@c;d e;f  g;h @Ai;j  ?@k;l zm;n   I= > > o;;vw;JE3zF   M\LLLC555! \ ) )bgm.Cy.P.PWCC V # #3 (=(J(JSCC 1 1 1++h..CC Z ' 'ECx):xx{888E3RtXFFFFFrrBrGc|}||}|r|d}d}tjtd5t |||ddddS#1swxYwYdS)NrTr'c tdNTest error messagerrs rfuncz,test_groupby_raises_string_udf..func,---rrr?r~rHrIrrF)rLr r"r,r+rMrs rtest_groupby_raises_string_udfrs B r  B W... y(< = = =CA((A,/A,groupby_func_npc  |}||}|r|d}tjdtjtdfi|\}} |r|tjurt}d} |rd} nd} t || |||d| dS) NrTr'rWzEagg function failed|Cannot perform reduction 'mean' with string dtypez-dtype 'str' does not support operation 'mean'using SeriesGroupBy.[sum|mean]!using DataFrameGroupBy.[sum|mean]r rP)r~nprwrirrR) rLr r"rr,rr+rMrJrKrPs rtest_groupby_raises_string_nprs B r  B W     S   JE3> bg % %E=736E3R"xPPPPPPrc |}t||}||}|r"|d}|dkrt|drJdSidddddddtdfd dd dd dd dd tdfdtdfddddddddddddddidddddddddddddtdfd td!fd"dd#dd$dd%dd&dd'td(d)d*gfd+dd,td-fd.td/f|\}} |d0vrd1|d2} n|dkr |rd3nd4} | d5} nd6} t || ||||| 7dS)8NrTr'rUrVrWrXrYz+cannot perform __mul__ with this index typerZr[r\r]r^z3datetime64 type does not support cumprod operationsr_z2datetime64 type does not support cumsum operationsr`rbrcrdrerfrgrhrirjrkrlrmrnz/cannot perform __truediv__ with this index typeroz%datetime64 type does not support prodrprqrrrsrtru|z1dtype datetime64\[ns\] does not support reductionz0datetime64 type does not support skew operationsrvrw/datetime64 type does not support sum operationsrxz/datetime64 type does not support var operations)rXrVryz&' with datetime64 dtypes is deprecatedrrr}r=r)rr~rrjoinrR) rLr r"rNr2r+rOrMrJrKrPrs rtest_groupby_raises_datetimers B "< 4 4D r  B W : % %r:.. . . . F+ z+ z+ + Y MN +  + J + *+ *+ ITU+ 9RS+  + + *+ + *+ *!+"  #++$ z%+&  '+( *)+* z++, *-+. :/+0 y"ST1+2 CD3+4 J5+6  7+8 z9+: ;+<  =+>   HHHF   ?+P zQ+R  LMS+T  LMU++VW+JE3Z~%%K|KKK  ! !):xx{888E3RthOOOOOOrc|}||}|r|d}d}tjtd5t |||ddddS#1swxYwYdS)NrTr'c tdrrrs rrz.test_groupby_raises_datetime_udf..funcmrrrr?r)rLr r"r2r+rMrs r test_groupby_raises_datetime_udfres B r  B W... y(< = = =Crc |}||}|r|d}tjtdftjdi|\}}|rd} nd} t |||||d| dS) NrTr'rrWrrr rr~rrwrrirR) rLr r"rr2r+rMrJrKrPs rtest_groupby_raises_datetime_nprts B r  B W MN JE3 736E3R"xPPPPPPrr)ror^rurxch|}|d}ttdd||gdS)NrrTz/timedelta64 type does not support .* operationsrA)r~rRr)rr6r+rMs rtest_groupby_raises_timedeltarsH B s  B9   rc |}t||}||}|r"|d}|dkrt|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;tddnd?} | d@} ndA} t | | ||||| dS)BNrTr'rUrVrWrXrY=unsupported operand type\(s\) for \*: 'Categorical' and 'int'rZr[r\zz(category type does not support cummax operations|category dtype not supported|cummax is not supported for category dtype)r]zz(category type does not support cummin operations|category dtype not supported|cummin is not supported for category dtype)r^z|(category type does not support cumprod operations|category dtype not supported|cumprod is not supported for category dtype)r_zz(category type does not support cumsum operations|category dtype not supported|cumsum is not supported for category dtype)r`zDunsupported operand type\(s\) for -: 'Categorical' and 'Categorical'rbrcSCannot setitem on a Categorical with a new category \(0\), set the categories firstrdrerfrgrhrirz2'Categorical' .* does not support reduction 'mean'2category dtype does not support aggregation 'mean'rjz4'Categorical' .* does not support reduction 'median'4category dtype does not support aggregation 'median'rkrlrmrnzDunsupported operand type\(s\) for /: 'Categorical' and 'Categorical'ro.category type does not support prod operationsrpNo matching signature foundrqrr1'Categorical' .* does not support reduction 'sem'1category dtype does not support aggregation 'sem'rsrtru0dtype category does not support reduction 'skew'.category type does not support skew operationsrv1'Categorical' .* does not support reduction 'std'1category dtype does not support aggregation 'std'rw-category type does not support sum operationsrx1'Categorical' .* does not support reduction 'var'1category dtype does not support aggregation 'var'rrr}r=)rr~rrrrrR) rLr r"rNusing_copy_on_writer<r+rOrMrJrKrrPs rtest_groupby_raises_categoryrs B "< 4 4D r  B W : % %r:.. . . . Fv zv zv v   L v v Jv  ) , : v  ) , : !v,  ) , ; -v8  ) , : 9vD   S EvL MvN  #   '  [v\ ]v^ *_v` *avb  cvvd zevf   HHHH   gvx   HHJJ   yvJ zKvL *MvN :OvP   S QvX LMYvZ Y =>[v\  ]v^   HHGG   _vp qvr  svt   HHFD   uvF   HHGG   GvX  JKYvZ   HHGG   [vvlmvJE3px):xx{888E3RtXFFFFFrc|}||}|r|d}d}tjtd5t |||ddddS#1swxYwYdS)NrTr'c tdrrrs rrz.test_groupby_raises_category_udf..func4rrrr?r)rLr r"r<r+rMrs r test_groupby_raises_category_udfr+s B r  B W... y(< = = =Crc |}||}|r|d}tjtdftjtdfi|\}}|rd} nd} t |||||d| dS) NrTr'rrrrr rr) rLr r"rr<r+rMrJrKrPs rtest_groupby_raises_category_npr;s B r  B W KL   @   JE3736E3R"xPPPPPPrc |}tgdgdd|d<t||}|||} |r"| d} |dkrt| drJdS| o/t d | jD} |sD|d kr>t|tr)t|d tr|dd gkr| rJd} |d krd } idddddddtdfdddddttfdfdttfdfdttfdfdttfdfdtdfddd |s td!fndd"dd#| r td$fndd%| r td$fndd&did'dd(td)fd*td+fd,dd-dd.dd/tdfd0td1fd2td3fd4dd5td6 d7d8gfd9dd:dd;td6 dtd6 d?d@gfdAtdBfdCtd6 dDdEgf|\} } |d kr |rdFndG} | dH}ndI}t| | || |||dS)JNr8r%Tr9r)r observedr'rUc3$K|] }|jV dSr)empty).0groups r z;test_groupby_raises_category_on_category..qs$'T'T 'T'T'T'T'T'TrrGrr FrVrWrXrYrrZr[r\zz(cummax is not supported for category dtype|category dtype not supported|category type does not support cummax operations)r]zz(cummin is not supported for category dtype|category dtype not supported|category type does not support cummin operations)r^z|(cumprod is not supported for category dtype|category dtype not supported|category type does not support cumprod operations)r_zz(cumsum is not supported for category dtype|category dtype not supported|category type does not support cumsum operations)r`rarbrcrrdrez(empty group due to unobserved categoriesrfrgrhrirrjrrkrlrmrnrorrprrqrrrrrrsrtrurrrvrrrwrrxrrrrr}r=)rrr~rrXgroupsvalues isinstancer)r|rrrrrR)rLr r"rNrrr<r+rOrM empty_groupsrJrKrrPs r(test_groupby_raises_category_on_categoryrVs B555'''BsG #< 4 4D rH - -B W : % %r:.. . . . F<TC'T'TAQAQASAS'T'T'T$T$TL   ;   r4   r!uc " "  3*    k d zd zd d   L d d Jd  ) , @ d  ) , @ !d,  ) , A -d8  ) , @ 9dD 67EdF GdH  #   '  UdV WdX  :IJJ ]d^  :IJJ cdd  eddf zgdh PQidj 9TUkdl zmdn *odp :qdr y"<=sdt LMudv Y =>wdx  ydz   HHGG   {dL MdN  OdP   HHDF   Qdb   HHGG   cdt  JKudv   HHGG   wddHIdJE3Lx):xx{888E3RtXFFFFFrcBtdgdgdgd}d}tjt|5|dd}dddn #1swxYwYt jtd 5|d ddddS#1swxYwYdS) Nrr r)rr r&z+DataFrame.groupby with axis=1 is deprecatedr?r)axisz'Cannot subset columns when using axis=1r )rrDrErCr~rHrIr)r+rKrMs r%test_subsetting_columns_axis_1_raisesrs' !A3aS11 2 2B 7C #M = = =%% ZZ!Z $ $%%%%%%%%%%%%%%% z)R S S S 3s#AAA> BBB)r=)+r1rnumpyrrHpandasrrrrpandas._testing_testingrDpandas.tests.groupbyrfixturearraydictzipr(r r"r,r2r6r<rRmark parametrizerrrwrirrrrrrrrrrr rrrs    888888  c C###,,,-- SSq666 7 788***++ +++ , ,ff5P5P5P.Q.QR      e}%%%&%             6666( > > >??oGoG@?oGd 455  65  455*RVRW,=>> Q Q?>65 QF > > >??BPBP@?BPJ 455  65  455*RVRW,=>>QQ?>65Q,!C!C!CDD  ED  > > >??LGLG@?LG^ 455  65  455*RVRW,=>>QQ?>65Q2 > > >??TGTG@?TGnr