o
    ç^iT	  ã                   @   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g d	¢¡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ddd	dd tt	t  
|¡ƒdƒ tdƒD ]}t j|||k g d¢dd }t|| dƒ q]d S )Nr   )é   é   gš™™™™™É?é
   gš™™™™™Ù?g333333ã?gš™™™™™©?©ÚsizeF©Úcopyé   é   )Úkernel_sizeÚmax_distÚrngÚconvert2labÚsigmaé   )r   gš™™™™™¹?g333333Ó?g      à?é   )Úbinsr	   )ÚnpÚrandomÚdefault_rngÚzerosÚnormalÚshapeÚastyper   r   ÚlenÚuniqueÚrangeÚ	histogramr   )r   r   ÚimgÚsegÚiÚhist© r)   úe/var/www/html/RAG/RAG_venv/lib/python3.10/site-packages/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s°J ‚|dd d …f |dd d …f k ¡ sÄJ ‚|d d …df |d d …df k ¡ sØJ ‚d S )NiaiÆ")r	   r
   r   r   r   r   r   g{®Gáz„?r   Fr   r/   )Ú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"   s<   
ÿù
(,r7   c                  C   sJ   t  d¡} tjtdd t| dd W d   ƒ d S 1 sw   Y  d S )N)r	   r
   r   z-Only RGB images can be converted to Lab space)ÚmatchT)r   )r   r   ÚpytestÚraisesÚ
ValueErrorr   )r%   r)   r)   r*   Útest_convert2lab_not_rgbJ   s   
ÿ"ýr<   )Únumpyr   r9   Úskimage.segmentationr   Úskimage._sharedr   Úskimage._shared.testingr   r   r   r   ÚparametrizeÚfloat32Úfloat64r+   r7   r<   r)   r)   r)   r*   Ú<module>   s    &