diff options
Diffstat (limited to 'python/pyarmnn/test')
-rw-r--r-- | python/pyarmnn/test/test_setup.py | 9 | ||||
-rw-r--r-- | python/pyarmnn/test/test_version.py | 5 |
2 files changed, 8 insertions, 6 deletions
diff --git a/python/pyarmnn/test/test_setup.py b/python/pyarmnn/test/test_setup.py index 8396ca0587..d1e6e0f472 100644 --- a/python/pyarmnn/test/test_setup.py +++ b/python/pyarmnn/test/test_setup.py @@ -1,4 +1,5 @@ # Copyright © 2020 Arm Ltd. All rights reserved. +# Copyright 2020 NXP # SPDX-License-Identifier: MIT import os import sys @@ -86,15 +87,15 @@ def test_gcc_serch_path(): def test_armnn_version(): - check_armnn_version('20190800', '20190800') + check_armnn_version('21.0.0', '21.0.0') def test_incorrect_armnn_version(): with pytest.raises(AssertionError) as err: - check_armnn_version('20190800', '20190500') + check_armnn_version('21.0.0', '21.1.0') - assert 'Expected ArmNN version is 201905 but installed ArmNN version is 201908' in str(err.value) + assert 'Expected ArmNN version is 21.1.0 but installed ArmNN version is 21.0.0' in str(err.value) def test_armnn_version_patch_does_not_matter(): - check_armnn_version('20190800', '20190801') + check_armnn_version('21.0.0', '21.0.1') diff --git a/python/pyarmnn/test/test_version.py b/python/pyarmnn/test/test_version.py index 2ea0fd85b4..14a91540fc 100644 --- a/python/pyarmnn/test/test_version.py +++ b/python/pyarmnn/test/test_version.py @@ -1,4 +1,5 @@ # Copyright © 2020 Arm Ltd. All rights reserved. +# Copyright 2020 NXP # SPDX-License-Identifier: MIT import os import importlib @@ -17,7 +18,7 @@ def test_dev_version(): importlib.reload(v) - assert "20.2.0.dev1" == v.__version__ + assert "21.0.0.dev1" == v.__version__ del os.environ["PYARMNN_DEV_VER"] del v @@ -29,7 +30,7 @@ def test_arm_version_not_affected(): importlib.reload(v) - assert "20200200" == v.__arm_ml_version__ + assert "21.0.0" == v.__arm_ml_version__ del os.environ["PYARMNN_DEV_VER"] del v |