
    `i                         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                     dS dS )    )get_thread_count)environ)configNc                       e Zd ZdZdZd ZdS )TestParallelEnvVariablezg
    Tests environment variables related to the underlying "parallel"
    functions for npyufuncs.
    Fc                 L   d}t          t          t          |t          j                            }d}|t          |<   	 t          j                     |                     |t          t                                           |                     |t          t          j                             n8# t          $ r+}| 	                    d|j
        d                    Y d}~nd}~ww xY w|t          |<   t          j                     dS # |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        /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/numba/tests/npyufunc/test_parallel_env_variable.pytest_num_threads_variablez1TestParallelEnvVariable.test_num_threads_variable   s"    "gc3(@AABBC	# """
 Wc*:*<*<&=&=>>>Wc&*B&C&CDDDD  	E 	E 	EMM8!&)DDDDDDDD	E CH """"" CH """"s0   B- AD -
C"7!CD C""D D#N)__name__
__module____qualname____doc___numba_parallel_test_r        r   r   r      s5         
 "# # # # #r    r   __main__)numba.np.ufunc.parallelr   osr   r   
numba.corer   unittestTestCaser   r   mainr   r    r   <module>r(      s    4 4 4 4 4 4             # # # # #h/ # # #< zHMOOOOO r    