单元测试检查main是否调用了方法

import unittest

def main():
    # 在main函数中调用了其他方法
    method1()
    method2()
    # ...

def method1():
    print("method1")

def method2():
    print("method2")

class TestMain(unittest.TestCase):
    def test_main_calls_method1(self):
        with patch('builtins.print') as mock_print:
            main()
            mock_print.assert_called_with("method1")

    def test_main_calls_method2(self):
        with patch('builtins.print') as mock_print:
            main()
            mock_print.assert_called_with("method2")

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