U +d3 @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 z ddl!Z!Wne"k rTYnJXz e!#Wn:e$k rZ%zdZ&e 'e&j(e%dW5dZ%[%XYnXdS)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+. Fcs0|dd|_tjdtt|j||dS)NZsuitez:x)popdiscovered_suitesys warnoptionsappendsuperr__init__)selfargskwargs __class__.run_tests_real__main__rz.prof)Ztimerz!Writing test profile data into %r)rRefleakTestRunnerr#hasattrr warningswarnunittestTextTestRunnerr ParallelTestRunnerr+r/r,r!ospathsplitextbasenamemodules__file__cProfileZProfiletime perf_counterenabledisableprintZ dump_statsZruncall)rr1filenameprrrr0Ns:   zNumbaTestProgram.runTests) __name__ __module__ __qualname____doc__rr!r r rr"r*r0 __classcell__rrrrrs    rcCsz tj}Wntk r&dd}YnXz tj}Wntk rNdd}YnXtjtjtjtjfD]}|dk rd|qdt t ||fS)NcSsdSN*rrrrrfunc1sz_refleak_cleanup..func1cSsdSrMrrrrrfunc2sz_refleak_cleanup..func2) r getallocatedblocksAttributeErrorr.stdoutstderr __stdout__ __stderr__flush_clear_type_cachegcZcollect)rOrPstreamrrr_refleak_cleanup{s   r[c@s eZdZdS)ReferenceLeakErrorN)rHrIrJrrrrr\sr\c@seZdZddZdS)IntPoolcCs|Srr)rkeyrrr __missing__szIntPool.__missing__N)rHrIrJr_rrrrr]sr]cs,eZdZdZdZddZfddZZS)RefleakTestResultr&c Cs|j|j}|j}dg||}dg||}t}tddD] }||qBd}} t|D]}t} || | s|j | j |j | j t ~ t\} } ||kr|| | |||<|| ||||<| | }} q`||fS)Nri8)rZrW repetitionswarmupr]rangerZ TestResultrun wasSuccessfulfailuresextenderrorsAssertionErrorr[) rrZrepcountZnwarmup rc_deltas alloc_deltasZ _int_pooliZ alloc_beforeZ rc_beforeresZ alloc_afterZrc_afterrrr _huntLeakss8      zRefleakTestResult._huntLeaksc sz||\}}Wn"tk r4|r.tYdSXdd}dd}d}|d|f|d|ffD]z\}}} | |r^d|||t|f} d }z t| Wntk rt} YnX|jr|j d ||f| || q^|st t ||dS) NcSst|Sr)anydeltasrrrcheck_rc_deltassz5RefleakTestResult.addSuccess..check_rc_deltascSs2d|dt|krdSt|tdks.dSdS)Nr&rT)rF)countlensetrrrrrcheck_alloc_deltass z8RefleakTestResult.addSuccess..check_alloc_deltasFZ referencesz memory blocksz%s leaked %s %s, sum=%sTz%s = %r )rprkrgsumr\ Exceptionr exc_infoZshowAllrZwriteZ addFailurer r` addSuccess) rrrlrmrtrzfailedrsZ item_namecheckermsgr}rrrrs8   zRefleakTestResult.addSuccess)rHrIrJrdrcrprrLrrrrr`s"r`c@seZdZeZdS)r3N)rHrIrJr` resultclassrrrrr3sr3cCs6t|tjr,g}|D]}|t|q|S|gSdS)z$Expand suite into list of tests N) isinstancer7Z TestSuiteri_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) rZr~getvaluerWtestsRunrhrirjskippedexpectedFailuresunexpectedSuccesses)rrrrr add_resultss zParallelTestResult.add_resultsN)rHrIrJrKrrrrrrsrc@s$eZdZdZdZddZddZdS)_MinimalResultz7 A minimal, picklable TestResult-alike object. )rhrjrrrrZ shouldStoprcCs d|_dS)zR Remove any unpicklable attributes from TestCase instance *case*. N)Z_outcomeForDoCleanups)rcaserrr fixup_casesz_MinimalResult.fixup_casecCsp|jD]}t||t||q|jD]\}}||q$|jD]\}}||q>|jD]\}}||qXdSr) __slots__setattrgetattrrrrjrh)rZoriginal_resultattrr_rrrr "s   z_MinimalResult.__init__N)rHrIrJrKrrr rrrrrsrc@s eZdZdZddZddZdS) _FakeStringIOz: A trivial picklable StringIO-alike for Python 2. cCs ||_dSr_value)rvaluerrrr 2sz_FakeStringIO.__init__cCs|jSrrrrrrr5sz_FakeStringIO.getvalueN)rHrIrJrKr rrrrrr-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||_||_dSr) runner_cls runner_args)rrrrrrr ?sz_MinimalRunner.__init__c Cs||j}t|d<|jf|}|}tt||j|_|j|_| |||W5QRXt |j |_ t |S)NrZ)rrrZ _makeResultrZinstallHandlerZregisterResultr/r,cleanup_objectrrZrr)rrrrrrrr__call__Fs    z_MinimalRunner.__call__c cs<t|j}z |VW5t|j|}|D] }|j|=q(XdS)zu A context manager which cleans up unwanted attributes on a test case (or any other object). N)ry__dict__)rrZ vanilla_attrsZspurious_attrsnamerrrrYs   z_MinimalRunner.cleanup_objectN) rHrIrJrKr r contextlibcontextmanagerrrrrrr9s rcs4eZdZdZeZddZddZfddZZ S)r9z\ A test runner which delegates the actual running to a pool of child processes. cKs tjj|f|||_||_dSr)rr8r rr)rrrrrrr pszParallelTestRunner.__init__cCsbt|j|j}t}|j}z.|||jD]}| ||j r*qDq*|WS||XdSr) rrrmultiprocessingZPoolZimap_unordered terminatejoin _test_listrr)rrZ child_runnerpoolimapZ child_resultrrr _run_innerus zParallelTestRunner._run_innercst||_tt||jSr)rrr r9rfr)rrrrrrfs zParallelTestRunner.run) rHrIrJrKrrr rrfrLrrrrr9hs r9z+Failed to enable faulthandler due to: {err})err)) collectionsrr@iorrYr:rr rAr7r5rrrmainrr[ RuntimeErrorr\ defaultdictr]ZTextTestResultr`r8r3rrobjectrrrr9 faulthandler ImportErrorrC BaseExceptionerr6formatrrrrsB aQ  /%