Пример #1
0
def test_mutable_edge_set_remove_edge(mutable_edge_set: MutableEdgeSet):
    node1 = Node('1')
    node2 = Node('2')
    edge = Edge('e1', (node1, node2))
    mutable_edge_set.add_edge(edge)
    mutable_edge_set.remove_edge(edge)

    assert edge not in mutable_edge_set
    assert mutable_edge_set.is_empty()
Пример #2
0
def test_mutable_edge_set_remove_edges(mutable_edge_set: MutableEdgeSet):
    edge_list = []

    for i in range(10):
        node2 = Node(f'{i}')
        node1 = Node(f'{i}_{i}')
        edge_list.append(Edge(f'e{i}', (node1, node2)))

    mutable_edge_set.add_edge(*edge_list)
    mutable_edge_set.remove_edge(*edge_list)

    assert all(edge not in mutable_edge_set for edge in edge_list)
    assert mutable_edge_set.is_empty()