diff options
Diffstat (limited to 'tests/test_utils_misc.py')
-rw-r--r-- | tests/test_utils_misc.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/tests/test_utils_misc.py b/tests/test_utils_misc.py index 011d09e..ae91850 100644 --- a/tests/test_utils_misc.py +++ b/tests/test_utils_misc.py @@ -1,10 +1,11 @@ -# 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 misc util functions.""" from unittest.mock import MagicMock import pytest +from mlia.utils.misc import get_pkg_version from mlia.utils.misc import yes @@ -23,3 +24,10 @@ def test_yes( """Test yes function.""" monkeypatch.setattr("builtins.input", MagicMock(return_value=response)) assert yes("some_prompt") == expected_result + + +@pytest.mark.parametrize("response", ["some version", FileNotFoundError()]) +def test_get_pkg_version(monkeypatch: pytest.MonkeyPatch, response: str) -> None: + """Test get_tosa_version.""" + monkeypatch.setattr("importlib.metadata.version", MagicMock(return_value=response)) + assert get_pkg_version("any name") == response |