def test_add_vertex(self): a = Vertex("a") b = Vertex("b") a.add_dart(b, 1) g = Graph([a, b]) g.add_vertex("c") self.assertTrue(len(g.vertices) == 3)
def test_remove_dart(self): a = Vertex("a") b = Vertex("b") a.add_dart(b, 1) # Remove the dart a.remove_dart(b) self.assertFalse(a.is_neighbor(b))
def test_remove_vertex(self): a = Vertex("a") b = Vertex("b") c = Vertex("c") a.add_dart(b, 1) a.add_dart(c, 1) b.add_dart(c, 1) g = Graph([a, b, c]) g.remove_vertex("a") self.assertTrue(len(g.vertices) == 2) self.assertTrue(len(g.get_vertex("b").neighbors) == 1) self.assertTrue(len(g.get_vertex("c").neighbors) == 0)
def test_add_dart(self): a = Vertex("a") b = Vertex("b") a.add_dart(b, 1) self.assertTrue(len(a.neighbors) != 0) self.assertTrue(len(b.neighbors) == 0)
def test_is_neighbor(self): a = Vertex("a") b = Vertex("b") a.add_dart(b, 1) self.assertTrue(a.is_neighbor(b)) self.assertFalse(b.is_neighbor(a))