o
    Eb                     @   s   d dl Z d dlZd dlZd dlZd dlmZ d dlm	Z	 d dl
mZ d dlmZ dd Zdd Zd	d
 Zejddddd ZdS )    N)get_fpu_mode)FPUModeChangeWarning)_pep440c                 C   s(   |  dd |  dd |  dd d S )NZmarkerszslow: 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 platforms)Zaddinivalue_line)Zconfig r   0/usr/lib/python3/dist-packages/scipy/conftest.pypytest_configure   s   r   c                 C   s2   t tjt dkr| |}|S | |}|S )Nz3.6.0)r   parsepytest__version__ZVersionZget_closest_markerZ
get_marker)itemnamemarkr   r   r   	_get_mark   s
   

r   c           	   	   C   s  t | d}|d ur'zttjdd}W n ty   d}Y nw |s'td t | d}|d urCt	dj
dk rCtd	|jd f  t |}|tj z
dd
lm} d}W n tyc   d}Y nw |rz	ttjd }W n ty   Y W d    d S w tdst d }t|| d}z||dd W n  ty   Y W d    d S w W d    d S W d    d S W d    d S 1 sw   Y  d S )NZxslowZSCIPY_XSLOW0Fz@very slow test; set environment variable SCIPY_XSLOW=1 to run itZxfail_on_32bitr      z(Fails on our 32-bit test platform(s): %s)threadpool_limitsTZPYTEST_XDIST_WORKER_COUNTZOMP_NUM_THREADS      Zblas)Zuser_api)r   intosenvironget
ValueErrorr	   skipnpZintpitemsizeZxfailargsnptZsuppress_warningsfilterZ PytestUnraisableExceptionWarningZthreadpoolctlr   	ExceptionKeyErrorgetenv	cpu_countmax)	r   r   vsupr   ZHAS_THREADPOOLCTLZxdist_worker_countZmax_openmp_threadsZthreads_per_workerr   r   r   pytest_runtest_setup   sT   




"r&   functionT)ZscopeZautousec                 c   s<    t  }dV  t  }||krtjd||tdd dS dS )z9
    Check FPU mode was not changed during the test.
    Nz6FPU mode changed from {0:#x} to {1:#x} during the testr   )category
stacklevel)r   warningswarnformatr   )ZrequestZold_modeZnew_moder   r   r   check_fpu_modeN   s   
r-   )r   r	   r*   Znumpyr   Znumpy.testingZtestingr   Zscipy._lib._fpumoder   Zscipy._lib._testutilsr   Z
scipy._libr   r   r   r&   Zfixturer-   r   r   r   r   <module>   s   	0