U
    /e                     @   sz   d dl Z d dlmZ d dlZed d dlZd dlmZ d dl	m
Z d dlmZ edd Zd	d
 Zdd Zdd ZdS )    N)contextmanagerZskimage)imsaveimread)tmpdirc              	   c   sl   t  \}t| D ]<}tj|d| }tjjdd|dd}t	||dd qtj|dV  W 5 Q R X d S )	Nzimage.%d.pngr      )sizeu1F)Zcheck_contrastz*.png)
r   rangeospathjoinnprandomrandintZastyper   )nshapedirnameifnx r   ?/tmp/pip-unpacked-wheel-dbjnr7gq/dask/array/tests/test_image.pyrandom_images   s    r   c               	   C   sp   t dd\} t| }|jdks"t|jdks0t|jdks>t| jdksPt| jdksbtW 5 Q R X d S )N            )r   r   r   r   ))   r   r   r   )r   )r   )r   Zuint8)r   	da_imreadr   AssertionErrorchunksdtypecompute)
globstringimr   r   r   test_imread   s    r'   c               	   C   sL   dd } t dd0}t|| d}| tjdddk s>tW 5 Q R X d S )	Nc                 S   s   t jdddS )N   r   r   i1r#   )r   ones)r   r   r   r   imread2%   s    z1test_imread_with_custom_function.<locals>.imread2r   r   r   )r   r)   r   r   r	   r+   r   r    r$   r   r,   allr!   )r-   r%   r&   r   r   r    test_imread_with_custom_function$   s    r0   c               	   C   sL   dd } t dd0}t|| d}| tjdddk s>tW 5 Q R X d S )	Nc                 S   s"   d| d d < | d d d d df S )Nr   r   r   )r   r   r   r   
preprocess.   s    z#test_preprocess.<locals>.preprocessr   r(   )r1   )r   r)   r   r	   r+   r.   )r1   r%   r&   r   r   r   test_preprocess-   s    r2   )r   
contextlibr   ZpytestZimportorskipZnumpyr   Z
skimage.ior   Zdask.array.imager   r    Z
dask.utilsr   r   r'   r0   r2   r   r   r   r   <module>   s   


	