o
    :ήc                     @   s   d dl Zd dlmZmZ ejg dg dg dgedZej	 ej
Zdd Zd	d
 Zdd Zdd Zedkr>e  dS dS )    N)pixel_graphcentral_pixel)   r   r   )r   r   r   )r   r   r   )dtypec                  C   sf   t tdd\} }| jdksJ t| jdksJ tjt| jdt	dg tj
|g d d S )N   connectivity)   r	      r   r   r	         )r   maskshapelendatanptestingassert_allcloseuniquesqrtassert_array_equal)gn r   K/tmp/pip-target-vg8gfxp4/lib/python/skimage/graph/tests/test_pixel_graph.pytest_small_graph   s
    r   c                  C   s   t tdd\} }t| |tjd\}}tj|d td}tj||d d |d |d d |d d g t| |\}}|dksDJ t| \}}|dksPJ d S )Nr   r   )r   r   r      r	   r   )	r   r   r   r   r   r   r   r   r   )r   r   pxdss2_r   r   r   test_central_pixel   s   
2r#   c                  C   s   dd } t ttd| d\}}td}tj|d ttd td  |  tj|d ttd td  d	  tj|g d
 d S )Nc                 S   s   t | | | S )N)r   abs)
values_src
values_dst	distancesr   r   r   	edge_func!   s   z%test_edge_function.<locals>.edge_funcr   )r   r   edge_functionr   r   r   r   r   )r   r   r   r   )	r   imager   r   r   r   r   r$   r   )r(   r   r   r!   r   r   r   test_edge_function    s   

((r-   c               	   C   sv   t ttddgd\} }t| jd }|dksJ tj| d dttd td    tj	|t
tj d S )Ng(\?)spacingr      r*   r+   )r   r,   r   arrayr   r   r   assert_almost_equalr$   r   arangesize)r   r   	num_edgesr   r   r   test_default_edge_func-   s   r5   __main__)numpyr   skimage.graph._graphr   r   r0   boolr   randomdefault_rngr   r,   r   r#   r-   r5   __name__r   r   r   r   <module>   s     

