jwZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z ddl m ZddlmZej"ddgddge ddgd ej$gd eeed gd d e gd e gd e gdgg dZej"d dgdZej"dZej"dZej"dZ d*dZej8j;dgddZej8j;dddgdZej8j;dddgej8j;dej@ejBgdZ"ej8j;dgddZ#ej8j;dddgd Z$ej8j;dddgej8j;dej@ejBgd!Z%ej8j;d"gd#d$Z&ej8j;dgdd%Z'ej8j;dddgd&Z(ej8j;dddgej8j;dej@ejBgd'Z)ej8jUd(ej8j;dgdd)Z+y)+N)Pandas4Warning) Categorical DataFrameGrouperSeries)get_groupby_method_argsab)keyc |dzS)Nxs a/root/aria/tools/markitdown-venv/lib/python3.12/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 ==rcRtgdgdtdtdd}|S)Nrrr xyzwtyuior r cd)rrangelistdfs rdf_with_string_colr..s- ,,qk"   B Ircrtgdgdtdtjddddddd d }|S) Nrrrir i`=r')rr*datetimer,s rdf_with_datetime_colr4;s? ,,q""4Ar2r6B   B Irc `tgdgdtdtgdgddd}|S)Nrrr r r r r r r r r r(r'T categoriesordered)rr*rr,s rdf_with_cat_colr:Hs6 ,,q=/  B Irctj||d5|B|dk(rt|||n|dk(r|j|g|nu|j|g|nat j ||5|dk(rt|||n,|dk(r|j|g|n|j|g|ddddddy#1swYxYw#1swYyxYw)NF)matchcheck_stacklevelmethodaggr<)tmassert_produces_warninggetattrr? transformpytestraises)klassmsghowgb groupby_funcargs warn_categorywarn_msgs r_call_and_checkrOYs # #X 6 =h)L)40|+d+ \1D1uC0 6(?-GB -t4E\BFF</$/ BLL55  666 6 666s%AC5AC7CC CCrI)r>r?rDc $|}t||}|j|}|r|d}|dk(rt|drJyidddddddtdfd dd dd ttfd fd ttfdfdttfdfdttfdfdtdfddddddddddddidtt j dfdtt j dfddd dd!dd"tdfd#tt j d$fd%td&fd'dd(td)fd*dd+dd,td)fd-td)fd.td)fd/dd0tt j d1f|\} } |r{|d2vrd3|d4} |d5vrmt} nf|d"k(r|djjd6k(rd7} nB|dk(r|djjd6k(rd8} n|d9vr| jd:d;} n|dk(rd<} |dk(r t} d=} nd} d>} t| | ||||| | y)?Nr"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 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 deprecatedrU) rgroupbyhasattr TypeErrorNotImplementedErrorreescape ValueErrordtypestoragereplacerrO) rIr"r$rKr.using_infer_stringr-rLrJrGrHrMrNs rtest_groupby_raises_stringrps B "< 4D r B W : %r:. .. ; z; z; ; Y 34 ;  ; J ;  ) , V ;  ) , V ;  ) , W ;&  ) , V ';. 67/;0 1;2 3;4 *5;6 *7;8  9;: z;;<   IIE F =;D   IIG H E;L zM;N *O;P :Q;R y"<=S;T   IIE F U;\ Y VW];^  _;`  ?@a;b c;d  e;f @Ag;h @Ai;j  ?@k;l zm;n   II= > o;vw;JE3z   =\N!LC==! \ )bgmm.C.Cy.PWC V #3 (=(=(JSC 1 1++h.C Z 'ECz!& < E3Rt]HUrr?rDc|}|j|}|r|d}d}tjtd5t |||dddy#1swYyxYw)NrQr)ctdNTest error messager~rs rfuncz,test_groupby_raises_string_udf..func,--rrr@r|rErFr~rC)rIr"r$r.r-rJrs rtest_groupby_raises_string_udfrs_ B r B W. y(< =C AAgroupby_func_npc |}|j|}|r|d}tjdtjtdfi|\}} |r(|tjurt}d|j d} t || |||dy)NrQr)rTzXCould not convert string .* to numeric|Cannot perform reduction 'mean' with string dtypezCannot perform reduction 'z' with string dtyper)r|nprurfr~__name__rO) rIr"r$rr.rr-rJrGrHs rtest_groupby_raises_string_nprs B r B W     @ JE3 bgg %E*?+C+C*DDWXE3R"=rc j|}t||}|j|}|r|d}|dk(rt|drJyidtdfdtdfdd 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)j d*d+gfd,td)j d*d-gfd.d d/td0fd1td2f|\}} |dk(r t } d3} nd} d4} t || ||||| | y)5NrQr)rRrSz3'all' with datetime64 dtypes is no longer supportedrVz3'any' with datetime64 dtypes is no longer supportedrWrTz+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{rU)rr|r}r~joinrrO) rIr"r$rKr4r-rLrJrGrHrMrNs rtest_groupby_raises_datetimers B "< 4D r B W : %r:. .. 3 PQ3 PQ3 3 Y MN 3  3 J 3 *3 *3 IUV3 9ST3  3 3 3 *3 *3  !3" z#3$  %3& *'3( z)3* *+3, :-3. y"ST/30 OP132 J334  536 z738 93:  ;3<   HHHG  =3N   HHHG  O3` za3b  LMc3d  MNe3fg3JE3jz!& < E3Rt]HUrc|}|j|}|r|d}d}tjtd5t |||dddy#1swYyxYw)NrQr)ctdrrrs rrz.test_groupby_raises_datetime_udf..funckrrrr@r)rIr"r$r4r-rJrs r test_groupby_raises_datetime_udfrcs_ B r B W. y(< =Crc|}|j|}|r|d}tjtt j dftj di|\}}t|||||dy)NrQr)z/datetime64[us] does not support operation 'sum'rTr)r|rrur~rrrfrO) rIr"r$rr4r-rJrGrHs rtest_groupby_raises_datetime_nprrsw B r B W   IIG H     JE3E3R"=rr)rlr\rrrsrvctgdgdtdtjdd}|j d}t t d d ||gy) Nrrrr)daysr'r rQz/timedelta64 type does not support .* operationsr>)rr*r3 timedeltar|rOr~)rr-rJs rtest_groupby_raises_timedeltarsX ,,q##+   B s B9   rc |}t||}|j|}|r|d}|dk(rt|drJyidddddddtdfd dd dd ttfd fd ttfdfdttfdfdttfdfdtdfddddddddddddidtdj ddgfdtdj d d!gfd"dd#dd$dd%td&fd'td(fd)td*fd+dd,tdj d-d.gfd/dd0dd1tdj d2d3gfd4tdj d5d6gfd7tdj d8d9gfd:td;fdgf|\}} |dk(r t } d?} nd} d@} t|| ||||| | y)ANrQr)rRrSrTrVrW=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{rU)rr|r}r~rrrrO) rIr"r$rKr:r-rLrJrGrHrMrNs rtest_groupby_raises_categoryrs B "< 4D r B W : %r:. .. x zx zx x   L x x Jx  ) , : x  ) , : !x,  ) , ; -x8  ) , : 9xD   S ExL MxN OxP *QxR *SxT  UxV zWxX   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  _xpqxJE3tz!& < E3Rt]HUrc|}|j|}|r|d}d}tjtd5t |||dddy#1swYyxYw)NrQr)ctdrrrs rrz.test_groupby_raises_category_udf..func:rrrr@r)rIr"r$r:r-rJrs r test_groupby_raises_category_udfr1sa B r B W. y(< =Crc|}|j|}|r|d}tjtdftjtdfi|\}}t |||||dy)NrQr)z/dtype category does not support operation 'sum'z0dtype category does not support operation 'mean'r)r|rrur~rfrO) rIr"r$rr:r-rJrGrHs rtest_groupby_raises_category_nprAsq B r B W MN   >   JE3E3R"=rz$ignore:In a future version, the keysc |}tgdgdd|d<t||}|j||}|r|d}|dk(rt|drJy| xr*t d |j j D} |d k(rd } id d dd dd dtdfdd dd dttfdfdttfdfdttfdfdttfdfdtdfdd dd d| rtd fnd d!| rtd fnd d"d d#d id$td%fd&td'fd(d d)d d*d d+tdfd,td-fd.td/fd0d d1td2jd3d4gfd5d d6d d7td2jd8d9gfd:td2jd;dd?gfd@tdAfdBtd2jdCdDgf|\} } |dk(r t} dE} nd} dF} t| | ||||| | y)GNr6r'Tr7r )r"observedr)rRc34K|]}|jywr)empty).0groups r z;test_groupby_raises_category_on_category..rs'T 'TsrDFrSrTrVrWrrXrYrZzz(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{rU) rrr|r}rVgroupsvaluesr~rrrrrO)rIr"r$rKrr:r-rLrJ empty_groupsrGrHrMrNs r(test_groupby_raises_category_on_categoryrWs B5'BsG #< 4D rH -B W : %r:. .. <TC'TAQAQAS'T$TL k f zf zf f   L f f Jf  ) , @ f  ) , @ !f,  ) , A -f8  ) , @ 9fD 67EfF GfH IfJ  IJ OfP  IJ UfV  WfX zYfZ 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  {fLMfJE3Pz!& < E3Rt]HUrrT),r3rnumpyrrE pandas.errorsrpandasrrrrpandas._testing_testingrApandas.tests.groupbyrfixturearraydictzipr*r"r$r.r4r:rOmark parametrizerrrurfrrrrrrrrfilterwarningsrrrrrs   ( 8  c C#,- Sq6t DE*+ + ,f5P.QR   e}%&      "KM6. >?qV@qVh 45 6  45*RVVRWW,=>>?6>@ >?IV@IVX 45 6  45*RVVRWW,=>>?6>(!KLM* >?OV@OVd 45 6  45*RVVRWW,=>>?6>(BC >?LV@DLVr