开始对没有返回值的函数进行单元测试

Question

我有一个基于函数的程序，它在函数内部接受用户输入，而不是函数之前的参数：例如，假设我的函数是

def my_function():
    a = input("a: ")
    b = input("b: ")
    print(a+b)

据我目前对单元测试的了解，这样的函数比像这样工作的函数更难单元测试：

def another_function(a,b):
    return(a+b)

那么，例如，我该如何测试一个看起来像 my_function 的函数呢？感觉好像只要输入不正确的输入并检查错误就可以很容易地手动测试，但是我必须编写一个测试套件来自动测试我的所有功能。

Answer 1

鉴于您的函数输入来自input，输出到print，您必须“模拟”这两个函数来测试my_function。例如，使用简单的手动模拟：

def my_function(input=input, print=print):
    a = input("a: ")
    b = input("b: ")
    print(a+b)

if __name__ == '__main__':
    inputs = ['hello', 'world']
    printed = []

    def mock_input(prompt):
        return inputs.pop(0)

    def mock_print(text):
        printed.append(text)

    my_function(mock_input, mock_print)
    assert len(inputs) == 0, 'not all input used'
    assert len(printed) == 1, '{} items printed'.format(len(printed))
    assert printed[0] == 'helloworld'

当您将其与以下内容进行比较时：

assert my_function('hello', 'world') == 'helloworld'

你会明白为什么后者更受欢迎！

您还可以使用适当的模拟库来更整洁地执行此操作，而无需将函数作为参数提供；参见例如How to supply stdin, files and environment variable inputs to Python unit tests?.