Merge branch 'global_data4' into global_data_new

SpiNNakerManchester · Dec 9, 2021 · fabb58b · fabb58b
2 parents 765404c + 52d89a5
commit fabb58b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/pacman/data/pacman_data_writer.py b/pacman/data/pacman_data_writer.py
@@ -74,6 +74,12 @@ def get_graph(self):
     def get_machine_graph(self):
         return self.__pacman_data._machine_graph
 
+    def get_runtime_graph(self):
+        return self.__pacman_data._runtime_graph
+
+    def get_runtime_machine_graph(self):
+        return self.__pacman_data._runtime_machine_graph
+
     def create_graphs(self, graph_label):
         """
         Creates the user/ original graphs based on the label

diff --git a/unittests/data/test_data.py b/unittests/data/test_data.py
@@ -69,6 +69,9 @@ def test_graphs(self):
             writer.runtime_graph
         with self.assertRaises(DataLocked):
             writer.runtime_machine_graph()
+        # the writer still has access to the runtime graphs
+        writer.get_runtime_graph()
+        writer.get_runtime_machine_graph()
 
         writer.start_run()
         writer.runtime_graph
@@ -89,6 +92,9 @@ def test_graphs(self):
             writer.runtime_graph
         with self.assertRaises(DataLocked):
             writer.runtime_machine_graph()
+        # the writer still has access to the runtime graphs
+        writer.get_runtime_graph()
+        writer.get_runtime_machine_graph()
 
         writer.stopping()
         writer.runtime_graph
@@ -109,6 +115,9 @@ def test_graphs(self):
             writer.runtime_graph
         with self.assertRaises(DataLocked):
             writer.runtime_machine_graph()
+        # the writer still has access to the runtime graphs
+        writer.get_runtime_graph()
+        writer.get_runtime_machine_graph()
 
     def test_graph_support(self):
         view = PacmanDataView()