
\c           @   s]   d  d l  Z  d  d l Z d  d l j Z d  d l m Z d  d l m Z d   Z	 d   Z
 d S(   iN(   t   inc(   t   HighLevelGraphc         C   sx   t  j d  t |   } t j d d d } | d } | d } | | } | j j |  t j j	 |  st t
  d  S(   Nt   graphvizi
   t   chunksi   i   i   (   i   (   t   pytestt   importorskipt   strt   dat   onest   daskt	   visualizet   ost   patht   existst   AssertionError(   t   tmpdirt   fnt   at   bt   ct   d(    (    s8   lib/python2.7/site-packages/dask/tests/test_highgraph.pyt   test_visualize
   s    


c          C   s   i d d 6}  i t  d f d 6} i |  d 6| d 6} i t   d 6d h d 6} t | |  } t |  i d d 6t  d f d 6k s t  d  S(   Ni   t   xt   yR   R   (   R    t   setR   t   dictR   (   R   R   t   layerst   dependenciest   hg(    (    s8   lib/python2.7/site-packages/dask/tests/test_highgraph.pyt
   test_basic   s    (   R   R   t
   dask.arrayt   arrayR   t   dask.utils_testR    t   dask.highlevelgraphR   R   R   (    (    (    s8   lib/python2.7/site-packages/dask/tests/test_highgraph.pyt   <module>   s   	