ó H`¾Tc@sdZddlZddlZddlmZmZddlmZddlm Z ddl Z ddl Z ddl Z ddlmZyddl Z Wnek r±dZ nXdd lmZmZmZmZdZdd „Zd efd „ƒYZd d„Zd„ZdS(s§Profiling support for unit and performance tests. These are special purpose profiling methods which operate in a more fine-grained way than nose's profiling plugin. iÿÿÿÿNi(t gc_collectt decorator(tconfig(tSkipTesti(tutil(tjythontpypytwin32tupdate_wrapperc snitƒd6td6td6td6td6d d6d d 6‰ˆd krOd ‰nt‡‡‡fd †ƒ}|S(sFunction profiling. @profiled() or @profiled(report=True, sort=('calls',), limit=20) Outputs profiling info for a decorated function. ttargetstreportt print_callerst print_calleestgraphicttimetcallstsorttlimittanonymous_targetc s.t|||Ž\}}}ˆjdˆdƒ}|rKtjdtƒn߈jdˆdƒ}|r*ˆjdˆdƒ}ˆjdˆdƒ} dˆfGH|ƒ} | j|Œ| rË| j| ƒn | jƒˆjdˆdƒ} | rþ| jƒnˆjdˆdƒ} | r*| jƒq*n|S( NR s runsnake %sR RRsProfile report for target '%s'R R ( t_profiletgettostsystemtfilenamet sort_statst print_statsR R ( tfntargstkwtelapsedt load_statstresultR R tsort_RtstatsR R (t target_optstprofile_configttarget(sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pytdecorate7s2     (stimeRN(tsettTruetFalsetNoneR(R$R"R%((R$R"R#sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pytprofiled"s    #tProfileStatsFilecBs\eZdZd„Zed„ƒZd„Zd„Zd„Zd„Z d„Z d„Z RS( s°"Store per-platform/fn profiling results in a file. We're still targeting Py2.5, 2.4 on 0.7 with no dependencies, so no json lib :( need to roll something silly cCsˆtjdk otjj|_tjj|ƒ|_tjj |jƒd|_ t j d„ƒ|_ |jƒ|jr„|jƒndS(NiÿÿÿÿcSs tjtƒS(N(t collectionst defaultdicttdict(((sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pytms(RtoptionsR)twrite_profilestwriteRtpathtabspathtfnametsplitt short_fnameR,R-tdatat_readt_write(tselfR((sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pyt__init__es  cCsÙtjjdtjj}djgtjdd!D]}t|ƒ^q1ƒ}|g}|j|ƒt rx|jdƒnt rŽ|jdƒnt r¤|jdƒntj j ƒ}|j|rÅdpÈd ƒdj|ƒS( Nt_t.iiRRtwint cextensionst nocextensions(Rtdbtnametdrivertjointsyst version_infotstrtappendRRRt requirementst_has_cextensions(R;t dbapi_keytvt py_versiontplatform_tokenst _has_cext((sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pyt platform_keyts2  cCs)t}||jko(|j|j|kS(N(t _current_testR8RQ(R;ttest_key((sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pyt has_statsˆsc CsÚt}|j|}||j}d|kr=g|d<}n |d}d|krdd|d<}n |d}t|ƒ|k}|s²|j|ƒ|jr©|jƒnd}n|d||f}|dcd7<|S(Ntcountst current_countitlinenoi(RRR8RQtlenRIR2R:R)( R;t callcountRStper_fnt per_platformRURVt has_countR((sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pyRs$          cCsNt}|j|}||j}|d}||d<|jrJ|jƒndS(NRUiÿÿÿÿ(RRR8RQR2R:(R;RYRSRZR[RU((sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pytreplaceªs     cCs d|jS(Ns # %s # This file is written out on a per-environment basis. # For each test in aaa_profiling, the corresponding function and # environment is located within this file. If it doesn't exist, # the test is skipped. # If a callcount does exist, it is compared to what we received. # assertions are raised if the counts do not match. # # To add a new callcount test, apply the function_call_count # decorator and re-run the tests using the --write-profiles # option - this file will be rewritten including the new count. # (R5(R;((sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pyt_header³s c Csñyt|jƒ}Wntk r'dSXx¸t|ƒD]ª\}}|jƒ}| s5|jdƒriq5n|jƒ\}}}|j|}||}g|jdƒD]} t| ƒ^q¥} | |d<|d|dàsRUs %s %s %s (R5RaR2R^tsortedR8RERg(R;RhRSRZRQR[Rk((sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pyR:Ös   "( t__name__t __module__t__doc__R<tpropertyRQRTRR]R^R9R:(((sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pyR+]s     gš™™™™™©?cs‡fd†}|S(s©Assert a target for a test case's function call count. The main purpose of this assertion is to detect changes in callcounts for various functions - the actual number is not as important. Callcounts are stored in a file keyed to Python version and OS platform information. This file is generated automatically for new tests, and versioned so that unexpected changes in callcounts will be detected. cs‡‡fd†}t|ˆƒS(Nc sDtdkrtdƒ‚ntjƒ rUtj rUˆ||Žtdtjƒ‚ntƒtˆ||Ž\}}}|ƒ}|j }tj |ƒ}|dkr­d}n |\} }d||fGH|j ƒ|r@t |ˆƒ} t ||ƒ| k} | r@tjrtj|ƒq=td|ˆd|fƒ‚q@n|S(NscProfile is not installedsŒNo profiling stats available on this platform for this function. Run tests with --write-profiles to add statistics to %s for this platform.sPstats calls: %d Expected %ssyAdjusted function call count %s not within %s%% of expected %s. Rerun with --write-profiles to regenerate this callcount.id(tcProfileR)Rt_profile_statsRTR2R7RRt total_callsRRRftabsR]tAssertionError( RRt timespentRt fn_resultR!RYtexpectedtexpected_counttline_notdeviancetfailed(Rtvariance(sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pytwrapñs<           (R(RR€(R(Rsb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pyR%ðs4((RR%((Rsb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pytfunction_call_countås 6cskd|j‰‡fd†}tjƒ}tjdtƒtƒdˆƒtjƒ}|||tƒdfS(Ns%s.profcs tjˆƒ}tjˆƒ|S(N(tpstatstStatsRtunlink(tst(R(sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pyR,s sresult = fn(*args, **kw)RR(RoRRstrunctxtglobalstlocals(RRRRtbegantended((Rsb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pyR)s   (RqRRFRRRtRtplugin.plugin_baseRR‚RR,Rst ImportErrorR)RRRRRRR*tobjectR+RR(((sb/var/www/send.findwatt.com/datamanager/lib/python2.7/site-packages/sqlalchemy/testing/profiling.pyt s&       " ;ˆ D