o
    Û¾ið  ã                   @   sV   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r)e ¡  dS 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 n ty3 } z|  d|jd ¡ W Y d}~n"d}~ww |  	|t t
ƒ ƒ¡ |  	|t tjƒ¡ W |t|< t ¡  dS W |t|< t ¡  dS |t|< t ¡  w )zO
        Tests the NUMBA_NUM_THREADS env variable behaves as expected.
        ÚNUMBA_NUM_THREADSÚ3154zCannot set NUMBA_NUM_THREADSr   N)ÚstrÚgetattrÚenvr   r   Úreload_configÚRuntimeErrorÚassertInÚargsÚassertEqualr   )ÚselfÚkeyÚcurrentÚthreadsÚe© r   úc/home/ubuntu/.local/lib/python3.10/site-packages/numba/tests/npyufunc/test_parallel_env_variable.pyÚtest_num_threads_variable   s&   €þøÿ
z1TestParallelEnvVariable.test_num_threads_variableN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Ú_numba_parallel_test_r   r   r   r   r   r      s    r   Ú__main__)Únumba.np.ufunc.parallelr   Úosr   r
   Ú
numba.corer   ÚunittestÚTestCaser   r   Úmainr   r   r   r   Ú<module>   s    ÿ