def test_translate_attr_modified():
mock_conn = DiGraph()
mock_conn.add_edge('B-1', 'B-2', Connection='Present')
translate = EndGraph._translate_attr_modified.im_func
# PDCs that get averaged are 3 (RCs), 6 (existent conn edge), and
# 18 (non-existent conn edge).
nt.assert_equal(
translate(EndGraph(), ('A-1', ['B-1', 'B-3']), ('A-2', ['B-2']), None,
mock_conn), {
'Connection': 'Present',
'PDC': 6.6
})
def test_translate_attr_original():
mock_conn = DiGraph()
mock_conn.add_edge('B-1', 'B-2', {'EC_Source': 'N', 'EC_Target': 'Nc'})
translate = EndGraph._translate_attr_original.im_func
# PDCs that get averaged are 3 (RCs), 6 (existent conn edge), and
# 18 (non-existent conn edge).
nt.assert_equal(
translate(EndGraph(), ('A-1', ['B-1', 'B-3']), ('A-2', ['B-2']), None,
mock_conn), {
'EC_Source': 'N',
'EC_Target': 'Nc',
'PDC': 6.6
})
def test_make_translation_dict():
translate = EndGraph._make_translation_dict.im_func
nt.assert_equal(translate(EndGraph(), None, 'A-1', 'B'), {'X': ['X']})
nt.assert_equal(translate(EndGraph(), None, 'A-1', 'A'), {'A-1': ['A-1']})
def setUp(self):
self.m = DiGraph()
self.c = DiGraph()
self.e = EndGraph()