관리-도구
편집 파일: assertions.cpython-38.pyc
U -?�f�z � @ s� d dl mZ d dlmZ d dlZd dlmZ d dlmZ d dlZd dl Z d dl Z ddl m Z ddl m Z dd l mZ dd l mZ dd lmZ dd lmZ d dl mZ d dl mZ d dl mZ d dl mZ d dl mZ d dlmZ d dlm Z d dl m Z d dlm Z dd� Z ej!dd� �Z"dd � Z#d d � Z$d d � Z%d!d"� Z&d#d$� Z'da(da)da*dod'd(�Z+ej!dpd)d*��Z,d+d,� Z-d-d.� Z.d/d0� Z/dqd1d2�Z0drd3d4�Z1dsd5d6�Z2dtd7d8�Z3dud9d:�Z4dvd;d<�Z5dwd=d>�Z6dxd?d@�Z7dydAdB�Z8dzdCdD�Z9d{dEdF�Z:e:Z;d|dGdH�Z<d}dIdJ�Z=e=Z>d~dKdL�Z?ddMdN�Z@dOdP� ZAdQdR� ZBdSdT� ZCdUdV� ZDdWdX� ZEdYdZ� ZFd[d\� ZGd�d]d^�ZHG d_d`� d`�ZIej!d�dadb��ZJd�dcdd�ZKd�dedf�ZLG dgdh� dh�ZMG didj� dj�ZNG dkdl� dl�ZOG dmdn� dn�ZPdS )�� )� annotations)� defaultdictN)�copy)� filterfalse� )� assertsql)�config)�engines)�mock)�db_spec)�fail� )�exc)�schema)�sql)�types)�util)�default)�url)� LABEL_STYLE_TABLENAME_PLUS_COL�� decoratorc O s t tj| f|�S )a� Context manager which expects one or more warnings. With no arguments, squelches all SAWarning 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_sqla_only�sa_excZ SAWarning��messages�kw� r �M/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/assertions.py�expect_warnings% s r c o sD t | �}t| t�r$|tj�s$dV n t||�� dV W 5 Q 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)r � isinstance�strr �_currentr )�dbr r �specr r r �expect_warnings_on5 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 s0 t � ddi�� | ||�W 5 Q R � S Q R X d S �N�assert_F)r ��fn�argsr �r r r �decorateN s z emits_warning.<locals>.decorater �r r, r r+ r � emits_warningF s r. c O s t tj| f|�S �N)r r ZSADeprecationWarningr r r r �expect_deprecatedV s ��r0 c O s t tj| f|�S r/ )r r ZBase20DeprecationWarningr r r r �expect_deprecated_20\ s ��r1 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 s6 t � f��ddi�� | ||�W 5 Q R � S Q R X d S r& )r% r( �r# r r r r, n s z"emits_warning_on.<locals>.decorater )r# r r, r r2 r �emits_warning_onb s r3 c s t � fdd��}|S )a+ Mark a test as immune from fatal deprecation warnings. With no arguments, squelches all SADeprecationWarning failures. Or pass one or more strings; these will be matched to the root of the warning description by warnings.filterwarnings(). As a special case, you may pass a function name prefixed with // and it will be re-written as needed to match the standard warning verbiage emitted by the sqlalchemy.util.deprecated decorator. Note that uses_deprecated does **not** assert that the warnings were in fact seen. c s0 t � ddi�� | ||�W 5 Q R � S Q R X d S r& )r0 r( r+ r r r, � s z!uses_deprecated.<locals>.decorater r- r r+ r �uses_deprecatedv s r4 TFc C s t | ||||dd�S )z�SQLAlchemy internal use only _expect_warnings(). Alembic is using _expect_warnings() directly, and should be updated to use this new interface. T)�regex� search_msgr' �raise_on_any_unexpected)�_expect_warnings)�exc_clsr r5 r6 r' r r r r � s �r c # s �s�rdd� |D �� nt |�� td k rftd k s4t�td k s@t�t�� � t�� � t| f7 ad V n�t� � �a� a| fa�r�dd� �ntj �� ������fdd�}t � d|��H z d V W 5 d a aa|r��r�tdd � �fd d ��D �� ��X W 5 Q R X d S ) Nc S s g | ]}t �|t jt jB ��qS r )�re�compile�I�S)�.0�msgr r r � <listcomp>� s z$_expect_warnings.<locals>.<listcomp>c _ s t d| ��d S )NzGot unexpected warning: %r��AssertionError)r? �argr r r r � real_warn� s z#_expect_warnings.<locals>.real_warnc s� t | t�r t| �}t| �} n|r*|d }nd }|r<t|t�sT�sP�| f|�|�S d S � s`�s`d S � D ]:}�rv|�| �s��r�|�| �s��sd|| krd��|� q�qd�s��| f|�|� d S )Nr )r �_EXC_CLS�typer! � issubclass�search�match�discard)r? rC r � exception�filter_��filtersr7 rD r5 r6 �seen�squelch_other_warningsr 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)r> �s)r5 r r � <genexpr>� s z#_expect_warnings.<locals>.<genexpr>) �list�_FILTERS�_SEENrB rE �extend�update�set�warnings�warnr �patch�join)r9 r r5 r6 r' r7 rP rQ r rM r r8 � s2 �r8 c C s t � dS )z�Check things that have to be finalized at the end of a test suite. Hardcoded at the moment, a modular system can be built here to support things like PG prepared transactions, tables all dropped, etc. N)�!_assert_no_stray_pool_connectionsr r r r �global_cleanup_assertions� s r` c C s t j�� d S r/ )r Ztesting_reaperZassert_all_closedr r r r r_ s r_ c C s6 t | | �}t|| �|k s2td||d | f ��d S )Nz:Given int value %s is not within %d%% of expected value %s�d )�int�absrB )�expectedZreceivedZvarianceZdeviancer r r �int_within_variance s ���re c C s$ t �|| �s t|p d| |f ��d S )Nz%r !~ %r)r: rI rB ��a�br? r r r �eq_regex s ri c C s | |ks t |pd| |f ��dS )z.Assert a == b, with repr messaging on failure.�%r != %rNrA rf r r r �eq_ s rk c C s | |ks t |pd| |f ��dS )z.Assert a != b, with repr messaging on failure.z%r == %rNrA rf r r r �ne_ s rl c C s | |ks t |pd| |f ��dS )z.Assert a <= b, with repr messaging on failure.rj NrA rf r r r �le_# s rm c C s" t | |�s t|pd| |f ��d S )Nz%r is not an instance of %r)r rB rf r r r �is_instance_of( s rn c C s t | d |d� d S �N�r? )�is_�rg r? r r r �is_none, s rs c C s t | d |d� d S ro )�is_notrr r r r � is_not_none0 s ru c C s t t| �d|d� d S )NTrp �rq �boolrr r r r �is_true4 s rx c C s t t| �d|d� d S )NFrp rv rr r r r �is_false8 s ry c C s | |ks t |pd| |f ��dS )z.Assert a is b, with repr messaging on failure.z %r is not %rNrA rf r r r rq < s rq c C s | |k s t |pd| |f ��dS )z2Assert a is not b, with repr messaging on failure.z%r is %rNrA rf r r r rt A s rt c C s | |ks t |pd| |f ��dS )z.Assert a in b, with repr messaging on failure.z %r not in %rNrA rf r r r �in_J s rz c C s | |ks t |pd| |f ��dS )z2Assert a in not b, with repr messaging on failure.z %r is in %rNrA rf r r r �not_inO s r{ c C s" | � |�s t|pd| |f ��dS )z>Assert a.startswith(fragment), with repr messaging on failure.z%r does not start with %rN)� startswithrB )rg �fragmentr? r r r � startswith_X s �r~ c C st t �dd| �} t �dd| �} t �dd| �} t �dd|�}t �dd|�}t �dd|�}| |kspt|pld| |f ��d S )Nz^\s+?|\n� z {2,}� z\trj )r: �subrB rf r r r �eq_ignore_whitespace` s r� c C s, | j | jk r(| js(ds(td| | j f ��dS )aH assert that any exception we're catching does not have a __context__ without a __cause__, and that __suppress_context__ is never set. Python 3 will report nested as exceptions as "during the handling of error X, error Y occurred". That's not what we want to do. we want these exceptions in a cause chain. FzZException %r was correctly raised but did not set a cause, within context %r as its cause.N)� __context__� __cause__�__suppress_context__rB )rK r r r � _assert_proper_exception_contextk s ����r� c O s t | |||dd�S )NT�� check_context��_assert_raises�� except_cls� callable_r* r r r r � assert_raises� s r� c O s t | |||�S r/ r� r� r r r �assert_raises_context_ok� s r� c O s t | ||||dd�S )NT�r? r� r� �r� r? r� r* �kwargsr r r �assert_raises_message� s �r� c O s. t | dg�� |||�W 5 Q R � S Q R X dS )a legacy adapter function for functions that were previously using assert_raises with SAWarning or similar. has some workarounds to accommodate the fact that the callable completes with this approach rather than stopping at the exception raise. z.*N�r )r� r� r* r� r r r � assert_warns� s r� c O s4 t | |gddd�� |||�W 5 Q R � S Q R X dS )aq legacy adapter function for functions that were previously using assert_raises with SAWarning or similar. has some workarounds to accommodate the fact that the callable completes with this approach rather than stopping at the exception raise. Also uses regex.search() to match the given message to the error string rather than regex.match(). TF)r6 r5 Nr� r� r r r �assert_warns_message� s �r� c O s t | ||||d�S ro r� r� r r r � assert_raises_message_context_ok� s r� c C s( t | ||��}|||� W 5 Q R X |jS r/ )�_expect_raises�error)r� r� r* r� r? r� �ecr r r r� � s r� c @ s e Zd ZdZdS )�_ErrorContainerN)�__name__� __module__� __qualname__r� r r r r r� � s r� c c s� t | t�rt| t�s t | t�r&td��t� }|r<t�� d }z|V d}W n~ | k r� } z`||_d}|d k r�t |�}t � ||t j �s�t d||f ��|r�|s�t|� tt |��d�� W 5 d }~X Y nX ~|s�t d��d S )NzCUse expect_warnings for warnings, not expect_raises / assert_raisesr FTz%r !~ %s�utf-8z#Callable did not raise an exception)r rF rG �Warning� TypeErrorr� �sys�exc_infor� r! r: rH �UNICODErB r� �print�encode)r� r? r� r� Z are_we_already_in_a_traceback�success�errZerror_as_stringr r r r� � s: ���� �$r� c C s t | |d�S )Nr� �r� )r� r� r r r � expect_raises� s r� c C s t | ||d�S )Nr� r� )r� r? r� r r r �expect_raises_message� s r� c @ s e Zd Zddd�ZdS )�AssertsCompiledSQLNFTc / s | rt �� �| �_|�_n�| r$d �n��d kr8t| dd ���d krJtjj�nh�dksZ�dkr��dkrlt �� �n t �d��| �_|�_n.�dkr�t �� �n t �t �r�t j � ���� � �|r�|�_i }i }|r�||d< |d k r�t|�|d< |r�d|d < |r�d|d < |�r d|d < |�rd|d < |�r&d|d < |�s:t| dd��rDtj|d< ddlm} t ||j��rp|�� }t|_|}|�r~||d< G dd� d�� G � fdd�d�}||�jf d�i|���t |tj��r�|j } | �r�t | � n|� � } | �r�t | d � t t�di ��} | � d��!dd �} t"d t �� | � d�� t#�$d d t ���} t%| |d | |�f � |d k �r�|�r��j&|dd!��t%�j'|� nt%��(|�|� | d k �r�|�r҈j&|dd!��t%t)�fd"d#��j*D ��| � n,�j(|dd!��t%t)�fd$d#��j*D ��| � | d k �rt%�j+| � |d k �r8t%�fd%d&��j,D �|� |d k �r\t%�fd'd&��j-D �|� |�rt�dd ��r�fd(d)�} | d*d�} |jf d| i|��}!t#�$d d t |!��}| dd�}"|jf d|"i|��}#t|#j*�}$|#j.}%|#j-|#j,B D ],}&|#j/|& ��|$k�r�fd+d#�|$D �}$�q�g �t0t�}'|$D ]�}(zZ|(|'k�rB|'|( d, })nd})|%�1|(|(�}*|�2d-|*f |)�d. }+��3|+� |'|( �3|+� W n0 t4k �r� d/|(|f },d�s�t5|,��Y nX �q$t6�fd0d1�t7d2t8���D ��}-d3d#� t9t:�|$��D �}.d4|$|.|f },t;|-|,� d S )5NZ __dialect__r Z default_qmarkZqmarkZdefault_enhanced�schema_translate_mapZ column_keysT� literal_binds�render_postcompile�"use_literal_execute_for_simple_int�for_executemany�render_schema_translateZassert_from_lintingFZlintingr )�orm�compile_kwargsc @ s e Zd Zdd� ZdS )z5AssertsCompiledSQL.assert_compile.<locals>.DontAccessc S s t d��d S )Nz=compiler accessed .statement; use compiler.current_executable)�NotImplementedError)�self�keyr r r �__getattribute__O s �zFAssertsCompiledSQL.assert_compile.<locals>.DontAccess.__getattribute__N)r� r� r� r� r r r r � DontAccessN s r� c sD e Zd Zdd� Zedd� �Zdd� Zdd� Zd d � Z� fd d �Z d S )z>AssertsCompiledSQL.assert_compile.<locals>.CheckCompilerAccessc S sb || _ i | _t|dd�| _| jr^|j| _t|d�r:|j| _t|d�rL|j| _t|d�r^|j| _d S )N�supports_executionF� _returning�_inline�_return_defaults) �test_statementZ _annotations�getattrr� Z_execution_options�hasattrr� r� r� )r� r� r r r �__init__V s � zGAssertsCompiledSQL.assert_compile.<locals>.CheckCompilerAccess.__init__c S s | j jS r/ )r� �_variant_mapping�r� r r r r� g s zOAssertsCompiledSQL.assert_compile.<locals>.CheckCompilerAccess._variant_mappingc S s | j �� S r/ )r� �_default_dialectr� r r r r� k s zOAssertsCompiledSQL.assert_compile.<locals>.CheckCompilerAccess._default_dialectc [ s | j jj| fd|i|��S )N�dialect)r� r; �__func__�r� r� r r r r r; n s ���zFAssertsCompiledSQL.assert_compile.<locals>.CheckCompilerAccess.compilec [ s | j jj| |f|�S r/ )r� � _compilerr� r� r r r r� s s ��zHAssertsCompiledSQL.assert_compile.<locals>.CheckCompilerAccess._compilerc sV t |d�rBtj�|d� � �� | jj|f|�W 5 Q R � S Q R X n| jj|f|�S d S )NZ statement)r� r r] �objectr� �_compiler_dispatch)r� �compilerr� �r� r r r� x s �����zQAssertsCompiledSQL.assert_compile.<locals>.CheckCompilerAccess._compiler_dispatchN) r� r� r� r� �propertyr� r� r; r� r� r r� r r �CheckCompilerAccessU s r� r� �paramsr� �ascii�ignorez SQL String: z[\n\t]r z%r != %r on dialect %r)Z escape_namesc s g | ]}� j | �qS r )� parameters�r> �x)�expanded_stater r r@ � s �z5AssertsCompiledSQL.assert_compile.<locals>.<listcomp>c s g | ] }� | �qS r r r� )�pr r r@ � s c s i | ]}� j | |j�qS r �� bind_namesZeffective_value�r> rh ��cr r � <dictcomp>� s � z5AssertsCompiledSQL.assert_compile.<locals>.<dictcomp>c s i | ]}� j | |j�qS r r� r� r� r r r� � s � c s t � �}| |_||_|S r/ )r � paramstyle� positional)r� r� �cp)r� r r � get_dialect� s z6AssertsCompiledSQL.assert_compile.<locals>.get_dialectZpyformatc s g | ]}|� kr|�qS r r )r> �v��namer r r@ � s ���z%%(%s)sr z#Expected to find bindparam %r in %rc 3 s" | ]}� |d � | k V qdS )r Nr )r> �i)� positionsr r rT � s �z4AssertsCompiledSQL.assert_compile.<locals>.<genexpr>r c S s g | ] \}}|�qS r r )r> �_r� r r r r@ � s zQOrder of parameters %s does not match the order in the statement %s. Statement %r)<r ZDefaultDialect�supports_default_values�supports_default_metavaluer� r r# r� ZStrCompileDialectr r! r �URL�creater� �default_schema_namerU r Z FROM_LINTINGZ sqlalchemyr� ZQueryZ _statement_20r Z _label_styler; �sqltypesZ TypeEngineZ_static_cache_key�hashZ_generate_cache_key�reprr� �decoder� r: r� rk Zconstruct_expanded_stater� Zconstruct_params�tupleZ positiontupZprefetchZliteral_execute_paramsZpost_compile_paramsZescaped_bind_namesr� r �get�index�append� ValueErrorrB �all�range�len�sorted�ziprx )/r� Zclause�resultr� Z checkparamsr� Zcheck_literal_executeZcheck_post_paramr� ZcheckpositionalZcheck_prefetchZuse_default_dialectZallow_dialect_selectr� r� r� r� r� r� r� Z from_lintingZcheck_param_orderr� r r� r� �stmtr� Zcache_key_no_warningsZ param_str�ccr� Zpyformat_dialectZ pyformat_cZ qmark_dialectZqmark_c�values�escapedZ post_paramZ pos_by_valuer� �start�esc�posr? �orderedrd r )r� r� r� r� r� r� r� r �assert_compile� s 3 � � ��� �� �� � ���z!AssertsCompiledSQL.assert_compile)NNFNNNNNFFTTFFNFNFTF)r� r� r� r r r r r r� � s* �r� c @ s e Zd Zddd�Zdd� ZdS ) �ComparesTablesFTc C sV t |j�t |j�kst�t|j|j�D ]�\}}t|j|j� ||j|j ksPt�|rpt|j|j� t|j|j� |r�d}t|j t |j ��s�t||j |j f ��n | � ||� t|j t j �r�t|j j |j j � |r�tdd� |jD �dd� |jD �� |jr&t|jtj�s&t�q&|�rRt |j�t |j�k�s,t�|jD ] }|jj|j d k �s2t��q2d S )Nz)Type '%s' doesn't correspond to type '%s'c S s h | ] }|j j�qS r ��columnr� �r> �fr r r � <setcomp> s z5ComparesTables.assert_tables_equal.<locals>.<setcomp>c S s h | ] }|j j�qS r r r r r r r s )r� r� rB r� rk r� Z primary_keyZnullabler rF �assert_types_baser� �String�lengthZ foreign_keysZserver_defaultr Z FetchedValue�columns)r� �tableZreflected_table� strict_typesZstrict_constraintsr� Z reflected_cr? r r r �assert_tables_equal� s< � � � z"ComparesTables.assert_tables_equalc C s* |j �|j �s&td|j|j |j f ��d S )Nz7On column %r, type '%s' doesn't correspond to type '%s')rF Z_compare_type_affinityrB r� )r� �c1�c2r r r r ( s ���z ComparesTables.assert_types_baseN)FT)r� r� r� r r r r r r r � s � +r c @ sr e Zd Zdd� Zdd� Zdd� Zdd� Zdd d �Zd d � Zdd� Z dd� Z e j dd� �Z dd� Ze j dd� �Zd S )�AssertsExecutionResultsc G s& t |�}tt|�� | �|||� d S r/ )rU r� r� � assert_list)r� r� �class_�objectsr r r � assert_result3 s z%AssertsExecutionResults.assert_resultc C sL | � t|�t|�kd|j � tdt|��D ]}| �||| || � q,d S )Nz9result list is not the same size as test list, for class r )r' r� r� r� � assert_row)r� r� r �list_r� r r r r 8 s ��z#AssertsExecutionResults.assert_listc C s� | � |j|kdt|� � |�� D ]�\}}t|t�r~t|d t�r`| �t||�|d |d � q�| � |d t||�|d � q"| � t||�|kd|t||�|f � q"d S )Nzitem class is not r r z'attribute %s value %s does not match %s) r' � __class__r� �itemsr r� rU r r� r )r� r Zrowobj�descr� �valuer r r r B s � ��z"AssertsExecutionResults.assert_rowc s� G dd� dt ��t�|�}�fdd�|D �}t�fdd�|�D ] }tdt|�j�jf � q>t|�t|�kr�tdt|�t|�f � t� � � �fd d �}|D ]<}|D ] }|||�r�|� |� q�q�td �jt |�f � q�d S ) z�As assert_result, but the order of objects is not considered. The algorithm is very expensive but not a big deal for the small numbers of rows that the test suite manipulates. c @ s e Zd Zdd� ZdS )zFAssertsExecutionResults.assert_unordered_result.<locals>.immutabledictc S s t | �S r/ )�idr� r r r �__hash__[ s zOAssertsExecutionResults.assert_unordered_result.<locals>.immutabledict.__hash__N)r� r� r� r r r r r � immutabledictZ s r! c s h | ] }� |��qS r r )r> �e)r! r r r _ s zBAssertsExecutionResults.assert_unordered_result.<locals>.<setcomp>c s t | � �S r/ )r )�o)�clsr r �<lambda>a � zAAssertsExecutionResults.assert_unordered_result.<locals>.<lambda>z#Unexpected type "%s", expected "%s"z+Unexpected object count "%s", expected "%s"c sv |� � D ]h\}}t|t�rZz$�jt| |�|d f|d �� W qp tk rV Y dS X qt| |� �|kr dS qdS )Nr r FT)r r r� �assert_unordered_resultr� rB )�objr$ r� r )�NOVALUEr� r r � _compare_itemo s �� zFAssertsExecutionResults.assert_unordered_result.<locals>._compare_itemz2Expected %s instance with attributes %s not found.T) �dictr Z IdentitySetr r rF r� r� r� �remover� ) r� r� r$ rd �foundZwrongr* Z expected_itemZ found_itemr )r) r$ r! r� r r' S s8 ���� ��z/AssertsExecutionResults.assert_unordered_resultNc C s |d krddl m} t�|�S )Nr )r# )r r# r Z assert_engine)r� r# r r r �sql_execution_asserter� s z.AssertsExecutionResults.sql_execution_asserterc G s* | � |�� }|� }W 5 Q R X |j|� |S r/ �r. r' )r� r# r� �rules�asserterr� r r r �assert_sql_execution� s z,AssertsExecutionResults.assert_sql_executionc C sX g }|D ]<}t |t�r0tjdd� |�� D �� }n tj|� }|�|� q| j||f|�� S )Nc S s g | ]\}}t �||��qS r )r � CompiledSQL)r> �kr� r r r r@ � s z6AssertsExecutionResults.assert_sql.<locals>.<listcomp>)r r+ r ZAllOfr r3 r� r2 )r� r# r� r0 Znewrules�ruleZnewruler r r � assert_sql� s � z"AssertsExecutionResults.assert_sqlc C s | � ||t�|��S r/ )r2 r �CountStatements)r� r# r� �countr r r �assert_sql_count� s �z(AssertsExecutionResults.assert_sql_countc g s* | � |�� }d V W 5 Q R X |j|� d S r/ r/ )r� r# r0 r1 r r r �assert_execution� s z(AssertsExecutionResults.assert_executionc C s | � |t�|��S r/ )r: r r7 )r� r# r8 r r r �assert_statement_count� s z.AssertsExecutionResults.assert_statement_countc # s� � fdd�t ||�D �}g }|D ]\}}}|�|�� � q z d V W 5 t ||�D ],\}\}}}|�d d d � |�t�|�� qPX d S )Nc s g | ]\}}� � |�||f�qS r )r. )r> r# r8 r� r r r@ � s �zKAssertsExecutionResults.assert_statement_count_multi_db.<locals>.<listcomp>)r� r� � __enter__�__exit__r' r r7 ) r� Zdbs�countsZrecsZ asserters�ctxr# r8 r1 r r� r � assert_statement_count_multi_db� s � z7AssertsExecutionResults.assert_statement_count_multi_db)N)r� r� r� r r r r' r. r2 r6 r9 � contextlib�contextmanagerr: r; r@ r r r r r 2 s 6 r c @ s e Zd Zdddd�dd�ZdS )�ComparesIndexesz schema.TablerU r! )r rd � dialect_namec C s t t|j�t|�� dd� |jD �}|D ]�}||d }t |j|d � dd� |d D �}t t|j�t|�� |D ]}t||jk� qr|�d�} | r�t t|j�t| �� t|j| |d �D ] \} } } | d kr�t | j | � q�|�d �r(|� d �|d kr(t |j | d |d |� d � � q(d S ) Nc S s i | ] }|j |�qS r r� )r> �idxr r r r� � s zEComparesIndexes.compare_table_index_with_expected.<locals>.<dictcomp>r� �uniquec S s g | ]}|d k r|�qS r/ r )r> r� r r r r@ � s zEComparesIndexes.compare_table_index_with_expected.<locals>.<listcomp>Z column_names� expressions�dialect_optionsZ_include�include) rk r� ZindexesrF r rx r� rG r� �textrH ) r� r rd rD Zidx_dict�exprE �colsr� ZexprsZidx_exp�expr�colr r r �!compare_table_index_with_expected� s6 ��� �z1ComparesIndexes.compare_table_index_with_expectedN)r� r� r� rO r r r r rC � s rC )TFT)TFTFF)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)NF)NF)T)T)Q� __future__r � collectionsr rA r � itertoolsr r: r� r[ r r r r r Z exclusionsr r r r r r r r r� Zenginer r Zsql.selectabler r r rB r% r. r0 r1 r3 r4 rV rW rE r r8 r` r_ re ri rk rl rm rn rs ru rx ry rq rt Zis_not_rz r{ Znot_in_r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r rC r r r r �<module> s� � � P � * 6