diff --git a/unittests/model_tests/test_abstract_process.py b/unittests/model_tests/test_abstract_process.py index b89d12395..4dc195e3b 100644 --- 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 index 77da6f499..e454d2963 100644 --- 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()