Python库项目结构最佳实践:导入和测试

时间:2016-08-04 17:17:52

标签: python python-unittest project-structure

我想重构一个python库,我在日常工作中经常使用它作为开源在github上发布。 在这样做之前,我希望符合某种python项目结构的最佳实践。 我将在下面描述我想要做的事情,我将非常感谢您的建议。

这是我的库(mylib)结构:

mylib/
   /examples/
       simple_example.py
   /mylib/
       __init__.py
       foo.py
       bar.py
   /tests/
       test_foo.py
       test_bar.py

以下是文件:

#foo.py
def Foo():
    print("foo.Foo")


#bar.py
import foo

def Bar():
    print("bar.Bar")
    foo.Foo()


#test_bar.py
from ..mylib import bar #doesnt work!

class TestBar(unittest.TestCase):
    def test_1(self):
        bar.Bar()
        self.assertEqual(True, True)

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


#simple_example.py
from .. import foo #doesnt work!
from .. import bar #doesnt work!

if __name__ == '__main__':  
    foo.Foo()
    bar.Bar()

我想做的是:

1-理想情况下从/ mylib / examples /:

执行simple_example.py
$cd myapp
$cd examples
$python simple_example.py
Traceback (most recent call last):
  File "simple_example.py", line 2, in <module>
    from .. import foo
SystemError: Parent module '' not loaded, cannot perform relative import

2-理想情况下从/ mylib / tests /:

执行单个测试文件
$cd myapp
$cd tests
$python test_bar.py
Traceback (most recent call last):
  File "test_bar.py", line 3, in <module>
    from ..mylib import bar
SystemError: Parent module '' not loaded, cannot perform relative import    

3-执行mylib root的所有测试

$cd myapp
$python -m unittest discover tests #same problem as above!

因此,问题出在simple_example.py和test_bar.py中的import语句中。 修复这些导入的最佳方法是什么?

请注意,我想使用python标准的lib unittest进行单元测试。

由于

查理

1 个答案:

答案 0 :(得分:1)

运行测试代码时,您希望进行绝对导入。这是因为当您进行单元测试等时,您应该假设您的图书馆&#39;安装在本地开发模式下进行测试 - 不要使用相对导入,因为你不在同一个包中。

以下是您在test_foo.py文件中导入的方法,例如:

# test_foo.py
from mylib.foo import Foo

# ... your test code here

通常,您应该只使用库代码的INSIDE相对导入,而不是在测试中使用=)

我希望这会有所帮助。

编辑:您还需要在开发模式下安装库才能运行。您可以通过以下两种方式之一完成此操作:

$ python setup.py develop

OR

$ pip install -e .

上述任何一个命令都会检查你的项目setup.py文件,该文件告诉Python你的软件包是如何构建/创建的,并将在本地安装它以便你可以运行测试/搞乱它。 / p>