^Mh V BddlZddlZddlZddlZddlmZddlZddlm Z ddl Z ddl Z ddl mZddlmZddlmZmZddlmZ ddlmZdZn #e$rd ZYnwxYw ddlZdZn #e$rd ZYnwxYwd Zd Ze jd d dZ ese jdZ!deiZ"ere#ee$r ddl%Z%e"&de%iej'e%j(ej)dkr e*de%j+dn #e*$rYnwxYw ddl,Z,e"&de,ie,j-en #e*$rYnwxYw ddl.Z.e"&de.in #e*$rYnwxYw ddl/Z0e"&de0jie0j1&dde0j1&de0j2edn #e*$rYnwxYwej3dvrNej4eZ5de5vrn9 de5DZ"n+#e6$r#de"7Z8e9e8wxYwde"vrdZe j:;d e"<Z=e j:>ed!"Z?d#Z@e jd$ZAe jd%ZBd;d'ZCe jDEejFGejHd(e jIJd)dd*e jIJd+ddddeKe jL,ejMNd-d+ZOe jIPeOered= version 2.0z2023.12) api_versiontorchcupy jax.numpyjax_enable_x64jax_default_device)1trueallc*i|]}|t|SrO)xp_available_backends).0backends r r`s0)))27;)))rMz!'--array-api-backend' must be in cudaxpzsTest involves masked arrays, object arrays, or other types that are not valid input when `SCIPY_ARRAY_API` is used.reasonc|j|dj}t|dkrtd|t |j|d}g}i}|D]}|jdr$d|d<|jdg|d<nR|jdr8|ds#d|d<|jdg|d<t|jdkr6|jd}| ||j d i||ji||fS) z%A helper for {skip,xfail}_xp_backends _xp_backendsrzmultiple backends: np_onlyT exceptionscpu_onlyrrO) keywordsr.lenr'listnode iter_markerskwargsr&appendupdate)rI skip_or_xfailargs_r backendsromarkerr_s r_backends_kwargs_from_requestrvsy   ;;; < AE 5zzA~~6u667777<,, -K-K-KLLMMGH F66 =  Y ' ' K $F9 #)=#4#4\2#F#FF< ]  z * * K::i(( K%)z"'-}'8'8r'J'J|$ v{  a  k!nG OOG $ $ $ FM 5 5Wfm4 5 5 5 V rMcjd|jvrdSt|d\}}t|||ddS)askip_xp_backends(backend=None, reason=None, np_only=False, cpu_only=False, exceptions=None) Skip a decorated test for the provided backend, or skip a category of backends. See ``skip_or_xfail_backends`` docstring for details. Note that, contrary to ``skip_or_xfail_backends``, the ``backend`` and ``reason`` arguments are optional single strings: this function only skips a single backend at a time. To skip multiple backends, provide multiple decorators. skip_xp_backendsNr)rrrjrvskip_or_xfail_xp_backendsrbrIrtros rrxrxsK!1114WFSSSHfb(F&IIIIIIrMcjd|jvrdSt|d\}}t|||ddS)axfail_xp_backends(backend=None, reason=None, np_only=False, cpu_only=False, exceptions=None) xfail a decorated test for the provided backend, or xfail a category of backends. See ``skip_or_xfail_backends`` docstring for details. Note that, contrary to ``skip_or_xfail_backends``, the ``backend`` and ``reason`` arguments are optional single strings: this function only xfails a single backend at a time. To xfail multiple backends, provide multiple decorators. xfail_xp_backendsNr-ryrzr|s rr~r~sK'"2224WGTTTHfb(F'JJJJJJrMr)cJtt|}|dd}|dd}|dg}|dx}rtd|d|r|rd}|r|s|std|ro|d d }t |t s"t |d krtd |jd kr|j|vr ||dS|rd}|gn|}trtdkr|jdkrd|vr ||n|jdkr2d|vr.d| dj j vr ||nN|jdkrCd|vr?| d D]} d| jvr |||Nt|D]@\} } |j| kr.|| d }|sd| }||?dSdS)an Skip based on the ``skip_xp_backends`` or ``xfail_xp_backends`` marker. See the "Support for the array API standard" docs page for usage examples. Parameters ---------- backends : tuple Backends to skip/xfail, e.g. ``("array_api_strict", "torch")``. These are overriden when ``np_only`` is ``True``, and are not necessary to provide for non-CPU backends when ``cpu_only`` is ``True``. For a custom reason to apply, you should pass a dict ``{'reason': '...'}`` to a keyword matching the name of the backend. reason : str, optional A reason for the skip/xfail in the case of ``np_only=True``. If unprovided, a default reason is used. Note that it is not possible to specify a custom reason with ``cpu_only``. np_only : bool, optional When ``True``, the test is skipped/xfailed for all backends other than the default NumPy backend. There is no need to provide any ``backends`` in this case. To specify a reason, pass a value to ``reason``. Default: ``False``. cpu_only : bool, optional When ``True``, the test is skipped/xfailed on non-CPU devices. There is no need to provide any ``backends`` in this case, but any ``backends`` will also be skipped on the CPU. Default: ``False``. exceptions : list, optional A list of exceptions for use with ``cpu_only`` or ``np_only``. This should be provided when delegation is implemented for some, but not all, non-CPU/non-NumPy backends. skip_or_xfail : str ``'skip'`` to skip, ``'xfail'`` to xfail. rgFrirhreasonsz.provide a single `reason=` kwarg; got reasons=z insteadz<`exceptions` is only valid alongside `cpu_only` or `np_only`rdz#do not run with non-NumPy backends.rz8please provide a singleton `reason` when using `np_only`rQrcNzTno array-agnostic implementation or delegation available for this backend and devicecpurUrTrrVz#do not run with array API backend: )getattrr(r&r' isinstancestrrk__name__rremptydevicetypedevices device_kind enumerate) rbrtrorrrgrirhrrddir_s rr{r{sFFM22MjjE**Gzz*e,,HL"--J**Y'''wVT'TTTUUU8Y8YwYWXXXH&KLL&#&& 53v;;??455 5 ;' ! !bk&C&C M ( ( ( ( 50%-RR:  5|u44{f$$z)A)A V,,,,,''G:,E,E 2 777!M0000 ++ :0M0M!,,..55AAM11% V4444#H-- - -JAw{g%%,,X66ML7LLF V,,,, - -rMz .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)rEmessage)zscipy.signal.cwtzscipy.signal.morletzscipy.signal.morlet2zscipy.signal.rickerzscipy.integrate.simpsonzscipy.interpolate.interp2dzscipy.linalg.kron)rEr) 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)rrEz"The maximum number of subdivisions)rz linalg.rstz stats.rstscipy.signal.normalize)_fixed_default_rngNignoreerror)r)dictRuntimeWarningDeprecationWarningscipyrIntegrationWarning UserWarningPendingDeprecationWarningrqsetscipy._lib._utilrrQrandomseedrGcatch_warningsrfilterwarnings simplefilterWarning) testknown_warningsdivide_by_zeror deprecatedr integration_wuser_wdctlegitrr*s 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.rs, %%%%%% ,,,,,,666666????????000000JJJJJ#!###"#' ' ' T...bj$/// C C0/ C ^^ !" 2&zz/3772&  $$&8:J%KLLL 7=)5 6 69O9O O O+GHH H33!             $$gu%5666  ....        $$fd^4444       $$k39%=>>> *D111 .  L0I0I!0LMMMM      m33%4:o66 $ $ $  &))#3)))%% & & &X:O:T:T:V:VXX joo% & """L{..t5J5Q5Q5S5STTk((I)KK >JJJ" K K K P-P-P-P-l 00GLL$$&& 66 $$ dt% $$  dTt$z566%:>>*D*9;;  !9::: d"OOO^Ob"9II( ~!!##'''##   ,%%%I!$ 8&><2)> $$I 3 7899I   !I"IId"d"szA AAA""A,+A,7ADDD"+EEEE66E>=E>A*G--G54G5 H,,(I