Fix-up merge_frames test with writeable header

dask · Jul 31, 2020 · e3d3d8c · e3d3d8c
1 parent ce88f34
commit e3d3d8c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/distributed/protocol/tests/test_protocol_utils.py b/distributed/protocol/tests/test_protocol_utils.py
@@ -15,7 +15,10 @@
     ],
 )
 def test_merge_frames(lengths, frames):
-    header = {"lengths": lengths}
+    header = {
+        "lengths": lengths,
+        "writeable": len(lengths) * (False,),
+    }
     result = merge_frames(header, frames)
 
     data = b"".join(frames)
@@ -25,7 +28,7 @@ def test_merge_frames(lengths, frames):
         data = data[i:]
 
     assert all(isinstance(f, memoryview) for f in result)
-    assert all(not f.readonly for f in result)
+    assert tuple(not f.readonly for f in result) == header["writeable"]
     assert list(map(ensure_bytes, result)) == expected