\ K]c@@sddlmZmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl mZddl mZmZmZddlmZmZmZmZddlmZdd lmZdd lmZe jdFkrddl Z nydd l!m"Z"e#Z$Wne%k rOe&Z$nXydd l!m'Z'e#Z(Wne%k re&Z(nXyddl)Z*e#Z+Wne%k re&Z+nXej,e(dZ-ej,e$dZ.e(oe'j/dkZ0ej,e0dZ1ej,e jdGkdZ2ej3j4dZ5ej3j4dZ6ej3j4do`ej7d dHkZ8ej9dIkZ:e:p{e8Z;e5 Z<dZ=e+rdZ>n dZ>dZ?dZ@deAfdYZBd eBfd!YZCd"eBfd#YZDd$eBfd%YZEd&eBfd'YZFd(ZGd)ZHejIZJd*eAfd+YZKd,ZLeHeJejMZNeHeLd-ZOe5seHeLd.ZPeHeLd/ZQneHeLd0ZRd1efd2YZSe jTdJkZUej,eUd6d7eSfd8YZVeVjWd9eSfd:YZXeXjWd;efd<YZYed=eYfd>YZZeZjWee2d?eYfd@YZ[ee1dAeYfdBYZ\edCefdDYZ]e^dEkrej_ndS(Ki(tprint_functiontabsolute_importN(tconfigtutils(tunittest_support(tjitt vectorizet guvectorizei(ttemp_directorytoverride_configtTestCasettag(tskip_unsupported(t linux_only(tqueuei(ttbbpool(tomppoolsOpenMP threadpool requiredsTBB threadpool requiredtGNUsGNU OpenMP only testssTest runs on Python 3 onlytwintdarwintwin32iii cC@stj||S(N(tnptones(tntv((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pytfooIscC@sEtjtj||ftj||f}|tj||S(N(RtdotRtarange(RRtx((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pytlinalgNs0cC@stj||S(N(RR(RR((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyRRscC@s||S(N((tatb((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyt ufunc_fooWscC@s|||d()i ( RR$R"RR3R4R5R R+R,(R%tsigR-RRR.R/((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyR0s  "(R(R)R0(((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyR6scK@s|jd}y]tjdkr.tjnx:ttt|dD]}tj |}|qKWWn t k r}|j |nXdS(NRiig?(ii( tgetRt PYVERSIONt faulthandlertenabletrangetinttlenR3tchoicet BaseExceptiontput(tfnlisttkwargstqt_tfnte((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pytchoosers #c@sfd}|S(Nc @s}i|d6}gtdD]$}dtd|fd|^q#}x|D]}|jqTWx|D]}|jqoW|jsg}x&|js|j|jtqWd}t|djg|D]}t |^qndS(NRittargettargsRCs)Error(s) occurred in delegated runner: %ss ( R<RHtstarttjointemptytappendR8tFalset RuntimeErrortrepr( RBRDtkwstitthstthterrorst_msgR(tparallel_classt queue_impl(s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyt run_compiles  4   ((RXRYRZ((RXRYs@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pytcompile_factoryst_proc_class_implcB@seZdZdZRS(cC@s ||_dS(N(t_method(R%tmethod((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyR'scO@sEtjdkrtj||Stj|j}|j||SdS(Nii(ii(RR9tmultiprocessingtProcesst get_contextR](R%RJRCtctx((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyR0s(R(R)R'R0(((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyR\s cC@sjtjdkr't|}tj}n9|dkr<d}ntj|}t|}|j}||fS(Niitdefault(ii(RR9R\R_tQueuetNoneRa(R^tprocRRb((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyt_get_mp_classess      tspawntforkt forkserverRctTestParallelBackendBasecB@s eZdZedeededeededeedeededeedeededdededddeedeededdededddeg Ze s<ededeedededeededeedededegZ ej e nddgZ e jdkre jd ere jd e jd qn e jd iedede gd6ededdgd6ededdgd6ed6ZddhZddZRS(s6 Base class for testing the parallel backends tnopythontcachetnogilRItparallelt threadingR3iitmultiprocessing_spawntmultiprocessing_forktmultiprocessing_forkservertmultiprocessing_defaulttconcurrent_jittconcurrect_vectorizetconcurrent_guvectorizetconcurrent_mix_usetompttbbcC@sCt|jj|_td|j|dkrAt|n|dkrZt|n|dkrst|n|dkrt|n|dkrt |n|dkr)t j d krtt g}n/ttg}t r|j t|j tntj|x+|D]}||qWntd |WdQXdS( Nt CACHE_DIRRpRrRsRqRtR3iisUnknown parallelism supplied %s(ii(Rt __class__R(t _cache_dirR t thread_impltfork_proc_impltforkserver_proc_impltspawn_proc_impltdefault_proc_implRR9t _HAVE_OS_FORKRNR3tshufflet ValueError(R%RBt parallelismtpstimpl((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyRZ s0               (ii(R(R)t__doc__R*tTrueR1R2R6t all_implst_parfors_unsupportedt parfor_implstextendRRR9RNRtrunnerst safe_backendsRZ(((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyRks@          RyRzt workqueuesThreading layer not explicittTestParallelBackendcB@seZdZedZRS(sv These are like the numba.tests.test_threadsafety tests but designed instead to torture the parallel backend. If a suitable backend is supplied via NUMBA_THREADING_LAYER these tests can be run directly. This test class cannot be run using the multiprocessing option to the test runner (i.e. `./runtests -m`) as daemon processes cannot have children. cC@s{xt|jD]i}x`|jjD]O\}}d|d|}d}|||}||_t|||q Wq WdS(Nttest_REc@sfd}|S(Nc@sBtj}|jr+d}|j|n|jddS(Ns)daemonized processes cannot have childrenR(R_tcurrent_processtdaemontskipTestRZ(R%tselfprocRW(Rtp(s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyt test_method@s   ((RRR((RRs@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pytmethgen?s(RRtitemsR(tsetattr(tclsRtnameRtmethnameRRF((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pytgenerate9s  (R(R)Rt classmethodR(((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pyR,stTestSpecificBackendcB@skeZdZeZied6ed6ejedd6Z dZ dZ e dZ e dZRS( s This is quite contrived, for each test in the TestParallelBackend tests it generates a test that will run the TestParallelBackend test in a new python process with an environment modified to ensure a specific threadsafe backend is used. This is with view of testing the backends independently and in an isolated manner such that if they hang/crash/have issues, it doesn't kill the test suite. RzRytRcC@stj|dtjdtjd|}tjd|j}zg|j|j\}}|jdkrt d|j|j fn|j |j fSWd|j Xd S( NtstdouttstderrtenvigN@is/process failed with code %s: stderr follows %s gr@(NN( t subprocesstPopentPIPERptTimertkillRKt communicatet returncodetAssertionErrortdecodetcancelRe(R%tcmdlineRtpopenttimeoutR!terr((s@lib/python2.7/site-packages/numba/tests/test_parallel_backend.pytrun_cmd`s      cC@sDtjj}t||ds       "                       R! P 6% >)