python - 断言某个错误发生

Question

给定以下代码:

def fun(param):
    if param == 'a':
        assert False
    else:
        return None

# fun('a') THIS THROWS AN AssertionError
# PSEUDOCODE FOR WHAT I WANT TO DO:
# assert (fun('a') throws an AssertionError)
assert fun('b') == None

我如何断言第一个函数调用 (fun('a')) 会给我一个 AssertionError？我需要为此 try catch 还是有更优雅的方法？

Answer 1

您可以使用 pytest为此:

import pytest
...
with pytest.raises(AssertionError):
    fun('a')

如果 fun('a') 不引发 AssertionError，这将引发错误。

或者，如果您正在使用 unittest，并且您在 TestCase 中，您可以使用 assert
Raises :

self.assertRaises(AssertionError, fun, 'a')

此外，正如其他答案所提到的，与断言 False 相比，raise 错误会更好。如果您确实引发了错误，您可以引发一个告诉用户更多有关错误的信息，或者引发您自己的自定义异常:

import pytest

class AError(Exception):
    pass

def fun(param):
    if param == 'a':
        raise AError("You passed 'a'; don't do that!")
    else:
        return None

with pytest.raises(AError):
    fun('a')