python在单元测试中模拟原始输入

时间:2022-10-14
本文介绍了python在单元测试中模拟原始输入的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着跟版网的小编来一起学习吧!

问题描述

假设我有这个 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+

这篇关于python在单元测试中模拟原始输入的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持跟版网!

上一条:如何模拟在 with 语句中使用的 open (使用 Python 中的 Mock 框架)? 下一条:一个模拟/存根 python 模块如何像 urllib

相关文章

最新文章