dAOddlZddlZddlZddlmZmZmZmZddl m Z ej ddgddgeddgdej gde eed gdegd egd egd gg d Zej ddg dZejdgddZejdddgdZejdddgejdejejgdZejdgddZejdddgdZejdddgejdejejgdZejdgddZejdddgdZejdddgejdejejgdZejdgddZd Z dS)!N) Categorical DataFrameGrouperSeries)get_groupby_method_argsab)keyc |dzS)Nxs \/var/www/html/t/fyr/venv311/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 =rhow)methodagg transformchtgdgdtdtdd}t||}||}|r"|d}|dkrt |drJdSid d d d d d dt d fdd dd dtt fdfdtt fdfdtt fdfdtt fdfdt dfdd dd dd dt dfdt d fd!d id"d d#t d$fd%t d&fd'd d(d d)d d*t dfd+t d,fd-t d.fd/d d0td&fd1d d2d d3t d&fd4td&fd5d d6t d&f|\}}|A|d7krt|||dS|d8kr|j |g|RdS|j |g|RdStj ||95|d7krt|||n#|d8kr|j |g|Rn|j |g|RddddS#1swxYwYdS):Nrrr xyzwtyuiorr cdr 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 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 r#r r" groupby_funcdfargsgbklassmsgs rtest_groupby_raises_stringrf*sH ,,,,,,qk""     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 9CD7/8 9CD9/:  ;//< z=/> LM?/@ 9ABA/B zC/D *E/F :G/H y"<=I/J NOK/L Y NOM/N  O/P  ?@Q/R S/T  U/V ?@W/X  ?@Y/Z z[/\  >?]//^_/JE3b } (?? %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>H''H+.H+r%r&cJtgdgdtdtdd}||}|r|d}d}t jt d 5t|||ddddS#1swxYwYdS) Nrrrr(r)r,r+c tdNTest error messagerYrs rfuncz,test_groupby_raises_string_udf..func,---rrjrS)rrUrVrWr]r^rYr\r#r r"rarcrls rtest_groupby_raises_string_udfros  ,,,,,,qk""     B r  B W... y(< = = =Cs1BBBgroupby_func_npctgdgdtdtdd}||}|r|d}tjdtjtd fi|\}}|t|||dStj || 5t|||ddddS#1swxYwYdS) Nrrrr(r)r,r+r/rCrS) rrUrVrWnprQrBrYr\r]r^r#r r"rprarcrdres rtest_groupby_raises_string_nprtsN ,,,,,,qk""     B r  B W   )MNJE3  }C))))) ]5 , , , . . GB  _ - - - . . . . . . . . . . . . . . . . . .s0CCCctgdgdtdtjddddddd d }t||}|| }|r"|d }|d krt |d rJdSiddddddd t dfdddddddddt dfdt dfddddddddddd dd!did"dd#dd$dd%dd&dd'dd(t d)fd*t d+fd,dd-dd.dd/dd0dd1t d2fd3dd4t d5fd6d|\}}|A|d7krt|||dS|d8kr|j|g|RdS|j |g|RdStj ||95|d7krt|||n#|d8kr|j|g|Rn|j |g|RddddS#1swxYwYdS):Nrrrr `=r)r,r+r-r.r/r1r2z+cannot perform __mul__ with this index typer3r4r5r6r7z3datetime64 type does not support cumprod operationsr8z2datetime64 type does not support cumsum operationsr9r;r<r=r>r?r@rArBrDrErFrGrHz/cannot perform __truediv__ with this index typerIz%datetime64 type does not support prodrJrKrLrMrNrOz1dtype datetime64\[ns\] does not support reductionrPrQ/datetime64 type does not support sum operationsrRr$r%rS) rrUdatetimerrWrXrYr\r%r&r]r^r_s rtest_groupby_raises_datetimer}s ,,,,,,q"4Ar2r6BB     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#: ;#<  =#> PQ?#@ zA#B  LMC#D zE##FG#JE3J } (?? %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>GGGc`tgdgdtdtjddddddd d }|| }|r|d }d }t jt d5t|||ddddS#1swxYwYdS)Nrrrrvrrwrxryrzr)r,r+c tdrirkrs rrlz.test_groupby_raises_datetime_udf..func rmrrjrS)rrUr|rWr]r^rYr\rns r test_groupby_raises_datetime_udfrs ,,,,,,q"4Ar2r6BB     B r  B W... y(< = = =Cs<B##B'*B'ctgdgdtdtjddddddd d }|| }|r|d }tjt d ftjdi|\}}|t|||dStj ||5t|||ddddS#1swxYwYdS)Nrrrrvrrwrxryrzr)r,r+r{r/rS) rrUr|rWrrrQrYrBr\r]r^rss rtest_groupby_raises_datetime_nprs] ,,,,,,q"4Ar2r6BB     B r  B W MN JE3  }C))))) ]5 , , , . . GB  _ - - - . . . . . . . . . . . . . . . . . .s;C""C&)C&c ~tgdgdtdtgdgddd}t||}||}|r"|d }|d krt |d rJdSid d d d dd d t dfdd dd dtt fdfdtt fdfdtt fdfdtt fdfdt dfdd d|s t dfnd dd d d d!d d"d id#d d$t d%fd&t d'fd(d d)d d*d d+t d,fd-t d.fd/t d0fd1d d2td3fd4d d5d d6t d7fd8td3fd9t d:fd;t dkr|j |g|RdS|j |g|RdStj || ?5|d=krt|||n#|d>kr|j |g|Rn|j |g|RddddS#1swxYwYdS)@Nrrr rrrrr r r r r*r)T categoriesorderedr,r+r-r.r/r1r2=unsupported operand type\(s\) for \*: 'Categorical' and 'int'r3r4r5zz(category type does not support cummax operations|category dtype not supported|cummax is not supported for category dtype)r6zz(category type does not support cummin operations|category dtype not supported|cummin is not supported for category dtype)r7z|(category type does not support cumprod operations|category dtype not supported|cumprod is not supported for category dtype)r8zz(category type does not support cumsum operations|category dtype not supported|cumsum is not supported for category dtype)r9zDunsupported operand type\(s\) for -: 'Categorical' and 'Categorical'r;r<SCannot setitem on a Categorical with a new category \(0\), set the categories firstr=r>r?r@rArBC'Categorical' with dtype category does not support reduction 'mean'rDE'Categorical' with dtype category does not support reduction 'median'rErFrGrHzDunsupported operand type\(s\) for /: 'Categorical' and 'Categorical'rI.category type does not support prod operationsrJzNo matching signature foundrKrL#Cannot cast object dtype to float64rMrNrOC'Categorical' with dtype category does not support reduction 'skew'rPrQ-category type does not support sum operationsrRB'Categorical' with dtype category does not support reduction 'var'r$r%rS)rrUrrrWrXrYrZr[r\r%r&r]r^) r#r r"r`using_copy_on_writerarbrcrdres rtest_groupby_raises_categoryr1s ,,,,,,q===///   B #< 4 4D r  B W : % %r:.. . . . FR zR zR R   L R R JR  ) , < R  ) , : !R,  ) , ; -R8  ) , : 9RD   S ERL MRN  #   )  [R\ ]R^ *_R` *aRb  cRRd zeRf   Q gRn   S oRv zwRx *yRz :{R|   S }RD LMERF Y =>GRH  IRJ  ABKRL MRN  ORP   Q QRX  ABYRZ  JK[R\   P ]RRdeRJE3h } (?? %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'>H22H69H6c Xtgdgdtdtgdgddd}||}|r|d }d }t jt d 5t|||ddddS#1swxYwYdS) Nrrrrr)Trr,r+c tdrirkrs rrlz.test_groupby_raises_category_udf..funcrmrrjrS)rrUrrWr]r^rYr\rns r test_groupby_raises_category_udfrs* ,,,,,,q===///   B r  B W... y(< = = =Cs8BB#&B#c tgdgdtdtgdgddd}||}|r|d }tjt d ftjt d fi|\}}|t|||dStj || 5t|||ddddS#1swxYwYdS) Nrrrrr)Trr,r+rrrS) rrUrrWrrrQrYrBr\r]r^rss rtest_groupby_raises_category_nprss ,,,,,,q===///   B r  B W KL   Q   JE3 }C))))) ]5 , , , . . GB  _ - - - . . . . . . . . . . . . . . . . . .s>C%%C),C)c 0ttgdgddgdtdtgdgddd}t||}|||}|r"|d }|d krt |d rJdSt d |jD} id d dd dd d tdfdd dd dttfdfdttfdfdttfdfdttfdfdtdfdd d|s tdfnd d d d!| r td"fnd d#| r td$fnd d%d id&d d'td(fd)td*fd+d d,d d-d d.tdfd/td0fd1td2fd3d d4td5fd6d d7d d8td9fd:td5fd;tdf|\} } | A|d?krt|||dS|d@kr|j |g|RdS|j|g|RdStj| | A5|d?krt|||n#|d@kr|j |g|Rn|j|g|RddddS#1swxYwYdS)BNrr)Trrr) rrrrr r r*r*r*)r observedr+r-c3$K|] }|jV dSr)empty).0groups r z;test_groupby_raises_category_on_category..s$CCuu{CCCCCCrr.r/r1r2rr3r4r5zz(cummax is not supported for category dtype|category dtype not supported|category type does not support cummax operations)r6zz(cummin is not supported for category dtype|category dtype not supported|category type does not support cummin operations)r7z|(cumprod is not supported for category dtype|category dtype not supported|category type does not support cumprod operations)r8zz(cumsum is not supported for category dtype|category dtype not supported|category type does not support cumsum operations)r9r:r;r<rr=r>z*attempt to get argmax of an empty sequencer?z*attempt to get argmin of an empty sequencer@rArBrrDrrErFrGrHrIrrJr0rKrLrrMrNrOrrPrQrrRrr$r%rS)rrrUrrWrXr1groupsvaluesrYrZr[r\r%r&r]r^) r#r r"r`rrrarbrc empty_groupsrdres r(test_groupby_raises_category_on_categoryrsD ===/// -,,q===///   B #< 4 4D rH - -B W : % %r:.. . . . FCC 0@0@0B0BCCCCCLP zP zP P   L P P JP  ) , B P  ) , @ !P,  ) , A -P8  ) , B 9PD 67EPF GPH  #   )  UPV WPX  :KLL ]P^  :KLL cPd  ePPf zgPh   Q iPp   S qPx zyPz *{P| :}P~ y"<=P@ LMAPB YOCPD  EPF  ABGPH IPJ  KPL   Q MPT  ABUPV  JKWPX   P YPP`aPJE3d } (?? %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>J  JJctdgdgdgd}|dd}tjtd5|d ddddS#1swxYwYdS) Nrr r)rr r*r)axisz'Cannot subset columns when using axis=1rSr )rrWr]r^r[)rarcs r%test_subsetting_columns_axis_1_raisesrts !A3aS11 2 2B Ca B z)R S S S 3s AA#&A#)!r|numpyrrr]pandasrrrrpandas.tests.groupbyrfixturearraydictziprUr r"mark parametrizerfrorQrBrtr}rrrrrrrr rrrs,   988888  c C###,,,-- SSq666 7 788***++ +++ , ,ff5P5P5P.Q.QR      e}%%%&% > > >??R2R2@?R2j 45565* 455*RVRW,=>>..?>65.6 > > >??F2F2@?F2R 45565, 455*RVRW,=>>..?>65.6 > > >??|2|2@?|2~ 455654 455*RVRW,=>>..?>65.D > > >??@2@2@?@2Fr