blob: 83606ab15bbaca3f8bd8b262fecafa7ef3dfc3bc [file] [log] [blame]
Richard Burtondc0c6ed2020-04-08 16:39:05 +01001# Copyright © 2020 Arm Ltd. All rights reserved.
Pavel Macenauer59e057f2020-04-15 14:17:26 +00002# Copyright 2020 NXP
Richard Burtondc0c6ed2020-04-08 16:39:05 +01003# SPDX-License-Identifier: MIT
4import os
5import importlib
6
7
8def test_rel_version():
9 import pyarmnn._version as v
10 importlib.reload(v)
11 assert "dev" not in v.__version__
12 del v
13
14
15def test_dev_version():
16 import pyarmnn._version as v
17 os.environ["PYARMNN_DEV_VER"] = "1"
18
19 importlib.reload(v)
20
Colm Donelan03bf98a2022-05-30 15:20:36 +010021 assert "30.0.0.dev1" == v.__version__
Richard Burtondc0c6ed2020-04-08 16:39:05 +010022
23 del os.environ["PYARMNN_DEV_VER"]
24 del v
25
26
27def test_arm_version_not_affected():
28 import pyarmnn._version as v
29 os.environ["PYARMNN_DEV_VER"] = "1"
30
31 importlib.reload(v)
32
Colm Donelan03bf98a2022-05-30 15:20:36 +010033 assert "30.0.0" == v.__arm_ml_version__
Richard Burtondc0c6ed2020-04-08 16:39:05 +010034
35 del os.environ["PYARMNN_DEV_VER"]
36 del v