设置 py.test

py.test 是可用于 Python 的几个第三方测试库之一。它可以使用安装 pip用

pip install pytest

要测试的代码

假设我们正在测试 projectroot/module/code.py 中的添加功能：

# projectroot/module/code.py
def add(a, b):
    return a + b

测试代码

我们在 projectroot/tests/test_code.py 中创建了一个测试文件。该文件必须以 test_ 开头才能被识别为测试文件。

# projectroot/tests/test_code.py
from module import code

def test_add():
    assert code.add(1, 2) == 3

运行测试

从 projectroot 我们只需运行 py.test：

# ensure we have the modules
$ touch tests/__init__.py
$ touch module/__init__.py
$ py.test
================================================== test session starts ===================================================
platform darwin -- Python 2.7.10, pytest-2.9.2, py-1.4.31, pluggy-0.3.1
rootdir: /projectroot, inifile:
collected 1 items

tests/test_code.py .

================================================ 1 passed in 0.01 seconds ================================================