python实用程序育儿法_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

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

小结

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

相关阅读