ujT ddlZddlZddlZddlZddlmZddlZddl m Z ddl m Z ejddgddgejddgdej gdeeed gdd ej(gd ej(gd ej(gd gg dZejd dgdZejdZejdZejdZ d+dZej6j9dgddZej6j9dddgdZej6j9dddgej6j9dej>ej@gdZ!ej6j9dgddZ"ej6j9dddgdZ#ej6j9dddgej6j9dej>ej@gd Z$ej6j9d!gd"d#Z%ej6j9d!gd$d%Z&ej6j9dgdd&Z'ej6j9dddgd'Z(ej6j9dddgej6j9dej>ej@gd(Z)ej6jUd)ej6j9dgdd*Z+y),N)Pandas4Warning)get_groupby_method_argsab)keyc |dzS)Nxs k/var/www/html/python-recuperauto-other/env/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 r byrs$ ==rFc|jSrrrs r groupby_seriesr 's ==rcftjgdgdtdtdd}|S)Nrrr xyzwtyuiorrcd)pd DataFramerangelistdfs r df_with_string_colr,,s1 ,,qk"   B Irctjgdgdtdtjddddddd d }|S) Nrrrir i`=r#)r&r'r(datetimer*s r df_with_datetime_colr29sC ,,q""4Ar2r6B   B Irc tjgdgdtdtjgdgddd}|S)Nrrr rrrrrrrrr$r#T categoriesordered)r&r'r( Categoricalr*s r df_with_cat_colr9Fs> ,,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_checkrNWs # #X 6 =h)L)40|+d+ \1D1uC0 6(?-GB -t4E\BFF</$/ BLL55  666 6 666s%AC5AC7CC CCrH)r=r>rCc $|}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)?Nrr%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->) rkrerfr\r[rsrnrurqrrrlz(dtype 'str' does not support operation '')rnrsrqrrpyarrowzBoperation 'truediv' not supported for dtype 'str' with dtype 'str'z>operation 'sub' not supported for dtype 'str' with dtype 'str')rZrYobjectstrz1Cannot perform reduction 'mean' with string dtype'DataFrameGroupBy.corrwith is deprecatedrT) rgroupbyhasattr TypeErrorNotImplementedErrorreescape ValueErrordtypestoragereplacerrN) rHrr rJr,using_infer_stringr+rKrIrFrGrLrMs r test_groupby_raises_stringrns 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>rCc|}|j|}|r|d}d}tjtd5t |||dddy#1swYyxYw)NrPr%ctdNTest error messager}r s r funcz,test_groupby_raises_string_udf..func,--rrr?r{rDrEr}rB)rHrr r,r+rIrs r test_groupby_raises_string_udfrs_ B r B W. y(< =C AAgroupby_func_npc |}|j|}|r|d}tjdtjtdfi|\}} |r(|tjurt}d|j d} t || |||dy)NrPr%rSzQCould not convert .* to numeric|Cannot perform reduction 'mean' with string dtypezCannot perform reduction 'z' with string dtyper )r{nprtrer}__name__rN) rHrr rr,rr+rIrFrGs r test_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)5NrPr%rQrRz-'all' with datetime64 dtypes is not supportedrUz-'any' with datetime64 dtypes is not supportedrVrSz+cannot perform __mul__ with this index typerWrXrYrZr[z4datetime64 type does not support operation 'cumprod'r\z3datetime64 type does not support operation 'cumsum'r]r_r`rarbrcrdrerfrgrhrirjz/cannot perform __truediv__ with this index typerkz1datetime64 type does not support operation 'prod'rlrmrnrorprq|z1dtype datetime64\[ns\] does not support operationz1datetime64 type does not support operation 'skew'rrz1datetime64 type does not support operation 'kurt'rsrtz/datetime64 type does not support operation 'sumruz0datetime64 type does not support operation 'var'rzrT)rr{r|r}joinrrN) rHrr rJr2r+rKrIrFrGrLrMs r test_groupby_raises_datetimers B "< 4D r B W : %r:. .. 3 JK3 JK3 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)NrPr%ctdrrr s r rz.test_groupby_raises_datetime_udf..funcirrrr?r)rHrr r2r+rIrs r test_groupby_raises_datetime_udfras_ B r B W. y(< =Crc|}|j|}|r|d}tjtt j dftj di|\}}t|||||dy)NrPr%z/datetime64[us] does not support operation 'sum'rSr )r{rrtr}rrrerN) rHrr rr2r+rIrFrGs r test_groupby_raises_datetime_nprpsw B r B W   IIG H     JE3E3R"=rr)rUrRrsrnrqrbractjtjdd}tjt |d5t |jgd|dddy#1swYyxYw)Nrrz is not supported for intervalr?)rrrr)r&Seriesinterval_rangerDrEr}rBr{)rsers r test_groupby_raises_intervalrsb ))B%%a+ ,C y4&0N(O P30 L)402333s #A44A=)rkr[rqrrructjgdgdtdtjdd}|j d}t td d ||gy) Nrrrr)daysr#rrPz/timedelta64 type does not support .* operationsr=)r&r'r(r1 timedeltar{rNr})rr+rIs r test_groupby_raises_timedeltars\ ,,q##+   B s B9   rc |}t||}|j|}|r|d}|dk(rt|drJyidddddddtdfd dd dd ttfd fd ttfdfdttfdfdttfdfdtdfddddddddddddidtdfdtdfddd dd!dd"td#fd$td%fd&td'fd(dd)td*fd+dd,dd-td.fd/td0fd1td2fd3td4fd5td6f|\}} |dk(r t } d7} nd} d8} t || ||||| | y)9NrPr%rQrRrSrUrV=unsupported operand type\(s\) for \*: 'Categorical' and 'int'rWrXrYzz(category type does not support cummax operations|category dtype not supported|cummax is not supported for category dtype)rZzz(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_r`rarbrcrdre2category dtype does not support aggregation 'mean'rf4category dtype does not support aggregation 'median'rgrhrirjzDunsupported operand type\(s\) for /: 'Categorical' and 'Categorical'rk.category type does not support prod operationsrlNo matching signature foundrmrn1category dtype does not support aggregation 'sem'rorprq.category type does not support skew operationsrr.category type does not support kurt operationsrs1category dtype does not support aggregation 'std'rt-category type does not support sum operationsru1category dtype does not support aggregation 'var'rzrT)rr{r|r}r~rrN) rHrr rJr9r+rKrIrFrGrLrMs r test_groupby_raises_categoryrs, B "< 4D r B W : %r:. .. U zU zU U   L U U JU  ) , : U  ) , : !U,  ) , ; -U8  ) , : 9UD   S EUL MUN OUP *QUR *SUT  UUV zWUX   @ YU`   B aUh ziUj *kUl :mUn   S oUv LMwUx Y =>yUz  {U|   ? }UD EUF  GUH   < IUP   < QUX   ? YU`  JKaUb   ? cUjkUJE3nz!& < E3Rt]HUrc|}|j|}|r|d}d}tjtd5t |||dddy#1swYyxYw)NrPr%ctdrrr s r rz.test_groupby_raises_category_udf..funcrrrr?r)rHrr r9r+rIrs r test_groupby_raises_category_udfrsa B r B W. y(< =Crc|}|j|}|r|d}tjtdftjtdfi|\}}t |||||dy)NrPr%z/dtype category does not support operation 'sum'z0dtype category does not support operation 'mean'r )r{rrtr}rerN) rHrr rr9r+rIrFrGs r test_groupby_raises_category_npr&sq B r B W MN   >   JE3E3R"=rz$ignore:In a future version, the keysc r|}tjgdgdd|d<t||}|j||}|r|d}|dk(rt |drJy| xr*t d |j jD} |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 d1td2fd3d d4d d5td6fd7td8fd9td:fd;tdf|\} } |dk(r t} d?} nd} d@} t| | ||||| | y)ANr4r#Tr5r)robservedr%rQc34K|]}|jywr)empty).0groups r z;test_groupby_raises_category_on_category..Ws'T 'TsrCFrRrSrUrVrrWrXrYzz(cummax is not supported for category dtype|category dtype not supported|category type does not support cummax operations)rZzz(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_r`raz(empty group due to unobserved categoriesrbrcrdrerrfrrgrhrirjrkrrlrrmrnrrorprqrrrrrsrrtrrurrzrT) r&r8rr{r|rUgroupsvaluesr}r~rrrN)rHrr rJrr9r+rKrI empty_groupsrFrGrLrMs r (test_groupby_raises_category_on_categoryr<s Bnn5'BsG #< 4D rH -B W : %r:. .. <TC'TAQAQAS'T$TL k M zM zM M   L M M JM  ) , @ M  ) , @ !M,  ) , A -M8  ) , @ 9MD 67EMF GMH IMJ  IJ OMP  IJ UMV  WMX zYMZ PQ[M\ 9TU]M^ z_M` *aMb :cMd y"<=eMf LMgMh Y =>iMj  kMl   ? mMt uMv  wMx   < yM@   < AMH   ? IMP  JKQMR   ? SMZ[MJE3^z!& < E3Rt]HUrrS),r1rnumpyrrD pandas.errorsrpandasr&pandas._testing_testingr@pandas.tests.groupbyrfixtureGrouperarraydictzipr(rrr r,r2r9rNmark parametrizerrrtrerrrrrrrrrfilterwarningsrr rr rs   (8  c  s#,- Sq6t DE -. BII1 2 BII1 2  "#"e}%&      "KM6. >?qV@qVh 45 6  45*RVVRWW,=>>?6>@ >?IV@IVX 45 6  45*RVVRWW,=>>?6>( D33!KLM* >?lV@lV^ 45 6  45*RVVRWW,=>>?6>(BC >?sV@DsVr