def prepare(cc): nodes = set() for i in range(cc.k): node = Node(i) node.updateParams(cc.data) nodes.add(node) for j in range(cc.n): flow = Flow(j) flow.updateParams(cc.data) flow.connectWithNode(nodes) flows = set() for node in nodes: for flow in node.flows: flows.add(flow) return (nodes, flows)