o
    i                     @   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   sh   t g dg dg dg} tj| dd}|| fD ]}tj|ddd\}}t|d t|g d	 qd S )
Nr      r   r   r   r   r   
null_valueTweakdirected
connection   r   r   r   )nparrayr   csgraph_from_denseconnected_componentsr   r   )XdeXspXn_componentslabels r   o/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/scipy/sparse/csgraph/tests/test_connected_components.pytest_weak_connections   s   

r   c                  C   s   t g dg dg dg} | | j }tj| dd}tj|dd}|| fD ]}tj|ddd\}}t|d |  t|g d	 q%||fD ]}tj|ddd\}}t|d
 |  t|g d qFd S )Nr   r   r   r   Tstrongr      )r   r   r   r   r   )	r   r   Tr   r   r   r   sortr   )X1deX2deX1spX2spr   r   r   r   r   r   test_strong_connections   s0   



r$   c               	   C   f   t g dg dg dg dg dg dg} tj| ddd\}}t|d	 |  t|g 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   r   r   r   r   r   r   r   r   Tr   r      )r   r   r   r   r      r   r   r   r   r   r   r   r   r   r   r   r   r   test_strong_connections22      

r.   c               	   C   r%   )Nr&   )r   r   r   r   r   r   r'   r(   r)   Tr
   r   r   )r   r   r   r   r   r   r,   r-   r   r   r   test_weak_connections2A   r/   r0   c                  C   sf   t g dg dg dg dg} tj| dd\}}t|d t|d |d	  t|d |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   )r   r   r   r   r   gr   r   r   r   r   test_ticket1876P   s   

r3   c                  C   s&   t d} t| \}}t|d d S )N)r+   r+   r   )r   onesr   r   r   r1   r   r   r   test_fully_connected_graph^   s   
r5   )numpyr   numpy.testingr   r   scipy.sparser   r   r$   r.   r0   r3   r5   r   r   r   r   <module>   s    