ó
ßüÚ\c           @   s|   d  d l  Z  d  d l Z d  d l Z d  d l Z d  d l m Z d  d l m Z d  d l m Z d Z	 d   Z
 d   Z d S(   i˙˙˙˙N(   t   Client(   t   cluster(   t   loops   
_worker_info = {}

def dask_setup(worker):
    _worker_info['address'] = worker.address

def get_worker_address():
    return _worker_info['address']
c         C   sÔ   d   } t  j   } zŞ t j j | d  } t | d   } | j t  Wd  QXt d i | g d 6 Q \ } } t	 | d d |  , } | j
 |  d   | D k s˛ t  Wd  QXWd  QXWd  t j |  Xd  S(	   Nc          S   s   d d  l  }  |  j   S(   Ni˙˙˙˙(   t   worker_infot   get_worker_address(   R   (    (    s=   lib/python2.7/site-packages/distributed/tests/test_preload.pyt   check_worker   s    s   worker_info.pyt   wt   worker_kwargst   preloadt   addressR   c         S   s!   i  |  ] } | d  | d   q S(   R	   (    (   t   .0t   worker(    (    s=   lib/python2.7/site-packages/distributed/tests/test_preload.pys
   <dictcomp>'   s   	 (   t   tempfilet   mkdtempt   ost   patht   joint   opent   writet   PRELOAD_TEXTR   R    t   runt   AssertionErrort   shutilt   rmtree(   R   R   t   tmpdirR   t   ft   st   workerst   c(    (    s=   lib/python2.7/site-packages/distributed/tests/test_preload.pyt   test_worker_preload_file   s    	" )c         C   s÷   d   } t  j   } t j j d |  zŞ t j j | d  } t | d   } | j t	  Wd  QXt
 d i d g d 6 Q \ } } t | d d	 |  , } | j |  d
   | D k sĹ t  Wd  QXWd  QXWd  t j j |  t j |  Xd  S(   Nc          S   s   d d  l  }  |  j   S(   Ni˙˙˙˙(   R   R   (   R   (    (    s=   lib/python2.7/site-packages/distributed/tests/test_preload.pyR   .   s    i    s   worker_info.pyR   R   R   R   R	   R   c         S   s!   i  |  ] } | d  | d   q S(   R	   (    (   R
   R   (    (    s=   lib/python2.7/site-packages/distributed/tests/test_preload.pys
   <dictcomp>@   s   	 (   R   R   t   sysR   t   insertR   R   R   R   R   R   R    R   R   t   removeR   R   (   R   R   R   R   R   R   R   R   (    (    s=   lib/python2.7/site-packages/distributed/tests/test_preload.pyt   test_worker_preload_module-   s    	)(   R   R   R   R   t   distributedR    t   distributed.utils_testR   R   R   R   R!   (    (    (    s=   lib/python2.7/site-packages/distributed/tests/test_preload.pyt   <module>   s   	