Python assertRaises 對用戶定義的異常

Question

下面的問題是由這篇文章中的討論引發的。

假設有兩個文件（ foob​​ar.py和foob​​ar_unittest.py ）。 文件foob​​ar.py包含一個具有兩個函數（ foo和bar ）的類 ( FooBar )。 函數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()

在foob​​ar.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

將工作。 然后，當然，您必須自己重新加載模塊。

如果您想知道我為什么使用 reload： 如何卸載（重新加載）Python 模塊？ .