ó
¦–Õ\c           @   s5  d  d l  Z  d  d l Z d  d l m Z m 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 Z d  d l m Z d  d l m Z m Z m Z m Z m Z m Z m Z m Z m Z m Z m Z m Z m Z d  d l m Z d  d l m Z m  Z  m! Z! d  d	 l" m# Z# m$ Z$ d  d
 l% m& Z& m' Z' m( Z( d  d l) m* Z* d „  Z+ e j, d ƒ Z- e+ d ƒ Z. e+ d ƒ Z/ e+ d ƒ Z0 e+ d ƒ Z1 e+ d ƒ Z2 e+ d ƒ Z3 d d „ Z4 d d d „ Z5 d „  Z6 d „  Z7 d „  Z8 e j9 j: d ƒ d „  ƒ Z; e j9 j: d ƒ d „  ƒ Z< e j9 j: d ƒ d „  ƒ Z= e j9 j: d ƒ d „  ƒ Z> e j9 j: d ƒ d  „  ƒ Z? e j9 j: d ƒ d! „  ƒ Z@ e j9 j: d ƒ d" „  ƒ ZA e j9 j: d ƒ d# „  ƒ ZB e j9 j: d ƒ d$ „  ƒ ZC d% „  ZD d& „  ZE e j9 j: d' ƒ d( „  ƒ ZF e j9 j: d' ƒ d) „  ƒ ZG e j9 j: d' ƒ d* „  ƒ ZH e j9 j: d' ƒ d+ „  ƒ ZI d, „  ZJ d- „  ZK d. „  ZL e j9 j: d ƒ d/ „  ƒ ZM d0 „  ZN d1 „  ZO d2 „  ZP e j9 j: d ƒ d3 „  ƒ ZQ e j9 jR d4 d eS d5 d5 d6 d7 d7 g  ds i  d eU eV g ƒ d8 „  ƒ ZW e j9 j: d ƒ d9 „  ƒ ZX e j9 j: d: ƒ e j9 jR d; dt ƒ dC „  ƒ ƒ ZY dD „  ZZ y+ d  d l[ Z[ e[ j\ dE d5 eV f g ƒ Z] Wn e^ k
 r¤d Z[ n XdF „  Z_ dG e f dH „  ƒ  YZ` dI „  Za e j9 j: dJ ƒ dK „  ƒ Zb e j9 j: dL ƒ dM „  ƒ Zc e j9 j: dL ƒ dN „  ƒ Zd e j9 j: dO ƒ dP „  ƒ Ze e j9 j: dQ ƒ dR „  ƒ Zf e j9 j: dO ƒ dS „  ƒ Zg e j9 j: dO ƒ dT „  ƒ Zh e j9 j: dU ƒ dV „  ƒ Zi e j9 j: dL ƒ dW „  ƒ Zj dX „  Zk e j9 j: dL ƒ e j9 j: e jl j dY dZ ƒd[ „  ƒ ƒ Zm e j9 j: e jl j dY dZ ƒd\ „  ƒ Zn e j9 j: dL ƒ e j9 j: e jl j dY dZ ƒd] „  ƒ ƒ Zo d^ „  Zp d_ „  Zq d` „  Zr da „  Zs db „  Zt dc „  Zu dd „  Zv de „  Zw df „  Zx e j9 j: dU ƒ dg „  ƒ Zy dh „  Zz di „  Z{ dj „  Z| dk „  Z} dl „  Z~ dm „  Z dn „  Z€ e j9 jR do dp dq g ƒ dr „  ƒ Z d S(u   iÿÿÿÿN(   t   addt   mul(   t   OrderedDict(   t   merge(   t   delayed(   t   computet   tokenizet   normalize_tokent   normalize_functiont	   visualizet   persistt   function_cachet   is_dask_collectiont   DaskMethodsMixint   optimizet   unpack_collectionst   named_schedulerst   get_scheduler(   t   Delayed(   t   tmpdirt   tmpfilet   ignoring(   t   inct   dec(   t   longt   unicodet   PY2(   t   Profilerc         C   s$   t  t ƒ  t j |  ƒ SWd  QXd  S(   N(   R   t   BaseExceptiont   pytestt   importorskipt   None(   t   path(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   import_or_none   s    t   toolzs
   dask.arrays   dask.bags   dask.dataframet   numpys   scipy.sparset   pandasi   c         C   s   d  S(   N(    (   t   at   bt   c(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   f1)   s    i   c         C   s   d  S(   N(    (   R%   R&   R'   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   f2-   s    c         C   s   d  S(   N(    (   R%   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   f31   s    c           C   s#  t  t ƒ s t ‚ t  d „  ƒ s' t ‚ t  t j t d d ƒƒ t  t j t d d ƒƒ k sc t ‚ t  t j t d d ƒƒ t  t j t d d ƒƒ k sŸ t ‚ t  t j t d d ƒƒ t  t j t d d ƒƒ k sÛ t ‚ t  t j t t ƒ ƒ t  t j t t ƒ ƒ k st ‚ t  t j t t ƒ ƒ t  t j t t ƒ ƒ k sGt ‚ t  t j t ƒ ƒ t  t j t ƒ ƒ k swt ‚ t  t j t ƒ ƒ t  t j t ƒ ƒ k s§t ‚ t  t j t d d ƒƒ t  t j t d d ƒƒ k sãt ‚ t  t j t d d ƒƒ t  t j t d d ƒƒ k st ‚ d  S(   Nc         S   s   |  S(   N(    (   R%   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   <lambda>9   t    R&   i   i   i   (	   R   R)   t   AssertionErrort   tzt   partialR(   t   composeR*   t   curry(    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_normalize_function5   s$    $$$!!00$c          C   s+   d }  t  t |  ƒ t t f ƒ s' t ‚ d  S(   Ni   i   i   (   i   i   i   (   t
   isinstanceR   t   strt   bytesR-   (   R%   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenizeR   s    s   not npc           C   sL   t  t j j d ƒ j d ƒ ƒ t  t j j d ƒ j d ƒ ƒ k sH t ‚ d  S(   NiÒ  iè  (   R   t   npt   randomt   RandomStatet   random_sampleR-   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt.   test_tokenize_numpy_array_consistent_on_valuesW   s    c           C   s&   t  t j j d ƒ j d d ƒ ƒ d  S(   Ni   t   dtypet   i2(   R   R7   R8   t   astype(    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt/   test_tokenize_numpy_array_supports_uneven_sizes]   s    c           C   s'   t  t j j d ƒ d  d  d … ƒ d  S(   Ni   i   (   R   R7   R8   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt'   test_tokenize_discontiguous_numpy_arrayb   s    c           C   s    t  t j d g d d ƒƒ d  S(   Ns   2000-01-01T12:00:00R<   s   M8[ns](   R   R7   t   array(    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_numpy_datetimeg   s    c           C   s–   t  t j d d d ƒƒ t  t j d d d ƒƒ k s< t ‚ t  t j d g d d d g ƒd
 ƒ t  t j d g d d d g ƒd
 ƒ k s’ t ‚ d  S(   Ng      ð?R<   t   f8i   i   R%   t   i4R&   t   i8i    (   i   i   (   R%   RD   (   R&   RE   (   i   i   (   R%   RD   (   R&   RE   (   R   R7   RA   R-   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_numpy_scalarl   s    <%c           C   sQ  t  t j d d d g d t ƒƒ t  t j d d d g d t ƒƒ k sN t ‚ t  t j d d  d g d t ƒƒ t  t j d d  d g d t ƒƒ k sœ t ‚ t  t j d	 d
 d g d t ƒƒ t  t j d d d g d t ƒƒ k sê t ‚ t rMt  t j t d d d ƒg d t ƒƒ t  t j t d d d ƒg d t ƒƒ k sMt ‚ n  d  S(   NR%   t   aat   aaaR<   i   s   Rebeca AlÃ³nt   encodings   utf-8(   i   R%   (   i   N(   i   RH   (   i   R%   (   i   N(   i   RH   (   R   R7   RA   t   objectR-   R   R   R   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt)   test_tokenize_numpy_array_on_object_dtypes   s    !-!-!-'c    
      C   s  t  d ƒ A }  t j d ƒ } t j |  | ƒ t t j |  d d ƒƒ } Wd  QXt  d ƒ A }  t j d ƒ } t j |  | ƒ t t j |  d d ƒƒ } Wd  QX| | k s° t ‚ t  d ƒ R}  t j j d d ƒ } t j |  | ƒ t j |  d d ƒ} t j |  d d ƒ} t | d d  d  … f ƒ } t | d d  d  … f ƒ } t | d d	 … d  d  … f ƒ } t | d  d  … d f ƒ }	 t	 t
 | | | |	 g ƒ ƒ d
 k s®t ‚ t | ƒ t | ƒ k sÌt ‚ t | d d  d  … f ƒ t | d d  d  … f ƒ k s
t ‚ Wd  QXd  S(   Ns   .npyi   t	   mmap_modet   rt   sizei
   i    i   i   i   (   i
   i
   (   R   R7   t   aranget   saveR   t   loadR-   R8   t   normalt   lent   set(
   t   fnt   xt   yt   zt   mmt   mm2R%   R&   R'   t   d(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_numpy_memmap€   s*    !!"*c          C   sš   t  d ƒ ˆ }  t  d ƒ s } t j d ƒ } t j |  | ƒ t j | | ƒ t j |  d d ƒ} | | } t | ƒ t | ƒ k sŠ t ‚ Wd  QXWd  QXd  S(   Ns   .npyi   RL   RM   (   R   R7   RO   RP   RQ   R   R-   (   t   fn1t   fn2RV   R%   R&   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt&   test_tokenize_numpy_memmap_no_filenameœ   s     
c          C   sp   t  t j ƒ d k s t ‚ t  t j ƒ d k s6 t ‚ t j d „  d d ƒ }  t  |  ƒ t  |  ƒ k sl t ‚ d  S(   Nt    02106e2c67daf452fb480d264e0dac21t    c99e52e912e4379882a9a4b387957a0bc         S   s   |  d S(   Ni   (    (   RV   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR+   °   R,   i   (   R   R7   t   sinR-   t   cost
   frompyfunc(   R   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt$   test_tokenize_numpy_ufunc_consistent©   s    c          C   sI   t  j t t d t ƒ}  t |  ƒ } d d d f } | | k sE t ‚ d  S(   NR'   s   cdask.tests.test_base
f3
p0
.s   cdask.tests.test_base
f2
p0
.s   cdask.tests.test_base
f1
p0
.(   s   cdask.tests.test_base
f2
p0
.(   R'   s   cdask.tests.test_base
f1
p0
.(   (   R'   s   cdask.tests.test_base
f1
p0
.(   R.   R/   R*   R)   R(   R   R-   (   t   ft   rest   sol(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt1   test_tokenize_partial_func_args_kwargs_consistent´   s    	c          C   sN   xG d t  d ƒ d d t d d d ƒ g D] }  t |  ƒ |  k s( t ‚ q( Wd  S(   Ni   gš™™™™™ñ?t   1i   i   (   R   t   sliceR   R-   (   t   i(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_normalize_base½   s    .s   not pdc          C   sŽ  t  j i d d d g d 6d d d  g d 6d d d d g ƒ}  t  j i d d d g d 6d d d  g d 6d d d d g ƒ} t |  ƒ t | ƒ k sš t ‚ d	 | j _ t |  ƒ t | ƒ k sÄ t ‚ t  j i d d d g d 6d
 d d
 g d 6ƒ }  t  j i d d d g d 6d
 d d
 g d 6ƒ } |  j j d ƒ |  d <t |  ƒ t | ƒ k sVt ‚ |  j j d ƒ | d <t |  ƒ t | ƒ k sŠt ‚ d  S(   Ni   i   i   RV   t   4t   asdRW   t   indext   fooR%   R&   t   categoryRX   (	   t   pdt	   DataFrameR   R   R-   Rp   t   nameRW   R>   (   R%   R&   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_pandasÂ   s    >>//c          C   sL   t  j i d d d g d 6d d d  g d 6d d d d g ƒ}  t |  ƒ d  S(	   Ni   i   i   s   x\ud83dRn   s	   asd\ud83ds   y\ud83dRp   (   Rs   Rt   R   R   (   t   df(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt$   test_tokenize_pandas_invalid_unicodeÓ   s    >c          C   s^   t  j i d d d g d j d ƒ 6d d j d ƒ d  g d 6d d d d g ƒ}  t |  ƒ d  S(   Ni   i   i   u   Ã¶t   utf8Rp   (   Rs   Rt   t   encodeR   R   (   Rw   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt(   test_tokenize_pandas_mixed_unicode_bytesÚ   s    Pc          C   sF   d t  f d „  ƒ  Y}  t j i d d  |  ƒ  g d 6ƒ } t | ƒ d  S(   Nt   NoPicklec           B   s   e  Z RS(    (   t   __name__t
   __module__(    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR|   â   s   Rq   RV   (   RJ   Rs   Rt   R   R   (   R|   Rw   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_pandas_no_pickleà   s    "c           C   s¦   t  d d d ƒt  d d d ƒk s* t ‚ t  d ƒ t  d d d ƒk sN t ‚ t  d d d ƒt  d d d ƒk sx t ‚ t  d d d ƒt  d d d ƒk s¢ t ‚ d  S(   Ni   RV   i   i   RW   (   R   R-   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_kwargsê   s    *$*c          C   sD   d t  f d „  ƒ  Y}  t |  d ƒ ƒ t |  d ƒ ƒ k s@ t ‚ d  S(   Nt   Fooc           B   s   e  Z d  „  Z d „  Z RS(   c         S   s   | |  _  d  S(   N(   RV   (   t   selfRV   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   __init__ô   s    c         S   s   d S(   Ns   a foo(    (   R‚   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   __repr__÷   s    (   R}   R~   Rƒ   R„   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR   ò   s   	i   i   (   RJ   R   R-   (   R   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_same_reprñ   s    c          C   s¬   d t  f d „  ƒ  Y}  |  d ƒ |  d ƒ } } t | ƒ t | ƒ k sM t ‚ t | ƒ t | ƒ k sk t ‚ t | ƒ } t j |  d „  ƒ t | ƒ } | | k s¨ t ‚ d  S(   NR   c           B   s   e  Z d  „  Z d „  Z RS(   c         S   s   | |  _  d  S(   N(   RV   (   R‚   RV   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyRƒ   ÿ   s    c         S   s   |  j  S(   N(   RV   (   R‚   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   __dask_tokenize__  s    (   R}   R~   Rƒ   R†   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR   þ   s   	i   i   c         S   s   |  j  d S(   Ni   (   RV   (   R‚   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR+     R,   (   RJ   R   R-   R   t   register(   R   R%   R&   t   beforet   after(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_methodý   s    c          C   s¹   t  d g ƒ t  d g ƒ k s$ t ‚ t  d g ƒ t  d ƒ k sE t ‚ t  d g ƒ t  d g ƒ k si t ‚ t j d ƒ }  t j d ƒ } d | d <t  |  g ƒ t  | g ƒ k sµ t ‚ d  S(   Ni   i   iÐ  i    iè  (   i   (   R   R-   R7   RO   (   RV   RW   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_sequences  s    $!$
c           C   s>   t  i d d 6d d 6ƒ t  i d d 6d d 6ƒ k s: t ‚ d  S(   Ni   RV   (   R   R-   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_dict  s    c           C   s:   t  d d d d h ƒ t  d d d d h ƒ k s6 t ‚ d  S(   Ni   i   RV   (   i   RV   (   i   RV   (   R   R-   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_set   s    c          C   s™   t  t ƒ ‡ d d l m }  |  d d g ƒ } |  d	 d
 g ƒ } |  d d g ƒ } t | ƒ t | ƒ k sq t ‚ t | ƒ t | ƒ k s t ‚ Wd  QXd  S(   Niÿÿÿÿ(   R   R%   i   R&   i   (   R%   i   (   R&   i   (   R%   i   (   R&   i   (   R&   i   (   R%   i   (   R   t   ImportErrort   collectionsR   R   R-   (   R   R%   R&   R'   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_ordered_dict$  s    c          C   sC   t  j d d t  j g d d ƒ}  t |  ƒ t |  ƒ k s? t ‚ d  S(   Nu   foou   JosÃ©R<   t   O(   R7   RA   t   nanR   R-   (   R%   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt$   test_tokenize_object_array_with_nans/  s    !RV   R%   g      ð?y              ð?c         C   s(   t  |  ƒ t  |  ƒ k s$ t |  ƒ ‚ d  S(   N(   R   R-   (   RV   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_base_types5  s    c          C   s€   t  j j d ƒ }  t  j |  j d ƒ ƒ } | j ƒ  } t | ƒ t | ƒ k sT t ‚ d | d *t | ƒ t | ƒ k s| t ‚ d  S(   NiÒ  id   i   i
   (   R7   R8   R9   t   asmatrixt   randt   copyR   R-   (   t   rngR%   R&   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_tokenize_numpy_matrix;  s    
s   not spt   cls_namet   diat   bsrt   coot   csct   csrt   dokt   lilc      	   C   sA  t  j j d ƒ } t j d  ƒ & t j d d d | ƒj |  ƒ } Wd  QX| j	 ƒ  } t
 | ƒ t
 | ƒ k ss t ‚ t | d ƒ r’ d | j d *n |  d k r« d | d
 <n t ‚ t
 | ƒ t
 | ƒ k sÏ t ‚ t j d  ƒ ? | j	 ƒ  j d	 ƒ } t  j d ƒ | j d *| j |  ƒ } Wd  QXt
 | ƒ t
 | ƒ k s=t ‚ d  S(   NiÒ  i
   i'  t   random_statet   datai   R    i   R   (   i   i   (   R7   R8   R9   R   t   warnsR   t   spR–   t   asformatR—   R   R-   t   hasattrR£   t
   ValueErrorRO   t   row(   Rš   R˜   R%   R&   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt    test_tokenize_dense_sparse_arrayF  s     'c          C   s   d t  f d „  ƒ  Y}  t d ƒ d } t | ƒ s8 t ‚ t d ƒ sK t ‚ t |  i  ƒ ƒ sc t ‚ t |  ƒ  ƒ sy t ‚ t |  ƒ sŒ t ‚ d  S(   Nt   DummyCollectionc           B   s   e  Z d d  „ Z d „  Z RS(   c         S   s   | |  _  d  S(   N(   t   dask(   R‚   t   dsk(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyRƒ   g  s    c         S   s   |  j  S(   N(   R¬   (   R‚   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   __dask_graph__j  s    N(   R}   R~   R   Rƒ   R®   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR«   f  s   i   i   (   RJ   R   R   R-   (   R«   RV   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_is_dask_collectione  s    t
   ADataClassc    
      C   sº  t  d ƒ d }  |  d } |  d } d „  } | |  | | d „  |  | | g Dƒ ƒ } t | Œ  \ } } t | ƒ d k s t ‚ | d d d	 g ƒ } | d d d	 d d d	 g ƒ } | | k sÄ t ‚ t d
 t | Œ \ } } t | ƒ d k sô t ‚ | | ƒ | k st ‚ t d d i d d 6ƒ \ } } | s8t ‚ | | ƒ d d i d d 6f k s`t ‚ d „  }	 t |  |	 d f |	 d d f g d
 t ƒ\ } } | | ƒ | |	 d f g ƒ d  S(   Ni   i   i   c         S   s—   |  | i |  d 6| |  6d d | g g d 6d d 6| d f d 6|  d d h d	 6t  d |  f g ƒ d
 6| f } t d  k	 r“ t d |  ƒ | d d
 <n  | S(   NR%   i   i   R&   i
   R'   R[   i   t   eRf   (   R   t   dataclassesR   R°   (   R%   R&   R'   t   iteratort   t(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   build‚  s    
	c         s   s   |  ] } | Vq d  S(   N(    (   t   .0Rl   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pys	   <genexpr>’  s    i   s   ~as   ~bs   ~ct   traverseR%   c          W   s   t  d ƒ ‚ d  S(   Ns   Shouldn't have been called(   R¨   (   t   args(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   fail§  s    (   R   R   RS   R-   t   False(
   R%   R&   R'   Rµ   R¸   R   t   repackt   resultRh   R¹   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_unpack_collections}  s*    

	((	!
t   Tuplec           B   s_   e  Z d	 Z e e j j ƒ Z d „  Z d „  Z	 d „  Z
 d „  Z d „  Z d „  Z d „  Z RS(
   t   _daskt   _keysc         C   s   | |  _  | |  _ d  S(   N(   R¿   RÀ   (   R‚   R­   t   keys(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyRƒ   ´  s    	c         C   s9   t  | t ƒ r5 t t |  j | j ƒ |  j | j ƒ St S(   N(   R3   R¾   R   R¿   RÀ   t   NotImplemented(   R‚   t   other(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   __add__¸  s    c         C   s   |  j  S(   N(   R¿   (   R‚   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR®   ¾  s    c         C   s   |  j  S(   N(   RÀ   (   R‚   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   __dask_keys__Á  s    c         C   s   |  j  S(   N(   RÀ   (   R‚   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR†   Ä  s    c         C   s
   t  d f S(   N(    (   t   tuple(   R‚   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   __dask_postcompute__Ç  s    c         C   s   t  |  j f f S(   N(   R¾   RÀ   (   R‚   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   __dask_postpersist__Ê  s    (   R¿   RÀ   (   R}   R~   t	   __slots__t   staticmethodR¬   t   threadedt   gett   __dask_scheduler__Rƒ   RÄ   R®   RÅ   R†   RÇ   RÈ   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR¾   °  s   						c          C   s?  i d d 6d d 6}  i t  d d f d 6t  d d f d 6} | j |  ƒ i t  d d f d 6t d f d	 6} | j |  ƒ t |  d d g ƒ } t | d d g ƒ } t | d d	 g ƒ } t j t ƒ  d | _ Wd  QXt | ƒ sç t	 ‚ t
 | ƒ t
 | ƒ k st	 ‚ t
 | ƒ t
 | ƒ k s#t	 ‚ | j ƒ  d k s;t	 ‚ t j | | | g ƒ d d d g f k skt	 ‚ | | | } | j ƒ  d k s‘t	 ‚ | j ƒ  } t | t ƒ s²t	 ‚ | j t t d t d d ƒ ƒ ƒ k sßt	 ‚ | j ƒ  d k s÷t	 ‚ t j | | | ƒ \ } }	 }
 | |	 |
 } | j | j k s;t	 ‚ d  S(   Ni   R%   i   R&   R'   R[   i   R±   Rf   i   i   i   t   abcdefi   (   i   i   (   i   i   (   i   i   (   i   i   (   i   i   i   i   i   i   (   i   i   i   i   i   i   (   R    t   updateR   R¾   R   t   raisest   AttributeErrorRq   R   R-   R   R   R¬   R
   R3   R¿   t   dictt   zipt   range(   R­   t   dsk2t   dsk3RV   RW   RX   R´   t   t2t   x2t   y2t   z2t   t3(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_custom_collectionÎ  s4    0-s   not dbc       	      s  d d l  m }  t j t d ƒ d d ƒ} t j t d ƒ } t j t d ƒ } | j	 | ƒ j	 | ƒ } g  ‰  |  d ‡  f d	 †  ƒ  | j
 d
 d d t ƒ Wd  QXt g  ˆ  D] } d | d k r¬ | ^ q¬ ƒ d k sÝ t ‚ t g  ˆ  D] } d | d k rç | ^ qç ƒ d k st ‚ g  ‰  |  d ‡  f d †  ƒ  | j
 d
 d ƒ Wd  QXt g  ˆ  D] } d | d k rW| ^ qWƒ d k sˆt ‚ t g  ˆ  D] } d | d k r’| ^ q’ƒ d k sÃt ‚ t g  ˆ  D] } d | d k rÍ| ^ qÍƒ d k sþt ‚ d  S(   Niÿÿÿÿ(   t   Callbackid   t   npartitionsi   i   i   t   pretaskc            s   ˆ  j  |  ƒ S(   N(   t   append(   t   keyR¸   (   RÁ   (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR+     R,   t	   schedulers   single-threadedt   optimize_graphR   i    R    c            s   ˆ  j  |  ƒ S(   N(   Rà   (   Rá   R¸   (   RÁ   (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR+     R,   i   s   add-from_sequence-mul(   t   dask.callbacksRÝ   t   dbt   from_sequenceRÔ   R.   R/   R    R   t   mapR   Rº   RS   R-   (   RÝ   R&   t   add1t   mul2t   ot   k(    (   RÁ   s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_compute_no_optö  s     ;;;;s   not dac          C   s’   t  j d ƒ j d ƒ }  t j |  d d ƒ} | d } | d } t | | ƒ \ } } t  j | |  d ƒ sr t ‚ t  j | |  d ƒ sŽ t ‚ d  S(	   Nid   i
   t   chunksi   i   i   (   i
   i
   (   i   i   (   R7   RO   t   reshapet   dat
   from_arrayR   t   allcloseR-   (   t   arrt   darrt   darr1t   darr2t   out1t   out2(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_compute_array  s    

c          C   s³   d d l  m }  t j d ƒ j d
 ƒ } t j | d d ƒ} | d | j d d	 ƒ } | j ƒ  } |  | | ƒ t	 | j
 ƒ j | j
 ƒ s‘ t ‚ t | j
 ƒ | j k s¯ t ‚ d  S(   Niÿÿÿÿ(   t	   assert_eqid   i
   Rí   i   i   t   axisi    (   i
   i
   (   i   i   (   t   dask.array.utilsRù   R7   RO   Rî   Rï   Rð   t   meanR
   RT   R¬   t   issubsetR-   RS   RÞ   (   Rù   Rò   RV   RW   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_persist_array  s    !s   not ddc          C   s½   t  j i d d d d g d 6d d d d g d 6ƒ }  t j |  d d ƒ} | j d } | j | j } t | | ƒ \ } } t  j j j	 | |  j d ƒ t  j j j	 | |  j |  j ƒ d  S(	   Ni   i   i   i   R%   i   R&   RÞ   (
   Rs   Rt   t   ddt   from_pandasR%   R&   R   t   utilt   testingt   assert_series_equal(   Rw   t   ddft   ddf1t   ddf2Rö   R÷   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_compute_dataframe)  s    5s   not dd or not dac          C   sÔ   t  j d ƒ j d ƒ }  t j |  d d ƒd } t j i d d d d g d	 6d d d d g d
 6ƒ } t j | d d ƒj	 d } t
 | | ƒ \ } } t  j | |  d ƒ s³ t ‚ t j j j | | j	 d ƒ d  S(   Nid   i
   Rí   i   i   i   i   i   R%   R&   RÞ   (   i
   i
   (   i   i   (   R7   RO   Rî   Rï   Rð   Rs   Rt   Rÿ   R   R%   R   Rñ   R-   R  R  R  (   Rò   Ró   Rw   R  t   arr_outt   df_out(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_compute_array_dataframe4  s    5c          C   sG   t  j d t j j d	 ƒ d d j d ƒ g ƒ }  t j |  d d ƒd  S(
   NR£   i   i   t   columnsu   Ã¶Ry   RÞ   i   (   i   i   (   Rs   Rt   R7   R8   Rz   Rÿ   R   (   Rw   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt-   test_compute_dataframe_valid_unicode_in_bytes?  s    	c          C   s>   t  j d t j j d ƒ d d g ƒ }  t j |  d d ƒd  S(	   NR£   i   i   R  s   \ud83dRÞ   i   (   i   i   (   Rs   Rt   R7   R8   Rÿ   R   (   Rw   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt&   test_compute_dataframe_invalid_unicodeH  s    	s   not da or not dbc             s¤   t  j d d d ƒ‰ t j d d d g ƒ ‰  t j t ‡  ‡ f d †  ƒ t ˆ ˆ  d d ƒ\ }  } t j	 |  t j d ƒ ƒ s… t
 ‚ | d d d g k s  t
 ‚ d  S(	   Ni   Rí   i   i   i   c              s   t  ˆ ˆ  ƒ S(   N(   R   (    (   R&   RV   (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR+   W  R,   Râ   s   single-threaded(   Rï   RO   Rå   Ræ   R   RÐ   R¨   R   R7   Rñ   R-   (   t   xxt   bb(    (   R&   RV   s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_compute_array_bagR  s    !c          C   s|   t  j d d d ƒ}  d } t |  | ƒ \ } } | |  j ƒ  k j ƒ  sN t ‚ | | k s` t ‚ t d ƒ d k sx t ‚ d  S(   Ni   Rí   i   i
   (   i   (   Rï   RO   R   t   allR-   (   RV   RW   R  t   yy(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_compute_with_literal^  s    c          C   sâ   t  d ƒ d }  |  d } |  d } t i |  d 6d d | g d 6| d f ƒ i d d 6d d d g d 6d f k sy t ‚ t |  | g | d	 t ƒ} | d
 d
 |  k s® t ‚ | d
 d | k sÈ t ‚ | d d k sÞ t ‚ d  S(   Ni   i   i   R%   R&   i   i   i   R·   i    (   i   i   (   R   R   R-   Rº   (   R%   R&   R'   Rg   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_compute_nestedj  s    

),t   reasons'   graphviz exception with Python -OO flagc          C   sä  t  j d ƒ t ƒ  È}  t j d d d ƒ} | j d t j j |  d ƒ ƒ t j j	 t j j |  d ƒ ƒ st t
 ‚ | j d t j j |  d ƒ ƒ t j j	 t j j |  d ƒ ƒ sº t
 ‚ t | d	 d d t j j |  d ƒ ƒt j j	 t j j |  d ƒ ƒ st
 ‚ i d	 d
 6t d
 d f d 6t d
 d	 f d 6} t | | d t j j |  d ƒ ƒt j j	 t j j |  d ƒ ƒ s|t
 ‚ t | d
 d d g ƒ } t | d t j j |  d ƒ ƒt j j	 t j j |  d ƒ ƒ sÚt
 ‚ Wd  QXd  S(   Nt   graphvizi   Rí   i   t   filenamet   mydasks
   mydask.pngs
   mydask.pdfi   R%   R&   R'   (   R   R   R   Rï   RO   R	   t   osR    t   joint   existsR-   R    R   R¾   (   R[   RV   R­   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_visualizew  s    ''%'-"'c         C   s…   t  j d ƒ t j j t |  ƒ d ƒ } t j i t d d f d 6g d | ƒt	 | ƒ  } | j
 ƒ  } Wd  QXd | k s t ‚ d  S(   NR  s
   myfile.doti   i   s   abc-xyzR  (   R   R   R  R    R  R4   R¬   R	   R    t   opent   readR-   (   R   RU   Rf   t   text(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_visualize_lists  s    &c          C   sš   t  j d ƒ t  j d ƒ t j d d d ƒ}  t d d ƒ V } |  j d d	 d
 | d d ƒ t | ƒ  } | j ƒ  } Wd  QXd | k s t ‚ Wd  QXd  S(   NR  s   matplotlib.pyploti   Rí   i   t	   extensiont   dott   colort   orderR  t   cmapt   RdBus   color="#(	   R   R   Rï   RO   R   R	   R  R  R-   (   RV   RU   Rf   R  (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_visualize_order›  s    c          B   sw   d d  l  }  d d l m } | d ƒ } |  j d } e | d d ƒ | j U| j } e | ƒ } d | k ss t ‚ d  S(   Niÿÿÿÿ(   t   dedents%   
    def inc():
        return x
    t   __main__s   <test>t   exect   cloudpickle(	   t   syst   textwrapR(  t   modulest   compilet   __dict__R   R   R-   (   R,  R(  t   defnR)  Rf   R´   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt5   test_use_cloudpickle_to_tokenize_functions_in__main__©  s    		c         C   sQ   t  |  ƒ }  x> |  D]6 } |  | d t k r t f |  | d |  | <q q W|  S(   Ni    i   (   RÒ   R   R   (   R­   RÁ   Rá   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt
   inc_to_decº  s
     c          C   s€   t  j t ƒ d ƒ }  |  j ƒ  d k s- t ‚ t  j j d t g ƒ  |  j ƒ  d k s^ t ‚ Wd  QX|  j ƒ  d k s| t ‚ d  S(   Ni   i   t   optimizationsi    (   R¬   R   R   R   R-   t   configRT   R3  (   RV   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_optimizations_keywordÂ  s
    c          C   sÓ  t  j t ƒ d ƒ }  t  j t ƒ |  ƒ } |  | } t |  | | d ƒ \ } } } } | d k sg t ‚ t | j  ƒ } t | j  ƒ | k s‘ t ‚ t | j  ƒ | k s¬ t ‚ t  j | | | ƒ t  j |  | | ƒ k sÜ t ‚ t  j |  | | d t g ƒ} t |  | | d t g ƒ\ }	 }
 } t  j |	 |
 | ƒ | k s?t ‚ t  j j	 d t g ƒ   t |  | | ƒ \ } } } Wd  QXxS t
 |	 |
 | g | | | g ƒ D]0 \ } } t | j  ƒ t | j  ƒ k s›t ‚ q›Wd  S(   Ni   R4  (   R¬   R   R   R   R-   RÒ   R   R3  R5  RT   RÓ   (   RV   RW   RX   RØ   RÙ   RÚ   t   constantR­   t   solst   x3t   y3t   z3t   x4t   y4t   z4R%   R&   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_optimizeÌ  s     
!0$!!.c          C   s®  t  j t ƒ d ƒ }  t  j t ƒ |  ƒ } |  | } t i |  d 6d d | g d 6| d f ƒ } | d d } | d d d } | d d } t | t ƒ s£ t ‚ t | t ƒ s¸ t ‚ t | t ƒ sÍ t ‚ t | j  ƒ t | j  ƒ k oÿ t | j  ƒ k n s
t ‚ t | Œ  i d d 6d d d g d 6d	 f k s?t ‚ t |  | g | d t	 ƒ} | d d |  k stt ‚ | d d | k sŽt ‚ | d j ƒ  d k sªt ‚ d  S(
   Ni   R%   i   R&   i    i   i   R·   (   i   i   (
   R¬   R   R   R   R3   R   R-   RÒ   R   Rº   (   R%   R&   R'   R¼   t   a2t   b2t   c2Rg   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_optimize_nestedé  s     
,=5c          C   s   d }  t  j t j d |  g ƒ } t t | j ƒ  ƒ ƒ } d | k sK t ‚ d d d d d d	 d
 g } x  | D] } | | k sm t ‚ qm Wd S(   sI   
    Startup time: `import dask` should not import too many modules.
    sY   if 1:
        import dask
        import sys

        print(sorted(sys.modules))
        s   -cR¬   s
   dask.arrays   dask.dataframeR#   R$   t   partdt   s3fst   distributedN(   t
   subprocesst   check_outputR,  t
   executableRT   t   evalt   decodeR-   (   t   codet   outR.  t	   blacklistt   mod(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_default_imports   s    	c           C   s"   t  d d d ƒ d k s t ‚ d  S(   Ni   i   i   (   i   i   i   (   R
   R-   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_persist_literals  s    c          C   sO  t  d ƒ d }  |  d } |  d } t i |  d 6d d | g d 6| d f ƒ } t | d d t ƒ sm t ‚ t | d d d t ƒ sŽ t ‚ t | d d t ƒ s« t ‚ t | Œ  i d d 6d d d g d 6d f k sà t ‚ t |  | g | d
 t ƒ} | d d |  k st ‚ | d d | k s/t ‚ | d j ƒ  d	 k sKt ‚ d  S(   Ni   i   i   R%   R&   i    i   i   i   R·   (   i   i   (   R   R
   R3   R   R-   R   Rº   (   R%   R&   R'   R¼   Rg   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_persist_nested  s    

,!5c          C   s©   t  d ƒ }  t  t ƒ |  ƒ } t  t ƒ | ƒ } t | ƒ \ } t | t ƒ sT t ‚ | j | j k sl t ‚ t | j ƒ d k s‡ t ‚ | j	 ƒ  | j	 ƒ  k s¥ t ‚ d  S(   Ni   (
   R   R   R
   R3   R   R-   Rá   RS   R¬   R   (   t   x1RØ   R9  R  (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_persist_delayed(  s    c          C   s€  t  j d d d ƒd }  t j d d d g ƒ j t ƒ } t j t ƒ  t	 |  | ƒ Wd  QXt	 |  | d d ƒ\ } } t
 | t  j ƒ s t ‚ t
 | t j ƒ s¨ t ‚ | j |  j k sÀ t ‚ | j | j k sØ t ‚ t | j ƒ | j k ot |  j ƒ k  n st ‚ t | j ƒ | j k o;t | j ƒ k  n sFt ‚ t j |  | ƒ s^t ‚ t | ƒ t | ƒ k s|t ‚ d  S(   Ni   Rí   i   i   i   Râ   s   single-threaded(   Rï   RO   Rå   Ræ   Rç   R   R   RÐ   R¨   R
   R3   t   ArrayR-   t   BagRu   RS   R¬   RÞ   R7   Rñ   t   list(   RV   R&   R  R  (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_persist_array_bag5  s    !77c          C   sP   x! t  d ƒ D] }  t d „  ƒ q Wd t t ƒ k  oA d k  n sL t ‚ d  S(   Niè  c         S   s   |  S(   N(    (   RV   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR+   M  R,   i2   iX  (   RÔ   R   RS   R   R-   (   Rl   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt$   test_normalize_function_limited_sizeK  s    c       
   C   sL  t  j d ƒ }  t  j d ƒ } |  j d d d ƒ} d „  } d d l m } | | d	 t j d ƒ d	 ƒ t j j d
 | ƒ + | | d	 t j d ƒ d d	 d ƒ Wd  QX| | d	 t j d ƒ d	 ƒ | j	 d d d ƒ} t j j d
 | ƒ R t j
 | d	 | j t ƒ d d ƒ\ } } | | t j d ƒ d d	 d ƒ Wd  QXd  S(   Ns
   dask.arrays   dask.bagi
   Rí   i   c         S   s   d „  |  j  ƒ  Dƒ S(   Nc         S   s(   i  |  ] \ } } t  d  | f | “ q S(   i   (   R   (   R¶   Rë   t   v(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pys
   <dictcomp>Y  s   	 (   t   items(   R­   RÁ   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   optimize_doubleX  s    iÿÿÿÿ(   Rù   i   t   array_optimizei   RÞ   Râ   s   single-threaded(   i   (   R   R   t   onesRû   Rù   R7   R¬   R5  RT   RÔ   R   Rç   R   (   Rï   Rå   RV   R\  Rù   R&   R  R  (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_optimize_globalsR  s    	,+c             s…   t  j d ƒ }  |  j d d d ƒ} | d  d d !d  d  d … d ‰  ‡  f d	 †  } t j j d
 d  d | ƒ  ˆ  j ƒ  Wd  QXd  S(   Ns
   dask.arrayi
   Rí   i   i	   i   i   i   c            s+   |  t  ˆ  j ƒ k s t ‚ t j |  | ƒ S(   N(   RÒ   R¬   R-   RÌ   (   R­   RÁ   (   RW   (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   my_getq  s    R]  Râ   (   i   (   R   R   R^  R¬   R5  RT   R   R   (   Rï   RV   R`  (    (   RW   s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_optimize_Nonek  s    "c          C   sô   d „  }  |  t  d <t t ƒ d ƒ } zÀ | j ƒ  d k s@ t ‚ | j d d ƒ d k s^ t ‚ t j j d d ƒ  | j ƒ  d k sŒ t ‚ Wd  QX| j ƒ  d k sª t ‚ t j j d d ƒ # | j d d ƒ d k sÞ t ‚ Wd  QXWd  t  d =Xd  S(   Nc         [   s
   d g g S(   Ni{   (    (   R­   RÁ   t   kwargs(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   schedulez  s    Rq   i   i   Râ   i{   t   threads(   R   R   R   R   R-   R¬   R5  RT   (   Rc  RV   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_scheduler_keywordy  s    	
(c          C   s\   t  t ƒ d ƒ }  t j t ƒ  } |  j d t j ƒ Wd  QXd t | j	 ƒ k sX t
 ‚ d  S(   Ni   RÌ   s
   scheduler=(   R   R   R   RÐ   t	   TypeErrorR   R¬   RÌ   R4   t   valueR-   (   RV   t   info(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_raise_get_keyword  s    c           C   s­   t  ƒ  d  k s t ‚ t  d d ƒ t j j k s6 t ‚ t  d d ƒ t j j k sW t ‚ t j j	 d d ƒ & t  d d ƒ t j j k sŽ t ‚ Wd  QXt  ƒ  d  k s© t ‚ d  S(   NRâ   Rd  t   sync(
   R   R   R-   R¬   RË   RÌ   t   localt   get_syncR5  RT   (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_get_scheduler˜  s    !!'c             sV   t  g ‰  ‡  f d †  }  t d „  ƒ ƒ  j d |  ƒ d k sB t ‚ ˆ  d sR t ‚ d  S(   Nc            s   t  ˆ  d <t j |  | ƒ S(   Ni    (   t   TrueR¬   RÌ   (   R­   RÁ   R¸   Rb  (   t   called(    s3   lib/python2.7/site-packages/dask/tests/test_base.pyRÌ   ¤  s    
c           S   s   d S(   Ni   (    (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyR+   ¨  R,   Râ   i   i    (   Rº   R   R   R-   (   RÌ   (    (   Ro  s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_callable_scheduler¡  s    	*Râ   Rd  t	   processesc         C   s©   t  d „  ƒ } g  t d ƒ D] } | | ƒ ^ q } d } t j j d | ƒ ) t ƒ   } t d |  | Œ } Wd  QXWd  QXd „  | j Dƒ } t | ƒ | k s¥ t	 ‚ d  S(   Nc         S   s   t  j d ƒ |  S(   Ng      à?(   t   timet   sleep(   RV   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyRf   °  s    i   i   t   num_workersRâ   c         S   s   h  |  ] } | j  ’ q S(    (   t	   worker_id(   R¶   Rl   (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pys	   <setcomp>º  s   	 (
   R   RÔ   R¬   R5  RT   R   R   t   resultsRS   R-   (   Râ   Rf   Rl   R%   Rt  t   proft   workers(    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   test_num_workers_config¬  s    % (    (   R›   Rœ   R   Rž   RŸ   R    R¡   (‚   R  R   t   operatorR    R   RG  R,  Rr  R   R   R"   R   R¬   R   t	   dask.baseR   R   R   R   R	   R
   R   R   R   R   R   R   R   t   dask.delayedR   t
   dask.utilsR   R   R   t   dask.utils_testR   R   t   dask.compatibilityR   R   R   t   dask.diagnosticsR   R!   R   R.   Rï   Rå   Rÿ   R7   R¥   Rs   R(   R)   R*   R2   R6   t   markt   skipifR;   R?   R@   RB   RF   RK   R\   R_   Re   Ri   Rm   Rv   Rx   R{   R   R€   R…   RŠ   R‹   RŒ   R   R   R“   t   parametrizeRn  R   R4   t   intR”   R™   Rª   R¯   R²   t   make_dataclassR°   RŽ   R½   R¾   RÜ   Rì   Rø   Rþ   R  R
  R  R  R  R  R  t   flagsR  R   R'  R2  R3  R6  R?  RC  RP  RQ  RR  RT  RX  RY  R_  Ra  Re  Ri  Rm  Rp  Ry  (    (    (    s3   lib/python2.7/site-packages/dask/tests/test_base.pyt   <module>   sÌ   X							
						!$	
	3	(	
	
			
															