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


猜你喜欢
- andom.sample(list, n)即是从list中随机选取n个不同的元素# -*- coding: utf-8 -*- import
- 如果你忘记了你的MYSQL的root口令的话,你可以通过下面的过程恢复。1. 向mysqld server 发
- turtle库是python的基础绘图库,官方手册这个库被介绍为一个最常用的用来给孩子们介绍编程知识的方法库,其主要是用于程序设计入门,是标
- 无论是我们上学时还之后的工作中,基本都需要用到电子证件照片,这类照片基本都对照片尺寸、背景色有要求,本文我们来看一下如何只用不到 20 行
- 1. DID(Differences-in-Differences)定义双重差分法,其主要被用于社会学中的政策效果评估。这种方法需要两个「差
- 浏览器的开发者在很早的时候就已经意识到, HTTP's 的无状态会对Web开发者带来很大的问题,于是(cookies)应运而生。 c
- 我是新手学习如何管理和导航MySQL数据库和表是要掌握的首要任务之一,下面的内容将主要对MySQL的数据库和表的一些常用命令进行总结,一些我
- 函数,对于人类来讲,能够发展到这个数学思维层次,是一个飞跃。可以说,它的提出,直接加快了现代科技和社会的发展,不论是现代的任何科技门类,乃至
- 代码很简单,这里就不多废话了,直接奉上:$(function(){ var w=
- create proc sp_PublicTurnPageWebSite( @TBName nvarchar(
- PyAutoGUI 不知道你有没有用过,它是一款用Python自动化控制键盘、鼠标的库。但凡是你不想手动重复操作的工作都可以用这个库来解决。
- 1、在Django配置搜索结果页的路由映射"""pachong URL Configuration
- PyCharm自身提供了大量实用的快捷键,但是由于自己之前其他软件的快捷键使用习惯与此不同,这就需要在PyCharm量身DIY属于自己的快捷
- 1.场景将URL动态生成二维码前端展示(微信支付等,)--》1.静态文件路径访问返回URL_name,(a标签,src 静态路由访问)2.流
- 用途logging模块是Python的内置模块,主要用于输出运行日志,可以灵活配置输出日志的各项信息。基本使用方法logging.basic
- 第一步:要使用vant组件安装好vant,npm i vant -S第二步:在你要用到的地方js中引入或者在src/main.js里面引入i
- 从前有三只小猪,长大自立了分别造房子住。老大搬来草堆堆出草屋,老二搬来木头搭出木屋,老三搬来砖头,砌墙,造烟囱,造出了坚固的砖房。一天晚上大
- 说明同学的代码中遇到一个数学公式牵扯到将生成指定的数字存储的一个列表中,那个熊孩子忽然懵逼的不会啦,,,给了博主一个表现的机会,,,哈哈哈好
- 运行截图运行效果:什么?你说你看不清烟花?那我换一种颜色,请点开看。实现过程准备工作使用语言和框架:python、pygame。安装pyga
- 问题:cudatoolkit cudnn 通过conda 虚拟环境安装,先前已经使用virtualenv安装tf,需要在conda虚拟环境中