Selenium Python - Unittest - при успешном выполнении функционального теста что-то делать

Question

Привет, ребята, мне нужно напечатать «Тест завершен» или сделать что-то после завершения test_case без проблем.

Это пример кода:

class Test_FIRST():


def test_01(self):

        code....

def test_02(self):

        if  Test_FIRST.test_01(self) == True:
            print("Test complete")
        else:
            print("Test not complete")

Этот способ не работает, также выведите «Тест не завершен» в случае успеха или неудачи.

И после этого вычислите время выполнения некоторого тестового примера и распечатайте его