class TestGraph(unittest.TestCase):
def setUp(self):
self.graph = Graph(4)
self.odd_graph = Graph(5)
def test_graph_average_degree_connectivity(self):
self.assertGreaterEqual(self.graph.average_degree_connectivity, 3)
def test_no_self_loops(self):
self.assertTrue(self.graph.no_self_loops())
def test_is_valid(self):
self.assertTrue(self.graph.is_valid())
def test_choose_random_nodes(self):
random_choice = Graph(5).choose_random_nodes()
self.assertNotIn(4, random_choice)
def test_odd_graph(self):
self.assertTrue(self.odd_graph.all_connected)
def test_odd_graph_is_valid(self):
self.assertTrue(self.odd_graph.is_valid())
def setUp(self): self.graph = Graph(4) self.odd_graph = Graph(5)
