
    fid                     (    d dl Z G d d          ZdS )    Nc                   2    e Zd Zd Zd Zd Zd Zd Zd ZdS )TestClosenessVitalityc                 n    t          j        d          }t          j        |          }|ddddk    sJ d S )N      r      r   )nxcycle_graphcloseness_vitalityselfGvitalitys      /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/networkx/algorithms/tests/test_vitality.pytest_unweightedz%TestClosenessVitality.test_unweighted   sB    N1(++qQ1--------    c                     t          j                    }t          j        |g dd           t          j        |d          }|ddddk    sJ d S )Nr   r   weightr      )r
   Graph	add_cycler   r   s      r   test_weightedz#TestClosenessVitality.test_weighted
   s^    HJJ
Q			!,,,,(8<<<qQ1--------r   c                     t          j        t          j        d                    }t          j        |          }|ddddk    sJ d S )Nr   r   r   )r
   DiGraphr   r   r   s      r   test_unweighted_digraphz-TestClosenessVitality.test_unweighted_digraph   sL    Jr~a(())(++qQ1--------r   c                     t          j                    }t          j        |g dd           t          j        |g dd           t          j        |d          }|ddddk    sJ d S Nr   r   r   )r   r	   r   r      )r
   r   r   r   r   s      r   test_weighted_digraphz+TestClosenessVitality.test_weighted_digraph   sy    JLL
Q			!,,,,
Q			!,,,,(8<<<qQ1--------r   c                     t          j                    }t          j        |g dd           t          j        |g dd           t          j        |d          }|ddddk    sJ d S r   )r
   MultiDiGraphr   r   r   s      r   test_weighted_multidigraphz0TestClosenessVitality.test_weighted_multidigraph   s{    O
Q			!,,,,
Q			!,,,,(8<<<qQ1--------r   c                     t          j        d          }t          j        |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   s     r   test_disconnecting_graphz.TestClosenessVitality.test_disconnecting_graph#   sB    
 M!$QQ///E%LL=@@@@@@r   N)	__name__
__module____qualname__r   r   r   r!   r$   r*    r   r   r   r      ss        . . .
. . .. . .
. . .. . .A A A A Ar   r   )networkxr
   r   r.   r   r   <module>r0      sM       %A %A %A %A %A %A %A %A %A %Ar   