U >v fAO@svddlZddlZddlZddlmZmZmZmZddl m Z ej ddgddgedddddddd d d d d d g e dddd d d d d d g e eed dddd d d d d d g ed d d d d d d d d g ed d d d d d d d d g ed d d d d d d d d g gg dddZej ddgdddZejddddgddZejdddgddZejdddgejdejejgddZejddddgd d!Zejdddgd"d#Zejdddgejdejejgd$d%Zejddddgd&d'Zejdddgd(d)Zejdddgejdejejgd*d+Zejddddgd,d-Zd.d/Z dS)0N) Categorical DataFrameGrouperSeries)get_groupby_method_argsab)keycCs|dS)Nxr r D/tmp/pip-unpacked-wheel-vdrwu74i/pandas/tests/groupby/test_raises.pyrr  )paramscCs|jSNparamrequestr r rbysrTFcCs|jSrrrr r rgroupby_series%srhowmethodagg transformc #Cstdddddddddg dddddddddg tdtdd}t||}|j|d}|rz|d }|d krzt|d rvtdSd d d td fd d ttfd fttfdfttfdfttfdftdfd d d tdftdfd d tdftdfd d d tdftdftdfd t dfd d tdft dfd tdfd"|\}}|dkrr|dkrDt |||n,|dkr`|j |f|n|j |f|n`t j||dL|dkrt |||n,|dkr|j |f|n|j |f|W5QRXdS)Nrr rrr xyzwtyuiorrcdrr$corrwithNzCould not convertzH(function|cummax) is not (implemented|supported) for (this|object) dtypezH(function|cummin) is not (implemented|supported) for (this|object) dtypezI(function|cumprod) is not (implemented|supported) for (this|object) dtypezH(function|cumsum) is not (implemented|supported) for (this|object) dtypeunsupported operand typez#'argmax' not allowed for this dtypez#'argmin' not allowed for this dtype.Could not convert xy?z?w?t?y?u?i?o? to numericz!could not convert string to floatz0can't multiply sequence by non-int of type 'str'z,cannot be performed against 'object' dtypes!"allanyZbfillr&countZcumcountZcummaxZcumminZcumprodZcumsumZdiffZffillZfillnafirstZidxmaxZidxminlastmaxmeanZmedianminZngroupZnuniqueZ pct_changeprodZquantileZrankZsemshiftsizeZskewZstdsumvarrrmatch)rrangelistrgroupbyhasattrAssertionError TypeErrorNotImplementedError ValueErrorgetattrrr pytestraises rrr groupby_funcdfargsgbklassmsgr r rtest_groupby_raises_string*s  /1     rMc Cstdddddddddg dddddddddg tdtdd}|j|d}|rV|d }d d }tjtd d t|||W5QRXdS)Nrr rrrr!r"r%r$cSs tddSNTest error messager@r r r rfuncsz,test_groupby_raises_string_udf..funcrOr9)rr;r<r=rDrEr@rCrrrrHrJrQr r rtest_groupby_raises_string_udfs rSgroupby_func_npc Cstdddddddddg dddddddddg tdtdd}|j|d}|rV|d }tjd tjtd fi|\}}|dkrt|||n(t j ||d t|||W5QRXdS) Nrr rrrr!r"r%r$r'r*r9) rr;r<r=npr7r2r@rCrDrErrrrTrHrJrKrLr r rtest_groupby_raises_string_nps* rWc #Cstdddddddddg dddddddddg tdtdddddd d d }t||}|j|d }|r|d }|dkrt|drtdSdddtdfddddtdftdfdddddddddddddtdftdfdddddtdfdtdfdd"|\}}|dkrH|dkrt|||n,|dkr6|j |f|n|j |f|n`t j ||dL|dkrrt|||n,|dkr|j |f|n|j |f|W5QRXdS)Nrr rrr `=r"r%r$r&r'z+cannot perform __mul__ with this index typez3datetime64 type does not support cumprod operationsz2datetime64 type does not support cumsum operationsz/cannot perform __truediv__ with this index typez%datetime64 type does not support prodz1dtype datetime64\[ns\] does not support reduction/datetime64 type does not support sum operationsr+rrr9) rr;datetimerr=r>r?r@rCrr rDrErFr r rtest_groupby_raises_datetimes~  #%     r_c Cstdddddddddg dddddddddg tdtdddddd d d }|j|d }|rd|d }dd}tjtddt|||W5QRXdS)Nrr rrrrXrYrZr[r\r"r%r$cSs tddSrNrPr r r rrQ sz.test_groupby_raises_datetime_udf..funcrOr9)rr;r^r=rDrEr@rCrRr r r test_groupby_raises_datetime_udfs r`c Cstdddddddddg dddddddddg tdtdddddd d d }|j|d }|rd|d }tjtdftjdi|\}}|dkrt|||n(t j ||dt|||W5QRXdS)Nrr rrrrXrYrZr[r\r"r%r$r]r'r9) rr;r^r=rUr7r@r2rCrDrErVr r rtest_groupby_raises_datetime_nps* rac #Cstdddddddddg dddddddddg tdtdddddddddg dddd gd d d }t||}|j|d }|r|d }|dkrt|drtdSdddtdfddttfdfttfdfttfdfttfdftdfd|stdfnddddddtdftdfdddtdftdftdfdt dfddtdft dftdftdfd "|\}} |dkr|d!krlt |||n,|d"kr|j |f|n|j |f|n`t j|| d#L|d!krt |||n,|d"kr|j |f|n|j |f|W5QRXdS)$Nrr rrrrrr#r$T categoriesZorderedr"r%r&r'=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 firstC'Categorical' with dtype category does not support reduction 'mean'E'Categorical' with dtype category does not support reduction 'median'zDunsupported operand type\(s\) for /: 'Categorical' and 'Categorical'.category type does not support prod operationszNo matching signature found#Cannot cast object dtype to float64C'Categorical' with dtype category does not support reduction 'skew'-category type does not support sum operationsB'Categorical' with dtype category does not support reduction 'var'r+rrr9)rr;rrr=r>r?r@rArBrCrr rDrE) rrrrGusing_copy_on_writerHrIrJrKrLr r rtest_groupby_raises_category1s   RT     rncCstdddddddddg dddddddddg tdtdddddddddg dddd gd d d }|j|d }|rv|d }dd}tjtddt|||W5QRXdS)Nrr rrrrrr#r$Trbr"r%cSs tddSrNrPr r r rrQsz.test_groupby_raises_category_udf..funcrOr9)rr;rr=rDrEr@rCrRr r r test_groupby_raises_category_udfs"  rocCstdddddddddg dddddddddg tdtdddddddddg dddd gd d d }|j|d }|rv|d }tjtdftjtdfi|\}}|dkrt|||n(t j ||dt|||W5QRXdS)Nrr rrrrrr#r$Trbr"r%rkrfr9) rr;rr=rUr7r@r2rCrDrErVr r rtest_groupby_raises_category_nps6  rpc #CsHttdddddddddg ddddgdddddddddddg td tdddddddddg ddddgddd }t||}|j||d }|r|d}|d krt|d rtdStd d|j D} dddt dfddt t fdft t fdft t fdft t fdft dfd|st dfndd| r&t dfnd| r6t dfndddt dft dfdddt dft dft dfdt dfddt dft dft dft d fd!"|\} } | dkr|d"krt |||n,|d#kr|j|f|n|j|f|n`tj| | d$L|d"krt |||n,|d#kr*|j|f|n|j|f|W5QRXdS)%Nrrr#r$Trbrrrr")robservedr&css|] }|jVqdSr)empty).0groupr r r sz;test_groupby_raises_category_on_category..r'rdzz(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)r)rez*attempt to get argmax of an empty sequencez*attempt to get argmin of an empty sequencerfrgrhr(rirjrkrlr+rrr9)rrr;rr=r>r?r-groupsvaluesr@rArBrCrr rDrE) rrrrGrqrmrHrIrJZ empty_groupsrKrLr r r(test_groupby_raises_category_on_categorys     PR     rxc CsJtdgdgdgd}|jddd}tjtdd|d W5QRXdS) Nrr r)rrr#r)Zaxisz'Cannot subset columns when using axis=1r9r)rr=rDrErB)rHrJr r r%test_subsetting_columns_axis_1_raisestsry)!r^ZnumpyrUrDZpandasrrrrZpandas.tests.groupbyrZfixturearraydictzipr;rrmarkZ parametrizerMrSr7r2rWr_r`rarnrorprxryr r r rsZ "2   U   I     "