U
    ƒ±Ëh†  ã                   @   sh   d dl Zd dlmZ d dlmZmZ dd„ Zdd„ Zdd	„ Z	d
d„ Z
dd„ Zdd„ Zdd„ Zdd„ ZdS )é    N)Úclear_border)Úassert_array_equalÚassert_c                  C   sÌ  t  dddddddddg	dddddddddg	dddddddddg	dddddddddg	dddddddddg	dddddddddg	g¡} t|  ¡ ƒ}|  ¡ }d|dd…dd…f< d|dd…df< t||ƒ t|  ¡ dƒ}t|t  |j¡ƒ t|  ¡ ddd}t|dt  | ¡ ƒ t  dddddddddg	dddddddddg	dddddddddg	dddddddddg	dddddddddg	dddddddddg	g¡ t	¡}t|  ¡ |d}|  ¡ }d|dd…dd…f< t||ƒ d S )Nr   é   é   é   éþÿÿÿ©Úbuffer_sizeZbgval)Úmask)
ÚnpÚarrayr   Úcopyr   ÚzerosÚshapeÚ	ones_likeZastypeÚbool)ÚimageÚresultÚrefr   © r   úP/tmp/pip-unpacked-wheel-gtraehzf/skimage/segmentation/tests/test_clear_border.pyÚtest_clear_border   s>    ûÿ	
ûûr   c                  C   sø   t  ddddgddddgddddgddddggddddgddddgddddgddddggddddgddddgddddgddddggg¡} t|  ¡ ƒ}|  ¡ }d|d< t||ƒ t|  ¡ dƒ}t|t  |j¡ƒ t|  ¡ ddd}t|dt  | ¡ ƒ d S )Nr   r   )r   r   r   r   r	   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   Útest_clear_border_3d,   s2    



ý



ý



ý÷
r   c               
   C   sž   t  dddddgdddddgdddddgdddddgg¡} t| ƒ}t  dddddgdddddgdddddgdddddgg¡}t||ƒ tt  | |k¡ ƒ d S )Nr   r   r   é   é   r   ©r   r   r   r   r   Úall©r   r   Úexpectedr   r   r   Útest_clear_border_non_binaryJ   s    ýý
r    c                  C   sj  t  dddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddggg¡} t| ƒ}t  dddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddggg¡}t||ƒ tt  | |k¡ ƒ d S )Nr   r   r   r   r   r   r   ©Úimage3dr   r   r   r   r   Útest_clear_border_non_binary_3dZ   sN    ýýýøÿýýýøÿ
r#   c               
   C   s˜   t  dddddgdddddgdddddgdddddgg¡} t| | d}t  dddddgdddddgdddddgdddddgg¡}t||ƒ t| |ƒ d S ©Nr   r   r   r   r   ©Úoutr   ©r   r   r   r   r   r   r   r   Ú$test_clear_border_non_binary_inplace~   s    ýý
r(   c                  C   sd  t  dddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddggg¡} t| | d}t  dddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddggg¡}t||ƒ t| |ƒ d S ©Nr   r   r   r   r   r%   r   r'   r!   r   r   r   Ú'test_clear_border_non_binary_inplace_3dŽ   sN    ýýýøÿýýýøÿ
r*   c               
   C   s¢   t  dddddgdddddgdddddgdddddgg¡} t  | ¡}t| |d}t  dddddgdddddgdddddgdddddgg¡}t||ƒ t||ƒ d S r$   ©r   r   Z
empty_liker   r   )r   r&   r   r   r   r   r   Ú test_clear_border_non_binary_out²   s    ý
ý
r,   c                  C   sn  t  dddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddggg¡} t  | ¡}t| |d}t  dddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddggg¡}t||ƒ t||ƒ d S r)   r+   )r"   r&   r   r   r   r   r   Ú#test_clear_border_non_binary_out_3dÂ   sP    ýýýøÿ
ýýýøÿ
r-   )Znumpyr   Zskimage.segmentationr   Zskimage._shared.testingr   r   r   r   r    r#   r(   r*   r,   r-   r   r   r   r   Ú<module>   s   %$$