Zengtudor/pybind11
1
0
Fork 0
mirror of https://github.com/pybind/pybind11.git synced 2024-11-11 08:03:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
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

15
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")
def np_dtype_long_double_or_none(name):
# Intentionally not using getattr(np, name, None), to be strict.
if m.std_is_same_double_long_double: # Windows. if m.std_is_same_double_long_double: # Windows.
np_float128 = None return None
np_complex256 = None return getattr(np, name)
else:
np_float128 = np.float128
np_complex256 = np.complex256
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
] ]