本文共 1751 字，大约阅读时间需要 5 分钟。
在Python编程中，当我们需要测试某些函数的行为时，使用 unittest.mock 库来模拟函数调用是一个非常有用的方法。这种技术可以帮助我们在测试过程中完全控制函数的行为，从而更好地验证我们的代码是否符合预期。
假设我们有一个需要测试的函数 add(a, b)，它的作用是返回两个数的和。为了模拟这个函数的行为，我们可以使用 unittest.mock 库。
导入必要的模块
首先，我们需要导入unittest.mock 库中的 patch 和 Mock 类，这些工具将帮助我们模拟函数的行为。通常，我们会在测试类中使用这些工具。创建测试类
我们将创建一个测试类TestMyFunction，在这个类中，我们将定义一个测试方法，用于验证我们的模拟函数是否正常工作。使用 @patch 装饰器
在测试方法中，我们使用@patch('MyModule.add') 装饰器来替换原本的 add 函数。这里的 MyModule 是包含我们要测试的模块名称。替换后的函数将被 Mock 对象代替，这样我们可以完全控制它的行为。设置 mock 对象的行为
在测试方法中，我们可以设置mock 对象的返回值、调用次数以及其他属性。例如，如果我们想模拟 add 函数返回一个固定的值，我们可以执行以下操作：mocked_add.return_value = 5
调用需要测试的函数
接下来，我们需要调用包含add 函数调用的函数。例如，如果我们有一个函数 do_something，它调用了 add 函数，那么我们可以执行：result = MyModule.do_something()
验证函数调用情况
最后，我们需要验证add 函数是否被正确调用。我们可以检查： 函数是否被调用过：
self.assertTrue(mocked_add.called)
函数是否被调用了一次：
self.assertEqual(mocked_add.call_count, 1)
函数是否接收了正确的参数：
self.assertEqual(mocked_add.call_args, (3, 4))
验证返回值
如果do_something 函数返回了一个值，我们可以检查这个值是否与预期一致：self.assertEqual(result, 10)
import unittestfrom unittest.mock import patch, Mockimport myModule # 假设这个模块包含我们要测试的函数class TestMyFunction(unittest.TestCase): @patch('MyModule.add') def test_my_function(self, mocked_add): # 设置 mock 对象的返回值 mocked_add.return_value = 5 # 调用需要测试的函数 result = MyModule.do_something() # 检查 add 函数是否被调用 self.assertTrue(mocked_add.called) self.assertEqual(mocked_add.call_count, 1) self.assertEqual(mocked_add.call_args, (3, 4)) # 检查 do_something 函数的返回值 self.assertEqual(result, 10) 最后，我们需要运行这个测试。可以在代码的末尾添加以下内容：
if __name__ == '__main__': unittest.main()
通过以上步骤，我们就成功地使用 unittest.mock 模拟了另一个函数的调用，并且可以在测试中验证函数的行为。这是一种非常有用的技术，特别是在单元测试时非常实用。
转载地址：http://coafk.baihongyu.com/