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
def test_removal_of_related_node_should_fail():
g = Graph()
g.relate(10, "is more than", 4)
assert set(g.nodes()) == {4, 10}
try:
g.remove(10)
except NotIsolatedError:
assert True
else:
assert False
g.separate(10, "is more than", 4)
g.remove(10)
assert set(g.nodes()) == {4}
def test_can_remove():
g = Graph()
g.add(10)
assert set(g.nodes()) == {10}
g.remove(10)
assert set(g.nodes()) == set()
