在Python中正确的使用断言-assert

阅读本文大概需要 3 分钟。

最近这两天流感频发,小伙伴们可要注意预防感冒,多锻炼身体。当一个人生病时就会觉得健康是一种莫大的幸福,主要是我中招了,感冒好了之后,我决定继续坚持锻炼身体。今天继续分享Python 的小碎碎 assert。

assert 的中文含义就是断言,assert something 意思就是我断定 something 是对的,如果不对,此处抛出异常,程序终止运行。这里的对或者错,在 Python 中就是 True 或 False。语句

assert 1 == 2

就相当于

if __debug__:
    if not 1 == 2: raise AssertionError

这里的 __debug__ 是一个常数,如果 Python 程序在执行时加了参数 -o ,比如 python test.py -o ,那么 __debug__ 的值就为 False,于是所有的 assert 语句都将失效。反之就会生效。

不过需要注意的是,直接对 __debug__ 赋值是非法的,因为它在解释器开始运行时就已经确定了,中途无法再进行修改。另外要注意的是,assert 后的表达式中不能使用括号,如下面的代码,会报语法错误的警告,但不会抛出异常,后面那行代码是正确的写法。

>>> assert(1==2,'this should fail')
<stdin>:1: SyntaxWarning: assertion is always true, perhaps remove parentheses?
>>> assert 1==2,'this should fail'
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AssertionError: this should fail
>>>

到这里你大概已经知道了,assert 可以用来做参数的检查,防止出现 bug,当出现问题时可以根据消息快速定位至问题处,提高 debug 效率。比如下面就是一种非常常见的用法:

def func(input):
    assert isinstance(input, list), 'input must be type of list'
    # 下面的操作都是基于前提:input必须是list
    if len(input) == 1:
        ...
    elif len(input) == 2:
        ...
    else:
        ... 

当然你可能会问了,我使用 if 语句来判断,不也是可以的么,当然可以,只是他们两者的使用场景不太一样。在重要的环节,比如判断用户是否是管理员,那么必须使用 if,因为 assert 是可以通过 -o 参数来关闭的。在一些非重要的环节,比如写代码的初期,程序还不够健壮,使用 assert 来断言,写代码的行数更少,也更优雅。其实,有没有 assert ,都不应该影响程序的正常运行,assert 的作用是可以让我们后期维护代码更加方便。

合理的使用 assert 可以增加代码的健壮度,同时也方便开发人员定位问题。当程序中出现的各种情况都是预料之中的时候,应该使用 if 来处理,当程序中要限定一些条件时,可以方便的使用 assert 简化处理。需要注意的是无论有没有 assert,都不应该影响程序的正常运行,assert 的作用是可以让我们后期维护代码更加方便,另外对于程序在运行时抛出的异常,看情况要进行捕获与否。

(完)

发布了45 篇原创文章 · 获赞 44 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/somenzz/article/details/104549187