U
    /e                     @   sP   d dl Z d dlZd dlmZ d dlmZ d dlmZ dd Zdd Z	d	d
 Z
dS )    N)
utils_test)HighLevelGraph)_check_warningc               	   C   s   ddi} dt jdfi}| |d}t dhd}t||}t |d|jd ksRtt |d|jd kslttjt	d	d
 t |d W 5 Q R X d S )Nx   y)a-layer	bee-layerr   abr	   zNo layer starts withmatchfoo)
r   incsetr   Z	hlg_layerlayersAssertionErrorpytestZraisesKeyError)r
   r   r   dependencieshg r   >/tmp/pip-unpacked-wheel-dbjnr7gq/dask/tests/test_utils_test.pytest_hlg_layer
   s    

r   c                  C   s   ddi} dt jdfi}dt jdfi}dtddgfi}| |||d}t dhdhdd	hd}t||}t |d
|jd ks~tt |d|jd kstt |d|jd |jd	 fkstd S )Nr   r   r   zr)r
   r   cdr
   r   r   r   r   )r   r   sumr   r   Zhlg_layer_topologicalr   r   )r
   r   r   r   r   r   r   r   r   r   test_hlg_layer_topological   s    
r    c               
   C   s   G dd dt } t 4 td td| d td|  W 5 Q R X W 5 Q R X tj| dd* td| d td|  W 5 Q R X W 5 Q R X d S )Nc                   @   s   e Zd ZdS )z&test__check_warning.<locals>.MyWarningN)__name__
__module____qualname__r   r   r   r   	MyWarning&   s   r$   errorTr   r   F)Warningwarningscatch_warningssimplefilterr   warnr   Zwarns)r$   r   r   r   test__check_warning%   s    

 r+   )r'   r   Zdaskr   Zdask.highlevelgraphr   Zdask.utils_testr   r   r    r+   r   r   r   r   <module>   s   