如何在python单元测试中永久模拟函数返回值

from unittest import TestCase
from unittest.mock import patch

# 假设我们要测试的函数是`add_numbers`，它调用了一个外部的函数`get_number`来获取数字并相加
def get_number():
    return 10

def add_numbers():
    num = get_number()
    return num + 5

class TestAddNumbers(TestCase):
    @patch('__main__.get_number')  # 使用`patch`装饰器来替换`get_number`函数
    def test_add_numbers(self, mock_get_number):
        mock_get_number.return_value = 20  # 模拟`get_number`函数的返回值为20
        result = add_numbers()
        self.assertEqual(result, 25)  # 断言函数的返回值是否符合预期

        mock_get_number.return_value = 30  # 可以在不同的测试用例中多次模拟返回值
        result = add_numbers()
        self.assertEqual(result, 35)

        # 如果不再需要模拟，可以使用`stop`方法停止模拟
        mock_get_number.stop()

        # 继续测试其他逻辑...

# 运行测试用例
if __name__ == '__main__':
    unittest.main()