对Python单元测试中方法内部的方法调用进行计数

from unittest import TestCase
from unittest.mock import patch

class MyClass:
    def method1(self):
        self.method2()
        self.method2()
        self.method3()

    def method2(self):
        pass

    def method3(self):
        pass

class MyTest(TestCase):
    @patch.object(MyClass, 'method2')
    def test_method1(self, mock_method2):
        obj = MyClass()
        obj.method1()

        # 断言方法2被调用了2次
        self.assertEqual(mock_method2.call_count, 2)