PiwZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z ddl m ZddlmZejddgddge ddgd ejgd eeed gd d e gd e gd e gdgg dZejd dgdZejdZejdZejdZ d+dZejdgddZejdddgdZejdddgejdej ej!gdZ"ejdgdd Z#ejdddgd!Z$ejdddgejdej ej!gd"Z%ejd#gd$d%Z&ejdgdd&Z'ejdddgd'Z(ejdddgejdej ej!gd(Z)ej*d)ejdgdd*Z+dS),N)Pandas4Warning) Categorical DataFrameGrouperSeries)get_groupby_method_argsab)keyc |dzS)Nxs t/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/groupby/test_raises.pyrs !a%) rrrr r r r T)strict rrrrrr r r r rrrrrrrr)paramsc|jSNparamrequests rbyr"s  =rFc|jSrrr s rgroupby_seriesr$)s =rcjtgdgdtdtdd}|S)Nrrr xyzwtyuior r cd)rrangelistdfs rdf_with_string_colr..sH ,,,,,,qk""     B Irctgdgdtdtjddddddd d }|S) Nrrrir i`=r')rr*datetimer,s rdf_with_datetime_colr4;sW ,,,,,,q"4Ar2r6BB     B Irc xtgdgdtdtgdgddd}|S)Nrrr r r r r r r r r r(r'T categoriesordered)rr*rr,s rdf_with_cat_colr:Hsd ,,,,,,q===///   B Irctj||d5|>|dkrt|||n|dkr|j|g|Rnz|j|g|Rnkt j||5|dkrt|||n#|dkr|j|g|Rn|j|g|Rdddn #1swxYwYddddS#1swxYwYdS)NF)matchcheck_stacklevelmethodaggr=)tmassert_produces_warninggetattrr@ transformpytestraises)klassmsghowgb groupby_funcargs warn_categorywarn_msgs r_call_and_checkrPYs #X   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/>B9- C9B= =CB= CCCrJ)r?r@rEc r|}t||}||}|r"|d}|dkrt|drJdSidddddddtdfd dd dd ttfd fd ttfdfdttfdfdttfdfdtdfddddddddddddidtt jdfdtt jdfddd dd!dd"tdfd#tt jd$fd%td&fd'dd(td)fd*dd+dd,td)fd-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 t} d=} nd} d>} t| | ||||| | dS)?Nr"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 typeffillfirstidxmaxidxminlastmaxmeanz-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 floatshiftsizeskewkurtstdsumvarz%agg function failed [how->var,dtype->) rlrfrgr]r\rtrorvrrrsrmz(dtype 'str' does not support operation '')rortrrrspyarrowzBoperation 'truediv' not supported for dtype 'str' with dtype 'str'z>operation 'sub' not supported for dtype 'str' with dtype 'str')r[rZobjectstrz1Cannot perform reduction 'mean' with string dtype'DataFrameGroupBy.corrwith is deprecatedr;) rgroupbyhasattr TypeErrorNotImplementedErrorreescape ValueErrordtypestoragereplacerrP) rJr"r$rLr.using_infer_stringr-rMrKrHrIrNrOs rtest_groupby_raises_stringrps 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 =;D   IG H H E;L zM;N *O;P :Q;R y"<=S;T   IE F F U;\ Y VW];^  _;`  ?@a;b c;d  e;f @Ag;h @Ai;j  ?@k;l zm;n   I= > > o;;vw;JE3zF   M\LLLC===! \ ) )bgm.Cy.P.PWCC V # #3 (=(J(JSCC 1 1 1++h..CC Z ' 'ECz!!& < E3Rt]HUUUUUrr@rEc|}||}|r|d}d}tjtd5t |||ddddS#1swxYwYdS)NrRr)c tdNTest error messager~rs rfuncz,test_groupby_raises_string_udf..func,---rrrAr|rFrGr~rD)rJr"r$r.r-rKrs rtest_groupby_raises_string_udfrs B r  B W... y(< = = =CA((A,/A,groupby_func_npc |}||}|r|d}tjdtjtdfi|\}} |r |tjurt}d|jd} t || |||ddS)NrRr)rUzXCould not convert string .* to numeric|Cannot perform reduction 'mean' with string dtypezCannot perform reduction 'z' with string dtyper)r|nprurfr~__name__rP) rJr"r$rr.rr-rKrHrIs rtest_groupby_raises_string_nprs B r  B W     @ JE3Y bg % %EX?+CXXXE3R"=====rc |}t||}||}|r"|d}|dkrt|drJdSidtdfdtdfdd dtd fd d d d d d dd dtdfdtdfdd dd dd dd dd dd dd idd dd dd dd dd dtd fd!td"fd#d d$d d%d d&d d'd d(td)d*d+gfd,td)d*d-gfd.d d/td0fd1td2f|\}} |dkr t } d3} nd} d4} t || ||||| | dS)5NrRr)rSrTz3'all' with datetime64 dtypes is no longer supportedrVz3'any' with datetime64 dtypes is no longer supportedrWrUz+cannot perform __mul__ with this index typerXrYrZr[r\z4datetime64 type does not support operation 'cumprod'r]z3datetime64 type does not support operation 'cumsum'r^r`rarbrcrdrerfrgrhrirjrkz/cannot perform __truediv__ with this index typerlz1datetime64 type does not support operation 'prod'rmrnrorprqrr|z1dtype datetime64\[ns\] does not support operationz1datetime64 type does not support operation 'skew'rsz1datetime64 type does not support operation 'kurt'rtruz/datetime64 type does not support operation 'sumrvz0datetime64 type does not support operation 'var'r{r;)rr|r}r~joinrrP) rJr"r$rLr4r-rMrKrHrIrNrOs rtest_groupby_raises_datetimers B "< 4 4D r  B W : % %r:.. . .. F3 PQ3 PQ3 3 Y MN 3  3 J 3 *3 *3 IUV3 9ST3  3 3 3 *3 *3  !3" z#33$  %3& *'3( z)3* *+3, :-3. y"ST/30 OP132 J334  536 z738 93:  ;3<   HHHG   =3N   HHHG   O3` za3b  LMc3d  MNe33fg3JE3jz!!& < E3Rt]HUUUUUrc|}||}|r|d}d}tjtd5t |||ddddS#1swxYwYdS)NrRr)c tdrrrs rrz.test_groupby_raises_datetime_udf..funckrrrrAr)rJr"r$r4r-rKrs r test_groupby_raises_datetime_udfrcs B r  B W... y(< = = =Crc|}||}|r|d}tjtt jdftjdi|\}}t|||||ddS)NrRr)z/datetime64[us] does not support operation 'sum'rUr)r|rrur~rrrfrP) rJr"r$rr4r-rKrHrIs rtest_groupby_raises_datetime_nprrs B r  B W   IG H H     JE3E3R"=====rr)rlr\rrrsrvctgdgdtdtjdd}|d}t t d d ||gdS) Nrrrr)daysr'r rRz/timedelta64 type does not support .* operationsr?)rr*r3 timedeltar|rPr~)rr-rKs rtest_groupby_raises_timedeltars ,,,,,,q#+++     B s  B9   rc $|}t||}||}|r"|d}|dkrt|drJdSidddddddtdfd dd dd ttfd fd ttfdfdttfdfdttfdfdtdfddddddddddddidtdddgfdtdd d!gfd"dd#dd$dd%td&fd'td(fd)td*fd+dd,tdd-d.gfd/dd0dd1tdd2d3gfd4tdd5d6gfd7tdd8d9gfd:td;fdgf|\}} |dkr t } d?} nd} d@} t|| ||||| | dS)ANrRr)rSrTrUrVrW=unsupported operand type\(s\) for \*: 'Categorical' and 'int'rXrYrZzz(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'r`rarbrcrdrerfrz2'Categorical' .* does not support operation 'mean'2category dtype does not support aggregation 'mean'rgz4'Categorical' .* does not support operation 'median'4category dtype does not support aggregation 'median'rhrirjrkzDunsupported operand type\(s\) for /: 'Categorical' and 'Categorical'rl.category type does not support prod operationsrmNo matching signature foundrnro1'Categorical' .* does not support operation 'sem'1category dtype does not support aggregation 'sem'rprqrr0dtype category does not support operation 'skew'.category type does not support skew operationsrs0dtype category does not support operation 'kurt'.category type does not support kurt operationsrt1'Categorical' .* does not support operation 'std'1category dtype does not support aggregation 'std'ru-category type does not support sum operationsrv1'Categorical' .* does not support operation 'var'1category dtype does not support aggregation 'var'r{r;)rr|r}r~rrrrP) rJr"r$rLr:r-rMrKrHrIrNrOs rtest_groupby_raises_categoryrs B "< 4 4D r  B W : % %r:.. . .. Fx zx zx x   L x x Jx  ) , : x  ) , : !x,  ) , ; -x8  ) , : 9xD   S ExL MxN OxP *QxR *SxT  UxV zWxxX   HHHH   Yxj   HHJJ   kx| z}x~ *x@ :AxB   S CxJ LMKxL Y =>MxN  OxP   HHGG   Qxb cxd  exf   HHFD   gxx   HHFD   yxJ   HHGG   Kx\  JK]x^   HHGG   _xxpqxJE3tz!!& < E3Rt]HUUUUUrc|}||}|r|d}d}tjtd5t |||ddddS#1swxYwYdS)NrRr)c tdrrrs rrz.test_groupby_raises_category_udf..func:rrrrAr)rJr"r$r:r-rKrs r test_groupby_raises_category_udfr1s B r  B W... y(< = = =Crc|}||}|r|d}tjtdftjtdfi|\}}t |||||ddS)NrRr)z/dtype category does not support operation 'sum'z0dtype category does not support operation 'mean'r)r|rrur~rfrP) rJr"r$rr:r-rKrHrIs rtest_groupby_raises_category_nprAs B r  B W MN   >   JE3E3R"=====rz$ignore:In a future version, the keysc |}tgdgdd|d<t||}|||}|r"|d}|dkrt|drJdS| o/t d |jD} |d krd } id d dd dd dtdfdd dd dttfdfdttfdfdttfdfdttfdfdtdfdd dd d| r td fnd d!| r td fnd d"d d#d id$td%fd&td'fd(d d)d d*d d+tdfd,td-fd.td/fd0d d1td2 d3d4gfd5d d6d d7td2 d8d9gfd:td2 d;dd?gfd@tdAfdBtd2 dCdDgf|\} } |dkr t} dE} nd} dF} t| | ||||| | dS)GNr6r'Tr7r )r"observedr)rSc3$K|] }|jV dSr)empty).0groups r z;test_groupby_raises_category_on_category..rs$'T'T 'T'T'T'T'T'TrrEFrTrUrVrWrrXrYrZzz(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^r_r`rarbz(empty group due to unobserved categoriesrcrdrerfrrgrrhrirjrkrlrrmrrnrorrrrprqrrrrrsrrrtrrrurrvrrr{r;) rrr|r}rVgroupsvaluesr~rrrrrP)rJr"r$rLrr:r-rMrK empty_groupsrHrIrNrOs r(test_groupby_raises_category_on_categoryrWsn B555'''BsG #< 4 4D rH - -B W : % %r:.. . .. F<TC'T'TAQAQASAS'T'T'T$T$TL k f zf zf f   L f f Jf  ) , @ f  ) , @ !f,  ) , A -f8  ) , @ 9fD 67EfF GfH IfJ  :IJJ OfP  :IJJ UfV  WfX zYffZ PQ[f\ 9TU]f^ z_f` *afb :cfd y"<=eff LMgfh Y =>ifj  kfl   HHGG   mf~ f@  AfB   HHDF   CfT   HHDF   Uff   HHGG   gfx  JKyfz   HHGG   {ffLMfJE3Pz!!& < E3Rt]HUUUUUrrU),r3rnumpyrrF pandas.errorsrpandasrrrrpandas._testing_testingrBpandas.tests.groupbyrfixturearraydictzipr*r"r$r.r4r:rPmark parametrizerrrurfrrrrrrrrfilterwarningsrrrrrs#   (((((( 888888  c C###,,,-- SSq666t D D DEE***++ +++ , ,ff5P5P5P.Q.QR      e}%%%&%         "KM6666. > > >??qVqV@?qVh 455  65  455*RVRW,=>>>>?>65>@ > > >??IVIV@?IVX 455  65  455*RVRW,=>>>>?>65>(!K!K!KLLML* > > >??OVOV@?OVd 455  65  455*RVRW,=>>>>?>65>(BCC > > >??LVLV@?DCLVLVLVr