def test_flow2(self):
flows, max_flow = ford_fulkerson(self.graph2, 1, 4)
self.__check_flow_constraints(1, 4, flows, self.graph2)
assert max_flow == 2000
def test_flow(self):
flows, max_flow = ford_fulkerson(self.graph, 1, 5)
self.__check_flow_constraints(1, 5, flows, self.graph)
assert max_flow == 6
