def test_multi_sinks(self):
edges = [make_edge(1, (('foo', 'bar'),
('baz', 'bad'))),
make_edge(2, (('foz', 'bas'),
('boz', 'bat')))]
with self.assertRaises(AssertionError):
build_keys(edges)
def test_multi_edge(self):
edges = [make_edge(1, (('foo', 'bar'),
('baz', 'bad'))),
make_edge(1, (('foz', 'bas'),
('boz', 'bat')))]
self.assertDictEqual(build_keys(edges),
{'bar' : 42,
'bad' : 42,
'bas' : 42,
'bat' : 42})
def test_complete_copy(self):
edges = [make_edge(1, (('foo', 'bar'),
('baz', 'bad'))),
make_edge(1, (('foz', 'bas'),
('boz', 'bat')))]
# set edges to None for complete copy
for edge in edges:
edge.map_ = None
self.assertDictEqual(build_keys(edges),
{'foo' : 42,
'baz' : 42,
'foz' : 42,
'boz' : 42})
def test_single_edge(self):
edges = [make_edge(1, (('foo', 'bar'),
('baz', 'bad')))]
self.assertDictEqual(build_keys(edges),
{'bar' : 42,
'bad' : 42})
