From 0efca3cadbad5517a59884576ddb90cfe7ac30f8 Mon Sep 17 00:00:00 2001
From: Diego Russo <diego.russo@arm.com>
Date: Mon, 30 May 2022 13:34:14 +0100
Subject: Add MLIA codebase

Add MLIA codebase including sources and tests.

Change-Id: Id41707559bd721edd114793618d12ccd188d8dbd
---
 tests/aiet/conftest.py | 139 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 139 insertions(+)
 create mode 100644 tests/aiet/conftest.py

(limited to 'tests/aiet/conftest.py')

diff --git a/tests/aiet/conftest.py b/tests/aiet/conftest.py
new file mode 100644
index 0000000..cab3dc2
--- /dev/null
+++ b/tests/aiet/conftest.py
@@ -0,0 +1,139 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+# pylint: disable=redefined-outer-name
+"""conftest for pytest."""
+import shutil
+import tarfile
+from pathlib import Path
+from typing import Any
+
+import pytest
+from click.testing import CliRunner
+
+from aiet.backend.common import get_backend_configs
+
+
+@pytest.fixture(scope="session")
+def test_systems_path(test_resources_path: Path) -> Path:
+    """Return test systems path in a pytest fixture."""
+    return test_resources_path / "systems"
+
+
+@pytest.fixture(scope="session")
+def test_applications_path(test_resources_path: Path) -> Path:
+    """Return test applications path in a pytest fixture."""
+    return test_resources_path / "applications"
+
+
+@pytest.fixture(scope="session")
+def test_tools_path(test_resources_path: Path) -> Path:
+    """Return test tools path in a pytest fixture."""
+    return test_resources_path / "tools"
+
+
+@pytest.fixture(scope="session")
+def test_resources_path() -> Path:
+    """Return test resources path in a pytest fixture."""
+    current_path = Path(__file__).parent.absolute()
+    return current_path / "test_resources"
+
+
+@pytest.fixture(scope="session")
+def non_optimised_input_model_file(test_tflite_model: Path) -> Path:
+    """Provide the path to a quantized dummy model file."""
+    return test_tflite_model
+
+
+@pytest.fixture(scope="session")
+def optimised_input_model_file(test_tflite_vela_model: Path) -> Path:
+    """Provide path to Vela-optimised dummy model file."""
+    return test_tflite_vela_model
+
+
+@pytest.fixture(scope="session")
+def invalid_input_model_file(test_tflite_invalid_model: Path) -> Path:
+    """Provide the path to an invalid dummy model file."""
+    return test_tflite_invalid_model
+
+
+@pytest.fixture(autouse=True)
+def test_resources(monkeypatch: pytest.MonkeyPatch, test_resources_path: Path) -> Any:
+    """Force using test resources as middleware's repository."""
+
+    def get_test_resources() -> Path:
+        """Return path to the test resources."""
+        return test_resources_path
+
+    monkeypatch.setattr("aiet.utils.fs.get_aiet_resources", get_test_resources)
+    yield
+
+
+@pytest.fixture(scope="session", autouse=True)
+def add_tools(test_resources_path: Path) -> Any:
+    """Symlink the tools from the original resources path to the test resources path."""
+    # tool_dirs = get_available_tool_directory_names()
+    tool_dirs = [cfg.parent for cfg in get_backend_configs("tools")]
+
+    links = {
+        src_dir: (test_resources_path / "tools" / src_dir.name) for src_dir in tool_dirs
+    }
+    for src_dir, dst_dir in links.items():
+        if not dst_dir.exists():
+            dst_dir.symlink_to(src_dir, target_is_directory=True)
+    yield
+    # Remove symlinks
+    for dst_dir in links.values():
+        if dst_dir.is_symlink():
+            dst_dir.unlink()
+
+
+def create_archive(
+    archive_name: str, source: Path, destination: Path, with_root_folder: bool = False
+) -> None:
+    """Create archive from directory source."""
+    with tarfile.open(destination / archive_name, mode="w:gz") as tar:
+        for item in source.iterdir():
+            item_name = item.name
+            if with_root_folder:
+                item_name = f"{source.name}/{item_name}"
+            tar.add(item, item_name)
+
+
+def process_directory(source: Path, destination: Path) -> None:
+    """Process resource directory."""
+    destination.mkdir()
+
+    for item in source.iterdir():
+        if item.is_dir():
+            create_archive(f"{item.name}.tar.gz", item, destination)
+            create_archive(f"{item.name}_dir.tar.gz", item, destination, True)
+
+
+@pytest.fixture(scope="session", autouse=True)
+def add_archives(
+    test_resources_path: Path, tmp_path_factory: pytest.TempPathFactory
+) -> Any:
+    """Generate archives of the test resources."""
+    tmp_path = tmp_path_factory.mktemp("archives")
+
+    archives_path = tmp_path / "archives"
+    archives_path.mkdir()
+
+    if (archives_path_link := test_resources_path / "archives").is_symlink():
+        archives_path.unlink()
+
+    archives_path_link.symlink_to(archives_path, target_is_directory=True)
+
+    for item in ["applications", "systems"]:
+        process_directory(test_resources_path / item, archives_path / item)
+
+    yield
+
+    archives_path_link.unlink()
+    shutil.rmtree(tmp_path)
+
+
+@pytest.fixture(scope="module")
+def cli_runner() -> CliRunner:
+    """Return CliRunner instance in a pytest fixture."""
+    return CliRunner()
-- 
cgit v1.2.1