ó
ßüÚ\c           @` sÍ   d  d l  m Z m Z m Z d  d l Z e j d ƒ d  d l m Z d  d l m	 Z	 d  d l
 m Z m Z m Z d  d l m Z m Z m Z d „  Z d	 „  Z e d
 e ƒ d „  ƒ Z d „  Z d „  Z d S(   i    (   t   print_functiont   divisiont   absolute_importNt   bokeh(   t   delayed(   t   wait(   t   progress_quadst
   nbytes_bart   progress_stream(   t   divt   gen_clustert   incc          C` sí  i i d d 6d d 6d d 6d 6i d d 6d	 d 6d d 6d
 6i d	 d 6d d 6d	 d 6d 6i d d 6d	 d 6d d 6d 6i d d 6d	 d 6d d 6d 6}  t  |  d d ƒ} | j d ƒ } t t | ƒ ƒ d k sÑ t ‚ i d d d g d 6d d d g d 6d	 d	 d g d 6d d d g d 6d	 d d	 g d 6d d d g d 6d d d g d 6d d d	 g d 6d d d	 g d
 6d	 d	 d g d 6d d d	 g d 6d d d g d 6d( d) d! g d" 6d+ d, d! g d$ 6d. d/ d g d% 6d1 d2 d4 g d& 6} | | k sét ‚ d  S(5   Ni   R   i   t   deci   t   addt   alli   i    t   memoryt   erredt   releasedt
   processingt   nrowst   colori   t   names	   show-namet   leftgÍÌÌÌÌÌì?gffffffþ?t   rightiÿÿÿÿt   topgš™™™™™é¿gÍÌÌÌÌÌü¿t   bottoms   3 / 5s   2 / 4s   1 / 1t   doneg      Ð?g      ð?s   released-locg      à?s
   memory-locs	   erred-locs   processing-locgÍÌÌÌÌÌì?g
×£p=
Ç?gÍÌÌÌÌÌÌ?gš™™™™™@gHáz®Gá?gÍÌÌÌÌÌÜ?gš™™™™™@gHáz®Gá?gÍÌÌÌÌÌÜ?gÍÌÌÌÌÌ@g
×£p=
ç?gÍÌÌÌÌÌì?gÍÌÌÌÌÌì?gffffffþ?(   R   t   popt   lent   sett   AssertionError(   t   msgt   dR   t   expected(    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pyt   test_progress_quads   s6    c          C` s¶   g  t  d ƒ D] }  d |  ^ q } i d „  | Dƒ d 6d „  | Dƒ d 6d „  | Dƒ d 6d	 „  | Dƒ d
 6d „  | Dƒ d 6} t | d d d d ƒ} t | d ƒ d k s² t ‚ d  S(   Niè  s   x-%dc         S` s   i  |  ] } d  | “ q S(   i   (    (   t   .0t   k(    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pys
   <dictcomp>6   s   	 R   c         S` s   i  |  ] } d  | “ q S(   i    (    (   R#   R$   (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pys
   <dictcomp>7   s   	 R   c         S` s   i  |  ] } d  | “ q S(   i    (    (   R#   R$   (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pys
   <dictcomp>8   s   	 R   c         S` s   i  |  ] } d  | “ q S(   i    (    (   R#   R$   (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pys
   <dictcomp>9   s   	 R   c         S` s   i  |  ] } d  | “ q S(   i    (    (   R#   R$   (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pys
   <dictcomp>:   s   	 R   R   i   t   ncolsi   R   i   (   t   rangeR   R   R   (   t   it   keysR   R    (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pyt   test_progress_quads_too_many3   s    #t   clientc         c` sh  |  j  t d g d t d ƒ ƒ } d } x& t d ƒ D] } t t ƒ | ƒ } q5 W|  j | ƒ } t | | g ƒ Vt | j d d ƒV} | j	 ƒ  V}	 |	 j
 d ƒ }
 |	 i i d d 6d d 6d	 6i d d 6d
 6i d d 6d d 6d 6i d d 6d 6i  d 6k st ‚ t |
 ƒ t |	 d	 ƒ k s%t ‚ t d „  |
 j ƒ  Dƒ ƒ sGt ‚ t |	 ƒ sYt ‚ | j ƒ  Vd  S(   Ni   i
   i   t   intervalg{®Gáz„?t   nbytesR	   R   R   R   i	   R   i   R   R   c         s` s   |  ] } | d  k Vq d S(   i    N(    (   R#   t   v(    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pys	   <genexpr>W   s    (   t   mapR	   R&   R   R   t   computeR   R   t   addresst   readR   R   R   R   t   valuesR   t   close(   t   ct   st   at   bt   futurest   xR'   t   futuret   commR   R,   (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pyt   test_progress_streamA   s&    """c          C` sÄ   i d d 6d d 6}  i d d g d 6d d g d 6d	 d
 g d 6d d g d 6d d g d 6d d g d 6d d g d 6} t  |  ƒ } | j d ƒ } t t | ƒ ƒ d k s® t ‚ | | k sÀ t ‚ d  S(   Niè  R   i¸  R   R   i    g      è?R   g      Ø?g      ì?t   centerg      ð?R   iK   i   t   percentgú~j¼t“h?gü©ñÒMbP?t   MBt   textR   i   (   R   R   R   R   R   (   R,   R!   t   resultR   (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pyt   test_nbytes_bar^   s    c          C` sÁ   g  t  d ƒ D] }  d |  ^ q } i d „  | Dƒ d 6d „  | Dƒ d 6d „  | Dƒ d 6d	 „  | Dƒ d
 6d „  | Dƒ d 6} t | d d ƒ} | j d ƒ } t t | ƒ ƒ d k  s½ t ‚ d  S(   Niè  s   fn%dc         S` s   i  |  ] } d  | “ q S(   i   (    (   R#   t   fn(    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pys
   <dictcomp>s   s   	 R   c         S` s   i  |  ] } d  | “ q S(   i   (    (   R#   RC   (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pys
   <dictcomp>t   s   	 R   c         S` s   i  |  ] } d  | “ q S(   i    (    (   R#   RC   (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pys
   <dictcomp>u   s   	 R   c         S` s   i  |  ] } d  | “ q S(   i    (    (   R#   RC   (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pys
   <dictcomp>v   s   	 R   c         S` s   i  |  ] } d  | “ q S(   i    (    (   R#   RC   (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pys
   <dictcomp>w   s   	 R   R   i   R   id   (   R&   R   R   R   R   R   (   R'   t	   funcnamesR   R    R   (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pyt"   test_progress_quads_many_functionsp   s    #(   t
   __future__R    R   R   t   pytestt   importorskipt   daskR   t   distributed.clientR   t'   distributed.diagnostics.progress_streamR   R   R   t   distributed.utils_testR	   R
   R   R"   R)   t   TrueR<   RB   RE   (    (    (    sQ   lib/python2.7/site-packages/distributed/diagnostics/tests/test_progress_stream.pyt   <module>   s   	!		