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)