U
    ?hT                     @   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 ]6}t j| ||k dd	d
ddgdd }t|| d qnd S )N)      g?
   g?g333333?r   sigma   g?333333?g      ?   )Zbins(   )	npzerosr   r   lenuniquerangeZ	histogramr   )imgsegihist r   ^/var/www/html/venv/lib/python3.8/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 ]2}t| |dd}t| }t| d	 | q&t 	 d d d
d d d
f }tdddD ]2}t||dd}t| }t| d	 | qd 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 5 Q R X tt t|| d W 5 Q R X tt	 t|d d t|d d W 5 Q R X 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    

