From 245d64c60d0ea30f5080ff53225b5169927e24d6 Mon Sep 17 00:00:00 2001 From: Matthew Bentham Date: Mon, 2 Dec 2019 12:59:43 +0000 Subject: Work in progress of python bindings for Arm NN Not built or tested in any way Signed-off-by: Matthew Bentham Change-Id: Ie7f92b529aa5087130f0c5cc8c17db1581373236 --- python/pyarmnn/test/test_supported_backends.py | 51 ++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) create mode 100644 python/pyarmnn/test/test_supported_backends.py (limited to 'python/pyarmnn/test/test_supported_backends.py') diff --git a/python/pyarmnn/test/test_supported_backends.py b/python/pyarmnn/test/test_supported_backends.py new file mode 100644 index 0000000000..443f8bac08 --- /dev/null +++ b/python/pyarmnn/test/test_supported_backends.py @@ -0,0 +1,51 @@ +# Copyright © 2019 Arm Ltd. All rights reserved. +# SPDX-License-Identifier: MIT. +import os +import platform +import pytest +import pyarmnn as ann + + +@pytest.fixture() +def get_supported_backends_setup(shared_data_folder): + options = ann.CreationOptions() + runtime = ann.IRuntime(options) + + get_device_spec = runtime.GetDeviceSpec() + supported_backends = get_device_spec.GetSupportedBackends() + + yield supported_backends + + +def test_ownership(): + options = ann.CreationOptions() + runtime = ann.IRuntime(options) + + device_spec = runtime.GetDeviceSpec() + + assert not device_spec.thisown + + +def test_to_string(): + options = ann.CreationOptions() + runtime = ann.IRuntime(options) + + device_spec = runtime.GetDeviceSpec() + expected_str = "IDeviceSpec {{ supportedBackends: [" \ + "{}" \ + "]}}".format(', '.join(map(lambda b: str(b), device_spec.GetSupportedBackends()))) + + assert expected_str == str(device_spec) + + +def test_get_supported_backends_cpu_ref(get_supported_backends_setup): + assert "CpuRef" in map(lambda b: str(b), get_supported_backends_setup) + + +@pytest.mark.juno +class TestNoneCpuRefBackends: + + @pytest.mark.parametrize("backend",["CpuAcc", "NpuAcc"]) + def test_get_supported_backends_cpu_acc(self, get_supported_backends_setup, backend): + assert backend in map(lambda b: str(b), get_supported_backends_setup) + -- cgit v1.2.1