fix test

jupyter · Dec 23, 2023 · 2bca5d2 · 2bca5d2
1 parent 259ec3d
commit 2bca5d2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_zmqstream.py b/tests/test_zmqstream.py
@@ -36,8 +36,8 @@ def on_recv1(msg):
 
     stream1.on_recv(on_recv1)
     stream2.send(b"ping")
-    msg = await future1
-    assert msg == b"ping"
+    msg_list = await future1
+    assert msg_list == [b"ping"]
     future2 = Future()
     future3 = Future()
 
@@ -52,8 +52,8 @@ def on_recv2(msg):
     stream1.send(b"pong")
     await future2
 
-    msg = await future3
-    assert msg == b"pong"
+    msg_list = await future3
+    assert msg_list == [b"pong"]
     stream1.close()
     stream2.close()
     context.term()