U
|
ý°d ã @ s d dl Z d dlZd dlZd dlmZ d dlZd dlmZm Z d dl
|
m Z dd Z dd Zdd Zd
|
d Zd d Zdd Zdd Zdd ZdS )é N)ÚPerformanceWarning)ÚIndexÚ
|
MultiIndexc C sX | ddg¡}t ddg¡}| |¡}| ddddg }t ||¡ t ||¡ | dg¡}| dddd dg }t ||¡ | d
|
¡}| ddd dg }t ||¡ t d g¡}tjtd d | d g¡ W 5 Q R X tjtd d | |¡ W 5 Q R X tjtdd | d
|
dg¡ W 5 Q R X t dd g¡}tjtd d | |¡ W 5 Q R X | j |dd}| ddddd dg }t ||¡ | j |dd}| dddddg }t ||¡ | j d
|
dgdd}| ddd dg }t ||¡ | d
|
dg¡}| dddg }t ||¡ d
|
ddg}tjtdd | |¡ W 5 Q R X | j |dd}| dddg }t ||¡ d S )N)ÚfooÚtwo)ZquxÚoner é é é Úbaré é r )r r z^15$©Úmatchz^'two'$r Úignore)Úerrors)Údropr Ú from_tuplesÚtmÚassert_index_equalÚpytestÚraisesÚKeyError)ÚidxÚdroppedÚindexZdropped2ÚexpectedZ mixed_index© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_drop.pyÚ test_drop sP
|
|
|
r c C sÞ | | d¡ }| d¡}|jdks&ttttdttdttdgt ddddddddg¡t ddddddddg¡t ddddddddg¡gdd d
|
gd }| d¡}|j d ks¸t| d ¡}| d¡}|
|
|¡sÚtd S ) Nr r Úsecondr r r r r r Úthree©ÚlevelsÚcodesÚnames)r r! ) Zget_locÚ droplevelÚnameÚAssertionErrorr r ÚrangeÚnpÚarrayr% Úequals)r r r r r r r Útest_droplevel_with_namesJ s
|
ýù
|
|
|
r- c C sL t ttdttdttdgt ddddddddg¡t ddddddddg¡t ddddddddg¡gdddgd } | d d
ddg¡}| d d
d¡ d¡}| |¡s¶t| d d
g ¡}| d d
}| |¡sâtd
|
}tj t
|
|d | d d
dddg¡ W 5 Q R X tj t d d | d d
dd g¡ W 5 Q R X d S )Nr r r r r r r r! r" zSCannot remove 3 levels from an index with 3 levels: at least one level must be leftr z'Level four not found'Zfour) r r r) r* r+ r& r, r( r r Ú
|
ValueErrorr )r r r Úmsgr r r Útest_droplevel_list` s( ýù
|
ÿ"r0 c C s¾ dddg} t j| ddgd}| ¡ s(ttjddddgd d
|
d d gd d ddggd}|jdddgdd}| ¡ }|j}| ¡ rtt
|
||¡ t t ¡ t
|
| d¡| d¡¡ W 5 Q R X d S )N)ÚaÚ )Úb1Úc1)Úb2Úc2ÚbÚc©r% r1 Údr r3 r4 r r5 r6 r )ÚcolumnsÚdata)r r; Úvalues)r r Z _is_lexsortedr( ÚpdZ DataFrameZ pivot_tableZ reset_indexr; r r Zassert_produces_warningr r )ZtuplesZ lexsorted_miZdfZnot_lexsorted_mir r r Útest_drop_not_lexsorted~ s
|
|
ÿ r? c C sN t jd| fgddgd}d}tjt|d |jt d¡dd W 5 Q R X d S ) NZblahr' Údater9 z@labels \[Timestamp\('2001-01-01 00:00:00'\)\] not found in levelr Z2001©Úlevel)r r r r r r r> Ú Timestamp)Z nulls_fixtureÚmir/ r r r Útest_drop_with_nan_in_index s rE c C sV t dddg¡} t ¡ t dt¡ | d¡}W 5 Q R X t dg¡}t ||¡ d S )N©r r )r r r ) r r ÚwarningsÚcatch_warningsÚ simplefilterr r r r )rD Úresultr r r r Ú'test_drop_with_non_monotonic_duplicates s
|
rK c C s t dddg¡} d}tjt|d | jddd W 5 Q R X tjt|d | jd dgdd W 5 Q R X d
|
}tjt|d | jtjgdd W 5 Q R X tjt|d | jtjd d d gdd W 5 Q R X t tjd fdg¡} d }tjt|d | jtjd dgdd W 5 Q R X d S )NrF )r r )r r z labels \[4\] not found in levelr r r rA r z!labels \[nan\] not found in levelr r z!labels \['a'\] not found in levelr1 )r r r r r r r* Únan)rD r/ r r r Ú1test_single_level_drop_partially_missing_elements§ s "rM c C s8 t jdgdd} | g ¡}tdgdd}t ||¡ d S )N)r )r7 r9 r r7 )r' )r r r& r r r )r rJ r r r r Ú#test_droplevel_multiindex_one_level¼ s
|
rN )rG Únumpyr* r Z pandas.errorsr Zpandasr> r r Zpandas._testingZ_testingr r r- r0 r? rE rK rM rN r r r r Ú<module> s :
|
|
