
    >ie>                    X    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dS )	    )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            d S # 1 swxY w Y   d S )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        :lib/python3.11/site-packages/dask/tests/test_utils_test.pytest_hlg_layerr      s"   	aA	z~s#$A++F"uuI;??L		-	-BC((BIi,@@@@@C((BIk,BBBBB	x'=	>	>	> ( (R'''( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s   C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 d S )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HHL		-	-B+B33ry~EEEE+B22binDDDD+B22ry~ryQT~6VVVVVVV    c                      G d dt                     } t          j                    5  t          j        d           t	          d| d          5  t          j        d|            d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   t          j        | d          5  t	          d| d          5  t          j        d|            d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nc                      e Zd ZdS )&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3 	, 	,M%+++	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	,, , , , , , , , , , , , , , ,
 
iu	-	-	- , ,E9e44 	, 	,M%+++	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	,, , , , , , , , , , , , , , , , , ,sk   &BA/#B/A3	3B6A3	7BB
B
'C39CC3C	C3"C	#C33C7:C7)
__future__r   r1   r   daskr   dask.highlevelgraphr   dask.utils_testr   r   r%   r6   r-   r&   r   <module>r;      s    " " " " " "         . . . . . . * * * * * *
( 
( 
(W W W, , , , ,r&   