Files
rk35xx-android14/external/armnn/delegate/python/test/conftest.py
T
hmz007 36ed224bac Rockchip Anroid14_SDK 20240628-rkr5 (2556df1a)
Signed-off-by: hmz007 <hmz007@gmail.com>
Change-Id: Ib87fdbe7a0be7dc961af3500fe9ea4589a127f9f
2024-10-29 18:12:02 +08:00

30 lines
913 B
Python

# Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
# SPDX-License-Identifier: MIT
import pytest
import os
@pytest.fixture(scope="module")
def test_data_folder(request):
"""
This fixture returns path to the folder with the shared test resources
"""
return str(os.path.join(request.fspath.dirname, "test_data"))
def pytest_addoption(parser):
"""
Adds the program option 'delegate-dir' to pytest
"""
parser.addoption("--delegate-dir",
action="append",
help="Directory of the armnn tflite delegate library",
required=True)
def pytest_generate_tests(metafunc):
"""
Makes the program option 'delegate-dir' available to all tests as a function fixture
"""
if "delegate_dir" in metafunc.fixturenames:
metafunc.parametrize("delegate_dir", metafunc.config.getoption("delegate_dir"))