今天学习的是对测试中的异常进行捕获,主要用到了两个方法,assertRaises()和assertRaisesRegex()方法
测试中的异常捕获
- 对需要测试的接口异常进行捕获
from unittest import TestCase
from unittest import main
def division(a, b):
return a / b
class MyTest(TestCase):
def setUp(self):
pass
def test_raise(self):
# 需要保证division传递了1,0两个参数,一定会抛出异常,且被Exception捕获
self.assertRaises(Exception, division, 1, 0)
def tearDown(self):
pass
# 主函数
if __name__ == '__main__':
main()
- 代码中定义了一个division除法函数,然后在MyTest类中定义了测试用例test_raise(),在这个测试用例中用到了
assertRaises(exception, callable, *args,**kwds)
断言方法,参数exception是待验证的异常类型,参数callable是待验证的方法,*args和kwds是待验证的方法参数。assertRaises()方法主要用于验证异常测试,在调用待测试函数callable时,在传入相应的测试数据后,如果测试通过则表明待测试函数抛出了预期的异常,如果报错则表明测试函数没有引发相应异常。 - 实例中我们给定的待验证方法的参数是1和0,1除以0不成立,满足待验证参数exception,所以执行代码不会引发异常。如果我们把0换成其他数字,则会报异常:
AssertionError:Exception not raised by division
通过正则表达式捕获异常
from unittest import TestCase
from unittest import main
def division(a, b):
return a / b
class MyTest(TestCase):
def setUp(self):
pass
def test_raise(self):
# 通过正则表达式'by zero' 匹配除零异常
self.assertRaisesRegex(Exception, "by zero", division, 1, 0)
def tearDown(self):
pass
# 主函数
if __name__ == '__main__':
main()
- 我们看这个实例,和上面的实例类似,不同在于断言方法
assertRaises()
变成了assertRaisesRegexp(exception, regexp, callable, *args, **kwds)
方法,assertRaisesRegexp()
方法比assertRaises()
方法多了一个regexp参数,是一个正则表达式对象。该方法使用正则表达式方式来匹配异常断言,如果捕获到了匹配正则表达式的异常则运行成功,否则报错,通过正则表达式捕获异常能更加灵活和适用更多的场景;