
    bi                    X    d dl mZ d dlZd dlZd dlmZ d Zd Zd Z G d d      Z	d	 Z
y)
    )annotationsN)globalmethodc                 
   t        j                  d       t        j                  d      } dgfd}| j                  dd      }|j                         j	                         dk(  sJ d   dk(  sJ t
        j                  j                  |      5  |j                         j	                         dk(  sJ 	 d d d        d   d	k(  sJ |j                         j	                         dk(  sJ d   d	k(  sJ y # 1 sw Y   AxY w)
Nnumpyz
dask.arrayr   c                H    d   dz   d<   t        j                  | |fi |S )Nr      )daskget)dskkeyskwargsvars      R/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/tests/test_context.pymygetztest_with_get.<locals>.myget   s*    Q!AxxT,V,,    
   )   )chunks)	schedulerr   )pytestimportorskiponessumcomputer	   configset)dar   xr   s      @r   test_with_getr   	   s    
 			\	*B#C- 	4 A557??"""q6Q;;	5	) 'uuw B&&&'q6Q;; 557??"""q6Q;;' 's   $C99Dc                      y)Nfoo r"   r   r   r!   r!           r   c                      y)Nbarr"   r"   r   r   r%   r%   $   r#   r   c                  @    e Zd Z ed      d        Z eede      Zy)Foof)keyc                      y)Nr   r"   r"   r   r   r(   zFoo.f)   s    r   g)r)   falseyN)__name__
__module____qualname__r   r(   r!   r%   r+   r"   r   r   r'   r'   (   s)    c  	Sc#.Ar   r'   c                    t               } | j                         dk(  sJ t        j                  j	                  d       5  | j                         dk(  sJ 	 d d d        t        j                  j	                  t
              5  | j                  t
        u sJ | j                         dk(  sJ 	 d d d        | j                  t
        u sJ | j                         dk(  sJ t        j                  j	                  d      5  | j                  t        u sJ | j                         dk(  sJ 	 d d d        y # 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   y xY w)	Nr   c                      y)N   r"   r"   r   r   <lambda>z#test_globalmethod.<locals>.<lambda>5   s    r   )r(   r2   r!   F)r+   r%   )r'   r(   r	   r   r   r!   r+   r%   )r   s    r   test_globalmethodr4   0   s   A335A::	9	% ssuzz 
3	 ssczzssu~~ 33#::335E>>	5	! ssczzssu~~    s$   D4*E  *E4D= E	E)
__future__r   r   r	   dask.contextr   r   r!   r%   r'   r4   r"   r   r   <module>r7      s/    "   %./ /r   