
    >ie
                    J   d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ  ej	        d          Z
d Zej                            dg d          d             Zej                            d	g d
          ej                            dddg          d                         ZdS )    )annotationsN)	cpu_countpsutilc                     t                      } t          | t                    sJ | t          j                     k    sJ | dk    sJ d S )N   )r   
isinstanceintos)counts    6lib/python3.11/site-packages/dask/tests/test_system.pytest_cpu_countr      sI    KKEeS!!!!!BLNN""""A::::::    dirname)zcpuacct,cpuzcpu,cpuacctNc                   d }|                     t          d|            G d d          }|                     t          d|           | rzd| z  t          j        d          d| z  t          j        d	          it
          j        fd
}|                     t
          d|           |                     t          dd           t                      }| r
|dk    sJ d S |dk    sJ d S )Nc                     dS N    r   r   r   mycpu_countz+test_cpu_count_cgroups.<locals>.mycpu_count       sr   r   c                      e Zd Zd ZdS ))test_cpu_count_cgroups.<locals>.MyProcessc                    g S Nr   selfs    r   cpu_affinityz6test_cpu_count_cgroups.<locals>.MyProcess.cpu_affinity       Ir   N__name__
__module____qualname__r   r   r   r   	MyProcessr      #        	 	 	 	 	r   r#   Processz"/sys/fs/cgroup/%s/cpu.cfs_quota_us2005z#/sys/fs/cgroup/%s/cpu.cfs_period_us10c                N    | v r                     |           S  | g|R i |S r   getpathargskwargsbuiltin_openpathss      r   myopenz&test_cpu_count_cgroups.<locals>.myopen,   s<    u}}yy&<6t666v666r   openplatformlinux   r   )	setattrr
   r   ioStringIObuiltinsr2   sysr   )r   monkeypatchr   r#   r1   r   r/   r0   s         @@r   test_cpu_count_cgroupsr<      s2      K555       
 	9555 607:BK<O<O1G;R[=N=N
  }	7 	7 	7 	7 	7 	7
 	Hff555CW555KKE ||||||||||||r   
group_name)/z/user.slicez/user.slice/more.slicequotamaxr&   c                   d }|                     t          d|            G d d          }|                     t          d|           |                    d          s| d}dt	          j        d|z            d	|z  t	          j        d
| z            it          j        fd}|                     t          d|           |                     t          dd           t                      }| dk    r
|dk    sJ d S |dk    sJ d S )Nc                     dS r   r   r   r   r   r   z.test_cpu_count_cgroups_v2.<locals>.mycpu_count?   r   r   r   c                      e Zd Zd ZdS ),test_cpu_count_cgroups_v2.<locals>.MyProcessc                    g S r   r   r   s    r   r   z9test_cpu_count_cgroups_v2.<locals>.MyProcess.cpu_affinityF   r   r   Nr   r   r   r   r#   rD   E   r$   r   r#   r%   r>   z/proc/self/cgroupz0::%sz/sys/fs/cgroup%scpu.maxz%s 10c                N    | v r                     |           S  | g|R i |S r   r)   r+   s      r   r1   z)test_cpu_count_cgroups_v2.<locals>.myopenU   s<    5==99T??"|D242226222r   r2   r3   r4   r@   r   r5   )
r6   r
   r   endswithr7   r8   r9   r2   r:   r   )	r?   r=   r;   r   r#   r1   r   r/   r0   s	          @@r   test_cpu_count_cgroups_v2rH   <   sW      K555       
 	9555s## &"%%%
 	R[:)=>>!J.GeO0L0LE =L3 3 3 3 3 3
 &&111Z111KKE~~|||||| ||||||r   )
__future__r   r9   r7   r
   r:   pytestdask.systemr   importorskipr   r   markparametrizer<   rH   r   r   r   <module>rO      s   " " " " " "  				 				 



  ! ! ! ! ! !		X	&	&   $H$H$HII" " JI"J 'U'U'UVV5&/22$ $ 32 WV$ $ $r   