o
    ]Zh/                     @   s.   d Z ddlZddlmZmZ G dd dZdS )z
ego graph
---------
    N)edges_equalnodes_equalc                   @   s   e Zd Zdd Zdd ZdS )TestGeneratorEgoc                 C   s   t d}t |d}t ||sJ |dd |dd |dd t |d}t t d|s6J t d}t |d}t| dgsKJ t j|dd	d
}t| dgs]J t j|ddd}t| g snJ d S )N   r               !   )r   r   T)
undirectedF)center)nxZ
star_graph	ego_graphZis_isomorphicadd_edgeZ
path_graphr   edges)selfGH r   Q/var/www/auris/lib/python3.10/site-packages/networkx/generators/tests/test_ego.pytest_ego   s   

zTestGeneratorEgo.test_egoc                 C   s   t  }|jddddd |jddddd |jddddd tt j|ddd g ds0J t j|dddd	}t| ddgsDJ t j|dddd
d}t| ddgsYJ t j|dddd	}t| g dsmJ d S )Nr   r   r   )weightdistancer   )radius)r   r   r   r   r   )r   r   T)r   r   r   r   )r   r   r   )r   ZGraphr   r   r   nodes)r   r   egr   r   r   test_ego_distance   s   "z"TestGeneratorEgo.test_ego_distanceN)__name__
__module____qualname__r   r   r   r   r   r   r   
   s    r   )__doc__Znetworkxr   Znetworkx.utilsr   r   r   r   r   r   r   <module>   s    