U
    /e                     @   s   d dl Zd dlZd dlmZ ejdddgdddgdd ddgd	ejdddgdddgdddd
gd	ejdd
dgd d d gddddgd	dZee	 Z
dd Zdd ZdS )    N                  )ab)index      	   ))xr   )r   r   )r   r   c               	   C   s   t dtdi} tj| dd} | jd | jd  }tjddi |	|j|
 }W 5 Q R X |	|j|
 }t|d	kstt|d	kstd S )
Nr   
   r   Znpartitionsr   r   zoptimization.fuse.ave-widthr      )pd	DataFramerangeddfrom_pandasr   daskconfigsetZ__dask_optimize__Z__dask_keys__lenAssertionError)dfsr   r	    r   P/tmp/pip-unpacked-wheel-dbjnr7gq/dask/dataframe/tests/test_optimize_dataframe.pytest_fuse_ave_width   s    r    c                  C   sr   ddl m}  ttdtdd}tj|dd}tdD ]}|jd |j |d< q:| |j	}t
|d	ksntd S )
Nr   )optimize_blockwiser   )r   yr   r   r   r   r   )Zdask.array.optimizationr!   r   r   r   r   r   r   r"   r   r   r   )r!   r   Zddf_graphr   r   r   test_optimize_blockwise   s    
r%   )Zpandasr   r   Zdask.dataframeZ	dataframer   r   Zdsklistvaluesdfsr    r%   r   r   r   r   <module>   s   $$$