VersionStrings as simple Enum

SpiNNakerManchester · Apr 12, 2024 · c1c3ca2 · c1c3ca2
1 parent 2a1816f
commit c1c3ca2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/unittests/model_tests/test_abstract_process.py b/unittests/model_tests/test_abstract_process.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 from spinn_utilities.config_holder import set_config
-from spinn_machine.version import ANY_VERSION
+from spinn_machine.version.version_strings import VersionStrings
 from spinnman.processes.abstract_multi_connection_process import (
     AbstractMultiConnectionProcess)
 from spinnman.messages.scp.impl import ReadMemory
@@ -41,7 +41,7 @@ def receive_scp_response(self, timeout=1.0):
 
 def test_error_print():
     unittest_setup()
-    set_config("Machine", "version", ANY_VERSION)
+    set_config("Machine", "versions", VersionStrings.ANY.text)
     connection = MockConnection(0, 0)
     process = MockProcess(RoundRobinConnectionSelector([connection]))
     with pytest.raises(SpinnmanGenericProcessException):

diff --git a/unittests/test_transceiver.py b/unittests/test_transceiver.py
@@ -15,7 +15,7 @@
 import unittest
 import struct
 from spinn_utilities.config_holder import set_config
-from spinn_machine.version import ANY_VERSION
+from spinn_machine.version.version_strings import VersionStrings
 from spinnman.config_setup import unittest_setup
 from spinnman.data import SpiNNManDataView
 from spinnman.data.spinnman_data_writer import SpiNNManDataWriter
@@ -84,7 +84,7 @@ def test_boot_board(self):
         trans._boot_board()
 
     def test_set_watch_dog(self):
-        set_config("Machine", "version", ANY_VERSION)
+        set_config("Machine", "versions", VersionStrings.ANY.text)
         connections = []
         connections.append(SCAMPConnection(remote_host=None))
         tx = MockExtendedTransceiver()