def test_residual_0(self): graph = happiness_upload._servermap_flow_graph(['peer0'], ['share0'], servermap={ 'peer0': ['share0'], }) flow = [[0 for _ in graph] for _ in graph] residual, capacity = happiness_upload.residual_network(graph, flow) # XXX no idea if these are right; hand-verify self.assertEqual(residual, [[1], [2], [3], []]) self.assertEqual( capacity, [[0, 1, 0, 0], [-1, 0, 1, 0], [0, -1, 0, 1], [0, 0, -1, 0]])
def test_residual_0(self): graph = happiness_upload._servermap_flow_graph( ['peer0'], ['share0'], servermap={ 'peer0': ['share0'], } ) flow = [[0 for _ in graph] for _ in graph] residual, capacity = happiness_upload.residual_network(graph, flow) # XXX no idea if these are right; hand-verify self.assertEqual(residual, [[1], [2], [3], []]) self.assertEqual(capacity, [[0, 1, 0, 0], [-1, 0, 1, 0], [0, -1, 0, 1], [0, 0, -1, 0]])
def test_trivial_flow_graph(self): self.assertEqual([], happiness_upload._servermap_flow_graph( set(), set(), {}))
def test_trivial_flow_graph(self): self.assertEqual( [], happiness_upload._servermap_flow_graph(set(), set(), {}) )