使用空用户输入对 Python 函数的返回值进行单元测试

Question

我在fruit.py 脚本中有一个函数，它根据用户输入返回一个枚举。

class Selection((IntEnum):
    APPLE = 0
    ORANGE = 1
    PEAR = 2
    MELON = 3
    GRAPE = 4

def get_input():
    selection = int(input("Input an integer from 0 to 4: "))
    fruit = Selection(selection)
    return fruit

现在我想在fruit_test.py 中测试如果用户输入为空，它是否会返回任何内容/引发错误：

import unittest
from unittest.mock import patch
import fruit

class TestCase(unittest.TestCase):
    @patch('builtins.input', return_value=int(''))
    def test_empty_input(self, input):
        result = fruit.get_input()
        self.assertEqual(result, "")

但我的测试以ValueError: invalid literal for int() with base 10: '' 失败。我知道这是因为我的补丁输入int('') 错误，但我不知道如何编写正确的测试格式。有谁知道如何解决这个问题？谢谢！

Answer 1

您可以将 get_input 函数转换为以下方式：

def get_input():
    selection = input("Input an integer from 0 to 4: ")
    if not selection.isnumeric() or not int(selection) Selection.__members__.values():
       return None # or False or "" or whatever you want
    fruit = Selection(int(selection))
    return fruit