如何测试在具有nosetests的函数中调用函数

发布时间：2020-12-16 23:04:03 所属栏目：Python 来源：网络整理

导读：我正在尝试为项目设置一些自动单元测试.我有一些功能,作为副作用偶尔会调用另一个功能.我想写一个单元测试,测试第二个函数被调用,但我很难过.下面是伪代码示例： def a(self): data = self.get() if len(data) 3500: self.b() # Bunch of other magic,which

我正在尝试为项目设置一些自动单元测试.我有一些功能,作为副作用偶尔会调用另一个功能.我想写一个单元测试,测试第二个函数被调用,但我很难过.下面是伪代码示例：

def a(self):
    data = self.get()
    if len(data) > 3500:
        self.b()

    # Bunch of other magic,which is easy to test.

def b(self):
    serial.write("x00x01x02")

我如何测试b() – 被调用？

解决方法

您可以使用 mock模块模拟函数b并检查它是否被调用.这是一个例子：

import unittest
from mock import patch


def a(n):
    if n > 10:
        b()

def b():
    print "test"


class MyTestCase(unittest.TestCase):
    @patch('__main__.b')
    def test_b_called(self,mock):
        a(11)
        self.assertTrue(mock.called)

    @patch('__main__.b')
    def test_b_not_called(self,mock):
        a(10)
        self.assertFalse(mock.called)

if __name__ == "__main__":
    unittest.main()

另见：

> Assert that a method was called in a Python unit test
> Mocking a class: Mock() or patch()?

希望有所帮助.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!