U
    /eM                     @   sN   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 )    N)globalmethodc               	      s   t d} dg  fdd}| jddd}|  dks>t d dksNttjj|d |  dksrtW 5 Q R X  d d	kst|  dkst d d	kstd S )
Nz
dask.arrayr   c                    s     d d  d< t j| |f|S )Nr      )daskget)Zdskkeyskwargsvar ;/tmp/pip-unpacked-wheel-dbjnr7gq/dask/tests/test_context.pymyget   s    ztest_with_get.<locals>.myget
   )   )chunks)Z	schedulerr   )	pytestZimportorskipZonessumZcomputeAssertionErrorr   configset)dar   xr
   r   r   test_with_get   s    
r   c                   C   s   dS )Nfoor
   r
   r
   r
   r   r      s    r   c                   C   s   dS )Nbarr
   r
   r
   r
   r   r   !   s    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   s   dS )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ttjjdd d |  dks:tW 5 Q R X tjjtd$ | jtksbt|  dksrtW 5 Q R X | jtkst|  dksttjjdd$ | jtkst|  d	kstW 5 Q R X d S )
Nr   c                   S   s   dS )N   r
   r
   r
   r
   r   <lambda>2       z#test_globalmethod.<locals>.<lambda>)r   r!   r   F)r   r   )	r   r   r   r   r   r   r   r   r   )r   r
   r
   r   test_globalmethod-   s    r$   )	r   r   Zdask.contextr   r   r   r   r   r$   r
   r
   r
   r   <module>   s   