xUnit:如何在失败的情况下运行额外的代码

import unittest

class MyTest(unittest.TestCase):
    def test_something(self):
        try:
            # 运行需要测试的代码
            result = 1 + 1
            self.assertEqual(result, 2)
        except AssertionError:
            # 当测试失败时运行额外的代码
            print("Test failed, running extra code...")
            # 运行额外的代码

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