Как проверить, какая инструкция вызвала исключение?

Question

У меня есть заявление вроде этого.

Я просто хочу знать, какой из двух утверждений assert сделал исключение.

try: 
    assert re.search("xyz", statement) 
    assert re.search("abc", statement) 
except AssertionError: 
    print "AssertionError : Expected Error message not found" 

Спасибо за ответ.

Answer 1

Как указано в документах The assert statement, вы можете дать выражение после выражения теста утверждения; что второе выражение будет передано в AssertionError. Вот простая демонстрация:

for n in (-5, 10, 20): 
    try: 
     assert 0 <= n, '%d is too low' % n 
     assert n <= 10, '%d is too high' % n 
     print('%d is ok' % n) 
    except AssertionError as err: 
     print "AssertionError:", err 

выход

AssertionError: -5 is too low                             
10 is ok                                  
AssertionError: 20 is too high  

---

Это второе выражение не должно быть строкой, это может быть что угодно. Поскольку утверждения должны использоваться только для проверки программной логики, а не для проверки пользовательских данных, я вообще не беспокоюсь о передаче строки с хорошо отформатированной строкой, я просто передаю кортеж, содержащий соответствующие значения, и, возможно, идентификационную строку. Например,

assert (a * b > c), ('Bad product', a, b, c) 

Answer 2

Вы можете распечатать последнее поднятое исключение с traceback.print_exc(). Пример:

>>> import traceback 
>>> try: 
... a = 1/0 
... except: 
... traceback.print_exc() 
... 
Traceback (most recent call last): 
    File "<stdin>", line 2, in <module> 
ZeroDivisionError: integer division or modulo by zero 

Существует также traceback.format_exc() в случае, если вы не хотите печатать.

Answer 3

Возможно, вы хотите отличать исключения? Тогда Getting exception details in Python может вам помочь. Копирование ответ здесь:

import sys 
try: 
    assert re.search("xyz", statement) 
    assert re.search("abc", statement) 
except AssertionError: 
    type, value, traceback = sys.exc_info() 

Затем вы можете распечатать информацию.

Answer 4

Вы можете использовать функции модуля traceback. Например, extract_tb возвращает список кортежей (названных кортежей в Python 3.5 и более новых), представляющих записи трассировки стека. Каждый кортеж содержит номер строки, а также строку исходного текста (если доступно).

import traceback 

try: 
    assert 1 
    assert None 
except AssertionError as e: 
    for x in traceback.extract_tb(e.__traceback__, limit=-1): 
     print(x.lineno, repr(x.line)) # Prints 5 'assert None'