a +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)dr) rrrr r r r rrrrrr r r r rrrrrrrr)paramscCs|jSNparamrequestr r rbysrTFcCs|jSrrrr 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 Cs|dkr dnt}tj||d|durp|dkr@t|||q|dkr\|j|g|Rq|j|g|Rnttj||dV|dkrt|||n.|dkr|j|g|Rn|j|g|RWdn1s0YWdn1s0YdS)Nr4matchmethodagg) FutureWarningtmassert_produces_warninggetattrr8 transformpytestraises)klassmsghowgb groupby_funcargswarn_msgZ warn_klassr r r_call_and_checkdsrGrB)r7r8r=c %Cs|}t||}|j|d}|r@|d}|dkr@t|drmean,dtype->object]z/agg function failed [how->median,dtype->object]z-agg function failed [how->prod,dtype->object]z4dtype 'object' does not support operation 'quantile'z!could not convert string to floatz%agg function failed [how->var,dtype->"allanyZbfillrIcountZcumcountcummaxcummincumprodcumsumdiffZffillfillnafirstZidxmaxZidxminlastmaxmeanmedianminZngroupZnunique pct_changeprodquantileZranksemshiftsizeskewstdsumvar) r]rYrZrSrRrcr_rerbr^z(dtype 'str' does not support operation '')r_rcrbr\ZpyarrowzBoperation 'truediv' not supported for dtype 'str' with dtype 'str'rTz>operation 'sub' not supported for dtype 'str' with dtype 'str')rQrPobjectstrz1Cannot perform reduction 'mean' with string dtyperUrrGroupBy.fillna is deprecatedr4) rgroupbyhasattr TypeErrorNotImplementedErrorreescape ValueErrorZdtypeZstoragereplacerG) rBrrrDr&using_infer_stringr%rErCr@rAkindrFr r rtest_groupby_raises_stringxs  ;=      rtr8r=cCsd|}|j|d}|r|d}dd}tjtddt|||Wdn1sV0YdS)NrHr!cSs tddSNTest error messagerlr r r rfuncsz,test_groupby_raises_string_udf..funcrvr5rjr>r?rlr<)rBrrr&r%rCrxr r rtest_groupby_raises_string_udfs rzgroupby_func_npc Csv|}|j|d}|r|d}tjdtjtdfi|\}} |rN|tjurJt}d} |rXd} nd} t|| |||d| d dS) NrHr!rJzEagg 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)rjnprdrYrlrG) rBrrr{r&rrr%rCr@rArFr r rtest_groupby_raises_string_nps(   rc #Cs|}t||}|j|d}|r@|d}|dkr@t|dr.funcrvr5ry)rBrrr+r%rCrxr r r test_groupby_raises_datetime_udfes rc Cs`|}|j|d}|r|d}tjtdftjdi|\}}|rBd} nd} t|||||d| ddS) NrHr!rrJr|r}r r~rjrrdrlrYrG) rBrrr{r+r%rCr@rArFr r rtest_groupby_raises_datetime_npts  rrx)r]rRrbrecCs&|}|jdd}ttdd||gdS)NrrHz/timedelta64 type does not support .* operationsr7)rjrGrl)rxr.r%rCr r rtest_groupby_raises_timedeltas rc &Csj|}t||}|j|d}|r@|d}|dkr@t|drdnd } | d!} nd"} t| | ||||| dS)#NrHr!rIrJ=unsupported operand type\(s\) for \*: 'Categorical' and 'int'zz(category type does not support cummax operations|category dtype not supported|cummax is not supported for category dtype)zz(category type does not support cummin operations|category dtype not supported|cummin is not supported for category dtype)z|(category type does not support cumprod operations|category dtype not supported|cumprod is not supported for category dtype)zz(category type does not support cumsum operations|category dtype not supported|cumsum is not supported for category dtype)zDunsupported operand type\(s\) for -: 'Categorical' and 'Categorical'SCannot setitem on a Categorical with a new category \(0\), set the categories firstrz2'Categorical' .* does not support reduction 'mean'2category dtype does not support aggregation 'mean'z4'Categorical' .* does not support reduction 'median'4category dtype does not support aggregation 'median'zDunsupported operand type\(s\) for /: 'Categorical' and 'Categorical'.category type does not support prod operationsNo matching signature found1'Categorical' .* does not support reduction 'sem'1category dtype does not support aggregation 'sem'0dtype category does not support reduction 'skew'.category type does not support skew operations1'Categorical' .* does not support reduction 'std'1category dtype does not support aggregation 'std'-category type does not support sum operations1'Categorical' .* does not support reduction 'var'1category dtype does not support aggregation 'var'rLrUrrrir4)rrjrkrlrmrrG) rBrrrDusing_copy_on_writer3r%rErCr@rArsrFr r rtest_groupby_raises_categorys       vx  rcCsd|}|j|d}|r|d}dd}tjtddt|||Wdn1sV0YdS)NrHr!cSs tddSrurwr r r rrx4sz.test_groupby_raises_category_udf..funcrvr5ry)rBrrr3r%rCrxr r r test_groupby_raises_category_udf+s rc Csd|}|j|d}|r|d}tjtdftjtdfi|\}}|rFd} nd} t|||||d| ddS) NrHr!rrr|r}r r~r) rBrrr{r3r%rCr@rArFr r rtest_groupby_raises_category_np;s   rc&Cs|}tgdgddd|d<t||}|j||d} |r\| d} |dkr\t| drXJdS| oxtd d | jD} |s|d krt|trt|d t r|dd gkr| rJd} |d krd} dddt dfddt t fdft t fdft t fdft t fdft dfd|st dfndd| r"t dfnd| r2t dfndddt dft dfdddt dft dft dfdt d ddgfddt d dd gft d d!d"gft d#ft d d$d%gfd&"|\} } |d'kr|rd(nd)} | d*}nd+}t| | || |||dS),Nr/rTr0r)robservedr!rIcss|] }|jVqdSr)empty).0groupr r r qrz;test_groupby_raises_category_on_category..r=rrFrJrzz(cummax is not supported for category dtype|category dtype not supported|category type does not support cummax operations)zz(cummin is not supported for category dtype|category dtype not supported|category type does not support cummin operations)z|(cumprod is not supported for category dtype|category dtype not supported|category type does not support cumprod operations)zz(cumsum is not supported for category dtype|category dtype not supported|category type does not support cumsum operations)rKrz(empty group due to unobserved categoriesrrrrrrrrrrrrrrrLrUrrrir4)rrrjrkrNgroupsvalues isinstancer#rhrlrmrprrG)rBrrrDrrr3r%rErCZ empty_groupsr@rArsrFr r r(test_groupby_raises_category_on_categoryVs          df  rcCstdgdgdgd}d}tjt|d|jddd}Wdn1sL0Ytjtd d|d Wdn1s0YdS) Nrr r)rrr z+DataFrame.groupby with axis=1 is deprecatedr5r)Zaxisz'Cannot subset columns when using axis=1r)rr:r;r9rjr>r?rp)r%rArCr r r%test_subsetting_columns_axis_1_raisess ,r)r4)+r*rnnumpyrr>ZpandasrrrrZpandas._testingZ_testingr:Zpandas.tests.groupbyrZfixturearraydictzipr"rrr&r+r.r3rGmarkZ parametrizertrzrdrYrrrrrrrrrrr r r rsv            r  # E