bip  ddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl Z ddl Z ddl Z dZddlmZddlmZmZmZmZmZmZmZmZddlmZdd lmZdd lm Z  dd l!m"Z"dZ# ddl%Z%dZ&d Z(d Z)e jTdddZ+e&se jTdZ,hdZ-e j\e de j^j`gZ1e2Z3e2Z4er` ddl5Z5e1jme j\e5de j^j`e jne5jpe jrdkrede5jtd ddl;Z;e1jme j\e;de j^j`e;jxeedk7re3j{dej|ddZ?e?dk(re;je;jne?dk7r eBd e?d! ddlCZCe1jme j\eCd"e j^j`e3j{d"ejd#d$eE%dd&lFmGZG[G ddlHZIe1jme j\eIjd'e j^j`e j^jgeIjjd(deIjjd)eIjededk7re3j{d'e4j{d' ddlNmOZPe1jme j\ePd*e j^j`e j^jge4j{d*e1Dchc]}|jc}ZReRe-z rJeSeeTrdejd+vrQe2ejeZWeWd,hk7r3eWeRz rd-eRd.eWZXeBeXe1Dcgc]}|jeWvr|c}Z1e jTe1/d0ZYe j^jed12Z[d3Z\d4e jd5e d6d7dfd8Z^e jTd9ZMere jjejjejd:e jjd;dd<e jjd=ddddefe j>ejjd?d=Zje jjeje#redNd@Zlele"_me2gdAe"_ne jpdBkr#e"jje2gdCgdDe"_odEdEdFdGdHdIe"_pe2dJge"_qdKgdLie"_rde"_se"jj{dMyy#e$rdZYwxYw#e$$rdZ#YwxYw#e'$rdZ&YwxYw#e$rYwxYw#e$rY?wxYw#e$rYwxYw#e$rYwxYw#e$rYwxYwcc}wcc}w)ON)contextmanager)LiteralTF) get_fpu_mode)SCIPY_ARRAY_API SCIPY_DEVICEarray_namespace default_xpis_cupyis_daskis_jaxis_torch)FPUModeChangeWarning)patch_lazy_xp_functions)_pep440) dt_configcn|jdd|jdd|jdd|jdd|jdd|jdd dd l} dd lm}t s6|jdd |jdd|jddt jdk(r,tjdkrtjddy y y #t$r|jdd YwxYw#t$r|jdd YwxYw)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 failurezUparallel_threads_limit(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 threadposix) forkserverT)force) addinivalue_linepytest_timeout Exceptionpytest_fail_slowrPARALLEL_RUN_AVAILABLEosnamesys version_infomultiprocessingset_start_method)configrrs I/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/scipy/conftest.pypytest_configurer')s^ I)+ IRT IBD IEG I .0 I /1 IS 4 "  ! "   F    Q  ww'c..8 ((TB 93 I G II S Q SSs$.C53D5DDD43D4cx|jd}|A ttjj dd}|st jd|jd}|Gtjdjdkr%t jd |jdtj5tjd t j  dd lm}d }|rc ttjd }tj*ds1tj,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): ignore)threadpool_limitsTPYTEST_XDIST_WORKER_COUNTOMP_NUM_THREADSblas)user_api)get_closest_markerintrenvironget ValueErrorpytestskipnpintpitemsizexfailargswarningscatch_warnings simplefilter PytestUnraisableExceptionWarning threadpoolctlr/rKeyErrorgetenv cpu_countmax)itemmarkvr/HAS_THREADPOOLCTLxdist_worker_countmax_openmp_threadsthreads_per_workers r&pytest_runtest_setuprRes  " "7 +D  BJJNN=#67A KKK L  " "#3 4D BGGAJ//!3 =diil^LM  "h(O(OP & 7 $    %(4O)P%Q" 99./%'\\^q%8"%();?Q)QST%U"%&86J5 A $ & %  & )$ !=67s)E%F0)E11F04F;F0 F E.-E.1 E?<F0>E??F0 F F0FF0 F-"F0,F--F00F9function)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)rrBwarnr)requestold_modenew_modes r&check_fpu_moder]sT ~H ~H8 .xm4}M!!3 CsAAcy)Nr3r_r&num_parallel_threadsrasr`>cupynumpytorcharray_api_strict jax.numpy dask.arrayrc)idmarksrez2.3z'array-api-strict must be >= version 2.3z2024.12) api_versionrdcpuSCIPY_DEFAULT_DTYPEfloat64)defaultfloat32zUSCIPY_DEFAULT_DTYPE env var, if set, can only be either 'float64' or 'float32'. Got 'z ' instead.rbr.z#cupyx.jit.rawkernel is experimentalrW)signalrfjax_enable_x64jax_default_devicerg)1trueallrvz!'--array-api-backend' must be in z; got )paramsc#>Kt|dt|d|j}t|jd}tr9t ||j5t |5|ddddddy|y#1swYxYw#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 r<r@rrZxpN)skip_or_xfail_xp_backendsparamremptyrrr rys r&rzrz&sgv.gw/ B ! %B%W G  B          s6AB B*B/B7BB BBBzsTest 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 jddrT|j jd xsd } |jD]!\}} |d k7s ||vs| j| #n|j jd drB|j jd xsd } t|z D]}||j| n]|j jddrA|j jd xsd} t|z D]}||j| t|jdk(r|jd}|tvrtd|dtt|j jd xsd|} ||jd| dD] } | |j vst| d|dt|jdkDs~td|j|jD 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>rnp_onlycpu_only eager_only exceptionszInvalid kwargs: rzUnknown backend(s): z; must be a subset of rFrz"do not run with non-NumPy backendsrcrzTno array-agnostic implementation or delegation available for this backend and devicerz*eager checks not executed on lazy backendsr3rzUnknown 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: )listnode iter_markersxp_known_backendssetkwargs TypeErrorr9r:itemsappendxp_skip_cpu_only_backendsxp_skip_eager_only_backendslenrAinsert) rZ skip_or_xfailrbackendreasonsmarkerinvalid_kwargsrinvalid_exceptionsrbackend_reasonskwargs r&_backends_kwargs_from_requestrPsD 7<<,, l-KLMG*;?@ @**<<= '*;;"&z4E'E"F F  F34F3GH4489J4K3LNO O ==  Y .]]&&x0X4XF,3MMO 3(g%'*C#**62 3]]  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 a2j-4MMO ( _Q' '  m=l s K>(LrZr)r<r@returnc|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. rN)rr~)keywordsrr|__name__getattrr;)rZrskip_xfail_reasonsrzrs r&r{r{ss`  %W-=-==6} B {{((#BKK0 v 6 V$ )r`cLt|rtjdt|rddgSt |rtjdt |r'|j j}d|DS|j jdgzS)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#293r~rkNzjax-ml/jax#26000c3@K|]}|jdk7s|yw)metaN)type).0devices r& zdevices..sF6 v0EFs)r r;r@r r r __array_namespace_info__devices)rzrs r&rrsr{  <=r{t} bz  ./|--/779FWFF & & ( 0 0 2dV ;;r`z .hypothesisnondeterministic)r deadline print_blob deterministic)r rrdatabase derandomizesuppress_health_checkSCIPY_HYPOTHESIS_PROFILEc#Kt}ddg}|D]}ttd||<g}|D]}tt||<ddlm}dd g}|D]}t|j d ||<gd }|D]}tt ||<tt d td ttd|j d}|j|td} ddl m } ddl } | 5| jjdtj 5|r5|j"|vr'tj$di||j"dnN|r|j"| vrdn9tj&dt(tj$ddtdddddddy#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)rWmessagerpr) 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)rrWz"The maximum number of subdivisions)rz linalg.rstz stats.rstscipy.signal.normalize)_fixed_default_rngNr.errorz.*odr.*)r.)dictRuntimeWarningDeprecationWarningscipyrIntegrationWarning UserWarningPendingDeprecationWarningupdaterscipy._lib._utilrrcrandomseedrBrCr filterwarningsrDWarning) testknown_warningsdivide_by_zeror deprecatedr integration_wuser_wdctlegitrr=s r&warnings_errors_and_rngrs  !? # BD#'0@$BN4  B   ED#'1C#DN4  E $55 " ND#'1M1M0L$NN4  N H >D#'#=N4  > k3HI979A'::<   c",- 8  !  IINN4 ((* DII7++Rtyy8QRdii50))'7;++HiAST       s7C5G70G 'BF>-G 5 G>G G  GG)zscipy.linalg.LinAlgErrorzscipy.fftpack.fftshiftzscipy.fftpack.ifftshiftzscipy.fftpack.fftfreqzscipy.special.sinczscipy.optimize.show_optionsrrz$scipy.io.matlab.MatlabObject.stridesz"scipy.io.matlab.MatlabObject.dtypez"scipy.io.matlab.MatlabOpaque.dtypez$scipy.io.matlab.MatlabOpaque.stridesz&scipy.io.matlab.MatlabFunction.stridesz$scipy.io.matlab.MatlabFunction.dtype2)zscipy.io.hb_readzscipy.io.hb_writez)scipy.sparse.csgraph.connected_componentsz&scipy.sparse.csgraph.depth_first_orderz"scipy.sparse.csgraph.shortest_pathz#scipy.sparse.csgraph.floyd_warshallzscipy.sparse.csgraph.dijkstraz!scipy.sparse.csgraph.bellman_fordzscipy.sparse.csgraph.johnsonzscipy.sparse.csgraph.yenz(scipy.sparse.csgraph.breadth_first_orderz*scipy.sparse.csgraph.reverse_cuthill_mckeez$scipy.sparse.csgraph.structural_rankz*scipy.sparse.csgraph.construct_dist_matrixz%scipy.sparse.csgraph.reconstruct_pathzscipy.ndimage.value_indiceszscipy.stats.mstats.describe)zscipy.stats.distributionszscipy.optimize.cython_optimizez scipy.testzscipy.show_configzscipy/special/_precomputez#scipy/interpolate/_interpnd_info.pyz'scipy/interpolate/_rbfinterp_pythran.pyzscipy/_build_utils/tempita.pyzscipy/_lib/array_api_compatzscipy/_lib/highszscipy/_lib/unuranzscipy/_lib/_gcutils.pyzscipy/_lib/doccer.pyzscipy/_lib/_uarrayz+scipy/linalg/_cython_signature_generator.pyzscipy/linalg/_generate_pyx.pyzscipy/linalg/_linalg_pythran.pyz$scipy/linalg/_matfuncs_sqrtm_triu.pyz1scipy/ndimage/utils/generate_label_testvectors.pyz scipy/optimize/_group_columns.pyz$scipy/optimize/_max_len_seq_inner.pyz"scipy/signal/_max_len_seq_inner.pyz%scipy/sparse/_generate_sparsetools.pyzscipy/special/_generate_pyx.pyzscipy/stats/_stats_pythran.pyzReST parser limitationz__cinit__ unexpected argumentznan in scalar_powerzintegration warning)zND_regular_grid.rstzextrapolation_examples.rstzsampling_pinv.rstzsampling_srou.rstzprobability_distributions.rstzintegrate.nquad(func,zio.rst)z octave_a.matzoctave_cells.matzoctave_struct.matz.text()N)ujsonr#rr!rBtempfile contextlibrtypingrrcr=r; hypothesishypothesis_available ImportErrorscipy._lib._fpumoderscipy._lib._array_apirrrr r r r r scipy._lib._testutilsr"scipy._lib.array_api_extra.testingr scipy._librscipy_doctest.conftestr HAVE_SCPDTModuleNotFoundErrorpytest_run_parallelrrr'rRfixturer]rarr|rLarray_api_backendsxp_available_backendsrrrrerparse __version__Versionset_array_api_strict_flagsrdset_default_deviceaddrHrnset_default_dtypermr:rbr FutureWarning cupyx.scipyrq jax.numpyjax thread_unsafer%rr dask.arrayarraydarhxp_available_backend_ids isinstancestrlowerloadsSCIPY_ARRAY_API_msgrzskipifskip_xp_invalid_argrFixtureRequestr{ configurationset_hypothesis_home_dirpathjoin gettempdirsettingsregister_profiler HealthCheckr8r9r load_profileruser_context_mgrskiplistpytest_extra_ignorepytest_extra_xfail pseudocodelocal_resources strict_check stopwords)pr|s00r&rs$  % !-7F0J#! 9Cx.bj$/ C0 C ^^ #FLLv{{'E'EF E!e  $$ FLL).@%{{== ? @ 7==)55 69O OGH H333!  $$ FLL7++00 2 3 !  . 5 % ) )' 2"))19E i  #E # #EMM 2  !%%,IZ9   $$ FLL&++00 2 3 "%%f-  ;m  '  $$ FLL{;;11;;,,. / 0 *D1 .  L0I!0LM 5 % ) )+ 6 $'' 4 $$ FLL ;;11;;,,. / 0 $'' 5/DDD'*;;; ; ?C( !O ! ! #+? ?ztzz/:; w &":::236:J9KM o%$9%88//% ! ,-!.!Hkk((I)K > B;%v'<'<;%-4_-E;%JN;%|#<# IIV"9II( ~~!!#'# ,%I!> 8&><2)> $I  789I !I"IH%]K! !J #"#v    *    $    &    (     E%sX!X%*X3$BY+B+Y A)YCY%A%Y19Y=)ZX"!X"%X0/X03X>=X>Y  Y  YYY"!Y"%Y.-Y.1Y:9Y: