U x^A!@sdZddlZddlZddlZddlZddlZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z z ddlZWnek rdZYnXdZdZGd d d eZdddZejdddZdS)zProfiling 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. N)config) gc_collect)jython)pypyupdate_wrapper)win32c@sTeZdZdZddZeddZddZdd Zd d Z d d Z ddZ ddZ dS)ProfileStatsFilez"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 cCsztjdk otjj|_|jp*tjdk o*tjj|_tj||_ tj |j d|_ t dd|_||jrv|dS)NcSs ttSN) collections defaultdictdictrrD/tmp/pip-install-dq5v43_d/SQLAlchemy/sqlalchemy/testing/profiling.py:z+ProfileStatsFile.__init__..)roptionsZforce_write_profiles force_writeZwrite_profileswriteospathabspathfnamesplit short_fnamerrdata_read_write)selffilenamerrr__init__0szProfileStatsFile.__init__cCstjjdtjj}dddtjddD}|g}||trP|dt r^|dt rl|d |tjj j r~d nd tj }||rd pd d|S)N_.cSsg|] }t|qSrstr).0vrrr Hsz1ProfileStatsFile.platform_key..rrrrwinZ nativeunicodeZ dbapiunicodeZ cextensionsZ nocextensions)rdbnameZdriverjoinsys version_infoappendrrr dialectZconvert_unicode requirementsZ_has_cextensions)r!Z dbapi_key py_versionZplatform_tokensZ _has_cextrrr platform_keyBs$     zProfileStatsFile.platform_keycCst}||jko|j|j|kSr ) _current_testrr5)r!test_keyrrr has_stats[szProfileStatsFile.has_statsc Cst}|j|}||j}d|kr.g|d<}n|d}d|krLd|d<}n|d}t||k}|s|||jr||d}n|d||f}|dd7<|S)Ncounts current_countrlinenor)r6rr5lenr1rr ) r! callcountr7per_fn per_platformr9r:Z has_countresultrrrr@as$    zProfileStatsFile.resultcCs\t}|j|}||j}|d}|d}|t|krB|||d<n||d<|jrX|dS)Nr9r:rr )r6rr5r<rr )r!r=r7r>r?r9r:rrrreplace|s   zProfileStatsFile.replacecCs d|jS)Na # %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. # )r)r!rrr_headerszProfileStatsFile._headerc Cszt|j}Wntk r$YdSXt|D]p\}}|}|r.|drNq.|\}}}|j|}||}dd|dD} | |d<|d|d<d|d <q.|dS) N#cSsg|] }t|qSr)intr(countrrrr*sz*ProfileStatsFile._read..,r9rr;rr:) openrIOError enumeratestrip startswithrrclose) r! profile_fr;liner7r5r9r>r?crrrrs    zProfileStatsFile._readcCstd|jt|jd}||t|jD]^}|j|}|d|t|D]8}||}ddd|dD}|d|||fqVq2|dS) NzWriting profile file %swz # TEST: %s rGcss|]}t|VqdSr r&rErrr sz*ProfileStatsFile._write..r9z %s %s %s ) printrrHrrBsortedrr.rM)r!rNr7r>r5r?rPrrrr s   zProfileStatsFile._writeN) __name__ __module__ __qualname____doc__r#propertyr5r8r@rArBrr rrrrr (s  r 皙?csfdd}|S)aAssert 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. csfdd}t|S)Nc s,td||W5QRSQRXdS)Nvariance)count_functions)argskw)fnr\rrwraps z3function_call_count..decorate..wrapr)r`rar[)r`rdecoratesz%function_call_count..decorater)r\rbrr[rfunction_call_counts rcc cs tdkrtdts0tjs0tdtjtt }| dV| t j |tjd}|j}t|}|dkrd}n|\}}td||f|d||rt||}t|||k}|stjrtjrt|ntd||d|tjfdS)NzcProfile is not installedzNo profiling stats available on this platform for this function. Run tests with --write-profiles to add statistics to %s for this platform.)streamzPstats calls: %d Expected %sZ cumulativezAdjusted function call count %s not within %s%% of expected %s, platform %s. Rerun with --write-profiles to regenerate this callcount.d)cProfilerZ_skip_test_exception_profile_statsr8rZ skip_testrrZProfileenabledisablepstatsZStatsr/stdoutZ total_callsr@rSZ sort_statsZ print_statsrDabsrrAAssertionErrorr5) r\prstatsr=expectedZexpected_countZline_noZdeviancefailedrrrr]sJ      r])rZ)rZ)rXr contextlibrrjr/rutilrrrr r rf ImportErrorr6rgobjectr rccontextmanagerr]rrrrs,