81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
import pytest
|
|
|
|
import numpy as np
|
|
from numpy.testing import assert_raises_regex
|
|
import numpy.lib._utils_impl as _utils_impl
|
|
|
|
from io import StringIO
|
|
|
|
|
|
def test_assert_raises_regex_context_manager():
|
|
with assert_raises_regex(ValueError, 'no deprecation warning'):
|
|
raise ValueError('no deprecation warning')
|
|
|
|
|
|
def test_info_method_heading():
|
|
# info(class) should only print "Methods:" heading if methods exist
|
|
|
|
class NoPublicMethods:
|
|
pass
|
|
|
|
class WithPublicMethods:
|
|
def first_method():
|
|
pass
|
|
|
|
def _has_method_heading(cls):
|
|
out = StringIO()
|
|
np.info(cls, output=out)
|
|
return 'Methods:' in out.getvalue()
|
|
|
|
assert _has_method_heading(WithPublicMethods)
|
|
assert not _has_method_heading(NoPublicMethods)
|
|
|
|
|
|
def test_drop_metadata():
|
|
def _compare_dtypes(dt1, dt2):
|
|
return np.can_cast(dt1, dt2, casting='no')
|
|
|
|
# structured dtype
|
|
dt = np.dtype([('l1', [('l2', np.dtype('S8', metadata={'msg': 'toto'}))])],
|
|
metadata={'msg': 'titi'})
|
|
dt_m = _utils_impl.drop_metadata(dt)
|
|
assert _compare_dtypes(dt, dt_m) is True
|
|
assert dt_m.metadata is None
|
|
assert dt_m['l1'].metadata is None
|
|
assert dt_m['l1']['l2'].metadata is None
|
|
|
|
# alignment
|
|
dt = np.dtype([('x', '<f8'), ('y', '<i4')],
|
|
align=True,
|
|
metadata={'msg': 'toto'})
|
|
dt_m = _utils_impl.drop_metadata(dt)
|
|
assert _compare_dtypes(dt, dt_m) is True
|
|
assert dt_m.metadata is None
|
|
|
|
# subdtype
|
|
dt = np.dtype('8f',
|
|
metadata={'msg': 'toto'})
|
|
dt_m = _utils_impl.drop_metadata(dt)
|
|
assert _compare_dtypes(dt, dt_m) is True
|
|
assert dt_m.metadata is None
|
|
|
|
# scalar
|
|
dt = np.dtype('uint32',
|
|
metadata={'msg': 'toto'})
|
|
dt_m = _utils_impl.drop_metadata(dt)
|
|
assert _compare_dtypes(dt, dt_m) is True
|
|
assert dt_m.metadata is None
|
|
|
|
|
|
@pytest.mark.parametrize("dtype",
|
|
[np.dtype("i,i,i,i")[["f1", "f3"]],
|
|
np.dtype("f8"),
|
|
np.dtype("10i")])
|
|
def test_drop_metadata_identity_and_copy(dtype):
|
|
# If there is no metadata, the identity is preserved:
|
|
assert _utils_impl.drop_metadata(dtype) is dtype
|
|
|
|
# If there is any, it is dropped (subforms are checked above)
|
|
dtype = np.dtype(dtype, metadata={1: 2})
|
|
assert _utils_impl.drop_metadata(dtype).metadata is None
|