Current File : //opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyc
�
�M!Vc @@ s� d d l m Z d d l Z d d l m Z d d l m Z d d l m Z m Z d d l
Z
d d l m Z d d l
m Z d d l Z d d
l m Z e j s� d d � Z d d � Z d d
� Z d d � Z d � Z d � Z n. d d l m Z m Z m Z m Z m Z m Z d d � Z d d � Z i Z d � Z d � Z e
j d � � Z! d � Z" d � Z# e
j e$ e$ d � � Z% d S( i ( t absolute_importNi ( t util( t default( t text_typet py3k( t decorator( t exci ( t mockc C@ s, | | k s( t | p" d | | f � � d S( s. Assert a == b, with repr messaging on failure.s %r != %rN( t AssertionError( t at bt msg( ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt eq_ s c C@ s, | | k s( t | p" d | | f � � d S( s. Assert a != b, with repr messaging on failure.s %r == %rN( R ( R R
R ( ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt ne_ s c C@ s, | | k s( t | p" d | | f � � d S( s. Assert a is b, with repr messaging on failure.s %r is not %rN( R ( R R
R ( ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt is_ s c C@ s, | | k s( t | p" d | | f � � d S( s2 Assert a is not b, with repr messaging on failure.s %r is %rN( R ( R R
R ( ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt is_not_ s c O@ sG y | | | � t } Wn | k
r0 t } n X| sC t d � � d S( Ns# Callable did not raise an exception( t Falset TrueR ( t
except_clst callable_t argst kwt success( ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt
assert_raises s
c O@ s� y# | | | � t s" t d � � Wn[ | k
r� } t j | t | � t j � si t d | | f � � t | � j d � GHn Xd S( Ns# Callable did not raise an exceptions %r !~ %ss utf-8( R R t ret searchR t UNICODEt encode( R R R R t kwargst e( ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt assert_raises_message* s
.( R R
R R R R c C@ s� t j d d | � } t j d d | � } t j d d | � } t j d d | � } t rr t j d d | � } n t j d d | � } | | k s� t | p� d | | f � � d S( Ns ^\s+?|\nt s {2,}t s !Ut us %r != %r( R t subR R ( R R
R ( ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt eq_ignore_whitespace8 s c C@ s_ t | � } t t | j d | � � j d d � j d d � | j d d � j d d � � d S( Nt dialects
R s ( t _get_dialectR R t compilet replace( t elementt
assert_stringR$ ( ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt assert_compiledH s
c C@ s� | d k s | d k r" t j � Sy t | SWnr t k
r� t t d | � j | � } | j � t | <} | d k r� t | _
n | d k r� t | _ n | SXd S( NR s sqlalchemy.dialects.%st
postgresqlt mssql(
t NoneR t DefaultDialectt _dialectst KeyErrort getattrt
__import__t dialectsR$ R t implicit_returningR t legacy_schema_aliasing( t namet dialect_modt d( ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyR% T s
c O@ s t t j | | � S( s� 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.
( t _expect_warningst sa_exct SAWarning( t messagesR ( ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt expect_warningse s
c o@ si t | � } t | t j � r6 | t j � r6 d Vn/ t | � sJ d Vn t | | � �
d VWd QXd S( s� Context manager which expects one or more warnings on specific
dialects.
The expect version **asserts** that the warnings were in fact seen.
N( t db_spect
isinstanceR t string_typest configt _currentt _is_excludedR= ( t dbR< R t spec( ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt expect_warnings_onu s " c @ s t � f d � � } | S( s� Decorator form of expect_warnings().
Note that emits_warning does **not** assert that the warnings
were in fact seen.
c @ s* t d t � � � | | | � SWd QXd S( Nt assert_( R= R ( t fnR R ( R< ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt decorate� s ( R ( R< RI ( ( R< sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt
emits_warning� s c @ s t � � f d � � } | S( sV 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 � � � � | | | � SWd QXd S( N( RF ( RH R R ( RD R< ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyRI � s ( R ( RD R< RI ( ( RD R< sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt emits_warning_on� s c #@ s� � r1 g | D] } t j | t j � ^ q
� n | � t � � � t j � d � � � � � f d � } t j d | � �
d VWd QX| r� � s� t d d j
� f d � � D� � � � n d S( Nc @ s� | r) t | � � r) � | | | | � S� s3 d Sxy � D]? } � rU | j | � sh � r: | | k r: � j | � Pq: q: W| d k r� � | | | � n � | | | | � d S( N( t
issubclasst matcht discardR- ( R t exceptiont argR t filter_( t exc_clst filterst real_warnt regext seen( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt our_warn� s
s
warnings.warns Warnings were not seen: %ss , c 3@ s( | ] } d � r | j n | Vq d S( s %rN( t pattern( t .0t s( RU ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pys <genexpr>� s ( R R& t It sett warningst warnR- R t patchR t join( RR R< RU RG R RW ( ( RR RS RT RU RV sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyR9 � s +
(& t
__future__R R R R t sqlalchemy.engineR t util.compatR R t
contextlibt sqlalchemy.utilR t
sqlalchemyR R: R] R t sqla_094R- R R
R R R R t sqlalchemy.testing.assertionsR# R* R/ R% R= t contextmanagerRF RJ RK R R9 ( ( ( sK /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/assertions.pyt <module> s6
.
Mini Shell