Generate test picture

.github/workflows/unit-build.yml

@@ -34,5 +34,10 @@ jobs:
     - name: Test with pytest
       run: |
         pytest --cov=golem test/unit
+    - name: Upload graph layout as an artifact
+      uses: actions/upload-artifact@v2
+      with:
+        name: graph-layout
+        path: 'test/unit/visualisation/graph_layout.png'
     - name: Codecov-coverage
       uses: codecov/codecov-action@v2

test/unit/visualisation/test_visualisation_utils.py

@@ -1,3 +1,7 @@
+import pathlib
+
+from matplotlib import pyplot as plt
+
 from golem.core.adapter import DirectAdapter
 from golem.core.dag.convert import graph_structure_as_nx_graph
 from golem.core.optimisers.fitness.multi_objective_fitness import MultiObjFitness
@@ -44,6 +48,17 @@ def test_hierarchy_pos():
     node_labels = {uid: str(node) for uid, node in nodes_dict.items()}
 
     pos = GraphVisualizer._get_hierarchy_pos_by_distance_to_primary_level(nx_graph, nodes_dict)
+    pos_by_label = {node_labels[uid]: pos[uid] for uid in pos}
+    print("pos_by_label:", pos_by_label)
+    descriptive_id = {nodes_dict[uid].descriptive_id: uid for uid in nodes_dict}
+    print("descriptive_id:", descriptive_id)
+    this_dir = pathlib.Path(__file__).parent
+
+    GraphVisualizer(graph).visualise(
+        nodes_layout_function=GraphVisualizer._get_hierarchy_pos_by_distance_to_primary_level,
+        save_path=this_dir / "graph_layout.png"
+    )
+
     print("pos:", pos)
     comparable_lists_y = make_comparable_lists(pos, real_hierarchy_levels_y,
                                                node_labels, 1, reverse=True)