mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-11 08:03:55 +00:00
1d1f81b278
This commit includes modifications that are needed to get pybind11 to work with PyPy. The full test suite compiles and runs except for a last few functions that are commented out (due to problems in PyPy that were reported on the PyPy bugtracker). Two somewhat intrusive changes were needed to make it possible: two new tags ``py::buffer_protocol()`` and ``py::metaclass()`` must now be specified to the ``class_`` constructor if the class uses the buffer protocol and/or requires a metaclass (e.g. for static properties). Note that this is only for the PyPy version based on Python 2.7 for now. When the PyPy 3.x has caught up in terms of cpyext compliance, a PyPy 3.x patch will follow.
63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
import pytest
|
|
from pybind11_tests import Matrix, ConstructorStats
|
|
|
|
with pytest.suppress(ImportError):
|
|
import numpy as np
|
|
|
|
|
|
@pytest.requires_numpy
|
|
def test_from_python():
|
|
with pytest.raises(RuntimeError) as excinfo:
|
|
Matrix(np.array([1, 2, 3])) # trying to assign a 1D array
|
|
assert str(excinfo.value) == "Incompatible buffer format!"
|
|
|
|
m3 = np.array([[1, 2, 3], [4, 5, 6]]).astype(np.float32)
|
|
m4 = Matrix(m3)
|
|
|
|
for i in range(m4.rows()):
|
|
for j in range(m4.cols()):
|
|
assert m3[i, j] == m4[i, j]
|
|
|
|
cstats = ConstructorStats.get(Matrix)
|
|
assert cstats.alive() == 1
|
|
del m3, m4
|
|
assert cstats.alive() == 0
|
|
assert cstats.values() == ["2x3 matrix"]
|
|
assert cstats.copy_constructions == 0
|
|
# assert cstats.move_constructions >= 0 # Don't invoke any
|
|
assert cstats.copy_assignments == 0
|
|
assert cstats.move_assignments == 0
|
|
|
|
|
|
# PyPy: Memory leak in the "np.array(m, copy=False)" call
|
|
# https://bitbucket.org/pypy/pypy/issues/2444
|
|
@pytest.unsupported_on_pypy
|
|
@pytest.requires_numpy
|
|
def test_to_python():
|
|
m = Matrix(5, 5)
|
|
|
|
assert m[2, 3] == 0
|
|
m[2, 3] = 4
|
|
assert m[2, 3] == 4
|
|
|
|
m2 = np.array(m, copy=False)
|
|
assert m2.shape == (5, 5)
|
|
assert abs(m2).sum() == 4
|
|
assert m2[2, 3] == 4
|
|
m2[2, 3] = 5
|
|
assert m2[2, 3] == 5
|
|
|
|
cstats = ConstructorStats.get(Matrix)
|
|
assert cstats.alive() == 1
|
|
del m
|
|
pytest.gc_collect()
|
|
assert cstats.alive() == 1
|
|
del m2 # holds an m reference
|
|
pytest.gc_collect()
|
|
assert cstats.alive() == 0
|
|
assert cstats.values() == ["5x5 matrix"]
|
|
assert cstats.copy_constructions == 0
|
|
# assert cstats.move_constructions >= 0 # Don't invoke any
|
|
assert cstats.copy_assignments == 0
|
|
assert cstats.move_assignments == 0
|