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)
