$Qe4z`P d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z ddl Z n
# e
$ r dZ Y nw xY wdZdZ G d d e Zd
d Zej dd ZdS )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 update_wrapperc R e Zd ZdZd Zed Zd Zd Zd Z d Z
d Z d Z d
S ) ProfileStatsFilezStore 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
c t j d uot j j | _ | j p t j d uot j j | _ t j | | _ t j
| j d | _ t j
d | _ | | j r| d S d S )Nc 4 t j t S N) collections defaultdictdict p/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/sqlalchemy/testing/profiling.pyz+ProfileStatsFile.__init__..8 s K+D11 r )r optionsforce_write_profiles force_writewrite_profileswriteospathabspathfnamesplit short_fnamer r data_read_write)selffilenames r __init__zProfileStatsFile.__init__. s N$ & N6>+N
%
N$ & H6>+H
W__X..
7==44R8 + 1 1
:
KKMMMMM r c t j j dz t j j z }t j j dk r8t j j t j j r|dz
}d d t j dd D }t j t j
t j
||g}| t j j j rdnd t j }| |rd
pd d | S ) N_sqlite_file.c , g | ]}t | S r str).0vs r
z1ProfileStatsFile.platform_key..K s E E E!s1vv E E Er r r
nativeunicode dbapiunicode cextensions
nocextensions)r dbnamedriverdialect_is_url_file_dburljoinsys version_infoplatformmachinesystemlowerpython_implementationappendconvert_unicode requirements_has_cextensions)r$ dbapi_key
py_versionplatform_tokens _has_cexts r platform_keyz ProfileStatsFile.platform_key@ sG INS(69+;; 9>X % %&)*;*K*K IM+
+
%
I XX E Es/?!/D E E EFF
O # # % % * , , 2 2 4 4
y 0
OO
'88:: y :] MoNNNxx(((r c J t }|| j v o| j | j | v S r )
_current_testr! rL )r$ test_keys r has_statszProfileStatsFile.has_stats^ s* ! Nd&749X;N&N
r c h t }| j | }|| j }d|vrg x|d< }n|d }d|vrdx|d< }n|d }t | |k }|s3| | | j r| d }n|d || f}|dxx dz
cc< |S )Ncounts
current_countr linenor )rN r! rL lenrD r r# ) r$ callcountrO per_fn per_platformrR rS has_countresults r rZ zProfileStatsFile.resultd s 8$ d /0 < ' '.0 0L "VV!(+F , . .<= =L )MM ( 9MKK-/ C MM) $ $ $z
FF!(+VM-BBF_%%%*%%%
r c t }| j | }|| j }|d }|d }|t | k r |||dz
< n||d< | j r| d S d S )NrR rS r r )rN r! rL rU r r# )r$ rV rO rW rX rR rS s r replacezProfileStatsFile.replace s 8$ d /0 h'$_5
3v;; & &(1F=1$ % %"F2J : KKMMMMM r c d| j z 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$ s r _headerzProfileStatsFile._header s
Z! r c t | j }n# t $ r Y d S w xY wt | D ]\ }}| }|r| d r1| \ }}}| j | }|| }d | d D } | |d< |dz |d< d|d< | d S ) N#c , g | ]}t | S r )intr/ counts r r1 z*ProfileStatsFile._read.. s ;;;U;;;r ,rR r rT r rS ) openr IOError enumeratestrip
startswithr r! close)
r$ profile_frT linerO rL rR rW rX cs
r r" zProfileStatsFile._read s TZ((II FF %i 0 0 . .LFD::<z*ProfileStatsFile._write.. s( L LESZZ L L L L L Lr rR
) printr rf r r^ sortedr! r< rk )r$ rl rO rW rL rX rn s r r# zProfileStatsFile._write s
(4:5777S)) ''' ty)) L LHYx(F OO . 9 : : : &v
L
L %l 3 HH L L\(5K L L LLL,,, JKKKK
L r N)
__name__
__module__ __qualname____doc__r& propertyrL rP rZ r\ r^ r" r# r r r r
r
& s $ ) ) X):
6 & & r r
皙?c fd}|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.
c 6 fd}t | S )Nc t D ]
} | i | t }t 5 |D ]
} | i |} |cd d d S # 1 swxY w Y d S )N)variance)rangecount_functions)
argskwwarm timerangetimervfntimesr warmups
r wrapz3function_call_count..decorate..wrap s f
DB e I (333
% ) )DT R BB
s AA Ar )r r r r r s ` r decoratez%function_call_count..decorate s> dB'''r r )r r r r s ``` r function_call_countr s/
(
(
(
(
(
(
( Or c # K t t j d t s-t j s!t j dt j z t t j }|
d V | t j
|t j }|j }t | }|d }n|\ }}t% d||fz | d | |rt+ || z }t- ||z
|k }|s t j rSt j r t | d S t3 d|d| dz d |d
t j d d S 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 %s
cumulativez Adjusted function call count z not within d z% of expected z , platform z;. Rerun with --write-profiles to regenerate this callcount.)cProfiler _skip_test_exception_profile_statsrP r skip_testr r ProfileenabledisablepstatsStatsr= stdout total_callsrZ ru
sort_stats print_statsrb absr r\ AssertionErrorrL ) r prstatsrV expectedexpected_countline_nodeviancefaileds r r r s )*EFFF # # % %
n.B
. 9
:
LLL BIIKKK EEEJJLLL
LCJ / / /E !I $$Y//H "* )Y,G GIII \""" y8+,,Y /008; ^/ #
&&y11111$n " !C &&333 r )r| r r )r| )rz r
contextlibr r? r r= r utilr r r ImportErrorrN r objectr
r contextmanagerr r r r r s:
! ! ! ! ! !OOOO HHH
X X X X Xv X X Xv 4 9 9 9 9 9 9s 3 ==