diff options
Diffstat (limited to 'tests/test_utils_misc.py')
-rw-r--r-- | tests/test_utils_misc.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/tests/test_utils_misc.py b/tests/test_utils_misc.py new file mode 100644 index 0000000..011d09e --- /dev/null +++ b/tests/test_utils_misc.py @@ -0,0 +1,25 @@ +# SPDX-FileCopyrightText: Copyright 2022, 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 yes + + +@pytest.mark.parametrize( + "response, expected_result", + [ + ["Y", True], + ["y", True], + ["N", False], + ["n", False], + ], +) +def test_yes( + monkeypatch: pytest.MonkeyPatch, expected_result: bool, response: str +) -> None: + """Test yes function.""" + monkeypatch.setattr("builtins.input", MagicMock(return_value=response)) + assert yes("some_prompt") == expected_result |