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

[zht0858]

在做自动化脚本开发，使用的 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()

[vTexEZDota]

try:
self.assertEqual(i,1)
except AssertionError as e:
self.aaa
self.assertEqueal(i,1)

你可以试试看。但是如果只做异常处理，这条用例会被认为正确执行了。

[jotsai]

我是在 teardown 里判断用例是否执行失败，失败就截图