From 59e057fbbf315419724bf71fe9c7f617248e7aee Mon Sep 17 00:00:00 2001 From: Pavel Macenauer Date: Wed, 15 Apr 2020 14:17:26 +0000 Subject: Integration of PyArmNN into CMake Change-Id: Ice37e693f4598a6b3c38bd38d89f1d35cdaa8a18 Signed-off-by: Pavel Macenauer --- python/pyarmnn/test/test_setup.py | 9 +++++---- python/pyarmnn/test/test_version.py | 5 +++-- 2 files changed, 8 insertions(+), 6 deletions(-) (limited to 'python/pyarmnn/test') 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 -- cgit v1.2.1