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'])