o hL@snUddlmZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddZddZddZddZddZ d4ddZGdddZej d5dd Z d6d"d#Z! d7d$d%Z"d8d&d'Z#iZ$d(e%d)<d*d+Z&d,d-Z'd.d/Z(d0d1Z)d2d3Z*dS)9) annotationsN)Any)Dict)exc)default)URL)_expect_warnings)eq_)is_)is_false)is_not_)is_true)ne_ decoratorcCs,|j|jur|jsJd||jfdSdS)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__) exceptionrn/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/alembic/testing/assertions.py _assert_proper_exception_contexts rcOst||||ddS)NT check_context_assert_raises except_cls callable_argskwrrr assert_raises+sr!cOst||||SNrrrrrassert_raises_context_ok/sr#cOst|||||ddS)NT)msgrrrr$rrkwargsrrrassert_raises_message3s r'cOst|||||dS)N)r$rr%rrr assert_raises_message_context_ok9sr(FcCsDt|||}||i|Wd|jS1swY|jSr")_expect_raiseserror)rrrr&r$recrrrr?s  rc@seZdZUdZded<dS)_ErrorContainerNrr*)__name__ __module__ __qualname__r*__annotations__rrrrr,Gs r,c cst}|r td}z|Vd}WnQ|yd}zE||_d}|durI|r6t||ks5J|d|nt|t|tjsIJ|d||rQ|sQt|t t| dWYd}~nd}~ww|skJddS)NrFTz != z !~ zutf-8z#Callable did not raise an exception) r,sysexc_infor*strresearchUNICODErprintencode)rr$r text_exactr+are_we_already_in_a_tracebacksuccesserrrrrr)Ks(  & r)TcCs t||dS)Nrr))rrrrr expect_raiseses r>cCst||||dS)N)r$rr9r=)rr$rr9rrrexpect_raises_messageisr?cCsXtdd|}tdd|}tdd|}tdd|}||ks*J|p(d||fdS)Nz^\s+?|\nz {2,} z%r != %r)r4sub)abr$rrreq_ignore_whitespaceqs  rEzDict[Any, Any] _dialect_modscCsL|dus|dkr tSt|}|dkrd|_|S|dkr$d|_|S)Nr postgresqlTmssqlF)rDefaultDialectrcreate get_dialectimplicit_returninglegacy_schema_aliasing)namedrrr _get_dialect}srPcOstt|fi|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. )rWarningmessagesr rrrexpect_warningss rTcstfdd}|S)zDecorator form of expect_warnings(). Note that emits_warning does **not** assert that the warnings were in fact seen. csHttgRddi||i|WdS1swYdS)Nassert_F)rDeprecationWarning)fnrr rSrrdecorates $z2emits_python_deprecation_warning..decorater)rSrYrrXr emits_python_deprecation_warningsrZcOttj|fi|Sr")rsa_excSADeprecationWarningrRrrrexpect_sqlalchemy_deprecatedr^cOr[r")rr\RemovedIn20WarningrRrrrexpect_sqlalchemy_deprecated_20r_ra)NF)NFF)T)TFr")+ __future__r contextlibr4r1typingrr sqlalchemyrr\sqlalchemy.enginerrsqlalchemy.testing.assertionsrr r r r r rsqlalchemy.utilrrr!r#r'r(rr,contextmanagerr)r>r?rErFr0rPrTrZr^rarrrrsL