def test_adjacent_error(): """error raised when nodes non-existant""" g = SimpleGraph() g.add_node('a') g.add_node('b') g.add_node('c') g.add_edge('a', 'c') g.add_edge('a', 'b') with pytest.raises(KeyError): g.adjacent('l', 'm')
def test_adjacent(): """adjacent correctly identify edges, new edge add node""" g = SimpleGraph() g.add_node('a') g.add_node('b') g.add_node('c') g.add_edge('a', 'd') g.add_edge('a', 'f') assert 'f' in g.nodes() assert g.adjacent('a', 'd') assert not g.adjacent('a', 'b')