
    0Ph                     d    d dl Zd dlZd dlmZ d dlmZ d dlmZ d dl	m
Z
 d Zd Zd Zd	 Zd
 ZdS )    N)connected_components)pairwise_distances)kneighbors_graph)_fix_connected_componentsc                      t          j        g d          d d d f         } t          | dd          }t          |          \  }}|dk    sJ t	          | |||          }t          |          \  }}|dk    sJ d S )Nr                  r
   distancen_neighborsmoder	   )nparrayr   r   r   Xgraphn_connected_componentslabelss       ^/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/sklearn/utils/tests/test_graph.pytest_fix_connected_componentsr   
   s    
###$$QQQW-AQAJ???E%9%%@%@"F!A%%%%%a0FOOE%9%%@%@"F!Q&&&&&&    c                     t          j        g d          d d d f         } t          | dd          }t          |          \  }}|dk    sJ t	          |           }t          ||||d          }t          |          \  }}|dk    sJ t          j        t          d	          5  t          ||||d           d d d            d S # 1 swxY w Y   d S )
Nr   r
   r   r   r	   precomputed)metriczdoes not work with a sparsematch)	r   r   r   r   r   r   pytestraisesRuntimeError)r   r   r   r   	distancess        r   )test_fix_connected_components_precomputedr%      sO   
###$$QQQW-AQAJ???E%9%%@%@"F!A%%%%"1%%I%50&  E &:%%@%@"F!Q&&&& 
|+H	I	I	I 
 
!50&	
 	
 	
 	

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
s   %CCCc                     t          j        g d          d d d f         } t          | dd          }t          |          \  }}t	          j        t          d          5  t          | |||d          }d d d            d S # 1 swxY w Y   d S )	Nr   r
   r   r   zUnknown moder   foor   )r   r   r   r   r!   r"   
ValueErrorr   r   s       r   (test_fix_connected_components_wrong_moder*   /   s    
###$$QQQW-AQAJ???E%9%%@%@"F	z	8	8	8 
 
)u,f5
 
 

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
s    BB	Bc                      t          j        g d          d d d f         } t          | dd          }t          |          \  }}t	          | |||d          }t          j        |j        dk              sJ d S )Nr   r	   r   r   r	   connectivityr   r(   )r   r   r   r   r   alldatar   s       r   /test_fix_connected_components_connectivity_moder0   ;   s    
qqq$w'AQANCCCE%9%%@%@"F%	5(&~  E 6%*/"""""""r   c                  2   t          j        g d          d d d f         } t          | dd          }t          j        |j        dk              sJ t          |          \  }}t          | |||d          }t          j        |j        dk              rJ d S )Nr,   r	   r   r   r(   )r   r   r   r.   r/   r   r   r   s       r   +test_fix_connected_components_distance_moder2   F   s    
qqq$w'AQAJ???E6%*/"""""%9%%@%@"F%	5(&z  E vejAo&&&&&&&r   )numpyr   r!   scipy.sparse.csgraphr   sklearn.metrics.pairwiser   sklearn.neighborsr   sklearn.utils.graphr   r   r%   r*   r0   r2    r   r   <module>r9      s         5 5 5 5 5 5 7 7 7 7 7 7 . . . . . . 9 9 9 9 9 9' ' '
 
 
.	
 	
 	
# # #
' 
' 
' 
' 
'r   