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))
