U
    h	                     @   sX   d dl Zd dlmZ d dlmZ d dlmZ dd Zdd Z	d	d
 Z
dd Zdd ZdS )    N)assert_array_equal)testing)compare_imagesc               	   C   sD   t jdt jd} t jdt jd}tt t| | W 5 Q R X d S )N
   r   dtype)r      )npzerosuint8r   Zraises
ValueErrorr   )img1img2 r   C/tmp/pip-unpacked-wheel-gtraehzf/skimage/util/tests/test_compare.py$test_compate_images_ValueError_shape	   s    r   c                  C   s   t jdt jd} d| ddddf< t | }d|ddddf< t j| t jd}d|ddddf< t| |dd	}t|| d S )
Nr   r            r   r	   Zdiffmethodr
   r   r   Z
zeros_likeZfloat64r   r   r   r   Zexpected_resultresultr   r   r   test_compare_images_diff   s    
r   c                  C   s   t jdt jd} d| ddddf< t | }d|ddddf< t j| t jd}d|ddddf< d|ddddf< t| |d	d
}t|| d S )Nr   r   r   r   r   r   r	   g      ?Zblendr   r   r   r   r   r   test_compare_images_blend   s    
r   c                  C   s   t jdt jd} t j| jdt jd}t| |dd}t ddddddddddddddddg}t ddddddddddddddddg}d	D ]}t||d d f | qd
D ]}t||d d f | qd S )N   r   r   r   Z
fill_valuer   checkerboardr                 ?)r   r	         r   	         )   r         r            r
   r   r   fullshaper   arrayr   r   r   resZexp_row1Zexp_row2ir   r   r   (test_compare_images_checkerboard_default'   s    **r5   c                  C   s   t jdt jd} t j| jdt jd}t| |ddd}t ddd	d	ddd	d	ddd	d	ddd	d	g}t d	d	ddd	d	ddd	d	ddd	d	ddg}d
D ]}t||d d f | qdD ]}t||d d f | qd S )Nr   r   r   r   r    )r#   r   )r   Zn_tilesr!   r"   )r   r	   r(   r   r   r%   r   r+   )r#   r$   r)   r*   r&   r'   r,   r-   r.   r2   r   r   r   &test_compare_images_checkerboard_tuple3   s    ""r6   )Znumpyr
   Zskimage._shared.testingr   Zskimage._sharedr   Zskimage.util.comparer   r   r   r   r5   r6   r   r   r   r   <module>   s   