o D2h3 @sddlZddlZddlZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z m Z mZGddde jZee _ddZGdd d eZGd d d ejZGd d d e jZGddde jZddZGddde jZGdddeZGdddeZGdddeZGddde jZ zddl!Z!Wn e"yYdSwze!#WdSe$yZ%zdZ&e 'e&j(e%dWYdZ%[%dSdZ%[%ww)N)StringIO)resultrunnersignalscs\eZdZdZdZdZdZfddZfddZfddZ fd d Z fd d Z Z S) NumbaTestProgramz A TestProgram subclass adding the following options: * a -R option to enable reference leak detection * a --profile option to enable profiling of the test run Currently the options are only added in 3.4+. Fcs4|dd|_tjdtt|j|i|dS)Nsuitez:x)popdiscovered_suitesys warnoptionsappendsuperr__init__)selfargskwargs __class__l/var/www/html/alexa/alex_system/speach-to-text/venv/lib/python3.10/site-packages/llvmlite/tests/customize.pyr!s zNumbaTestProgram.__init__cs(|jdur |j|_dStt|dSN)r testr r createTestsrrrrr(s  zNumbaTestProgram.createTestscsVtt|}|jdur|jdddddd|jddd dd d|jd d dd d|S)N-Rz --refleakrefleak store_truezDetect reference / memory leaks)destactionhelp-mz--multiprocess multiprocesszParallelize testsz --profileprofilezProfile the test run)r r_getParentArgParser testRunner add_argument)rparserrrrr#.s   z$NumbaTestProgram._getParentArgParsercsbtjdkrd|vr|dd|_d|vr|dd|_tt|||jdkr/d|_ dSdS)N)rTr r) r version_inforemoverr!r r parseArgs verbositybuffer)rargvrrrr+?s     zNumbaTestProgram.parseArgsc sjrt_ttdstdn jdurtj_j r*t jj j j d_fdd}jr{tjtjtjdjdd}tjtjd }|z||W|td |f||dS|td |f||w|dS) Ngettotalrefcountz^detecting reference leaks requires a debug build of Python, only memory leaks will be detected)r,failfastr-csttdSr)r rrunTestsr)rrrrrun_tests_real`sz1NumbaTestProgram.runTests..run_tests_real__main__rz.prof)timerz!Writing test profile data into %r)rRefleakTestRunnerr$hasattrr warningswarnunittestTextTestRunnerr!ParallelTestRunnerr,r0r-r"ospathsplitextbasenamemodules__file__cProfileProfiletime perf_counterenableruncalldisableprint dump_stats)rr2filenameprrrr1NsB      zNumbaTestProgram.runTests) __name__ __module__ __qualname____doc__rr"r!rrr#r+r1 __classcell__rrrrrs    rcCsztj}Wn tydd}Ynwztj}Wn ty%dd}YnwtjtjtjtjfD] }|dur:|q0t t ||fS)NcSdSN*rrrrrfunc1z_refleak_cleanup..func1cSrRrSrrrrrfunc2rVz_refleak_cleanup..func2) r getallocatedblocksAttributeErrorr/stdoutstderr __stdout__ __stderr__flush_clear_type_cachegccollect)rUrWstreamrrr_refleak_cleanup{s"       rcc@s eZdZdS)ReferenceLeakErrorN)rMrNrOrrrrrdsrdc@seZdZddZdS)IntPoolcCs|Srr)rkeyrrr __missing__rVzIntPool.__missing__N)rMrNrOrgrrrrres recs,eZdZdZdZddZfddZZS)RefleakTestResultr'c Cs|j|j}|j}dg||}dg||}t}tddD]}||q!d}} t|D]B}t} || | sO|j | j |j | j t ~ t\} } ||krm|| | |||<|| ||||<| | }} q0||fS)Nri8)rbr^ repetitionswarmupreranger TestResultrun wasSuccessfulfailuresextenderrorsAssertionErrorrc) rrrepcountnwarmup rc_deltas alloc_deltas _int_pooli alloc_before rc_beforeres alloc_afterrc_afterrrr _huntLeakss8      zRefleakTestResult._huntLeaksc sz ||\}}Wnty|rJYdSwdd}dd}d}|d|f|d|ffD]:\}}} | |rhd|||t|f} d }zt| tyTt} Ynw|jrb|j d ||f| || q.|sut t ||dSdS) NcSst|Sr)anydeltasrrrcheck_rc_deltassz5RefleakTestResult.addSuccess..check_rc_deltascSs2d|dt|kr dSt|tdksdSdS)Nr'rT)rF)countlensetrrrrcheck_alloc_deltass z8RefleakTestResult.addSuccess..check_alloc_deltasF referencesz memory blocksz%s leaked %s %s, sum=%sTz%s = %r )rrtrpsumrd Exceptionr exc_infoshowAllrbwrite addFailurer rh addSuccess) rrrwrxrrfailedr item_namecheckermsgrrrrrs:       zRefleakTestResult.addSuccess)rMrNrOrlrkrrrQrrrrrhs "rhc@seZdZeZdS)r5N)rMrNrOrh resultclassrrrrr5sr5cCs2t|tjrg}|D] }|t|q |S|gS)z$Expand suite into list of tests ) isinstancer9 TestSuiterr_flatten_suite)rtestsxrrrrs rc@seZdZdZddZdS)ParallelTestResultzA A TestResult able to inject results from other results. cCsv|j|j|j|j|j7_|j|j|j|j|j|j|j |j |j |j dS)zI Add the results from the other *result* to this result. N) rbrgetvaluer^testsRunrqrrrsskippedexpectedFailuresunexpectedSuccesses)rrrrr add_resultss zParallelTestResult.add_resultsN)rMrNrOrPrrrrrrs rc@s$eZdZdZdZddZddZdS)_MinimalResultz7 A minimal, picklable TestResult-alike object. )rqrsrrrrb shouldStoprcCs d|_dS)zR Remove any unpicklable attributes from TestCase instance *case*. N)_outcomeForDoCleanups)rcaserrr fixup_cases z_MinimalResult.fixup_casecCsp|jD] }t||t||q|jD] \}}||q|jD] \}}||q|jD] \}}||q,dSr) __slots__setattrgetattrrrrsrq)roriginal_resultattrr_rrrr"s    z_MinimalResult.__init__N)rMrNrOrPrrrrrrrrs  rc@s eZdZdZddZddZdS) _FakeStringIOz: A trivial picklable StringIO-alike for Python 2. cCs ||_dSr_value)rvaluerrrr2s z_FakeStringIO.__init__cCs|jSrrrrrrr5sz_FakeStringIO.getvalueN)rMrNrOrPrrrrrrr-s rc@s.eZdZdZddZddZejddZdS) _MinimalRunnerzu A minimal picklable object able to instantiate a runner in a child process and run a test case with it. cCs||_||_dSr) runner_cls runner_args)rrrrrrr?s z_MinimalRunner.__init__cCs|j}t|d<|jdi|}|}tt||j|_|j|_| | ||Wdn1s9wYt |j |_ t |S)Nrbr)rrr _makeResultrinstallHandlerregisterResultr0r-cleanup_objectrrbrr)rrrrrrrr__call__Fs    z_MinimalRunner.__call__c cs\t|j}z|VWt|j|}|D]}|j|=qdSt|j|}|D]}|j|=q&w)zu A context manager which cleans up unwanted attributes on a test case (or any other object). N)r__dict__)rr vanilla_attrsspurious_attrsnamerrrrYs   z_MinimalRunner.cleanup_objectN) rMrNrOrPrr contextlibcontextmanagerrrrrrr9s rcs4eZdZdZeZddZddZfddZZ S)r;z\ A test runner which delegates the actual running to a pool of child processes. cKs$tjj|fi|||_||_dSr)rr:rrr)rrrrrrrps zParallelTestRunner.__init__cCslt|j|j}t}|j}z|||jD] }|||jr!nq|W| | S| | wr) rrrmultiprocessingPoolimap_unordered _test_listrr terminatejoin)rr child_runnerpoolimap child_resultrrr _run_innerus   zParallelTestRunner._run_innercst||_tt||jSr)rrr r;ror)rrrrrros zParallelTestRunner.run) rMrNrOrPrrrrrorQrrrrr;hs r;z+Failed to enable faulthandler due to: {err})err)) collectionsrrBiorr`r<rr rDr9r7rrrmainrrc RuntimeErrorrd defaultdictreTextTestResultrhr:r5rrobjectrrrr; faulthandler ImportErrorrF BaseExceptionerr8formatrrrrsJ aQ  /%