Подсчет предупреждений в Python 2.4

Question

У меня есть несколько тестов, которые должны подсчитать количество предупреждений, выданных функцией. В Python 2.6 это просто, используя

with warnings.catch_warnings(record=True) as warn:
    ...
    self.assertEquals(len(warn), 2)

К сожалению, with недоступен в Python 2.4, так что еще можно использовать? Я не могу просто проверить, было ли одно предупреждение (используя фильтр предупреждений с action='error' и try / catch), потому что количество предупреждений является значительным.

Answer 1

Я собирался предложить тот же обходной путь, что и Игнасио, чуть более полный пример тестирования кода:

import warnings

def setup_warning_catcher():
    """ Wrap warnings.showwarning with code that records warnings. """


    caught_warnings = []
    original_showwarning = warnings.showwarning

    def custom_showwarning(*args,  **kwargs):
        caught_warnings.append(args[0])
        return original_showwarning(*args, **kwargs)

    warnings.showwarning = custom_showwarning
    return caught_warnings


caught_warnings_list = setup_warning_catcher()

# trigger warning here

assert len(caught_warnings_list) == 1

Answer 2

То, что вы можете сделать, это продублировать поведение warnings.catch_warnings() самостоятельно. Сохраните текущее значение warnings.showwarning и замените его функцией, которая сохраняет предупреждение в списке, затем после обычной проверки длины списка и восстановления warnings.showwarning.

oldsw = warnings.showwarning
warnings.showwarning = myshowwarning
  ...
self.assertEquals(len(somewarninglist), 2)
warnings.showwarning = oldsw