o
    ßË·e/  ã                   @   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)ÚnxÚ
star_graphÚ	ego_graphÚis_isomorphicÚadd_edgeÚ
path_graphr   Úedges)ÚselfÚGÚH© r   úY/var/www/ideatree/venv/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   Ú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__Únetworkxr   Únetworkx.utilsr   r   r   r   r   r   r   Ú<module>   s    