Python公开课 - 断言处理

前言

assert断言语句用来声明某个条件是真的,其作用是测试一个条件(condition)是否成立,如果不成立,则抛出异常。

一般来说在做单元测试的时候用的比较多,在生产环境代码运行的情况下,不建议使用断言,会让程序abort掉。

什么时候使用断言

  • 保护性的编程 正常情况下,并不是防范当前代码发生错误,而防范由于以后的代码变更发生错误。
  • 运行时序逻辑的检查 这种情况一般都是很严重的,防止脏数据或者异常数据进入业务系统,主动进行中断处理
  • 单元测试代码 开发或测试人员编写单元测试代码的时候,也会使用断言

Python中断言使用

编辑test.py文件

def compare(a, b):
    assert a>b, 'error: b is bigger'
    print('compare: %d %d' %(a, b))

compare(1, 3)

输出:
Traceback (most recent call last):
  File "test8.py", line 5, in <module>
    compare(1, 3)
  File "test8.py", line 2, in compare
    assert a>b, 'error: b is bigger'
AssertionError: error: b is bigger

当然,我们也可以通过-O参数来忽略断言,运行python3 -O test.py

输出:
compare: 1 3

单元测试中常用断言

Python中的unittest框架用来进行单元测试非常方便

import unittest

class WidgetTestCase(unittest.TestCase):

    def test_upper(self):
        self.assertEqual('foo'.upper(), 'FOO')

unittest.main()

unittest中一些常用断言方法如下:

assertEqual(a, b)   a == b   
assertNotEqual(a, b)    a != b   
assertTrue(x)   bool(x) is True  
assertFalse(x)  bool(x) is False     
assertIs(a, b)  a is b  3.1
assertIsNot(a, b)   a is not b  3.1
assertIsNone(x) x is None   3.1
assertIsNotNone(x)  x is not None   3.1
assertIn(a, b)  a in b  3.1
assertNotIn(a, b)   a not in b  3.1
assertIsInstance(a, b)  isinstance(a, b)    3.2
assertNotIsInstance(a, b)   not isinstance(a, b)    3.2

小结

断言在软件系统中有非常重要的作用,写的好可以让你的系统更稳定,也可以让你有更多面对对象的时间,而不是在调试代码。

相关阅读