
    wg                         d dl mZ d dlmZ d dlmZ d dlZ G d dej                        Z	e
dk(  r ej                          yy)    )get_thread_count)environ)configNc                       e Zd ZdZdZd Zy)TestParallelEnvVariablezg
    Tests environment variables related to the underlying "parallel"
    functions for npyufuncs.
    Fc                    d}t        t        t        |t        j                              }d}|t        |<   	 t        j
                          | j                  |t        t                            | j                  |t        t        j                               |t        |<   t        j
                          y# t        $ r)}| j                  d|j                  d          Y d}~Kd}~ww xY w# |t        |<   t        j
                          w xY 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assertEqualr   RuntimeErrorassertInargs)selfkeycurrentthreadses        t/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/numba/tests/npyufunc/test_parallel_env_variable.pytest_num_threads_variablez1TestParallelEnvVariable.test_num_threads_variable   s     "gc3(@(@ABC	#  "
 Wc*:*<&=>Wc&*B*B&CD CH  "  	EMM8!&&)DD	E CH  "s0   B5 AC* 5	C'>C"C* "C''C* *D	N)__name__
__module____qualname____doc___numba_parallel_test_r        r   r   r      s    
 "#r    r   __main__)numba.np.ufunc.parallelr   osr   r   
numba.corer   unittestTestCaser   r   mainr   r    r   <module>r(      s=    4   #h// #< zHMMO r    