File: //lib/python3/dist-packages/samba/tests/kcc/__pycache__/graph_utils.cpython-310.pyc
o
�/a� � @ sh d Z ddlZddlZddlmZ ddlmZmZmZmZm Z m
Z
ddl Z dd� Z G dd� dej
j�ZdS ) z Tests for samba.kcc.graph_utils� N)�
GraphError)�verify_graph_complete�verify_graph_connected�*verify_graph_connected_under_edge_failures�verify_graph_forest�,verify_graph_connected_under_vertex_failures� verify_graph_no_lonely_verticesc C s� t | �dk rdS t| �}t� }t� }|�|�� � |�|�� � |�t|�� |rA|�� }|�� }||f}|�|� |�|� |s)t|�S )N� � )�len�set�add�pop�tuple�update)�vertices� remaining�used�edges�v�w�er
r
�=/usr/lib/python3/dist-packages/samba/tests/kcc/graph_utils.py� make_tree# s
�r c sd e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z � Z
S )�UndirectedGraphTestsc s t t| ��� td�}td�}tt�|d��}tt�|d��}tt|dd � |d d� ��}||d |d fg }t|�}t|�}|||g| _ || || || g| _
||| |g| _ |||g| _ |||g| _
|||g| _|| || || g| _d|df| _d S )N�abcdefgh�ijkr � ���r r
)�superr �setUpr � itertools� combinations�list�zipr �complete_graph�disconnected_clusters�graph_with_unreachables�ring�line�tree�forest�unconnected_graph) �selfr � vertices2r �edges2�
line_edges�
ring_edgesr* �tree2�� __class__r
r r 9 s4 �� �zUndirectedGraphTests.setUpc G s | j t|g|�R � S �N)� assertRaisesr )r- �fn�argsr
r
r �assertGraphErrorY s z%UndirectedGraphTests.assertGraphErrorc C sh t }| j|g| j�R � | j|g| j�R � | j|g| j�R � | j|g| j�R � | �|| j� � d S r5 )r r9 r& r' r( r* � assertIsNoner% �r- r7 r
r
r �test_graph_complete\ s z(UndirectedGraphTests.test_graph_completec C s� t }| j|g| j�R � | j|g| j�R � | j|g| j�R � | j|g| j�R � | �|| j� � | �|| j� � | �|| j � � | �|| j
� � d S r5 ) r r9 r& r' r+ r, r: r) r( r% r* r; r
r
r �test_graph_connectedf s z)UndirectedGraphTests.test_graph_connectedc C s� t }| j|g| j�R � | j|g| j�R � | j|g| j�R � | �|| j� � | �|| j� � | �|| j� � | �|| j � � d S r5 )
r r9 r& r' r( r: r) r* r+ r, r; r
r
r �test_graph_forests s z&UndirectedGraphTests.test_graph_forestc C �x t }| j|g| j�R � | j|g| j�R � | j|g| j�R � | j|g| j�R � | �|| j� � | �|| j� � d S r5 ) r r9 r) r* r+ r& r: r( r% r; r
r
r �(test_graph_connected_under_edge_failures s z=UndirectedGraphTests.test_graph_connected_under_edge_failuresc C r? r5 ) r r9 r) r* r+ r& r: r( r% r; r
r
r �*test_graph_connected_under_vertex_failures� s z?UndirectedGraphTests.test_graph_connected_under_vertex_failuresc C � d S r5 r
�r- r
r
r � test_graph_multi_edge_forest� � z1UndirectedGraphTests.test_graph_multi_edge_forestc C s� t }| j|g| j�R � | j|g| j�R � | �|| j� � | �|| j� � | �|| j� � | �|| j� � | �|| j � � d S r5 )
r r9 r, r' r: r( r% r) r* r+ r; r
r
r � test_graph_no_lonely_vertices� s z2UndirectedGraphTests.test_graph_no_lonely_verticesc C rB r5 r
rC r
r
r � test_graph_no_unknown_vertices� rE z3UndirectedGraphTests.test_graph_no_unknown_vertices)�__name__�
__module__� __qualname__r r9 r< r= r> r@ rA rD rF rG �
__classcell__r
r
r3 r r 7 s
r )�__doc__�samba� samba.tests�samba.kcc.graph_utilsr r r r r r r r! r �tests�TestCaser r
r
r
r �<module> s