def test_results_graph_bands(): bundles = [ Bundle('a', 'b'), ] # Mock flow data bundle_flows = { bundles[0]: pd.DataFrame.from_records([ ('a1', 'b1', 'm', 3), ], columns=('source', 'target', 'material', 'value')) } view_graph = LayeredGraph() view_graph.add_node('a', node=ProcessGroup()) view_graph.add_node('b', node=ProcessGroup()) view_graph.add_edges_from([ ('a', 'b', { 'bundles': bundles }), ]) view_graph.ordering = Ordering([ [['a'], []], [[], ['b']], ]) # Do partition based on flows stored in bundles Gr, groups = results_graph(view_graph, bundle_flows) assert Gr.ordering == Ordering([ # rank 1 [['a^*'], []], # rank 2 [[], ['b^*']], ])
def test_dummy_nodes_left_RL(): G = _twonodes(2, 'R', 0, 'L') assert set(G.nodes()) == {'x', 'y', '__x_y_1', '__x_y_2'} assert set(G.edges()) == {('x', '__x_y_2'), ('__x_y_2', '__x_y_1'), ('__x_y_1', 'y')} assert G.ordering == Ordering([[['y']], [['__x_y_1']], [['x', '__x_y_2']]])
def test_dummy_nodes_right_RR(): G = _twonodes(0, 'R', 2, 'R') assert set(G.nodes()) == {'x', 'y', '__x_y_1'} assert set(G.edges()) == {('x', '__x_y_1'), ('__x_y_1', 'y')} assert G.ordering == Ordering([[['x']], [['__x_y_1']], [['y']]])
def test_dummy_nodes_simple(): G = _twonodes(0, 'R', 1, 'R', bundle_key=27) assert set(G.nodes()) == {'x', 'y'} assert set(G.edges()) == {('x', 'y')} assert G.ordering == Ordering([[['x']], [['y']]]) assert G['x']['y']['bundles'] == [27]
def test_results_graph_overall(): material_partition = Partition.Simple('material', ['m', 'n']) c_partition = Partition.Simple('process', ['c1', 'c2']) view_graph = LayeredGraph() view_graph.add_node('a', node=ProcessGroup(title='Node a')) view_graph.add_node('b', node=ProcessGroup()) view_graph.add_node('c', node=ProcessGroup(partition=c_partition)) view_graph.add_node('via', node=Waypoint(partition=material_partition)) view_graph.add_edges_from([ ('a', 'via', { 'bundles': [0], 'flow_partition': material_partition }), ('b', 'via', { 'bundles': [1], 'flow_partition': material_partition }), ('via', 'c', { 'bundles': [0, 1], 'flow_partition': material_partition }), ]) view_graph.ordering = Ordering([[['a', 'b']], [['via']], [['c']]]) # Mock flow data bundle_flows = { 0: pd.DataFrame.from_records([ ('a1', 'c1', 'm', 3), ('a2', 'c1', 'n', 1), ], index=(0, 1), columns=('source', 'target', 'material', 'value')), 1: pd.DataFrame.from_records([ ('b1', 'c1', 'm', 1), ('b1', 'c2', 'm', 2), ('b1', 'c2', 'n', 1), ], index=(2, 3, 4), columns=('source', 'target', 'material', 'value')) } # Do partition based on flows stored in bundles Gr, groups = results_graph(view_graph, bundle_flows) assert sorted(Gr.nodes(data=True)) == [ ('a^*', { 'direction': 'R', 'type': 'process', 'title': 'Node a' }), ('b^*', { 'direction': 'R', 'type': 'process', 'title': 'b' }), ('c^c1', { 'direction': 'R', 'type': 'process', 'title': 'c1' }), ('c^c2', { 'direction': 'R', 'type': 'process', 'title': 'c2' }), ('via^m', { 'direction': 'R', 'type': 'group', 'title': 'm' }), ('via^n', { 'direction': 'R', 'type': 'group', 'title': 'n' }), ] assert sorted(Gr.edges(keys=True, data=True)) == [ ('a^*', 'via^m', ('m', '*'), { 'measures': { 'value': 3 }, 'original_flows': [0], 'bundles': [0] }), ('a^*', 'via^n', ('n', '*'), { 'measures': { 'value': 1 }, 'original_flows': [1], 'bundles': [0] }), ('b^*', 'via^m', ('m', '*'), { 'measures': { 'value': 3 }, 'original_flows': [2, 3], 'bundles': [1] }), ('b^*', 'via^n', ('n', '*'), { 'measures': { 'value': 1 }, 'original_flows': [4], 'bundles': [1] }), ('via^m', 'c^c1', ('m', '*'), { 'measures': { 'value': 4 }, 'original_flows': [0, 2], 'bundles': [0, 1] }), ('via^m', 'c^c2', ('m', '*'), { 'measures': { 'value': 2 }, 'original_flows': [3], 'bundles': [0, 1] }), ('via^n', 'c^c1', ('n', '*'), { 'measures': { 'value': 1 }, 'original_flows': [1], 'bundles': [0, 1] }), ('via^n', 'c^c2', ('n', '*'), { 'measures': { 'value': 1 }, 'original_flows': [4], 'bundles': [0, 1] }), ] assert Gr.ordering == Ordering([ [['a^*', 'b^*']], [['via^m', 'via^n']], [['c^c1', 'c^c2']], ]) # Only includes groups where the title is not the same as the single node # title assert groups == [ { 'id': 'via', 'title': '', 'type': 'group', 'nodes': ['via^m', 'via^n'] }, { 'id': 'c', 'title': '', 'type': 'process', 'nodes': ['c^c1', 'c^c2'] }, ]
def test_results_graph_time_partition(): time_partition = Partition.Simple('time', [1, 2]) view_graph = LayeredGraph() view_graph.add_node('a', node=ProcessGroup()) view_graph.add_node('b', node=ProcessGroup()) view_graph.add_edges_from([ ('a', 'b', { 'bundles': [0] }), ]) view_graph.ordering = Ordering([[['a']], [['b']]]) # Mock flow data bundle_flows = { 0: pd.DataFrame.from_records([ ('a1', 'b1', 'm', 1, 3), ('a2', 'b1', 'n', 1, 1), ('a2', 'b2', 'n', 1, 2), ('a1', 'b1', 'm', 2, 1), ('a1', 'b1', 'n', 2, 3), ], columns=('source', 'target', 'material', 'time', 'value')), } # Do partition based on flows stored in bundles Gr, groups = results_graph(view_graph, bundle_flows, time_partition=time_partition) assert sorted(Gr.edges(keys=True, data=True)) == [ ('a^*', 'b^*', ('*', '1'), { 'measures': { 'value': 6 }, 'original_flows': [0, 1, 2], 'bundles': [0] }), ('a^*', 'b^*', ('*', '2'), { 'measures': { 'value': 4 }, 'original_flows': [3, 4], 'bundles': [0] }), ] # Now add a material partition too material_partition = Partition.Simple('material', ['m', 'n']) Gr, groups = results_graph(view_graph, bundle_flows, material_partition, time_partition) assert sorted(Gr.edges(keys=True, data=True)) == [ ('a^*', 'b^*', ('m', '1'), { 'measures': { 'value': 3 }, 'original_flows': [0], 'bundles': [0] }), ('a^*', 'b^*', ('m', '2'), { 'measures': { 'value': 1 }, 'original_flows': [3], 'bundles': [0] }), ('a^*', 'b^*', ('n', '1'), { 'measures': { 'value': 3 }, 'original_flows': [1, 2], 'bundles': [0] }), ('a^*', 'b^*', ('n', '2'), { 'measures': { 'value': 3 }, 'original_flows': [4], 'bundles': [0] }), ]
def test_results_graph_elsewhere_stubs(): b_partition = Partition.Simple('process', ['b1', 'b2']) view_graph = LayeredGraph() view_graph.add_node('a', node=ProcessGroup()) view_graph.add_node('b', node=ProcessGroup(partition=b_partition), from_elsewhere_bundles=[1]) view_graph.add_edge('a', 'b', bundles=[0]) view_graph.ordering = Ordering([[['a']], [['b']]]) # Mock flow data bundle_flows = { 0: pd.DataFrame.from_records([ ('a1', 'b1', 'm', 3), ('a2', 'b1', 'n', 1), ], index=(0, 1), columns=('source', 'target', 'material', 'value')), 1: pd.DataFrame.from_records([ ('x1', 'b1', 'm', 1), ('x3', 'b2', 'n', 5), ], index=(2, 3), columns=('source', 'target', 'material', 'value')) } # Do partition based on flows stored in bundles Gr, groups = results_graph(view_graph, bundle_flows) assert sorted(Gr.nodes(data=True)) == [ ('a^*', { 'direction': 'R', 'type': 'process', 'title': 'a' }), ('b^b1', { 'direction': 'R', 'type': 'process', 'title': 'b1', 'from_elsewhere_edges': [ (('*', '*'), { 'measures': { 'value': 1 }, 'original_flows': [2], 'bundles': [1] }), ] }), ('b^b2', { 'direction': 'R', 'type': 'process', 'title': 'b2', 'from_elsewhere_edges': [ (('*', '*'), { 'measures': { 'value': 5 }, 'original_flows': [3], 'bundles': [1] }), ] }), ] assert sorted(Gr.edges(keys=True, data=True)) == [ ('a^*', 'b^b1', ('*', '*'), { 'measures': { 'value': 4 }, 'original_flows': [0, 1], 'bundles': [0] }), ]