U
|
¸ý°d¬~ ã @ 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/dqd1d2Z0drd3d4Z1dsd5d6Z2dtd7d8Z3dud9d:Z4dvd;d<Z5dwd=d>Z6dxd?d@Z7dydAdBZ8dzdCdDZ9d{dEdFZ:e:Z;d|dGdHZ<d}dIdJZ=e=Z>d~dKdLZ?ddMdNZ@dOdP ZAdQdR ZBdSdT ZCdUdV ZDdWdX ZEdYdZ ZFd[d\ ZGdd]d^ZHG d_d` d`ZIej!ddadbZJddcddZKddedfZLG dgdh dhZMG didj djZNG dkdl dlZOG dmdn dnZPdS )é )Ú 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 úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/assertions.pyÚexpect_warnings% s r c o sD t | }t| tr$|tjs$dV n t|| dV W 5 Q R X dS )zContext 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 )zDecorator 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 fddi | ||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 | ||||ddS )zSQLAlchemy 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 srdd |D nt | td k rftd k s4ttd k s@tt ¡ t ¡ t| f7 ad V nt a a| fardd 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 | tr t| }t| } n|r*|d }nd }|r<t|tsTsP| f||S d S s`s`d S D ]:}rv| | ¡sr| | ¡ssd|| krd |¡ q´qds´| f|| d S )Nr )r Ú_EXC_CLSÚtyper! Ú
|
issubclassÚsearchÚmatchÚdiscard)r? rC r Ú exceptionZfilter_©Ú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 rO rP r rL 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 ÿýþrd 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 rh c C s | |ks t |pd| |f dS )z.Assert a == b, with repr messaging on failure.ú%r != %rNrA re r r r Úeq_ s rj c C s | |ks t |pd| |f dS )z.Assert a != b, with repr messaging on failure.z%r == %rNrA re r r r Úne_ s rk c C s | |ks t |pd| |f dS )z.Assert a <= b, with repr messaging on failure.ri NrA re r r r Úle_# s rl c C s" t | |s t|pd| |f d S )Nz%r is not an instance of %r)r rB re r r r Úis_instance_of( s rm c C s t | d |d d S ©N©r? )Úis_©rf r? r r r Úis_none, s rr c C s t | d |d d S rn )Úis_notrq r r r Ú is_not_none0 s rt c C s t t| d|d d S )NTro ©rp Úboolrq r r r Úis_true4 s rw c C s t t| d|d d S )NFro ru rq r r r Úis_false8 s rx c C s | |ks t |pd| |f dS )z.Assert a is b, with repr messaging on failure.z %r is not %rNrA re r r r rp < s rp c C s | |k s t |pd| |f dS )z2Assert a is not b, with repr messaging on failure.z%r is %rNrA re r r r rs A s rs c C s | |ks t |pd| |f dS )z.Assert a in b, with repr messaging on failure.z %r not in %rNrA re r r r Úin_J s ry c C s | |ks t |pd| |f dS )z2Assert a in not b, with repr messaging on failure.z %r is in %rNrA re r r r Únot_inO s rz 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 )rf Ú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\tri )r: ÚsubrB re 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 | |||ddS )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 | ||||ddS )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 | ||||dS rn 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 | trt| ts t | tr&tdt }|r<t ¡ d }z|V d}W n~ | k
|
rÈ } z`||_d}|d k rt |}t
|
||t
|
j ¡st d||f |r¦|s¦t| tt | d¡ W 5 d }~X Y nX ~|sØt dd 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 | |dS )Nr ©r )r r r r r Ú expect_raisesì s r© c C s t | ||dS )Nr r¨ )r r? r r r r Úexpect_raises_messageð s rª c @ s e Zd ZdddZdS )ÚAssertsCompiledSQLNFTc / s | rt ¡ | _|_n| r$d nd kr8t| dd d krJtjjnhdksZdkrdkrlt ¡ n
|
t d¡| _|_n.dkrt ¡ 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| ddrDtj|d< ddlm} t ||jrp| ¡ }t|_|}|r~||d< G dd d G fddd}||jf di|t |tjrÚ|j } | rôt | n| ¡ } | rôt | d t tdi } | d¡ !dd ¡} t"d t
|
| d¡ t# $d d t
|
¡} t%| |d | |f |d k r|rj&|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 | |rtdd rfd(d)} | d*d} |jf d| i|}!t# $d d t
|
|!¡}| dd}"|jf d|"i|}#t|#j*}$|#j.}%|#j-|#j,B D ],}&|#j/|& |$kræfd+d#|$D }$qæg t0t}'|$D ]}(zZ|(|'krB|'|( d, })nd})|% 1|(|(¡}*| 2d-|*f |)¡d. }+ 3|+¡ |'|( 3|+¡ W n0 t4k
|
r² d/|(|f },ds®t5|,Y nX q$t6fd0d1t7d2t8D }-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 dd 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|dr:|j| _t|drL|j| _t|dr^|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 |drBtj |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 | |jqS r ©Ú
|
bind_namesZeffective_value©r> rg ©Úcr r Ú
|
<dictcomp>¸ s ÿ z5AssertsCompiledSQL.assert_compile.<locals>.<dictcomp>c s i | ]} j | |jqS 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 rS í 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_namerT r Z FROM_LINTINGZ
|
sqlalchemyr² ZQueryZ _statement_20r Z _label_styler; ÚsqltypesZ
|
TypeEngineZ_static_cache_keyÚhashZ_generate_cache_keyÚreprr¥ Údecoder¤ r: r rj Zconstruct_expanded_staterÑ Zconstruct_paramsÚtupleZ positiontupZprefetchZliteral_execute_paramsZpost_compile_paramsZescaped_bind_namesr× r ÚgetÚindexÚappendÚ
|
ValueErrorrB ÚallÚrangeÚlenÚsortedÚziprw )/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² ZstmtrÍ Zcache_key_no_warningsZ param_strÚccrß Zpyformat_dialectZ
|
pyformat_cZ qmark_dialectZqmark_cÚvaluesÚescapedZ
|
post_paramZ pos_by_valuerà ÚstartÚescÚposr? Zorderedrc 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 ZdddZdd ZdS ) ÚComparesTablesFTc C sV t |jt |jkstt|j|jD ]ä\}}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|jtjs&tq&|rRt |jt |jks,t|jD ] }|jj|j d k s2tq2d S )Nz)Type '%s' doesn't correspond to type '%s'c S s h | ] }|j jqS r ©Úcolumnrâ ©r> Úfr r r Ú <setcomp> s z5ComparesTables.assert_tables_equal.<locals>.<setcomp>c S s h | ] }|j jqS r r r r r r r s )r÷ rÚ rB rù rj 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µ Zc1Ú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/ )rT r¤ rî Ú assert_list)rµ rú Úclass_Zobjectsr 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|tr~t|d tr`| 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ð rT r r¾ r )rµ r ZrowobjÚdescr¶ Úvaluer r r r B s
|
ÿ
|
ÿþz"AssertsExecutionResults.assert_rowc sÞ G dd dt t |¡}fdd|D }tfdd|D ] }tdt|jjf q>t|t|krtdt|t|f t fd d
|
}|D ]<}|D ] }|||r¤| |¡ qq¤td jt
|
|f qd 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|trZz$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 rc Ú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 |tr0tjdd | ¡ 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 r. ró r- )rµ r# r r+ ZnewrulesZruleZnewruler r r Ú
|
assert_sql s
|
ÿ
|
z"AssertsExecutionResults.assert_sqlc C s | ||t |¡¡S r/ )r- 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# r+ r, r r r Úassert_execution§ s z(AssertsExecutionResults.assert_executionc C s | |t |¡¡S r/ )r4 r r1 )rµ r# r2 r r r Úassert_statement_count s z.AssertsExecutionResults.assert_statement_countc # s fddt ||D }g }|D ]\}}}| | ¡ ¡ q z
|
d V W 5 t ||D ],\}\}}}| d d d ¡ | t |¡¡ qPX d S )Nc s g | ]\}} |¡||fqS r )r) )r> r# r2 r r r r@ ² s ÿzKAssertsExecutionResults.assert_statement_count_multi_db.<locals>.<listcomp>)rù ró Ú __enter__Ú__exit__r' r r1 ) rµ ZdbsÚcountsZrecsZ assertersÚctxr# r2 r, r r r Ú assert_statement_count_multi_db° s
|
þ
|
z7AssertsExecutionResults.assert_statement_count_multi_db)N)r r r r r r r" r) r- r0 r3 Ú
|
contextlibÚcontextmanagerr4 r5 r: r r r r r 2 s
|
6
|
|
r c @ s e Zd ZddddddZdS )ÚComparesIndexesz schema.TablerT r! )r rc Ú dialect_namec C s t t|jt| dd |jD }|D ]ð}||d }t |j|d dd |d D }t t|jt| |D ]}t||jk qr| d¡} | rÚt t|jt| 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) rj r÷ Zindexesr@ r rw rñ rA rù ÚtextrB ) rµ r rc r> Zidx_dictÚexpr? ÚcolsrÚ ÚexprsZidx_expÚexprÚcolr r r Ú!compare_table_index_with_expected s6
|
ÿÿþ þz1ComparesIndexes.compare_table_index_with_expectedN)r r r rJ r r r r r= Á s r= )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 r; r Ú itertoolsr r: r¡ rZ r~ r r r r
|
Z
|
exclusionsr r r r r r r r rì Zenginer r Zsql.selectabler r r r< r% r. r0 r1 r3 r4 rU rV rE r r8 r_ r^ rd rh rj rk rl rm rr rt rw rx rp rs Zis_not_ry rz Znot_in_r} r r
r r r r r r r r r r© rª r« r r r= r r r r Ú<module>
|
s¬
|
û
|
ù P
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
ÿ
|
*
|
|
|
6
|
