Edit File: assertions.cpython-37.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�kw�successr 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 r0 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 r: 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 � string_types�configZ_currentZ _is_excludedr: )�dbr9 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)r: )�fnr r )r9 r r �decorate� s z emits_warning.<locals>.decorate)r )r9 rC r )r9 r � emits_warning� s rD 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@ )rB r r )r> r9 r r rC � s z"emits_warning_on.<locals>.decorate)r )r> r9 rC r )r> r9 r �emits_warning_on� s rE 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 � 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)rG �s)rQ r r � <genexpr>� s z#_expect_warnings.<locals>.<genexpr>)N)�set�warnings�warnr �patchr �join)rN r9 rQ rA rS r )rN rO rP rQ rR r r7 � s r7 )N)N)N)N)N)N)TT)$� __future__r r r$ r Zsqlalchemy.enginer Z util.compatr r � contextlibZsqlalchemy.utilr Z sqlalchemyr r8 rX r Zsqla_094r r r r r r# Z sqlalchemy.testing.assertionsr( r0 r1 r, r: �contextmanagerr@ rD rE r7 r r r r �<module> s6