diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_backend_tosa_compat.py | 5 | ||||
| -rw-r--r-- | tests/test_cli_main.py | 29 | ||||
| -rw-r--r-- | tests/test_utils_py_manager.py | 30 |
3 files changed, 52 insertions, 12 deletions
diff --git a/tests/test_backend_tosa_compat.py b/tests/test_backend_tosa_compat.py index 4c4dc5a..52aaa44 100644 --- a/tests/test_backend_tosa_compat.py +++ b/tests/test_backend_tosa_compat.py @@ -10,6 +10,7 @@ from unittest.mock import MagicMock import pytest +from mlia.backend.errors import BackendUnavailableError from mlia.backend.tosa_checker.compat import get_tosa_compatibility_info from mlia.backend.tosa_checker.compat import Operator from mlia.backend.tosa_checker.compat import TOSACompatibilityInfo @@ -31,7 +32,9 @@ def test_compatibility_check_should_fail_if_checker_not_available( """Test that compatibility check should fail if TOSA checker is not available.""" replace_get_tosa_checker_with_mock(monkeypatch, None) - with pytest.raises(Exception, match="TOSA checker is not available"): + with pytest.raises( + BackendUnavailableError, match="Backend tosa-checker is not available" + ): get_tosa_compatibility_info(test_tflite_model) diff --git a/tests/test_cli_main.py b/tests/test_cli_main.py index 5f6beb6..925f1e4 100644 --- a/tests/test_cli_main.py +++ b/tests/test_cli_main.py @@ -15,10 +15,12 @@ from unittest.mock import MagicMock import pytest import mlia +from mlia.backend.errors import BackendUnavailableError from mlia.cli.main import backend_main from mlia.cli.main import CommandInfo from mlia.cli.main import main from mlia.core.context import ExecutionContext +from mlia.core.errors import InternalError from tests.utils.logging import clear_loggers @@ -358,6 +360,33 @@ def test_commands_execution_backend_main( MagicMock(side_effect=KeyboardInterrupt()), ["Execution has been interrupted"], ], + [ + False, + MagicMock( + side_effect=BackendUnavailableError( + "Backend sample is not available", "sample" + ) + ), + ["Error: Backend sample is not available."], + ], + [ + False, + MagicMock( + side_effect=BackendUnavailableError( + "Backend tosa-checker is not available", "tosa-checker" + ) + ), + [ + "Error: Backend tosa-checker is not available.", + "Please use next command to install it: " + 'mlia-backend install "tosa-checker"', + ], + ], + [ + False, + MagicMock(side_effect=InternalError("Unknown error")), + ["Internal error: Unknown error"], + ], ], ) def test_verbose_output( diff --git a/tests/test_utils_py_manager.py b/tests/test_utils_py_manager.py index e41680d..c9047f1 100644 --- a/tests/test_utils_py_manager.py +++ b/tests/test_utils_py_manager.py @@ -1,6 +1,7 @@ # SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 """Tests for python package manager.""" +import subprocess # nosec import sys from unittest.mock import MagicMock @@ -16,13 +17,16 @@ def test_get_package_manager() -> None: assert isinstance(manager, PyPackageManager) -@pytest.fixture(name="mock_check_call") -def mock_check_call_fixture(monkeypatch: pytest.MonkeyPatch) -> MagicMock: +@pytest.fixture(name="mock_check_output") +def mock_check_output_fixture(monkeypatch: pytest.MonkeyPatch) -> MagicMock: """Mock check_call function.""" - mock_check_call = MagicMock() - monkeypatch.setattr("mlia.utils.py_manager.check_call", mock_check_call) + mock_check_output = MagicMock() - return mock_check_call + monkeypatch.setattr( + "mlia.utils.py_manager.subprocess.check_output", mock_check_output + ) + + return mock_check_output def test_py_package_manager_metadata() -> None: @@ -32,14 +36,14 @@ def test_py_package_manager_metadata() -> None: assert manager.packages_installed(["pytest", "mlia"]) -def test_py_package_manager_install(mock_check_call: MagicMock) -> None: +def test_py_package_manager_install(mock_check_output: MagicMock) -> None: """Test package installation.""" manager = PyPackageManager() with pytest.raises(ValueError, match="No package names provided"): manager.install([]) manager.install(["mlia", "pytest"]) - mock_check_call.assert_called_once_with( + mock_check_output.assert_called_once_with( [ sys.executable, "-m", @@ -48,18 +52,20 @@ def test_py_package_manager_install(mock_check_call: MagicMock) -> None: "install", "mlia", "pytest", - ] + ], + stderr=subprocess.STDOUT, + text=True, ) -def test_py_package_manager_uninstall(mock_check_call: MagicMock) -> None: +def test_py_package_manager_uninstall(mock_check_output: MagicMock) -> None: """Test package removal.""" manager = PyPackageManager() with pytest.raises(ValueError, match="No package names provided"): manager.uninstall([]) manager.uninstall(["mlia", "pytest"]) - mock_check_call.assert_called_once_with( + mock_check_output.assert_called_once_with( [ sys.executable, "-m", @@ -69,5 +75,7 @@ def test_py_package_manager_uninstall(mock_check_call: MagicMock) -> None: "--yes", "mlia", "pytest", - ] + ], + stderr=subprocess.STDOUT, + text=True, ) |