def __init__(self, graph, debug=False): self.graph = graph self.spanning_tree = SpanningTree(graph.vertexCount()) self.edge_stack = [] self.best_price = 0 self.best = None self.debug = debug
def test_edgeCandidates(self): self.graph.addEdge(0, 1) self.graph.addEdge(1, 2) self.graph.addEdge(1, 3) self.graph.addEdge(1, 4) self.graph.addEdge(2, 3) self.graph.addEdge(0, 4) tree = SpanningTree(5) tree.addEdge(Edge(0, 1)) actual = self.graph.edgeCandidates(tree) expected = [Edge(1, 2), Edge(1, 3), Edge(1, 4), Edge(0, 4)] self.assertEqual(set(expected), set(actual))
def __init__(self, graph, comm, debug=False): self.graph = graph self.spanning_tree = SpanningTree(graph.vertexCount()) self.edge_stack = [] self.best_price = sys.maxint self.best = None self.color = Token.BLACK self.initial_token_sent = False self.counter = 0 self.finished = False self.rank = comm.Get_rank() self.comm_size = comm.Get_size() self.comm = comm self.logger = self.setupLogging(debug) self.mpi_logger = self.logger self.total_edges = 0 self.total_solutions = 0 self.all_trees = set([])
def setUp(self): self.spanning_tree = SpanningTree(7)