a
    nÝEb\  ã                   @   sÒ  d dl 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)ÚSeriesz=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ÚsÚresultÚexpected© r   úe/home/ja/django-apps/lartica_env/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_descÚserr   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 ¡ s|J ‚|  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)r   r   r   r   r   r   Útest_drop_with_ignore_errors.   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>C   ó    z(test_drop_empty_list.<locals>.<listcomp>)r	   Údtype)r   Úobjectr   r   r   )r	   r   r   Zseriesr   r   r-   r   Útest_drop_empty_list?   s    r2   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	   r0   r   r   )r1   r   r    r!   ÚKeyErrorr   )r   r	   r   r0   r"   r   r   r   Útest_drop_non_empty_listI   s    
r5   c                  C   sn   t g d¢ƒ} d}tjt|d�� |  dd¡}W d   ƒ n1 s@0    Y  t ddgddgd�}t ||¡ d S )	Nr*   zqIn a future version of pandas all arguments of Series\.drop except for the argument 'labels' will be keyword-onlyr   r   r   r   r   r   )r   r   Zassert_produces_warningÚFutureWarningr   r   )r"   Úmsgr   r   r   r   r   Útest_drop_pos_args_deprecationY   s    ÿ*r8   )r    Zpandasr   Zpandas._testingZ_testingr   ÚmarkZparametrizer   r'   r(   r4   Ú
ValueErrorr#   r)   r2   r5   r8   r   r   r   r   Ú<module>   sD    øþ

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

ýþ
