a
    Rel                     @   sl   d dl Zd dlmZmZ d dlmZmZ dd Zdd 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)csgraph	csr_arrayc                  C   sh   t g dg dg dg} tj| dd}|| fD ]0}tj|ddd\}}t|d t|g d	 q2d S )
Nr      r   r   r   r   r   Z
null_valueTweakdirected
connection   r   r   r   )nparrayr   csgraph_from_denseconnected_componentsr   r   )ZXdeZXspXn_componentslabels r   h/var/www/sunrise/env/lib/python3.9/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 ]8}tj|ddd\}}t|d |  t|g d	 qJ||fD ]8}tj|ddd\}}t|d
 |  t|g d qd S )Nr   r   r   r	   Tstrongr      )r   r   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   sf   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   s    

r'   c               	   C   sf   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 )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   s    

r(   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    

r+   c                  C   s&   t d} t| \}}t|d d S )N)r$   r$   r   )r   Zonesr   r   r   r)   r   r   r   test_fully_connected_graph^   s    
r,   c                  C   sh   t dgtjdgdggtjdfdd} | jjtjks8J tj| dd\}}|dksVJ t|ddg d S )Nr   r   dtyper   r   shapeF)r   	r   r   r   int64indicesr.   r   r   r   r*   nr   r   r   r   test_int64_indices_undirectedf   s
    &r7   c                  C   sj   t dgtjdgdggtjdfdd} | jjtjks8J tj| ddd\}}|d	ksXJ t|ddg d S )
Nr   r   r-   r/   r0   Tr   r   r   r2   r5   r   r   r   test_int64_indices_directedo   s    &
r8   )numpyr   Znumpy.testingr   r   Zscipy.sparser   r   r   r   r'   r(   r+   r,   r7   r8   r   r   r   r   <module>   s   	