U
|
ý°di ã @ sH d Z ddlZddlZddlmZmZ ddlmZmZ dd Z dd Z
|
dS ) zY
|
Test the consistency of the groupby API, both internally and with other pandas objects.
|
é N)Ú DataFrameÚSeries)ÚDataFrameGroupByÚ SeriesGroupByc C sþ |dkr"d}| j tjj|d¡ |dkrDd}| j tjj|d¡ |dkr^tt|rZtd S tt|}tt |}t
|
t |¡j }|dkrdh}nt
|
t |¡j }t
|
t
|
}}|d krÄd
|
d d h}nô|d krÖdd h}nâ|dkræd h}nÒ|dkrd d
|
dh}dddh}n²|dkr$d d
|
dh}ddh}n|dkr:d d
|
dh}n~|dkrPddd h}nh|dkrjddh}dh}nN|d kr|dh}n<|d krd
|
h}d h}n$|d kr¦dh}n|d kr¸d d h}||@ |ksÊt||@ |ksÜt||8 }||8 }||ksútd S )!N©ÚfirstÚlastz?first and last are entirely different between frame and groupby©Úreason)Úcumcountz DataFrame has no such methodÚngroupÚsizeÚself©ÚanyÚallÚkwargsÚ bool_onlyÚaxis)ÚcountÚ numeric_only)Znunique©ÚmaxÚminÚskipnaÚ min_countÚengineÚ engine_kwargs©ZmeanZstdÚsumÚvar©ZmedianÚprodZsem©ZbackfillZbfillZffillÚpadÚdowncastÚinplace©ZcummaxZcumminÚargs©ZcumprodZcumsum©Z
|
pct_change©Zrank©ZquantileÚmethod)ÚnodeÚ
|
add_markerÚpytestÚmarkÚxfailÚhasattrr ÚAssertionErrorÚgetattrr ÚsetÚinspectÚ signatureÚ
|
parameters) ÚrequestÚ groupby_funcÚmsgZ frame_methodÚ gb_methodÚresultÚexpectedÚexclude_expectedÚexclude_result© rB ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_api_consistency.pyÚtest_frame_consistency sb
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
rD c C sü |dkr"d}| j tjj|d¡ |dkrDd}| j tjj|d¡ |dkr^tt|rZtd S tt|}tt |}t
|
t |¡j }|dkrdh}nt
|
t |¡j }t
|
t
|
}}|d krÄd
|
d d h}nò|d krÔd h}nâ|dkròd d
|
dh}dddh}nÄ|dkrd d
|
dh}ddh}n¦|dkr&d d
|
dh}n|dkr<ddd h}nz|dkrVddh}dh}n`|dkrhdh}nN|d krd
|
h}d h}n6|d krdh}n$|d kr¦dd
|
h}n|d kr¶dh}||@ |ksÈt||@ |ksÚt||8 }||8 }||ksøtd S ) Nr z@first and last are entirely different between Series and groupbyr )r ZcorrwithzSeries has no such methodr r r r r r r )Zdiffr r r r r r r! r# r% r& r' r( r r) r* r+ )ZidxminZidxmaxr, )r. r/ r0 r1 r2 r3 r r4 r5 r r6 r7 r8 r9 ) r: r; r< Z series_methodr= r> r? r@ rA rB rB rC Útest_series_consistencyR sb
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
rE ) Ú__doc__r7 r0 Zpandasr r Zpandas.core.groupby.genericr r rD rE rB rB rB rC Ú<module> s ?
|
