\ K]c@`sddlmZmZmZddljZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlmZmZmZmZmZmZddlmZddlmZmZddlmZyddl mZWn!ek r>dd lm ZnXd Z!d e"fd YZ#d ej$fdYZ%dej&fdYZ'dZ(ddZ)dZ*dZ+de,fdYZ-dej.fdYZ/dej0fdYZ1dej2fdYZ3dej0fdYZ4d e"fd!YZ5d"e"fd#YZ6d$e"fd%YZ7d&Z8d'ej2fd(YZ9dS()i(tprint_functiontdivisiontabsolute_importN(tresulttrunnertsignalstsuitetloadertcasei(t TestLoader(t PYVERSIONtStringIO(tconfig(t TimeoutError(tEmptyc`sfd}|S(sO Create a decorator allowing tests to be tagged with the *known_tags*. c`sFx0D](}|krtd|fqqWfd}|S(s Tag a test method with the given tags. Can be used in conjunction with the --tags command-line argument for runtests.py. sunknown tag: %rc`st| s/t|ts/|jjd r>tdny |j}Wn!tk rnt}|_nX|j |S(Nttest_s(@tag(...) should be used on test methods( tcallablet isinstancettypet__name__t startswitht TypeErrorttagstAttributeErrortsettupdate(tfuncts(R(s1lib/python2.7/site-packages/numba/testing/main.pytdecorate+s   (t ValueError(RttR(t known_tags(Rs1lib/python2.7/site-packages/numba/testing/main.pyttag!s    ((RR ((Rs1lib/python2.7/site-packages/numba/testing/main.pytmake_tag_decoratorst TestListercB`s eZdZdZdZRS(s5Simply list available tests rather than running them.cC`sdS(N((tself((s1lib/python2.7/site-packages/numba/testing/main.pyt__init__<scC`sltjtjdtdd}t||_x!|jD]}t|jq7Wtdt |j|S(Nt descriptionst verbosityis%d tests found( RtTextTestResulttsyststderrtTruet_flatten_suitet _test_listtprinttidtlen(R#ttestRR((s1lib/python2.7/site-packages/numba/testing/main.pytrun?s (Rt __module__t__doc__R$R1(((s1lib/python2.7/site-packages/numba/testing/main.pyR":s t SerialSuitecB`seZdZdZRS(sAA simple marker to make sure tests in this suite are run serially. Note: As the suite is going through internals of unittest, it may get unpacked and stuffed into a plain TestSuite. We need to set an attribute on the TestCase objects to remember they should not be run in parallel. cC`sVt|tjs3x=|D]}|j|qWnt|_tt|j|dS(N(RtunittesttTestCasetaddTesttFalset_numba_parallel_test_tsuperR4(R#R0R((s1lib/python2.7/site-packages/numba/testing/main.pyR7Qs   (RR2R3R7(((s1lib/python2.7/site-packages/numba/testing/main.pyR4HstNumbaTestProgramcB`sweZdZeZeZeZeZdZ dZ dZ dZ dZ dZdZdZddZdZRS( s3 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 * a -m option for parallel execution * a -l option to (only) list tests Currently the options are only added in 3.4+. i*cO`stjr"tjdtt_n|jdd}t||dR:R;R$(R#targstkwargsR<((s1lib/python2.7/site-packages/numba/testing/main.pyR$ss   c C`stt|j}|jdkrI|jddddddddn|jd d dd d d dtdtjdd|jdddddddd|jddddt dd|jddddt dd|jddddt dd|jddddddd |S(!Ns-Rs --refleaktdesttrefleaktactiont store_truethelpsDetect reference / memory leakss-ms--multiprocesst multiprocesstnargst?RtconstsParallelize testss-ls--listtlistsList tests without running thems--tagsRsAComma-separated list of tags to select a subset of the test suites--exclude-tagst exclude_tagssDComma-separated list of tags to de-select a subset of the test suites--randomt random_selects$Random proportion of tests to selects --profiletprofilesProfile the test run( R:R;t_getParentArgParsert testRunnerRCt add_argumenttinttmultiprocessingt cpu_counttstrtfloat(R#tparser((s1lib/python2.7/site-packages/numba/testing/main.pyRUs.  c C`sud}xB|D]:}||kr |dkr4|}qGtd|q q W|dk rq|j|}yh||kr||dj}|j|n4d|kr|jddj}n tdWn'tk rd}t||nX|jdrtdnd|kr;d}t||n|d j dd }t ||||j|ndS( Nsargument %s supplied repeatedlyit=t unreachables,%s requires at least one tag to be specifiedt-s,tag starts with '-', probably a syntax errors1%s argument contains '=', probably a syntax errorit_( RCRtindextstriptremovetsplittAssertionErrort IndexErrorRtreplacetsetattr( R#targvttagstrtfoundtxtposnttag_argstmsgtattr((s1lib/python2.7/site-packages/numba/testing/main.pyt _handle_tagss4         cC`sOd|kr%|jdt|_ntdkrd|kr|jd}tj}yh||d}|dkryt|}Wn'tk rd}t ||nX|j|nWnt k rnX||_ |jdn|j |d|j |d nt t|j|t|d  sF|jj rb|jf|_|jn|jrg|jjd D]}|j^q~}t|j|d d |_n|jrg|jjd D]}|j^q}t|j|d d|_n|jr0t|j|j|j|_n|jdkrKt|_ndS(Ns-liis-mis--s7Expected an integer argument to option `-m`, found "%s"s--tagss--exclude-tagsR0t,tmodetincludetexcludei(ii( RdR*RQR RbRYRZRXt BaseExceptionRRgRMRrR:R;t parseArgsthasattrR0tcountTestCasest defaultTestt testNamest createTestsRReRct_choose_tagged_testsRRRSt_choose_random_testst random_seedR&tbuffer(R#Rjt dashm_posntnprocstm_optionRpRR((s1lib/python2.7/site-packages/numba/testing/main.pyRxsJ             + + cC`sdS(N((R#RjtLoader((s1lib/python2.7/site-packages/numba/testing/main.pyt _do_discoverysc `sjr4t_ttdsjtjdqjn6jrLt_njdkrjt j _nj rj rj dkrd}t|ntjj djdjdj_nfd}jr~tjjtjjtjd jd d }tjd tj}|jz|j|Wd|j t!d |f|j"|Xn|dS(Ntgettotalrefcounts^detecting reference leaks requires a debug build of Python, only memory leaks will be detectedisSValue specified for the number of processes to use in running the suite must be > 0R&tfailfastRc`sttjdS(N(R:R;trunTests((R#(s1lib/python2.7/site-packages/numba/testing/main.pytrun_tests_real#st__main__is.profttimers!Writing test profile data into %r(#RItRefleakTestRunnerRVRyR(R@RARQR"RCR5tTextTestRunnerRMR>RtParallelTestRunnerR&RRRTtostpathtsplitexttbasenametmodulest__file__tcProfiletProfilettimet perf_countertenabletruncalltdisableR-t dump_stats(R#RpRtfilenametp((R#s1lib/python2.7/site-packages/numba/testing/main.pyR s<            N(RR2R3R8RIRTRMRQRCRRRRSRR$RURrRxRR(((s1lib/python2.7/site-packages/numba/testing/main.pyR;_s    % > cC`sTt|tjttfrIg}x!|D]}|jt|q(W|S|gSdS(s6 Expand nested suite into list of test cases. N(RR5t TestSuiteRQttupletextendR+(R0ttestsRm((s1lib/python2.7/site-packages/numba/testing/main.pyR+6s  RucC`sg}t|}xt|D]}t|tjs=tt||j}y |j}Wnt k ronXt|dd}|dkr|dk r||@r|j |qq|dkr|dks||@ r|j |qqt d|qWtj |S(s Select tests that are tagged/not tagged with at least one of the given tags. Set mode to 'include' to include the tests with tags, or 'exclude' to exclude the tests with the tags. RRuRvsInvalid 'mode' supplied: %s.N(RR+RR5R6Rftgetattrt_testMethodNametim_funcRRCRERR(RRRttselectedR0Rt found_tags((s1lib/python2.7/site-packages/numba/testing/main.pyR~Cs$     cC`s~tj}|j|t|tjr:t|}n|j|tt ||}t |dd}tj|S(s7 Choose a given proportion of tests at random. tkeycS`s |jS(N(R.(R((s1lib/python2.7/site-packages/numba/testing/main.pytkt( trandomtRandomtseedRR5RR+tsampleRXR/tsorted(RtratioRtrnd((s1lib/python2.7/site-packages/numba/testing/main.pyRbs  "cC`stj}y tj}Wntk r2d}nXx?tjtjtjtjfD]}|dk rR|j qRqRWtj t j ||fS(NcS`sdS(Ni*((((s1lib/python2.7/site-packages/numba/testing/main.pyRxR( R(tgetallocatedblocksRRtstdoutR)t __stdout__t __stderr__RCtflusht_clear_type_cachetgctcollect(tfunc1tfunc2tstream((s1lib/python2.7/site-packages/numba/testing/main.pyt_refleak_cleanuprs    %   tReferenceLeakErrorcB`seZRS((RR2(((s1lib/python2.7/site-packages/numba/testing/main.pyRstIntPoolcB`seZdZRS(cC`s|S(N((R#R((s1lib/python2.7/site-packages/numba/testing/main.pyt __missing__s(RR2R(((s1lib/python2.7/site-packages/numba/testing/main.pyRstRefleakTestResultcB`s&eZdZdZdZdZRS(iic C`s;|jj|j}|j}dg||}dg||}t}xtddD]}||qZWxt|D]}tj}|j||j s|j j |j |j j |j t n~t\} } ||kr || | |||<|| | |||s   (RR2R3R$Rt contextlibtcontextmanagerR(((s1lib/python2.7/site-packages/numba/testing/main.pyRs  cC`sg}g}d}t|tjrgxa|D]2}t|\}}|j||j|q.Wn$||r|g}n t|}||fS(s: Split test suite into parallel and serial tests. cS`sJ|j}t||}|j|kr:|jdkr:tSt|dtS(Nt testFailureR9(RRRR8R*(R0t method_nametmethod((s1lib/python2.7/site-packages/numba/testing/main.pytis_parallelizable_test_caseTs  (RR5Rt_split_nonparallel_testsRR+(R0tptestststestsR RRR((s1lib/python2.7/site-packages/numba/testing/main.pyR Ms      RcB`s>eZdZeZdZdZdZdZdZ RS(s\ A test runner which delegates the actual running to a pool of child processes. i,cK`s2tjj||||_||_||_dS(N(RRR$RRR(R#RRRG((s1lib/python2.7/site-packages/numba/testing/main.pyR$ws  c C`s t|j|j}d}gtdt|j|D]}|j|||!^q7}x|D]|}tj|j}zVy|j ||||Wn|j n"X|j r|j Pn |j Wd|j Xq^W|j st|j}|j||SdS(Nii(RRRRR/t_ptestsRYtPoolRt_run_parallel_testst terminateRtclosetjoinR4t_stestsR1( R#Rt child_runnert chunk_sizeRtsplitted_testsRtpoolR ((s1lib/python2.7/site-packages/numba/testing/main.pyt _run_inner}s*9      c C`std|D}|j||}xtry|j|j}Wndtk r[dStk r}ddjdt|D} | f|j d|_ |q+X|j ||j |j |j r+t|_ dSq+WdS(Ncs`s|]}|jVqdS(N(R.(t.0R((s1lib/python2.7/site-packages/numba/testing/main.pys ss3Tests didn't finish before timeout (or crashed): %sRcs`s|]}d|VqdS(s- %r N((Rttid((s1lib/python2.7/site-packages/numba/testing/main.pys si(Rtimap_unorderedR*t__next__ttimeoutt StopIterationR RRRFRtdiscardRR( R#RRRRt remaining_idstitt child_resultteRp((s1lib/python2.7/site-packages/numba/testing/main.pyRs"       cC`s1t|\|_|_tt|j|jS(N(R R RR:RR1R(R#R0((s1lib/python2.7/site-packages/numba/testing/main.pyR1s( RR2R3RRRR$RRR1(((s1lib/python2.7/site-packages/numba/testing/main.pyRms  " (:t __future__RRRtnumba.unittest_supporttunittest_supportR5t collectionsRRRRYRRR(RR@RRRRRRR t numba.utilsR R tnumbaR R t ImportErrortQueueRR!tobjectR"RR4tmainR;R+R~RRt RuntimeErrorRt defaultdictRR'RRRRRRRR R(((s1lib/python2.7/site-packages/numba/testing/main.pytsJ          .     N /