o 4BiJ#@s<ddlZddlZddlZddlZddlZddlmZddl Z ddl Z ddl m Z ddl mZddlmZddlmZmZddZdd Zd d Ze jd d dddZdeiZereeerz ddlZedeiWn eytYnwzddlZedeie eWn eyYnwz ddl!Z!ede!iWn eyYnwe"dvre#eZ$de$vrnz dde$DZWne%yde&Z'e(e'wdevrdZe j)*de+Z,e jddZ-e j./ej01e2de j3j4ddd d e j3j4d!dd dd e5e j6d"ej78d#d!Z9e j3:e9dS)$N) get_fpu_mode)FPUModeChangeWarning)_pep440)SCIPY_ARRAY_API SCIPY_DEVICEcCs`|dd|dd|ddzddl}Wnty'|ddYnw|dddS)Nmarkerszslow: 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 platformsrz.timeout: mark a test for a non-default timeoutzskip_if_array_api(*backends, reasons=None, np_only=False, cpu_only=False): mark the desired skip configuration for the `skip_if_array_api` fixture.)addinivalue_linepytest_timeout Exception)configr r N/var/www/html/Trade-python/venv/lib/python3.10/site-packages/scipy/conftest.pypytest_configures&  rcCs2ttjtdkr||}|S||}|S)Nz3.6.0)rparsepytest __version__Versionget_closest_marker get_marker)itemnamemarkr r r _get_mark#s   rc Cst|d}|dur'z ttjdd}Wn tyd}Ynw|s'tdt|d}|durCt dj dkrCt d |j dt |}|tjz dd lm}d }Wn tycd}Ynw|rz ttjd }WntyYWddSwtd std}t||d}z||ddWn tyYWddSwWddSWddSWddS1swYdS)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)rintosenvironget ValueErrorrskipnpintpitemsizexfailargsnptsuppress_warningsfilter PytestUnraisableExceptionWarning threadpoolctlrr KeyErrorgetenv cpu_countmax) rrvsuprHAS_THREADPOOLCTLxdist_worker_countmax_openmp_threadsthreads_per_workerr r r pytest_runtest_setup+sT            "r?functionT)scopeautouseccsFt}dVt}||kr!tjd|dd|ddtdddSdS)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_moder r r check_fpu_mode\s rJnumpyarray_api_strictpytorchcupy)1trueallcCsi|]}|t|qSr )xp_available_backends).0backendr r r srUz!'--array-api-backend' must be in cudaxpc Cs"d|jvrdS|jdj}|jdj}|dd}|dd}|r9|ddg}|d}|jd kr7tj|d dS|rcd }trctd krc|jd krOtj|d n|jdkrcd t dj j vrctj|d |dur|dd}t |D]\}} |j| kr|sd| }n||}tj|d qqdSdS)a Skip based on the ``skip_if_array_api`` marker. Parameters ---------- *backends : tuple Backends to skip, 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``. reasons : list, optional A list of reasons for each skip. When ``np_only`` is ``True``, this should be a singleton list. Otherwise, this should be a list of reasons, one for each corresponding backend in ``backends``. If unprovided, default reasons are used. Note that it is not possible to specify a custom reason with ``cpu_only``. Default: ``None``. np_only : bool, optional When ``True``, the test is skipped 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 singleton list to ``reasons``. Default: ``False``. cpu_only : bool, optional When ``True``, the test is skipped 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``. skip_if_array_apiNnp_onlyFcpu_onlyreasonsz#do not run with non-NumPy backends.rrK)reasonz4do not run with `SCIPY_ARRAY_API` set and not on CPUcpurNtorchz#do not run with array API backend: )keywordsr/kwargsr(__name__rr*rrr^emptydevicetype enumerate) rWrGbackendsr`rYrZr[r\irTr r r rXs>               rXz .hypothesisnondeterministic)rdeadline print_blob deterministic)rrirjdatabase derandomizesuppress_health_checkSCIPY_HYPOTHESIS_PROFILE);jsonr&rEtempfilerKr+ numpy.testingtestingr0r hypothesisscipy._lib._fpumoderscipy._lib._testutilsr scipy._librscipy._lib._array_apirrrrr?fixturerJrR isinstancestrrLupdate ImportErrorr^set_default_devicerNlowerloadsSCIPY_ARRAY_API_r5keysmsgr)r parametrizevaluesarray_api_compatiblerX configurationset_hypothesis_home_dirpathjoin gettempdirsettingsregister_profilelist HealthCheckr'r(ro load_profiler r r r s     1        =