Skip to content

Commit

Permalink
Handle pickle protocol 5 separately
Browse files Browse the repository at this point in the history
  • Loading branch information
jakirkham committed Sep 9, 2021
1 parent 0ae7863 commit 881d643
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion distributed/protocol/tests/test_pickle.py
Expand Up @@ -75,11 +75,22 @@ def test_pickle_out_of_band():
def test_pickle_empty():
x = MemoryviewHolder(bytearray()) # Empty view
header, frames = serialize(x, serializers=("pickle",))

assert header["serializer"] == "pickle"
assert len(frames) >= 1
assert isinstance(frames[0], bytes)
header["writeable"] = (False,) * len(frames)

if HIGHEST_PROTOCOL >= 5:
assert len(frames) == 2
assert len(header["writeable"]) == 1

header["writeable"] = (False,) * len(frames)
else:
assert len(frames) == 1
assert len(header["writeable"]) == 0

y = deserialize(header, frames)

assert isinstance(y, MemoryviewHolder)
assert isinstance(y.mv, memoryview)
assert y.mv == x.mv
Expand Down

0 comments on commit 881d643

Please sign in to comment.