gVxddlZddlZddlZddlZddlmZmZmZm 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 l/var/www/html/nodeJS/PythonScripts/venv3.11/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 p|}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|\}} |dkr |rd2nd3} | d4} nd5} t|| ||||| dS)6Nr 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 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->rrGroupBy.fillna is deprecatedr=) rgroupbyhasattr TypeErrorNotImplementedErrorreescape ValueErrorrR) rLr r"rNr,r+rOrMrJrKkindrPs rtest_groupby_raises_stringrxs8 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 NO_;`  a;b  ?@c;d e;f  g;h @Ai;j  ?@k;l zm;n   I= > > o;;vw;JE3zx):xx{888E3RtXFFFFFrrBrGc|}||}|r|d}d}tjtd5t |||ddddS#1swxYwYdS)NrTr'c tdNTest error messager}rs rfuncz,test_groupby_raises_string_udf..func,---rrr?r{rHrIr}rF)rLr r"r,r+rMrs rtest_groupby_raises_string_udfrs B r  B W... y(< = = =CA((A,/A,groupby_func_npc |}||}|r|d}tjdtjtt jdfi|\}}|rd} nd} t|||||d| dS) NrTr'rWrjusing SeriesGroupBy.[sum|mean]!using DataFrameGroupBy.[sum|mean]r rP)r{nprxrir}rrrR) rLr r"rr,r+rMrJrKrPs rtest_groupby_raises_string_nprs B r  B W     IE F F   JE3736E3R"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`rbrcrdrerfrgrhrirkrlrmrnroz/cannot perform __truediv__ with this index typerpz%datetime64 type does not support prodrqrrrsrtrurv|z1dtype datetime64\[ns\] does not support reductionz0datetime64 type does not support skew operationsrwrx/datetime64 type does not support sum operationsryz/datetime64 type does not support var operations)rXrV'z&' with datetime64 dtypes is deprecatedrrrzr=r)rr{r|r}joinrR) 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..funcDrrrr?r)rLr r"r2r+rMrs r test_groupby_raises_datetime_udfr<s B r  B W... y(< = = =Crc |}||}|r|d}tjtdftjdi|\}}|rd} nd} t |||||d| dS) NrTr'rrWrrr rr{rrxr}rirR) rLr r"rr2r+rMrJrKrPs rtest_groupby_raises_datetime_nprKs B r  B W MN JE3 736E3R"xPPPPPPrr)rpr^rvrych|}|d}ttdd||gdS)NrrTz/timedelta64 type does not support .* operationsrA)r{rRr})rr6r+rMs rtest_groupby_raises_timedeltarcsH 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'rkz4'Categorical' .* does not support reduction 'median'4category dtype does not support aggregation 'median'rlrmrnrozDunsupported operand type\(s\) for /: 'Categorical' and 'Categorical'rp.category type does not support prod operationsrqzNo matching signature foundrrrs1'Categorical' .* does not support reduction 'sem'1category dtype does not support aggregation 'sem'rtrurv0dtype category does not support reduction 'skew'.category type does not support skew operationsrw1'Categorical' .* does not support reduction 'std'1category dtype does not support aggregation 'std'rx-category type does not support sum operationsry1'Categorical' .* does not support reduction 'var'1category dtype does not support aggregation 'var'rrrzr=)rr{r|r}r~rrR) rLr r"rNusing_copy_on_writer<r+rOrMrJrKrrPs rtest_groupby_raises_categoryrrs 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..func rrrr?r)rLr r"r<r+rMrs r test_groupby_raises_category_udfrs 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_nprs 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}nd3}t| | || |||dS)INr8r%Tr9r)r observedr'rUc3$K|] }|jV dSr)empty).0groups r z;test_groupby_raises_category_on_category..Hs$'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 categoriesrfrgrhrirrkrrlrmrnrorprrqr=rrrsrrrrtrurvrrrwrrrxrryrrrrrz)rrr{r|rXgroupsvalues isinstancer)strr}r~rrrR)rLr r"rNrrr<r+rOrM empty_groupsrJrKrrPs r(test_groupby_raises_category_on_categoryr-s 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 YOwdx  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 parametrizerrrxrirrrrrrrrrrr rrrs    888888  c C###,,,-- SSq666 7 788***++ +++ , ,ff5P5P5P.Q.QR      e}%%%&%             6666( > > >??PGPG@?PGf 455  65  455*RVRW,=>>QQ?>65Q2 > > >??BPBP@?BPJ 455  65  455*RVRW,=>>QQ?>65Q,!C!C!CDD  ED  > > >??LGLG@?LG^ 455  65  455*RVRW,=>>QQ?>65Q2 > > >??TGTG@?TGnr