Python unittest 只在断言失败是怎么触发另一个方法函数？

在做自动化脚本开发，使用的 unittest 框架， 想在断言失败的时候截图。下面这个例子无论成功或失败都会触发指定的函数，请问怎么修改，只在失败的时候触发函数呢？

import unittest
class loginTests(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        pass

    @classmethod
    def tearDownClass(cls):
        pass

    def aaa(self):
        print('触发函数')

    def test1(self):
        i =2
        self.assertEqual(i,1,self.aaa())

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