Python中的Assert语句怎么使用

2023-06-15 14:03:09 python assert 语句

这篇文章主要介绍了Python中的Assert语句怎么使用的相关知识,内容详细易懂,操作简单快捷,具有一定借鉴价值,相信大家阅读完这篇Python中的Assert语句怎么使用文章都会有所收获,下面我们一起来看看吧。

1. 引言

如果你是一个Python新手,那么你可能会在一些代码中看到

assert
关键字的语句。

2. 基础语法

Assert语句的基础语法如下:

assert boolean_expression

这里,我们期望上述

boolean_expression
取值为True,而assert关键字的作用是帮助我们再次检查它是否真的为True。

  • 此时如果我们断言是

    True
    ,那么一切都好,没有什么错,我们可以像往常一样继续往下执行。

  • 但是,如果我们断言为False,则会发生如下断言错误

Traceback (most recent call last):
  File "/your/path/a.py", line 2, in <module>
    assert False
AssertionError

3. 基础用法

我们可以将assert关键字作为代码中的某种健全性检查。比如在我们的代码中的某个点上,如果我们期望某些东西以某种方式出现,此时我们就可以使用

assert
关键字来验证它。

例如,假设我们有一个复杂的函数

get_price()
,它主要用于检索某个物品的价格。这里,假设价格不能是负数。

price = get_price()
assert price >= 0

上述代码中,语句

assert price>0
主要用于检查并验证
price
不是负数。如果函数
get_price()
有错误并返回负数,则会发生断言错误并停止运行程序。

4. 断言并输出提示信息

更进一步,我们可以添加断言消息,该消息与断言错误一起显示。 语法如下:

assert boolean_expression, message

举例如下:

price = -1
assert price > 0, "price must not be negative"

输出如下:

Traceback (most recent call last):
  File "/your/path/a.py", line 3, in <module>
    assert price > 0, "price must not be negative"
AssertionError: price must not be negative

相关文章