aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorBenjamin Klimczak <benjamin.klimczak@arm.com>2023-11-24 12:23:16 +0000
committerBenjamin Klimczak <benjamin.klimczak@arm.com>2023-11-24 17:02:41 +0000
commit2b5f217fb21e2d100baabd6161c8470b65669ed9 (patch)
tree1f4b3e8d9e1cfcb968f01f901357d54a14a3def5 /tests
parent82a5fe34d7464bba70577c734dc446111adb4d93 (diff)
downloadmlia-2b5f217fb21e2d100baabd6161c8470b65669ed9.tar.gz
MLIA-1037 Fix default backend validation
Add a check to see if the default backends are installed when no backend is provided via CLI. Change-Id: I27dd9f35cfeec187f44cba06915d1be5a3a052b5 Signed-off-by: Benjamin Klimczak <benjamin.klimczak@arm.com>
Diffstat (limited to 'tests')
-rw-r--r--tests/test_cli_command_validators.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/tests/test_cli_command_validators.py b/tests/test_cli_command_validators.py
index 1ac82db..e140749 100644
--- a/tests/test_cli_command_validators.py
+++ b/tests/test_cli_command_validators.py
@@ -5,6 +5,7 @@ from __future__ import annotations
import argparse
from contextlib import ExitStack
+from unittest.mock import MagicMock
import pytest
@@ -160,3 +161,20 @@ def test_validate_backend(
with exit_stack:
assert validate_backend(input_target_profile, input_backends) == output_backends
+
+
+def test_validate_backend_default_available() -> None:
+ """Test default backend validation with available backend."""
+ backends = validate_backend("cortex-a", None)
+ assert backends
+ assert backends == ["armnn-tflite-delegate"]
+
+
+def test_validate_backend_default_unavailable(monkeypatch: pytest.MonkeyPatch) -> None:
+ """Test default backend validation with unavailable backend."""
+ monkeypatch.setattr(
+ "mlia.cli.command_validators.default_backends",
+ MagicMock(return_value=["UNKNOWN_BACKEND"]),
+ )
+ with pytest.raises(argparse.ArgumentError):
+ validate_backend("cortex-a", None)