2@jdZddlZddlZddlmZddlZddlZddlZddlZddl 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@mAZAddlBmCZC ddlDZDn #eE$rdZDYnwxYw ddlFZFeFjGn #eE$rYnwxYwe,ZHdeH_Ie,ZJdeJ_Ke,ZLe,ZMdeM_Ne!jOgdZPePdZQe jRdkZSeSZTe4jUeTdZVe4jWe%jXdkdZYe4jWe%jXdkdZZe4jWe%jXdkdZ[e4jUe%jXdkoe j\ddZ]e4jWe%jXdkdZ^e4jUe j\d oej_d!kd"Z`e4jUeSd#Zae8jbd$kZce4jUecd%Zde j\d oej_d&ko e8jbd'kZeefege jhid(dZj e4jUejd)Zkele d*d+ Zme4jUemd,Znd-Zod.Zpd/Zqd0Zrd1Zsd2Ztd3Zud4Zve4jUeDduevZwe4jWe)jxd5Zyd6Zze4jUe j\d  ezZ{d7Z|e4jUe j\d e|Z}d8Z~e4jUe j\de~Zej_d9kZe4jUdd:Ze4jUej_d;kd<Zefe jhid=dZe4jWed>Ze4jUed?Ze4jUej_d;kd@Ze4jUej_d;kdAZejZejdBkZe4jUedCZeo ej_dDkZejdEko ej_dFkZe4jUedGZ ddlZDdZn #eE$rdHZYnwxYwe4jWedIZ ddlZDdZn #eE$rdHZYnwxYwe4jWedJZe jhidKddLkZe4jWedMZ ddlZdZn#eEef$rdHZYnwxYwe4jWedNZdOZGdPdQe4jZGdRdSeZejdTZejdUZdVZdWZe jdXkr)e jejeZn:e jejedYe jZdZZd[Zd\Zd]Zd^Zd_Zejd`ZdaZdbZejdcZGdddeeZGdfdgeZGdhdieeZejdjZejdkZdlZdmZedfdnZd}doZdpZGdqdreZdsZejdtZd~dvZddxZdyZGdzd{e.Zd|ZdS)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) important long_running always_testr!zparfors not supported) zneeds Python 3.10 or laterzneeds Python 3.10r# zneeds Python 3.12r# winz(Not supported on Python 3.13+ on Windowsr#zneeds Python 3.14+linuxaarch64zNot supported on Linux aarch64zNot supported on 32 bitrzNot supported on numpy 2.0+x86_64)_NUMBA_REDUCED_TESTINGzSkipped for reduced testing_is_gil_enabledcdS)NTr6DC:\Projects\Historical\venv\Lib\site-packages\numba/tests/support.pyr9sdr7zWSkipped [NOT APPLICABLE] if using a free-threading build and free-threading is enabled.cNtjdkrtj|S|S)N)r# r PYVERSIONunittestexpectedFailurefns r8expected_failure_py311rB% '!!'+++ r7cNtjdkrtj|S|S)Nr%r<r@s r8expected_failure_py312rErCr7cNtjdkrtj|S|S)Nr'r<r@s r8expected_failure_py313rGrCr7cNtjdkrtj|S|S)Nr*r<r@s r8expected_failure_py314rIrCr7cNtjdkrtj|S|S)N)r#r<r@s r8expected_failure_py315rLrCr7cNtjdkrtj|S|S)Nr.)r numpy_versionr>r?r@s r8expected_failure_np2rOs&"f,,'+++ r7c ddl}|jS#t$rYnwxYwtjdr td5}|D]O}|dr8t| ddzccdddSP dddn #1swxYwYn#t$rYnwxYwdS)z Best-effort lookup of currently available system memory, in bytes. Returns None if it can't be determined (psutil not installed, and not running on Linux where /proc/meminfo can be read directly). rNr,z /proc/meminfoz MemAvailable:r1i) psutilvirtual_memory available ImportErrorsysplatform startswithopenintsplitOSError)rQflines r8available_memory_bytesr^sa   $$&&00       |w'' o&& ;!;;D77;"4::<<?33d::: ; ; ; ; ; ; ; ;;; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;    D  4sO ,,C AB># C 0B>2 C >CC CC CCzSciPy needed for testz requires cffizlinux only testzWindows-only testznon Windows testarmv7lz Test disabledppc64lez?Hits: 'Parameter area must exist to pass an argument in memory'NUMBA_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 macOSarm64WindowsARM64zOAArch64 uimm12 fixup failure on win-arm64 for large UniTuple(unicode) argumentsFzLAPACK needs SciPy 1.0+zBLAS needs SciPy 1.0+ SUBPROC_TEST1zneeds subprocess harnesszTest needs setuptoolscvtjddtjddtjddS)zUse in testing within a ` warnings.catch_warnings` block to filter out warnings that are unrelated/internally generated by Numba. ignore typeguard)modulez.*TBB_INTERFACE_VERSION.*znumba\.np\.ufunc\.parallel.*)actionmessagecategoryrkN)warningsfilterwarningsr NumbaWarningr6r7r8ignore_internal_warningsrrAsR  H[9999 8$@%+%8#BDDDDDDr7cneZdZdZedZdZejdZ ejdZ dZ ejdZ e ejfZeefefejfeejfgZefefejgZeefgZeejfZeej fZ!dZ"d Z#d Z$d Z% ddZ& ddZ' ddZ(ddZ)dZ*dZ+dS)TestCaseTc@tjdS)N*)nprandom RandomStateselfs r8rxzTestCase.randomTsy$$R(((r7c|t|trtj|} |`dS#t $rYdSwxYw)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) isinstancestrrUmodules__warningregistry__AttributeError)r{rks r8reset_module_warningszTestCase.reset_module_warningsXsU fc " " )[(F ***    DD s - ;;c#Ktjtjttf}||5}|VddddS#1swxYwYdS)zr A context manager that asserts the enclosed code block fails compiling in nopython mode. N)r LoweringError TypingError TypeErrorNotImplementedError assertRaises)r{_accepted_errorscms r8assertTypingErrorzTestCase.assertTypingErrores #0&2D%':<   / 0 0 BHHH                  sAAAc'Ktjd|D}dVtjd|D}t|||D]'\}}}||kr|d|||fz(dS)z A context manager that asserts the given objects have the same reference counts before and after executing the enclosed block. c6g|]}tj|Sr6rU getrefcount.0xs r8 z+TestCase.assertRefCount..{"===++===r7Nc6g|]}tj|Sr6rrs r8rz+TestCase.assertRefCount..rr7z-Refcount changed from %d to %d for object: %r)gccollectzipfail)r{objects old_refcounts new_refcountsoldnewobjs r8assertRefCountzTestCase.assertRefCountps ==W===   ==W===   wGG - -MCcczz I #sO,--- - -r7c tjd|D}|d}tt|ddD]/}||}||kr|d|d|d|d0dS)Nc6g|]}tj|Sr6rrs r8rz0TestCase.assertRefCountEqual..s" 2 2 2Qcoa  2 2 2r7rr1z(Refcount for objects does not match. #0(z) != #(z) does not match.)rrrangelenr)r{rrcrc_0irc_is r8assertRefCountEqualzTestCase.assertRefCountEquals 2 2' 2 2 2!us7||$$QRR( I IAa5Dt|| H $HH,-HH04HHHIII I Ir7c#2Ktj}dVtj}|j|jz }|j|jz }|j|jz }|j|jz }|||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 assertEqualr{rr total_alloc total_freetotal_mi_alloc total_mi_frees r8assertNoNRTLeakzTestCase.assertNoNRTLeaks (** (**i#)+ X(  4 ck1  jH J J J N P P P P Pr7c^t|tjrdSt|tjrdS|jD]}t||rdS|jD]}t||rdS|jD]}t||rdS|jD]}t||rdSdS)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) r}rwrrEnum_sequence_typesets_exact_typesets_complex_types_approx_typesets)r{numeric_objecttps r8_detect_familyzTestCase._detect_familys nbj 1 1 9 ndi 0 0 6) " "B."-- "!zz "&  B."-- ww % ! !B."-- ! yy !' % %B."-- %$}} %yr7ctjdkr.s5"""'qyyS\) yyr7)sizendimrstridesr)r{rs `r8 _fix_strideszTestCase._fix_stridessW 8q==3> !""""+.s{CI+F+F""" "r7c||j|jd||j|jd|||||ddS)zG Test that two arrays have the same shape and strides. z shapes differzitemsizes differzstrides differN)rrrr)r{firstseconds r8assertStridesEqualzTestCase.assertStridesEquals flODDD :LMMM **51143D3DV3L3L) + + + + +r7rr1NFc  ||||||||dS#t$r}t|} Yd}~nd}~wwxYw|d|d|d| dS)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 z and : )_assertPreciseEqualAssertionErrorr~r) r{rrpreculpsmsgignore_sign_on_zeroabs_tolexc failure_msgs r8assertPreciseEqualzTestCase.assertPreciseEqualsH   $ $UFD$#W . . . F  # # #c((KKKKKK # EEE666;;OPPPPPs A=Ac  d"fd }|} |} d| d| d} r| dz } | | | | } | dkro|j} | |j|j|jd |j|jd |jj|jjd ||d |j| kr| | }|j| kr| | }t|j |j D] \}} |||||!dS| d kret|t|t||D] \}} |||||!dS| dkrd}n| dvrd}n| dkrG|j|j |j|j|||dS| dkr#|j|jd}n Jdt#|dr0t#|dr |j|jt%|jt%|jkr2d|jd|jd} r| dz } |  t+j|rt+j|rdSn#t.$rYnwxYw|o|dkr't1jt5|j}n*t%|t8r|}nt;d|zt=||z |krdS|p|dk}|sX|dkrR|dkrd}n|dkrd}nt;d|d ||z d!z z}|t=|t=|zz}nd}t%|jr0||j |j |||j!|j!|dSt%|t0j"t0j#fr_t1j$|r<t1j$|t1j$|dS||||dS||||dS)#z-Recursive workhorse for assertPreciseEqual().Nc|7||cxkrdks*ntj|stj|r||sp tjd|tjd|kr0|d|dSdS#t $rYdSwxYwdS|||dS)Ngrr1 != )deltar)mathisinfrcopysignr_formatMessagerassertAlmostEqual)rrrrrr{s r8_assertNumberEqualz8TestCase._assertPreciseEqual.._assertNumberEquals> &!7!7!7!7C!7!7!7!7:e$$"8(, 6(:(:"8  C 888*=E22dmAv6N6NNN II $ 3 3C5:UUFF5D!E!EFFFFFON %&&ufEs&KKKKKsAB44 CCzType Family mismatch. (r)rrrzdifferent number of dimensionszdifferent shapeszdifferent mutabilityzdifferent stridesrrT)rrFrrrzunexpected familyrzMismatching return types (z vs. epsz)abs_tol is not "eps" or a float, found %ssingledouble5zunsupported precision r/r1N)%rrrrrrflags writeablerastyperflatrrassertIs __class__valuehasattrr} _bool_typesrcmathisnanrrwfinfotyperfloat ValueErrorabsrrealimag timedelta64 datetime64isnat)r{rrrrrrrr first_family second_familyassertion_messagecompare_familyrabexact_comparisonrtolbitskrs` `` r8rzTestCase._assertPreciseEqualss  L L L L L L L L"**511 ++F33 BN,, MM  1  33 0  }:KLLL& Y & &OOEK00E   UDOOFL$A$A B B B   UZ= ? ? ?   U[&,/ 1 1 1   U[2FL4J3 5 5 5   T..u55!!&))+> @ @ @{e## U++|u$$u--EJ 44 G G1((AtT3) > >E6** G G1((AtT3)""T))+>tw DGOOx!! jtt!EFFFdTkAo&AUc&kk12EEE eT0 1 1 5  uz6; > > >  uz6; > > > > >  > ? ? 5x 9  %"(62B2BCCCCC""5&%88888  ufe 4 4 4 4 4s3(N N+*N+<rgc`|j}t|j} |||f} dd| D} g} |F|D]1\} }| | | |2t jg| dd| }tj }||d< tj d|d<n#t$rYnwxYwtj |in|}||tj|tjtj||d }d |jd |j}||jd |d}||jvr||n|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. .c3K|]}||V dSrr6rs r8 z2TestCase.subprocess_test_runner..s"'J'JaAMMMMM'J'Jr7Nz-mznumba.runtestsrfCOVERAGE_RCFILECOVERAGE_PROCESS_STARTT)stdoutstderrtimeoutenvuniversal_newlinesz captured stdout: z captured stderr: rz NO TESTS RANOK) __module__r__name__joinitemsappendrU executableosenvironcopyKeyErrorpytypesMappingProxyTypeupdate subprocessrunPIPEr!r"r returncodeskipTestassertIn)r{ test_module test_class test_nameenvvarsr#r_subproc_test_envthemodtheclspartsfully_qualified_test flags_argsflagrcmdenv_copystatusstreams no_tests_rans r8subprocess_test_runnerzTestCase.subprocess_test_runners2d$j)4"xx'J'J5'J'J'JJJ  ${{}} . . e!!T),,,!!U*----~% %D%2B%#%:??$$#4  13)r-r.getr(rrJr')r{argskwargs class_namer=funcr#s r8innerz?TestCase.run_test_in_subprocess..wrapper..inners~:>>.$774=HH"&!8J//$(O#-"&- ' '*.- 0DJJJJJr7r6)rRrSr=r#s` r8wrapperz0TestCase.run_test_in_subprocess..wrappers.       "Lr7)r}r1 FunctionType) maybefuncr#r=rTs `` r8run_test_in_subprocesszTestCase.run_test_in_subprocesssN      ( i!5 6 6 79%% %Nr7cv|}td|tjfi}|dt |t Gddt}tj |fd}t|d}||fS)zUse to generate a dummy type unique to this test. Returns a python Dummy class and a corresponding Numba type DummyType.zDummyTypeFor{}my_dummyceZdZdS)'TestCase.make_dummy_type..DummyN)r(r' __qualname__r6r7r8Dummyr[ s Dr7r]cSrr6)valc dummy_types r8 typeof_dummyz.TestCase.make_dummy_type..typeof_dummys  r7cNt|jSr)rcontextget_dummy_value)typrr`s r8 unbox_dummyz-TestCase.make_dummy_type..unbox_dummysqy88::;; ;r7) idrformatrOpaquerrobjectrregisterr)r{test_id DummyTyper]rbrgras @r8make_dummy_typezTestCase.make_dummy_types '')))0099EL?BOO Yz** !y!!+...     F     e $ $     % $  y   < <   <ir7cTddlm}|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)numba.pycc.platformrqr8)r{rqs r8skip_if_no_external_compilerz%TestCase.skip_if_no_external_compilersN @?????&&(( F MMD E E E E E F Fr7)rr1NFN)NNNrNrg)NrN),r(r'r\ longMessagerrxr contextlibrrrrrboolrwbool_rrYr~integerbytesbytes_rrrinexactrtuplelistrfloating _float_typescomplexfloatingrrrrrrrrJrWrorsr6r7r8rtrtNsK))_)   ---*IIIPPP$"K"SFSFRZMry)+O7*rz; $-2;'Lr12N<    " " "+++DE9>#'.Q.Q.Q.Q`EF:?$(N5N5N5N5`>BEG=@8888t:   0 F F F F Fr7rtceZdZdZdZdS) SerialMixinz-Mixin to mark test for serial execution. FN)r(r'r\__doc___numba_parallel_test_r6r7r8rr's !r7rc#Ktt|}tt|| dVtt||dS#tt||wxYw)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_values r8override_configr/sg%%I FD%   ) i(((((i((((s A A#c#hKtj|}|tj|<tj dV|tj|=n|tj|<tjdS#|tj|=n|tj|<tjwxYw)zk Return a context manager that temporarily sets an Numba config environment *name* to *value*. N)r-r.rNr reload_config)rrrs r8override_env_configr>s *..  CBJt    ; 4   #BJt  ; 4   #BJt s A==4B1c~t|dd}i}t|||||S)zj Given a *code* string, compile it with globals *globs* and return the function named *name*. zr)compilerstripeval)rcodeglobsconss r8compile_functionrUs=  H 5 5B BUB d8Or7z numba-testsntriQc\ tjtdS#t$rYdSwxYwr)r-mkdir _trashcan_dirFileExistsErrorr6r7r8_create_trashcan_dirros?        s  ++crtjtz }ttjt D]t}tjt |} tj|}|j |krtj |d^#t$r }Yd}~md}~wwxYwdS)NT) ignore_errors) time_trashcan_timeoutsortedr-listdirrpathr)statst_mtimeshutilrmtreer[)freshness_thresholdrAstes r8_purge_trashcan_dirrus)++(99RZ ..//   W\\- , , B{000 b5555    DDDD   s*5B  B4/B4c^ttj|dzt}|S)N-)prefixdir)rtempfilemkdtempr)rrs r8_create_trashcan_subdirrs,  6C<] C C CD Kr7c<tt|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)rs r8temp_directoryrs "6 * **r7cjddl}|jt|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__rUr)modnamers r8import_dynamicrs< I!!!w ;w r7c#Ktt|}tt|tj tt|Vtt||dS#tt||wxYw)zReturn a context manager used by captured_stdout/stdin/stderr that temporarily replaces the sys stream *stream_name* with a StringIO.N)rrUrioStringIO) stream_name orig_stdouts r8captured_outputrsz#{++K Cbkmm,,,/c;'''''[+.....[+....s A//Bc tdS)zCapture the output of sys.stdout: with captured_stdout() as stdout: print("hello") self.assertEqual(stdout.getvalue(), "hello ") r!rr6r7r8captured_stdoutr 8 $ $$r7c tdS)zCapture the output of sys.stderr: with captured_stderr() as stderr: print("hello", file=sys.stderr) self.assertEqual(stderr.getvalue(), "hello ") r"rr6r7r8captured_stderrrrr7c#Kt5}tdd5|Vdddn #1swxYwYddddS#1swxYwYdS)N DEBUG_CACHET)rr)outs r8capture_cache_logrs   c ]D 1 1  III               s,A 3 A 7 A 7 A  AAceZdZdZdZdZdS)EnableNRTStatsMixinz,Mixin to enable the NRT statistics counters.c,tjdSr)_nrtmemsys_enable_statsrzs r8setUpzEnableNRTStatsMixin.setUps  """""r7c,tjdSr)rmemsys_disable_statsrzs r8tearDownzEnableNRTStatsMixin.tearDowns !#####r7N)r(r'r\rrrr6r7r8rrs866###$$$$$r7rc*eZdZdZdZdZdZdZdS) MemoryLeakTc\tjtj|_dSr)rrrr_MemoryLeak__init_statsrzs r8memory_leak_setupzMemoryLeak.memory_leak_setups$ !688r7c@|jr|dSdSr)_MemoryLeak__enable_leak_checkassert_no_memory_leakrzs r8memory_leak_teardownzMemoryLeak.memory_leak_teardowns/  # )  & & ( ( ( ( ( ) )r7c |j}tj}|j|jz }|j|jz }|j|jz }|j|jz }||||||dSr)rrrrrrrrrs r8rz MemoryLeak.assert_no_memory_leaks(**i#)+ X(  4 ck1  j111 77777r7cd|_dS)NF)rrzs r8disable_leak_checkzMemoryLeak.disable_leak_checks#(   r7N)r(r'r\rrrrrr6r7r8rrsV999 )))888)))))r7rc(eZdZfdZfdZxZS)MemoryLeakMixinc~tt||dSr)superrrrr{rs r8rzMemoryLeakMixin.setUps7 ot$$**,,,      r7ctj|tt|dSr)rrrrrrrs r8rzMemoryLeakMixin.tearDownsB  !!### ot$$--/////r7)r(r'r\rr __classcell__)rs@r8rrsQ!!!!!000000000r7rc#Kddlm}dg}i}d} |D]}|d}|}|ddD]}t||}|d}t||}t |s Jd|z||||f<t |||dV|D]\\}}}t |||dS#|D]\\}}}t |||wxYw) 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 td)Nzcodegen forbidden by test case) RuntimeError)rOrPs r8rzforbid_codegen..fails;<<||t|dS)NT)recordalwaysrr1)rocatch_warnings simplefilterr~rmrrnr)r{messagesrncatchfoundrms r8check_warningsz!CheckWarningsMixin.check_warningss   $D 1 1 1 U  !( + + + EEE                 A  AI&&$$QZ:::QJE  H .....s=AAN)r(r'r\rurRuntimeWarningr9r6r7r8r/r/s90> / / / / / /r7r/c |sdSg}|D]X\}}t|trd|}|d||Yd|S)Nz"{}"z{}={}z, )r*r}r~rir+r)) jit_optionsrkeyrs r8_format_jit_optionsr?s r C!''))// U eS ! ! )MM%((E 7>>#u--.... 99S>>r7c+K td}dttjdd}t j||dz}td i|}t|d5}||}| |dddn #1swxYwYtj d |tj|}|Vtj|dtj|t'j|dS#tj|dtj|t'j|wxYw) 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})``. test_temp_moduleztest_temp_module_{}r_z.pyr)r=Nrr6)rrir~uuiduuid4replacer-rr)r?rXwriterUinsertr import_modulerpopremoverr) source_linesr=tempdirtemp_module_nametemp_module_pathr\linesr:s r8create_temp_modulerPs !344077    % %c3 / /117<<1AE1IJJ)88K88 "C ( ( A ''K'@@E GGENNN                7###-.>??  ($///     g ($///     gs1BE),C E)CE)C;E))AF>c6|g}tjg|zd|gz}tj|tjtj|}||\}}|jdkr-d}t||j|fz||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)r!r"r$)r#rz/process failed with code %s: stderr follows %s ) rUr,r4Popenr6 communicater7rdecode) rrr$r#rEpopenrerrrs r8run_in_subprocessrYs } > e #tTl 2C  S$.O > > >E   11HC 1ASE$4cjjll#CCDDD 8Or7r$c  tjd5}tj}t jd}|t d|ddt|dd|d |j g}tj |tj  j }tj| j}|d fd } tj j } | ||  tj|t,jn d} | |  | |} | |n-#| |wxYw dddn #1swxYwY| 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)r!r<cPj}|dkrd|d|}t|dS)Nrz0strace exited non-zero, process return code was:z. )r7r)problemretrrWs r8 check_returnzstrace..check_returnsJ"Caxx+++!(++"3'''xr7)rz)If this is SIGKILL, increase the timeout?)r<)rNamedTemporaryFiler-getpidrwhichr r~r)rr4rTr6pid threadingTimerkillrThreadrUflushpollsignalSIGINTwait readlinescancel)worksyscallsr#ntf parent_pid strace_binaryrE strace_pidthread_timeoutr` thread_commsr^ strace_datarWs @r8r[r[s:  $U + +4$sY[[  X..  FGG GS__SXXh''SX  Z_>>>Y "%*== ( ( ( ( ( (  $%+53DEEEL     DFFF IIKKKzz||# FM2222D W%%% JJLLL LNNN--//K        ! ! # # # #        ! ! # # # # #i4$4$4$4$4$4$4$4$4$4$4$4$4$4$4$l s+CG<B?G)G<*G,,G<<HHctjdsdSd}d} t||g}n#t$rYdSwxYw|d|vS)z)Checks if strace is supported and workingr,FcHtjtjddgdS)NrSzexit())r4r5rUr,r6r7r8 force_clonez%strace_supported..force_clones!h788888r7cloner<)rUrVrWr[r(r))r{syscalltraces r8strace_supportedrs < " "7 + +u999G{WJ// uu bggenn $$s: AAceZdZdZdZdS)IRPreservingTestPipelinez Same as the standard pipeline, but preserves the func_ir into the metadata store after legalisation, useful for testing IR changesctj|jd}d|_|t t ||gS)Nir_preserving_custom_pipeF)rdefine_nopython_pipelinestate _finalizedadd_pass_afterrrfinalize)r{pipelines r8define_pipelinesz)IRPreservingTestPipeline.define_pipelines.sV%> J355 $ N;;;zr7N)r(r'r\rrr6r7r8rr*s/HH     r7rc ddl}ddlm}tjtjt }tj|ddd}tj|stdttj|d5}| }dddn #1swxYwY| ||}|d}td }|dd D]I} | d d } | D]$} || d | dxxdz cc<%Jgd} tjj|g| R} tj| stdttj| d5}| }dddn #1swxYwY| ||}|dddd } | D]$} || d | dxxdz cc<%tdtdt!|D]8\}}|D]\}}td|d|dd|9td}t!|D]*\}}|D]\}}||||<+tdtdd}t!||D]8\}}|D]\}}td|dd|d|9dS)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 availablertjobsc*ttSrrrYr6r7r8r9z$print_azure_matrix..MsK$4$4r7r/ parametersmatrixNUMPYPYTHONr1)rr buildscriptsazurezazure-windows.ymlz$'azure-windows.yml' is not availablestrategyzNumPy | Python | Countz----------------------- z | z<4z | c*ttSrrr6r7r8r9z$print_azure_matrix..is+c"2"2r7z Python | NumPy | Countc^t|dddS)Nrrr1)rYrZ)rs r8r9z$print_azure_matrix..os!s1Q4::c??1-..r7)r>)yamlrr-rdirnameabspathrr)isfilerrXreadloadrvaluesr"rr*)rr base_path azure_piper\datapipe_yml templates py2np_maptmpltrsetupwinpath azure_windows windows_ymlnpverpyspyvercountrev_mapsorternpss r8print_azure_matrixr<sv KKK 9 9::Iit5JKKJ 7>>* % %ECDDD bgooj))4 0 0Avvxxyyfy--H I4455I2A2::|$X.]]__ : :E eGn %eHo 6 6 6 9 6 6 6 6 :IHHGGL5W555M 7>>- ( (CABBB bgoom,,d 3 3qvvxx))D)00K  #J / 9F66%.!%/222A52222 "### #$$$Y__..//:: sIIKK : :LE5 8e8888888 9 9 9 9 :2233GY__..//** sIIKK * *LE5$)GEN5 ! ! * $%%% #$$$ . .FW]]__&999:: sIIKK : :LE5 8e888E8888 9 9 9 9 :::s$ C**C.1C..HHH)T)NNrQ)r$)rrru collectionsrrrrrVr-rkrr4rUrrerrrmultiprocessingrror)rrCrrr1 functoolsrnumpyrwrrrrr r r r numba.core.typingr numba.core.compilerrrrrrnumba.core.typed_passesrnumba.core.untyped_passesrr>numba.core.runtimernumba.nprrrnumba.core.extendingrrrrrrnumba.core.datamodel.modelsrscipyrTcoverageprocess_startupenable_pyobj_flagsenable_pyobjectforce_pyobj_flagsforce_pyobjectno_pyobj_flags nrt_flagsnrtmake_tag_decoratortagr!r_32bitis_parfors_unsupportedskipIfskip_parfors_unsupported skipUnlessr=skip_unless_py10_or_laterskip_unless_py10skip_unless_py312rWskip_if_py313plus_on_windowsskip_unless_py314_or_latermachineskip_if_linux_aarch64 skip_if_32bitrN IS_NUMPY_2skip_if_numpy_2numpy_sincos_low_precisionrvrYr.rNREDUCED_TESTINGskip_if_reduced_testingr_free_threadingskip_if_freethreadingrBrErGrIrLrOr^_msgskip_unless_scipy SUPPORTEDskip_unless_cffi _lnx_reason linux_only _win_reason windows_only_non_win_reasonskip_if_windows _is_armv7l disabled_testskip_ppc64le_issue4563 has_typeguardskip_unless_typeguardskip_if_typeguardskip_ppc64le_issue6465skip_ppc64le_invalid_ctr_loopuname_unamesystemIS_MACOSskip_macos_fenv_errorsIS_MACOS_ARM64 IS_WIN_ARM64skip_win_arm64_unittuple_uimm12scipy.linalg.cython_lapack has_lapack needs_lapackscipy.linalg.cython_blashas_blas needs_blas _exec_condneeds_subprocess setuptoolshas_setuptoolsr[needs_setuptoolsrrrtrkrrrrrrrr) gettempdirgetuidrrrrrrrrrrrrrrrrr r(rr rr/r?rPrYr[rrrr6r7r8r s  ######    %%%%%% %%%%%% 99999999999999((((((55555555555555322222000000$$$$$$""""""222222433333LLLL EEEOOOH   D UWW%)"EGG#'  EGG  !g !M!M!MNNc-   *8? 0H/ Ow  '8& Ow (H' Ow /x _BCL$;$;E$B$B/   1X0 Ow (LG$$H)9)9););y)H$  (ABB  (F 2 !(/*"?AALG$$*)9)9););x)G*') $ss2:>>*BAFFGGHH *(//*GIIDggc#4llCCEEE(*FHH  ,#HOETM488&8&z';_MM X_!8!8!A!AA; O O ! x3<#:#:5#A#AA;OO $!(/#,"9"9%"@"@/RR X   8 + o66 ))9)9););y)H+KMM RZ^^$91==>> ++-$HO3 ))9)9););y)H+=??!0H)#!!    =H $(3557fn7} )Gfn.G #2(/## %%%%JJJJJ#x":#<>> ####HHHHH!X +B C C Z^^ND 1 1S 8 &8&z3MNNNNWNNN '8&~7NOO D D DVFVFVFVFVFx VFVFVFr"""""&"""  ) ) ) , 7d??GLL!4!4!6!6 FFMMGLL!4!4!6!6,9MM929;;;!GIIM       + + +     ///%%%%%%  $$$$$&$$$))))))))8 0 0 0 0 0): 0 0 0 ***D ,7?MMMM"$$$$N / / / / / / / / :&>>>>B%%%&|$6:6:6:6:6:sZCC#"C#'C;;DDTT&%T&;UU  U V V$#V$