3
Od                 @   sX   d dl Zd dlmZmZ d dlmZ dd Zdd Zdd	 Z	d
d Z
dd Zdd ZdS )    N)assert_equalassert_array_almost_equal)csgraphc              C   st   t jdddgdddgdddgg} tj| dd}x>|| fD ]2}tj|ddd\}}t|d t|dddg q:W d S )Nr      )
null_valueTweak)directed
connection   )nparrayr   csgraph_from_denseconnected_componentsr   r   )ZXdeZXspXn_componentslabels r   h/var/www/html/virt/lib64/python3.6/site-packages/scipy/sparse/csgraph/tests/test_connected_components.pytest_weak_connections   s    
r   c              C   s   t jdddgdddgdddgg} | | j }tj| dd}tj|dd}xF|| fD ]:}tj|ddd\}}t|d |j  t|dddg qRW xF||fD ]:}tj|ddd\}}t|d |j  t|dddg qW d S )	Nr   r   )r   Tstrong)r   r	      r
   )	r   r   Tr   r   r   r   sortr   )ZX1deZX2deZX1spZX2spr   r   r   r   r   r   test_strong_connections   s$    


r   c              C   s   t jddddddgddddddgddddddgddddddgddddddgddddddgg} tj| ddd\}}t|d |j  t|dddddd	g d S )
Nr   r   Tr   )r   r	      r
   r      )r   r   r   r   r   r   r   )r   r   r   r   r   r   test_strong_connections22   s    
r   c              C   s   t jddddddgddddddgddddddgddddddgddddddgddddddgg} tj| ddd\}}t|d |j  t|ddddddg d S )Nr   r   Tr   )r   r	   r
   )r   r   r   r   r   r   r   )r   r   r   r   r   r   test_weak_connections2A   s    
r   c              C   sv   t jddddgddddgddddgddddgg} tj| dd\}}t|d t|d |d  t|d |d  d S )Nr   r   r   )r	   r
   r   )r   r   r   r   r   )gr   r   r   r   r   test_ticket1876P   s    


r   c              C   s&   t jd} tj| \}}t|d d S )Nr   r   )r   r   )r   Zonesr   r   r   )r   r   r   r   r   r   test_fully_connected_graph^   s    
r    )Znumpyr   Znumpy.testingr   r   Zscipy.sparser   r   r   r   r   r   r    r   r   r   r   <module>   s   