U
    ?hd                     @   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(td S )N      r      r   )nxcycle_graphcloseness_vitalityAssertionErrorselfGZvitality r   Y/var/www/html/venv/lib/python3.8/site-packages/networkx/algorithms/tests/test_vitality.pytest_unweighted   s    

z%TestClosenessVitality.test_unweightedc                 C   sD   t  }t j|dddgdd t j|dd}|ddddks@td S )Nr   r   r   weightr      r   )r   ZGraph	add_cycler	   r
   r   r   r   r   test_weighted
   s    z#TestClosenessVitality.test_weightedc                 C   s2   t t d}t |}|ddddks.td S )Nr   r   r   )r   DiGraphr   r	   r
   r   r   r   r   test_unweighted_digraph   s    
z-TestClosenessVitality.test_unweighted_digraphc                 C   sZ   t  }t j|dddgdd t j|dddgdd t j|dd}|ddddksVtd S Nr   r   r   r   r      r   )r   r   r   r	   r
   r   r   r   r   test_weighted_digraph   s
    z+TestClosenessVitality.test_weighted_digraphc                 C   sZ   t  }t j|dddgdd t j|dddgdd t j|dd}|ddddksVtd S r   )r   ZMultiDiGraphr   r	   r
   r   r   r   r   test_weighted_multidigraph   s
    z0TestClosenessVitality.test_weighted_multidigraphc                 C   s*   t d}t j|ddtd ks&tdS )zwTests that the closeness vitality of a node whose removal
        disconnects the graph is negative infinity.

        r   r   )nodeinfN)r   Z
path_graphr	   floatr
   )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   )Znetworkxr   r   r   r   r   r   <module>   s   