mit neuen venv und exe-Files
This commit is contained in:
80
venv3_12/Lib/site-packages/numpy/lib/tests/test_utils.py
Normal file
80
venv3_12/Lib/site-packages/numpy/lib/tests/test_utils.py
Normal file
@@ -0,0 +1,80 @@
|
||||
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
|
||||
Reference in New Issue
Block a user