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

Как написать тест, который выходит из строя, только если функция не генерирует ожидаемое исключение?

Short Answer:

Использовать метод self.assertRaises в качестве менеджера контекста:

    def test_1_cannot_add_int_and_str(self):
        with self.assertRaises(TypeError):
            1 + '1'

Демонстрация

довольно легко продемонстрировать в оболочке Python.

Библиотека unittest

В Python 2.7 или 3:

import unittest

В Python 2.6 вы можете установить backport из 2.7's unittest, называемой unittest2 , и просто псевдоним, который как unittest:

import unittest2 as unittest

Примеры тестов

Теперь вставьте в свою оболочку Python следующую тест безопасности типа Python:

class MyTestCase(unittest.TestCase):
    def test_1_cannot_add_int_and_str(self):
        with self.assertRaises(TypeError):
            1 + '1'
    def test_2_cannot_add_int_and_str(self):
        import operator
        self.assertRaises(TypeError, operator.add, 1, '1')

В тестовом режиме в качестве менеджера контекста используется assertRaises, который гарантирует, что ошибка будет правильно захвачена и очищена, при записи.

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

Я думаю, что это гораздо проще, легко читаемо и поддерживается только для использования диспетчера контекстов.

Запуск тестов

Для запуска тестов:

unittest.main(exit=False)

В Python 2.6 вам, вероятно, понадобится следующее :

unittest.TextTestRunner().run(unittest.TestLoader().loadTestsFromTestCase(MyTestCase))

И ваш терминал должен вывести следующее:

..
----------------------------------------------------------------------
Ran 2 tests in 0.007s

OK

И мы видим, что, как мы ожидаем, попытка добавить 1 и '1' приведет к TypeError.

---

Для получения более подробного вывода попробуйте следующее:

unittest.TextTestRunner(verbosity=2).run(unittest.TestLoader().loadTestsFromTestCase(MyTestCase))