诗歌+赛顿+测试(鼻子测试)

2024-04-20 07:47:43 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用诗歌来构建带有cython扩展的包。现在我想为它编写测试(最好使用nosetest)。问题是我需要预编译二进制文件,通常使用setup.py build_clib build_ext --inplace

对我来说,最好的解决方案是运行测试,而不在目录中创建额外的.py.sh文件,因为我已经有了build.py。在虚拟环境中安装包后运行测试是可以的,就像它是在readthedocs服务器上实现的一样

我还熟悉了taskipy,所以我的pyproject.toml中的一些bash命令也可以。欢迎使用pyproject.toml的任何其他软件包

也许有一些诗歌的钩子,因为它在创建.whl分发文件时会使诗歌变得通俗易懂

在此方面的任何帮助都将不胜感激

UPDTox看起来像是合适的工具,但它在目录中没有看到pyproject.toml。非常欢迎在软件包或教程中链接到带有tox和cython的repos


Tags: 文件pybuild目录setup二进制clib解决方案
2条回答

下面是对已接受答案(pytest示例)的一个轻微修改,当src代码和测试代码在不同的目录中分开时,它现在也可以工作

例如

.
|_ src/
    |_ your_package/
        |_ some_cython_module.pyx
|_ test/
    |_ conftest.py
    |_ some_cython_module_test.py
# conftest.py

from distutils.dist import Distribution
from pathlib import Path

from Cython.Build import cythonize


def pytest_sessionstart(session):
    difflib_source = Path('src/your_package/some_cython_module.pyx')
    dist = Distribution(attrs={'ext_modules': cythonize(str(difflib_source))})
    build_ext_cmd = dist.get_command_obj('build_ext')
    build_ext_cmd.ensure_finalized()
    build_ext_cmd.inplace = 1
    build_ext_cmd.run()
    session.fib_obj = Path(build_ext_cmd.get_ext_fullpath(
        str(difflib_source.parents[0]) + '/' + difflib_source.stem
    ))


def pytest_sessionfinish(session):
    session.fib_obj.unlink()

如果扩展是发行版的一部分,那么除了运行poetry install-poetry将在项目的可编辑安装中构建扩展之外,您不需要做任何事情

在其他情况下,您可以在测试中嵌入调用distutils命令,作为套件设置/拆卸的一部分。我对nose不是很熟悉,但这里有一个简单的例子。假设我有一个fib.pyx(这是Cython书中的一个例子):

def fib(long n):
    '''Returns the nth Fibonacci number.'''
    cdef long a=0, b=1, i
    for i in range(n):
        a, b = a + b, a
    return a

一个test_fib.py模块,用于构建fib库并在测试成功时将其删除:

from distutils.dist import Distribution
from distutils.core import Extension
from pathlib import Path
from Cython.Build import cythonize


fib_source = Path('fib.pyx')

# distutils magic. This is essentially the same as calling
# python setup.py build_ext  inplace
dist = Distribution(attrs={'ext_modules': cythonize(fib_source.name)})
build_ext_cmd = dist.get_command_obj('build_ext')
build_ext_cmd.ensure_finalized()
build_ext_cmd.inplace = 1
build_ext_cmd.run()

fib_obj = Path(build_ext_cmd.get_ext_fullpath(fib_source.stem))

# the lib was built, so the import will succeed now
from fib import fib


def teardown_module():
    # remove built library
    fib_obj.unlink()

    # if you also want to clean the build dir:
    from distutils.dir_util import remove_tree
    remove_tree(build_ext_cmd.build_lib)
    remove_tree(build_ext_cmd.build_temp)


# sample tests

def test_zero():
    assert fib(0) == 0


def test_ten():
    assert fib(10) == 55

您可能正在自定义build.py中自定义setup_kwargs。要重用此代码,请调整dist初始化,例如:

from build import build

setup_kwargs = {}
build(setup_kwargs)
dist = Distribution(attrs=setup_kwargs)
...

pytest示例

使用pytest可以更方便地组织事情。创建一个名为conftest.py的文件,并将安装/拆卸代码提取到挂钩:

# conftest.py

from distutils.core import Extension
from distutils.dist import Distribution
from distutils.dir_util import remove_tree
from pathlib import Path
from Cython.Build import cythonize


def pytest_sessionstart(session):
    fib_source = Path('fib.pyx')
    dist = Distribution(attrs={'ext_modules': cythonize(fib_source.name)})
    build_ext_cmd = dist.get_command_obj('build_ext')
    build_ext_cmd.ensure_finalized()
    build_ext_cmd.inplace = 1
    build_ext_cmd.run()
    session.fib_obj = Path(build_ext_cmd.get_ext_fullpath(fib_source.stem))


def pytest_sessionfinish(session):
    session.fib_obj.unlink()

现在,测试变得更加干净,整个测试会话只运行一次设置代码。上述测试示例,请重新查看:

from fib import fib


def test_zero():
    assert fib(0) == 0


def test_ten():
    assert fib(10) == 55

相关问题 更多 >