o i @sUdZddlmZmZmZmZddlmZddlm Z m Z ddl Z ddl m Z ddlmZmZzddlZWn ey@dZYnwd Zed d ZgZeged fed <ded fddZe ddZGdd d e ZGdddeZejddZdS)zw Implement code coverage support. Currently contains logic to extend ``coverage`` with lines covered by the compiler. )OptionalSequenceCallable no_type_check) defaultdict)ABCabstractmethodN)cache)irconfigFTcCsd}tr tj}|S)z>Get active coverage instance or return None if not found. N)coverage_availablecoverageCoveragecurrent)covrX/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/numba/misc/coverage_support.pyget_active_coverages r NotifyLocBase _the_registryreturncCs&tjsgSttddddtDS)zC Returns a list of the registered NotifyLocBase instances. cSs|duSNr)xrrr,sz+get_registered_loc_notify..css|]}|VqdSrr).0factoryrrr -sz,get_registered_loc_notify..)r JIT_COVERAGElistfilterrrrrrget_registered_loc_notify%s   r cs:tjddtdusJdtjfdd}S)zm Make a singleton ``CoverageData``. Avoid writing to disk. Other processes can corrupt the file. T)no_diskNzno active Coverage instancecsdSr)get_dataupdaterrcovdatarr _finalize:sz%_get_coverage_data.._finalize)r CoverageDataratexitregister)r&rr$r_get_coverage_data0s r*c@s6eZdZdZedejddfddZed ddZdS) rzAInterface for notifying visiting of a ``numba.core.ir.Loc``. locrNcCdSrrselfr+rrrnotifyEzNotifyLocBase.notifycCr,rrr.rrrcloseIr0zNotifyLocBase.close)rN) __name__ __module__ __qualname____doc__rr Locr/r2rrrrrBs c@s0eZdZdZddZdejfddZddZd S) NotifyCompilerCoveragez Use to notify ``coverage`` about compiled lines. The compiled lines are under the "numba_compiled" context in the coverage data. cCstt|_dSr)rset _arcs_datar1rrr__init__UszNotifyCompilerCoverage.__init__r+cCs.|jdr|j|j|j|jfdSdS)Nz.py)filenameendswithr:addliner-rrrr/Xs zNotifyCompilerCoverage.notifycCsHt}|j|d||jWddS1swYdS)Nnumba_compiled)r*_lock set_contextadd_arcsr:)r.r%rrrr2]s  "zNotifyCompilerCoverage.closeN) r3r4r5r6r;r r7r/r2rrrrr8Ns  r8cCstdurtSdSr)rr8rrrr_register_coverage_notifierds rD)r6typingrrrr collectionsrabcrrr( functoolsr numba.corer r r ImportErrorr rr__annotations__r r*rr8appendrDrrrrs.