a PfV@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 %Cs6|}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]z,cannot be performed against 'object' dtypes!z!could not convert string to floatz%agg function failed [how->var,dtype->"allanyZbfillrIcountZcumcountZcummaxZcummincumprodZcumsumdiffZffillfillnafirstZidxmaxZidxminlastmaxmeanZmedianminZngroupZnuniqueZ pct_changeprodZquantileZrankZsemshiftsizeskewZstdsumvarrSrrGroupBy.fillna is deprecatedr4) rgroupbyhasattr TypeErrorNotImplementedErrorreescape ValueErrorrG) rBrrrDr&r%rErCr@rAkindrFr r rtest_groupby_raises_stringxs  ;=  rhr8r=cCsd|}|j|d}|r|d}dd}tjtddt|||Wdn1sV0YdS)NrHr!cSs tddSNTest error messagerbr r r rfuncsz,test_groupby_raises_string_udf..funcrjr5r`r>r?rbr<)rBrrr&r%rCrlr r rtest_groupby_raises_string_udfs rngroupby_func_npc Csf|}|j|d}|r|d}tjdtjttdfi|\}}|rHd} nd} t|||||d| ddS) NrHr!rJrLusing SeriesGroupBy.[sum|mean]!using DataFrameGroupBy.[sum|mean]r rF)r`npr]rWrbrdrerG) rBrrror&r%rCr@rArFr r rtest_groupby_raises_string_nps  rtc #Cs|}t||}|j|d}|r@|d}|dkr@t|dr.funcrjr5rm)rBrrr+r%rCrlr r r test_groupby_raises_datetime_udf<s rzc Cs`|}|j|d}|r|d}tjtdftjdi|\}}|rBd} nd} t|||||d| ddS) NrHr!rvrJrprqr rrr`rsr]rbrWrG) rBrrror+r%rCr@rArFr r rtest_groupby_raises_datetime_npKs  r|rl)rYrQr\r^cCs&|}|jdd}ttdd||gdS)NrrHz/timedelta64 type does not support .* operationsr7)r`rGrb)rlr.r%rCr r rtest_groupby_raises_timedeltacs r}c &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 firstruz2'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 operationszNo 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'rMrSrrr_r4)rr`rarbrcrxrG) rBrrrDusing_copy_on_writer3r%rErCr@rArgrFr r rtest_groupby_raises_categoryrs       vx  rcCsd|}|j|d}|r|d}dd}tjtddt|||Wdn1sV0YdS)NrHr!cSs tddSrirkr r r rrl sz.test_groupby_raises_category_udf..funcrjr5rm)rBrrr3r%rCrlr r r test_groupby_raises_category_udfs rc Csd|}|j|d}|r|d}tjtdftjtdfi|\}}|rFd} nd} t|||||d| ddS) NrHr!rrrprqr rrr{) rBrrror3r%rCr@rArFr r rtest_groupby_raises_category_nps   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 Hrz;test_groupby_raises_category_on_category..r=rrFrJr~zz(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 categoriesrrrr4rurrrrrrrrrrMrSrrr_)rrr`rarOgroupsvalues isinstancer#strrbrcrfrxrG)rBrrrDrrr3r%rErCZ empty_groupsr@rArgrFr r r(test_groupby_raises_category_on_category-s          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;r9r`r>r?rf)r%rArCr r r%test_subsetting_columns_axis_1_raisess ,r)r4)+r*rdnumpyrsr>ZpandasrrrrZpandas._testingZ_testingr:Zpandas.tests.groupbyrZfixturearraydictzipr"rrr&r+r.r3rGmarkZ parametrizerhrnr]rWrtryrzr|r}rrrrrr r r rsv            S   E