From dc0c6ed9f8b993e63f492f203d7d7080ab4c835c Mon Sep 17 00:00:00 2001 From: Richard Burton Date: Wed, 8 Apr 2020 16:39:05 +0100 Subject: 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 Signed-off-by: Derek Lamberti --- python/pyarmnn/test/test_version.py | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 python/pyarmnn/test/test_version.py (limited to 'python/pyarmnn/test/test_version.py') 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 -- cgit v1.2.1