def test_addOutEdge(self): v4 = Vertex("4", 200, 100) v5 = Vertex("5", 100, 100) v45 = Edge(v4, v5, 2) v54 = Edge(v5, v4, 3) nose.tools.assert_raises(GraphError, v5.addOutEdge, v45) v5.addOutEdge(v54) assert v5.getNumOutEdges() == 1 nose.tools.assert_raises(GraphError, v5.addOutEdge, v54)
def test_getOutEdge(self): v4 = Vertex("4", 200, 100) v5 = Vertex("5", 100, 100) e45 = Edge(v4, v5, 2) e54 = Edge(v5, v4, 3) v5.addOutEdge(e54) assert e54 == v5.getOutEdge("4") nose.tools.assert_raises(GraphError, v5.getOutEdge, "10") assert v5.hasOutEdge("4") assert not v5.hasOutEdge("11")