(Python单元测试)针对同一python文件中的不同方法，对同一对象进行不同的补丁

import unittest
from unittest.mock import patch

# 假设有一个名为MyClass的类，其中包含两个方法method1和method2
class MyClass:
    def method1(self):
        return "Original method1"

    def method2(self):
        return "Original method2"

# 编写一个测试类，继承自unittest.TestCase
class MyTestClass(unittest.TestCase):
    def test_method1_with_patch(self):
        # 创建一个MyClass的实例
        my_obj = MyClass()

        # 使用patch修饰器，将method1替换为新的实现
        @patch.object(my_obj, 'method1')
        def test_patch_method1(mock_method1):
            # 设置mock_method1的返回值
            mock_method1.return_value = "Patched method1"

            # 调用被补丁过的method1
            result = my_obj.method1()

            # 断言返回值是否符合预期
            self.assertEqual(result, "Patched method1")

        # 执行测试函数
        test_patch_method1()

    def test_method2_with_patch(self):
        # 创建一个MyClass的实例
        my_obj = MyClass()

        # 使用patch修饰器，将method2替换为新的实现
        @patch.object(my_obj, 'method2')
        def test_patch_method2(mock_method2):
            # 设置mock_method2的返回值
            mock_method2.return_value = "Patched method2"

            # 调用被补丁过的method2
            result = my_obj.method2()

            # 断言返回值是否符合预期
            self.assertEqual(result, "Patched method2")

        # 执行测试函数
        test_patch_method2()

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