aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/test
diff options
context:
space:
mode:
authorPavel Macenauer <pavel.macenauer@linaro.org>2020-04-15 14:17:26 +0000
committerPavel Macenauer <pavel.macenauer@linaro.org>2020-05-21 16:07:49 +0200
commit59e057fbbf315419724bf71fe9c7f617248e7aee (patch)
tree33a60c365494a3bf1b65a06ad5a450d69763c4bd /python/pyarmnn/test
parent0800bf541a3c6f5f2d6ff842a30e20ed48fb2eac (diff)
downloadarmnn-59e057fbbf315419724bf71fe9c7f617248e7aee.tar.gz
Integration of PyArmNN into CMake
Change-Id: Ice37e693f4598a6b3c38bd38d89f1d35cdaa8a18 Signed-off-by: Pavel Macenauer <pavel.macenauer@nxp.com>
Diffstat (limited to 'python/pyarmnn/test')
-rw-r--r--python/pyarmnn/test/test_setup.py9
-rw-r--r--python/pyarmnn/test/test_version.py5
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