
    
3jm                         S SK Jr  S SKr\R                  " SS9S 5       r\R                  S 5       rS rS rS	 rS
 r	S r
g)    )
nvidia_smiNmodule)scopec                 ,    [         R                  " 5       $ )N)r   getInstance)requests    U/home/wildlama/miniconda3/lib/python3.13/site-packages/pynvml_utils/tests/test_smi.pysmir
      s    !!##    c                 p    U R                  S5      S   nUS:  d   e[        S[        U5      -   S-   SS9  U$ )Ncountr   [z GPUs] )end)DeviceQueryprintstr)r
   results     r	   ngpusr      s>    __W%g.FA::	#F
h
&C0Mr   c                 r    UR                  S5      n[        U 5       H  nUS   U   S   S   S:  a  M   e   g )Nzmemory.freegpufb_memory_usagefreer   )r   range)r   r
   r   is       r	   test_query_memoryr      s@    __]+F5\e}Q 126:a??? r   c                 r    [        U 5       H(  nUR                  S5      S   U   S   S   nUS:  a  M(   e   g )Nzutilization.gpur   utilizationgpu_utilr   r   r   r   r
   r   r   s       r	   test_gpu_utilizationr"      s?    5\!23E:1=mLZX{{ r   c                 r    [        U 5       H(  nUR                  S5      S   U   S   S   nUS:  a  M(   e   g )Nzutilization.memoryr   r   memory_utilr   r    r!   s       r	   test_memory_utilizationr%   &   sD    5\!56u=a@O
 {{	 r   c                 r    [        U 5       H(  nUR                  S5      S   U   S   nUS   S:X  a  M(   e   g )Npstater   performance_stater   Pr    r!   s       r	   test_pstater*   /   s@    5\*51!45HIayC r   c                     [        U 5       HJ  nUR                  S5      S   U   S   S   nUR                  S5      S   U   S   S   nUS:  a  X4:  a  MJ   e   g )Nztemperature.gpur   temperaturegpu_tempgpu_temp_max_thresholdr   r    )r   r
   r   tempmax_temps        r	   test_temperaturer1   6   sl    5\01%8;MJ:V??#45e<Q?N$
 qt// r   )pynvml_utilsr   pytestfixturer
   r   r   r"   r%   r*   r1    r   r	   <module>r6      sZ    #  h$  $  @ 0r   