测试在对象初始化 Python 期间是否调用了某个方法

Question

I have a superclass that calls a certain independent method during its initialization. Something like

class MasterClass:
    def __init__(self, *args, **kwargs):
        if type(self).__name__ == "SpecificClass":
            call_a_module_method()

I want to test that a subclass of this class called SpecificClass will get call_a_module_method method called.

Answer 1

As Kevin Lee adviced you can use isinstance check or, if for some other reasons you do not want to directly check class in your tests, the option is to use mock :

import unittest
from mock import patch

 @patch('module_name.call_a_module_method')
 def test_method_called(self, mock):
     instance = SpecificClass()
     self.assertTrue(mock.called)