如何在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)