U
    >vf                     @   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   J/tmp/pip-unpacked-wheel-vdrwu74i/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 )NN   )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  gGz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   gffffffc              	   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   







