B Ꮚ\!3 @sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddlm Z m Z m Z ddl mZGdddejZee_ddZGd d d eZGd d d ejZGd dde jZGddde jZddZGddde jZGdddeZGdddeZGdddeZGddde jZ y ddl!Z!Wne"k rTYnJXy e!#Wn:e$k rZ%zdZ&e 'e&j(e%dWddZ%[%XYnXdS)N)resultrunnersignals)StringIOcs\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+. Fcs0|dd|_tjdtt|j||dS)NZsuitez:x)popdiscovered_suitesys warnoptionsappendsuperr__init__)selfargskwargs) __class__7lib/python3.7/site-packages/llvmlite/tests/customize.pyr"s zNumbaTestProgram.__init__cs&|jdk r|j|_ntt|dS)N)r testr r createTests)r)rrrr*s  zNumbaTestProgram.createTestscsVtt|}|jdkr,|jdddddd|jddd dd d|jd d dd d|S)Nz-Rz --refleakrefleak store_truezDetect reference / memory leaks)destactionhelpz-mz--multiprocess multiprocesszParallelize testsz --profileprofilezProfile the test run)r r_getParentArgParser testRunner add_argument)rparser)rrrr0s   z$NumbaTestProgram._getParentArgParsercs^tjdkr:d|kr"|dd|_d|kr:|dd|_tt|||jdkrZd|_ dS)N)z-RTz-mr) r version_inforemoverrr r parseArgs verbositybuffer)rargv)rrrr&As    zNumbaTestProgram.parseArgsc sjr"t_ttds4tdnjdkr4tj_j rTt jj j j d_fdd}jrtjtjtjdjdd}tjtjd }|z||Wd|td |f||Xn|dS) Ngettotalrefcountz^detecting reference leaks requires a debug build of Python, only memory leaks will be detected)r'failfastr(csttdS)N)r rrunTestsr)rrrrrun_tests_realasz1NumbaTestProgram.runTests..run_tests_real__main__rz.prof)Ztimerz!Writing test profile data into %r)rRefleakTestRunnerrhasattrr warningswarnunittestTextTestRunnerrParallelTestRunnerr'r+r(rospathsplitextbasenamemodules__file__cProfileZProfiletimeZ perf_counterenableZruncalldisableprintZ dump_stats)rr-filenamep)r)rrr,Ps0     zNumbaTestProgram.runTests) __name__ __module__ __qualname____doc__rrrrrrr&r, __classcell__rr)rrrs    rcCsy tj}Wntk r&dd}YnXy tj}Wntk rNdd}YnXx,tjtjtjtjfD]}|dk rf|qfWt t ||fS)NcSsdS)N*rrrrrsz"_refleak_cleanup..cSsdS)NrHrrrrrrIs) r getallocatedblocksAttributeErrorr*stdoutstderr __stdout__ __stderr__flush_clear_type_cachegcZcollect)Zfunc1Zfunc2streamrrr_refleak_cleanup|s   rTc@s eZdZdS)ReferenceLeakErrorN)rCrDrErrrrrUsrUc@seZdZddZdS)IntPoolcCs|S)Nr)rkeyrrr __missing__szIntPool.__missing__N)rCrDrErXrrrrrVsrVcs,eZdZdZdZddZfddZZS)RefleakTestResultr"c Cs|j|j}|j}dg||}dg||}t}xtddD] }||qDWxt|D]}t}||| s|j |j |j |j t ~t\} } ||kr|| | |||<|| | |||<| | } } q^W||fS)Nri8)rSrP repetitionswarmuprVrangerZ TestResultrun wasSuccessfulfailuresextenderrorsAssertionErrorrT) rrZrepcountZnwarmup rc_deltas alloc_deltasZ _int_pooliZresZ alloc_afterZrc_afterZ rc_beforeZ alloc_beforerrr _huntLeakss,    zRefleakTestResult._huntLeaksc sy||\}}Wn tk r2|r.tdSXdd}dd}d}x|d|f|d|fgD]z\}}} | |r^d|||t|f} d }y t| Wntk rt} YnX|jr|j d ||f| || q^W|st t ||dS) NcSst|S)N)any)deltasrrrcheck_rc_deltassz5RefleakTestResult.addSuccess..check_rc_deltascSs2d|dt|krdSt|tdks.dSdS)Nr"rT)rF)countlenset)rjrrrcheck_alloc_deltass z8RefleakTestResult.addSuccess..check_alloc_deltasFZ referencesz memory blocksz%s leaked %s %s, sum=%sTz%s = %r )rhrdr`sumrU Exceptionr exc_infoZshowAllrSwriteZ addFailurer rY addSuccess) rrrerfrkrqZfailedrjZ item_nameZcheckermsgrt)rrrrvs0   zRefleakTestResult.addSuccess)rCrDrEr]r\rhrvrGrr)rrrYsrYc@seZdZeZdS)r/N)rCrDrErY resultclassrrrrr/sr/cCs:t|tjr0g}x|D]}|t|qW|S|gSdS)z$Expand suite into list of tests N) isinstancer3Z TestSuiterb_flatten_suite)rZtestsxrrrrzs   rzc@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) rSrugetvaluerPtestsRunrarbrcskippedexpectedFailuresunexpectedSuccesses)rrrrr add_resultss zParallelTestResult.add_resultsN)rCrDrErFrrrrrr|sr|c@s$eZdZdZdZddZddZdS)_MinimalResultz7 A minimal, picklable TestResult-alike object. )rarcrrrrS shouldStopr~cCs d|_dS)zR Remove any unpicklable attributes from TestCase instance *case*. N)Z_outcomeForDoCleanups)rcaserrr fixup_casesz_MinimalResult.fixup_casecCsx |jD]}t||t||qWx|jD]\}}||q*Wx|jD]\}}||qHWx|jD]\}}||qfWdS)N) __slots__setattrgetattrrrrcra)rZoriginal_resultattrr_rrrrs z_MinimalResult.__init__N)rCrDrErFrrrrrrrr src@s eZdZdZddZddZdS) _FakeStringIOz: A trivial picklable StringIO-alike for Python 2. cCs ||_dS)N)_value)rvaluerrrr-sz_FakeStringIO.__init__cCs|jS)N)r)rrrrr}0sz_FakeStringIO.getvalueN)rCrDrErFrr}rrrrr(src@s.eZdZdZddZddZejddZdS) _MinimalRunnerzu A minimal picklable object able to instantiate a runner in a child process and run a test case with it. cCs||_||_dS)N) runner_cls runner_args)rrrrrrr:sz_MinimalRunner.__init__c Cs||j}t|d<|jf|}|}tt||j|_|j|_| |||WdQRXt |j |_ t |S)NrS)rrrZ _makeResultrZinstallHandlerZregisterResultr+r(cleanup_objectrrSr}r)rrrrrrrr__call__As    z_MinimalRunner.__call__c cs@t|j}z |VWdt|j|}x|D] }|j|=q*WXdS)zu A context manager which cleans up unwanted attributes on a test case (or any other object). N)rp__dict__)rrZ vanilla_attrsZspurious_attrsnamerrrrTs    z_MinimalRunner.cleanup_objectN) rCrDrErFrr contextlibcontextmanagerrrrrrr4srcs4eZdZdZeZddZddZfddZZ S)r5z\ A test runner which delegates the actual running to a pool of child processes. cKs tjj|f|||_||_dS)N)rr4rrr)rrrrrrrkszParallelTestRunner.__init__cCs`t|j|j}t}|j}z,x&|||jD]}|||jr,Pq,W|S| | XdS)N) rrrmultiprocessingZPoolZimap_unordered _test_listrrZ terminatejoin)rrZ child_runnerZpoolZimapZ child_resultrrr _run_innerps zParallelTestRunner._run_innercst||_tt||jS)N)rzrr r5r_r)rr)rrrr_s zParallelTestRunner.run) rCrDrErFr|rxrrr_rGrr)rrr5cs r5z+Failed to enable faulthandler due to: {err})err)) collectionsrr<rRr6rr r=r3r1rrrZsixrmainrrT RuntimeErrorrU defaultdictrVZTextTestResultrYr4r/rzr|objectrrrr5Z faulthandler ImportErrorr> BaseExceptionerwr2formatrrrrsB aM  /%