从模块导入__init__.py时，发出测试用户定义的异常

Question

我正在尝试将大型模块分成多个文件，同时将其统一在一个逻辑模块中，如Ch. 10 of the Python Cookbook

中所述

但是，在我的单元测试中，我在测试异常时收到意外错误。当我试图强制异常时，我希望引发异常package.MyExceptionError。但是，我的测试失败了，因为我收到异常package.module.MyExceptionError。另一方面，当我从已通过__init__.py导入的模块调用函数时，我可以按预期将其称为package.my_function()。

我已经查看了几个类似布局的PyPi模块的代码，但是我无法确定我做错了什么，或者我误解了它应该如何工作？

下面是一个虚拟包，其中包含复制问题所需的最少代码。

我的文件结构是：

Spam/
    spam/
        __init__.py
        eggs.py

    test/
        test_eggs.py

以下是spam/eggs.py的内容：

class EggError(Exception):
    pass



def egg(b=True):

    if b == True:
        return 0

    else:
        raise EggError('Spam!')



if __name__ == '__main__':
    pass

以下是spam/__init__.py的内容：

from .egg import egg
from .egg import EggError

__all__ = ['egg', 'EggError', ]

if __name__ == '__main__':
    pass

以下是test/test_spam.py的内容：

import unittest
import spam


class Test(unittest.TestCase):

    def test_egg(self):
        self.assertEquals(spam.egg(), 0)

    def test_EggError(self):
        self.assertRaises(spam.EggError, spam.egg(False))


if __name__ == "__main__":
    unittest.main()

这些是我运行测试时得到的结果：

Finding files... done.
Importing test modules ... done.

======================================================================
ERROR: test_EggError (test.test_spam.Test)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "K:\Scripting\Python\err_test\test\test_spam.py", line 11, in test_EggError
    self.assertRaises(spam.EggError, spam.egg(False))
  File "K:\Scripting\Python\err_test\spam\eggs.py", line 12, in egg
    raise EggError('Spam!')
spam.eggs.EggError: Spam!

----------------------------------------------------------------------
Ran 2 tests in 0.415s

FAILED (errors=1)

Answer 1

在spam.egg()有机会捕获异常之前，您正在调用assertRaises()。您应该按如下方式编写该测试：

self.assertRaises(spam.EggError, spam.egg, False)

assertRaises()会为您致电spam.egg()。

如果您发现这很难阅读，您可以使用上下文管理器样式：

with self.assertRaises(spam.EggError):
    spam.egg(False)