
    sgl                     \    d dl Zd dlmZmZ d dlmZmZ d Zd Z	d Z
d Zd Zd	 Zd
 Zd Zy)    N)assert_equalassert_array_almost_equal)csgraph	csr_arrayc                      t        j                  g dg dg dg      } t        j                  | d      }|| fD ]7  }t        j                  |dd      \  }}t        |d       t        |g d	       9 y )
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labelss        g/var/www/html/venv/lib/python3.12/site-packages/scipy/sparse/csgraph/tests/test_connected_components.pytest_weak_connectionsr      sx    
((I  C 
$
$SQ
7C#X 5((T4:< 	f 	\1%!&)45    c                     t        j                  g dg dg dg      } | | j                  z   }t        j                  | d      }t        j                  |d      }|| fD ]G  }t        j
                  |dd      \  }}t        |d       |j                          t        |g d	       I ||fD ]G  }t        j
                  |dd      \  }}t        |d
       |j                          t        |g d       I y )Nr   r
   r   r   Tstrongr      )r   r	   r   r   r   )	r   r   Tr   r   r   r   sortr   )X1deX2deX1spX2spr   r   r   s          r   test_strong_connectionsr(      s    88Y  !D $&&=D%%dq9D%%dq9D4Z 5((T4<> 	f 	\1%!&)45 4Z 5((T4<> 	f 	\1%!&)45r   c            	          t        j                  g dg dg dg dg dg dg      } t        j                  | dd      \  }}t	        |d	       |j                          t        |g d
       y )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   s      r   test_strong_connections2r2   2   sg    
$$$$$$& 	'A 	$$Q08	: L& q!
KKMf&89r   c            	          t        j                  g dg dg dg dg dg dg      } t        j                  | dd      \  }}t	        |d	       |j                          t        |g d
       y )Nr*   )r	   r   r   r   r   r   r+   r,   r-   Tr   r   r   )r   r   r	   r	   r	   r	   r0   r1   s      r   test_weak_connections2r4   A   sg    
$$$$$$& 	'A 	$$Q06	8 L& q!
KKMf&89r   c                      t        j                  g dg dg dg dg      } t        j                  | d      \  }}t	        |d       t	        |d   |d	          t	        |d   |d
          y )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   s      r   test_ticket1876r8   P   si     	,  	!A #77hOL&q!F1I&F1I&r   c                  v    t        j                  d      } t        j                  |       \  }}t	        |d       y )N)r/   r/   r	   )r   onesr   r   r   r6   s      r   test_fully_connected_graphr;   ^   s0     	A"77:L&q!r   c                  "   t        dgt        j                  dgdggt        j                        fd      } | j                  j
                  t        j                  k(  sJ t        j                  | d      \  }}|dk(  sJ t        |ddg       y )Nr	   r   dtyper   r   shapeF)r   	r   r   r   int64indicesr>   r   r   r   r7   nr   s      r   test_int64_indices_undirectedrG   f   su    A31#s288<=VLA99??bhh&&&,,Q?IAv6M6fq!f-r   c                  $   t        dgt        j                  dgdggt        j                        fd      } | j                  j
                  t        j                  k(  sJ t        j                  | dd      \  }}|d	k(  sJ t        |ddg       y )
Nr	   r   r=   r?   r@   Tr    r   r   rB   rE   s      r   test_int64_indices_directedrI   o   s{    A31#s288<=VLA99??bhh&&&,,Q8@BIAv6M6fq!f-r   )numpyr   numpy.testingr   r   scipy.sparser   r   r   r(   r2   r4   r8   r;   rG   rI    r   r   <module>rN      s6     A +5 58::'"..r   