o
    :ήc                     @   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mZ e
 dd Zdd	 Zed
d dgdd Zdd Zdd Zdd ZdS )    N)data)felzenszwalb)testing)assert_greatertest_parallelassert_equalassert_array_equalassert_warnsassert_no_warningsexpected_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      ?   )bins(   )	npzerosr   r   lenuniquerange	histogramr   )imgsegihist r    S/tmp/pip-target-vg8gfxp4/lib/python/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   coinsr   aranger   bincountravelr   mincoffee)r   r'   segmentscountsr-   r    r    r!   test_minsize   s   r0   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   r2   )r1   )
r   r   moveaxisr
   r   r	   RuntimeWarningr   raises
ValueError)r1   grey_imgrgb_imgthree_d_imgr    r    r!   test_3D-   s    



"r;   c                  C   s   t d} tdg t| dd W d    n1 sw   Y  tdg t| dddd W d    d S 1 s9w   Y  d S )	Nr3   z'`multichannel` is a deprecated argumentT)multichannelz%Providing the `multichannel` argumentr   g?   )r   r   r   r   )r9   r    r    r!    test_3D_multichannel_deprecation?   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    r!   
test_colorG   s   
r@   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   r?   r    r    r!   test_mergingV   s
   rC   )numpyr   skimager   skimage.segmentationr   skimage._sharedr   skimage._shared.testingr   r   r   r   r	   r
   r   r"   r0   parametrizer;   r>   r@   rC   r    r    r    r!   <module>   s    $

