diff options
author | Annie Tallund <annie.tallund@arm.com> | 2022-12-14 15:55:19 +0100 |
---|---|---|
committer | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2023-02-08 15:17:12 +0000 |
commit | 09ecc5c8acb758e8def33155feb746a34dd7b560 (patch) | |
tree | 65c39a7f7929b745b9c5a31ab48bb4c6e97cb3ea /tests/test_utils_filesystem.py | |
parent | 6fbcffb4ee039438a409fbc92e38fa5d1d118833 (diff) | |
download | mlia-09ecc5c8acb758e8def33155feb746a34dd7b560.tar.gz |
MLIA-590 Support path to custom target profiles
- Start using TOML format for target profile
- Add support for loading custom target profile files
Change-Id: I6be019d4341e93115440ccdbdb6dafdc1c85b966
Diffstat (limited to 'tests/test_utils_filesystem.py')
-rw-r--r-- | tests/test_utils_filesystem.py | 59 |
1 files changed, 13 insertions, 46 deletions
diff --git a/tests/test_utils_filesystem.py b/tests/test_utils_filesystem.py index 9dd51e1..954f9e3 100644 --- a/tests/test_utils_filesystem.py +++ b/tests/test_utils_filesystem.py @@ -1,21 +1,18 @@ -# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 """Tests for the filesystem module.""" import contextlib -import json from pathlib import Path -from unittest.mock import MagicMock import pytest from mlia.utils.filesystem import all_files_exist from mlia.utils.filesystem import all_paths_valid from mlia.utils.filesystem import copy_all +from mlia.utils.filesystem import get_builtin_supported_profile_names from mlia.utils.filesystem import get_mlia_resources +from mlia.utils.filesystem import get_mlia_target_profiles_dir from mlia.utils.filesystem import get_profile -from mlia.utils.filesystem import get_profiles_data -from mlia.utils.filesystem import get_profiles_file -from mlia.utils.filesystem import get_supported_profile_names from mlia.utils.filesystem import get_vela_config from mlia.utils.filesystem import sha256 from mlia.utils.filesystem import temp_directory @@ -34,50 +31,20 @@ def test_get_vela_config() -> None: assert get_vela_config().name == "vela.ini" -def test_profiles_file() -> None: - """Test profiles file getter.""" - assert get_profiles_file().is_file() - assert get_profiles_file().name == "profiles.json" +def test_get_mlia_target_profiles() -> None: + """Test target profiles getter.""" + assert get_mlia_target_profiles_dir().is_dir() -def test_profiles_data() -> None: - """Test profiles data getter.""" - assert list(get_profiles_data().keys()) == [ - "ethos-u55-256", - "ethos-u55-128", - "ethos-u65-512", - "ethos-u65-256", - "tosa", - "cortex-a", - ] - - -def test_profiles_data_wrong_format( - monkeypatch: pytest.MonkeyPatch, tmp_path: Path -) -> None: - """Test if profile data has wrong format.""" - wrong_profile_data = tmp_path / "bad.json" - with open(wrong_profile_data, "w", encoding="utf-8") as file: - json.dump([], file) - - monkeypatch.setattr( - "mlia.utils.filesystem.get_profiles_file", - MagicMock(return_value=wrong_profile_data), - ) - - with pytest.raises(Exception, match="Profiles data format is not valid"): - get_profiles_data() - - -def test_get_supported_profile_names() -> None: +def test_get_builtin_supported_profile_names() -> None: """Test profile names getter.""" - assert list(get_supported_profile_names()) == [ - "ethos-u55-256", + assert get_builtin_supported_profile_names() == [ + "cortex-a", "ethos-u55-128", - "ethos-u65-512", + "ethos-u55-256", "ethos-u65-256", + "ethos-u65-512", "tosa", - "cortex-a", ] @@ -86,11 +53,11 @@ def test_get_profile() -> None: assert get_profile("ethos-u55-256") == { "target": "ethos-u55", "mac": 256, - "system_config": "Ethos_U55_High_End_Embedded", "memory_mode": "Shared_Sram", + "system_config": "Ethos_U55_High_End_Embedded", } - with pytest.raises(Exception, match="Unable to find target profile unknown"): + with pytest.raises(Exception, match=r"File not found:*"): get_profile("unknown") |