U mdAO@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 Y/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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&countZcumcountZcummaxZcumminZcumprodZcumsumdiffZffillZfillnafirstZidxmaxZidxminlastmaxmeanZmedianminZngroupZnuniqueZ pct_changeprodZquantileZrankZsemshiftsizeZskewZstdsumvarrrmatch)rrangelistrgroupbyhasattrAssertionError TypeErrorNotImplementedError ValueErrorgetattrrr pytestraises rrr groupby_funcdfargsgbklassmsgr r rtest_groupby_raises_string*s  /1     rNc Cstdddddddddg dddddddddg tdtdd}|j|d}|rV|d }d d }tjtd d t|||W5QRXdS)Nrr rrrr!r"r%r$cSs tddSNTest error messagerAr r r rfuncsz,test_groupby_raises_string_udf..funcrPr:)rr<r=r>rErFrArDrrrrIrKrRr r rtest_groupby_raises_string_udfs rTgroupby_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*r:) rr<r=r>npr8r3rArDrErFrrrrUrIrKrLrMr r rtest_groupby_raises_string_nps* rXc #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+rrr:) rr<datetimerr>r?r@rArDrr rErFrGr r rtest_groupby_raises_datetimes~  #%     r`c Cstdddddddddg dddddddddg tdtdddddd d d }|j|d }|rd|d }dd}tjtddt|||W5QRXdS)Nrr rrrrYrZr[r\r]r"r%r$cSs tddSrOrQr r r rrR sz.test_groupby_raises_datetime_udf..funcrPr:)rr<r_r>rErFrArDrSr r r test_groupby_raises_datetime_udfs rac Cstdddddddddg dddddddddg tdtdddddd d d }|j|d }|rd|d }tjtdftjdi|\}}|dkrt|||n(t j ||dt|||W5QRXdS)Nrr rrrrYrZr[r\r]r"r%r$r^r'r:) rr<r_r>rVr8rAr3rDrErFrWr r rtest_groupby_raises_datetime_nps* rbc #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+rrr:)rr<rrr>r?r@rArBrCrDrr rErF) rrrrHusing_copy_on_writerIrJrKrLrMr r rtest_groupby_raises_category1s   RT     rocCstdddddddddg dddddddddg tdtdddddddddg dddd gd d d }|j|d }|rv|d }dd}tjtddt|||W5QRXdS)Nrr rrrrrr#r$Trcr"r%cSs tddSrOrQr r r rrRsz.test_groupby_raises_category_udf..funcrPr:)rr<rr>rErFrArDrSr r r test_groupby_raises_category_udfs"  rpcCstdddddddddg dddddddddg tdtdddddddddg dddd gd d d }|j|d }|rv|d }tjtdftjtdfi|\}}|dkrt|||n(t j ||dt|||W5QRXdS)Nrr rrrrrr#r$Trcr"r%rlrgr:) rr<rr>rVr8rAr3rDrErFrWr r rtest_groupby_raises_category_nps6  rqc #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$Trcrrrr")robservedr&css|] }|jVqdSr)empty).0groupr r r sz;test_groupby_raises_category_on_category..r'rezz(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)rfz*attempt to get argmax of an empty sequencez*attempt to get argmin of an empty sequencergrhrir(rjrkrlrmr+rrr:)rrr<rr>r?r@r-groupsvaluesrArBrCrDrr rErF) rrrrHrrrnrIrJrKZ empty_groupsrLrMr r r(test_groupby_raises_category_on_categorys     PR     ryc CsJtdgdgdgd}|jddd}tjtdd|d W5QRXdS) Nrr r)rrr#r)Zaxisz'Cannot subset columns when using axis=1r:r)rr>rErFrC)rIrKr r r%test_subsetting_columns_axis_1_raisestsrz)!r_numpyrVrEZpandasrrrrZpandas.tests.groupbyrZfixturearraydictzipr<rrmarkZ parametrizerNrTr8r3rXr`rarbrorprqryrzr r r rsZ "2   U   I     "