如何更改 Python AssertionError 中的消息？

Question

我正在编写以下内容，在比较两个多行 Unicode 文本块时，我尝试生成一个体面的错误消息。进行比较的内部方法引发了一个断言，但默认解释对我来说毫无用处

我需要在代码中添加一些内容，如下所示：

def assert_long_strings_equal(one, other):
    lines_one = one.splitlines()
    lines_other = other.splitlines()
    for line1, line2 in zip(lines_one, lines_other):
        try:
            my_assert_equal(line1, line2)
        except AssertionError, error:
            # Add some information to the printed result of error??!
            raise

我无法弄清楚如何更改我捕获的 assertionerror 中打印的错误消息。我总是得到AssertionError: u'something' != 'something else'，它只显示输出的第一行。

如何更改断言消息以打印出我想要的任何内容？

如果相关，我将使用nose 运行测试。

Answer 1

您可以在创建异常时传递所需的消息。

raise AssertionError(line1 + ' != ' + line2)

希望这会有所帮助。

Answer 2

assert expression, info

例如，

>>> assert False, "Oopsie"
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AssertionError: Oopsie

---

来自docs：

断言语句是一种方便的方法 将调试断言插入到 程序：

assert_stmt ::=  "assert" expression
["," expression] 

简单的形式， assert expression，相当于

if __debug__:
    if not expression:
        raise AssertionError 

扩展形式

assert expression1, expression2

等价于

if __debug__:
    if not expression1:
        raise AssertionError(expression2)

这些等价假设 __debug__ 和 AssertionError 指的是内置变量 名字。在当前的实现中， 内置变量__debug__ 是 正常情况下为真，假 请求优化时 （命令行选项 -O）。目前的 代码生成器不为 优化时的断言语句 在编译时请求。注意 没有必要包括 表达式的源代码 错误消息中失败；它会 显示为堆栈的一部分 跟踪。

Answer 3

您想获取捕获的异常，将其转换为字符串，将其与一些额外的字符串信息结合起来，然后引发新的异常。

x = 3
y = 5
try:
    assert( x == y )
except AssertionError, e:
    raise( AssertionError( "Additional info. %s"%e ) )

Answer 4

使用e.args，e.message 已弃用。

try:
    assert False, "Hello!"
except AssertionError as e:
    e.args += ('some other', 'important', 'information', 42)
    raise

这会保留原始回溯。它的最后一部分看起来像这样：

AssertionError: ('Hello!', 'some other', 'important', 'information', 42)

适用于 Python 2.7 和 Python 3。