U
|
ý°dF ã @ sL d Z ddlmZ ddlZddlZddlmZmZm Z ddl
|
m Z ddl mZmZmZ dddd d
|
d d d ddg
|
Zd gZejdd Zejdd Zejdd Zej de¡ej dddg¡ej dddg¡ej d ddg¡d d Zd d Zd!d" Zd#d$ Z ej d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6g¡d7d8 Z d9d: Z d;d< Z dS )=zV
|
test methods relating to generic function evaluation
|
the so-called white/black lists
|
é )Úascii_lowercaseN)Ú DataFrameÚSeriesÚ
|
date_range)Úgroupby_other_methodsÚreduction_kernelsÚtransformation_kernelsÚsumÚprodÚminÚmaxÚmedianÚmeanÚskewÚstdÚvarÚsemc
|
C sB t ddddddddgddddddddgtj d¡tj d¡dS )NÚfooÚbarZoneZtwoZthreeé )ÚAÚBÚCÚD)r ÚnpÚrandomZrandn© r r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_allowlist.pyÚdf& s
|
|
üÿr c C sR t tt¡} d}| t j dd|¡¡}t|d tt j |¡ t|d}|S )Né
|
r é )ÚfloatsÚletters) r ÚarrayÚlistr Útaker Úrandintr r )r" ÚNZrandom_lettersr r r r Ú
|
df_letters2 s þÿr( c C s
|
t dgS )Nr )r r r r r Ú raw_frame@ s r) ÚopÚaxisé ÚskipnaTFÚsortc sÐ dkr| }n| j }tkrx|jd |d}t|d}|jdd fdd¡}|rj|j d}t ||¡ nT|jd |d}t| }|jdd fdd¡}|rÀ|j d}t ||¡ d S ) Nr )Úlevelr+ r. )r- ©r/ c s t | dS )N)r+ r- ©Úgetattr©Úh©r+ r* r- r r Ú<lambda>V ó z3test_regression_allowlist_methods.<locals>.<lambda>©r+ c s t | dS )Nr8 r1 r3 )r+ r* r r r6 ^ r7 )ÚTÚAGG_FUNCTIONS_WITH_SKIPNAÚgroupbyr2 ÚapplyZ
|
sort_indexÚtmÚassert_frame_equal) r) r* r+ r- r. ÚframeZgroupedÚresultÚexpectedr r5 r Ú!test_regression_allowlist_methodsE s$ ÿ rB c C sÖ | }| j }ddddddddd d
|
d d d g }dd t|D }| |¡ |D ]}||fD ]z}| |j¡}dt| dt|j d}dt|j dt| d} | d| }
|
tj t
|
|
|
d t || W 5 Q R X qTqHd S )NÚevalÚqueryÚabsÚwhereÚmaskZalignr; ZclipZastypeÚatÚcombineZ consolidateZconvert_objectsc S s g | ]}| d ¡r|qS )Úto_©Ú
|
startswith)Ú.0Úmethodr r r Ú
|
<listcomp>w s
|
z*test_groupby_blocklist.<locals>.<listcomp>z (?:^Cannot.+z.+'z!'.+try using the 'apply' method$)z(?:^'z' object has no attribute z$)ú|)Úmatch) r! ÚdirÚextendr; r" ÚreprÚtypeÚ__name__ÚpytestZraisesÚAttributeErrorr2 ) r( r ÚsZ blocklistZ
|
to_methodsÚblÚobjÚgbZdefined_but_not_allowedZ not_definedÚmsgr r r Útest_groupby_blocklistd s8 ó
|
ÿÿr^ c B C s¶ | j dd}dd t|D }ddddd d
|
d d d ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFhB}||ks²td S )GNÚsecondr0 c S s h | ]}| d ¡s|qS ©Ú_rK )rM Úvr r r Ú <setcomp> s
|
z&test_tab_completion.<locals>.<setcomp>r r r ZaggZ aggregater< ZboxplotÚfilterÚfirstZ get_groupÚgroupsÚhistÚindicesÚlastr r r r ZngroupsZnthÚohlcZplotr
|
Úsizer r Z transformr r ÚcountZnuniqueÚheadÚdescribeÚcummaxÚquantileÚrankÚcumprodÚtailÚresampleÚcumminÚfillnaZcumsumZcumcountZngroupÚallÚshiftr r% Ú
|
pct_changeÚanyÚcorrZcorrwithZcovÚdtypesÚndimÚdiffÚidxmaxÚidxminÚffillÚbfillZrollingZ expandingÚpipeÚsampleZewmZ value_counts)r; rR ÚAssertionError)ÚmframeÚgrpÚresultsrA r r r Útest_tab_completion s ¾Dr c C s2 | j dd}dD ] }t||}|j|kstqd S )Nr_ r0 )r r
|
r r re ri )r; r2 rV r
)r r ÚnameÚfr r r Ú test_groupby_function_renameÚ s
|
r rN rl r{ ro ru rr rn rq rp r~ rx rw rz r r r r ry c C sd t dt| d}|| _| dg¡dg }| dg | d ¡}t|| }t|| }t ||¡ d S )NÚ2014©Zperiodsr r )r ÚlenÚindexr; r2 r= r> )r rN ÚrngÚgÚg_expÚresÚexpr r r Ú#test_groupby_selection_with_methodsá s r c C sð t dt| d}d| j_|| _| dg¡dg }| dg | d ¡}t | d¡| d¡¡ t |j |j ¡ t |
|
dd ¡|
|
dd ¡¡ t | d ¡ ¡ | d ¡ ¡ ¡ t | d ¡ ¡ | d ¡ ¡ ¡ t | d
|
d ¡| d
|
d ¡¡ d S ) Nr r r r r r c S s | ¡ S )N)r ©Úxr r r r6 r7 z6test_groupby_selection_other_methods.<locals>.<lambda>r c S s t | dkS )Né )r r r r r r6 r7 )r r Úcolumnsr r r; r= r> rv r| r< rt r rj rd )r r r r r r r Ú$test_groupby_selection_other_methods s ÿr c C sÀ | | jd d
df ¡}dd t|D t| j }t|}|t8 }|t8 }|t8 }tt@ r`ttt@ rlttt@ rxt|rd| d}t|ttB tB }||kr¼d|| d}t|d S )Nr c S s h | ]}| d ¡s|qS r` rK )rM ra r r r rc s
|
z/test_all_methods_categorized.<locals>.<setcomp>z?
|
There are uncategorized methods defined on the Grouper class:
|
aG .
|
|
Was a new method recently added?
|
|
Every public method On Grouper must appear in exactly one the
|
following three lists defined in pandas.core.groupby.base:
|
- `reduction_kernels`
|
- `transformation_kernels`
|
- `groupby_other_methods`
|
see the comments in pandas/core/groupby/base.py for guidance on
|
how to fix this test.
|
zI
|
Some methods which are supposed to be on the Grouper class
|
are missing:
|
z.
|
|
They're still defined in one of the lists that live in pandas/core/groupby/base.py.
|
If you removed a method, you should update them
|
) r; ZilocrR Úsetr r r r r
)r r ÚnamesZ new_namesr] Zall_categorizedr r r Ú test_all_methods_categorized s( þ ýr ) Ú__doc__Ústringr Únumpyr rW Zpandasr r r Zpandas._testingZ_testingr= Zpandas.core.groupby.baser r r Z AGG_FUNCTIONSr: Zfixturer r( r) ÚmarkZ parametrizerB r^ r r r r r r r r r Ú<module> sp ö
|
|
|
,Jïþ
|
|
