o dAO@sddlZddlZddlZddlmZmZmZmZddl m Z ej ddgddgeddddgd e gd e eed gd egd egd egd gg d ddZej ddgd ddZejdgdddZejdddgddZejdddgejdejejgddZejdgddd Zejdddgd!d"Zejdddgejdejejgd#d$Zejdgdd%d&Zejdddgd'd(Zejdddgejdejejgd)d*Zejdgdd+d,Zd-d.Z dS)/N) Categorical DataFrameGrouperSeries)get_groupby_method_argsab)keycCs|dS)Nxr r e/var/www/html/visualizacion-main/env/lib/python3.10/site-packages/pandas/tests/groupby/test_raises.pysr) rrrr r r r rrrrrr r r r rrrrrrrr)paramscC|jSNparamrequestr r rbysrTFcCrrrrr r rgroupby_series%srhow)methodagg transformc Cstgdgdtdtdd}t||}|j|d}|r/|d}|dkr/t|dr-JdSid d d d d d dtd fdd dd dttfdfdttfdfdttfdfdttfdfdtdfdd dd dd dtdfdtd fd!d id"d d#td$fd%td&fd'd d(d d)d d*tdfd+td,fd-td.fd/d d0td&fd1d d2d d3td&fd4td&fd5d d6td&f|\}}|dur|d7krt |||dS|d8kr|j |g|RdS|j |g|RdSt j ||d9>|d7krt |||n |d8kr|j |g|Rn|j |g|RWddSWddSWddS1s;wYdS):Nrrr xyzwtyuiorrcdrr&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 typeffillfillnafirstidxmaxz#'argmax' not allowed for this dtypeidxminz#'argmin' not allowed for this dtypelastmaxmean.Could not convert xy?z?w?t?y?u?i?o? to numericmedianz!could not convert string to floatminngroupnunique pct_changeprodz0can't multiply sequence by non-int of type 'str'quantilez,cannot be performed against 'object' dtypes!ranksemshiftsizeskewstdsumvarr r!match)rrangelistrgroupbyhasattr TypeErrorNotImplementedError ValueErrorgetattrr!r"pytestraises rrr groupby_funcdfargsgbklassmsgr r rtest_groupby_raises_string*s    !"#$%&'()*+,-./1  $rar!r"cCstgdgdtdtdd}|j|d}|r|d}dd }tjtd d t|||WddS1s;wYdS) Nrrrr#r$r'r&cStdNTest error messagerTr r r rfuncz,test_groupby_raises_string_udf..funcrdrN)rrPrQrRrXrYrTrWrrrr\r^rfr r rtest_groupby_raises_string_udfs "rigroupby_func_npcCstgdgdtdtdd}|j|d}|r|d}tjdtjtd fi|\}}|dur8t|||dSt j ||d t|||WddS1sRwYdS) Nrrrr#r$r'r&r*r>rN) rrPrQrRnprLr=rTrWrXrYrrrrjr\r^r_r`r r rtest_groupby_raises_string_nps(  "rmc CsVtgdgdtdtddddddd d }t||}|j|d }|r6|d }|d kr6t|d r4JdSiddddddd tdfdddddddddtdfdtdfddddddddddd dd!did"dd#dd$dd%dd&dd'dd(td)fd*td+fd,dd-dd.dd/dd0dd1td2fd3dd4td5fd6d|\}}|dur|d7krt|||dS|d8kr|j|g|RdS|j |g|RdSt j ||d9<|d7krt|||n|d8kr|j|g|Rn|j |g|RWddSWddSWddS1s$wYdS):Nrrrr `=r$r'r&r(r)r*r,r-z+cannot perform __mul__ with this index typer.r/r0r1r2z3datetime64 type does not support cumprod operationsr3z2datetime64 type does not support cumsum operationsr4r6r7r8r9r:r;r<r=r?r@rArBrCz/cannot perform __truediv__ with this index typerDz%datetime64 type does not support prodrErFrGrHrIrJz1dtype datetime64\[ns\] does not support reductionrKrL/datetime64 type does not support sum operationsrMr r!rN) rrPdatetimerrRrSrTrWr!r"rXrYrZr r rtest_groupby_raises_datetimes        !"#%$ruc Cstgdgdtdtddddddd d }|j|d }|r$|d }d d}tjtddt|||WddS1sBwYdS)Nrrrrnrrorprqrrr$r'r&cSrbrcrer r r rrf rgz.test_groupby_raises_datetime_udf..funcrdrN)rrPrtrRrXrYrTrWrhr r r test_groupby_raises_datetime_udfs "rvc Cstgdgdtdtddddddd d }|j|d }|r$|d }tjtd ftjdi|\}}|dur?t|||dSt j ||dt|||WddS1sYwYdS)Nrrrrnrrorprqrrr$r'r&rsr*rN) rrPrtrRrkrLrTr=rWrXrYrlr r rtest_groupby_raises_datetime_nps(  "rwc Cstgdgdtdtgdgdddd}t||}|j|d}|r6|d }|d kr6t|d r4JdSid d d d dd d tdfdd dd dttfdfdttfdfdttfdfdttfdfdtdfdd d|svtdfnd dd d d d!d d"d id#d d$td%fd&td'fd(d d)d d*d d+td,fd-td.fd/td0fd1d d2td3fd4d d5d d6td7fd8td3fd9td:fd;tdkr|j |g|RdS|j |g|RdSt j || d?>|d=krt |||n |d>kr!|j |g|Rn|j |g|RWddSWddSWddS1sFwYdS)@Nrrr rrrrrrrrr%r$T categoriesorderedr'r&r(r)r*r,r-=unsupported operand type\(s\) for \*: 'Categorical' and 'int'r.r/r0zz(category type does not support cummax operations|category dtype not supported|cummax is not supported for category dtype)r1zz(category type does not support cummin operations|category dtype not supported|cummin is not supported for category dtype)r2z|(category type does not support cumprod operations|category dtype not supported|cumprod is not supported for category dtype)r3zz(category type does not support cumsum operations|category dtype not supported|cumsum is not supported for category dtype)r4zDunsupported operand type\(s\) for -: 'Categorical' and 'Categorical'r6r7SCannot setitem on a Categorical with a new category \(0\), set the categories firstr8r9r:r;r<r=C'Categorical' with dtype category does not support reduction 'mean'r?E'Categorical' with dtype category does not support reduction 'median'r@rArBrCzDunsupported operand type\(s\) for /: 'Categorical' and 'Categorical'rD.category type does not support prod operationsrEzNo matching signature foundrFrG#Cannot cast object dtype to float64rHrIrJC'Categorical' with dtype category does not support reduction 'skew'rKrL-category type does not support sum operationsrMB'Categorical' with dtype category does not support reduction 'var'r r!rN)rrPrrrRrSrTrUrVrWr!r"rXrY) rrrr[using_copy_on_writer\r]r^r_r`r r rtest_groupby_raises_category1s"   "&'./01237;<=>BCDEFGHLMNRT  $rc Cstgdgdtdtgdgdddd}|j|d}|r$|d }d d }tjtd d t|||WddS1sBwYdS)Nrrrrxr$Tryr'r&cSrbrcrer r r rrfrgz.test_groupby_raises_category_udf..funcrdrN)rrPrrRrXrYrTrWrhr r r test_groupby_raises_category_udfs$ "rc Cstgdgdtdtgdgdddd}|j|d}|r$|d }tjtd ftjtd fi|\}}|durAt|||dSt j ||d t|||WddS1s[wYdS) Nrrrrxr$Tryr'r&rr~rN) rrPrrRrkrLrTr=rWrXrYrlr r rtest_groupby_raises_category_nps6  "rc Csttgdgdddgdtdtgdgdddd}t||}|j||d}|r>|d }|d kr>t|d rt d?f|\} } | dur |d@krt |||dS|dAkr|j |g|RdS|j|g|RdStj| | dB>|d@kr5t |||n |dAkrD|j |g|Rn|j|g|RWddSWddSWddS1siwYdS)CNrxr$Tryrr) rrrrrrr%r%r%)robservedr&r(css|]}|jVqdSr)empty).0groupr r r sz;test_groupby_raises_category_on_category..r)r*r,r-r|r.r/r0zz(cummax is not supported for category dtype|category dtype not supported|category type does not support cummax operations)r1zz(cummin is not supported for category dtype|category dtype not supported|category type does not support cummin operations)r2z|(cumprod is not supported for category dtype|category dtype not supported|category type does not support cumprod operations)r3zz(cumsum is not supported for category dtype|category dtype not supported|category type does not support cumsum operations)r4r5r6r7r}r8r9z*attempt to get argmax of an empty sequencer:z*attempt to get argmin of an empty sequencer;r<r=r~r?rr@rArBrCrDrrEr+rFrGrrHrIrJrrKrLrrMrr r!rN)rrrPrrRrSr,groupsvaluesrTrUrVrWr!r"rXrY) rrrr[rrr\r]r^ empty_groupsr_r`r r r(test_groupby_raises_category_on_categorys,   "#$+, / 2348<=>?@ABCDEFJKLP R    $rcCs`tdgdgdgd}|jddd}tjtdd |d WddS1s)wYdS) Nrr r)rrr%r)axisz'Cannot subset columns when using axis=1rNr)rrRrXrYrV)r\r^r r r%test_subsetting_columns_axis_1_raisests  "r)!rtnumpyrkrXpandasrrrrpandas.tests.groupbyrfixturearraydictziprPrrmark parametrizerarirLr=rmrurvrwrrrrrr r r rs\      U   I     "