How do you test that a Python function throws an exception?


We write a unittest that fails only if a function doesn't throw an expected exception.

We also test if a Python function throws an exception.

For example, see the sample code we paste into the Python shell to test Python's type-safety:

import unittest
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')
 unittest.main(exit=False)

Running the tests

And the terminal outputs the following:

..

----------------------------------------------------------------------

Ran 2 tests in 0.001s

OK

Test one uses assertRaises as a context manager, which ensures that the error is properly caught and cleaned up, while recorded.

We could also write it without the context manager, see test two. The first argument would be the error type you expect to raise, the second argument, the function you are testing, and the remaining args and keyword args will be passed to that function.

It's far more simple, and readable to just use the context manager.

We see that as we expect, attempting to add a 1 and a '1' result in a TypeError.

raja
Published on 07-Dec-2017 17:36:57
Advertisements