Skip to content

Commit b6bec32

Browse files
committed
Handle pickle protocol 5 separately
1 parent 0ae7863 commit b6bec32

1 file changed

Lines changed: 12 additions & 1 deletion

File tree

distributed/protocol/tests/test_pickle.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,11 +75,22 @@ def test_pickle_out_of_band():
7575
def test_pickle_empty():
7676
x = MemoryviewHolder(bytearray()) # Empty view
7777
header, frames = serialize(x, serializers=("pickle",))
78+
7879
assert header["serializer"] == "pickle"
7980
assert len(frames) >= 1
8081
assert isinstance(frames[0], bytes)
81-
header["writeable"] = (False,) * len(frames)
82+
83+
if HIGHEST_PROTOCOL >= 5:
84+
assert len(frames) == 2
85+
assert header["writeable"]
86+
87+
header["writeable"] = (False,) * len(frames)
88+
else:
89+
assert len(frames) == 1
90+
assert not header["writeable"]
91+
8292
y = deserialize(header, frames)
93+
8394
assert isinstance(y, MemoryviewHolder)
8495
assert isinstance(y.mv, memoryview)
8596
assert y.mv == x.mv

0 commit comments

Comments
 (0)