:jXx 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)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 jTe10d1ZYe j^jed23Z[d4Z\d5e jd6e d7d8dfd9Z^e jTd:ZMere jjejjejd;e jjdddddefe j?ejjd@d>Zje jjeje#rqedMdAZlele"_me2gdBe"_ngdCe"_odDdDdEdFdGdHe"_pe2dIge"_qdJgdKie"_rde"_se"jj{dLyy#e$rdZYwxYw#e$$rdZ#YwxYw#e'$rdZ&YwxYw#e$rYwxYw#e$rY)wxYw#e$rYwxYw#e$rYwxYw#e$rYwxYwcc}wcc}w)NN)contextmanager)LiteralTF) get_fpu_mode)SCIPY_ARRAY_API SCIPY_DEVICEarray_namespace default_xpis_cupyis_daskis_jaxis_torch)FPUModeChangeWarning)patch_lazy_xp_functions)version) dt_configc|jdd|jdd|jdd|jdd|jdd|jdd|jdd d d l} d d lm}t s6|jdd|jdd|jddt jdk(r@tjdkr,tjdk7rtjddy y 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` fixturezuses_xp_capabilities(status, funcs=None, reason=None): mark whether pytest markers for array API backends are generated from the xp_capabilities entries for one or more functionsrNz.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)cygwin forkserverT)force) addinivalue_linepytest_timeout Exceptionpytest_fail_slowrPARALLEL_RUN_AVAILABLEosnamesys version_infoplatformmultiprocessingset_start_method)configrrs c/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/scipy/conftest.pypytest_configurer))s I)+ IRT IBD IEG I .0 I /1 I/1IS 4 "  ! "   F    Q  ww'c..8S\\X=U ((TB >V83 I G II S Q SSs$DD<D98D9<EEcx|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_markerintr environget ValueErrorpytestskipnpintpitemsizexfailargswarningscatch_warnings simplefilter PytestUnraisableExceptionWarning threadpoolctlr1rKeyErrorgetenv cpu_countmax)itemmarkvr1HAS_THREADPOOLCTLxdist_worker_countmax_openmp_threadsthreads_per_workers r(pytest_runtest_setuprTks  " "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)rrDwarnr)requestold_modenew_modes r(check_fpu_moder_sT ~H ~H8 .xm4}M!!3 CsAAcy)Nr5rar(num_parallel_threadsrcsrb>cupynumpytorcharray_api_strict jax.numpy dask.arrayre)idmarksrgz2.3z'array-api-strict must be >= version 2.3z2025.12) api_versionrfcpuSCIPY_DEFAULT_DTYPEfloat64)defaultfloat32zUSCIPY_DEFAULT_DTYPE env var, if set, can only be either 'float64' or 'float32'. Got 'z ' instead.rdr0z#cupyx.jit.rawkernel is experimentalrY)signalrhjax_enable_x64jax_default_matmul_precisionjax_default_deviceri)1trueallryz!'--array-api-backend' must be in z; got )paramsc#Kt|dt|d|jjdtjdd|j }t |jd}tr9t||j 5t|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>rBuses_xp_capabilitiesNztest uses `xp` fixture without drawing from `xp_capabilities` but is not explicitly marked with ``pytest.mark.uses_xp_capabilities(False)``r)rZr\xp) skip_or_xfail_xp_backendsnoder8rDr[paramremptyrrr r}s r(r~r~0sgv.gw/||&&'=>F  ;   B ! %B%W G  B          s6BC CB7!C)C7C <CC CzsTest 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]b}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|jdk(rt!|jdt"t%dzr|j\}} |tvrtd|dtt| r0|j jd } ||jd| dD] } | |j vst| d|3t|jdkDsMtd|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 backendsrerzTno array-agnostic implementation or delegation available for this backend and devicerz*eager checks not executed on lazy backendsr5rzUnknown backend: z; must be one of z#do not run with array API backend: )rrrrz is mutually exclusive with r4Nz,Please specify only one backend per marker: )listr iter_markersxp_known_backendssetkwargs TypeErrorr;r<itemsappendxp_skip_cpu_only_backendsxp_skip_eager_only_backendslenrCinsert isinstancebooltype) r\ skip_or_xfailrbackendreasonsmarkerinvalid_kwargsrinvalid_exceptionsrbackend_reasonskwarg conditions r(_backends_kwargs_from_requestrhs 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  "z&++a.$dBS'T"( GY// #4WI>3378I3J2K"MNN**84 ''62L VFMM)$w.J7)%TUU V  !>v{{mL IFR-4MMO ( _Q' '  U=T s O 7Or\r)r>rBreturnc|d|jvryt||}|j}|j|vr-||j}|sJt t |}||yy)a 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(backend, condition, /, *, reason='skip because condition') @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(backend, condition, /, *, reason='xfail because condition') @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=)r\rskip_xfail_reasonsr~rs r(rrssd  %W-=-==6} B {{((#BKK0 v 6 V$ )rbc\t|rtjdt|rddgSt |rtjdt |r'|j j}d|DSt|j jdzS)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#293rrmNzjax-ml/jax#26000c3@K|]}|jdk7s|yw)metaN)r).0devices r( zdevices..sF6 v0EFsN) r r=rBr r r __array_namespace_info__devicestuple)r~rs r(rrsr{  <=r{t} bz  ./|--/779FWFF ,,.668 9G CCrbz .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)rYmessagerrr) 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)rrYz"The maximum number of subdivisions)rz linalg.rstz stats.rstscipy.signal.normalize)_fixed_default_rngNr0errorz.*odr.*)r0)dictRuntimeWarningDeprecationWarningscipyrIntegrationWarning UserWarningPendingDeprecationWarningupdaterscipy._lib._utilrrerandomseedrDrEr!filterwarningsrFWarning) testknown_warningsdivide_by_zeror! deprecatedr integration_wuser_wdctlegitrr?s r(warnings_errors_and_rngrEs  !? # 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.dtypezscipy.interpolate.lagrangez/scipy.interpolate.approximate_taylor_polynomialzscipy.interpolate.padezscipy.spatial.tsearchz"scipy.spatial.minkowski_distance_pz scipy.spatial.minkowski_distancezscipy.spatial.distance_matrix)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/_externalzscipy/_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(r)ujsonr%r r"rDtempfile contextlibrtypingrrer?r= hypothesishypothesis_available ImportErrorscipy._lib._fpumoderscipy._lib._array_apirrrr r r r r scipy._lib._testutilsr'scipy._external.array_api_extra.testingr!scipy._external.packaging_versionrscipy_doctest.conftestr HAVE_SCPDTModuleNotFoundErrorpytest_run_parallelrrr)rTfixturer_rcrrrNarray_api_backendsxp_available_backendsrrrrgrparse __version__Versionset_array_api_strict_flagsrfset_default_deviceaddrJrpset_default_dtyperor<rdr FutureWarning cupyx.scipyrs jax.numpyjax thread_unsafer'rr dask.arrayarraydarjxp_available_backend_idsrstrlowerloadsSCIPY_ARRAY_API_msgr~skipifskip_xp_invalid_argrFixtureRequestr configurationset_hypothesis_home_dirpathjoin gettempdirsettingsregister_profiler HealthCheckr:r;r load_profileruser_context_mgrskiplistpytest_extra_ignorepytest_extra_xfail pseudocodelocal_resources strict_check stopwords)prs00r(rs  % !-7K50J#! ?CD.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 8)D .  L0I!0LM 5 % ) )+ 6 $'' 4 $$ FLL ;;11;;,,. / 0 $'' 5/DDD'*;;; ; ?C( !O ! ! #+? ?ztzz/:; w &":::236:J9KM o%$9%88//% ! ,-/./dkk((I)K R j=%v'<'<=%-4_-E=%JN=%@"D"DJ44  (X((*M:(( $4)(( $4":#9#9:) "zz~~.H.= ?$$%=> IIV"9II:%I!@ 8&><2)> $I  789I !I"IH%Ce! !J #"#B    *    $    .    (     E%sX!X*X$BX++B+X7A)YC*Y,A%YY'Y,X  X XXX('X(+X43X47Y?YY  Y YYY$#Y$