我正在尝试测试一个依赖多个用户输入来返回某个值的函数。
我已经在这里寻找了多个答案,但没有一个能够解决我的问题。我看到了参数化、模拟和猴子补丁的东西,但没有任何帮助。我认为很大程度上是因为我没有清楚地理解正在做的事情背后的概念,并且我无法适应我的问题。我看到了为此使用外部文件的建议,但我不想依赖于此。我正在尝试使用 pytest 和 python 3.7.3
我想测试的功能是这样的
def function():
usr_input = input('please enter a number: ')
while True:
if int(usr_input) < 5:
usr_input = input('please, enter a value less then 5: ')
else:
break
return usr_input
我想知道当插入的值无效时如何传递两个输入值来测试函数。示例:发送值 6 和 2,断言期望值 2 并通过测试。我的其他测试如下所示:
def test_input(monkeypatch):
monkeypatch.setattr('builtins.input', lambda x: 6)
test = function()
assert test == 2
但是,对于这种情况,它们会循环。是否可以仅使用参数化或其他简单代码来做到这一点?
EDIT
正如 wim 在接受的答案中指出的那样,我在“if”中添加了一个 int() ,只是为了防止未来的读者感到困惑。我假设演员阵容是可能的。
这里有两个问题,您需要将输入转换为数字,否则比较将失败,将字符串与数字进行比较:usr_input < 5
。请注意,真实的input
永远不会返回数字,只返回字符串。
一旦你清除了这个问题,你就可以进行猴子补丁了input
具有在调用时可以返回不同值的可调用对象:
def fake_input(the_prompt):
prompt_to_return_val = {
'please enter a number: ': '6',
'please, enter a value less then 5: ': '2',
}
val = prompt_to_return_val[the_prompt]
return val
def test_input(monkeypatch):
monkeypatch.setattr('builtins.input', fake_input)
test = function()
assert test == 2
如果你安装了插件pytest-mock https://github.com/pytest-dev/pytest-mock,您可以使用模拟 API 更轻松地完成此操作:
def test_input(mocker):
mocker.patch('builtins.input', side_effect=["6", "2"])
test = function()
assert test == 2
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)