diff options
Diffstat (limited to 'tests/test_cli_config.py')
-rw-r--r-- | tests/test_cli_config.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/tests/test_cli_config.py b/tests/test_cli_config.py index b007052..1487f11 100644 --- a/tests/test_cli_config.py +++ b/tests/test_cli_config.py @@ -7,7 +7,7 @@ from unittest.mock import MagicMock import pytest -from mlia.cli.config import get_ethos_u_default_backends +from mlia.cli.config import get_default_backends from mlia.cli.config import is_corstone_backend @@ -27,9 +27,15 @@ from mlia.cli.config import is_corstone_backend ["Vela", "Corstone-300", "New backend"], ["Vela", "Corstone-300", "New backend"], ], + [["ArmNNTFLiteDelegate"], ["ArmNNTFLiteDelegate"]], + [["TOSA-Checker"], ["TOSA-Checker"]], + [ + ["ArmNNTFLiteDelegate", "Corstone-300"], + ["ArmNNTFLiteDelegate", "Corstone-300"], + ], ], ) -def test_get_ethos_u_default_backends( +def test_get_default_backends( monkeypatch: pytest.MonkeyPatch, available_backends: list[str], expected_default_backends: list[str], @@ -40,7 +46,7 @@ def test_get_ethos_u_default_backends( MagicMock(return_value=available_backends), ) - assert get_ethos_u_default_backends() == expected_default_backends + assert get_default_backends() == expected_default_backends def test_is_corstone_backend() -> None: |