
    >ie+                       d dl mZ d dlZd dlZd dlmZ  ej        g dg ddg d           ej        g dg ddg d	           ej        g d
g ddg d          dZ	 e
e	                                          Zd Zd ZdS )    )annotationsN)         )         )ab)r   r   r   )index)r   r   r   )r   r	      )   r   	   )r   r   r   )r   r   r   ))xr   )r   r   )r   r   c                    t          j        dt          d          i          } t          j        | d          } | j        dz   | j        dz   z   }t          j                            ddi          5  |	                    |j        |
                                          }d d d            n# 1 swxY w Y   |	                    |j        |
                                          }t          |          d	k    sJ t          |          d	k    sJ d S )
Nr   
   r   npartitionsr   r   zoptimization.fuse.ave-widthr      )pd	DataFramerangeddfrom_pandasr   daskconfigset__dask_optimize____dask_keys__len)dfsr
   r   s       Llib/python3.11/site-packages/dask/dataframe/tests/test_optimize_dataframe.pytest_fuse_ave_widthr$      s#   	sE"II&	'	'B		*	*	*B	bdQhA	7;	<	< ; ;(9(9::; ; ; ; ; ; ; ; ; ; ; ; ; ; ; 	
AFAOO$5$566Aq66R<<<<q66R<<<<<<s   0.B**B.1B.c                 8   ddl m}  t          j        t	          d          t	          d          d          }t          j        |d          }t	          d          D ]}|j        dz   |j        z   |d<    | |j	                  }t          |          d	k    sJ d S )
Nr   )optimize_blockwiser   )r   yr   r   r   r   r   )dask.array.optimizationr&   r   r   r   r   r   r   r'   r   r    )r&   r!   ddf_graphs        r#   test_optimize_blockwiser,      s    ::::::	E"IIE"II66	7	7B
.
+
+
+C2YY % %519su$Csx((Eu::??????    )
__future__r   pandasr   r   dask.dataframe	dataframer   r   dsklistvaluesdfsr$   r,    r-   r#   <module>r7      s   " " " " " "            bl;;999MMMbl;;999MMMbl;;999MMM 
 d3::<<      r-   