data_specification_loaded to turn off requires data spec / reset

spinn_utilities/data/utils_data_view.py

@@ -536,8 +536,7 @@ def get_requires_data_generation(cls) -> bool:
         Set to True at the start and by any change that could require
         data generation or mapping
         Remains True during the first run after a data change
-        Only set to False at the *end* of the first run
-
+        Only set to False after the data is loaded.
         """
         return cls.__data._requires_data_generation
 
@@ -546,7 +545,7 @@ def set_requires_data_generation(cls) -> None:
         """
         Sets `requires_data_generation` to True.
 
-        Only the end of a run can set it to False
+        Set to False after data is loaded
         """
         cls.check_user_can_act()
         cls.__data._requires_data_generation = True
@@ -588,7 +587,6 @@ def _mock_has_run(cls) -> None:
         """
         cls.__data._run_status = RunStatus.NOT_RUNNING
         cls.__data._reset_status = ResetStatus.HAS_RUN
-        cls.__data._requires_data_generation = False
         cls.__data._requires_mapping = False
 
     @classmethod

spinn_utilities/data/utils_data_writer.py

@@ -146,7 +146,6 @@ def finish_run(self) -> None:
                 f"{self.__data._run_status}")
         self.__data._run_status = RunStatus.NOT_RUNNING
         self.__data._reset_status = ResetStatus.HAS_RUN
-        self.__data._requires_data_generation = False
         self.__data._requires_mapping = False
 
     def _hard_reset(self) -> None:
@@ -205,6 +204,23 @@ def soft_reset(self) -> None:
             f"Unexpected call to reset while reset status is "
             f"{self.__data._reset_status}")
 
+    def data_specification_loaded(self) -> None:
+        """
+        Used to indicate that load data specification has run/
+    .
+        Causes get_requires_data_generation to report False
+        """
+        self.__data._requires_data_generation = False
+
+    @classmethod
+    def set_requires_data_generation(cls) -> None:
+        """
+        Sets `requires_data_generation` to True.
+
+        Set to False after data is loaded
+        """
+        cls.__data._requires_data_generation = True
+
     def request_stop(self) -> None:
         """
         Used to indicate a user has requested a stop.

unittests/data/test_utils_data.py

@@ -1030,13 +1030,20 @@ def test_requires(self):
         self.assertTrue(writer.get_requires_mapping())
         # Can not be changed during run
         with self.assertRaises(SimulatorRunningException):
-            writer.set_requires_data_generation()
+            UtilsDataView.set_requires_data_generation()
         with self.assertRaises(SimulatorRunningException):
             writer.set_requires_mapping()
+        writer.data_specification_loaded()
+        self.assertFalse(writer.get_requires_data_generation())
+        with self.assertRaises(SimulatorRunningException):
+            UtilsDataView.set_requires_data_generation()
+        writer.set_requires_data_generation()
+        self.assertTrue(writer.get_requires_data_generation())
         writer.finish_run()
 
+        # Stay as it was
+        self.assertTrue(writer.get_requires_data_generation())
         # False after run
-        self.assertFalse(writer.get_requires_data_generation())
         self.assertFalse(writer.get_requires_mapping())
 
         # Setting requires mapping sets both to True