rjA'UdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddl m Z ddl mZddl mZdd l mZd d lmZd d lmZd d lmZd dlmZd dlmZd dlmZddlmZddlmZ ddlZdZded< da ded< ddZ!GddZ" d ddZ#ejHdd dZ%y#e$rdZYIwxYw)!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. ) annotationsN)Any)Callable) DefaultDict) Generator)Optional)Tuple)config) provision) PlatformKey) ProfilesFile)TestKey) gc_collect) freethreading)has_compiled_extProfileStatsFile_profile_statsr _current_testcR|atjrtjyyN)rr force_write reset_count)id_s g/var/www/html/python-recuperauto-other/env/lib/python3.12/site-packages/sqlalchemy/testing/profiling.py_start_current_testrCs!M!!""$"cveZdZdZ d d dZeddZddZddZddZ ddZ dd Z dd Z dd Z y)raStore 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. The parsing and rendering of the file itself, as well as of the platform keys used within it, is in :mod:`.testing.profiles_file`, which is shared with ``tools/profiles.py``. This class adds the bookkeeping needed while the test suite runs. Nc6tjduxrtjj|_|jxs.tjduxrtjj|_t ||_|jj|_ tjj|jd|_ ||_ ||_tj t"|_|j r|j'yy)N)r optionsforce_write_profilesrwrite_profileswriterprofilespathfnameossplit short_fnamedumpsort collections defaultdictint_current_counts_write)selffilenamer-r,s r__init__zProfileStatsFile.__init__Ys NN$ & N6>>+N+N %% NN$ & H6>>+H+H %X. ]]'' 77==4R8   # #C (  :: KKM rcVtj}|Jd|j|S)Nzno database configured)r dbplatform_key_for)r3r7s r platform_keyzProfileStatsFile.platform_keyvs, YY~777~$$R((rc 4g}|jjr|jd|jt j |t jjdt jj}tr|dz }ttjtjjtj j||j"|j$t'|t)S)Nasync.t)machinesystemimplementationpythondialectdriver dbapi_flags cextensions)rBis_asyncappendextendr profile_platform_tokenssys version_infomajorminorrr platformr>r?lowerpython_implementationnamerCtupler)r3r7rD py_versions rr8z!ProfileStatsFile.platform_key_for|s ::     w ' 9<z0ProfileStatsFile.has_platform..s-  DMM77A A s'*)r%r8anyr& test_keys)r3r7r9s` @r has_platformzProfileStatsFile.has_platformsF ::,,R0   MM335   rcZ|jjt|jduSrr& counts_forrr9r3s r has_statszProfileStatsFile.has_statss) MM $ $]D4E4E F  rcjt}|j}|jj||}|j||f}t ||kr0|j ||jr|jd}n!|jj||||f}|dz|j||f<|S)Nr ) rr9r&setdefault_countsr1lenrGr%r2 lineno_for)r3 callcountrXr9counts current_countresults rrizProfileStatsFile.results (( 00<H,,X|-CD  v;- ' MM) $zz F ((<@}%F8Eq7HX|34 rch|jjt|j}|g|ddyyrr^)r3rgs rrzProfileStatsFile.reset_counts2))-9J9JK  F1I rct}|j}|jj||}|sJd|j||f}|t |kr |||dz <n||d<|j r|jyy)Nz1replace() called before result() recorded a countr r!)rr9r&r_r1rdr%r2)r3rfrXr9rgrhs rreplacezProfileStatsFile.replaces (( ))(LAJJJv,,X|-CD 3v; &(1F=1$ %"F2J :: KKM rchtd|jz|jjy)NzWriting profile file %s)printr(r&r%r`s rr2zProfileStatsFile._writes$ '$**45 r) cumulativeN)r4strr-rpr,z Optional[str]returnNone)rqr )r7rrqr )r7rrqbool)rqrs)rfr0rqz#Optional[Tuple[Optional[int], int]])rqrr)rfr0rqrr)__name__ __module__ __qualname____doc__r5propertyr9r8r\rarirrlr2rrrrKsq  !"    :))  2 * * rc2ddlm}|dfd }|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. r) decoratorct D] }||i| t}t5|D] }||i|} cdddS#1swYyxYw)N)variance)rangecount_functions) fnargskwwarm timerangetimervtimesr}warmups rwrapz!function_call_count..wrapsh&M D OO %L h / ! %__ %   s A  A)rCallable[..., Any]rrrrrqr)sqlalchemy.utilr{)r}rrr{rs``` rfunction_call_countrs!* Krc #Kttjdtj s6tj s&tj dtjzttj}|jd|jtj|tj}|j }tj#|}|d}n|\}}t%d||fz|j&t)j*dtj,|j/tj0rtt2j4j7tj0\}}t8j+dd} |d| |xsd } |j;| t%d | ztj<rtj?|y|rqtA||z} tC||z | kD} | rOtj rtj?|ytEd |d |d zd|dtjFd yyw)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[, ]r<r!_z.profilezDumped stats to file %szAdjusted function call count z not within dz% of expected z , platform z;. Rerun with --write-profiles to regenerate this callcount.)$cProfiler _skip_test_exceptionrrar% skip_testr+rProfileenabledisablepstatsStatsrJstdout total_callsrirn sort_statsrer*r- print_statsr,r)r'splitextr dump_statsrrlr0absAssertionErrorr9) r}prstatsrfexpectedexpected_countline_nobaseext test_namedumpfiledeviancefaileds rrrs))*EFF  # # %n.B.B  .99 : L    BIIK JJL LLCJJ /E!!I$$Y/H"* (I~+F FGEbhhw(;(;<= GG$$^%8%89 c!'',R0 $i1B 1BC " '(23!!y) y8+,Y/08; ##&&y1$ "!C&&33     sI'I))rrrqrr)皙?r r)r}floatrr0rr0rqr)r)r}rrqzGenerator[None, None, None])&rw __future__rr. contextlibr)rNrrrJtypingrrrrrr r r profiles_filer rrutilrrrr ImportErrorr__annotations__rrrrcontextmanagerrryrrrs#  &'" #$( ' w%PPh;<#&47: AA[Hs;B??C C