diff options
Diffstat (limited to 'python/pyarmnn/examples/tests/test_common_utils.py')
-rw-r--r-- | python/pyarmnn/examples/tests/test_common_utils.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/python/pyarmnn/examples/tests/test_common_utils.py b/python/pyarmnn/examples/tests/test_common_utils.py index 28d68ea235..254eba63f8 100644 --- a/python/pyarmnn/examples/tests/test_common_utils.py +++ b/python/pyarmnn/examples/tests/test_common_utils.py @@ -2,9 +2,13 @@ # SPDX-License-Identifier: MIT import os +import time +import cv2 +import numpy as np from context import cv_utils from context import utils +from utils import Profiling def test_get_source_encoding(test_data_folder): @@ -17,3 +21,22 @@ def test_read_existing_labels_file(test_data_folder): label_file = os.path.join(test_data_folder, "labelmap.txt") labels_map = utils.dict_labels(label_file) assert labels_map is not None + + +def test_preprocess(test_data_folder): + content_image = "messi5.jpg" + target_shape = (1, 256, 256, 3) + padding = True + image = cv2.imread(os.path.join(test_data_folder, content_image)) + image = cv_utils.preprocess(image, np.float32, target_shape, True, padding) + + assert image.shape == target_shape + + +def test_profiling(): + profiler = Profiling(True) + profiler.profiling_start() + time.sleep(1) + period = profiler.profiling_stop_and_print_us("Sleep for 1 second") + assert (1_000_000 < period < 1_002_000) + |