Edit File: assertions.cpython-37.opt-1.pyc
B �M!V � @ s* d dl mZ d dlZddlmZ d dlmZ ddlmZm Z d dl Z d dl m Z d dl mZ d dlZd d lmZ ejs�d)d d �Zd*d d�Zd+dd�Zd,dd�Zdd� Zdd� Zn d dlmZmZmZmZmZmZ d-dd�Zd.dd�Zi Z d d � Z d d � Z e j d d!� �Z d"d#� Z!d$d%� Z"e j d/d'd(��Z#dS )0� )�absolute_importN� )�util)�default)� text_type�py3k)� decorator)�exc� )�mockc C s | |ks t |pd| |f ��dS )z.Assert a == b, with repr messaging on failure.z%r != %rN)�AssertionError)�a�b�msg� r �K/opt/alt/python37/lib/python3.7/site-packages/alembic/testing/assertions.py�eq_ s r c C s | |ks t |pd| |f ��dS )z.Assert a != b, with repr messaging on failure.z%r == %rN)r )r r r r r r �ne_ s r c C s | |ks t |pd| |f ��dS )z.Assert a is b, with repr messaging on failure.z %r is not %rN)r )r r r r r r �is_ s r c C s | |k s t |pd| |f ��dS )z2Assert a is not b, with repr messaging on failure.z%r is %rN)r )r r r r r r �is_not_ s r c O s< y|||� d}W n | k r* d}Y nX |s8t d��d S )NFTz#Callable did not raise an exception)r )� except_cls� callable_�args�kwZsuccessr r r � assert_raises s r c O sx y|||� dst d��W nX | k rr } z:t�|t|�tj�sPt d||f ��tt|��d�� W d d }~X Y nX d S )NFz#Callable did not raise an exceptionz%r !~ %szutf-8)r �re�searchr �UNICODE�print�encode)r r r r �kwargs�er r r �assert_raises_message* s r" )r r r r r" r c C sz t �dd| �} t �dd| �} t �dd|�}t �dd|�}trLt �dd|�}nt �dd|�}| |ksvt|prd| |f ��d S )Nz^\s+?|\n� z {2,}� z!U�uz%r != %r)r �subr r )r r r r r r �eq_ignore_whitespace8 s r' c C sB t |�}tt| j|d���dd��dd�|�dd��dd�� d S )N)�dialect� r# � )� _get_dialectr r �compile�replace)�elementZ assert_stringr( r r r �assert_compiledH s r/ c C s| | d ks| dkrt �� S yt| S tk rv ttd| �j| �}|�� t| < }| dkrdd|_n| dkrrd|_ |S X d S )Nr zsqlalchemy.dialects.%sZ postgresqlTZmssqlF) r ZDefaultDialect� _dialects�KeyError�getattr� __import__Zdialectsr( Zimplicit_returningZlegacy_schema_aliasing)�nameZ dialect_mod�dr r r r+ T s r+ c O s t tj| f|�S )a� Context manager which expects one or more warnings. With no arguments, squelches all SAWarnings emitted via sqlalchemy.util.warn and sqlalchemy.util.warn_limited. Otherwise pass string expressions that will match selected warnings via regex; all non-matching warnings are sent through. The expect version **asserts** that the warnings were in fact seen. Note that the test suite sets SAWarning warnings to raise exceptions. )�_expect_warnings�sa_excZ SAWarning)�messagesr r r r �expect_warningse s r9 c o sV t | �}t| tj�r&|tj�s&dV n,t| � s6dV n t||�� dV W dQ R X dS )z�Context manager which expects one or more warnings on specific dialects. The expect version **asserts** that the warnings were in fact seen. N)Zdb_spec� isinstancer Z string_typesZconfigZ_currentZ _is_excludedr9 )�dbr8 r �specr r r �expect_warnings_onu s r= c s t � fdd��}|S )z�Decorator form of expect_warnings(). Note that emits_warning does **not** assert that the warnings were in fact seen. c s$ t � ddi�� | ||�S Q R X d S )N�assert_F)r9 )�fnr r )r8 r r �decorate� s z emits_warning.<locals>.decorate)r )r8 r@ r )r8 r � emits_warning� s rA c s t � �fdd��}|S )aV Mark a test as emitting a warning on a specific dialect. With no arguments, squelches all SAWarning failures. Or pass one or more strings; these will be matched to the root of the warning description by warnings.filterwarnings(). Note that emits_warning_on does **not** assert that the warnings were in fact seen. c s$ t � f��� � | ||�S Q R X d S )N)r= )r? r r )r; r8 r r r@ � s z"emits_warning_on.<locals>.decorate)r )r; r8 r@ r )r; r8 r �emits_warning_on� s rB Tc # s� �rdd� |D ��n|�t ���tj�d � ����fdd� }t�d|�� d V W d Q R X |r��r�tdd��fdd ��D �� ��d S ) Nc S s g | ]}t �|t j��qS r )r r, �I)�.0r r r r � <listcomp>� s z$_expect_warnings.<locals>.<listcomp>c s� |r t |� �s �| |f|�|�S �s(d S x^�D ]*}�r@|�| �sL�s.|| kr.��|� P q.W |d krv�| f|�|� n�| |f|�|� d S )N)� issubclass�match�discard)r Z exception�argr Zfilter_)�exc_cls�filters� real_warn�regex�seenr r �our_warn� s z"_expect_warnings.<locals>.our_warnz warnings.warnzWarnings were not seen: %sz, c 3 s | ]}d � r|j n| V qdS )z%rN)�pattern)rD �s)rM r r � <genexpr>� s z#_expect_warnings.<locals>.<genexpr>)N)�set�warnings�warnr Zpatchr �join)rJ r8 rM r> rO r )rJ rK rL rM rN r r6 � s r6 )N)N)N)N)N)N)TT)$Z __future__r r r# r Zsqlalchemy.enginer Z util.compatr r � contextlibZsqlalchemy.utilr Z sqlalchemyr r7 rT r Zsqla_094r r r r r r" Z sqlalchemy.testing.assertionsr' r/ r0 r+ r9 �contextmanagerr= rA rB r6 r r r r �<module> s6