o
    ßË·ed  ã                   @   s   d dl ZG dd„ dƒZdS )é    Nc                   @   s<   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ ZdS )ÚTestClosenessVitalityc                 C   s,   t  d¡}t  |¡}|ddddœksJ ‚d S )Né   é   ©r   é   r   )ÚnxÚcycle_graphÚcloseness_vitality©ÚselfÚGÚvitality© r   ú^/var/www/ideatree/venv/lib/python3.10/site-packages/networkx/algorithms/tests/test_vitality.pyÚtest_unweighted   s   

z%TestClosenessVitality.test_unweightedc                 C   sB   t  ¡ }t j|g d¢dd t j|dd}|ddddœksJ ‚d S )Nr   r   ©Úweightr   é   )r   ÚGraphÚ	add_cycler	   r
   r   r   r   Útest_weighted
   s   z#TestClosenessVitality.test_weightedc                 C   s2   t  t  d¡¡}t  |¡}|ddddœksJ ‚d S )Nr   r   r   )r   ÚDiGraphr   r	   r
   r   r   r   Útest_unweighted_digraph   s   
z-TestClosenessVitality.test_unweighted_digraphc                 C   óV   t  ¡ }t j|g d¢dd t j|g d¢dd t j|dd}|ddddœks)J ‚d S ©Nr   r   r   )r   r   r   r   é   )r   r   r   r	   r
   r   r   r   Útest_weighted_digraph   ó
   z+TestClosenessVitality.test_weighted_digraphc                 C   r   r   )r   ÚMultiDiGraphr   r	   r
   r   r   r   Útest_weighted_multidigraph   r   z0TestClosenessVitality.test_weighted_multidigraphc                 C   s*   t  d¡}t j|ddtdƒ ksJ ‚dS )zwTests that the closeness vitality of a node whose removal
        disconnects the graph is negative infinity.

        r   r   )ÚnodeÚinfN)r   Ú
path_graphr	   Úfloat)r   r   r   r   r   Útest_disconnecting_graph#   s   
 z.TestClosenessVitality.test_disconnecting_graphN)	Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r$   r   r   r   r   r      s    r   )Únetworkxr   r   r   r   r   r   Ú<module>   s    