From 3e3dcb9bd5abb88adcd85b4f89e8a81e7f6fa293 Mon Sep 17 00:00:00 2001 From: Dmitrii Agibov Date: Fri, 27 Jan 2023 09:12:50 +0000 Subject: MLIA-595 Remove old backend configuration mechanism - Remove old backend configuration code - Install backends into directory ~/.mlia - Rename targets/backends in registry to make it consistent across codebase. Change-Id: I9c8b012fe863280f1c692940c0dcad3ef638aaae --- tests/test_backend_registry.py | 20 +++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-) (limited to 'tests/test_backend_registry.py') diff --git a/tests/test_backend_registry.py b/tests/test_backend_registry.py index 703e699..e3e2da2 100644 --- a/tests/test_backend_registry.py +++ b/tests/test_backend_registry.py @@ -9,6 +9,8 @@ import pytest from mlia.backend.config import BackendType from mlia.backend.config import System +from mlia.backend.registry import get_supported_backends +from mlia.backend.registry import get_supported_systems from mlia.backend.registry import registry from mlia.core.common import AdviceCategory @@ -35,7 +37,7 @@ from mlia.core.common import AdviceCategory BackendType.CUSTOM, ), ( - "TOSA-Checker", + "tosa-checker", [AdviceCategory.COMPATIBILITY], [System.LINUX_AMD64], BackendType.WHEEL, @@ -79,3 +81,19 @@ def test_backend_registry( cfg.supported_systems ), f"Supported systems differs: {advices} != {cfg.supported_advice}" assert cfg.type == type_ + + +def test_get_supported_backends() -> None: + """Test function get_supported_backends.""" + assert get_supported_backends() == [ + "ArmNNTFLiteDelegate", + "Corstone-300", + "Corstone-310", + "Vela", + "tosa-checker", + ] + + +def test_get_supported_systems() -> None: + """Test function get_supported_systems.""" + assert get_supported_systems() -- cgit v1.2.1