Python 3 - 具有多个输入的单元测试和使用模拟的打印语句

Question

我正在研究 Python，几周前我创建了一个游戏，用户需要猜测用户自己定义的间隔之间的数字。现在我正在学习 Unittest，我决定为游戏编写一个测试模块。然而，由于它需要来自用户的 4 个输入（其中两个定义了将生成随机数的范围，一个是用户的猜测，最后一个是一个 Y/N 问题，供用户决定他是否想要继续。import randomdef main():    print('Welcome to the guess game!')    while True:        try:            low_param = int(input('Please enter the lower number: '))            high_param = int(input('Please enter the higher number: '))             if high_param <= low_param:                print('No, first the lower number, then the higher number!')            else:                break        except:            print('You need to enter a number!')    while True:        try:            result = random.randint(low_param, high_param)            guess = int(input(f'Please enter a number between {low_param} and {high_param}: '))            if low_param <= guess <= high_param:                if result == guess:                    print('Nice, dude!')                    break                else:                    print ('Not yet, chap')                while True:                    try_again = input('Would you like to try again? (Y/N) ')                    if try_again.lower() == 'n':                        break                    elif try_again.lower() == 'y':                        print('If you consider yourself capable...')                        break                    else:                        pass                if try_again.lower() == 'n':                    print('Ok, maybe next time, pal :v')                    break                            else:                print(f'Your guess must be between {low_param} and {high_param}')        except:            print('Are you sure you entered a number?')if __name__ == '__main__':    main()在测试中，我想创建一些方法来验证以下情况：1 - low_param 或 high_param 不是数字 2 - low_param 高于 high_param 3 - 猜测高于 high_param 4 - 猜测低于 low_param 5 - 猜测是字符串 6 - try_again 既不是 Y 也不是 N我设法在第一种方法上模拟了一个输入，但是我不知道如何将 print 语句断言为情况输出。对于其他情况，我需要模拟多个输入，然后我就卡住了。我该如何解决这两个问题？

Answer 1

你的第一个问题是跳出循环。print您可以通过向引发异常的模拟函数添加副作用来做到这一点，并在测试中忽略该异常。mockedprint也可用于检查打印的消息：

@patch('randomgame.print')

@patch('randomgame.input', create=True)

def test_params_input_1(self, mock_input, mock_print):

    mock_input.side_effect = ['foo']

    mock_print.side_effect = [None, Exception("Break the loop")]

    with self.assertRaises(Exception):

        main()

    mock_print.assert_called_with('You need to enter a number!')

请注意，您必须将副作用添加到第二个 print调用，因为第一个调用用于发出欢迎消息。

第二个测试的工作方式完全相同（如果以相同的方式编写），但有一个问题：在您的代码中，您捕获的是通用异常而不是特定异常，因此您的“中断”异常也将被捕获。这通常是不好的做法，因此与其解决此问题，不如捕获转换int失败时引发的特定异常：

while True:

    try:

        low_param = int(input('Please enter the lower number: '))

        high_param = int(input('Please enter the higher number: '))

        if high_param <= low_param:

            print('No, first the lower number, then the higher number!')

        else:

            break

    except ValueError:  # catch a specific exception

        print('You need to enter a number!')

代码中的第二个try/catch块也是如此。