python 测试，如何处理“继续”语句循环

Question

我正在学习如何使用 Python。 我有一个 function 里面有一个条件，如果提供了无效的输入，它应该重新启动循环，直到提供有效的输入。

不幸的是，这种“重新启动”行为在我的测试中导致了无限循环（它循环提供了错误的输入）。 如何暂停、中断或将 output 限制为一个实例，以便测试返回的字符串？

function：

def confirm_user_choice(choice: str):
    while True:
        user_response = input(f"\nYou chose '{choice}', is this correct? y/n ")
        if user_response == "y":
            return True
        elif user_response == "n":
            return False
        else:
            print("\nSelect either 'y' (yes) or 'n' (no)")

测试：

import unittest
from unittest import mock
from src.utils.utils import addValues, confirm_user_choice


class TestConfirmUserChoice(unittest.TestCase):
    def test_yes(self):
        with mock.patch("builtins.input", return_value="y"):
            result = confirm_user_choice("y")
        self.assertEqual(result, True)

    def test_no(self):
        with mock.patch("builtins.input", return_value="n"):
            result = confirm_user_choice("n")
        self.assertEqual(result, False)

    def test_invalid_input(self):
        with mock.patch("builtins.input", return_value="apple"):   <-- triggers func else case
            result = confirm_user_choice("apple")
        self.assertEqual(result, False)

Answer 1

您有部分 function：在正确的输入上，它将返回 Boolean 值，但它可能根本不会返回，并且您无法测试无限循环确实是无限的。

为了使其更具可测试性，允许 function 采用默认为sys.stdin的可选可迭代值，允许您控制 function 读取的内容（以及尝试这样做的时间。）

def confirm_user_choice(choice: str, responses: Optional[Iterable[str]] = None):
    if responses is None:
        # An infinite stream of calls to input()
        responses = iter(lambda: input(f"\nYou chose '{choice}', is this correct? y/n "), None)

    for user_response in responses:
        if user_response == "y":
            return True
        elif user_response == "n":
            return False
        else:
            print("\nSelect either 'y' (yes) or 'n' (no)")
    else:
        # Note: cannot be raised from the default value of responses
        raise ValueError("Unexpected end of responses")

现在您的测试可以简单地通过预设的响应列表，然后捕获预期的ValueError ，或者查看返回的 Boolean 值。

import unittest
from src.utils.utils import addValues, confirm_user_choice


class TestConfirmUserChoice(unittest.TestCase):
    def test_yes(self):
        result = confirm_user_choice("y", ["y"])
        self.assertTrue(result)

    def test_eventual_yes(self):
        result = confirm_user_choice("y", ["apple", "pear", "y"])
        self.assertTrue(result)

    def test_no(self):
        result = confirm_user_choice("y", ["n"])
        self.assertFalse(result)

    def test_no_valid_input(self):
        with self.assertRaises(ValueError):
            result = confirm_user_choice(["apple"])

Answer 2

continue在您的代码中不执行任何操作
continue允许您忽略某些循环实例的部分代码。

例如：

for i in range(2):
   if i < 1:
      continue
   print(i)

Output：

1

对于您想要做的事情，不要忘记当满足条件时， while应该结束。 因此，使用while True:绕过条件，然后使用 if 退出循环有点适得其反。

只需使用 while 条件：

user_response = ""
while user_response not in ["y", "n"]:
    user_response = input("y/n ? ")
print(user_response)

快乐编程

Answer 3

我自己是 Python 的新手，但据我了解，单元测试会调查 function 如何根据函数的返回值或引发的异常（如果有）处理不同的输入。

您的 function 仅在用户输入“y”或“n”或引发错误时退出（例如，如果用户提供 Crtl-Z）。 当用户输入“apple”时，您的 while 循环不会中断。 pytest（或类似的）没有返回值来检查。

如果您真的想对此进行测试，则必须重写您的 function 以便更加模块化。 它必须具有至少三个不同的返回值，包括一个暗示输入无效的返回值。