ó ¿b›]c@sŠddlZddlZddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z e r´d„Z e Z Zn.e rÐd „ZeZ ZnejZ d „Zd „Zd „Zd efd„ƒYZd„Zd„Zd„Zd„Zd„Zd„Zed„ƒZd„Zdefd„ƒYZddd„Z d„Z!dS(iÿÿÿÿNi(t decorator(t defaultdict(tjython(tpy2k(tpypycGs/tjƒtjdƒtjƒtjƒdS(s aggressive gc.collect for tests.gš™™™™™¹?i(tgctcollectttimetsleep(targs((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytjython_gc_collects     cGstjƒtjƒdS(N(RR(R ((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytpypy_gc_collect$s cCsdS(N((((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytlazy_gc-sc#s¨tƒ}trCyddl}|j|ƒWqCtk r?qCXnddl}|j|ƒxE|D]=‰x4td|jƒD] ‰ˆj‡‡fd†fVq|WqcWdS(Niÿÿÿÿiþÿÿÿcsˆj|ˆƒS(N(tdumps(td(tpickle_tprotocol(s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytBt( tsetRtcPickletaddt ImportErrortpickletrangetHIGHEST_PROTOCOLtloads(tpicklersRR((RRs6lib/python2.7/site-packages/sqlalchemy/testing/util.pyR1s      cCsNt|tƒrt||ƒS|tjdd|ƒjtjƒtd|ƒS(Nt1t0i (t isinstancetfloattroundtdecimaltDecimalt to_integralt ROUND_FLOORtpow(tvaluetprec((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pyt round_decimalEs   t RandomSetcBs>eZd„Zd„Zd„Zd„Zd„Zd„ZRS(cCs,ttj|ƒƒ}tj|ƒt|ƒS(N(tlistRt__iter__trandomtshuffletiter(tselftl((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pyR+Ps cCsFtjdt|ƒdƒ}ttj|ƒƒ|}|j|ƒ|S(Nii(R,trandinttlenR*RR+tremove(R/tindextitem((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytpopUs cCsttj||ƒƒS(N(R)Rtunion(R/tother((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pyR7[scCsttj||ƒƒS(N(R)Rt difference(R/R8((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pyR9^scCsttj||ƒƒS(N(R)Rt intersection(R/R8((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pyR:ascCs t|ƒS(N(R)(R/((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytcopyds(t__name__t __module__R+R6R7R9R:R;(((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pyR)Os      cCs‚ttƒ}x%|D]\}}||j|ƒqWxGt|ƒD]5\}}x&||D]}|||krXtSqXWqAWtSdS(sATrue if the given sorting conforms to the given partial ordering.N(RRRt enumeratetFalsetTrue(ttuplestsorted_elementstdepstparenttchildtitnodetn((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytconforms_partial_orderinghs  csVttƒ‰x%|D]\}}ˆ|j|ƒqW‡‡fd†‰tˆ|ƒƒS(Nc3s†t|ƒdkr t|ƒVnbx_|D]W}t|ƒj|gƒ}|jˆ|ƒs'x#ˆ|ƒD]}|g|VqeWq'q'WdS(Ni(R2R*RR9R:(telementstelemtsubsett sub_ordering(t_all_orderingstedges(s6lib/python2.7/site-packages/sqlalchemy/testing/util.pyRN{s (RRRR.(RARJRDRE((RNROs6lib/python2.7/site-packages/sqlalchemy/testing/util.pytall_partial_orderingsvs   cCsLy ||_Wn8tk rGtj|j|j||j|jƒ}nX|S(scReturn a function with a given __name__. Will assign to __name__ and return the original function if possible on the Python implementation, otherwise a new function will be constructed. This function should be phased out as much as possible in favor of @decorator. Tests that "generate" many named tests should be modernized. (R<t TypeErrorttypest FunctionTypet__code__t __globals__t __defaults__t __closure__(tfntname((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytfunction_named‰s   %cOsr|jƒ}y-||||Ž}|jdddƒ|SWn2tjƒ}|j|Œ}|sf‚qn|SnXdS(sRun the given function under the given contextmanager, simulating the behavior of 'with' to support older Python versions. This is not necessary anymore as we have placed 2.6 as minimum Python version, however some tests are still using this structure. N(t __enter__t__exit__tNonetsystexc_info(tctxRXtargtkwtobjtresultR_traise_((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytrun_as_contextmanagers  cCsd„|DƒS(sŠConverts the results of sql execution into a plain set of column tuples. Useful for asserting the results of an unordered query. cSsh|]}t|ƒ’qS((ttuple(t.0trow((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pys ¼s ((tresults((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytrowset¶scCstst|ƒ‚dS(N(R?tAssertionError(tmsg((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytfail¿sc Osœddlm}ddlm}ddlm}|j|jƒ}|d}t|ddƒ}||_ z|||ŽSWd|j ||jƒ||_ XdS( s>Provide bound MetaData for a single test, dropping afterwards.i(tconfig(tenginesiÿÿÿÿ(tschemaitmetadataN( RRoRpt sqlalchemyRqtMetaDatatdbtgetattrR]Rrtdrop_all_tables( RXR RbRoRpRqRrR/t prev_meta((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytprovide_metadataÃs  cs?ddlm‰ddlm‰t‡‡‡fd†ƒ}|S(siForce the given table names to be dropped after test complete, isolating for foreign key cycles i(Roiÿÿÿÿ(tinspectcs8z|||ŽSWdtˆjˆˆjƒdˆƒXdS(Nt include_names(RwRu(RXR Rb(RoRztnames(s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytgoÞs(RRoRsRzR(R|R}((RoRzR|s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytforce_drop_namesÖstadictcBs&eZdZd„Zd„ZeZRS(s,Dict keys available as attributes. Shadows.cCs1y ||SWntk r,tj||ƒSXdS(N(tKeyErrortdictt__getattribute__(R/tkey((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pyR‚ìs  cGs!tg|D]}||^q ƒS(N(Rg(R/tkeysRƒ((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pyt__call__òs(R<R=t__doc__R‚R…tget_all(((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pyRés  c Cs›ddlm}m}m}m}m}ddlm} m} |dk rYt |ƒ}n|j ƒ0} x&t |j d|ƒƒD] \} } | rÙ|dk r±| |kr±qn| j| || |ƒd|ƒƒƒq| r|jjsñqnx™| D]Ž\} }|dk r"| |kr"qøn|| |ƒ|d|ƒ|d|ƒd|ƒ}| j| ||jjg|jjgd|ƒƒƒqøWqqWWdQXdS(Niÿÿÿÿ(tColumntTabletIntegerRttForeignKeyConstraint(t DropTabletDropConstraintRqtxtyRY(RsRˆR‰RŠRtR‹tsqlalchemy.schemaRŒRR]Rtconnecttreversedtget_sorted_table_and_fkc_namestexecutetdialecttsupports_altertcRŽR(tenginet inspectorRqR{RˆR‰RŠRtR‹RŒRtconnttnametfkcstfkcttb((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pyRwøs8( "     cst‡fd†ƒ}|S(Ncs#z|||ŽSWdˆjƒXdS(N(t_clear(RXRaRb(t event_cls(s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytdecorate's(R(R R¡((R s6lib/python2.7/site-packages/sqlalchemy/testing/util.pytteardown_events&s("R!RR,R^RRRtutilRRRRRR t gc_collectR R RRR(RR)RIRPRZRfRkRnRyR~RRR]RwR¢(((s6lib/python2.7/site-packages/sqlalchemy/testing/util.pyts@                   .