U
|
¸ý°dî( ã @ sÎ d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z ddl
|
m Z ddl m Z ddl mZ z ddlZW n ek
|
r dZY nX dZdad d
|
ZG d d d ZdddZejdddZdS )z§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.
|
|
é )Ú annotationsNé )Úconfig)Ú
|
gc_collecté )Úhas_compiled_extc C s | a tjrt ¡ d S ©N)Ú _current_testÚ_profile_statsÚ force_writeÚ reset_count)Zid_© r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/profiling.pyÚ_start_current_test6 s r c @ s^ e Zd ZdZdddZedd Zdd Zd
|
d Zd d Z dd Z
|
dd Z dd Z dd Z dS )ÚProfileStatsFilezøStore per-platform/fn profiling results in a file.
|
|
There was no json module available when this was written, but now
|
the file format which is very deterministically line oriented is kind of
|
handy in any case for diffs and merges.
|
|
Ú
|
cumulativeNc C s t jd k ot jj| _| jp*t jd k o*t jj| _tj |¡| _ tj
|
| j ¡d | _ t dd ¡| _|| _|| _| ¡ | jr| ¡ d S )Néÿÿÿÿc S s
|
t t¡S r )Ú collectionsÚ defaultdictÚdictr r r r Ú<lambda>Q ó z+ProfileStatsFile.__init__.<locals>.<lambda>)r ÚoptionsZforce_write_profilesr Zwrite_profilesÚwriteÚosÚpathÚabspathÚfnameÚsplitÚ short_fnamer r ÚdataÚdumpÚsortÚ_readÚ_write)ÚselfÚfilenamer" r! r r r Ú__init__G s ÿÿÿzProfileStatsFile.__init__c C s¦ t jjd t jj }t jjdkr:t jj t jj¡r:|d7 }d dd tj dd
D ¡}t
|
¡ t
|
¡ ¡ t
|
¡ ¡ ||g}| d ¡ t }| |rd
|
pd ¡ d |¡S ) NÚ_ZsqliteÚ_fileÚ.c S s g | ] }t |qS r ©Ústr)Ú.0Úvr r r Ú
|
<listcomp>f s z1ProfileStatsFile.platform_key.<locals>.<listcomp>r r Z dbapiunicodeZ cextensionsZ nocextensions)r ÚdbÚnameZdriverÚdialectZ_is_url_file_dbÚurlÚjoinÚsysÚ version_infoÚplatformÚmachineÚsystemÚlowerÚpython_implementationÚappendr )r% Z dbapi_keyÚ
|
py_versionZplatform_tokensZ _has_cextr r r Ú platform_key[ s ÿ
|
|
û
|
z ProfileStatsFile.platform_keyc C s t }|| jko | j| j| kS r ©r r r> )r% Útest_keyr r r Ú has_statsu s ÿzProfileStatsFile.has_statsc C s¦ t }| j| }|| j }d|kr.g |d< }n|d }d|krLd |d< }n|d }t||k}|s| |¡ | jr|| ¡ d }n|d || f}|d d7 < |S )NÚcountsÚ current_countr Úlinenor )r r r> Úlenr< r r$ ) r% Ú callcountr@ Úper_fnÚ per_platformrB rC Z has_countÚresultr r r rI { s$
|
|
|
zProfileStatsFile.resultc C sP t }|| jkrd S | j| }| j|kr*d S || j }d|krLg |d d d
< d S )NrB r? )r% r@ rG rH r r r r s
|
|
|
|
z ProfileStatsFile.reset_countc C s\ t }| j| }|| j }|d }|d }|t|k rB|||d < n||d< | jrX| ¡ d S )NrB rC r r )r r r> rE r r$ )r% rF r@ rG rH rB rC r r r Úreplace£ s
|
|
zProfileStatsFile.replacec C s
|
d| j S )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% r r r Ú_header° s ðzProfileStatsFile._headerc
|
C s¬ zt | j}W n tk
|
r$ Y d S X t|D ]p\}}| ¡ }|r.| d¡rNq.| ¡ \}}}| j| }|| }dd | d¡D } | |d< |d |d< d|d < q.| ¡ d S )
|
Nú#c S s g | ] }t |qS r )Úint©r- Úcountr r r r/ Ð s z*ProfileStatsFile._read.<locals>.<listcomp>ú,rB r rD r rC ) Úopenr ÚOSErrorÚ enumerateÚstripÚ
|
startswithr r Úclose)
|
r% Ú profile_frD Úliner@ r> rB rG rH Úcr r r r# Ã s
|
|
zProfileStatsFile._readc C s t d| j t| jd}| | ¡ ¡ t| jD ]^}| j| }| d| ¡ t|D ]8}|| }d dd |d D ¡}| d|||f ¡ qVq2| ¡ d S ) NzWriting profile file %sÚwz
|
# TEST: %s
|
|
rP c s s | ]}t |V qd S r r+ rN r r r Ú <genexpr>à s z*ProfileStatsFile._write.<locals>.<genexpr>rB z %s %s %s
|
) Úprintr rQ r rK Úsortedr r4 rV )r% rW r@ rG r> rH rY r r r r$ Ö s
|
zProfileStatsFile._write)r N)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r' Úpropertyr> rA rI r rJ rK r# r$ r r r r r > s
|
|
r ç©?c s$ ddl m} | fdd}|S )a© 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.
|
|
r )Ú decoratorc
|
sZ t D ]}| || qt }td& |D ]}| ||}q0|W 5 Q R £ S Q R X d S )N)Úvariance)ÚrangeÚcount_functions)ÚfnÚargsÚkwZwarmZ timerangeÚtimeÚrv©Útimesre Úwarmupr r Úwrapö s z!function_call_count.<locals>.wrap)Zsqlalchemy.utilrd )re rn ro rd rp r rm r Úfunction_call_countå s rq c c st t d krt d¡t ¡ s0tjs0t dtj ¡ t t ¡ }|
|
¡ d V | ¡ t j |tjd}|j}t |¡}|d krd }n|\}}td||f |jt dtj¡ | ¡ tjrtj tj¡\}}t d¡d } d|| |pêd f }
|
| |
|
¡ td
|
|
|
tj rt |¡ nV|rpt || } t!|| | k} | rptjrVt |¡ nt"d || d |tj#f d S ) 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.)Ústreamz Pstats calls: %d Expected %sz[, ]r* r z%s_%s%sz.profilezDumped stats to file %szAdjusted function call count %s not within %s%% of expected %s, platform %s. Rerun with --write-profiles to regenerate this callcount.éd )$ÚcProfiler Z_skip_test_exceptionr
|
rA r Z skip_testr r ZProfileÚenableÚdisableÚpstatsZStatsr5 ÚstdoutZ total_callsrI r\ Z
|
sort_statsÚrer r" Z print_statsr! r r Úsplitextr Z
|
dump_statsr rJ rM ÚabsÚAssertionErrorr> ) re ÚprÚstatsrF ÚexpectedZexpected_countÚline_noÚbaseÚextZ test_nameZdumpfileZdevianceÚfailedr r r rg sZ
|
ýÿ
|
|
üüÿrg )rc r r )rc )ra Ú
|
__future__r r Ú
|
contextlibr r7 rw ry r5 Ú r Úutilr r rt Ú ImportErrorr
|
r r r rq Úcontextmanagerrg r r r r Ú<module>
|
s.
|
(
|
|
