o
    Nrf                     @  sZ   d dl mZ d dlZd dlZd dlmZ dd Zdd Zdd	 ZG d
d dZ	dd Z
dS )    )annotationsN)globalmethodc                    s   t d t d} dg  fdd}| jddd}|  dks$J  d dks,J tjj|d	 |  dks>J W d    n1 sHw   Y   d d
ksUJ |  dks_J  d d
ksgJ d S )Nnumpyz
dask.arrayr   c                   s$    d d  d< t j| |fi |S )Nr      )daskget)Zdskkeyskwargsvar `/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/dask/tests/test_context.pymyget   s   ztest_with_get.<locals>.myget
   )   )chunks)Z	schedulerr   )pytestZimportorskipZonessumZcomputer   configset)dar   xr   r
   r   test_with_get	   s   

r   c                   C     dS )Nfoor   r   r   r   r   r          r   c                   C  r   )Nbarr   r   r   r   r   r   $   r   r   c                   @  s,   e Zd Zedddd ZeededZdS )Foof)keyc                   C  r   )Nr   r   r   r   r   r   r   )   s   zFoo.fg)r   ZfalseyN)__name__
__module____qualname__r   r   r   r   r    r   r   r   r   r   (   s    
r   c                  C  s  t  } |  dksJ tjjdd d |  dksJ W d    n1 s'w   Y  tjjtd | jtu s;J |  dksCJ W d    n1 sMw   Y  | jtu sYJ |  dksaJ tjjdd | jtu spJ |  d	ksxJ W d    d S 1 sw   Y  d S )
Nr   c                   S  r   )N   r   r   r   r   r   <lambda>5   s    z#test_globalmethod.<locals>.<lambda>)r   r$   r   F)r    r   )r   r   r   r   r   r   r    r   )r   r   r   r   test_globalmethod0   s   "r&   )
__future__r   r   r   Zdask.contextr   r   r   r   r   r&   r   r   r   r   <module>   s    