o _~#gV@sddlZddlZddlZddlZddlmZmZmZm Z ddl m Z ddl mZejddgddgeddddgd egd eeed gd e gd e gd e gd gg d ddZejddgd ddZejddZejddZejddZejddZd>ddZejdgd d!d"Zejdd#d$gd%d&Zejdd#d$gejd'ejej gd(d)Z!ejdgd d*d+Z"ejdd#d$gd,d-Z#ejdd#d$gejd'ejej gd.d/Z$ejd0gd1d2d3Z%ejdgd d4d5Z&ejdd#d$gd6d7Z'ejdd#d$gejd'ejej gd8d9Z(ejdgd d:d;Z)dsr) rrrr r r r rrrrrr r r r rrrrrrrr)paramscC|jSNparamrequestr r rbysrTFcCrrrrr r rgroupby_series'srcCs&tgdgdtdtdd}|S)Nrrr xyzwtyuiorrcd)rrangelistdfr r rdf_with_string_col,sr'c Cs4tgdgdtdtddddddd d }|S) Nrrrir i`=r )rr#datetimer%r r rdf_with_datetime_col9sr,cCs*tgdgdtdtjddd}|S)Nrrrr)daysr )rr#r+ timedeltar%r r rdf_with_timedelta_colFs r/c Cs4tgdgdtdtgdgdddd}|S)Nrrr rrrrrrrrr!r T categoriesordered)rr#rr%r r rdf_with_cat_colSs r4c CsH|dkrdnt}tj||d|dur8|dkr t|||nb|dkr.|j|g|Rn\|j|g|RnBtj||d+|dkrLt|||n|dkrZ|j|g|Rn |j|g|RWdn(1smwYWddSWddSWddSWddSWddS1swYdS)Nr5matchmethodagg) FutureWarningtmassert_produces_warninggetattrr9 transformpytestraises)klassmsghowgb groupby_funcargswarn_msg warn_klassr r r_call_and_checkds2 "rIrC)r8r9r>c Cs|}t||}|j|d}|r |d}|dkr t|drJdSidddddddtdfd dd dd ttfd fd ttfdfdttfdfdttfdfdtdfddddddddddddidddttdfdttdfd dd!dd"dd#tdfd$ttd%fd&td'fd(dd)td*fd+dd,dd-td*fd.td*fd/dd0ttd1f|\}} |dkr|rd2nd3} | d4} nd5} t|| ||||| dS)6Nrr"corrwithallNr5anybfillzCould 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 typeffillfillnafirstidxmaxidxminlastmaxmean-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]quantilez,cannot be performed against 'object' dtypes!ranksemz!could not convert string to floatshiftsizeskewstdsumvarz%agg function failed [how->var,dtype->rrGroupBy.fillna is deprecatedr5) rgroupbyhasattr TypeErrorNotImplementedErrorreescape ValueErrorrI) rCrrrEr'r&rFrDrArBkindrGr r rtest_groupby_raises_stringxs   #'()*+/01234567;=  ryr9r>cCf|}|j|d}|r|d}dd}tjtddt|||WddS1s,wYdS)NrJr"cStdNTest error messagersr r r rfuncz,test_groupby_raises_string_udf..funcr}r6rqr?r@rsr=)rCrrr'r&rDrr r rtest_groupby_raises_string_udf "rgroupby_func_npc Csf|}|j|d}|r|d}tjdtjttdfi|\}}|r$d} nd} t|||||d| ddS) NrJr"rMr`using SeriesGroupBy.[sum|mean]!using DataFrameGroupBy.[sum|mean]r rG)rqnprnr_rsrurvrI) rCrrrr'r&rDrArBrGr r rtest_groupby_raises_string_nps  rc Cs|}t||}|j|d}|r |d}|dkr t|drJdSidddddddtdfd dd dd dd dd tdfdtdfddddddddddddddidddddddddddddtdfd td!fd"dd#dd$dd%dd&dd'td(d)d*gfd+dd,td-fd.td/f|\}} |d0vrd1|d2} n|dkr|rd3nd4} | d5} nd6} t|| ||||| d7dS)8NrJr"rKrLrMrNrOz+cannot perform __mul__ with this index typerPrQrRrSrTz3datetime64 type does not support cumprod operationsrUz2datetime64 type does not support cumsum operationsrVrXrYrZr[r\r]r^r_rarbrcrdrez/cannot perform __truediv__ with this index typerfz%datetime64 type does not support prodrgrhrirjrkrl|z1dtype datetime64\[ns\] does not support reductionz0datetime64 type does not support skew operationsrmrn/datetime64 type does not support sum operationsroz/datetime64 type does not support var operations)rNrL'z&' with datetime64 dtypes is deprecatedrrrpr5r)rrqrrrsjoinrI) rCrrrEr,r&rFrDrArBrGrxr r rtest_groupby_raises_datetimes       ()*+-  rcCrz)NrJr"cSr{r|r~r r r rrDrz.test_groupby_raises_datetime_udf..funcr}r6r)rCrrr,r&rDrr r r test_groupby_raises_datetime_udf<rrc Cs`|}|j|d}|r|d}tjtdftjdi|\}}|r!d} nd} t|||||d| ddS) NrJr"rrMrrr rrqrrnrsr_rI) rCrrrr,r&rDrArBrGr r rtest_groupby_raises_datetime_npKs  rr)rfrTrlrocCs&|}|jdd}ttdd||gdS)NrrJz/timedelta64 type does not support .* operationsr8)rqrIrs)rr/r&rDr r rtest_groupby_raises_timedeltacs rc Cs|}t||}|j|d}|r |d}|dkr t|drJdSidddddddtdfd dd dd ttfd fd ttfdfdttfdfdttfdfdtdfddd|s`tdfndddddddddidddtddd gfd!tdd"d#gfd$dd%dd&dd'td(fd)td*fd+td,fd-dd.tdd/d0gfd1dd2dd3tdd4d5gfd6tdd7d8gfd9td:fd;tddnd?} | d@} ndA} t| | ||||| dS)BNrJr"rKrLrMrNrO=unsupported operand type\(s\) for \*: 'Categorical' and 'int'rPrQrRzz(category type does not support cummax operations|category dtype not supported|cummax is not supported for category dtype)rSzz(category type does not support cummin operations|category dtype not supported|cummin is not supported for category dtype)rTz|(category type does not support cumprod operations|category dtype not supported|cumprod is not supported for category dtype)rUzz(category type does not support cumsum operations|category dtype not supported|cumsum is not supported for category dtype)rVzDunsupported operand type\(s\) for -: 'Categorical' and 'Categorical'rXrYSCannot setitem on a Categorical with a new category \(0\), set the categories firstrZr[r\r]r^r_rz2'Categorical' .* does not support reduction 'mean'2category dtype does not support aggregation 'mean'raz4'Categorical' .* does not support reduction 'median'4category dtype does not support aggregation 'median'rbrcrdrezDunsupported operand type\(s\) for /: 'Categorical' and 'Categorical'rf.category type does not support prod operationsrgzNo matching signature foundrhri1'Categorical' .* does not support reduction 'sem'1category dtype does not support aggregation 'sem'rjrkrl0dtype category does not support reduction 'skew'.category type does not support skew operationsrm1'Categorical' .* does not support reduction 'std'1category dtype does not support aggregation 'std'rn-category type does not support sum operationsro1'Categorical' .* does not support reduction 'var'1category dtype does not support aggregation 'var'rrrpr5)rrqrrrsrtrrI) rCrrrEusing_copy_on_writer4r&rFrDrArBrxrGr r rtest_groupby_raises_categoryrs2    "&'./0123<EFGHLMNOXYZclmvx  rcCrz)NrJr"cSr{r|r~r r r rr rz.test_groupby_raises_category_udf..funcr}r6r)rCrrr4r&rDrr r r test_groupby_raises_category_udfs "rc Csd|}|j|d}|r|d}tjtdftjtdfi|\}}|r#d} nd} t|||||d| ddS) NrJr"rrrrr rr) rCrrrr4r&rDrArBrGr r rtest_groupby_raises_category_nps   rcCsz|}tgdgddd|d<t||}|j||d} |r.| d} |dkr.t| dr,JdS| ogfd?t d7 d@dAgfdBt dCfdDt d7 dEdFgf|\} } |d!kr/|r'dGndH} | dI}nd4}t| | || |||dS)JNr0r Tr1r)robservedr"rKcss|]}|jVqdSr)empty).0groupr r r Hsz;test_groupby_raises_category_on_category..r>rrFrLrMrNrOrrPrQrRzz(cummax is not supported for category dtype|category dtype not supported|category type does not support cummax operations)rSzz(cummin is not supported for category dtype|category dtype not supported|category type does not support cummin operations)rTz|(cumprod is not supported for category dtype|category dtype not supported|category type does not support cumprod operations)rUzz(cumsum is not supported for category dtype|category dtype not supported|category type does not support cumsum operations)rVrWrXrYrrZr[z(empty group due to unobserved categoriesr\r]r^r_rrarrbrcrdrerfrrgr5rhrirrrrjrkrlrrrmrrrnrrorrrrrp)rrrqrrrNgroupsvalues isinstancer$strrsrtrwrrI)rCrrrErrr4r&rFrD empty_groupsrArBrxrGr r r(test_groupby_raises_category_on_category-s:      "#$+, / 23456789:;<=FGHQZ[d f rcCstdgdgdgd}d}tjt|d|jddd}Wdn1s&wYtjtd d |d WddS1sBwYdS) Nrr r)rrr!z+DataFrame.groupby with axis=1 is deprecatedr6r)axisz'Cannot subset columns when using axis=1r)rr;r<r:rqr?r@rw)r&rBrDr r r%test_subsetting_columns_axis_1_raisess "r)r5)+r+runumpyrr?pandasrrrrpandas._testing_testingr;pandas.tests.groupbyrfixturearraydictzipr#rrr'r,r/r4rImark parametrizeryrrnr_rrrrrrrrrrr r r rsx            S   E