Python assertRaises 处理用户定义的异常答案

【问题标题】：Python assertRaises on user-defined exceptionsPython assertRaises 处理用户定义的异常
【发布时间】：2016-02-18 19:36:42
【问题描述】：

this post的讨论引发了以下问题。

假设有两个文件（foobar.py 和 foobar_unittest.py）。文件 foobar.py 包含一个类 (FooBar)，该类具有两个函数 (foo 和 bar)。函数 bar 引发内置异常，函数 foo 引发用户定义异常。

# foobar.py
class MyException(Exception):
    pass
class FooBar:
    def __init__(self):
        pass
    def bar(self):
        raise ValueError('Hello World.')
    def foo(self):
        raise MyException('Hello World.')

# foobar_unittest.py
import unittest
import foobar as fb
class MyException(Exception):
    pass
class FooBarTestCases(unittest.TestCase):
    def test_bar(self):
        with self.assertRaises(ValueError):
            fb.FooBar().bar()
    def test_foo(self):
        with self.assertRaises(MyException):
            fb.FooBar().foo()
if __name__ == '__main__':
    unittest.main()

在 foobar.py 上运行单元测试时，为什么引发用户定义异常 (foo) 的函数无法通过测试？

>>> python2.7 foobar_unittest.py 
.E
======================================================================
ERROR: test_foo (__main__.FooBarTestCases)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "foobar_unittest.py", line 11, in test_foo
    fb.FooBar().foo()
  File "/a_path/foobar.py", line 9, in foo
    raise MyException('Hello World.')
MyException: Hello World.

----------------------------------------------------------------------
Ran 2 tests in 0.000s

FAILED (errors=1)

【问题讨论】：

您定义了两个不同的MyException 类。您引发的异常与您试图捕获的异常不同。
@VincentSavard 如果我要删除 foobar_unittest.py 中定义 MyException 的两行，那么我会收到以下错误：NameError: global name 'MyException' is not defined。

标签： python class exception-handling message-passing assertraises

【解决方案1】：

从 foobar 导入 MyException，不要重新定义它。

import unittest
from foobar import MyException
import foobar as fb

class FooBarTestCases(unittest.TestCase):
    def test_bar(self):
        with self.assertRaises(ValueError):
            fb.FooBar().bar()
    def test_foo(self):
        with self.assertRaises(MyException):
            fb.FooBar().foo()
if __name__ == '__main__':
    unittest.main()

这段代码现在应该可以作为

..
----------------------------------------------------------------------
Ran 2 tests in 0.001s

OK

【讨论】：

是的，这就是上述问题的答案。谢谢！

【解决方案2】：

请注意，如果您使用 reload 导入您的异常，也会发生同样的情况（发生在我身上）。

在我的单元测试中，我有相关的导入，例如

from importlib import reload
import foobar
reload(foobar)
from foobar import MyException

无论出于何种原因，这也不起作用。写这个就像

from foobar import MyException

会起作用。然后，当然，您必须自己重新加载模块。

如果你想知道我为什么要使用 reload：How do I unload (reload) a Python module?。

【讨论】：