Python模拟修补另一个函数中的函数

Question

def f1():
    return 10, True

def f2():
    num, stat = f1()
    return 2*num, stat

如何使用python的模拟库修补f1()，并返回自定义结果以便测试f2()？

编辑： 我的考试有什么问题吗？这似乎不起作用，所有的测试都失败了

from foo.bar import f2
from mock import patch

class MyTest(TestCase):

    def test_f2_1(self):
        with patch('project.module.f1') as some_func:
            some_func.return_value = (20, False)
            num, stat = f2()
            self.assertEqual((num, stat), (40, False))

   @patch('project.module.f1')
   def test_f2_2(self, some_func):
       some_func.return_value = (20, False)
       num, stat = f2()
       self.assertEqual((num, stat), (40, False))

Answer 1

假设您正在使用这个mock库：

def f1():
    return 10, True

def f2():
    num, stat = f1()
    return 2*num, stat

import mock

print f2()   # Unchanged f1 -> prints (20, True)

with mock.patch('__main__.f1') as MockClass:       # replace f1 with MockClass 
    MockClass.return_value = (30, True)     # Change the return value

    print f2()     # f2 with changed f1 -> prints (60, True)

如果您的代码被分成多个模块，您可能需要用模块/函数的路径替换__main__.f1。