aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/test/test_version.py
diff options
context:
space:
mode:
authorRichard Burton <richard.burton@arm.com>2020-04-08 16:39:05 +0100
committerJim Flynn <jim.flynn@arm.com>2020-04-10 16:11:09 +0000
commitdc0c6ed9f8b993e63f492f203d7d7080ab4c835c (patch)
treeea8541990b13ebf1a038009aa6b8b4b1ea8c3f55 /python/pyarmnn/test/test_version.py
parentfe5a24beeef6e9a41366e694f41093565e748048 (diff)
downloadarmnn-dc0c6ed9f8b993e63f492f203d7d7080ab4c835c.tar.gz
Add PyArmNN to work with ArmNN API of 20.02
* Add Swig rules for generating python wrapper * Add documentation * Add tests and testing data Change-Id: If48eda08931514fa21e72214dfead2835f07237c Signed-off-by: Richard Burton <richard.burton@arm.com> Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Diffstat (limited to 'python/pyarmnn/test/test_version.py')
-rw-r--r--python/pyarmnn/test/test_version.py35
1 files changed, 35 insertions, 0 deletions
diff --git a/python/pyarmnn/test/test_version.py b/python/pyarmnn/test/test_version.py
new file mode 100644
index 0000000000..2ea0fd85b4
--- /dev/null
+++ b/python/pyarmnn/test/test_version.py
@@ -0,0 +1,35 @@
+# Copyright © 2020 Arm Ltd. All rights reserved.
+# SPDX-License-Identifier: MIT
+import os
+import importlib
+
+
+def test_rel_version():
+ import pyarmnn._version as v
+ importlib.reload(v)
+ assert "dev" not in v.__version__
+ del v
+
+
+def test_dev_version():
+ import pyarmnn._version as v
+ os.environ["PYARMNN_DEV_VER"] = "1"
+
+ importlib.reload(v)
+
+ assert "20.2.0.dev1" == v.__version__
+
+ del os.environ["PYARMNN_DEV_VER"]
+ del v
+
+
+def test_arm_version_not_affected():
+ import pyarmnn._version as v
+ os.environ["PYARMNN_DEV_VER"] = "1"
+
+ importlib.reload(v)
+
+ assert "20200200" == v.__arm_ml_version__
+
+ del os.environ["PYARMNN_DEV_VER"]
+ del v