def test_buffers_size_link_bandwidth_default_size(self):
topo = fnss.line_topology(4)
fnss.set_capacities_constant(topo, 8, 'Mbps', [(0, 1)])
fnss.set_capacities_constant(topo, 16, 'Mbps', [(1, 2)])
fnss.set_buffer_sizes_link_bandwidth(topo, buffer_unit='bytes', default_size=10)
self.assertEquals(topo.graph['buffer_unit'], 'bytes')
self.assertEquals(topo.adj[0][1]['buffer'], 1000000)
self.assertEquals(topo.adj[1][2]['buffer'], 2000000)
self.assertEquals(topo.adj[2][3]['buffer'], 10)
fnss.clear_buffer_sizes(topo)
self.assertTrue('capacity' not in topo.adj[2][3])
self.assertRaises(ValueError, fnss.set_buffer_sizes_link_bandwidth, topo)
def tearDown(self):
fnss.clear_buffer_sizes(self.topo)
def tearDown(self):
fnss.clear_buffer_sizes(self.topo)
