a
    h                     @   sl   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d Zdd	 Zd
d Zdd Zdd ZdS )    N)connected_components)pairwise_distances)kneighbors_graph)_fix_connected_componentsc                  C   sj   t g dd d d f } t| ddd}t|\}}|dks@J t| |||}t|\}}|dksfJ d S )Nr                  r   distanceZn_neighborsmoder   )nparrayr   r   r   Xgraphn_connected_componentslabels r   Z/var/www/html/assistant/venv/lib/python3.9/site-packages/sklearn/utils/tests/test_graph.pytest_fix_connected_components
   s    r   c                  C   s   t g dd d d f } t| ddd}t|\}}|dks@J t| }t||||dd}t|\}}|dksrJ tjtdd	" t||||dd W d    n1 s0    Y  d S )
Nr   r   r   r   r   Zprecomputed)Zmetriczdoes not work with a sparsematch)	r   r   r   r   r   r   pytestraisesRuntimeError)r   r   r   r   Z	distancesr   r   r   )test_fix_connected_components_precomputed   s    

r   c                  C   sx   t g dd d d f } t| ddd}t|\}}tjtdd" t| |||dd}W d    n1 sj0    Y  d S )	Nr   r   r   r   zUnknown moder   Zfoor   )r   r   r   r   r   r   
ValueErrorr   r   r   r   r   (test_fix_connected_components_wrong_mode/   s    
r!   c                  C   s^   t g dd d d f } t| ddd}t|\}}t| |||dd}t |jdksZJ d S )Nr   r   r
   r   r   Zconnectivityr   r   )r   r   r   r   r   alldatar   r   r   r   /test_fix_connected_components_connectivity_mode;   s    
r%   c                  C   sr   t g dd d d f } t| ddd}t |jdks<J t|\}}t| |||dd}t |jdkrnJ d S )Nr"   r   r   r   r   )r   r   r   r#   r$   r   r   r   r   r   r   +test_fix_connected_components_distance_modeF   s    
r&   )numpyr   r   Zscipy.sparse.csgraphr   Zsklearn.metrics.pairwiser   Zsklearn.neighborsr   Zsklearn.utils.graphr   r   r   r!   r%   r&   r   r   r   r   <module>   s   