jA'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 ddlZn #e$rdZYnwxYwdZded< da ded< d'dZ!GddZ" d(d)d$Z#ej$d*d+d&Z%dS),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_testid_returnNonecX|atjrtdSdSN)rr force_write reset_count)rs /home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib64/python3.11/site-packages/sqlalchemy/testing/profiling.py_start_current_testrCs4M!%""$$$$$%%cpeZdZdZ ddd Zedd ZddZddZd dZ d!dZ d"dZ d#dZ d"dZ dS)$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. cumulativeNfilenamestrsortdump Optional[str]rrctjduotjj|_|jptjduotjj|_t ||_|jj|_ tj |j d|_ ||_ ||_tjt"|_|jr|dSdS)N)r optionsforce_write_profilesrwrite_profileswriterprofilespathfnameossplit short_fnamer&r% collections defaultdictint_current_counts_write)selfr#r%r&s r__init__zProfileStatsFile.__init__Ys N$ & N6>+N % N$ & H6>+H %X.. ]' 7==44R8   #C ( (  :  KKMMMMM  r r c\tj}| Jd||S)Nzno database configured)r dbplatform_key_for)r9r<s r platform_keyzProfileStatsFile.platform_keyvs- Y~~7~~~$$R(((r r<rc g}|jjr|d|t j|t jjdt jj }tr|dz }ttj tjtj||j|jt'|t)S)Nasync.t)machinesystemimplementationpythondialectdriver dbapi_flags cextensions)rGis_asyncappendextendr profile_platform_tokenssys version_infomajorminorrr platformrCrDlowerpython_implementationnamerHtupler)r9r<rI py_versions rr=z!ProfileStatsFile.platform_key_for|s :  (   w ' ' ' 9r9s r z0ProfileStatsFile.has_platform..sL   DM77AA A      r )r-r=anyr. test_keys)r9r<r>s` @r has_platformzProfileStatsFile.has_platformss : 4,,R00       M3355      r cP|jt|jduSrr. counts_forrr>r9s r has_statszProfileStatsFile.has_statss' M $ $]D4E F F  r callcountr6#Optional[Tuple[Optional[int], int]]cft}|j}|j||}|j||f}t ||kr3|||jr|d}n#|j ||||f}|dz|j||f<|S)Nr ) rr>r.setdefault_countsr7lenrLr-r8 lineno_for)r9rhr^r>counts current_countresults rrpzProfileStatsFile.results ( 00<HH,X|-CD  v;;- ' ' MM) $ $ $z  FF ((<@@}%F8Eq7HX|34 r cf|jt|j}| g|dd<dSdSrrd)r9rns rrzProfileStatsFile.reset_counts;))-9JKK  F111III  r c t}|j}|j||}|s Jd|j||f}|t |kr |||dz <n||d<|jr|dSdS)Nz1replace() called before result() recorded a countr r))rr>r.rer7rlr-r8)r9rhr^r>rnros rreplacezProfileStatsFile.replaces ( ))(LAAJJJJJJ,X|-CD 3v;; & &(1F=1$ % %"F2J :  KKMMMMM  r cftd|jz|jdS)NzWriting profile file %s)printr0r.r-rfs rr8zProfileStatsFile._writes2 '$*4555 r )r"N)r#r$r%r$r&r'rr)rr )r<rrr )r<rrrY)rrY)rhr6rri)rr)rhr6rr)__name__ __module__ __qualname____doc__r:propertyr>r=rbrgrprrsr8r rrrKs   !" :)))X)     2    *    *    r 皙?variancefloattimesr6warmupCallable[..., Any]c8ddlm}|d fd }|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) decoratorfnrargsrkwrct D] }||i| t}t5|D] }||i|} |cdddS#1swxYwYdS)N)r})rangecount_functions) rrrwarm timerangetimervrr}rs rwrapz!function_call_count..wraps&MM  D BOOOOO%LL h / / /  ! % %R___                  sAAA)rrrrrrrr)sqlalchemy.utilr)r}rrrrs``` rfunction_call_countrsR*)))))Y Kr Generator[None, None, None]c #Kttjdts-tjs!tjdtjzttj }| dV| tj |tj}|j}t|}|d}n|\}}t%d||fz|jt)jdtj|tjr~t2jtj\}}t8dd} |d| |pd } || t%d | ztjrt|dS|r{tA||z} tC||z | k} | rStjrt|dStEd |d |d zd|dtj#d dSdS)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[, ]rAr)_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_exceptionrrgr- skip_testr3rProfileenabledisablepstatsStatsrOstdout total_callsrpru sort_statsrer2r% print_statsr&r1r/splitextr dump_statsrrsr6absAssertionErrorr>) r}prstatsrhexpectedexpected_countline_nobaseext test_namedumpfiledeviancefaileds rrrs)*EFFF  # # % % n.B   .9 :   LLL    BIIKKK EEEJJLLL LCJ / / /E!I$$Y//H"* (I~+F FGGGEbhw(;<<== 4G$$^%899 c!'',,R0 $ii1B 1BC """ '(2333!y))))) y8+,,Y/008;  # &&y11111$n " !C&&333      r )rrrr)r|r r)r}r~rr6rr6rr)r|)r}r~rr)&ry __future__rr4 contextlibr1rSrrrOtypingrrrrrr r r profiles_filer rrutilrrrr ImportErrorr__annotations__rrrrcontextmanagerrr{r rrsj#"""""  &&&&&&''''''"""""" ######OOOOHHH$('''' %%%%PPPPPPPPh;<: AAAAAAAs;BB  B