U
    `»{jè  ã                   @   sö  d dl Z d dlmZmZ d dlmZ e j dddgddgdgd dgdgfddgddgdgddgdgfdddgdddgdgd ddgddgfdddgdddgdd ddgddgfdddgdddgdgd dgdgfdddgdddgdd dgdgfg¡d	d
„ ƒZ	e j de
dƒedƒdd edfe
dƒedƒdd edfe
dƒedƒddedfg¡dd„ ƒZdd„ Ze j ddddgdddgg¡e j dg dgdgg¡dd„ ƒƒZe j dddddgddgfddddgddgfddgd dgddgfg¡dd „ ƒZd!d"„ Zd#d$„ ZdS )%é    N)ÚIndexÚSeriesz=data, index, drop_labels, axis, expected_data, expected_indexé   é   ÚoneÚtwoÚ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/var/www/html/bwave/python/venv/lib/python3.8/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ÚcolumnszNo axis named columnsc              	   C   s8   t | |d�}tj||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 tdƒtdƒd�} | jddd�}t || ¡ | jddgdd�}| jd	d … }t ||¡ t d
dgddgd�} | j ¡ r|t	‚| jj
tksŒt	‚|  d¡}t dgdgd�}t ||¡ d S )Nr   r   r   r   Úignore)Úerrorsr   Údr   r   TF)r   ÚrangeÚlistr   r   r   Zilocr
   Z	is_objectÚAssertionErrorÚdtypeÚbool)r   r   r   r   r   r   Útest_drop_with_ignore_errors1   s    
r,   r
   r   c                    s@   ‡ fdd„| D ƒ}t | td� ˆ ¡}t |td�}t ||¡ d S )Nc                    s   g | ]}|ˆ kr|‘qS 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   Zseriesr   r   r/   r   Útest_drop_empty_listC   s    r2   zdata, index, drop_labelsé   FTc              	   C   sF   | d krt nd }t| ||d�}tjtdd�� | |¡ W 5 Q R X d S )N)r	   r
   r*   r   r   )r1   r   r!   r"   ÚKeyErrorr   )r	   r
   r   r*   r   r   r   r   Útest_drop_non_empty_listM   s    
r5   c               	   C   s\   t dddgƒ} d}tjt|d�� |  dd¡}W 5 Q R X t ddgddgd�}t ||¡ d S )Nr   r   r   zqIn a future version of pandas all arguments of Series\.drop except for the argument 'labels' will be keyword-onlyr   r   r   )r   r   Zassert_produces_warningÚFutureWarningr   r   )r   Úmsgr   r   r   r   r   Útest_drop_pos_args_deprecation]   s    ÿr8   c                 C   sZ   t dtdddg| d�d�}t|jd gƒ}| |¡}t dtdg| d�d�}t ||¡ d S )Néd   r   r   )r*   r   )r   r   r
   r   r   r   )Zany_numeric_ea_dtypeZdfÚidxr   r   r   r   r   Útest_drop_index_ea_dtypej   s
    
r;   )r!   Zpandasr   r   Zpandas._testingZ_testingr   ÚmarkZparametrizer   r'   r(   r4   Ú
ValueErrorr#   r,   r2   r5   r8   r;   r   r   r   r   Ú<module>   sF   $" øþ

ûþ

ýþ
