def test_fan_in_out_capacities_directed(self):
dir_topology = fnss.DirectedTopology()
dir_topology.add_edge(0, 1)
dir_topology.add_edge(1, 0)
dir_topology.add_edge(1, 2)
dir_topology.add_edge(3, 2)
fnss.set_capacities_constant(dir_topology, 10, 'Mbps')
in_cap, out_cap = fnss.fan_in_out_capacities(dir_topology)
self.assertEquals({0: 10, 1: 10, 2: 20, 3: 0}, in_cap)
self.assertEquals({0: 10, 1: 20, 2: 0, 3: 10}, out_cap)
def test_fan_in_out_capacities_directed(self):
dir_topology = fnss.DirectedTopology()
dir_topology.add_edge(0, 1)
dir_topology.add_edge(1, 0)
dir_topology.add_edge(1, 2)
dir_topology.add_edge(3, 2)
fnss.set_capacities_constant(dir_topology, 10, 'Mbps')
in_cap, out_cap = fnss.fan_in_out_capacities(dir_topology)
self.assertEqual({0: 10, 1: 10, 2: 20, 3: 0}, in_cap)
self.assertEqual({0: 10, 1: 20, 2: 0, 3: 10}, out_cap)
def test_fan_in_out_capacities_undirected(self):
topology = fnss.star_topology(3)
fnss.set_capacities_constant(topology, 10, 'Mbps')
in_cap, out_cap = fnss.fan_in_out_capacities(topology)
self.assertEquals({0: 30, 1: 10, 2: 10, 3: 10}, in_cap)
self.assertEquals(in_cap, out_cap)
def test_fan_in_out_capacities_undirected(self):
topology = fnss.star_topology(3)
fnss.set_capacities_constant(topology, 10, 'Mbps')
in_cap, out_cap = fnss.fan_in_out_capacities(topology)
self.assertEqual({0: 30, 1: 10, 2: 10, 3: 10}, in_cap)
self.assertEqual(in_cap, out_cap)
