Ejemplos de Graph.add en Python

Ejemplo n.º 1

Archivo: memgraph_test.py Proyecto: nigelsmall/memgraph

def test_removal_of_non_existent_node_should_fail():
    g = Graph()
    g.add(10)
    assert set(g.nodes()) == {10}
    try:
        g.remove(12)
    except KeyError:
        assert True
    else:
        assert False

Ejemplo n.º 2

Archivo: memgraph_test.py Proyecto: nigelsmall/memgraph

def test_can_remove():
    g = Graph()
    g.add(10)
    assert set(g.nodes()) == {10}
    g.remove(10)
    assert set(g.nodes()) == set()