Настроить AssertionError в Python

Question

Я пытаюсь добавить текст для всех ошибок утверждения в моем коде.

Это мой код:

class AssertionError(Exception): 
    def __init__(self, msg): 
     Exception.__init__(self, msg) 
     self.message = msg + "+ SOME TEXT" 

assert 1 == 2, "FAIL" 

Результат

__main__.AssertionError: FAIL 

Я ожидал увидеть результат: "FAIL + НЕКОТОРЫХ TEXT"

---

Проблема с UnitTest также. Я хочу добавить текст для всех неудавшихся тестов (без обновления всего текстового сообщения).

import unittest 

class TestCase(unittest.TestCase): 
    def test1(self): 
     self.assertTrue(False, "FAIL!") 

    def test2(self): 
     self.assertLessEqual(10, 2, "FAIL!") 

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

Answer 1

Проблема в том, что вы ничего не делаете с self.message = msg + "+ SOME TEXT". Вам необходимо передать пользовательское сообщение, которое вы хотите: Exception.__init__.

Это будет работать для вас:

class AssertionError(Exception): 
    def __init__(self, msg): 
     self.message = msg + " SOME TEXT" 
     super().__init__(self, self.message) 
assert 1 == 2, "FAIL" 

Если вы хотите, чтобы просмотреть сообщение в будущем, вы можете использовать Try/за исключением и поймать пользовательское сообщение, как это:

try: 
    assert 1 == 2, "FAIL" 
except AssertionError as e: 
    print(e.message) 

Answer 2

Это похоже на Morgan's answer, но использует немного другой способ достичь того же результата:

>>> class AssertionError(AssertionError): 
    def __init__(self, msg): 
     super().__init__(msg + ' SOME TEXT') 

>>> assert 1 == 2, 'FAIL' 
Traceback (most recent call last): 
    File "<pyshell#4>", line 1, in <module> 
    assert 1 == 2, 'FAIL' 
AssertionError: FAIL SOME TEXT 

Answer 3

Этот вариант сохраняет исключение именно так, как он был поднят и изменяет его строковое представление только:

class AssertionError(AssertionError): 
    def __str__(self): 
     return super().__str__() + "SOME TEXT" 

(кредит: подклассов взяты из ответа NOCTIS')