File: //opt/alt/python37/lib/python3.7/site-packages/alembic/testing/__pycache__/warnings.cpython-37.pyc
B
�M!Vd � @ sB d Z ddlmZ ddlZddlmZ ddlZdd� Zd
dd �Z dS )z�NOTE: copied/adapted from SQLAlchemy master for backwards compatibility;
this should be removable when Alembic targets SQLAlchemy 0.9.4.
� )�absolute_importN)�excc C s4 t jdtjd� t jdtjd� t jdtjd� dS )z/Set global warning behavior for the test suite.�ignore)�category�errorN)�warnings�filterwarnings�sa_excZSAPendingDeprecationWarningZSADeprecationWarningZ SAWarning� r
r
�I/opt/alt/python37/lib/python3.7/site-packages/alembic/testing/warnings.py�
setup_filters s
r Fc C s| ddl m} tjdd��}t�d� | � }W dQ R X xB|D ]:}|�d�}|rdt�|t|j ��stt
�q:||t|j �� q:W |S )z9Assert that each of the given warnings are emitted by fn.� )�eq_T)�record�alwaysNr )Z
assertionsr r �catch_warningsr �pop�re�match�str�message�AssertionError)�fnZwarning_msgsZregexr �log�result�warningZpopwarnr
r
r �assert_warnings s
r )F)
�__doc__�
__future__r r Z
sqlalchemyr r r r r r
r
r
r �<module> s