msgpack-python/test/test_memoryview.py at main · DiamondLightSource/msgpack-python · GitHub

History

111 lines (75 loc) · 3.07 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

#!/usr/bin/env python

from array import array

from msgpack import packb, unpackb

def make_array(f, data):

a = array(f)

a.frombytes(data)

return a

def _runtest(format, nbytes, expected_header, expected_prefix, use_bin_type):

# create a new array

original_array = array(format)

original_array.fromlist([255] * (nbytes // original_array.itemsize))

original_data = original_array.tobytes()

view = memoryview(original_array)

# pack, unpack, and reconstruct array

packed = packb(view, use_bin_type=use_bin_type)

unpacked = unpackb(packed, raw=(not use_bin_type))

reconstructed_array = make_array(format, unpacked)

# check that we got the right amount of data

assert len(original_data) == nbytes

# check packed header

assert packed[:1] == expected_header

# check packed length prefix, if any

assert packed[1 : 1 + len(expected_prefix)] == expected_prefix

# check packed data

assert packed[1 + len(expected_prefix) :] == original_data

# check array unpacked correctly

assert original_array == reconstructed_array

def test_fixstr_from_byte():

_runtest("B", 1, b"\xa1", b"", False)

_runtest("B", 31, b"\xbf", b"", False)

def test_fixstr_from_float():

_runtest("f", 4, b"\xa4", b"", False)

_runtest("f", 28, b"\xbc", b"", False)

def test_str16_from_byte():

_runtest("B", 2**8, b"\xda", b"\x01\x00", False)

_runtest("B", 2**16 - 1, b"\xda", b"\xff\xff", False)

def test_str16_from_float():

_runtest("f", 2**8, b"\xda", b"\x01\x00", False)

_runtest("f", 2**16 - 4, b"\xda", b"\xff\xfc", False)

def test_str32_from_byte():

_runtest("B", 2**16, b"\xdb", b"\x00\x01\x00\x00", False)

def test_str32_from_float():

_runtest("f", 2**16, b"\xdb", b"\x00\x01\x00\x00", False)

def test_bin8_from_byte():

_runtest("B", 1, b"\xc4", b"\x01", True)

_runtest("B", 2**8 - 1, b"\xc4", b"\xff", True)

def test_bin8_from_float():

_runtest("f", 4, b"\xc4", b"\x04", True)

_runtest("f", 2**8 - 4, b"\xc4", b"\xfc", True)

def test_bin16_from_byte():

_runtest("B", 2**8, b"\xc5", b"\x01\x00", True)

_runtest("B", 2**16 - 1, b"\xc5", b"\xff\xff", True)

def test_bin16_from_float():

_runtest("f", 2**8, b"\xc5", b"\x01\x00", True)

_runtest("f", 2**16 - 4, b"\xc5", b"\xff\xfc", True)

def test_bin32_from_byte():

_runtest("B", 2**16, b"\xc6", b"\x00\x01\x00\x00", True)

def test_bin32_from_float():

_runtest("f", 2**16, b"\xc6", b"\x00\x01\x00\x00", True)

def test_multidim_memoryview():

# See https://github.com/msgpack/msgpack-python/issues/526

view = memoryview(b"\00" * 6)

data = view.cast(view.format, (3, 2))

packed = packb(data)

assert packed == b"\xc4\x06\x00\x00\x00\x00\x00\x00"

def test_unpack_noncontiguous_memoryview():

# Use a multi-byte value so the padded stride-2 view is non-contiguous.

packed = packb(2**32)

padded = bytearray()

for byte in packed:

padded.append(byte)

padded.append(0)

noncont = memoryview(bytes(padded))[::2]

assert not noncont.c_contiguous

assert unpackb(noncont) == 2**32

msgpack-python/test/test_memoryview.py at main · DiamondLightSource/msgpack-python · GitHub

Uh oh!

Expand file tree

Breadcrumbs

test_memoryview.py

Latest commit

History

Breadcrumbs

test_memoryview.py

File metadata and controls