
    Cdq                    `    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
dS )
    )annotationsN)globalmethodc                 <   t          j        d          } dgfd}|                     dd          }|                                                                dk    sJ d         dk    sJ t
          j                            |          5  |                                                                dk    sJ 	 d d d            n# 1 swxY w Y   d         dk    sJ |                                                                dk    sJ d         dk    sJ d S )	Nz
dask.arrayr   c                F    d         dz   d<   t          j        | |fi |S )Nr      )daskget)dskkeyskwargsvars      7lib/python3.11/site-packages/dask/tests/test_context.pymygetztest_with_get.<locals>.myget   s.    Q!AxT,,V,,,    
   )   )chunks)	schedulerr   )pytestimportorskiponessumcomputer   configset)dar   xr   s      @r   test_with_getr   	   sZ   		\	*	*B#C- - - - - 	4  A5577??""""q6Q;;;;	5	)	) ' 'uuww  B&&&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' 'q6Q;;;; 5577??""""q6Q;;;;;;s   -C		CCc                     dS )Nfoo r!   r   r   r    r           5r   c                     dS )Nbarr!   r!   r   r   r$   r$   #   r"   r   c                  T    e Zd Z ed          d             Z eede          ZdS )Foof)keyc                     dS )Nr   r!   r!   r   r   r'   zFoo.f(   s    qr   g)r(   falseyN)__name__
__module____qualname__r   r'   r    r$   r*   r!   r   r   r&   r&   '   sM        \c   	Sc#...AAAr   r&   c                    t                      } |                                 dk    sJ t          j                            d           5  |                                 dk    sJ 	 d d d            n# 1 swxY w Y   t          j                            t
                    5  | j        t
          u sJ |                                 dk    sJ 	 d d d            n# 1 swxY w Y   | j        t
          u sJ |                                 dk    sJ t          j                            d          5  | j        t          u sJ |                                 dk    sJ 	 d d d            d S # 1 swxY w Y   d S )	Nr   c                     dS )N   r!   r!   r   r   <lambda>z#test_globalmethod.<locals>.<lambda>4   s    1 r   )r'   r1   r    F)r*   r$   )r&   r'   r   r   r   r    r*   r$   )r   s    r   test_globalmethodr3   /   s   A3355A::::	99	%	%  ssuuzzzzz               
3		  sczzzzssuu~~~~~               3#::::3355E>>>>	5	!	!  sczzzzssuu~~~~~                 s6   
A22A69A6"+CC!C/+E((E,/E,)
__future__r   r   r   dask.contextr   r   r    r$   r&   r3   r!   r   r   <module>r6      s    " " " " " "   % % % % % %  ,    / / / / / / / /    r   