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 categoriesZordered)rr"rr$r r rdf_with_cat_colSs r2c 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)Nr3matchmethodagg) FutureWarningtmassert_produces_warninggetattrr7 transformpytestraises)klassmsghowgb groupby_funcargswarn_msgZ warn_klassr r r_call_and_checkdsrFrA)r6r7r<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->"allanyZbfillrHcountZcumcountcummaxcummincumprodcumsumdiffZffillfillnafirstZidxmaxZidxminlastmaxmeanmedianminZngroupZnunique pct_changeprodquantileZranksemshiftsizeskewstdsumvar) r\rXrYrRrQrbr^rdrar]z(dtype 'str' does not support operation '')r^rbrar[ZpyarrowzBoperation 'truediv' not supported for dtype 'str' with dtype 'str'rSz>operation 'sub' not supported for dtype 'str' with dtype 'str')rPrOobjectstrz1Cannot perform reduction 'mean' with string dtyperTrrGroupBy.fillna is deprecatedr3) rgroupbyhasattr TypeErrorNotImplementedErrorreescape ValueErrorZdtypeZstoragereplacerF) rArrrCr&using_infer_stringr%rDrBr?r@kindrEr r rtest_groupby_raises_stringxs  ;=      rsr7r<cCsd|}|j|d}|r|d}dd}tjtddt|||Wdn1sV0YdS)NrGr!cSs tddSNTest error messagerkr r r rfuncsz,test_groupby_raises_string_udf..funcrur4rir=r>rkr;)rArrr&r%rBrwr r rtest_groupby_raises_string_udfs rygroupby_func_npc Csv|}|j|d}|r|d}tjdtjtdfi|\}} |rN|tjurJt}d} |rXd} nd} t|| |||d| d dS) NrGr!rIzEagg 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 rE)rinprcrXrkrF) rArrrzr&rqr%rBr?r@rEr r rtest_groupby_raises_string_nps(   rc #Cs|}t||}|j|d}|r@|d}|dkr@t|dr.funcrur4rx)rArrr+r%rBrwr r r test_groupby_raises_datetime_udfes rc Cs`|}|j|d}|r|d}tjtdftjdi|\}}|rBd} nd} t|||||d| ddS) NrGr!rrIr{r|r r}rir~rcrkrXrF) rArrrzr+r%rBr?r@rEr r rtest_groupby_raises_datetime_npts  rrw)r\rQrardcCs&|}|jdd}ttdd||gdS)NrrGz/timedelta64 type does not support .* operationsr6)rirFrk)rwr.r%rBr r rtest_groupby_raises_timedeltas rc &Csj|}t||}|j|d}|r@|d}|dkr@t|drdnd } | d!} nd"} t| | ||||| dS)#NrGr!rHrI=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'rKrTrrrhr3)rrirjrkrlrrF) rArrrCusing_copy_on_writer2r%rDrBr?r@rrrEr r rtest_groupby_raises_categorys       vx  rcCsd|}|j|d}|r|d}dd}tjtddt|||Wdn1sV0YdS)NrGr!cSs tddSrtrvr r r rrw4sz.test_groupby_raises_category_udf..funcrur4rx)rArrr2r%rBrwr r r test_groupby_raises_category_udf+s rc Csd|}|j|d}|r|d}tjtdftjtdfi|\}}|rFd} nd} t|||||d| ddS) NrGr!rrr{r|r r}r) rArrrzr2r%rBr?r@rEr 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!rHcss|] }|jVqdSr)empty).0groupr r r qrz;test_groupby_raises_category_on_category..r<rrFrIrzz(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)rJrz(empty group due to unobserved categoriesrrrrrrrrrrrrrrrKrTrrrhr3)rrrirjrMgroupsvalues isinstancer#rgrkrlrorrF)rArrrCrrr2r%rDrBZ empty_groupsr?r@rrrEr 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 deprecatedr4r)Zaxisz'Cannot subset columns when using axis=1r)rr9r:r8rir=r>ro)r%r@rBr r r%test_subsetting_columns_axis_1_raisess ,r)r3)+r*rmnumpyr~r=ZpandasrrrrZpandas._testingZ_testingr9Zpandas.tests.groupbyrZfixturearraydictzipr"rrr&r+r.r2rFmarkZ parametrizersryrcrXrrrrrrrrrrr r r rsv            r  # E