diff options
Diffstat (limited to 'Lib')
| -rw-r--r-- | Lib/test/test_buffer.py | 15 | ||||
| -rw-r--r-- | Lib/test/test_sys.py | 5 |
2 files changed, 18 insertions, 2 deletions
diff --git a/Lib/test/test_buffer.py b/Lib/test/test_buffer.py index 0976fa9..d921a5a 100644 --- a/Lib/test/test_buffer.py +++ b/Lib/test/test_buffer.py @@ -2462,6 +2462,21 @@ class TestBufferProtocol(unittest.TestCase): self.assertEqual(m.tobytes(), b'') self.assertEqual(m.tolist(), []) + check_sizeof = support.check_sizeof + + def test_memoryview_sizeof(self): + check = self.check_sizeof + vsize = support.calcvobjsize + base_struct = 'Pnin 2P2n2i5P P' + per_dim = '3n' + + items = list(range(8)) + check(memoryview(b''), vsize(base_struct + 1 * per_dim)) + a = ndarray(items, shape=[2, 4], format="b") + check(memoryview(a), vsize(base_struct + 2 * per_dim)) + a = ndarray(items, shape=[2, 2, 2], format="b") + check(memoryview(a), vsize(base_struct + 3 * per_dim)) + def test_memoryview_struct_module(self): class INT(object): diff --git a/Lib/test/test_sys.py b/Lib/test/test_sys.py index ed6b47d..dc241a6 100644 --- a/Lib/test/test_sys.py +++ b/Lib/test/test_sys.py @@ -846,6 +846,9 @@ class SizeofTest(unittest.TestCase): check(x, vsize('n2Pi') + x.__alloc__()) # bytearray_iterator check(iter(bytearray()), size('nP')) + # bytes + check(b'', vsize('n') + 1) + check(b'x' * 10, vsize('n') + 11) # cell def get_cell(): x = 42 @@ -965,8 +968,6 @@ class SizeofTest(unittest.TestCase): check(int(PyLong_BASE), vsize('') + 2*self.longdigit) check(int(PyLong_BASE**2-1), vsize('') + 2*self.longdigit) check(int(PyLong_BASE**2), vsize('') + 3*self.longdigit) - # memoryview - check(memoryview(b''), size('Pnin 2P2n2i5P Pn')) # module check(unittest, size('PnPPP')) # None |