def test_addInEdge(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.addInEdge, v54) v5.addInEdge(v45) assert v5.getNumInEdges() == 1 nose.tools.assert_raises(GraphError, v5.addInEdge, v45)
def test_incidentEdge(self): v4 = Vertex("4", 200, 100) v5 = Vertex("5", 100, 100) e45 = Edge(v4, v5, 2) e54 = Edge(v5, v4, 3) v5.addInEdge(e45) assert e45 == v5.getInEdge("4") nose.tools.assert_raises(GraphError, v5.getInEdge, "40") assert v5.hasInEdge("4") assert not v5.hasInEdge("100")