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

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2021-12-10 08:16:36 +00:00
parent 00a85a6969
commit 4be007670c
1 changed files with 22 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests/test_numpy_scalars.py
View File

@ -7,7 +7,8 @@ from pybind11_tests import numpy_scalars as m
np = pytest.importorskip("numpy") np = pytest.importorskip("numpy")
SCALAR_TYPES = dict([ SCALAR_TYPES = dict(
[
(np.bool_, False), (np.bool_, False),
(np.int8, -7), (np.int8, -7),
(np.int16, -15), (np.int16, -15),
@ -21,33 +22,34 @@ SCALAR_TYPES = dict([
(np.double, 1.25), (np.double, 1.25),
(np.complex64, 1 - 0.125j), (np.complex64, 1 - 0.125j),
(np.complex128, 1 - 0.25j), (np.complex128, 1 - 0.25j),
]) ]
)
ALL_TYPES = [int, bool, float, bytes, str] + list(SCALAR_TYPES) ALL_TYPES = [int, bool, float, bytes, str] + list(SCALAR_TYPES)
def type_name(tp): def type_name(tp):
try: try:
return tp.__name__.rstrip('_') return tp.__name__.rstrip("_")
except BaseException: except BaseException:
# no numpy # no numpy
return str(tp) return str(tp)
@pytest.fixture(scope='module', params=list(SCALAR_TYPES), ids=type_name) @pytest.fixture(scope="module", params=list(SCALAR_TYPES), ids=type_name)
def scalar_type(request): def scalar_type(request):
return request.param return request.param
def expected_signature(tp): def expected_signature(tp):
s = 'str' if sys.version_info[0] >= 3 else 'unicode' s = "str" if sys.version_info[0] >= 3 else "unicode"
t = type_name(tp) t = type_name(tp)
return 'test_{t}(x: {t}) -> Tuple[{s}, {t}]\n'.format(s=s, t=t) return "test_{t}(x: {t}) -> Tuple[{s}, {t}]\n".format(s=s, t=t)
def test_numpy_scalars(scalar_type): def test_numpy_scalars(scalar_type):
expected = SCALAR_TYPES[scalar_type] expected = SCALAR_TYPES[scalar_type]
name = type_name(scalar_type) name = type_name(scalar_type)
func = getattr(m, 'test_' + name) func = getattr(m, "test_" + name)
assert func.__doc__ == expected_signature(scalar_type) assert func.__doc__ == expected_signature(scalar_type)
for tp in ALL_TYPES: for tp in ALL_TYPES:
value = tp(1) value = tp(1)