Python中优雅使用assert断言的方法实例
作者:王坤祥 发布时间:2021-03-27 12:56:58
目录
什么是 assert 断言
断言和异常的使用场景
使用断言的几个原则
建议不使用断言的情况:
总结
什么是 assert 断言
Assert statements are a convenient way to insert debugging assertions into a program
断言声明是用于程序调试的一个便捷方式。断言可以看做是一个 debug 工具,Python 的实现也符合这个设计哲学,在 Python 中 assert 语句的执行是依赖于__debug__这个内置变量的,其默认值为True。当__debug__为True时,assert 语句才会被执行。
对于一般的声明,assert expression 等价于
if __debug__:
if not expression: raise AssertionError
assert 可以同时声明两个 expression,例如 assert expression1, expression2 等价于
if __debug__:
if not expression1: raise AssertionError(expression2)
如果执行脚本文件时加上-O参数, __debug__则为False
举一个例子,假设我们有一个脚本 testAssert.py,内容为:
print(__debug__)
assert 1 > 2
当使用python assert.py运行时,__debug__会输出 True,assert 1 > 2 语句会抛出 AssertionError 异常。
当使用python -O assert.py运行时,__debug__会输出 False,assert 1 > 2 语句由于没有执行不会报任何异常。
断言和异常的使用场景
先说结论:
检查先验条件使用断言,检查后验条件使用异常
举个例子来说明一下,在开发中我们经常会遇到读取本地文件的场景。我们定义一个 read_file 方法。
def read_file(path):
assert isinstance(file_path, str)
...
read_file 函数要求在开始执行的时候满足一定条件:file_path 必须是 str 类型,这个条件就是先验条件,如果不满足,就不能调用这个函数,如果真的出现了不满足条件的情况,证明代码中出现了 bug,这时候我们就可以使用 assert 语句来对 file_path 的类型进行推断,提醒程序员修改代码,也可以使用 if...raise...语句来实现 assert,但是要繁琐很多。在很多优秀的 Python 项目中都会看到使用 assert 进行先验判断的情况,平时可以多多留意。
read_file 函数在被调用执行后,依然需要满足一定条件,比如 file_path 所指定的文件需要是存在的,并且当前用户有权限读取该文件,这些条件称为后验条件,对于后验条件的检查,我们需要使用异常来处理。
def read_file(file_path):
assert isinstance(file_path, str)
if not check_exist(file_path):
raise FileNotFoundError()
if not has_privilege(file_path):
raise PermissionError()
文件不存在和没有权限,这两种情况并不属于代码 bug,是代码逻辑的一部分,上层代码捕获异常后可能会执行其他逻辑,因此我们不能接受这部分代码在生产环境中被忽略,这属于后验条件。并且,相比于 assert 语句只能抛出 AssertionError,使用异常可以抛出更详细的错误,方便上层代码针对不同错误执行不同的逻辑。
使用断言的几个原则
使用断言捕捉不应该发生的非法情况。不要混淆非法情况与错误情况之间的区别,后者是必然存在的并且是一定要作出处理的。
使用断言对函数的参数进行确认。
在编写函数时,要进行反复的考查,并且自问:“我打算做哪些假定?”一旦确定了的假定,就要使用断言对假定进行检查。
一般教科书都鼓励程序员们进行防错性的程序设计,但要记住这种编程风格会隐瞒错误。当进行防错性编程时,如果“不可能发生”的事情的确发生了,则要使用断言进行报警。
断言也可以用于代码测试,用作一个做事毛手毛脚的开发人员的单元测试,只要能你接受当使用-O标志时这个测试什么都不做。我有时也会在代码中用"assert Fasle"来对还没有实现的分支作标记,当然我希望他们失败。如果稍微更细节一些,或许触发NotImplementedError是更好的选择
另一个断言用得好的地方就是检查程序中的不变量。一个不变量是一些你能相信为真的条件,除非一个缺陷导致它变成假。如果有一个缺陷,越早发现越好,因此我们需要对其进行测试,但我们不想因为这些测试而影响代码执行速度。因此采用断言,它能在开发时生效而在产品中失效。
断言也是一个很好的检查点注释。为了替代如 * 释:
#当我们执行到这里,我们知道n>2
#你可以确保在运行时用以下断言:
assert n > 2
建议不使用断言的情况:
不要用于测试用户提供的数据,或者那些需要在所有情况下需要改变检查的地方
不要用于检查你认为在通常使用中可能失败的地方。断言用于非常特别的失败条件。你的用户绝不看到一个AssertionError,如果看到了,那就是个必须修复的缺陷。
特别地不要因为断言只是比一个明确的测试加一个触发异常矮小而使用它。断言不是懒惰的代码编写者的捷径。
不要将断言用于公共函数库输入参数的检查,因为你不能控制调用者,并且不能保证它不破坏函数的合约。
不要将断言用于你期望修改的任何错误。换句话,你没有任何理由在产品代码捕获一个AssertionError异常。
不要太多使用断言,它们使代码变得晦涩难懂。
来源:https://xie.infoq.cn/article/62388d990f852ad93863f35e5
猜你喜欢
- 代码如下:Create PROCEDURE Batch_Delete @TableName nvarchar(100), --表
- 如下所示:aa = [1,2,3,4,5]aa.index(max(aa))如果aa是numpy数组:aa = numpy.array([1
- 题目描述724. 寻找数组的中心下标 - 力扣(LeetCode) (leetcode-cn.com)给你一个整数数组 nums
- 在日常学习工作过程中,我们难免需要复用以前的项目,这里讲下复用 Django 项目并重命名的过程。1.修改项目名称,使用 pycharm -
- 前段时间在网上找了一个“完美的”JavaScript对象克隆的函数,感觉还不错,但随后便出现了一些问题,发现这个克隆并不好用,在使用发现了如
- Image and text elements that appear in another element are called floa
- 本篇文章适合css新手学习,对于已经掌握了css的朋友们也可以通过本片文章来复习知识。作者通过实践,认为在有些情况下css的代码是可以更加简
- 进程的概念程序是没有运行的代码,静态的;进程是运行起来的程序,进程是一个程序运行起来之后和资源的总称;程序只有一个,但同一份程序可以有多个进
- MFCC梅尔倒谱系数(Mel-scaleFrequency Cepstral Coefficients,简称MFCC)。MFCC通常有以下之
- Python是动态语言,在创建对象后,可以动态地绑定属性和方法定义类:class Student: #定义类 &nb
- 问题描述:被搜索名字为:andy这时搜索andy正常,但是搜索a就搜不到。解决办法,在索引配置文件中的index中添加min_infix_l
- 一.使用DOM生成和读取XML文件 实例一: <?php //Creates XML string and XML document
- 在当今用户的显示器越来越大的今天,之前的1024*768固宽布局有点越来越不合时宜,对大屏幕的用户而言,两侧空空的留白给人第一眼的印象是严重
- 问:怎样才能取得局域网中所有SQL Server的实例?答:请参考以下的具体步骤:SmoApplication.EnumAvailableS
- 如何在聊天室实现趣味答题并计分功能?这个创意确实很好,我们可用在聊天室框架中加入一隐含帧(5分钟刷新一次)的做法来实现这一功能。questi
- 偶然在Google发现了他们的用户体验设计原则,因此翻译作一下记录。1.以人为本 —他们的生活、他们的工作和他们的梦想2.珍惜每一毫秒的时间
- 一年一度的双十一就快到了,各种砍价、盖楼、挖现金的口令将在未来一个月内充斥朋友圈、微信群中。玩过多次双十一活动的小编表示一顿操作猛如虎,一看
- php mysql获取表字段名称和字段信息的三种方法先给出本实例中使用的表的信息:使用desc获取表字段信息php代码如下:<?php
- 1 包简介1.1 工作空间go语言的工作空间必须由 bin、pkg、src三个目录组成,可以在GOPATH环境变量中添加多个工作空间,但不能
- 你的主页或者你管理的网站有各种密码需要保护,把密码直接放在数据库或者文件中存在不少安全隐患,所以密码加密后存储是最常见的做法。在ASP.NE