o 1&i @sdZddlZddlZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlZddlZddlZddlZddlmZddlZddlZddlZddlmZddlZddl m!Z!mZddl"m#Z#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)dd l*m+Z+m,Z,m-Z-m.Z.m/Z/dd l0m1Z1dd l2m3Z3ddl4Z4dd l5m6Z6dd l7m8Z8ddl5m9Z:ddl;mZ>m?Z?m@Z@mAZAddlBmCZCzddlDZDWn eEydZDYnwzddlFZFWn eEyYnweFGe,ZHdeH_Ie,ZJdeJ_Ke,ZLe,ZMdeM_Ne!OgdZPePdZQe jRdkZSeSZTe4UeTdZVe4We%jXdkdZYe4We%jXdkdZZe4We%jXdkdZ[e4Ue%jXdkoMe j\ddZ]e4Ue j\do^e^dkd Z_e4UeSd!Z`e8jad"kZbe4Uebd#Zcedeee jfgd$dZh e4Uehd%Zid&d'Zjd(d)Zkd*d+Zld,d-Zmd.Zne4UeDduenZoe4We)jpd/Zqd0Zre4Ue j\d erZsd1Zte4Ue j\d etZue^d2kZve4Udd3Zwe4Ue^d4kd5Zxede jfgd6dZye4Weyd7Zze4Ueyd8Z{e4Ue^d4kd9Z|e4Ue^d4kd:Z}e~Zejd;kZe4Ued<Zeoej^d=kZzddlZDdZWn eEy)d>ZYnwe4Wed?ZzddlZDdZWn eEyDd>ZYnwe4Wed@Ze jfgdAddBkZe4WedCZzddlZdZWneEefypd>ZYnwe4WedDZdEdFZGdGdHdHe4jZGdIdJdJeZejdKdLZejdMdNZdOdPZdQZe jdRkre jeeZne jedSee fZdTZdUdVZdWdXZdYdZZd[d\Zd]d^Zejd_d`ZdadbZdcddZejdedfZGdgdhdheZGdidjdjeZGdkdldleeZejdmdnZejdodpZdqdrZdsdtZedfdudvZddwdxZdydzZGd{d|d|eZd}d~ZejddZdddZdddZddZGddde.ZddZdS)z& Assorted utilities for use in tests. N) defaultdict)contextmanager)cached_property)testingtypes)errorstypingutilsconfigcpu) cffi_utils) compile_extraFlags DEFAULT_FLAGS CompilerBaseDefaultPassBuilder)IRLegalization) PreserveIR)rtsys) numpy_support) _nrt_python)overload_method typeof_implregister_modelunbox NativeValuemodels) OpaqueModelT)Z importantZ long_running always_testrzparfors not supported) zneeds Python 3.10 or laterzneeds Python 3.10r zneeds Python 3.12r winz'Not supported on Python 3.13 on Windowslinuxaarch64zNot supported on Linux aarch64zNot supported on 32 bitrzNot supported on numpy 2.0+Z_NUMBA_REDUCED_TESTINGzSkipped for reduced testingcCtjdkr t|S|S)N)r r PYVERSIONunittestexpectedFailurefnr3>C:\wamp64\www\opt\env\Lib\site-packages\numba/tests/support.pyexpected_failure_py311  r5cCr+)Nr"r-r1r3r3r4expected_failure_py312r6r7cCr+)Nr$r-r1r3r3r4expected_failure_py313r6r8cCr+)Nr))r numpy_versionr/r0r1r3r3r4expected_failure_np2r6r:zSciPy needed for testz requires cffizlinux only testzWindows-only testarmv7lz Test disabledppc64lez?Hits: 'Parameter area must exist to pass an argument in memory'ZNUMBA_USE_TYPEGUARDzTypeguard is not enabledzBroken if Typeguard is enabledz>Hits: 'mismatch in size of parameter area' in LowerCall_64SVR4zInvalid PPC CTR loopDarwinz-fenv.h-like functionality unreliable on macOSarm64FzLAPACK needs SciPy 1.0+zBLAS needs SciPy 1.0+ SUBPROC_TEST1zneeds subprocess harnesszTest needs setuptoolscCs&tjdddtjddtjdddS)zUse in testing within a ` warnings.catch_warnings` block to filter out warnings that are unrelated/internally generated by Numba. ignoreZ typeguard)modulez.*TBB_INTERFACE_VERSION.*znumba\.np\.ufunc\.parallel.*)actionmessagecategoryrBN)warningsfilterwarningsrZ NumbaWarningr3r3r3r4ignore_internal_warningss  rHc@seZdZdZeddZddZejddZ ejdd Z d d Z ejd d Z e ejfZeefefejfeejfgZefefejgZeefgZeejfZeej fZ!ddZ"ddZ#ddZ$ddZ%   d(ddZ&   d(ddZ'   d)d d!Z(d*d"d#Z)d$d%Z*d&d'Z+dS)+TestCaseTcCs tjdS)N*)nprandomZ RandomStateselfr3r3r4rL s zTestCase.randomcCs4t|tr tj|}z|`WdStyYdSw)z Reset the warnings registry of a module. This can be necessary as the warnings module is buggy in that regard. See http://bugs.python.org/issue4180 N) isinstancestrsysmodules__warningregistry__AttributeError)rNrBr3r3r4reset_module_warningss    zTestCase.reset_module_warningsccsHtjtjttf}|| }|VWddS1swYdS)zr A context manager that asserts the enclosed code block fails compiling in nopython mode. N)rZ LoweringErrorZ TypingError TypeErrorNotImplementedError assertRaises)rNZ_accepted_errorscmr3r3r4assertTypingErrors "zTestCase.assertTypingErrorcgsddd|D}dVtdd|D}t|||D]\}}}||kr/|d|||fqdS)z A context manager that asserts the given objects have the same reference counts before and after executing the enclosed block. cSg|]}t|qSr3rQ getrefcount.0xr3r3r4 0z+TestCase.assertRefCount..NcSr[r3r\r^r3r3r4ra3rbz-Refcount changed from %d to %d for object: %r)gccollectzipfail)rNobjectsZ old_refcountsZ new_refcountsoldnewobjr3r3r4assertRefCount)szTestCase.assertRefCountc Gsjtdd|D}|d}tt|ddD]}||}||kr2|d|d|d|dqdS) NcSr[r3r\r^r3r3r4ra;rbz0TestCase.assertRefCountEqual..rz(Refcount for objects does not match. #0(z) != #(z) does not match.)rcrdrangelenrf)rNrgrcZrc_0iZrc_ir3r3r4assertRefCountEqual9s zTestCase.assertRefCountEqualccsht}dVt}|j|j}|j|j}|j|j}|j|j}|||d|||ddS)z| A context manager that asserts no NRT leak was created during the execution of the enclosed block. Nz-number of data allocs != number of data freesz3number of meminfo allocs != number of meminfo frees)rget_allocation_statsallocfreemi_allocmi_free assertEqualrNrhriZ total_allocZ total_freeZtotal_mi_allocZ total_mi_freer3r3r4assertNoNRTLeakCs    zTestCase.assertNoNRTLeakcCst|tjrdSt|tjrdS|jD] }t||rdSq|jD] }t||r+dSq!|jD] }t||r9dSq/|jD] }t||rGdSq=dS)z This function returns a string description of the type family that the object in question belongs to. Possible return values are: "exact", "complex", "approximate", "sequence", and "unknown" ndarrayenumsequenceexactcomplex approximateunknown) rOrKr{r|Enum_sequence_typesets_exact_typesets_complex_types_approx_typesets)rNZnumeric_objecttpr3r3r4_detect_family^s*          zTestCase._detect_familycCs0tjdkrtjdkr|tdkrtdS|S)z7 Fix the given *dtype* for comparison. win32rZint32Zint64)rQplatformmaxsizerKdtype)rNrr3r3r4 _fix_dtype|s zTestCase._fix_dtypecs2jdkr dgjSfddtjjDS)z Return the strides of the given array, fixed for comparison. Strides for 0- or 1-sized dimensions are ignored. rcs"g|] \}}|dkr|jqS)rl)itemsize)r_Zstrideshapearrr3r4ras z)TestCase._fix_strides..)sizendimrestridesr)rNrr3rr4 _fix_stridess    zTestCase._fix_stridescCsB||j|jd||j|jd|||||ddS)zG Test that two arrays have the same shape and strides. z shapes differzitemsizes differzstrides differN)rxrrr)rNfirstsecondr3r3r4assertStridesEquals zTestCase.assertStridesEqualr~rlNFc Csbz||||||||WdSty$}z t|} WYd}~nd}~ww|d||| fdS)ag Versatile equality testing function with more built-in checks than standard assertEqual(). For arrays, test that layout, dtype, shape are identical, and recursively call assertPreciseEqual() on the contents. For other sequences, recursively call assertPreciseEqual() on the contents. For scalars, test that two scalars or have similar types and are equal up to a computed precision. If the scalars are instances of exact types or if *prec* is 'exact', they are compared exactly. If the scalars are instances of inexact types (float, complex) and *prec* is not 'exact', then the number of significant bits is computed according to the value of *prec*: 53 bits if *prec* is 'double', 24 bits if *prec* is single. This number of bits can be lowered by raising the *ulps* value. ignore_sign_on_zero can be set to True if zeros are to be considered equal regardless of their sign bit. abs_tol if this is set to a float value its value is used in the following. If, however, this is set to the string "eps" then machine precision of the type(first) is used in the following instead. This kwarg is used to check if the absolute difference in value between first and second is less than the value set, if so the numbers being compared are considered equal. (This is to handle small numbers typically of magnitude less than machine precision). Any value of *prec* other than 'exact', 'single' or 'double' will raise an error. Nzwhen comparing %s and %s: %s)_assertPreciseEqualAssertionErrorrPrf) rNrrpreculpsmsgignore_sign_on_zeroabs_tolexcZ failure_msgr3r3r4assertPreciseEquals$zTestCase.assertPreciseEqualc sd fdd }|} |} d| | f} r"| df7} j| | | d| } | dkr|j} | |j|j|jd|j|jd |jj|jjd ||d |j| krt| | }|j| kr~| | }t |j |j D]\}} |||||qdS| d krjt |t |dt ||D]\}} |||||qdS| d krd}n5| dvrd}n.| dkr|j|j |j|j|||dS| dkr|j|jd}nJdt|drt|dr|j|jt|jt|jkr2d|j|jf} r-| df7} | zt|rBt|rBWdSWn tyMYnw|dur{|dkratt|j}nt|trj|}ntd|t|||kr{dS|p|d k}|s|d kr|dkrd}n|dkrd}ntd|fd||d}|t|t|}nd}t|jr||j |j |||j!|j!|dSt|tj"tj#frt$|rt$|t$|dS||||dS||||dS)!z-Recursive workhorse for assertPreciseEqual().Ncs|dus||krdksnt|st|rOj||dsMztd|td|kr@d||fWdSWdStyLYdSwdSj|||ddS)Ngrrlz%s != %s)deltar)mathisinfrxcopysignrfZ_formatMessagerVZassertAlmostEqual)rrrrrrNr3r4_assertNumberEquals,   z8TestCase._assertPreciseEqual.._assertNumberEqualz Type Family mismatch. (%s != %s)z: %srr{zdifferent number of dimensionszdifferent shapeszdifferent mutabilityzdifferent stridesr}r~T)rrFr|rrzunexpected familyrz$Mismatching return types (%s vs. %s)epsz)abs_tol is not "eps" or a float, found %ssingledouble5zunsupported precision %rr*rlN)%rrxrrrrflagsZ writeablerZastypereZflatrroZassertIs __class__valuehasattrrO _bool_typesrfcmathisnanrVrKZfinfotyperfloat ValueErrorabsrrealimagZ timedelta64Z datetime64Zisnat)rNrrrrrrrrZ first_familyZ second_familyZassertion_messageZcompare_familyrabZexact_comparisonZrtolbitskrr3rr4rs                      zTestCase._assertPreciseEqual<r@cCs@|j}t|j} |||f} ddd| D} g} |dur2|D]\} }| | | |q!tjg| dd| }tj }||d<z tj d|d <Wn t yYYnwt |durbin|}||tj|tjtj||d d }d |jd |j}||jd|d}||jvr|||S|d|j|S)a Runs named unit test(s) as specified in the arguments as: test_module.test_class.test_name. test_module must always be supplied and if no further refinement is made with test_class and test_name then all tests in the module will be run. The tests will be run in a subprocess with environment variables specified in `envvars`. If given, envvars must be a map of form: environment variable name (str) -> value (str) If given, flags must be a map of form: flag including the `-` (str) -> value (str) It is most convenient to use this method in conjunction with @needs_subprocess as the decorator will cause the decorated test to be skipped unless the `SUBPROC_TEST` environment variable is set to the same value of ``_subproc_test_env`` (this special environment variable is set by this method such that the specified test(s) will not be skipped in the subprocess). Following execution in the subprocess this method will check the test(s) executed without error. The timeout kwarg can be used to allow more time for longer running tests, it defaults to 60 seconds. .css|] }|dur|VqdSrr3r^r3r3r4 ysz2TestCase.subprocess_test_runner..Nz-mznumba.runtestsr?ZCOVERAGE_RCFILEZCOVERAGE_PROCESS_STARTT)stdoutstderrtimeoutenvuniversal_newlinesz captured stdout: z captured stderr: rz NO TESTS RANOK) __module__r__name__joinitemsappendrQ executableosenvironcopyKeyErrorpytypesMappingProxyTypeupdate subprocessrunPIPErrrx returncodeskipTestZassertIn)rN test_module test_class test_nameenvvarsrr_subproc_test_envZthemodZtheclspartsZfully_qualified_testZ flags_argsflagrcmdZenv_copystatusstreamsZ no_tests_ranr3r3r4subprocess_test_runner]sF          zTestCase.subprocess_test_runnercs&fdd}t|tjr||S|S)zRuns the decorated test in a subprocess via invoking numba's test runner. kwargs timeout and envvars are passed through to subprocess_test_runner.csfdd}|S)NcsHtjddjkr|jj}|j|j|jjddS|dS)Nr?)rrrrrr)rrgetrrrr)rNargskwargs class_name)rfuncrr3r4inners z?TestCase.run_test_in_subprocess..wrapper..innerr3)rrrr)rr4wrappersz0TestCase.run_test_in_subprocess..wrapper)rOr FunctionType)Z maybefuncrrrr3rr4run_test_in_subprocesss zTestCase.run_test_in_subprocesscsp|}td|tjfi}|dt|tGdddt}t |fdd}t |dd}||fS) zUse to generate a dummy type unique to this test. Returns a python Dummy class and a corresponding Numba type DummyType.zDummyTypeFor{}Zmy_dummyc@s eZdZdS)z'TestCase.make_dummy_type..DummyN)rr __qualname__r3r3r3r4DummysrcsSrr3)valcZ dummy_typer3r4 typeof_dummysz.TestCase.make_dummy_type..typeof_dummycSst|jSr)rcontextZget_dummy_value)typrjrr3r3r4 unbox_dummysz-TestCase.make_dummy_type..unbox_dummy) idrformatrZOpaquerrobjectrregisterr)rNZtest_idZ DummyTyperrrr3rr4make_dummy_types  zTestCase.make_dummy_typecCs$ddlm}|s|ddSdS)a Call this to ensure the test is skipped if no suitable external compiler is found. This is a method on the TestCase opposed to a stand-alone decorator so as to make it "lazy" via runtime evaluation opposed to running at test-discovery time. r)external_compiler_worksz(No suitable external compiler was found.N)Znumba.pycc.platformrr)rNrr3r3r4skip_if_no_external_compilers z%TestCase.skip_if_no_external_compiler)r~rlNFN)NNNrNr@)NrN),rrrZ longMessagerrLrU contextlibrrZrkrrrzboolrKZbool_rintrPintegerbytesbytes_rrrZinexactrtuplelistrZfloatingZ _float_typesZcomplexfloatingrrrrrrrrrrrr3r3r3r4rIsP            0  : rIc@seZdZdZdZdS) SerialMixinz-Mixin to mark test for serial execution. FN)rrr__doc__Z_numba_parallel_test_r3r3r3r4rsrc cs@tt|}tt||z dVWtt||dStt||w)z Return a context manager that temporarily sets Numba config variable *name* to *value*. *name* must be the name of an existing variable in numba.config. N)getattrr setattr)namer old_valuer3r3r4override_configs   rc csxtj|}|tj|<tzdVW|durtj|=n|tj|<tdS|dur2tj|=n|tj|<tw)zk Return a context manager that temporarily sets an Numba config environment *name* to *value*. N)rrrr Z reload_config)r rrhr3r3r4override_env_configs        rcCs(t|dd}i}t|||||S)zj Given a *code* string, compile it with globals *globs* and return the function named *name*. zr)compilerstripeval)r codeZglobsconsr3r3r4compile_functions rz numba-testsntz%s.%siQcCs&zttWdStyYdSwr)rmkdir _trashcan_dirFileExistsErrorr3r3r3r4_create_trashcan_dir"s  rc Cs|tt}tttD].}tjt|}zt|}|j |kr(t j |ddWq t y;}zWYd}~q d}~wwdS)NT) ignore_errors) time_trashcan_timeoutsortedrlistdirrpathrstatst_mtimeshutilrmtreeOSError)Zfreshness_thresholdr2ster3r3r4_purge_trashcan_dir(s    r)cCsttj|dtd}|S)N-)prefixdir)r)tempfilemkdtempr)r+r!r3r3r4_create_trashcan_subdir5sr/cCstt|S)a Create a temporary directory with the given *prefix* that will survive at least as long as this process invocation. The temporary directory will be eventually deleted when it becomes stale enough. This is necessary because a DLL file can't be deleted while in use under Windows. An interesting side-effect is to be able to inspect the test files shortly after a test suite run. )rr/)r+r3r3r4temp_directory:s r0cCs"ddl}|t|tj|S)z Import and return a module of the given name. Care is taken to avoid issues due to Python's internal directory caching. rN) importlibinvalidate_caches __import__rQrR)modnamer1r3r3r4import_dynamicJs r5c csJtt|}tt|tztt|VWtt||dStt||w)zReturn a context manager used by captured_stdout/stdin/stderr that temporarily replaces the sys stream *stream_name* with a StringIO.N)r rQr ioStringIO) stream_name orig_stdoutr3r3r4captured_outputWs  r:cCtdS)zCapture the output of sys.stdout: with captured_stdout() as stdout: print("hello") self.assertEqual(stdout.getvalue(), "hello ") rr:r3r3r3r4captured_stdoutbr=cCr;)zCapture the output of sys.stderr: with captured_stderr() as stderr: print("hello", file=sys.stderr) self.assertEqual(stderr.getvalue(), "hello ") rr<r3r3r3r4captured_stderrkr>r?c csnt)}tdd |VWdn1swYWddSWddS1s0wYdS)NZ DEBUG_CACHET)r=r)outr3r3r4capture_cache_logus "rAc@s eZdZdZddZddZdS)EnableNRTStatsMixinz,Mixin to enable the NRT statistics counters.cC tdSr)_nrtZmemsys_enable_statsrMr3r3r4setUp zEnableNRTStatsMixin.setUpcCrCr)rDZmemsys_disable_statsrMr3r3r4tearDownrFzEnableNRTStatsMixin.tearDownN)rrrr rErGr3r3r3r4rB|s rBc@s0eZdZdZddZddZddZdd Zd S) MemoryLeakTcCstt|_dSr)rcrdrrs_MemoryLeak__init_statsrMr3r3r4memory_leak_setupszMemoryLeak.memory_leak_setupcCs|jr |dSdSr)_MemoryLeak__enable_leak_checkassert_no_memory_leakrMr3r3r4memory_leak_teardowns zMemoryLeak.memory_leak_teardowncCsZ|j}t}|j|j}|j|j}|j|j}|j|j}||||||dSr)rIrrsrtrurvrwrxryr3r3r4rLs     z MemoryLeak.assert_no_memory_leakcCs d|_dS)NF)rKrMr3r3r4disable_leak_checks zMemoryLeak.disable_leak_checkN)rrrrKrJrMrLrNr3r3r3r4rHs  rHcs(eZdZfddZfddZZS)MemoryLeakMixincstt||dSr)superrOrErJrMrr3r4rEs zMemoryLeakMixin.setUpcs"t|tt|dSr)rcrdrMrPrOrGrMrQr3r4rGszMemoryLeakMixin.tearDown)rrrrErG __classcell__r3r3rQr4rOs rOc csddlm}dg}i}dd}zP|D]6}|d}|}|ddD]}t||}q"|d}t||}t|s=Jd |||||f<t|||qdVW|D] \\}}}t|||qRdS|D] \\}}}t|||qew) z Forbid LLVM code generation during the execution of the context manager's enclosed block. If code generation is invoked, a RuntimeError is raised. r)codegenz%CPUCodeLibrary._finalize_final_modulec_std)Nzcodegen forbidden by test case) RuntimeError)rrr3r3r4rfszforbid_codegen..failrNz%r should be callable) numba.corerSsplitr callabler r) rSZ patchpointsrhrfr rrjattrnamerr3r3r4forbid_codegens,     rZc csddlm}t|j}|t|}t\}}z$t ||t |dVW|t |t ||t |dS|t |t ||t |w)zt Temporarily redirect *fd* to a pipe's write end and return a file object wrapping the pipe's read end. r) _helperlibrN) numbar[ctypesCDLL__file__Z_numba_flush_stdoutrduppipedup2r6openclose)fdr[Zlibnumbasaver\wr3r3r4 redirect_fds"          ricCtj}t|S)zRedirect C stdout )rQ __stdout__filenorirfr3r3r4redirect_c_stdout rncCrj)zRedirect C stderr )rQ __stderr__rlrirmr3r3r4redirect_c_stderrrorqcCst|}t|||dS)aKSpawn a new process to run `func` with a temporary cache directory. The childprocess's stdout and stderr will be captured and redirected to the current process's stdout and stderr. Returns ------- ret : dict exitcode: 0 for success. 1 for exception-raised. stdout: str stderr: str )verbose)r0run_in_new_process_in_cache_dir)rZcache_dir_prefixrr cache_dirr3r3r4run_in_new_process_cachings rucCstd}|}td|T|jt||gd}|||}|}|r=| r=t t d ddt ||r[| r[t t j dt d ddt j dt |t j dWd n1sewY|j||d S) aSpawn a new process to run `func` with a temporary cache directory. The childprocess's stdout and stderr will be captured and redirected to the current process's stdout and stderr. Similar to ``run_in_new_process_caching()`` but the ``cache_dir`` is a directory path instead of a name prefix for the directory path. Returns ------- ret : dict exitcode: 0 for success. 1 for exception-raised. stdout: str stderr: str ZspawnZNUMBA_CACHE_DIR)targetrSTDOUTPr*)fileZSTDERRN)exitcoderr)mpZ get_contextQueuerProcess_remote_runnerstartr get_nowaitstripprintcenterrQrrz)rrtrrctxqoutprocrrr3r3r4rss,     rsc Cst9}t}z|Wntytd}Ynwd}Wdn1s)wY||Wdn1s?wY||t|dS)z+Used by `run_in_new_process_caching()` rlrN) r?r= Exception traceback print_excputgetvaluerQexit)r2rrrrzr3r3r4r~.s   r~c@seZdZejefddZdS)CheckWarningsMixinccstjdd}tddVWdn1swYd}|D]}|D]}|t|jvr;||j||d7}q'q#||t|dS)NT)recordalwaysrrl)rFcatch_warnings simplefilterrPrDrxrEro)rNmessagesrEZcatchfoundrhmr3r3r4check_warnings?s z!CheckWarningsMixin.check_warningsN)rrrrrRuntimeWarningrr3r3r3r4r>srcKsN|sdSg}|D]\}}t|trd|}|d||q d|S)Nz"{}"z{}={}z, )rrOrPrrr) jit_optionsr@keyrr3r3r4_format_jit_optionsMs   rc kszetd}dttdd}tj||d}t d i|}t |d}|j|d}| |Wdn1s>wYt j d |t|}|VWt j|dt j|t|dSt j|dt j|t|w) aA context manager that creates and imports a temporary module from sources provided in ``source_lines``. Optionally it is possible to provide jit options for ``jit_module`` if it is explicitly used in ``source_lines`` like ``jit_module({jit_options})``. Ztest_temp_moduleztest_temp_module_{}r*_z.pyrh)rNrr3)r0rrPuuiduuid4replacerr!rrrdwriterQinsertr1 import_modulerRpopremover$r%)Z source_linesrtempdirZtemp_module_nameZtemp_module_pathflinesrr3r3r4create_temp_moduleXs*        rc Csr|durg}tjg|d|g}tj|tjtj|d}|j|d\}}|jdkr5d}t||j|f||fS)zRun a snippet of Python code in a subprocess with flags, if any are given. 'env' is passed to subprocess.Popen(). 'timeout' is passed to popen.communicate(). Returns the stdout and stderr of the subprocess after its termination. N-c)rrr)rrz/process failed with code %s: stderr follows %s ) rQrrPopenr communicaterrdecode) rrrrrpopenr@errrr3r3r4run_in_subprocessvs  rr!c s6td}t}td}|durtd|ddt|dd|d |j g}t j |t j d j }t|j}|dfd d } z;tjjd} | ||duret|tjnd} | | | |} W| |n | |wWd| S1swY| S)a)Runs strace whilst executing the function work() in the current process, captures the listed syscalls (list of strings). Takes an optional timeout in seconds, default is 10, if this is exceeded the process will be sent a SIGKILL. Returns a list of lines that are output by strace. zw+tstraceNz'No valid 'strace' binary could be foundz-qz-pz-e,z-o)rrcs*j}|dkrd|d|}t|dS)Nrz0strace exited non-zero, process return code was:z. )rrT)problemretrrr3r4 check_returnszstrace..check_return)rvz)If this is SIGKILL, increase the timeout?)r)r-NamedTemporaryFilergetpidr$whichrrPrr rrrpid threadingTimerkillrThreadrflushpollsignalSIGINTwait readlinescancel) ZworkZsyscallsrZntfZ parent_pidZ strace_binaryrZ strace_pidZthread_timeoutrZ thread_commsrZ strace_datar3rr4rsH        66rcCsPtjdsdSdd}d}zt||g}Wn ty YdSw|d|vS)z)Checks if strace is supported and workingr'FcSsttjddgdS)Nrzexit())rrrQrr3r3r3r4 force_clonesz%strace_supported..force_clonecloner)rQr startswithrrr)rZsyscalltracer3r3r4strace_supporteds  rc@seZdZdZddZdS)IRPreservingTestPipelinez Same as the standard pipeline, but preserves the func_ir into the metadata store after legalisation, useful for testing IR changescCs.t|jd}d|_|tt||gS)NZir_preserving_custom_pipeF)rZdefine_nopython_pipelinestateZ _finalizedZadd_pass_afterrrfinalize)rNZpipeliner3r3r4define_pipeliness z)IRPreservingTestPipeline.define_pipelinesN)rrrr rr3r3r3r4rs rc Csddl}ddlm}tjtjt}tj|ddd}tj|s't dt tj|d }| }Wdn1s?wY|j ||d}|d}t d d }|dd D]} | d d } | D]} || d| dd7<qgq[gd} tjj|g| R} tj| st dt tj| d }| }Wdn1swY|j ||d}|dddd } | D]} || d| dd7<qtdtdt|D]\}}|D]\}}td|d|dd|qqt dd }t|D]\}}|D] \}}||||<qq tdtddd }t||dD]\}}|D]\}}td|dd|d|q<q4dS)a-This is a utility function that prints out the map of NumPy to Python versions and how many of that combination are being tested across all the declared config for azure-pipelines. It is useful to run when updating the azure-pipelines config to be able to quickly see what the coverage is.rN)Loader..zazure-pipelines.ymlz&'azure-pipelines.yml' is not availablertjobscSttSrrrr3r3r3r4z$print_azure_matrix..r* parametersmatrixZNUMPYZPYTHONrl)rrZ buildscriptsZazurezazure-windows.ymlz$'azure-windows.yml' is not availableZstrategyzNumPy | Python | Countz----------------------- z | z<4z | cSrrrr3r3r3r4rrz Python | NumPy | CountcSst|dddS)Nrrrl)rrW)r`r3r3r4r"rb)r)yamlrrr!dirnameabspathr`risfilerTrdreadloadrvaluesrrr)rr base_pathZ azure_piperdataZpipe_ymlZ templatesZ py2np_mapZtmpltrsetupZwinpathZ azure_windowsZ windows_ymlZnpverZpyspyvercountZrev_mapZsorterZnpsr3r3r4print_azure_matrixs^           r)T)NNr)r!)r rr collectionsrr|rcrrrrr$rrQr-rrr6r^multiprocessingr{rFrrrr1rr functoolsrnumpyrKr]rrVrrr r r Znumba.core.typingr Znumba.core.compilerr rrrrZnumba.core.typed_passesrZnumba.core.untyped_passesrr/Znumba.core.runtimerZnumba.nprrrDZnumba.core.extendingrrrrrrZnumba.core.datamodel.modelsrZscipy ImportErrorZcoverageZprocess_startupZenable_pyobj_flagsZenable_pyobjectZforce_pyobj_flagsZforce_pyobjectZno_pyobj_flagsZ nrt_flagsZnrtZmake_tag_decoratortagrrZ_32bitZis_parfors_unsupportedZskipIfZskip_parfors_unsupportedZ skipUnlessr.Zskip_unless_py10_or_laterZskip_unless_py10Zskip_unless_py312rZskip_if_py313_on_windowsmachineZskip_if_linux_aarch64Z skip_if_32bitr9Z IS_NUMPY_2Zskip_if_numpy_2rrrrZREDUCED_TESTINGZskip_if_reduced_testingr5r7r8r:Z_msgZskip_unless_scipyZ SUPPORTEDZskip_unless_cffiZ _lnx_reasonZ linux_onlyZ _win_reasonZ windows_onlyZ _is_armv7lZ disabled_testZskip_ppc64le_issue4563Z has_typeguardZskip_unless_typeguardZskip_if_typeguardZskip_ppc64le_issue6465Zskip_ppc64le_invalid_ctr_loopunameZ_unamesystemZIS_MACOSZskip_macos_fenv_errorsZIS_MACOS_ARM64Zscipy.linalg.cython_lapackZ has_lapackZ needs_lapackZscipy.linalg.cython_blasZhas_blasZ needs_blasZ _exec_condZneeds_subprocess setuptoolsZhas_setuptoolsr&Zneeds_setuptoolsrHrIrrrrrrr r!r gettempdirgetuidrrr)r/r0r5r:r=r?rArBrHrOrZrirnrqrrursr~rrrrrrrrr3r3r3r4s                          V           "  '   A