U
|
ý°dºQ ã @ sv d dl Z d dlZd dlZd dlmZmZmZmZ d dl m
|
Z
|
ej ddgddgedddd d d d d d
|
d
|
d
|
d d g e d d d d d
|
d
|
d
|
d d g ¡e eed d 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 ed d d d d d d d d g gg
|
ddd Zej ddgddd Zej ddddg¡dd Zej dddg¡dd Zej dddg¡ej d ejejg¡d d Zej ddddg¡d d! Zej dddg¡d"d# Zej dddg¡ej d ejejg¡d$d% Zej ddddg¡d&d' Z ej dddg¡d(d) Z ej dddg¡ej d ejejg¡d*d+ Z ej ddddg¡d,d- Z d.d/ Z dS )0é N)Ú CategoricalÚ DataFrameÚGrouperÚSeries)Úget_groupby_method_argsÚaÚb)Úkeyc C s | d S )Né © ©Úxr r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_raises.pyÚ<lambda> ó r é r
|
é é é )Úparamsc C s | j S ©N©Úparam©Úrequestr r r Úby s r TFc C s | j S r r r r r r Úgroupby_series% s r ÚhowÚmethodÚaggÚ transformc # C sÖ t dddddddddg dddddddddg tdtdd}t||}|j|d}|rz|d }|d
|
krzt|d
|
rvtd S d d d td fd d ttfd fttfdfttfdfttfdftdfd d d tdftdfd d tdftdfd d d tdftdftdfd t dfd d tdft dfd tdfd"| \}}|d krr| dkrDt
|
||| n,| dkr`|j |f| n|j |f| n`t j||dL | dkrt
|
||| n,| dkr¸|j |f| n|j |f| W 5 Q R X d S ) Nr r
|
r r r Ú xyzwtyuio©r r ÚcÚd©r r$ Úcorrwith©NÚ zCould 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) dtypeúunsupported 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!©"ÚallÚanyZbfillr& ÚcountZcumcountZcummaxZcumminZcumprodZcumsumZdiffZffillZfillnaÚfirstZidxmaxZidxminÚlastÚmaxÚmeanZmedianÚminZngroupZnuniqueZ
|
pct_changeÚprodZquantileZrankZsemÚshiftÚsizeZskewZstdÚsumÚvarr r ©Úmatch)r ÚrangeÚlistr ÚgroupbyÚhasattrÚAssertionErrorÚ TypeErrorÚNotImplementedErrorÚ
|
ValueErrorÚgetattrr r ÚpytestÚraises© r r r Ú groupby_funcÚdfÚargsÚgbÚklassÚmsgr r r Útest_groupby_raises_string* s üÿ
|
þþþþÒ/Ñ1
|
|
|
|
|
rM c C s t dddddddddg dddddddddg tdtdd}|j|d}|rV|d }d
|
d }tjtd d t|| | W 5 Q R X d S )Nr r
|
r r r r! r" r% r$ c S s t dd S ©NúTest error message©r@ r r r r Úfunc s z,test_groupby_raises_string_udf.<locals>.funcrO r9 )r r; r< r= rD rE r@ rC ©r r r rH rJ rQ r r r Ú test_groupby_raises_string_udf s üÿ rS Úgroupby_func_npc C s¶ t dddddddddg dddddddddg tdtdd}|j|d}|rV|d }tjd
|
tjtd fi| \}}|d krt|| | n(t j
|
||d t|| | W 5 Q R X d S ) Nr r
|
r r r r! r" r% r$ r' r* r9 ) r r; r< r= Únpr7 r2 r@ rC rD rE ©r r r rT rH rJ rK rL r r r Ú test_groupby_raises_string_np s* üÿ þýrW c # C s¬ t dddddddddg dddddddddg tdt dddddd d
|
¡d }t||}|j|d }|r|d }|dkrt|drtd S dddtdfddddtdftdfdddddddddddddtdftdfdddddtdfdtdfdd"| \}}|d krH| dkrt||| n,| dkr6|j |f| n|j
|
|f| n`t j ||dL | dkrrt||| n,| dkr|j |f| n|j
|
|f| W 5 Q R X d S )Nr r
|
r r r éÕ é
|
é é é`= 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+ r r r9 ) r r; Údatetimer r= r> r? r@ rC r r rD rE rF r r r Ú test_groupby_raises_datetime³ s~ üÿ
|
Þ#Ý%
|
|
|
|
|
r_ c C s t dddddddddg dddddddddg tdt dddddd d
|
¡d }|j|d }|rd|d }dd }tjtdd t|| | W 5 Q R X d S )Nr r
|
r r r rX rY rZ r[ r\ r" r% r$ c S s t dd S rN rP r r r r rQ s z.test_groupby_raises_datetime_udf.<locals>.funcrO r9 )r r; r^ r= rD rE r@ rC rR r r r Ú test_groupby_raises_datetime_udfý s üÿ r` c C sÄ t dddddddddg dddddddddg tdt dddddd d
|
¡d }|j|d }|rd|d }tjtdftjdi| \}}|d krt|| | n(t j
|
||d t|| | W 5 Q R X d S )Nr r
|
r r r rX rY rZ r[ r\ r" r% r$ r] r' r9 ) r r; r^ r= rU r7 r@ r2 rC rD rE rV r r r Ú test_groupby_raises_datetime_np s* üÿ þýra c
|
# C sþ t dddddddddg dddddddddg tdtdddddddddg dddd gd
|
d d }t||}|j|d }|r|d }|dkrt|drtd S dddtdfddttfdfttfdfttfdfttfdftdfd|sætdfnddddddtdftdfdddtdftdftdfdt d fddtd ft d ftd ftd fd "| \}} |d kr| d!krlt
|
||| n,| d"kr|j |f| n|j |f| n`t j|| d#L | d!krÄt
|
||| n,| d"krà|j |f| n|j |f| W 5 Q R X d S )$Nr r
|
r r r r r r# 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 firstúC'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 float64úC'Categorical' with dtype category does not support reduction 'skew'ú-category type does not support sum operationsúB'Categorical' with dtype category does not support reduction 'var'r+ r r r9 )r r; r r r= r> r? r@ rA rB rC r r rD rE )
|
r r r rG Úusing_copy_on_writerH rI rJ rK rL r r r Ú test_groupby_raises_category1 s¼
|
ýüÿ
|
þþþþþþûþþþþþþ²R®T
|
|
|
|
|
rn c C sª t dddddddddg dddddddddg tdtdddddddddg dddd gd
|
d d }|j|d }|rv|d }dd }tjtdd t|| | W 5 Q R X d S )Nr r
|
r r r r r r# r$ Trb r" r% c S s t dd S rN rP r r r r rQ Å s z.test_groupby_raises_category_udf.<locals>.funcrO r9 )r r; r r= rD rE r@ rC rR r r r Ú test_groupby_raises_category_udf± s"
|
ýüÿ ro c C sÚ t dddddddddg dddddddddg tdtdddddddddg dddd gd
|
d d }|j|d }|rv|d }tjtdftjtdfi| \}}|d kr®t|| | n(t j
|
||d t|| | W 5 Q R X d S )Nr r
|
r r r r r r# r$ Trb r" r% rk rf r9 ) r r; r r= rU r7 r@ r2 rC rD rE rV r r r Ú test_groupby_raises_category_npÌ s6
|
ýüÿ þþúrp c # C sH t tdddddddddg ddddgdddddddddddg td tdddddddddg ddddgddd
|
}t||}|j||d }|r®|d }|d kr®t|d rªtd S td 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
|
d fdt d fddt
|
d ft d ft
|
d ft
|
d fd!"| \}
|
} |
|
d krä| d"kr¶t ||| n,| d#krÒ|j|f| n|j|f| n`tj|
|
| d$L | d"krt ||| n,| d#kr*|j|f| n|j|f| W 5 Q R X d S )%Nr r r# r$ Trb r r r r" )r Úobservedr& c s s | ] }|j V qd S r )Úempty)Ú.0Úgroupr r r Ú <genexpr> s z;test_groupby_raises_category_on_category.<locals>.<genexpr>r' rd 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)r) re z*attempt to get argmax of an empty sequencez*attempt to get argmin of an empty sequencerf rg rh r( ri rj rk rl r+ r r r9 )r r r; r r= r> r? r- ÚgroupsÚvaluesr@ rA rB rC r r rD rE ) r r r rG rq rm rH rI rJ Z empty_groupsrK rL r r r Ú(test_groupby_raises_category_on_categoryð sÆ
|
ý
|
ýøÿ
|
þþþþþûþÿ ÿ þþþþ´P°R
|
|
|
|
|
rx c C sJ t dgdgdgd} | jddd}tjtdd |d W 5 Q R X d S )
|
Nr r
|
r )r r r# r )Zaxisz'Cannot subset columns when using axis=1r9 r )r r= rD rE rB )rH rJ r r r Ú%test_subsetting_columns_axis_1_raisest s ry )!r^ ÚnumpyrU rD Zpandasr r r r Zpandas.tests.groupbyr ZfixtureÚarrayÚdictÚzipr; r r ÚmarkZ parametrizerM rS r7 r2 rW r_ r` ra rn ro rp rx ry r r r r Ú<module> sZ "2öÿ
|
|
|
U
|
|
I
|
|
|
"
|
|
