def testDensity(self): numVertices = 10 graph = DictGraph(True) for i in range(numVertices): graph.setVertex(i, 0) graph.addEdge(0, 1) self.assertEquals(graph.density(), float(1)/45) graph.addEdge(0, 2) self.assertEquals(graph.density(), float(2)/45) graph = DictGraph(False) for i in range(numVertices): graph.setVertex(i, 0) graph.addEdge(0, 1) self.assertEquals(graph.density(), float(1)/90) graph.addEdge(0, 2) self.assertEquals(graph.density(), float(2)/90) #Test a graph with 1 vertex graph = DictGraph(True) graph.setVertex(0, 12) self.assertEquals(graph.density(), 0) graph.addEdge(0, 0) self.assertEquals(graph.density(), 1)