o[weeRvddlZddlZddlZddlZddlmZmZmZm Z ddl m Z ddl mZejddgddgeddgdejgdeeed gde gd e gd e gd gg d Zejddg dZejdZejdZejdZejdZdZejdgddZejdddgdZejdddgejdejej gdZ!ejdgddZ"ejdddgdZ#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 @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 Ircd|A|dkrt|||dS|dkr|j|g|RdS|j|g|RdStj||5|dkrt|||n#|dkr|j|g|Rn|j|g|RddddS#1swxYwYdS)Nmethodaggmatch)getattrr? transformpytestraises)klassmsghowgb groupby_funcargss r_call_and_checkrLdsR } (?? %GB % %t , , , , E\\ BF< '$ ' ' ' ' ' ' BL - - - - - - - ]5 , , , 2 2h)L))4000|+d+++++ \1D1111  2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2s>B%%B),B)rH)r>r?rCcF|}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|\}} t|| ||||dS)2Nr r'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 ValueErrorrL) rHr r"rJr,r+rKrIrFrGs rtest_groupby_raises_stringr|vs 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   ID E E o;;vw;JE3zE3Rt<<<<.func,---rrr@rurDrErwrB)rHr r"r,r+rIrs rtest_groupby_raises_string_udfrs B r  B W... y(< = = =CA((A,/A,groupby_func_npc `|}||}|r|d}tjdtjtt jdfi|\}}|rd} nd} tjt| 5t|||||dddddS#1swxYwYdS) NrNr'rQreusing SeriesGroupBy.[sum|mean]!using DataFrameGroupBy.[sum|mean]r@r ) runprsrdrwryrztmassert_produces_warning FutureWarningrL) rHr r"rr,r+rIrFrGwarn_msgs rtest_groupby_raises_string_nprs" B r  B W     IE F F   JE3736 #M B B BBBsC_bAAABBBBBBBBBBBBBBBBBBsB##B'*B'c |}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|\}} d} d0|d1} |d2vrt } t j| | 35t|| ||||ddddS#1swxYwYdS)4NrNr'rOrPrQrSrTz+cannot perform __mul__ with this index typerUrVrWrXrYz3datetime64 type does not support cumprod operationsrZz2datetime64 type does not support cumsum operationsr[r]r^r_r`rarbrcrdrfrgrhrirjz/cannot perform __truediv__ with this index typerkz%datetime64 type does not support prodrlrmrnrorprq|z1dtype datetime64\[ns\] does not support reductionz0datetime64 type does not support skew operationsrrrs/datetime64 type does not support sum operationsrtz/datetime64 type does not support var operations'z&' with datetime64 dtypes is deprecated)rSrPr@) rrurvrwjoinrrrrL) rHr r"rJr2r+rKrIrFrGwarnrs rtest_groupby_raises_datetimers7 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 DG<GGGH~%% #D 9 9 9AAsC\4@@@AAAAAAAAAAAAAAAAAAs6EEEc|}||}|r|d}d}tjtd5t |||ddddS#1swxYwYdS)NrNr'c tdrrrs rrz.test_groupby_raises_datetime_udf..func=rrrr@r)rHr r"r2r+rIrs r test_groupby_raises_datetime_udfr5s B r  B W... y(< = = =Crc <|}||}|r|d}tjtdftjdi|\}}|rd} nd} t jt| 5t|||||dddddS#1swxYwYdS) NrNr'rrQrrr@r rurrsrwrdrrrrL) rHr r"rr2r+rIrFrGrs rtest_groupby_raises_datetime_nprDs B r  B W MN JE3 736 #M B B BBBsC_bAAABBBBBBBBBBBBBBBBBBs/BBBr)rkrYrqrtch|}|d}ttdd||gdS)NrrNz/timedelta64 type does not support .* operationsr>)rurLrw)rr6r+rIs rtest_groupby_raises_timedeltar]sH 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;tddNrNr'rOrPrQrSrT=unsupported operand type\(s\) for \*: 'Categorical' and 'int'rUrVrWzz(category type does not support cummax operations|category dtype not supported|cummax is not supported for category dtype)rXzz(category type does not support cummin operations|category dtype not supported|cummin is not supported for category dtype)rYz|(category type does not support cumprod operations|category dtype not supported|cumprod is not supported for category dtype)rZzz(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'r]r^SCannot setitem on a Categorical with a new category \(0\), set the categories firstr_r`rarbrcrdrz2'Categorical' .* does not support reduction 'mean'2category dtype does not support aggregation 'mean'rfz4'Categorical' .* does not support reduction 'median'4category dtype does not support aggregation 'median'rgrhrirjzDunsupported operand type\(s\) for /: 'Categorical' and 'Categorical'rk.category type does not support prod operationsrlzNo matching signature foundrmrn1'Categorical' .* does not support reduction 'sem'1category dtype does not support aggregation 'sem'rorprq0dtype category does not support reduction 'skew'.category type does not support skew operationsrr1'Categorical' .* does not support reduction 'std'1category dtype does not support aggregation 'std'rs-category type does not support sum operationsrt1'Categorical' .* does not support reduction 'var'1category dtype does not support aggregation 'var')rrurvrwrxrrL) rHr r"rJusing_copy_on_writer<r+rKrIrFrGs rtest_groupby_raises_categoryrls 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   [vvlmvJE3pE3Rt<<<<.funcrrrr@r)rHr r"r<r+rIrs r test_groupby_raises_category_udfrs B r  B W... y(< = = =Crc J|}||}|r|d}tjtdftjtdfi|\}}|rd} nd} t jt| 5t|||||dddddS#1swxYwYdS) NrNr'rrrrr@r r) rHr r"rr<r+rIrFrGrs rtest_groupby_raises_category_nprs B r  B W KL   @   JE3736 #M B B BBBsC_bAAABBBBBBBBBBBBBBBBBBs6BBBcB|}tgdgdd|d<t||}|||} |r"| d} |dkrt| drJdSt d | jD} id d d d d d dtdfdd dd dttfdfdttfdfdttfdfdttfdfdtdfdd d|s tdfnd dd d| r td fnd d!| r td"fnd d#d id$d d%td&fd'td(fd)d d*d d+d d,tdfd-td.fd/td0fd1d d2td3 d4d5gfd6d d7d d8td3 d9d:gfd;td3 dtd?fd@td3 dAdBgf|\} } t| | || ||dS)CNr8r%Tr9r)r observedr'rOc3$K|] }|jV dSr)empty).0groups r z;test_groupby_raises_category_on_category..>s$CCuu{CCCCCCrrPrQrSrTrrUrVrWzz(cummax is not supported for category dtype|category dtype not supported|category type does not support cummax operations)rXzz(cummin is not supported for category dtype|category dtype not supported|category type does not support cummin operations)rYz|(cumprod is not supported for category dtype|category dtype not supported|category type does not support cumprod operations)rZzz(cumsum is not supported for category dtype|category dtype not supported|category type does not support cumsum operations)r[r\r]r^rr_r`z*attempt to get argmax of an empty sequenceraz*attempt to get argmin of an empty sequencerbrcrdrrfrrgrhrirjrkrrlrRrmrnrrrrorprqrrrrrrrsrrtrr) rrrurvrSgroupsvaluesrwrxr{rrL) rHr r"rJrrr<r+rKrI empty_groupsrFrGs r(test_groupby_raises_category_on_categoryr#s- B555'''BsG #< 4 4D rH - -B W : % %r:.. . . . FCC 0@0@0B0BCCCCCLd zd zd d   L d d Jd  ) , @ d  ) , @ !d,  ) , A -d8  ) , @ 9dD 67EdF GdH  #   '  UdV WdX  :KLL ]d^  :KLL cdd  eddf zgdh PQidj 9TUkdl zmdn *odp :qdr y"<=sdt LMudv YOwdx  ydz   HHGG   {dL MdN  OdP   HHDF   Qdb   HHGG   cdt  JKudv   HHGG   wddHIdJE3LE3Rt<<<< BBB)+r1rynumpyrrDpandasrrrrpandas._testing_testingrpandas.tests.groupbyrfixturearraydictzipr(r r"r,r2r6r<rLmark parametrizer|rrsrdrrrrrrrrrrr rrrs     888888  c C###,,,-- SSq666 7 788***++ +++ , ,ff5P5P5P.Q.QR      e}%%%&%             222$ > > >??K=K=@?K=\ 455  65  455*RVRW,=>>BB?>65B4 > > >??AAAA@?AAH 455  65  455*RVRW,=>>BB?>65B.!C!C!CDD  ED  > > >??G=G=@?G=T 455  65  455*RVRW,=>>BB?>65B4 > > >??B=B=@?B=Jr