如果函数没有抛出预期的异常，那么如何写一个单元测试失败？
答案 0 :(得分：544)
使用unittest模块中的TestCase.assertRaises（或TestCase.failUnlessRaises），例如：
import mymod
class MyTestCase(unittest.TestCase):
def test1(self):
self.assertRaises(SomeCoolException, mymod.myfunc)
答案 1 :(得分：389)
从Python 2.7开始，您可以使用上下文管理器来获取抛出的实际Exception对象：
import unittest
def broken_function():
raise Exception('This is broken')
class MyTestCase(unittest.TestCase):
def test(self):
with self.assertRaises(Exception) as context:
broken_function()
self.assertTrue('This is broken' in context.exception)
if __name__ == '__main__':
unittest.main()
http://docs.python.org/dev/library/unittest.html#unittest.TestCase.assertRaises
在 Python 3.5 中，您必须在context.exception中包含str，否则您将获得TypeError
self.assertTrue('This is broken' in str(context.exception))
答案 2 :(得分：260)
我之前回答中的代码可以简化为：
def test_afunction_throws_exception(self):
self.assertRaises(ExpectedException, afunction)
如果功能接受参数，只需将它们传递给assertRaises：
def test_afunction_throws_exception(self):
self.assertRaises(ExpectedException, afunction, arg1, arg2)
答案 3 :(得分：112)
如何测试Python函数是否会引发异常？
如果一个函数没有抛出，那么如何编写一个失败的测试 预期的例外？
使用self.assertRaises方法作为上下文管理器：
def test_1_cannot_add_int_and_str(self):
with self.assertRaises(TypeError):
1 + '1'
最佳实践方法很容易在Python shell中演示。
unittest库
在Python 2.7或3中：
import unittest
在Python 2.6中，您可以安装一个名为unittest2的2.7 unittest库的反向端口，并将其别名为unittest：
import unittest2 as unittest
现在，在Python shell中粘贴Python的类型安全性的以下测试：
class MyTestCase(unittest.TestCase):
def test_1_cannot_add_int_and_str(self):
with self.assertRaises(TypeError):
1 + '1'
def test_2_cannot_add_int_and_str(self):
import operator
self.assertRaises(TypeError, operator.add, 1, '1')
测试一个使用assertRaises作为上下文管理器，这可以确保在记录时正确捕获和清除错误。
我们也可以在没有上下文管理器的情况下编写，参见测试二。第一个参数是您希望引发的错误类型，第二个参数，您正在测试的函数，剩余的args和关键字args将传递给该函数。
我认为使用上下文管理器更加简单，易读和易于维护。
运行测试：
unittest.main(exit=False)
在Python 2.6中，您可能need the following：
unittest.TextTestRunner().run(unittest.TestLoader().loadTestsFromTestCase(MyTestCase))
您的终端应输出以下内容：
..
----------------------------------------------------------------------
Ran 2 tests in 0.007s
OK
<unittest2.runner.TextTestResult run=2 errors=0 failures=0>
我们发现，正如我们所料，尝试在1中添加'1'和TypeError结果。
要获得更详细的输出，请尝试以下方法：
unittest.TextTestRunner(verbosity=2).run(unittest.TestLoader().loadTestsFromTestCase(MyTestCase))
答案 4 :(得分：41)
您的代码应遵循此模式（这是一个单元测试模块样式测试）：
def test_afunction_throws_exception(self):
try:
afunction()
except ExpectedException:
pass
except Exception as e:
self.fail('Unexpected exception raised:', e)
else:
self.fail('ExpectedException not raised')
在Python上＆lt; 2.7此构造对于检查预期异常中的特定值很有用。 unittest函数assertRaises仅检查是否引发了异常。
答案 5 :(得分：17)
由于我还没有看到任何有关如何使用上下文管理器检查是否在接受的列表中发现特定异常的详细解释，或者我将添加我的其他异常详细信息（在python 3.8上进行了检查）。
如果我只想检查例如TypeError的函数是否在上升，我会写：
with self.assertRaises(TypeError):
function_raising_some_exception(parameters)
如果我想检查函数是提高TypeError还是IndexError，我会写：
with self.assertRaises((TypeError,IndexError)):
function_raising_some_exception(parameters)
如果我想获得有关引发异常的更多详细信息，我可以在这样的上下文中捕获它：
# Here I catch any exception
with self.assertRaises(Exception) as e:
function_raising_some_exception(parameters)
# Here I check actual exception type (but I could
# check anything else about that specific exception,
# like it's actual message or values stored in the exception)
self.assertTrue(type(e.exception) in [TypeError,MatrixIsSingular])
答案 6 :(得分：14)
来自：http://www.lengrand.fr/2011/12/pythonunittest-assertraises-raises-error/
首先，这是文件dum_function.py中相应的（仍为dum：p）函数：
def square_value(a):
"""
Returns the square value of a.
"""
try:
out = a*a
except TypeError:
raise TypeError("Input should be a string:")
return out
以下是要执行的测试（仅插入此测试）：
import dum_function as df # import function module
import unittest
class Test(unittest.TestCase):
"""
The class inherits from unittest
"""
def setUp(self):
"""
This method is called before each test
"""
self.false_int = "A"
def tearDown(self):
"""
This method is called after each test
"""
pass
#---
## TESTS
def test_square_value(self):
# assertRaises(excClass, callableObj) prototype
self.assertRaises(TypeError, df.square_value(self.false_int))
if __name__ == "__main__":
unittest.main()
我们现在准备测试我们的功能了！以下是尝试运行测试时发生的情况：
======================================================================
ERROR: test_square_value (__main__.Test)
----------------------------------------------------------------------
Traceback (most recent call last):
File "test_dum_function.py", line 22, in test_square_value
self.assertRaises(TypeError, df.square_value(self.false_int))
File "/home/jlengrand/Desktop/function.py", line 8, in square_value
raise TypeError("Input should be a string:")
TypeError: Input should be a string:
----------------------------------------------------------------------
Ran 1 test in 0.000s
FAILED (errors=1)
TypeError会引发actullay，并生成测试失败。问题是这正是我们想要的行为：s。
要避免此错误，只需在测试调用中使用lambda运行该函数：
self.assertRaises(TypeError, lambda: df.square_value(self.false_int))
最终输出：
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
完美！
......对我来说也很完美!!
Thansk很多Julien Lengrand-Lambert先生
答案 7 :(得分：10)
您可以构建自己的contextmanager来检查是否引发了异常。
import contextlib
@contextlib.contextmanager
def raises(exception):
try:
yield
except exception as e:
assert True
else:
assert False
然后您可以像这样使用raises：
with raises(Exception):
print "Hola" # Calls assert False
with raises(Exception):
raise Exception # Calls assert True
如果您使用的是pytest，则此功能已经实施。你可以pytest.raises(Exception)：
示例：
def test_div_zero():
with pytest.raises(ZeroDivisionError):
1/0
结果：
pigueiras@pigueiras$ py.test
================= test session starts =================
platform linux2 -- Python 2.6.6 -- py-1.4.20 -- pytest-2.5.2 -- /usr/bin/python
collected 1 items
tests/test_div_zero.py:6: test_div_zero PASSED
答案 8 :(得分：10)
我几乎到处都使用 doctest [1]因为我喜欢同时记录和测试我的功能的事实。
看一下这段代码：
def throw_up(something, gowrong=False):
"""
>>> throw_up('Fish n Chips')
Traceback (most recent call last):
...
Exception: Fish n Chips
>>> throw_up('Fish n Chips', gowrong=True)
'I feel fine!'
"""
if gowrong:
return "I feel fine!"
raise Exception(something)
if __name__ == '__main__':
import doctest
doctest.testmod()
如果将此示例放在模块中并从命令行运行它，则会评估和检查两个测试用例。
[1] Python documentation: 23.2 doctest -- Test interactive Python examples
答案 9 :(得分：9)
查看unittest模块的assertRaises方法。
答案 10 :(得分：6)
我刚刚发现Mock library提供了一个assertRaisesWithMessage（）方法（在其unittest.TestCase子类中），它不仅会检查引发预期的异常，还会检查它是否与预期的消息一起引发：
from testcase import TestCase
import mymod
class MyTestCase(TestCase):
def test1(self):
self.assertRaisesWithMessage(SomeCoolException,
'expected message',
mymod.myfunc)
答案 11 :(得分：2)
您可以使用unittest模块中的assertRaises
import unittest
class TestClass():
def raises_exception(self):
raise Exception("test")
class MyTestCase(unittest.TestCase):
def test_if_method_raises_correct_exception(self):
test_class = TestClass()
# note that you dont use () when passing the method to assertRaises
self.assertRaises(Exception, test_class.raises_exception)
答案 12 :(得分：1)
您可以使用上下文管理器运行错误的函数，并使用assertRaisesMessage来断言该函数引发带有特定消息的异常
with self.assertRaisesMessage(SomeException,'Some error message e.g 404 Not Found'):
faulty_funtion()
答案 13 :(得分：0)
这里有很多答案。该代码显示了如何创建异常，如何在我们的方法中使用该异常以及最后如何在单元测试中验证正确的异常。
import unittest
class DeviceException(Exception):
def __init__(self, msg, code):
self.msg = msg
self.code = code
def __str__(self):
return repr("Error {}: {}".format(self.code, self.msg))
class MyDevice(object):
def __init__(self):
self.name = 'DefaultName'
def setParameter(self, param, value):
if isinstance(value, str):
setattr(self, param , value)
else:
raise DeviceException('Incorrect type of argument passed. Name expects a string', 100001)
def getParameter(self, param):
return getattr(self, param)
class TestMyDevice(unittest.TestCase):
def setUp(self):
self.dev1 = MyDevice()
def tearDown(self):
del self.dev1
def test_name(self):
""" Test for valid input for name parameter """
self.dev1.setParameter('name', 'MyDevice')
name = self.dev1.getParameter('name')
self.assertEqual(name, 'MyDevice')
def test_invalid_name(self):
""" Test to check if error is raised if invalid type of input is provided """
self.assertRaises(DeviceException, self.dev1.setParameter, 'name', 1234)
def test_exception_message(self):
""" Test to check if correct exception message and code is raised when incorrect value is passed """
with self.assertRaises(DeviceException) as cm:
self.dev1.setParameter('name', 1234)
self.assertEqual(cm.exception.msg, 'Incorrect type of argument passed. Name expects a string', 'mismatch in expected error message')
self.assertEqual(cm.exception.code, 100001, 'mismatch in expected error code')
if __name__ == '__main__':
unittest.main()
答案 14 :(得分：0)
对于 await/async aiounittest 有一个稍微不同的模式：
https://aiounittest.readthedocs.io/en/latest/asynctestcase.html#aiounittest.AsyncTestCase
async def test_await_async_fail(self):
with self.assertRaises(Exception) as e:
await async_one()
答案 15 :(得分：0)
如果在此类中将 stock_id 设置为 Integer 将引发错误，则这将引发 TypeError，如果发生这种情况，测试将通过，否则将失败
def set_string(prop, value):
if not isinstance(value, str):
raise TypeError("i told you i take strings only ")
return value
class BuyVolume(ndb.Model):
stock_id = ndb.StringProperty(validator=set_string)
from pytest import raises
buy_volume_instance: BuyVolume = BuyVolume()
with raises(TypeError):
buy_volume_instance.stock_id = 25
答案 16 :(得分：-2)
尽管所有答案都很好，但我一直在寻找一种方法来测试函数是否引发异常，而无需依赖单元测试框架并且不必编写测试类。
我最终写了以下内容：
def assert_error(e, x):
try:
e(x)
except:
return
raise AssertionError()
def failing_function(x):
raise ValueError()
def dummy_function(x):
return x
if __name__=="__main__":
assert_error(failing_function, 0)
assert_error(dummy_function, 0)
它在右行失败：
Traceback (most recent call last):
File "assert_error.py", line 16, in <module>
assert_error(dummy_function, 0)
File "assert_error.py", line 6, in assert_error
raise AssertionError()
AssertionError