Python Assert

最新推荐文章于 2024-07-19 16:48:59 发布

lvhdbb

最新推荐文章于 2024-07-19 16:48:59 发布

阅读量141

点赞数

分类专栏： python

本文链接：https://blog.youkuaiyun.com/lvhdbb/article/details/96770151

版权

20 篇文章

订阅专栏

本文探讨了Python中默认断言的不足及其改进方案，包括自定义断言信息、使用测试框架如py.test和unittest，以及推荐了assertpy库。断言的改进能提高代码的可读性和调试效率，选择合适的断言方式取决于具体需求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Python中的断言用起来非常简单，你可以在assert后面跟上任意判断条件，如果断言失败则会抛出异常。

其实assert看上去不错，然而用起来并不爽。就比如有人告诉你程序错了，但是不告诉哪里错了。很多时候这样的assert还不如不写，写了我就想骂娘。直接抛一个异常来得更痛快一些。

一个稍微改进一丢丢的方案就是把必要的信息也放到assert语句后面，比如这样。

看上去还行吧，但是其实写的很蛋疼。假如你是一名测试汪，有成千上万的测试案例需要做断言做验证，相信你面对以上做法，心中一定有千万只那种马奔腾而过。

不管你是你是搞测试还是开发的，想必听过不少测试框架。你猜到我要说什么了吧？对，不用测试框架里的断言机制，你是不是洒。

py.test

py.test 是一个轻量级的测试框架，所以它压根就没写自己的断言系统，但是它对Python自带的断言做了强化处理，如果断言失败，那么框架本身会尽可能多地提供断言失败的原因。那么也就意味着，用py.test实现测试，你一行代码都不用改。

import pytest
def test_case():
expected = "Hello"
actual = "hello"
assert expected == actual
if __name__ == '__main__':
pytest.main()
"""
================================== FAILURES ===================================
__________________________________ test_case __________________________________
def test_case():
expected = "Hello"
actual = "hello"
> assert expected == actual
E assert 'Hello' == 'hello'
E - Hello
E ? ^
E + hello
E ? ^
assertion_in_python.py:7: AssertionError
========================== 1 failed in 0.05 seconds ===========================
""""

unittest

Python自带的unittest单元测试框架就有了自己的断言方法self.assertXXX()，而且不推荐使用assert XXX语句。

ptest

我非常喜欢ptest，感谢Karl大神写了这么一个测试框架。ptest中的断言可读性很好，而且通过IDE的智能提示你能轻松完成各种断言语句。

from ptest.decorator import *
from ptest.assertion import *
@TestClass()
class TestCases:
@Test()
def test1(self):
actual = 'foo'
expected = 'bar'
assert_that(expected).is_equal_to(actual)
"""
Start to run following 1 tests:
------------------------------
...
[demo.assertion_in_python.TestCases.test1@Test] Failed with following message:
...
AssertionError: Unexpectedly that the str <bar> is not equal to str <foo>.
"""