U
    ?h                     @   sR   d dl mZ d dlmZ d dlmZ d dlZG dd dejZ	e
dkrNe  dS )    )get_thread_count)environ)configNc                   @   s   e Zd ZdZdZdd ZdS )TestParallelEnvVariablezg
    Tests environment variables related to the underlying "parallel"
    functions for npyufuncs.
    Fc              
   C   s   d}t tt|tj}d}|t|< zjzt  W n4 tk
rd } z| d|jd  W 5 d}~X Y n&X | 	|t t
  | 	|t tj W 5 |t|< t  X dS )zO
        Tests the NUMBA_NUM_THREADS env variable behaves as expected.
        NUMBA_NUM_THREADSZ3154zCannot set NUMBA_NUM_THREADSr   N)strgetattrenvr   r   Zreload_configRuntimeErrorZassertInargsZassertEqualr   )selfkeycurrentthreadse r   a/var/www/html/venv/lib/python3.8/site-packages/numba/tests/npyufunc/test_parallel_env_variable.pytest_num_threads_variable   s    $z1TestParallelEnvVariable.test_num_threads_variableN)__name__
__module____qualname____doc__Z_numba_parallel_test_r   r   r   r   r   r      s   r   __main__)Znumba.np.ufunc.parallelr   osr   r	   Z
numba.corer   ZunittestZTestCaser   r   mainr   r   r   r   <module>   s   