我也倾向于把我的单元测试放在文件本身中，正如Jeremy Cantrell上面所指出的，尽管我倾向于不把测试函数放在主体中，而是把所有东西都放在一个文件中

if __name__ == '__main__':
   do tests...

块。这最终会将文档添加到文件中，作为如何使用您正在测试的python文件的“示例代码”。

我应该补充一点，我倾向于编写非常紧凑的模块/类。如果你的模块需要大量的测试，你可以把它们放在另一个测试中，但即使这样，我仍然会补充:

if __name__ == '__main__':
   import tests.thisModule
   tests.thisModule.runtests

这让任何阅读源代码的人都知道到哪里去寻找测试代码。

我使用tests/目录，然后使用相对导入导入主要应用程序模块。在MyApp/tests/foo.py中，可能有:

from .. import foo

导入MyApp。foo模块。

我更喜欢顶层测试目录。这确实意味着进口变得更加困难。对此，我有两个解决方案:

使用setuptools。然后你可以将test_suite='tests.runalltests.suite'传递给setup()，并可以简单地运行测试 在运行测试时设置PYTHONPATH: PYTHONPATH=。python测试/ runalltests.py

下面是M2Crypto中的代码是如何支持这些东西的:

http://svn.osafoundation.org/m2crypto/trunk/setup.py http://svn.osafoundation.org/m2crypto/trunk/tests/alltests.py

如果您更喜欢使用鼻子测试运行测试，则可能需要做一些不同的事情。

我建议你在GitHub上查看一些主要的Python项目，并获得一些想法。

当你的代码变大，你添加了更多的库，最好在你有setup.py的同一个目录下创建一个测试文件夹，并为每种测试类型镜像你的项目目录结构(unittest, integration，…)

例如，如果你有一个这样的目录结构:

myPackage/
    myapp/
       moduleA/
          __init__.py
          module_A.py
       moduleB/
          __init__.py
          module_B.py
setup.py

添加测试文件夹后，您将拥有如下目录结构:

myPackage/
    myapp/
       moduleA/
          __init__.py
          module_A.py
       moduleB/
          __init__.py
          module_B.py
test/
   unit/
      myapp/
         moduleA/
            module_A_test.py
         moduleB/
            module_B_test.py
   integration/
          myapp/
             moduleA/
                module_A_test.py
             moduleB/
                module_B_test.py
setup.py

许多正确编写的Python包使用相同的结构。Boto包就是一个很好的例子。 检查https://github.com/boto/boto

只有一个测试文件

如果只有1个测试文件，建议将其放在顶级目录中:

module/
    lib/
        __init__.py
        module.py
    test.py

在CLI下运行测试

python test.py

许多测试文件

如果有很多测试文件，将其放在tests文件夹中:

module/
    lib/
        __init__.py
        module.py
    tests/
        test_module.py
        test_module_function.py

# test_module.py

import unittest
from lib import module

class TestModule(unittest.TestCase):
    def test_module(self):
        pass

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

在CLI下运行测试

# In top-level /module/ folder
python -m tests.test_module
python -m tests.test_module_function

使用单元测试发现

单元测试发现将在包文件夹中找到所有测试。

在tests/文件夹中创建__init__.py

module/
    lib/
        __init__.py
        module.py
    tests/
        __init__.py
        test_module.py
        test_module_function.py

在CLI下运行测试

# In top-level /module/ folder

# -s, --start-directory (default current directory)
# -p, --pattern (default test*.py)

python -m unittest discover

参考

用于测试布局的pytest良好实践 unittest

单元测试框架

鼻子 nose2 pytest

如果测试很简单，就把它们放在文档字符串中——大多数Python测试框架都可以使用它:

>>> import module
>>> module.method('test')
'testresult'

对于其他更复杂的测试，我将它们放在../tests/test_module.py或tests/test_module.py中。