def setUp(self):
self.g = Graph()
self.g.add_connected_vertice(0, 0)
self.g.add_connected_vertice(3, 4)
self.g.add_connected_vertice(0, 4)
self.g.add_connected_vertice(3, 0)
self.g.add_connected_vertice(10, 40)
self.mst = [[
Vertice(0, 0),
Vertice(0, 4),
Vertice(3, 0),
Vertice(3, 4),
Vertice(10, 40),
]]
self.three_groups = [
[Vertice(0, 0), Vertice(3, 0)],
[Vertice(3, 4), Vertice(0, 4)],
[Vertice(10, 40)],
]
def test_add_connected_vertice(self):
vertices = []
edges = []
self.assertEqual(self.graph.edges, edges)
self.assertEqual(self.graph.vertices, vertices)
self.graph.add_connected_vertice(0, 0)
vertices.append(Vertice(0, 0))
self.assertEqual(self.graph.edges, edges)
self.assertEqual(self.graph.vertices, vertices)
self.graph.add_connected_vertice(3, 4)
vertices.append(Vertice(3, 4))
edges.append(Edge(vertices[0], vertices[1]))
self.assertEqual(self.graph.edges, edges)
self.graph.add_connected_vertice(0, 4)
vertices.append(Vertice(0, 4))
edges.append(Edge(vertices[0], vertices[2]))
edges.append(Edge(vertices[1], vertices[2]))
self.assertEqual(self.graph.edges, edges)
self.assertEqual(self.graph.vertices, vertices)
def setUp(self):
self.v1 = Vertice(0, 0)
self.v2 = Vertice(3, 4)
self.edge = Edge(self.v1, self.v2)
def test_repr(self):
vertice = Vertice(9, 1)
self.assertEqual(str(vertice), repr(vertice))
def test_str(self):
vertice = Vertice(9, 1)
self.assertEqual('(9, 1)', str(vertice))
def test_euclidian_distance(self):
v1 = Vertice(0, 0)
v2 = Vertice(3, 4)
self.assertEqual(v1 - v2, 5)
def test_init(self):
vertice = Vertice(10, 23)
self.assertEqual(vertice.x, 10)
self.assertEqual(vertice.y, 23)