Yhnk  ddlZddlZddlZddlZddlmZddlmZddlZ ddl m Z ddl Z ddlZddlmZddlmZmZmZmZmZmZmZddlmZddlmZddlmZ dd lm Z d Z!n #e"$rd Z!YnwxYw ddl#Z#d Z$n #e%$rd Z$YnwxYwd Z&d Z'e j(dd dZ)e$se j(dZ*hdZ+de iZ,e-Z.e-Z/er ddl0Z0e,1de0iej2e0j3ej4dkr e5de0j6dn #e5$rYnwxYw ddl7Z7e,1de7ie7j8eedkre.9dej:ddZ;e;dkre7j<e7j=ne;dkre>de;d n #e5$rYnwxYw ddl?Z?e,1d!e?ie.9d!ej@d"d#eA$dd%lBmCZC[Cn #e5$rYnwxYw ddlDZEe,1d&eEjieEjF1d'd eEjF1d(eEjGededkre.9d&e/9d&n #e5$rYnwxYw ddlHmIZJe,1d)eJie/9d)n #e5$rYnwxYweKeeLr[ejMd*vrJejNeZOeOd+gkr3 d,eODZ,n%#eP$rd-eQe,d.eOZRe>eRwxYwe-e,e+z rJe-e,dhz ZSe j(d/e,TD0d1ZUe jVWed23ZXd4ZYd5e jZd6ed7d8dfd9Z[e j(d:ZGej\]ej^_ej`d;ejabddd dd eQejc?ejded@d>Zfejagefe!redOdAZhehe _ie-gdBe _je j3dCkr%e jj1e-gdDgdEe _kdFdFdGdHdIdJe _le-dKge _mdLgdMie _nd e _oe jp9dNdSdS)PN)contextmanager)Literal) get_fpu_mode)SCIPY_ARRAY_API SCIPY_DEVICEarray_namespace default_xpis_cupyis_daskis_jax)FPUModeChangeWarning)patch_lazy_xp_functions)_pep440) dt_configTFcX|dd|dd|dd|dd|dd|dd dd l}n&#t$r|dd YnwxYw dd lm}n&#t$r|dd YnwxYwt sD|dd |dd|ddd Sd S)a Add pytest markers to avoid PytestUnknownMarkWarning This needs to contain all markers that are SciPy-specific, as well as dummy fallbacks for markers defined in optional test packages. Note that we need both the registration here *and* in `pytest.ini`. markerszslow: Tests that are very slow.zHxslow: mark test as extremely slow (not run unless explicitly requested)z8xfail_on_32bit: mark test as failing on 32-bit platformsz;array_api_backends: test iterates on all array API backendszskip_xp_backends(backends, reason=None, np_only=False, cpu_only=False, eager_only=False, exceptions=None): mark the desired skip configuration for the `skip_xp_backends` fixturezxfail_xp_backends(backends, reason=None, np_only=False, cpu_only=False, eager_only=False, exceptions=None): mark the desired xfail configuration for the `xfail_xp_backends` fixturerNz.timeout: mark a test for a non-default timeout)parse_durationz8fail_slow: mark a test for a non-default timeout failurezOparallel_threads(n): run the given test function in parallel using `n` threads.z8thread_unsafe: mark the test function as single-threadedzCiterations(n): run the given test function `n` times in each thread)addinivalue_linepytest_timeout Exceptionpytest_fail_slowrPARALLEL_RUN_AVAILABLE)configrrs `/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/scipy/conftest.pypytest_configurer$s  I)+++ IRTTT IBDDD IEGGG I .000 I /111 I III G I I I I IIS 4333333 SSS Q S S S S SS "    ! " " "   F      Q        s$B B.-B.2B99 CCc|d}|W ttjdd}n#t $rd}YnwxYw|st jd|d}|?tj dj dkr"t j d |j dtj5}|t j dd lm}d }n#t&$rd}YnwxYw|r ttjd }n#t($rYddddSwxYwtjd sTtjdz}t/||zd} ||dn#t&$rYddddSwxYwddddS#1swxYwYdS)Nxslow SCIPY_XSLOW0Fz@very slow test; set environment variable SCIPY_XSLOW=1 to run itxfail_on_32bitrz&Fails on our 32-bit test platform(s): )threadpool_limitsTPYTEST_XDIST_WORKER_COUNTOMP_NUM_THREADSblas)user_api)get_closest_markerintosenvironget ValueErrorpytestskipnpintpitemsizexfailargsnptsuppress_warningsfilter PytestUnraisableExceptionWarning threadpoolctlr"rKeyErrorgetenv cpu_countmax) itemmarkvsupr"HAS_THREADPOOLCTLxdist_worker_countmax_openmp_threadsthreads_per_workers rpytest_runtest_setuprGZs  " "7 + +D  BJNN=#6677AA   AAA  L KK L L L  " "#3 4 4D BGAJJ/!33 LdilLLMMM  C 6:;;; & 7 7 7 7 7 7 $   & & & %    &    %(4O)P%Q%Q""   )$ 9.// %'\^^q%8"%();?Q)QST%U%U"%%&86JJJJJ =67s-A AA G9DG DGDGD76G7 EGEAG FG F8)G7F88GG G function)scopeautousec#Kt}dVt}||kr'tjd|dd|ddtddSdS)z9 Check FPU mode was not changed during the test. NzFPU mode changed from z#xz to z during the testr)category stacklevel)rwarningswarnr )requestold_modenew_modes rcheck_fpu_moderSs ~~H EEE~~H8 !xU!!U!!!3 C C C C C CcdS)Nr&rVrTrnum_parallel_threadsrWsqrT>cupynumpytorcharray_api_strict jax.numpy dask.arrayrYr[z2.3z'array-api-strict must be >= version 2.3z2024.12) api_versionrZcpuSCIPY_DEFAULT_DTYPEfloat64)defaultfloat32zUSCIPY_DEFAULT_DTYPE env var, if set, can only be either 'float64' or 'float32'. Got 'z ' instead.rXignorez#cupyx.jit.rawkernel is experimentalrL)signalr\jax_enable_x64jax_default_devicer])1trueallrkc*i|]}|t|SrV)xp_available_backends.0backends r rqs0)))27;)))rTz!'--array-api-backend' must be in z; got c`g|]+\}}tj||tjj,S))idmarks)r/paramr@array_api_backends)rokrAs r rxsB 1 Lq >???rT)paramsc#<Kt|dt|d|j}t|d}trEt |j||t |5|VddddS#1swxYwYdS|VdS)aRun the test that uses this fixture on each available array API library. You can select all and only the tests that use the `xp` fixture by passing `-m array_api_backends` to pytest. You can select where individual tests run through the `@skip_xp_backends`, `@xfail_xp_backends`, and `@skip_xp_invalid_arg` pytest markers. Please read: https://docs.scipy.org/doc/scipy/dev/api-dev/array_api.html#adding-tests r0r4r)xprP monkeypatchN)skip_or_xfail_xp_backendsruremptyrrr )rPr|r{s rr{r{s"gv...gw/// B ! % %B }g;    ^^  HHH                  s9B  BBzsTest involves masked arrays, object arrays, or other types that are not valid input when `SCIPY_ARRAY_API` is used.reasonc t|j|d}dtD}|D]}t |jhdz }|rt d|t |jdg}|tz }t|tz x}r'td|dtt|jdd rE|jd pd }t|z D]} ||  |n|jd d rE|jd pd }t|z D]} ||  |n_|jdd rD|jd pd}t|z D]} ||  |t|jdkr|jd} | tvr'td| dtt|jd pd| }|| d|dD]} | |jvrt| d|  ut|jdkrtd|jd|DS)ztA helper for {skip,xfail}_xp_backends. Return dict of {backend to skip/xfail: top reason to skip/xfail it} _xp_backendsci|]}|gSrVrVrns rrqz1_backends_kwargs_from_request..=s<<rnp_onlycpu_only eager_only exceptionszInvalid kwargs: rzUnknown backend(s): z; must be a subset of rFrz"do not run with non-NumPy backendsrzTno array-agnostic implementation or delegation available for this backend and devicerz*eager checks not executed on lazy backendsr&rzUnknown backend: z; must be one of z#do not run with array API backend: )rrrrz is mutually exclusive with z,Please specify only one backend per marker: c*i|]\}}|||dS)rrV)rorpbackend_reasonss rrqz1_backends_kwargs_from_request..rs8   ( G_Q'   rT)listnode iter_markersxp_known_backendssetkwargs TypeErrorr-r.xp_skip_np_only_backendsappendxp_skip_cpu_only_backendsxp_skip_eager_only_backendslenr5insertitems) rP skip_or_xfailrreasonsmarkerinvalid_kwargsrinvalid_exceptionsrrpkwargs r_backends_kwargs_from_requestr7s 7<,, -K-K-KLLMMG<<*;<<< 6C 0 0 ''//// v{  q k!nG/// "MW"M"M378I3J3J"M"MNNN]&&x00A?g??  G  # #Av . . .L V VFM))$%T%T7%T%TUUU* V   ! !Lv{LL "   ,3MMOO    rTrPr)r0r4returnc|d|jvrdSt||}|j}|j|vr4||j}|sJt t |}||dSdS)aP Helper of the `xp` fixture. Skip or xfail based on the ``skip_xp_backends`` or ``xfail_xp_backends`` markers. See the "Support for the array API standard" docs page for usage examples. Usage ----- :: skip_xp_backends = pytest.mark.skip_xp_backends xfail_xp_backends = pytest.mark.xfail_xp_backends ... @skip_xp_backends(backend, *, reason=None) @skip_xp_backends(*, cpu_only=True, exceptions=(), reason=None) @skip_xp_backends(*, eager_only=True, exceptions=(), reason=None) @skip_xp_backends(*, np_only=True, exceptions=(), reason=None) @xfail_xp_backends(backend, *, reason=None) @xfail_xp_backends(*, cpu_only=True, exceptions=(), reason=None) @xfail_xp_backends(*, eager_only=True, exceptions=(), reason=None) @xfail_xp_backends(*, np_only=True, exceptions=(), reason=None) Parameters ---------- backend : str, optional Backend to skip/xfail, e.g. ``"torch"``. Mutually exclusive with ``cpu_only``, ``eager_only``, and ``np_only``. cpu_only : bool, optional When ``True``, the test is skipped/xfailed on non-CPU devices, minus exceptions. Mutually exclusive with ``backend``. eager_only : bool, optional When ``True``, the test is skipped/xfailed for lazy backends, e.g. those with major caveats when invoking ``__array__``, ``__bool__``, ``__float__``, or ``__complex__``, minus exceptions. Mutually exclusive with ``backend``. np_only : bool, optional When ``True``, the test is skipped/xfailed for all backends other than the default NumPy backend and the exceptions. Mutually exclusive with ``backend``. Implies ``cpu_only`` and ``eager_only``. reason : str, optional A reason for the skip/xfail. If omitted, a default reason is used. exceptions : list[str], optional A list of exceptions for use with ``cpu_only``, ``eager_only``, or ``np_only``. This should be provided when delegation is implemented for some, but not all, non-CPU/non-NumPy backends. rN)rr)keywordsrru__name__getattrr/)rPrskip_xfail_reasonsr{rs rr}r}ws` %%%W-===6} B {(((#BK0  66  V$$$$$$ )(rTc t|rtjdt|rddgSt |rtjd|dgzS)aoFixture that returns a list of all devices for the backend, plus None. Used to test input->output device propagation. Usage ----- from scipy._lib._array_api import xp_device def test_device(xp, devices): for d in devices: x = xp.asarray(..., device=d) y = f(x) assert xp_device(y) == xp_device(x) zdata-apis/array-api-compat#293rr_Nzjax-ml/jax#26000)r r/r4r r __array_namespace_info__devices)r{s rrrsr{{>  <====r{{t} bzz0  .//// & & ( ( 0 0 2 2dV ;;rTz .hypothesisnondeterministic)namedeadline print_blob deterministic)rrrdatabase derandomizesuppress_health_checkSCIPY_HYPOTHESIS_PROFILEc#Kt}ddg}|D]}ttd||<gd}|D]}tt||<ddlm}d d g}|D]}t|jd ||<gd }|D]}tt ||<tt d tdttd|jd}||td} ddl m } ddl } | 5| j dtj5|r*|j|vr!tjdi||jdVn.|r|j| vrdVntjdt(dVdddn #1swxYwYddddS#1swxYwYdS)zmTemporarily turn (almost) all warnings to errors. Filter out known warnings which we allow. zscipy.linalg.normzscipy.ndimage.center_of_masszdivide by zero)rLmessage)zscipy.signal.cwtzscipy.signal.morletzscipy.signal.morlet2zscipy.signal.rickerzscipy.integrate.simpsonzscipy.interpolate.interp2dzscipy.linalg.kronrer) integratezscipy.special.ellip_normalzscipy.special.ellip_harm_2zThe occurrence of roundoff)zscipy.stats.anderson_ksampzscipy.stats.kurtosistestzscipy.stats.normaltestscipy.sparse.linalg.normzExited at iterationzthe matrix subclass is not)rrLz"The maximum number of subdivisions)rz linalg.rstz stats.rstscipy.signal.normalize)_fixed_default_rngNrderror)rd)dictRuntimeWarningDeprecationWarningscipyrIntegrationWarning UserWarningPendingDeprecationWarningupdaterscipy._lib._utilrrYrandomseedrNcatch_warningsrfilterwarnings simplefilterWarning) testknown_warningsdivide_by_zeror deprecatedr integration_wuser_wdctlegitrr1s rwarnings_errors_and_rngrs7  !? # B BD#'0@$B$B$BN4     E ED#'1C#D#D#DN4 ######55 " N ND#'1M0L$N$N$NN4 HHH > >D#'#=#=#=N4 k3HIII97999A':<<<    c""",-- 877777   ! !   INN4 (**  DI77+II.r s  %%%%%% ,,,,,,766666FFFFFF000000JJJJJ#!###"#3 3 3 l...bj$/// C C0/ C ^^ ### " CEE!cee[&  $$&8:J%KLLL 7=)5 6 69O9O O O+GHH H33!             $$gu%5666  ... 5 % ) )' 2 2 2")19EEE i   #E #EM 2 2 2 2  ! !*9%,999          $$fd^444!%%f---  ;m     '&&&&& F        $$k39%=>>> *D111 .  L0I0I!0LMMM 5 % ) )+ 6 6 6 $'' 4444        $$lB%7888 $'' 5555       ?C((& !O ! # #+? ? ?%4:o66 w & & &))#3)))%% & & &P$455PP=MPP joo% & 3$ % %(9 99993455 A%++--## #Lk((I)KK = = = @;%v'<;%-4_-E;%JN;%;%;%;%|<<>*D*9;;  !9::: t&OOO^Ob"9II( ~!!##'''##   ,%%%I!> 8&><2)> $$I 3 7899I   !I"IH%%%%%it&t&sA##A-,A-1A88BBAD77D?>D?B G GGA H$$H,+H,0BK KK2L LL M"M9