mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-25 06:35:12 +00:00
1be0a0a610
For single-file extensions, a convenient pattern offered by cython
is to place the source files directly in the python source tree
(`foo/__init__.py`, `foo/ext.pyx`), deriving the package names from
their filesystem location. Adapt this pattern for pybind11, using an
`intree_extensions` helper, which should be thought of as the moral
equivalent to `cythonize`.
Differences with cythonize: I chose not to include globbing support
(`intree_extensions(glob.glob("**/*.cpp"))` seems sufficient), nor to
provide extension-customization kwargs (directly setting the attributes
on the resulting Pybind11Extension objects seems sufficient).
We could choose to have `intree_extension` (singular instead) and make
users write `[*map(intree_extension, glob.glob("**/*.cpp"))]`; no strong
opinion here.
Co-authored-by: Aaron Gokaslan <skylion.aaron@gmail.com>
144 lines
3.8 KiB
Python
144 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
from textwrap import dedent
|
|
|
|
import pytest
|
|
|
|
DIR = os.path.abspath(os.path.dirname(__file__))
|
|
MAIN_DIR = os.path.dirname(os.path.dirname(DIR))
|
|
|
|
|
|
@pytest.mark.parametrize("parallel", [False, True])
|
|
@pytest.mark.parametrize("std", [11, 0])
|
|
def test_simple_setup_py(monkeypatch, tmpdir, parallel, std):
|
|
monkeypatch.chdir(tmpdir)
|
|
monkeypatch.syspath_prepend(MAIN_DIR)
|
|
|
|
(tmpdir / "setup.py").write_text(
|
|
dedent(
|
|
u"""\
|
|
import sys
|
|
sys.path.append({MAIN_DIR!r})
|
|
|
|
from setuptools import setup, Extension
|
|
from pybind11.setup_helpers import build_ext, Pybind11Extension
|
|
|
|
std = {std}
|
|
|
|
ext_modules = [
|
|
Pybind11Extension(
|
|
"simple_setup",
|
|
sorted(["main.cpp"]),
|
|
cxx_std=std,
|
|
),
|
|
]
|
|
|
|
cmdclass = dict()
|
|
if std == 0:
|
|
cmdclass["build_ext"] = build_ext
|
|
|
|
|
|
parallel = {parallel}
|
|
if parallel:
|
|
from pybind11.setup_helpers import ParallelCompile
|
|
ParallelCompile().install()
|
|
|
|
setup(
|
|
name="simple_setup_package",
|
|
cmdclass=cmdclass,
|
|
ext_modules=ext_modules,
|
|
)
|
|
"""
|
|
).format(MAIN_DIR=MAIN_DIR, std=std, parallel=parallel),
|
|
encoding="ascii",
|
|
)
|
|
|
|
(tmpdir / "main.cpp").write_text(
|
|
dedent(
|
|
u"""\
|
|
#include <pybind11/pybind11.h>
|
|
|
|
int f(int x) {
|
|
return x * 3;
|
|
}
|
|
PYBIND11_MODULE(simple_setup, m) {
|
|
m.def("f", &f);
|
|
}
|
|
"""
|
|
),
|
|
encoding="ascii",
|
|
)
|
|
|
|
subprocess.check_call(
|
|
[sys.executable, "setup.py", "build_ext", "--inplace"],
|
|
stdout=sys.stdout,
|
|
stderr=sys.stderr,
|
|
)
|
|
|
|
# Debug helper printout, normally hidden
|
|
for item in tmpdir.listdir():
|
|
print(item.basename)
|
|
|
|
assert (
|
|
len([f for f in tmpdir.listdir() if f.basename.startswith("simple_setup")]) == 1
|
|
)
|
|
assert len(list(tmpdir.listdir())) == 4 # two files + output + build_dir
|
|
|
|
(tmpdir / "test.py").write_text(
|
|
dedent(
|
|
u"""\
|
|
import simple_setup
|
|
assert simple_setup.f(3) == 9
|
|
"""
|
|
),
|
|
encoding="ascii",
|
|
)
|
|
|
|
subprocess.check_call(
|
|
[sys.executable, "test.py"], stdout=sys.stdout, stderr=sys.stderr
|
|
)
|
|
|
|
|
|
def test_intree_extensions(monkeypatch, tmpdir):
|
|
monkeypatch.syspath_prepend(MAIN_DIR)
|
|
|
|
from pybind11.setup_helpers import intree_extensions
|
|
|
|
monkeypatch.chdir(tmpdir)
|
|
root = tmpdir
|
|
root.ensure_dir()
|
|
subdir = root / "dir"
|
|
subdir.ensure_dir()
|
|
src = subdir / "ext.cpp"
|
|
src.ensure()
|
|
(ext,) = intree_extensions([src.relto(tmpdir)])
|
|
assert ext.name == "ext"
|
|
subdir.ensure("__init__.py")
|
|
(ext,) = intree_extensions([src.relto(tmpdir)])
|
|
assert ext.name == "dir.ext"
|
|
|
|
|
|
def test_intree_extensions_package_dir(monkeypatch, tmpdir):
|
|
monkeypatch.syspath_prepend(MAIN_DIR)
|
|
|
|
from pybind11.setup_helpers import intree_extensions
|
|
|
|
monkeypatch.chdir(tmpdir)
|
|
root = tmpdir / "src"
|
|
root.ensure_dir()
|
|
subdir = root / "dir"
|
|
subdir.ensure_dir()
|
|
src = subdir / "ext.cpp"
|
|
src.ensure()
|
|
(ext,) = intree_extensions([src.relto(tmpdir)], package_dir={"": "src"})
|
|
assert ext.name == "dir.ext"
|
|
(ext,) = intree_extensions([src.relto(tmpdir)], package_dir={"foo": "src"})
|
|
assert ext.name == "foo.dir.ext"
|
|
subdir.ensure("__init__.py")
|
|
(ext,) = intree_extensions([src.relto(tmpdir)], package_dir={"": "src"})
|
|
assert ext.name == "dir.ext"
|
|
(ext,) = intree_extensions([src.relto(tmpdir)], package_dir={"foo": "src"})
|
|
assert ext.name == "foo.dir.ext"
|