def test_remove_edge():
trek = Space()
captain = trek.add_vertex('Captain')
kirk = trek.add_vertex(name='James T. Kirk')
rank = trek.add_edge(kirk, captain, 'has_rank')
trek.remove_edge(rank.id)
with pytest.raises(MissingEdge):
trek.edge_by_id(rank.id)
def test_create_edge():
trek = Space()
captain = trek.add_vertex('Captain')
kirk = trek.add_vertex('James Kirk')
rank = trek.add_edge(kirk, captain, 'has_rank')
assert rank.start == kirk
assert rank.end == captain
assert rank.name == 'has_rank'
assert rank.id is not None
def test_adjacent_edges():
trek = Space()
captain = trek.add_vertex('Captain')
kirk = trek.add_vertex(name='James T. Kirk')
pike = trek.add_vertex(name='Christopher Pike')
edges = set()
for person in [kirk, pike]:
edges.add(trek.add_edge(person, captain, 'has_rank'))
assert trek.adjacent_edges(captain) == edges
def test_remove_vertex():
trek = Space()
captain = trek.add_vertex('Captain')
kirk = trek.add_vertex(name='James T. Kirk')
pike = trek.add_vertex(name='Christopher Pike')
edges = []
for person in [kirk, pike]:
edges.append(trek.add_edge(person, captain, 'has_rank'))
trek.remove_vertex(captain.id)
for edge in edges:
with pytest.raises(MissingEdge):
trek.edge_by_id(edge.id)
assert trek.vertex_by_id(kirk.id) == kirk
assert trek.vertex_by_id(pike.id) == pike
def test_find_edge_by_id():
trek = Space()
kirk = trek.add_vertex(name='James T. Kirk')
captain = trek.add_vertex('Captain')
rank = trek.add_edge(kirk, captain, 'has_rank')
assert trek.edge_by_id(rank.id) == rank
def test_find_vertex_by_id():
trek = Space()
kirk = trek.add_vertex(name='James T. Kirk')
assert trek.vertex_by_id(kirk.id) == kirk