
    bi>                    V    d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm	Z	 d Z
d Zd Zy)	    )annotationsN)
utils_test)HighLevelGraph)_check_warningc                    ddi} dt         j                  dfi}| |d}t               dhd}t        ||      }t        j                  |d      |j
                  d   u sJ t        j                  |d      |j
                  d   u sJ t        j                  t        d	
      5  t        j                  |d       d d d        y # 1 sw Y   y xY w)Nx   y)a-layer	bee-layerr   abr   zNo layer starts withmatchfoo)	r   incsetr   	hlg_layerlayerspytestraisesKeyError)r   r   r   dependencieshgs        U/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/tests/test_utils_test.pytest_hlg_layerr      s    	aA	z~~s#$A+F"uI;?L		-BC(BIIi,@@@@C(BIIk,BBBB	x'=	> (R'( ( (s   !CC
c                    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        j
                  |d
      |j                  d   u sJ t        j
                  |d      |j                  d   u sJ t        j
                  |d      |j                  d   |j                  d	   fv sJ y )Nr   r	   r
   zr)r   r   cdr   r   r    r!   r   )r   r   sumr   r   hlg_layer_topologicalr   )r   r   r    r!   r   r   r   s          r   test_hlg_layer_topologicalr%      s    	aA	z~~s#$A	z~~s#$A	sS#J A111-FcU#c3ZHL		-B++B3ryy~EEE++B2biinDDD++B2ryy~ryyQT~6VVVV    c                     G d dt               } t        j                         5  t        j                  d       t	        d| d      5  t        j
                  d|        d d d        d d d        t        j                  | d      5  t	        d| d      5  t        j
                  d|        d d d        d d d        y # 1 sw Y   ^xY w# 1 sw Y   bxY w# 1 sw Y   *xY w# 1 sw Y   y xY w)Nc                      e Zd Zy)&test__check_warning.<locals>.MyWarningN)__name__
__module____qualname__ r&   r   	MyWarningr)   (   s    r&   r.   errorTr   r   F)Warningwarningscatch_warningssimplefilterr   warnr   warns)r.   s    r   test__check_warningr6   '   s    G  
	 	 	" ,g&D)U3 	,MM%+	,,
 
iu	- ,E9e4 	,MM%+	,, ,	, 	,, ,	, 	,, ,sG   #CB;CCC*C;C	 CCC	CC()
__future__r   r1   r   daskr   dask.highlevelgraphr   dask.utils_testr   r   r%   r6   r-   r&   r   <module>r;      s'    "    . *
(W,r&   