
    Cdu                        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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   s    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   s    Ir   N)__name__
__module____qualname__r   r   r   r   	MyProcessr      s#        	 	 	 	 	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builtinsr*   sysr   )r   monkeypatchr   r   r)   r   r'   r(   s         @@r   test_cpu_count_cgroupsr4      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   )
__future__r   r1   r/   r
   r2   pytestdask.systemr   importorskipr   r   markparametrizer4   r   r   r   <module>r;      s    " " " " " "  				 				 



  ! ! ! ! ! !		X	&	&   $H$H$HII" " JI" " "r   