如何在模拟基类的同时为子类编写Unittest？

import unittest
from unittest.mock import Mock

class MyBaseClass:
    def base_method(self):
        # 基类的方法逻辑
        return "base"

class MySubClass(MyBaseClass):
    def sub_method(self):
        # 子类的方法逻辑
        return "sub"

class MyTestCase(unittest.TestCase):
    def test_sub_method(self):
        # 创建子类的模拟对象
        sub_mock = Mock(spec=MySubClass)
        # 配置模拟对象的行为
        sub_mock.sub_method.return_value = "mocked sub"
        # 创建基类的模拟对象
        base_mock = Mock(spec=MyBaseClass)
        # 配置模拟对象的行为
        base_mock.base_method.return_value = "mocked base"
        # 将子类的模拟对象设置为基类的模拟对象的属性
        base_mock.sub = sub_mock

        # 调用子类的方法
        result = base_mock.sub.sub_method()

        # 验证测试结果
        self.assertEqual(result, "mocked sub")

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