| # 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")) |