python unittest模块，将故障记录到文件

import unittest

class FaultRecordingTestResult(unittest.TextTestResult):
    def addFailure(self, test, err):
        # 将故障记录到文件
        with open('fault.log', 'a') as f:
            f.write(f"Failure in {test}: {err}\n")
        super().addFailure(test, err)

    def addError(self, test, err):
        # 将错误记录到文件
        with open('fault.log', 'a') as f:
            f.write(f"Error in {test}: {err}\n")
        super().addError(test, err)

class MyTestCase(unittest.TestCase):
    def test_something(self):
        self.assertEqual(2 + 2, 4)

if __name__ == '__main__':
    # 使用自定义测试结果报告类
    unittest.main(testRunner=unittest.TextTestRunner(resultclass=FaultRecordingTestResult))