Simplify ensure_memoryview test with array (#6322)

dask · May 11, 2022 · 9f02e7a · 9f02e7a
1 parent 1937be7
commit 9f02e7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/distributed/tests/test_utils.py b/distributed/tests/test_utils.py
@@ -1,4 +1,3 @@
-import array
 import asyncio
 import contextvars
 import functools
@@ -8,6 +7,7 @@
 import socket
 import traceback
 import warnings
+from array import array
 from collections import deque
 from time import sleep
 
@@ -255,7 +255,7 @@ def test_ensure_memoryview_empty():
 
 
 def test_ensure_memoryview():
-    data = [b"1", memoryview(b"1"), bytearray(b"1"), array.array("b", [49])]
+    data = [b"1", memoryview(b"1"), bytearray(b"1"), array("B", b"1")]
     for d in data:
         result = ensure_memoryview(d)
         assert isinstance(result, memoryview)