B
    T\#                 @   sT   d dl mZ d dlmZ d dlZdd Zdd Zdd ZG d	d
 d
e	Z
dd ZdS )    )globalmethodNc           	      s   dg  fdd} t jddd}|  dks4t d dksDttjj| d |  dkshtW d Q R X  d dkst|  dkst d dkstd S )	Nr   c                s     d d  d< t j| |f|S )Nr      )daskget)Zdskkeyskwargs)var 6lib/python3.7/site-packages/dask/tests/test_context.pymyget	   s    ztest_with_get.<locals>.myget
   )   )Zchunks)Z	schedulerr   )daZonessumZcomputeAssertionErrorr   configset)r   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 d Q R X tjjtd$ | jtksbt|  dksrtW d Q R X | jtkst|  dksttjjdd$ | jtkst|  d	kstW d Q R X d S )
Nr   c               S   s   dS )N   r	   r	   r	   r	   r
   <lambda>0   s    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    )Zdask.contextr   Z
dask.arrayZarrayr   r   r   r   r   objectr   r    r	   r	   r	   r
   <module>   s   