U
    ?h	                  
   @   s   d dl Z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d	d
dd ddgdd Zdd ZdS )    N)
quickshift)testing)assert_greaterrun_in_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	dg t|dddd	dd W 5 Q R X t
tt |d tdD ]6}t j|||k dddddgdd }t|| d qd S )Nr   )      g?
   g?g333333?g?sizeFcopy      )kernel_sizemax_distrngconvert2labsigmaz&`random_seed` is a deprecated argument)r   r   Zrandom_seedr   r      g?g333333?g      ?   )Zbinsr	   )nprandomdefault_rngzerosnormalshapeastyper   r   Zexpected_warningsr   lenuniquerangeZ	histogramr   )r   r   imgsegihist r'   \/var/www/html/venv/lib/python3.8/site-packages/skimage/segmentation/tests/test_quickshift.py	test_grey	   s(    

 
 $r)   channel_axisr   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sbt|dd d f |dd d f k st|d d df |d d df k std S )Niai")r	   r
   r   r   r   r   r   g{Gz?r   Fr   r-   )sourceZdestination   )r   r   r   r   r*   r   r   )r   r   r   r   r   r*   	   )r   r   r   r   r   r   r   Zmoveaxisr   r   r    r!   r   AssertionErrorall)r   r*   r   r#   r$   Zseg2r'   r'   r(   
test_color    s4    

 *r3   c               	   C   s4   t d} tjtdd t| dd W 5 Q R X d S )N)r	   r
   r   z-Only RGB images can be converted to Lab space)matchT)r   )r   r   pytestZraises
ValueErrorr   )r#   r'   r'   r(   test_convert2lab_not_rgbA   s    
 r7   )numpyr   r5   Zskimage.segmentationr   Zskimage._sharedr   Zskimage._shared.testingr   r   r   r   ZparametrizeZfloat32Zfloat64r)   r3   r7   r'   r'   r'   r(   <module>   s   