python unittest之异常测试

exception.png

前言

在python unittest测试框架中，提供了一系列的方法用于验证程序的异常。

下面和我一起看看在unittest中，如何使用异常验证来进行断言，完成对应的测试工作

assertRaises(exception, callable, *args, **kwds)

参数说明:

assertRaises(
exception,  # 待验证异常类型
callable,  # 待验证方法
*args,  # 待验证方法参数
**kwds # 待验证方法参数(dict类型)
)

功能说明：

验证异常测试，验证异常（第一个参数）是当调用待测试函数时，在传入相应的测试数据后，如果测试通过，则表明待测试函数抛出了预期的异常，否则测试失败。

下面我们通过一个示例来进行演示，如果验证做除法时抛出除数不能为0的异常ZeroDivisionError。

# _*_ coding:utf-8 _*_

__author__ = '苦叶子'

import unittest
import sys
reload(sys)
sys.setdefaultencoding("utf-8")

# 除法函数
def div(a, b):
    return a/b
    
# 测试用例
class demoRaiseTest(unittest.TestCase):
    def test_raise(self):
        self.assertRaises(ZeroDivisionError, div, 1, 0)
        
# 主函数
if __name__ == '__main__':
    unittest.main()

test_raise方法使用了assertRaises方法来断言验证div方法除数为零时抛出的异常。

运行python raise_demo.py结果如下

.
-------------------------------------
Ran 1 test in 0.000s

OK

你还可以尝试调整下数据，如下：

def test_raise(self):
    
    self.assertRaises(ZeroDivisionError, div, 1,1)

执行结果如下:

F
=====================================
FAIL: test_raise (__main__.demoRaiseTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "raise_demo.py", line 18, in test_raise
    self.assertRaises(ZeroDivisionError, div, 1,1)
AssertionError: ZeroDivisionError not raised

----------------------------------
Ran 1 test in 0.000s

assertRaisesRegexp(exception, regexp, callable, *args, **kwds)

这里就不对参数进行说明了，该方法使用正则表达式方式来匹配异常断言，能更加灵活，实用更多的场景。

通常regexp参数是一个正常表达式，或包含正则表达式的字符串（使用re.search()函数）

下面跟我一下看下assertRaisesRegexp的实际应用示例

# _*_ coding:utf-8 _*_

__author__ = '苦叶子'

import unittest
import re
import sys
reload(sys)
sys.setdefaultencoding("utf-8")

# 除法函数
def div(a, b):
    return a/b

# 测试用例
class demoRaiseTest(unittest.TestCase):
    def test_raise_regexp(self):
        self.assertRaisesRegexp(
        ZeroDivisionError, "integer division or modulo by zero", div, 2, 0)
        
# 主函数
if __name__ == '__main__':
    unittest.main()

运行python test_raise_regexp.py上述示例，结果如下:

.
-----------------------------------
Ran 1 test in 0.001s

OK

我们修改下除数，把

self.assertRaisesRegexp(
        ZeroDivisionError, "integer division or modulo by zero", div, 2, 0)

改为：

self.assertRaisesRegexp(
        ZeroDivisionError, "integer division or modulo by zero", div, 2, 2)

则执行失败，因为匹配失败了。

F
===================================
FAIL: test_raise_regexp (__main__.demoRaiseTest)
------------------------------------
Traceback (most recent call last):
  File "test_raise_regexp.py", line 19, in test_raise_regexp
    ZeroDivisionError, "integer division or modulo by zero",
AssertionError: ZeroDivisionError not raised

------------------------------------
Ran 1 test in 0.001s

FAILED (failures=1)

总结

今天我们就如何对被测对象的抛出的异常进行断言验证，做了基本的说明和示例，大家可以基于上述示例进行修改理解。深入掌握异常断言原理和方法。

python unittest之异常测试