FilesExpand file tree

 readme-python3

/

test_buffer.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

49 lines (37 loc) · 1.32 KB

 readme-python3

/

test_buffer.py

Copy path

Top

File metadata and controls

• Code
• Blame

49 lines (37 loc) · 1.32 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

from pytest import raises

from msgpack import Packer, packb, unpackb

def test_unpack_buffer():

from array import array

buf = array("b")

buf.frombytes(packb((b"foo", b"bar")))

obj = unpackb(buf, use_list=1)

assert [b"foo", b"bar"] == obj

def test_unpack_bytearray():

buf = bytearray(packb((b"foo", b"bar")))

obj = unpackb(buf, use_list=1)

assert [b"foo", b"bar"] == obj

expected_type = bytes

assert all(type(s) is expected_type for s in obj)

def test_unpack_memoryview():

buf = bytearray(packb((b"foo", b"bar")))

view = memoryview(buf)

obj = unpackb(view, use_list=1)

assert [b"foo", b"bar"] == obj

expected_type = bytes

assert all(type(s) is expected_type for s in obj)

def test_packer_getbuffer():

packer = Packer(autoreset=False)

packer.pack_array_header(2)

packer.pack(42)

packer.pack("hello")

buffer = packer.getbuffer()

assert isinstance(buffer, memoryview)

assert bytes(buffer) == b"\x92*\xa5hello"

if Packer.__module__ == "msgpack._cmsgpack": # only for Cython

# cython Packer supports buffer protocol directly

assert bytes(packer) == b"\x92*\xa5hello"

with raises(BufferError):

packer.pack(42)

buffer.release()

packer.pack(42)

assert bytes(packer) == b"\x92*\xa5hello*"