pytest-pipeline 0.3.0

pytest-pipeline

pytest-pipeline是用于数据分析管道功能测试的pytest插件。它们通常是具有多个输入和/或输出文件/目录的长时间运行的脚本或可执行文件。该插件适用于端到端测试，您可以在管道运行之前和之后测试条件（输出文件、校验和等）。

pytest-pipeline已在Python版本2.7、3.3、3.4、3.5和3.6上进行了测试。

安装

pip install pytest-pipeline

安装指南

在我们的示例中，我们将使用一个超级简单的管道，该管道写入文件并打印到stdout

#!/usr/bin/env python

from __future__ import print_function

if __name__ == "__main__":

    with open("result.txt", "w") as result:
        result.write("42\n")
    print("Result computed")

目前这只是一个简单的脚本，但它应该足以说明插件。此外，如果您想跟上来，将上述文件保存为run_pipeline。

有了上述管道，以下是您的测试看起来像pytest_pipeline

import os
import shutil
import unittest
from pytest_pipeline import PipelineRun, mark, utils

# we can subclass `PipelineRun` to add custom methods
# using `PipelineRun` as-is is also possible
class MyRun(PipelineRun):

    # before_run-marked functions will be run before the pipeline is executed
    @mark.before_run
    def test_prep_executable(self):
        # copy the executable to the run directory
        shutil.copy2("/path/to/run_pipeline", "run_pipeline")
        # ensure that the file is executable
        assert os.access("run_pipeline", os.X_OK)

# a pipeline run is treated as a test fixture
run = MyRun.class_fixture(cmd="./run_pipeline", stdout="run.stdout")

# the fixture is bound to a unittest.TestCase using the usefixtures mark
@pytest.mark.usefixtures("run")
# tests per-pipeline run are grouped in one unittest.TestCase instance
class TestMyPipeline(unittest.TestCase):

    def test_result_md5(self):
        assert utils.file_md5sum("result.txt") == "50a2fabfdd276f573ff97ace8b11c5f4"

    def test_exit_code(self):
        # the run fixture is stored as the `run_fixture` attribute
        assert self.run_fixture.exit_code == 0

    # we can check the stdout that we capture as well
    def test_stdout(self):
        assert open("run.stdout", "r").read().strip() == "Result computed"

如果上面的测试保存为test_demo.py，您可以通过执行py.test -v test_demo.py来运行测试。您应该看到执行了三个测试，全部四个都通过了。

获取和提供帮助

请使用问题跟踪器来报告错误或功能请求。您也可以分叉并提交一个pull请求。

许可证

查看LICENSE。

变更日志