ó
¦–Õ\c           @   s»   d  d l  m Z d  d l m Z d  d l m Z d  d l m Z d  d l m	 Z	 d  d l
 m Z d  d l Z e j d ƒ Z g  Z d	 „  Z d
 „  Z d „  Z d „  Z d „  Z d „  Z d S(   iÿÿÿÿ(   t   Callback(   t   Cache(   t   get_sync(   t   get(   t   add(   t   sleepNt   cacheyc         C   s   t  j |  ƒ |  d S(   Ni   (   t   flagt   append(   t   x(    (    s4   lib/python2.7/site-packages/dask/tests/test_cache.pyt   inc   s    c          C   s6  t  j d ƒ }  t |  ƒ } | - t i t d f d 6d ƒ d k sJ t ‚ Wd  QXt d g k se t ‚ |  j d d k s~ t ‚ | j sŽ t ‚ | j sž t ‚ x t r´ t j	 ƒ  q¡ Wi t d f d 6t d f d 6t
 d d f d 6} |   t | d ƒ d k st ‚ Wd  QXt d g k s"t ‚ t j s2t ‚ d  S(   Ni'  i   R	   i   t   yt   zi   (   R   R   R   R
   t   AssertionErrorR   t   datat
   starttimest	   durationst   popR   R    t   active(   t   ct   cct   dsk(    (    s4   lib/python2.7/site-packages/dask/tests/test_cache.pyt
   test_cache   s    .	0!c          C   sa   t  d d d ƒ}  t |  j t j  ƒ s- t ‚ |  j j d k sE t ‚ |  j j d k s] t ‚ d  S(   Ni'  t   limiti   (   R   t
   isinstancet   cacheR   R   t   available_bytesR   (   R   (    (    s4   lib/python2.7/site-packages/dask/tests/test_cache.pyt   test_cache_with_number,   s    c          C   s´   t  d ƒ }  t j d ƒ } d d l m } m } | j | d ƒ d d ƒ} | j | d ƒ d d ƒ} |  A | j ƒ  d k j ƒ  sŒ t	 ‚ | j ƒ  d k j ƒ  sª t	 ‚ Wd  QXd  S(	   Ni'  s
   dask.arrayiÿÿÿÿ(   t   onest   zerosi   t   chunksi
   i    (
   R   t   pytestt   importorskipt   numpyR   R   t
   from_arrayt   computet   allR   (   R   t   daR   R   R   t   o(    (    s4   lib/python2.7/site-packages/dask/tests/test_cache.pyt   test_cache_correctness3   s    c         G   s   t  |  ƒ d g | S(   Ni    (   R   (   t   durationt   sizet   args(    (    s4   lib/python2.7/site-packages/dask/tests/test_cache.pyt   f?   s    
c          C   s   i t  d d f d 6t  d d d f d 6}  t d ƒ } |  t |  d ƒ Wd  QX| j j j d | j j j d k  s{ t ‚ d  S(   Ng{®Gáz„?i
   R	   gíµ ÷Æ°>i   R   i'  (   R+   R   R   R   t   scorert   costR   (   R   R   (    (    s4   lib/python2.7/site-packages/dask/tests/test_cache.pyt   test_prefer_cheap_dependentD   s
    )(   t   dask.callbacksR    t
   dask.cacheR   t
   dask.localR   t   dask.threadedR   t   operatorR   t   timeR   R   R    R   R   R
   R   R   R'   R+   R.   (    (    (    s4   lib/python2.7/site-packages/dask/tests/test_cache.pyt   <module>   s   					