class GraphTest(unittest.TestCase):
V1_LABEL = 1
V2_LABEL = 2
V3_LABEL = 3
V4_LABEL = 4
V5_LABEL = 5
E1_LABEL = 1
E2_LABEL = 2
E3_LABEL = 3
if settings.INTEGER_VERTICES:
v1 = 1
v2 = 2
v3 = 3
v4 = 4
v5 = 5
else:
v1 = Vertex(V1_LABEL)
v2 = Vertex(V2_LABEL)
v3 = Vertex(V3_LABEL)
v4 = Vertex(V4_LABEL)
v5 = Vertex(V5_LABEL)
e1 = Edge(E1_LABEL, v1, v2, 3)
e2 = Edge(E2_LABEL, v2, v1, 5)
e3 = Edge(E3_LABEL, v4, v5, 10)
def setUp(self):
self.g = Graph()
def test_get_vertex(self):
self.g.add_vertex(self.v1)
self.assertEqual(self.g.get_vertex(self.V1_LABEL), self.v1)
def test_get_edge_ends(self):
self.assertTupleEqual(self.g.get_edge_ends(self.e1), (self.v1, self.v2))
def test_get_vertices_count(self):
self.assertEqual(self.g.get_vertices_count(), 0)
self.g.add_vertex(self.v1)
self.assertEqual(self.g.get_vertices_count(), 1)
self.g.add_vertex(self.v2)
self.assertEqual(self.g.get_vertices_count(), 2)
def test_get_vertex_position(self):
if settings.INTEGER_VERTICES:
return
self.g.add_vertex(self.v1)
self.g.add_vertex(self.v2)
self.g.add_vertex(self.v3)
self.assertEqual(self.g.get_vertex_position(self.v1), 0)
self.assertEqual(self.g.get_vertex_position(self.v2), 1)
self.assertEqual(self.g.get_vertex_position(self.v3), 2)
