您如何测试Python函数引发异常？

我们编写一个仅在函数未引发预期异常时失败的单元测试。

我们还测试了Python函数是否引发异常。

例如，请参阅我们粘贴到Python shell中的示例代码以测试Python的类型安全性：

示例

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)

运行测试

输出结果

终端输出以下内容-

..

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

Ran 2 tests in 0.001s

OK

测试人员将assertRaises用作上下文管理器，以确保在记录错误的同时正确捕获并清除该错误。

我们也可以不使用上下文管理器来编写它，请参见测试二。第一个参数是您希望引发的错误类型，第二个参数是您要测试的函数，其余的args和关键字args将传递给该函数。

仅使用上下文管理器就更简单易读。

我们看到了，正如我们期望的那样，尝试将1和'1'相加会导致TypeError。