a
    PfB                     @   s  d dl Z d dlmZmZ d dl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g d
g ddgd ddgddgfg d
g ddd ddgddgfg d
g ddgd dgdgfg d
g d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g dg dge j	
dg dgdggdd Ze j	
d dg ddd!gfdg d"dd!gfddgd dgd#d$gfgd%d& Zd'd( ZdS ))    N)IndexSeries)is_bool_dtypez=data, index, drop_labels, axis, expected_data, expected_index      onetworows)r   r   r   )r   r   r   c           	      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   ^/var/www/ai-form-bot/venv/lib/python3.9/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   sL   t | |d}tj||d |j||d W d    n1 s>0    Y  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} t| js|J | jj	t
ksJ | 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   r   dtypebool)r   r   r   r   r   r   test_drop_with_ignore_errors1   s    
r,   r   )r   r   r   )r   r   r   r   c                    s@    fdd| D }t | td }t |td}t|| d S )Nc                    s   g | ]}| vr|qS r   r   ).0ir   r   r   
<listcomp>G       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    r3   zdata, index, drop_labels   r   r   r   FTc                 C   sZ   | d u rt nd }t| ||d}tjtdd || W d    n1 sL0    Y  d S )N)r
   r   r*   r   r    )r2   r   r"   r#   KeyErrorr   )r
   r   r   r*   r   r   r   r   test_drop_non_empty_listM   s    
r7   c                 C   sX   t dtg d| dd}t|jd g}||}t dtdg| dd}t|| d S )Nd   r5   )r*   r   r   )r   r   r   r   r   r   )Zany_numeric_ea_dtypeZdfidxr   r   r   r   r   test_drop_index_ea_dtype]   s
    
r:   )r"   Zpandasr   r   Zpandas._testingZ_testingr   Zpandas.api.typesr   markZparametrizer   r(   r)   r6   
ValueErrorr$   r,   r3   r7   r:   r   r   r   r   <module>   sF    
	


