问题描述
假设我有这个 python 代码:
Suppose I have this python code:
def answer(): ans = raw_input('enter yes or no') if ans == 'yes': print 'you entered yes' if ans == 'no': print 'you entered no'
如何为此编写单元测试?我知道我必须使用模拟",但我不明白如何.谁能举个简单的例子?
How do I write a unittest for this ? I know i have to use 'Mock' but I don't understand how. Can anyone make some simple example ?
推荐答案
你不能给输入打补丁,但是你可以把它包装起来使用 mock.patch().这是一个解决方案:
You can't patch input but you can wrap it to use mock.patch(). Here is a solution:
from unittest.mock import patch from unittest import TestCase def get_input(text): return input(text) def answer(): ans = get_input('enter yes or no') if ans == 'yes': return 'you entered yes' if ans == 'no': return 'you entered no' class Test(TestCase): # get_input will return 'yes' during this test @patch('yourmodule.get_input', return_value='yes') def test_answer_yes(self, input): self.assertEqual(answer(), 'you entered yes') @patch('yourmodule.get_input', return_value='no') def test_answer_no(self, input): self.assertEqual(answer(), 'you entered no')
请记住,此代码段仅适用于 Python 3.3+ 版本
Keep in mind that this snippet will only work in Python versions 3.3+