Call x.tobytes() once and assign it

Avoid repeated copies while testing that don't add value here.
dask · May 6, 2022 · 46c1b8f · 46c1b8f
1 parent 99a1179
commit 46c1b8f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/distributed/protocol/tests/test_protocol.py b/distributed/protocol/tests/test_protocol.py
@@ -48,10 +48,11 @@ def test_compression_1():
     pytest.importorskip("lz4")
     np = pytest.importorskip("numpy")
     x = np.ones(1000000)
-    frames = dumps({"x": Serialize(x.tobytes())})
-    assert sum(map(nbytes, frames)) < x.nbytes
+    b = x.tobytes()
+    frames = dumps({"x": Serialize(b)})
+    assert sum(map(nbytes, frames)) < nbytes(b)
     y = loads(frames)
-    assert {"x": x.tobytes()} == y
+    assert {"x": b} == y
 
 
 def test_compression_2():