Zengtudor
/
pybind11
mirror of https://github.com/pybind/pybind11.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

pytest-compatible access to np.float128, np.complex256

This commit is contained in:
Ralf W. Grosse-Kunstleve 2023-05-19 00:17:30 -07:00
parent d9e3bd3248
commit e9a289c50f
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/test_buffers.py
View File

@ -10,12 +10,13 @@ from pybind11_tests import buffers as m
np = pytest.importorskip("numpy") np = pytest.importorskip("numpy")
if m.std_is_same_double_long_double: # Windows.
np_float128 = None def np_dtype_long_double_or_none(name):
np_complex256 = None # Intentionally not using getattr(np, name, None), to be strict.
else: if m.std_is_same_double_long_double: # Windows.
np_float128 = np.float128 return None
np_complex256 = np.complex256 return getattr(np, name)
CPP_NAME_FORMAT_NP_DTYPE_TABLE = [ CPP_NAME_FORMAT_NP_DTYPE_TABLE = [
item item
@ -32,10 +33,10 @@ CPP_NAME_FORMAT_NP_DTYPE_TABLE = [
("std::uint64_t", "Q", np.uint64), ("std::uint64_t", "Q", np.uint64),
("float", "f", np.float32), ("float", "f", np.float32),
("double", "d", np.float64), ("double", "d", np.float64),
("long double", "g", np_float128), ("long double", "g", np_dtype_long_double_or_none("float128")),
("std::complex<float>", "Zf", np.complex64), ("std::complex<float>", "Zf", np.complex64),
("std::complex<double>", "Zd", np.complex128), ("std::complex<double>", "Zd", np.complex128),
("std::complex<long double>", "Zg", np_complex256), ("std::complex<long double>", "Zg", np_dtype_long_double_or_none("complex256")),
] ]
if item[-1] is not None if item[-1] is not None
] ]