今天学习的是对测试中的异常进行捕获,主要用到了两个方法,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参数,是一个正则表达式对象。该方法使用正则表达式方式来匹配异常断言,如果捕获到了匹配正则表达式的异常则运行成功,否则报错,通过正则表达式捕获异常能更加灵活和适用更多的场景;

参考:https://www.9xkd.com/user/plan-view.html?id=1932068715


Summer
83 声望9 粉丝

关于Python学习的一些小知识


« 上一篇
封装mock