U
    ¾mœd  ã                
   @   s|   d dl Z d dlZd dlmZ d dlmZ ej de	ddfe	ddfe	ddfe	dd	fe
dd	fg¡ej d
de
g¡dd„ ƒƒZdS )é    N)Úrewrite_warningz$target_category, target_message, hitúTarget messageTZTargetzget messzMissed messageFÚnew_categoryc              
   C   sh   d}|r|r|n| }|}nt }d}tj||d��. t|| ||ƒ� tjdt d� W 5 Q R X W 5 Q R X d S )NzRewritten messager   )Úmatch)ÚmessageÚcategory)ÚFutureWarningÚtmZassert_produces_warningr   ÚwarningsÚwarn)Ztarget_categoryZtarget_messageÚhitr   Znew_messageZexpected_categoryZexpected_message© r   ú_/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/tests/util/test_rewrite_warning.pyÚtest_rewrite_warning
   s       ÿr   )r
   ZpytestZpandas.util._exceptionsr   Zpandas._testingZ_testingr	   ÚmarkZparametrizer   ÚDeprecationWarningr   r   r   r   r   Ú<module>   s&   ûþ
þþ