def test_keep_one_level():
# Not mocked: _find_bottom_of_hierarchy,
# _remove_level_from_hierarchy, _summate_connections.
hierarchy = {
'A-J': {
'A-A': {}
},
'A-B': {
'A-I': {
'A-F': {
'A-K': {},
'A-L': {}
},
'A-H': {}
},
'A-D': {
'A-E': {}
}
}
}
mock_conn = DiGraph()
mock_conn.add_edges_from([('A-D', 'A-Z'), ('A-Y', 'A-F'), ('A-J', 'A-X')])
mapp = MapGraph()
mapp.add_nodes_from(
['A-A', 'A-B', 'A-D', 'A-E', 'A-F', 'A-H', 'A-I', 'A-J', 'A-K', 'A-L'])
mapp.cong = mock_conn
mapp._keep_one_level(hierarchy, 'A')
nt.assert_equal(mapp.cong.edges(), [('A-J', 'A-X'), ('A-D', 'A-Z'),
('A-Y', 'A-F')])
nt.assert_equal(mapp.nodes(), ['A-H', 'A-J', 'A-D', 'A-F'])
def test_keep_one_level():
# Not mocked: _find_bottom_of_hierarchy,
# _remove_level_from_hierarchy, _summate_connections.
hierarchy = {'A-J': {'A-A': {}},
'A-B': {'A-I': {'A-F': {'A-K': {},
'A-L': {}},
'A-H': {}},
'A-D': {'A-E': {}}}}
mock_conn = DiGraph()
mock_conn.add_edges_from([('A-D', 'A-Z'), ('A-Y', 'A-F'), ('A-J', 'A-X')])
mapp = MapGraph()
mapp.add_nodes_from(['A-A', 'A-B', 'A-D', 'A-E', 'A-F', 'A-H', 'A-I',
'A-J', 'A-K', 'A-L'])
mapp.cong = mock_conn
mapp._keep_one_level(hierarchy, 'A')
nt.assert_equal(mapp.cong.edges(), [('A-J', 'A-X'), ('A-D', 'A-Z'),
('A-Y', 'A-F')])
nt.assert_equal(mapp.nodes(), ['A-H', 'A-J', 'A-D', 'A-F'])
