B f@shUddlmZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl mZddl mZdd l mZdd l mZdd l mZdd l mZdd l mZddlmZddlmZddZddZddZddZddZd5ddZGdddZej d6d d!Z!d7d#d$Z"d8d%d&Z#d9d'd(Z$iZ%d)e&d*<d+d,Z'd-d.Z(d/d0Z)d1d2Z*d3d4Z+dS):) annotationsN)Any)Dict)exc)default)_expect_warnings)eq_)is_)is_false)is_not_)is_true)ne_) decorator) sqla_compatcCs,|j|jk r(|js(ds(td||jfdS)aHassert 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__AssertionError) exceptionr\C:\Users\Lenovo\AppData\Local\Temp\pip-unpacked-wheel-q6xnih48\alembic\testing\assertions.py _assert_proper_exception_contexts rcOst||||ddS)NT) check_context)_assert_raises) except_cls callable_argskwrrr assert_raises,srcOst||||S)N)r)rrrrrrrassert_raises_context_ok0sr cOst|||||ddS)NT)msgr)r)rr!rrkwargsrrrassert_raises_message4sr#cOst|||||dS)N)r!)r)rr!rrr"rrr assert_raises_message_context_ok:sr$Fc Cs(t|||}|||WdQRX|jS)N)_expect_raiseserror)rrrr"r!recrrrr@src@seZdZUdZded<dS)_ErrorContainerNrr&)__name__ __module__ __qualname__r&__annotations__rrrrr(Hs r(c cst}|rtd}y|Vd}Wn||k r}z^||_d}|dk rnt|t|tjsnt|d||r~|s~t |t t| dWdd}~XYnX|stddS)NrFTz !~ zutf-8z#Callable did not raise an exception) r(sysexc_infor&researchstrUNICODErrprintencode)rr!rr'Zare_we_already_in_a_tracebacksuccesserrrrrr%Ls &$r%TcCs t||dS)N)r)r%)rrrrr expect_raisesasr7cCst|||dS)N)r!r)r%)rr!rrrrexpect_raises_messageesr8cCsXtdd|}tdd|}tdd|}tdd|}||ksTt|pPd||fdS)Nz^\s+?|\nz {2,} z%r != %r)r/subr)abr!rrreq_ignore_whitespaceis r>zDict[Any, Any] _dialect_modscCsN|dks|dkrtSt|}|dkr8d|_n|dkrFd|_|SdS)NrZ postgresqlTZmssqlF)rZDefaultDialectrZ _create_url get_dialectZimplicit_returningZlegacy_schema_aliasing)namedrrr _get_dialectusrCcOstt|f|S)aContext 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. )rWarning)messagesrrrrexpect_warningss rFcstfdd}|S)zDecorator form of expect_warnings(). Note that emits_warning does **not** assert that the warnings were in fact seen. c s*ttfddi |||SQRXdS)NZassert_F)rDeprecationWarning)fnrr)rErrdecoratesz2emits_python_deprecation_warning..decorate)r)rErIr)rEr emits_python_deprecation_warningsrJcOsttj|f|S)N)rsa_excZSADeprecationWarning)rErrrrexpect_sqlalchemy_deprecatedsrLcOsttj|f|S)N)rrKZRemovedIn20Warning)rErrrrexpect_sqlalchemy_deprecated_20srM)NF)NF)T)T)N), __future__r contextlibr/r-typingrrZ sqlalchemyrrKZsqlalchemy.enginerZsqlalchemy.testing.assertionsrrr r r r r Zsqlalchemy.utilrutilrrrr r#r$rr(contextmanagerr%r7r8r>r?r,rCrFrJrLrMrrrrsF