o i3 @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)NZsuitez:x)popdiscovered_suitesys warnoptionsappendsuperr__init__)selfargskwargs __class__g/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/llvmlite/tests/customize.pyr !s zNumbaTestProgram.__init__cs(|jdur |j|_dStt|dSN)rtestr 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)rTrr) 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+r/r,r!ospathsplitextbasenamemodules__file__cProfileZProfiletime perf_counterenableZruncalldisableprintZ dump_stats)rr1filenameprrrr0NsB      zNumbaTestProgram.runTests) __name__ __module__ __qualname____doc__rr!r r rr"r*r0 __classcell__rrrrrs    rcCsztj}Wn tydd}Ynwztj}Wn ty%dd}YnwtjtjtjtjfD] }|dur:|q0t t ||fS)NcSdSN*rrrrrfunc1z_refleak_cleanup..func1cSrNrOrrrrrfunc2rRz_refleak_cleanup..func2) r getallocatedblocksAttributeErrorr.stdoutstderr __stdout__ __stderr__flush_clear_type_cachegcZcollect)rQrSstreamrrr_refleak_cleanup{s"       r^c@s eZdZdS)ReferenceLeakErrorN)rIrJrKrrrrr_sr_c@seZdZddZdS)IntPoolcCs|Srr)rkeyrrr __missing__rRzIntPool.__missing__N)rIrJrKrbrrrrr`s r`cs,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)r]rZ repetitionswarmupr`rangerZ TestResultrun wasSuccessfulfailuresextenderrorsAssertionErrorr^) rrZrepcountZnwarmup rc_deltas alloc_deltasZ _int_pooliZ alloc_beforeZ rc_beforeresZ alloc_afterZrc_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)countlensetrurrrcheck_alloc_deltass z8RefleakTestResult.addSuccess..check_alloc_deltasFZ referencesz memory blocksz%s leaked %s %s, sum=%sTz%s = %r )rsrnrjsumr_ Exceptionr exc_infoZshowAllr]writeZ addFailurer rc addSuccess) rrrorprwr}failedrvZ item_namecheckermsgrrrrrs:       zRefleakTestResult.addSuccess)rIrJrKrgrfrsrrMrrrrrcs "rcc@seZdZeZdS)r4N)rIrJrKrc resultclassrrrrr4sr4cCs2t|tjrg}|D] }|t|q |S|gS)z$Expand suite into list of tests ) isinstancer8Z TestSuiterl_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) r]rgetvaluerZtestsRunrkrlrmskippedexpectedFailuresunexpectedSuccesses)rrrrr add_resultss zParallelTestResult.add_resultsN)rIrJrKrLrrrrrrs rc@s$eZdZdZdZddZddZdS)_MinimalResultz7 A minimal, picklable TestResult-alike object. )rkrmrrrr] shouldStoprcCs d|_dS)zR Remove any unpicklable attributes from TestCase instance *case*. N)Z_outcomeForDoCleanups)rcaserrr fixup_cases z_MinimalResult.fixup_casecCsp|jD] }t||t||q|jD] \}}||q|jD] \}}||q|jD] \}}||q,dSr) __slots__setattrgetattrrrrmrk)rZoriginal_resultattrr_rrrr "s    z_MinimalResult.__init__N)rIrJrKrLrrr rrrrrs  rc@s eZdZdZddZddZdS) _FakeStringIOz: A trivial picklable StringIO-alike for Python 2. cCs ||_dSr_value)rvaluerrrr 2s z_FakeStringIO.__init__cCs|jSrrrrrrr5sz_FakeStringIO.getvalueN)rIrJrKrLr rrrrrr-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)Nr]r)rrrZ _makeResultrZinstallHandlerZregisterResultr/r,cleanup_objectrr]rr)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__)rrZ vanilla_attrsZspurious_attrsnamerrrrYs   z_MinimalRunner.cleanup_objectN) rIrJrKrLr r contextlibcontextmanagerrrrrrr9s rcs4eZdZdZeZddZddZfddZZ S)r:z\ A test runner which delegates the actual running to a pool of child processes. cKs$tjj|fi|||_||_dSr)rr9r rr)rrrrrrr ps zParallelTestRunner.__init__cCslt|j|j}t}|j}z|||jD] }|||jr!nq|W| | S| | wr) rrrmultiprocessingZPoolZimap_unordered _test_listrr terminatejoin)rrZ child_runnerpoolimapZ child_resultrrr _run_innerus   zParallelTestRunner._run_innercst||_tt||jSr)rrr r:rir)rrrrrris zParallelTestRunner.run) rIrJrKrLrrr rrirMrrrrr:hs r:z+Failed to enable faulthandler due to: {err})err)) collectionsrrAiorr\r;rr rBr8r6rrrmainrr^ RuntimeErrorr_ defaultdictr`ZTextTestResultrcr9r4rrobjectrrrr: faulthandler ImportErrorrD BaseExceptionerr7formatrrrrsJ aQ  /%