class TestGraphMST(unittest.TestCase):
"""This unit test includes tests for the Edge data type,
EdgeWeightedGraph, and KruskalMST."""
def setUp(self):
vertices = [
{1:16,2:12,3:21},
{3:17,4:20},
{3:28,5:31},
{4:18,5:19,6:23},
{6:11},
{6:27},
{}
]
self.g = EdgeWeightedGraph(7)
for i, adj in enumerate(vertices):
for j, weight in adj.items():
self.g.add_edge(Edge(i, j, weight))
self.mst = KruskalMST(self.g)
def test_edge_2_5_graph(self):
for e in self.g.adj(2):
if e.other(2) == 5:
self.assertEqual( e.get_weight(), 31 )
def test_num_edges_graph(self):
self.assertEqual( len(list(self.g.edges())), 12 )
def test_num_edges_mst(self):
self.assertEqual( len(list(self.mst.edges())), 6 )
def test_weight_mst(self):
total_weight = 0
for e in self.mst.edges():
total_weight += e.get_weight()
self.assertEqual( total_weight, 93 )
def setUp(self):
vertices = [
{1: 16, 2: 12, 3: 21},
{3: 17, 4: 20},
{3: 28, 5: 31},
{4: 18, 5: 19, 6: 23},
{6: 11},
{6: 27},
{},
]
self.g = EdgeWeightedGraph(7)
for i, adj in enumerate(vertices):
for j, weight in adj.items():
self.g.add_edge(Edge(i, j, weight))
self.mst = KruskalMST(self.g)
class TestGraphMST(unittest.TestCase):
"""This unit test includes tests for the Edge data type,
EdgeWeightedGraph, and KruskalMST."""
def setUp(self):
vertices = [
{1: 16, 2: 12, 3: 21},
{3: 17, 4: 20},
{3: 28, 5: 31},
{4: 18, 5: 19, 6: 23},
{6: 11},
{6: 27},
{},
]
self.g = EdgeWeightedGraph(7)
for i, adj in enumerate(vertices):
for j, weight in adj.items():
self.g.add_edge(Edge(i, j, weight))
self.mst = KruskalMST(self.g)
def test_edge_2_5_graph(self):
for e in self.g.adj(2):
if e.other(2) == 5:
self.assertEqual(e.get_weight(), 31)
def test_num_edges_graph(self):
self.assertEqual(len(list(self.g.edges())), 12)
def test_num_edges_mst(self):
self.assertEqual(len(list(self.mst.edges())), 6)
def test_weight_mst(self):
total_weight = 0
for e in self.mst.edges():
total_weight += e.get_weight()
self.assertEqual(total_weight, 93)
def setUp(self):
vertices = [
{1:16,2:12,3:21},
{3:17,4:20},
{3:28,5:31},
{4:18,5:19,6:23},
{6:11},
{6:27},
{}
]
self.g = EdgeWeightedGraph(7)
for i, adj in enumerate(vertices):
for j, weight in adj.items():
self.g.add_edge(Edge(i, j, weight))
self.mst = KruskalMST(self.g)