U
    »mœd–  ã                   @   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   sV   |   dd¡ |   dd¡ |   dd¡ zdd l}W n  tk
rP   |   dd¡ Y nX d S )NÚ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 platformsr   z.timeout: mark a test for a non-default timeout)Zaddinivalue_lineÚpytest_timeoutÚ	Exception)Úconfigr   © r	   úG/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/scipy/conftest.pyÚpytest_configure   s     ÿÿÿ ÿr   c                 C   s0   t  tj¡t  d¡kr"|  |¡}n
|  |¡}|S )Nz3.6.0)r   ÚparseÚpytestÚ__version__ÚVersionZget_closest_markerZ
get_marker)ÚitemÚnameÚmarkr	   r	   r
   Ú	_get_mark   s    
r   c           	   	   C   sp  t | dƒ}|d k	rPzttj dd¡ƒ}W n tk
r@   d}Y nX |sPt d¡ t | dƒ}|d k	rˆt 	d¡j
dk rˆt d	|jd f ¡ t ¡ �Ö}| tj¡ zdd
lm} d}W n tk
rÊ   d}Y nX |�rbzttjd ƒ}W n" tk
�r   Y W 5 Q R £ d S X t d¡�sbt ¡ d }t|| dƒ}z||dd� W n" tk
�r`   Y W 5 Q R £ d S X W 5 Q R X 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   r   Ú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#   s<    





r*   ÚfunctionT)ÚscopeZautousec                 c   s6   t ƒ }dV  t ƒ }||kr2tjd ||¡tdd� 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   )ÚrequestZold_modeZnew_moder	   r	   r
   Úcheck_fpu_modeS   s     ÿ þr3   )r   r   r/   Únumpyr   Znumpy.testingÚtestingr"   Zscipy._lib._fpumoder   Zscipy._lib._testutilsr   Z
scipy._libr   r   r   r*   Zfixturer3   r	   r	   r	   r
   Ú<module>   s   0