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

Question

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

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

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

>>> function()
Enter speed:
10

是否有任何方法可以使用任何函数传递输入或跳过一行并为变量赋值
提前谢谢

Answer 1

可以编写function以接受测试模式，最简单的方法是使用默认的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)