diff options
author | Eric Wieser <wieser.eric@gmail.com> | 2020-02-08 21:22:03 +0000 |
---|---|---|
committer | Eric Wieser <wieser.eric@gmail.com> | 2020-02-08 21:23:35 +0000 |
commit | 48c0b1426ff81c8da3832fd6fb162dfbc1aa4601 (patch) | |
tree | 1546e45a8b6ca7f6cff7fca42ab7bd26faebad02 /numpy | |
parent | 2d1e8f38e973f88aeb29dc51caa0f57ab86efc67 (diff) | |
download | numpy-48c0b1426ff81c8da3832fd6fb162dfbc1aa4601.tar.gz |
MAINT,TST: Tidy test_datetime_memoryview a little
Diffstat (limited to 'numpy')
-rw-r--r-- | numpy/core/tests/test_scalarbuffer.py | 17 |
1 files changed, 8 insertions, 9 deletions
diff --git a/numpy/core/tests/test_scalarbuffer.py b/numpy/core/tests/test_scalarbuffer.py index b8c6dd4aa..15226d10b 100644 --- a/numpy/core/tests/test_scalarbuffer.py +++ b/numpy/core/tests/test_scalarbuffer.py @@ -76,24 +76,23 @@ class TestScalarPEP3118: assert_equal(mv_x.itemsize, mv_a.itemsize) assert_equal(mv_x.format, mv_a.format) + def _as_dict(self, m): + return dict(strides=m.strides, shape=m.shape, itemsize=m.itemsize, + ndim=m.ndim, format=m.format) + def test_datetime_memoryview(self): # gh-11656 # Values verified with v1.13.3, shape is not () as in test_scalar_dim - def as_dict(m): - return dict(strides=m.strides, shape=m.shape, itemsize=m.itemsize, - ndim=m.ndim, format=m.format) dt1 = np.datetime64('2016-01-01') dt2 = np.datetime64('2017-01-01') - expected = {'strides': (1,), 'itemsize': 1, 'ndim': 1, - 'shape': (8,), 'format': 'B'} + expected = dict(strides=(1,), itemsize=1, ndim=1, shape=(8,), + format='B') v = memoryview(dt1) - res = as_dict(v) - assert_equal(res, expected) + assert self._as_dict(v) == expected v = memoryview(dt2 - dt1) - res = as_dict(v) - assert_equal(res, expected) + assert self._as_dict(v) == expected dt = np.dtype([('a', 'uint16'), ('b', 'M8[s]')]) a = np.empty(1, dt) |