在Python中使用unittest进行相对导入

问题描述

我正在尝试使用Python单元测试和相对导入，但我似乎搞不清楚。我知道有很多相关的问题，但到目前为止，这些问题都没有帮助。如果这是重复的，很抱歉，但我真的很感激任何人的帮助。我试图使用PEP 328http://www.python.org/dev/peps/pep-0328/中的语法，但我一定是搞错了。

我的目录结构是：

project/
    __init__.py
    main_program.py
    lib/
        __init__.py
        lib_a
        lib_b
    tests/
        __init__.py
        test_a
        test_b

我使用以下命令运行测试：

python -m unittest test_module1 test_module2

test_a需要同时导入lib/lib_a和main_Program。这是我尝试用于导入的test_a代码：

from ..lib import lib_a as lib
from ...project import main_program

两者都引发此错误：

ValueError: Attempted relative import in non-package

我的所有init.py文件当前均为空。

如有任何具体建议，我们将不胜感激！

编辑：

这可能就是答案：Python Packages? 我还在核实这是否会奏效。

编辑II：

为澄清起见，此时我尝试以3种不同的方式运行测试文件：

project/tests $ python -m unittest test_a
project/tests $ python -m test_a
project/tests $ ./test_a

所有三个都失败，错误与上面相同。当我在项目目录中使用相同的三种语法时，我得到这个错误：

ValueError: Attempted relative import beyond toplevel package

再次感谢。

---

解决方案

根据我的经验，如果您的项目根目录不是包，则最简单，如下所示：

project/
  test.py
  run.py
  package/
    __init__.py
    main_program.py
    lib/
      __init__.py
      lib_a
      lib_b
    tests/
      __init__.py
      test_a
      test_b

然而，从python3.2开始，unittest模块提供了-t选项，允许您设置顶级目录，因此您可以(从package/)：

python -m unittest discover -t ..

有关更多详细信息，请访问unittest docs。