o
    :ήcM                     @   s   d dl Zd dlmZ d dlmZ d dlmZmZm	Z	m
Z
 e edejejgdd Zedejejgedg d	d
d ZdS )    N)
quickshift)testing)assert_greatertest_parallelassert_equalassert_array_equaldtypec                 C   s   t jd}t d}d|d ddd f< d|dd d df< d|dd dd f< |d|j|jd 7 }|j| d	d
}t|dddd	dd}tt	t 
|d tdD ]}t j|||k g ddd }t|| d qSd S )Nr   )      g?
   g?g333333?g?sizeFcopy      )kernel_sizemax_distrandom_seedconvert2labsigma   )r   g?g333333?g      ?   )binsr	   )nprandomdefault_rngzerosnormalshapeastyper   r   lenuniquerange	histogramr   )r   rndimgsegihist r*   Q/tmp/pip-target-vg8gfxp4/lib/python/skimage/segmentation/tests/test_quickshift.py	test_grey   s   

r,   channel_axis)r   r   r   c              	   C   s  t jd}t d}d|d dd ddf< d|dd d ddf< d|dd dd df< |d|j|jd 7 }d||dk< d||dk < |j| d	d
}t j|d|d}t|dddd|d}t	t
t |d t|d dd df d t|dd d df d t|d ddd f d t|dd dd f d t|dddd	d|d}t
t |dksJ |dd d f |dd d f k sJ |d d df |d d df k sJ d S )Niai")r	   r
   r   r   r   r   r   g{Gz?r   Fr   r0   )sourcedestination   )r   r   r   r   r-   r   r   )r   r   r   r   r   r-   	   )r   r   r   r   r   r   r    moveaxisr   r   r!   r"   r   all)r   r-   r%   r&   r'   seg2r*   r*   r+   
test_color   s2   

(,r8   )numpyr   skimage.segmentationr   skimage._sharedr   skimage._shared.testingr   r   r   r   parametrizefloat32float64r,   r8   r*   r*   r*   r+   <module>   s    