File: conftest.py

package info (click to toggle)
armnn 23.08-5
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 53,580 kB
  • sloc: cpp: 337,440; python: 4,755; sh: 1,708; makefile: 42; asm: 15; xml: 6
file content (30 lines) | stat: -rw-r--r-- 913 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# 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"))