如何使用 @patch 装饰器来修补内置的 input() 函数?
例如,这是我想测试的 Question.py 中的一个函数,其中包含对 input() 的调用:
def query_yes_no(question, default="yes"):
""" Adapted from http://stackoverflow.com/questions/3041986/python-command-line-yes-no-input """
valid = {"yes": True, "y": True, "ye": True, "no": False, "n": False}
if default is None:
prompt = " [y/n] "
elif default == "yes":
prompt = " [Y/n] "
elif default == "no":
prompt = " [y/N] "
else:
raise ValueError("invalid default answer: '%s'" % default)
while True:
sys.stdout.write(question + prompt)
choice = input().lower()
if default is not None and choice == '':
return valid[default]
elif choice in valid:
return valid[choice]
else:
sys.stdout.write("Please respond with 'yes' or 'no' "
"(or 'y' or 'n').\n")
这是我的测试,它给了我错误“ImportError:没有名为'的模块builtins'":
import unittest
from unittest.mock import patch
import question
class TestQueryYesNo(unittest.TestCase):
@patch('__builtins__.input.return_value', 'y')
def test_query_y(self):
answer = question.query_yes_no("Blah?")
self.assertTrue(answer)
__内置__ http://docs.python.org/2/library/__builtin__.html模块重命名为builtins http://docs.python.org/3/library/builtins.html在Python 3中,替换如下:
@patch('builtins.input', lambda *args: 'y')
UPDATE
input
有一个可选参数。更新了代码以接受可选参数。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)