
    Uil                     ^    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dS )    N)assert_equalassert_array_almost_equal)csgraph	csr_arrayc                      t          j        g dg dg dg          } t          j        | d          }|| fD ]>}t          j        |dd          \  }}t          |d           t          |g d	           ?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labelss        /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/scipy/sparse/csgraph/tests/test_connected_components.pytest_weak_connectionsr      s    
(IIIIIII    C 
$SQ
7
7
7C#X 5 5(T4:< < < 	f 	\1%%%!&)))44445 5    c                     t          j        g dg dg dg          } | | j        z   }t          j        | d          }t          j        |d          }|| fD ]R}t          j        |dd          \  }}t          |d           |                                 t          |g d	           S||fD ]R}t          j        |dd          \  }}t          |d
           |                                 t          |g d           Sd 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   s          r   test_strong_connectionsr(      sI   8YYYYYYY  ! !D $&=D%dq999D%dq999D4Z 5 5(T4<> > > 	f 	\1%%%!&)))44444Z 5 5(T4<> > > 	f 	\1%%%!&)))44445 5r   c            	          t          j        g dg dg dg dg dg dg          } t          j        | 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   s      r   test_strong_connections2r2   2   s    
$$$$$$$$$$$$$$$$$$& 	' 	'A 	$Q08	: 	: 	: L& q!!!
KKMMMf&8&8&899999r   c            	          t          j        g dg dg dg dg dg dg          } t          j        | 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	   r0   r1   s      r   test_weak_connections2r4   A   s    
$$$$$$$$$$$$$$$$$$& 	' 	'A 	$Q06	8 	8 	8 L& q!!!
KKMMMf&8&8&899999r   c                     t          j        g dg dg dg dg          } t          j        | 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   s      r   test_ticket1876r8   P   s     	,,,,,,,,,  	! 	!A #7hOOOL&q!!!F1I&&&F1I&&&&&r   c                  |    t          j        d          } t          j        |           \  }}t	          |d           d S )N)r/   r/   r	   )r   onesr   r   r   r6   s      r   test_fully_connected_graphr;   ^   s<     	A"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           d S )Nr	   r   dtyper   r   shapeF)r   	r   r   r   int64indicesr>   r   r   r   r7   nr   s      r   test_int64_indices_undirectedrG   f   s    A31#s28<<<=VLLLA9?bh&&&&,Q???IAv6666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           d S )
Nr	   r   r=   r?   r@   Tr    r   r   rB   rE   s      r   test_int64_indices_directedrI   o   s    A31#s28<<<=VLLLA9?bh&&&&,Q8@B B BIAv6666fq!f-----r   )numpyr   numpy.testingr   r   scipy.sparser   r   r   r(   r2   r4   r8   r;   rG   rI    r   r   <module>rN      s        A A A A A A A A + + + + + + + +5 5 5 5 5 58: : :: : :' ' '" " ". . .. . . . .r   