mock_info

SpiNNakerManchester · Oct 9, 2023 · 021a35f · 021a35f
1 parent c26a423
commit 021a35f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/unittests/interface/interface_functions/test_front_end_common_app_finisher.py b/unittests/interface/interface_functions/test_front_end_common_app_finisher.py
@@ -20,7 +20,7 @@
 from spinn_front_end_common.interface.config_setup import unittest_setup
 from spinn_front_end_common.interface.interface_functions import (
     application_finisher)
-from spinnman.model.cpu_infos import CPUInfos
+from spinnman.model import CPUInfo, CPUInfos
 from spinnman.transceiver import Transceiver
 
 
@@ -55,9 +55,11 @@ def get_cpu_infos(
                 if (x, y, p) in core_states:
                     if hasattr(states, "__iter__"):
                         if core_states[x, y, p] in states:
-                            cores_in_state.add_processor(x, y, p, None)
+                            cores_in_state.add_info(CPUInfo.mock_info(
+                                x, y, p, p, core_states[x, y, p]))
                     elif core_states[x, y, p] == states:
-                        cores_in_state.add_processor(x, y, p, None)
+                        cores_in_state.add_info(CPUInfo.mock_info(
+                            x, y, p, p, core_states[x, y, p]))
 
         self._current_state += 1
         return cores_in_state