在python (范例)中,使用断言方便

・1 分钟阅读

最近,我学习了断言的简单用法,所以,我想做一个备忘录。

当结果为false时,assert返回AssertionError 。

 

>>> assert False
AssertionError:

如果将断言与逗号(, )和字符串连接,那么字符串将在下面的错误之后,显示。

 

>>> assert False, "it's wrong"
AssertionError: it's wrong

因此,如果要在函数中声明参数,可以轻松编写简单的检查:

 

def do_something(mode="train"):
 assert mode in ['train', 'test'], "mode only takes either train or test"

通过这样写,一个简单的检查变得小菜一碟。

 

>>> do_something(mode='something')
AssertionError: mode only takes either train or test.

今天就是这样了。

Xajhqffl profile image