U
    ?h/  ã                   @   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&t‚| dd¡ | dd¡ | dd¡ t  |d¡}t  t  d¡|¡slt‚t  d¡}t  |d¡}t| ¡ dgƒs–t‚t j|dd	d
}t| ¡ dgƒsºt‚t j|ddd}t| ¡ g ƒsÜt‚d S )Né   r   é   é   é   é   é!   )r   r   T)Ú
undirectedF)Úcenter)	ÚnxZ
star_graphÚ	ego_graphZis_isomorphicÚAssertionErrorÚadd_edgeZ
path_graphr   Úedges)ÚselfÚGÚH© r   úT/var/www/html/venv/lib/python3.8/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 ¡ ddddgƒsdt‚t j|dddd}t| ¡ ddgƒsŒt‚t j|dddd	d
}t| ¡ ddgƒs¶t‚t j|dddd}t| ¡ dddgƒsàt‚d S )Nr   r   r   )ÚweightÚdistancer   )Úradiusr   )r   r   T)r   r   r   r   )r   ZGraphr   r   r   Znodesr   )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   