python - 如何在 python 测试用例中提供输入运行时

Question

在为 python 方法编写测试用例时，如何自动向变量提供输入而不提示用户输入？

def function():
    speed = input('Enter speed:\n')
    if speed == 10:
        pass
    else:
        pass

对于上面的代码，如果我们调用该方法，它会提示用户输入速度。

>>> function()
Enter speed:
10

有没有办法使用任何函数传递输入或跳过一行并为变量赋值？
提前致谢

Answer 1

可以编写函数来接受测试模式，最简单的方法是使用默认的None参数:

def function(speed=None):
    if speed is None:
        speed = input('Enter speed:\n')
    if speed == 10:
        pass
    else:
        pass

这样，它仍然可以像以前一样被调用(它将运行input部分):

function()

或者在测试期间使用不同的“输入”:

function(5)
function(10)
for i in range(33,50):
    function(i)