U
    ¾mœdû  ã                   @   s(  d dl Z d dlmZ d dlmZ eddƒd*dd„ƒZddd	œZeddeƒd+d
d„ƒZdd„ Z	edde	ƒd,dd„ƒZ
e j ddefdg¡dd„ ƒZe j dee ¡ ƒ¡dd„ ƒZe j ddddg¡dd„ ƒZe j dddd g¡dd „ ƒZd!d"„ Zd#d$„ Zeddƒd-d%d&„ƒZe j ddd'g¡d(d)„ ƒZdS ).é    N)Údeprecate_kwargÚoldÚnewFc                 C   s   | S ©N© ©r   r   r   ú_/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/tests/util/test_deprecate_kwarg.pyÚ_f1   s    r	   T)ÚyesÚnoc                 C   s   | S r   r   r   r   r   r   Ú_f2   s    r   c                 C   s   | d S )Né   r   ©Úxr   r   r   Ú_f3_mapping   s    r   c                 C   s   | S r   r   r   r   r   r   Ú_f3   s    r   z	key,klass)r   Nc              	   C   s4   d}t  |¡� tf | |iŽ|ks&t‚W 5 Q R X d S )NéN   )ÚtmÚassert_produces_warningr	   ÚAssertionError)ÚkeyÚklassr   r   r   r   Útest_deprecate_kwarg   s    r   r   c              	   C   s0   t  t¡� t| d�t|  ks"t‚W 5 Q R X d S ©N©r   )r   r   ÚFutureWarningr   Ú_f2_mappingsr   ©r   r   r   r   Útest_dict_deprecate_kwarg&   s    r   Zbogusi90  g®Gáz®ó¿c              	   C   s,   t  t¡� t| d�| kst‚W 5 Q R X d S r   )r   r   r   r   r   r   r   r   r   Útest_missing_deprecate_kwarg,   s    r   r   r   gffffffö¿c              	   C   s0   t  t¡� t| d�t| ƒks"t‚W 5 Q R X d S r   )r   r   r   r   r   r   r   r   r   r   Útest_callable_deprecate_kwarg2   s    r    c               	   C   s,   d} t jt| d�� tdd� W 5 Q R X d S )Nz=((can only|cannot) concatenate)|(must be str)|(Can't convert)©ÚmatchZhellor   )ÚpytestÚraisesÚ	TypeErrorr   )Úmsgr   r   r   Ú"test_callable_deprecate_kwarg_fail8   s    r'   c               	   C   s8   d} t jt| d�� tdddƒddd„ƒ}W 5 Q R X d S )	NzAmapping from old to new argument values must be dict or callable!r!   r   r   r   c                 S   s   | S r   r   r   r   r   r   Úf4D   s    z$test_bad_deprecate_kwarg.<locals>.f4)N)r#   r$   r%   r   )r&   r(   r   r   r   Útest_bad_deprecate_kwarg?   s    
r)   c                 C   s   | |fS r   r   )r   Ú	unchangedr   r   r   Ú_f4I   s    r+   r*   c              	   C   sV   d}| dkrt }|df}nd }d|f}t |¡� tf | |iŽ|ksHt‚W 5 Q R X d S )Né	   r   T)r   r   r   r+   r   )r   r   r   Úexpectedr   r   r   Útest_deprecate_keywordN   s    
r.   )F)F)r   )TT)r#   Zpandas.util._decoratorsr   Zpandas._testingZ_testingr   r	   r   r   r   r   ÚmarkZparametrizer   r   ÚlistÚkeysr   r   r    r'   r)   r+   r.   r   r   r   r   Ú<module>   s0   







