aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/test/test_setup.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/pyarmnn/test/test_setup.py')
-rw-r--r--python/pyarmnn/test/test_setup.py9
1 files changed, 5 insertions, 4 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')