Џkhnk B ddlZddlZddlZddlZddlmZddlmZddlZ ddl m Z ddl Z ddlZddlmZddlmZmZmZmZmZmZmZddlmZddlmZddlmZ dd lm Z d Z! ddl#Z#d Z$d Z&d Z'e jPdd dZ)e$se jPdZ*hdZ+de iZ,e-Z.e-Z/er6 ddl0Z0e,jcde0iejde0jfejhdkre5de0jld ddl7Z7e,jcde7ie7jpeedk7re.jsdejtddZ;e;dk(re7jxe7jzne;dk7r e>de;d  ddl?Z?e,jcd!e?ie.jsd!ejd"d#eA$dd%lBmCZC[C ddlDZEe,jcd&eEjieEjjcd'd eEjjcd(eEjededk7re.jsd&e/jsd& ddlHmIZJe,jcd)eJie/jsd)eKeeLr@ejd*vr-ejeZOeOd+gk7r eODcic]}|e,| c}Z,e-e,e+z rJe-e,dhz ZSe jPe,jDcgc].\}}e j||e jj.0c}}/d0ZXe jjed12ZZd3Z[d4e jd5ed6d7dfd8Z]e jPd9ZGejjejjejd:ejjd;dd <ejjd=dd dd eQej>ejjd?d=Zhejjehe!redNd@Zjeje _ke-gdAe _le jfdBkr#e jjce-gdCgdDe _mdEdEdFdGdHdIe _ne-dJge _odKgdLie _pd e _qe jjsdMyy#e"$rd Z!YwxYw#e%$rd Z$YwxYw#e5$rY+wxYw#e5$rYwxYw#e5$rYwwxYw#e5$rYwxYw#e5$rYwxYwcc}w#eP$rd,eQe,d-eOZRe>eRwxYwcc}}w)ON)contextmanager)Literal) get_fpu_mode)SCIPY_ARRAY_API SCIPY_DEVICEarray_namespace default_xpis_cupyis_daskis_jax)FPUModeChangeWarning)patch_lazy_xp_functions)_pep440) dt_configTFc|jdd|jdd|jdd|jdd|jdd|jdd dd l} dd lm}t s7|jdd |jdd|jddy y #t$r|jdd YbwxYw#t$r|jdd Y|wxYw)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 J/var/www/teggl/fontify/venv/lib/python3.12/site-packages/scipy/conftest.pypytest_configurer$s. I)+ IRT IBD IEG I .0 I /1 IS 4 "  ! "   F    Q  " I G II S Q SSs$.B73C7CCC65C6cn|jd}|A ttjj dd}|st jd|jd}|Gtjdjdkr%t jd |jdtj5}|jt j  dd lm}d }|rc ttjd }tj*d s1tj,dz}t/||zd} |ddddy#t $rd}Y*wxYw#t&$rd}YwxYw#t($r YdddywxYw#t&$r YdddywxYw#1swYyxYw)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  BJJNN=#67A KKK L  " "#3 4D BGGAJ//!3 =diil^LM   C 6::; & 7 $    %(4O)P%Q" 99./%'\\^q%8"%();?Q)QST%U"%&86J5 A $ & %  & )$ !=67s)E F+$E,,F+/E= ;F+ F E)(E), E:7F+9E::F+= FF+FF+ F(F+'F((F++F4function)scopeautousec#Kt}dt}||k7r&tjd|dd|ddtdyyw)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_moderSsT ~H ~H8 .xm4}M!!3 CsAAcy)Nr&rUrnum_parallel_threadsrWsrV>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])1trueallrkz!'--array-api-backend' must be in z; got )idmarks)paramsc#Kt|dt|d|j}t|jd}tr1t |j||t |5|dddy|y#1swYyxYww)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_backendsparamremptyrrr )rPrqrps rrprps}"gv.gw/ B ! %B}}g; ^ H    sA(B *A=/B =BB zsTest involves masked arrays, object arrays, or other types that are not valid input when `SCIPY_ARRAY_API` is used.reasonc t|jj|d}tDcic]}|g}}|D]}t |j hdz }|rt d|t |j jdg}|tz }t|tz x}rtd|dtt|j jddrB|j jd xsd } t|z D]}||j| n|j jd drB|j jd xsd } t|z D]}||j| n]|j jd drA|j jd xsd} t|z D]}||j| t|jdk(r|jd}|tvrtd|dtt|j jd xsd|} ||jd| dD] } | |j vst| d|Rt|jdkDsltd|j|j!D cic] \}} | r|| dc} }Scc}wcc} }w)ztA helper for {skip,xfail}_xp_backends. Return dict of {backend to skip/xfail: top reason to skip/xfail it} _xp_backends>rvnp_onlycpu_only eager_only exceptionszInvalid kwargs: r|zUnknown backend(s): z; must be a subset of ryFrvz"do not run with non-NumPy backendsrzzTno array-agnostic implementation or delegation available for this backend and devicer{z*eager checks not executed on lazy backendsr&rzUnknown backend: z; must be one of z#do not run with array API backend: )rzryr{r|z is mutually exclusive with z,Please specify only one backend per marker: )listnode iter_markersxp_known_backendssetkwargs TypeErrorr-r.xp_skip_np_only_backendsappendxp_skip_cpu_only_backendsxp_skip_eager_only_backendslenr5insertitems) rP skip_or_xfailrbackendreasonsmarkerinvalid_kwargsr|invalid_exceptionsrvkwargbackend_reasonss r_backends_kwargs_from_requestr7s4 7<<,, l-KLMG*;?@ @**<<= '*;;"&z4E'E"F F  F34F3GH4489J4K3LNO O ==  Y .]]&&x0X4XF3j@ 0 ''/ 0]]  z5 1]]&&x0/. 5zA 0 ''/ 0]]  |U 3]]&&x0>< 6C 0 ''/ 0 v{{ q kk!nG// #4WI>3378I3J2K"MNN]]&&x0A5gY?  G  # #Av .L VFMM)$w.J7)%TUU V  !>v{{mL _1h-4MMO ( _Q' '  k=j s K,K1rPr)r0r4returnc|d|jvryt||}|j}|j|vr-||j}|sJt t |}||yy)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. rxN)rru)keywordsrrs__name__getattrr/)rPrskip_xfail_reasonsrprvs rrrrrwss`  %W-=-==6} B {{((#BKK0 v 6 V$ )rVct|rtjdt|rddgSt |rtjd|j j 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#293rur_Nzjax-ml/jax#26000)r r/r4r r __array_namespace_info__devices)rps rrrs`r{  <=r{t} bz  ./ & & ( 0 0 2dV ;;rVz .hypothesisnondeterministic)namedeadline print_blob deterministic)rrrdatabase derandomizesuppress_health_checkSCIPY_HYPOTHESIS_PROFILEc#~Kt}ddg}|D]}ttd||<gd}|D]}tt||<ddlm}d d g}|D]}t|j d ||<gd }|D]}tt ||<tt d tdttd|j d}|j|td} ddl m } ddl } | 5| jjdtj 5|r5|j"|vr'tj$di||j"dn3|r|j"| vrdntj&dt(dddddddy#1swYxYw#1swYyxYww)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_rngNerror)rd)dictRuntimeWarningDeprecationWarningscipyrIntegrationWarning UserWarningPendingDeprecationWarningupdaterscipy._lib._utilrrYrandomseedrNcatch_warningsrfilterwarnings simplefilterWarning) testknown_warningsdivide_by_zeror deprecatedr integration_wuser_wdctlegitrr1s rwarnings_errors_and_rngrs  !? # BD#'0@$BN4  B   ED#'1C#DN4  E $55 " ND#'1M1M0L$NN4  N H >D#'#=N4  > k3HI979A'::<   c",- 8  !  IINN4 ((* DII7++I.rsr % ,7F0J#! 3 l.bj$/ C0 C ^^ # " E!e  $$&8:J%KL 7==)55 69O OGH H333!  $$gu%56   . 5 % ) )' 2"))19E i  #E # #EMM 2  !%%,IZ9   $$fd^4!%%f-  ;m  '   $$k399%=> *D1 .  L0I!0LM 5 % ) )+ 6 $'' 4  $$lB%78 $'' 5 ?C( !O ! ! #+? ?%4::o6 w & &$4)27;;)% $ %(9 99945 A%++- 1FLLq > >?# #Lkk((I)K = @;%v'<'<;%-4_-E;%JN;%|<>*D*9;  !9: OOb"9II( ~~!!#'# ,%I!> 8&><2)> $I  789I !I"IH%iyJ #"#b    &               ) &:456f=M