class test_routingtable(TestCase): def setUp(self): self.connections_orig = [('a', 'b'), ('b', 'c')] self.ids_orig = set([i[0] for i in self.connections_orig]+\ [i[1] for i in self.connections_orig]) self.t = RoutingTable() for c in self.connections_orig: self.t[c[0], c[1]] = 1 def test_ids(self): assert set(self.t.ids) == self.ids_orig def test_connections(self): assert set(self.t.connections) == set(self.connections_orig) def test_setitem(self): t = RoutingTable() t['a', 'b'] = 1 assert t.data.has_node('a') assert t.data.has_node('b') assert t.data.has_edge('a', 'b') def test_getitem(self): t = RoutingTable() t.data.add_node('a') t.data.add_node('b') t.data.add_edge('a', 'b', {'data': 1}) assert t['a', 'b'] == 1 def test_src_ids(self): for i in self.connections_orig: assert i[0] in self.t.src_ids(i[1]) def test_dest_ids(self): for i in self.connections_orig: assert i[1] in self.t.dest_ids(i[0])
class test_routingtable(TestCase): def setUp(self): self.connections_orig = [('a', 'b'), ('b', 'c')] self.ids_orig = set([i[0] for i in self.connections_orig]+\ [i[1] for i in self.connections_orig]) self.t = RoutingTable() for c in self.connections_orig: self.t[c[0], c[1]] = 1 def test_ids(self): assert set(self.t.ids) == self.ids_orig def test_connections(self): assert set(self.t.connections) == set(self.connections_orig) def test_setitem(self): t = RoutingTable() t['a', 'b'] = 1 assert t.data.has_node('a') assert t.data.has_node('b') assert t.data.has_edge('a', 'b') def test_getitem(self): t = RoutingTable() t.data.add_node('a') t.data.add_node('b') t.data.add_edge('a', 'b', {'data':1}) assert t['a', 'b'] == 1 def test_src_ids(self): for i in self.connections_orig: assert i[0] in self.t.src_ids(i[1]) def test_dest_ids(self): for i in self.connections_orig: assert i[1] in self.t.dest_ids(i[0])
class test_routingtable(TestCase): def setUp(self): self.connections_orig = [('a', 'b'), ('b', 'c')] self.ids_orig = set([i[0] for i in self.connections_orig]+\ [i[1] for i in self.connections_orig]) self.t = RoutingTable() for c in self.connections_orig: self.t[c[0], c[1]] = 1 def test_ids(self): assert set(self.t.ids) == self.ids_orig def test_connections(self): assert set(self.t.connections) == set(self.connections_orig) def test_setitem(self): t = RoutingTable() t['a', 'b'] = 1 assert t.data.has_node('a') assert t.data.has_node('b') assert t.data.has_edge('a', 'b') def test_getitem_scalar(self): t = RoutingTable() t.data.add_node('a') t.data.add_node('b') t.data.add_edge('a', 'b', {'data': 1}) assert t['a', 'b'] == 1 assert t['a', 'b', 'data'] == 1 def test_getitem_non_dict(self): t = RoutingTable() t.data.add_node('a') t.data.add_node('b') t.data.add_edge('a', 'b', {'data': [1, 2]}) assert t['a', 'b'] == [1, 2] assert t['a', 'b', 'data'] == [1, 2] def test_getitem_dict(self): t = RoutingTable() t.data.add_node('a') t.data.add_node('b') t.data.add_edge('a', 'b', {'x': 1, 'y': 2}) assert t['a', 'b'] == {'x': 1, 'y': 2} assert t['a', 'b', 'x'] == 1 def test_src_ids(self): for i in self.connections_orig: assert i[0] in self.t.src_ids(i[1]) assert self.t.src_ids('d') == [] def test_dest_ids(self): for i in self.connections_orig: assert i[1] in self.t.dest_ids(i[0]) assert self.t.src_ids('d') == [] def test_subtable(self): t = RoutingTable() t['a', 'b'] = 1 t['b', 'c'] = 1 t['c', 'd'] = 1 t['d', 'a'] = 1 s = t.subtable(['a', 'b', 'c']) assert set(s.ids) == set(['a', 'b', 'c']) assert set(s.connections) == set([('a', 'b'), ('b', 'c')])
class test_routingtable(TestCase): def setUp(self): self.connections_orig = [('a', 'b'), ('b', 'c')] self.ids_orig = set([i[0] for i in self.connections_orig]+\ [i[1] for i in self.connections_orig]) self.t = RoutingTable() for c in self.connections_orig: self.t[c[0], c[1]] = 1 def test_ids(self): assert set(self.t.ids) == self.ids_orig def test_connections(self): assert set(self.t.connections) == set(self.connections_orig) def test_setitem(self): t = RoutingTable() t['a', 'b'] = 1 assert t.data.has_node('a') assert t.data.has_node('b') assert t.data.has_edge('a', 'b') def test_getitem_scalar(self): t = RoutingTable() t.data.add_node('a') t.data.add_node('b') t.data.add_edge('a', 'b', **{'data': 1}) assert t['a', 'b'] == 1 assert t['a', 'b', 'data'] == 1 def test_getitem_non_dict(self): t = RoutingTable() t.data.add_node('a') t.data.add_node('b') t.data.add_edge('a', 'b', **{'data': [1, 2]}) assert t['a', 'b'] == [1, 2] assert t['a', 'b', 'data'] == [1, 2] def test_getitem_dict(self): t = RoutingTable() t.data.add_node('a') t.data.add_node('b') t.data.add_edge('a', 'b', **{'x': 1, 'y': 2}) assert t['a', 'b'] == {'x': 1, 'y': 2} assert t['a', 'b', 'x'] == 1 def test_src_ids(self): for i in self.connections_orig: assert i[0] in self.t.src_ids(i[1]) assert self.t.src_ids('d') == [] def test_dest_ids(self): for i in self.connections_orig: assert i[1] in self.t.dest_ids(i[0]) assert self.t.src_ids('d') == [] def test_subtable(self): t = RoutingTable() t['a', 'b'] = 1 t['b', 'c'] = 1 t['c', 'd'] = 1 t['d', 'a'] = 1 s = t.subtable(['a', 'b', 'c']) assert set(s.ids) == set(['a', 'b', 'c']) assert set(s.connections) == set([('a', 'b'), ('b', 'c')])