diff --git a/unittests/model_tests/test_abstract_process.py b/unittests/model_tests/test_abstract_process.py index baa7a0d25..b89d12395 100644 --- a/unittests/model_tests/test_abstract_process.py +++ b/unittests/model_tests/test_abstract_process.py @@ -13,6 +13,7 @@ # limitations under the License. from spinn_utilities.config_holder import set_config +from spinn_machine.version import ANY_VERSION from spinnman.processes.abstract_multi_connection_process import ( AbstractMultiConnectionProcess) from spinnman.messages.scp.impl import ReadMemory @@ -40,7 +41,7 @@ def receive_scp_response(self, timeout=1.0): def test_error_print(): unittest_setup() - set_config("Machine", "version", 5) + set_config("Machine", "version", ANY_VERSION) 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 d0584a830..77da6f499 100644 --- a/unittests/test_transceiver.py +++ b/unittests/test_transceiver.py @@ -15,6 +15,7 @@ import unittest import struct from spinn_utilities.config_holder import set_config +from spinn_machine.version import ANY_VERSION from spinnman.config_setup import unittest_setup from spinnman.data import SpiNNManDataView from spinnman.data.spinnman_data_writer import SpiNNManDataWriter @@ -83,7 +84,7 @@ def test_boot_board(self): trans._boot_board() def test_set_watch_dog(self): - set_config("Machine", "version", 5) + set_config("Machine", "version", ANY_VERSION) connections = [] connections.append(SCAMPConnection(remote_host=None)) tx = MockExtendedTransceiver()