o
    e                     @   s   d dl Zd dlmZ d dlmZ d dlmZ d dlm	Z	m
Z
mZmZmZmZ e
 dd Zdd	 Zed
d dgdd Zdd Zdd ZdS )    N)data)felzenszwalb)testing)assert_greaterrun_in_parallelassert_equalassert_array_equalassert_warnsassert_no_warningsc                  C   s   t d} d| d ddd f< d| dd d df< d| dd dd f< t| dd}ttt |d tdD ]}t j| ||k g d	d
d }t|| d q7d S )N)      g?
   g?g333333?r   sigma   )r   g?333333?g      ?   )Zbins(   )	npzerosr   r   lenuniquerangeZ	histogramr   )imgsegihist r   \D:\Projects\ConvertPro\env\Lib\site-packages\skimage/segmentation/tests/test_felzenszwalb.py	test_grey   s   
r   c                  C   s   t  ddddf } tdddD ]}t| |dd}t| }t| d	 | qt 	 d d d
d d d
f }tdddD ]}t||dd}t| }t| d	 | qBd S )Nr      r      r   d      )min_sizer   r   r   )
r   Zcoinsr   Zaranger   ZbincountZravelr   mincoffee)r   r$   segmentscountsr&   r   r   r   test_minsize   s   r)   channel_axisc                 C   s   t d}t d}t d}t |d| }t  t|dd t|d d t|| d W d    n1 s6w   Y  tt t|| d W d    n1 sPw   Y  tt	 t|d d t|d d W d    d S 1 srw   Y  d S )N)r   r   )r   r   r#   )r   r   r   r+   )r*   )
r   r   Zmoveaxisr
   r   r	   RuntimeWarningr   Zraises
ValueError)r*   Zgrey_imgZrgb_imgZthree_d_imgr   r   r   test_3D,   s    



"r.   c                  C   s   t d} d| d dd ddf< d| dd d ddf< d| dd dd df< t| 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 d S )	N)r   r   r#   r   r   r      r   r   r#   )r   r   r   r   r   r   r   r   r   r   r   r   
test_color>   s   
r1   c                  C   sj   t ddgddgg} t| dddd}ttt |d t|dd d f d t|dd d f d d S )Nr   r   gffffff?r   r/   )scaler   r$   )r   arrayr   r   r   r   r   r0   r   r   r   test_mergingM   s
   r4   )numpyr   Zskimager   Zskimage.segmentationr   Zskimage._sharedr   Zskimage._shared.testingr   r   r   r   r	   r
   r   r)   Zparametrizer.   r1   r4   r   r   r   r   <module>   s     

