o $j\@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)NrrrZ 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 rT categoriesordered)rr"rr$r r rdf_with_cat_colSs r3c 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)Nr4matchmethodagg) FutureWarningtmassert_produces_warninggetattrr8 transformpytestraises)klassmsghowgb groupby_funcargswarn_msgZ warn_klassr r r_call_and_checkds2 "rGrB)r7r8r=c CsH|}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|\} } |r|d2vrd3|d4} |d5vrt} n0|d#kr|djj d6krd7} n!|dkr|djj d6krd8} n|d9vr| d:d;} n|dkrd<} |dkr|rd=nd>} | d?} nd@} t | | ||||| dS)ANrr!corrwithallNr4anybfillzCould 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 typeffillfillnafirstidxmaxidxminlastmaxmeanz-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 floatshiftsizeskewstdsumvarz%agg function failed [how->var,dtype->) rcr]r^rSrRrjrfrlrirdz(dtype 'str' does not support operation '')rfrjriZpyarrowzBoperation 'truediv' not supported for dtype 'str' with dtype 'str'z>operation 'sub' not supported for dtype 'str' with dtype 'str')rQrPobjectstrz1Cannot perform reduction 'mean' with string dtyperrGroupBy.fillna is deprecatedr4) rgroupbyhasattr TypeErrorNotImplementedErrorreescape ValueErrorZdtypeZstoragereplacerG) rBrrrDr&using_infer_stringr%rErCr@rAkindrFr r rtest_groupby_raises_stringxs   #'()*+/01234567;=    r{r8r=cCf|}|j|d}|r|d}dd}tjtddt|||WddS1s,wYdS)NrHr!cStdNTest error messagersr r r rfuncz,test_groupby_raises_string_udf..funcrr5rqr>r?rsr<)rBrrr&r%rCrr r rtest_groupby_raises_string_udf "rgroupby_func_npc Csv|}|j|d}|r|d}tjdtjtdfi|\}} |r'|tjur%t}d} |r,d} nd} t|| |||d| d dS) NrHr!rKzEagg function failed|Cannot perform reduction 'mean' with string dtypez-dtype 'str' does not support operation 'mean'using SeriesGroupBy.[sum|mean]!using DataFrameGroupBy.[sum|mean]r rF)rqnprkr]rsrG) rBrrrr&ryr%rCr@rArFr 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)8NrHr!rIrJrKrLrMz+cannot perform __mul__ with this index typerNrOrPrQrRz3datetime64 type does not support cumprod operationsrSz2datetime64 type does not support cumsum operationsrTrVrWrXrYrZr[r\r]r^r_r`rarbz/cannot perform __truediv__ with this index typercz%datetime64 type does not support prodrdrerfrgrhri|z1dtype datetime64\[ns\] does not support reductionz0datetime64 type does not support skew operationsrjrk/datetime64 type does not support sum operationsrlz/datetime64 type does not support var operations)rLrJrmz&' with datetime64 dtypes is deprecatedrrrpr4r)rrqrrrsjoinrG) rBrrrDr+r%rErCr@rArFrzr r rtest_groupby_raises_datetimes       ()*+-  rcCr|)NrHr!cSr}r~rr r r rrmrz.test_groupby_raises_datetime_udf..funcrr5r)rBrrr+r%rCrr r r test_groupby_raises_datetime_udferrc Cs`|}|j|d}|r|d}tjtdftjdi|\}}|r!d} nd} t|||||d| ddS) NrHr!rrKrrr rrqrrkrsr]rG) rBrrrr+r%rCr@rArFr r rtest_groupby_raises_datetime_npts  rr)rcrRrirlcCs&|}|jdd}ttdd||gdS)NrrHz/timedelta64 type does not support .* operationsr7)rqrGrs)rr.r%rCr r rtest_groupby_raises_timedeltas 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)BNrHr!rIrJrKrLrM=unsupported operand type\(s\) for \*: 'Categorical' and 'int'rNrOrPzz(category type does not support cummax operations|category dtype not supported|cummax is not supported for category dtype)rQzz(category type does not support cummin operations|category dtype not supported|cummin is not supported for category dtype)rRz|(category type does not support cumprod operations|category dtype not supported|cumprod is not supported for category dtype)rSzz(category type does not support cumsum operations|category dtype not supported|cumsum is not supported for category dtype)rTzDunsupported operand type\(s\) for -: 'Categorical' and 'Categorical'rVrWSCannot setitem on a Categorical with a new category \(0\), set the categories firstrXrYrZr[r\r]rz2'Categorical' .* does not support reduction 'mean'2category dtype does not support aggregation 'mean'r^z4'Categorical' .* does not support reduction 'median'4category dtype does not support aggregation 'median'r_r`rarbzDunsupported operand type\(s\) for /: 'Categorical' and 'Categorical'rc.category type does not support prod operationsrdNo matching signature foundrerf1'Categorical' .* does not support reduction 'sem'1category dtype does not support aggregation 'sem'rgrhri0dtype category does not support reduction 'skew'.category type does not support skew operationsrj1'Categorical' .* does not support reduction 'std'1category dtype does not support aggregation 'std'rk-category type does not support sum operationsrl1'Categorical' .* does not support reduction 'var'1category dtype does not support aggregation 'var'rrrpr4)rrqrrrsrtrrG) rBrrrDusing_copy_on_writer3r%rErCr@rArzrFr r rtest_groupby_raises_categorys2    "&'./0123<EFGHLMNOXYZclmvx  rcCr|)NrHr!cSr}r~rr r r rr4rz.test_groupby_raises_category_udf..funcrr5r)rBrrr3r%rCrr r r test_groupby_raises_category_udf+s "rc Csd|}|j|d}|r|d}tjtdftjtdfi|\}}|r#d} nd} t|||||d| ddS) NrHr!rrrrr rr) rBrrrr3r%rCr@rArFr r rtest_groupby_raises_category_np;s   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}ndJ}t| | || |||dS)KNr/rTr0r)robservedr!rIcss|]}|jVqdSr)empty).0groupr r r qsz;test_groupby_raises_category_on_category..r=rrFrJrKrLrMrrNrOrPzz(cummax is not supported for category dtype|category dtype not supported|category type does not support cummax operations)rQzz(cummin is not supported for category dtype|category dtype not supported|category type does not support cummin operations)rRz|(cumprod is not supported for category dtype|category dtype not supported|category type does not support cumprod operations)rSzz(cumsum is not supported for category dtype|category dtype not supported|category type does not support cumsum operations)rTrUrVrWrrXrYz(empty group due to unobserved categoriesrZr[r\r]rr^rr_r`rarbrcrrdrrerfrrrrgrhrirrrjrrrkrrlrrrrrpr4)rrrqrrrLgroupsvalues isinstancer#rorsrtrwrrG)rBrrrDrrr3r%rErCZ empty_groupsr@rArzrFr r r(test_groupby_raises_category_on_categoryVs:      "#$+, / 23456789:;<=FGHQZ[d f rcCstdgdgdgd}d}tjt|d|jddd}Wdn1s&wYtjtd d |d WddS1sBwYdS) Nrr r)rrr z+DataFrame.groupby with axis=1 is deprecatedr5r)Zaxisz'Cannot subset columns when using axis=1r)rr:r;r9rqr>r?rw)r%rArCr r r%test_subsetting_columns_axis_1_raisess "r)r4)+r*runumpyrr>ZpandasrrrrZpandas._testingZ_testingr:Zpandas.tests.groupbyrZfixturearraydictzipr"rrr&r+r.r3rGmarkZ parametrizer{rrkr]rrrrrrrrrrr r r rsx            r  # E