Python assertRaises 用户定义的异常

Question

以下问题是由中的讨论引发的。

假设有两个文件（foobar.py 和 foobar_unittest.py）。文件 foobar.py 包含一个 class (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)

Answer 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

Answer 2

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

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

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

无论出于何种原因，这也行不通。写这个就像

from foobar import MyException

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

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

Python assertRaises 用户定义的异常

Python assertRaises on user-defined exceptions

python

exception-handling

class

message-passing

assertraises