U
|
®ý°d¥ ã @ sú d dl Z d dlmZmZ d dlmZ d dlmZ e j
|
dddgddgdgd dgdgfddgddgdgd dgdgfdddgdddgdgd ddgddgfdddgdddgdd ddgddgfdddgdddgdgd dgdgfdddgdddgdd dgdgfg¡d
|
d Z e j
|
d e d e ddd edfe d e ddd edfe d e dddedfg¡dd Zdd Ze j
|
dddd gddd gg¡e j
|
dg dgd gg¡dd Ze j
|
d dddd gdd gfddddgdd gfdd gd dgd d gfg¡d d! Zd"d# ZdS )$é N)ÚIndexÚSeries)Ú is_bool_dtypez=data, index, drop_labels, axis, expected_data, expected_indexé é ZoneZtwoÚrowsc C s6 t | |d}|j||d}t ||d}t ||¡ d S )N)ÚdataÚindex©Úaxis)r ÚdropÚtmÚassert_series_equal) r r r Ú drop_labelsZ expected_dataÚexpected_indexÚserÚresultÚexpected© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_drop.pyÚ%test_drop_unique_and_non_unique_index s r z6data, index, drop_labels, axis, error_type, error_descé ÚabcÚbcúnot found in axis)ÚaÚcolumnszNo axis named columnsc C s8 t | |d}tj||d |j||d W 5 Q R X d S )N©r ©Úmatchr
|
)r ÚpytestÚraisesr )r r r r Z
|
error_typeZ
|
error_descr r r r Útest_drop_exception_raised! s r" c C s¶ t tdtdd} | jddd}t || ¡ | jddgdd}| jd d
}t ||¡ t d
|
dgd d gd} t| js|t | jj
|
t kst | d ¡}t dgd gd}t ||¡ d S ) Nr r r r Úignore)Úerrorsr Údr r TF) r ÚrangeÚlistr r r Zilocr r ÚAssertionErrorÚdtypeÚbool)r r r r r r Ú test_drop_with_ignore_errors1 s
|
r+ r r c s@ fdd| D }t | td ¡}t |td}t ||¡ d S )Nc s g | ]}| kr|qS r r )Ú.0Úi©r r r Ú
|
<listcomp>G s z(test_drop_empty_list.<locals>.<listcomp>)r r) )r Úobjectr r r )r r r Zseriesr r r. r Útest_drop_empty_listC s r1 zdata, index, drop_labelsé FTc C sF | d kr t nd }t| ||d}tjtdd | |¡ W 5 Q R X d S )N)r r r) r r )r0 r r r! ÚKeyErrorr )r r r r) r r r r Útest_drop_non_empty_listM s
|
r4 c C sZ t dtdddg| dd}t|jd g}| |¡}t dtdg| dd}t ||¡ d S )Néd r r )r) r )r r r r r r )Zany_numeric_ea_dtypeZdfÚidxr r r r r Útest_drop_index_ea_dtype] s
|
|
r7 )r Zpandasr r Zpandas._testingZ_testingr Zpandas.api.typesr ÚmarkZ parametrizer r& r' r3 Ú
|
ValueErrorr" r+ r1 r4 r7 r r r r Ú<module> sF $" øþ
|
ûþ
|
|
ýþ
|
|
