mirror of
https://github.com/aykhans/AzSuicideDataVisualization.git
synced 2025-04-21 18:23:35 +00:00
25 lines
723 B
Python
25 lines
723 B
Python
"""Test serialize/deserialize messages with buffers"""
|
|
|
|
import os
|
|
|
|
from jupyter_client.session import Session
|
|
from ..base.zmqhandlers import (
|
|
serialize_binary_message,
|
|
deserialize_binary_message,
|
|
)
|
|
|
|
def test_serialize_binary():
|
|
s = Session()
|
|
msg = s.msg('data_pub', content={'a': 'b'})
|
|
msg['buffers'] = [ memoryview(os.urandom(3)) for i in range(3) ]
|
|
bmsg = serialize_binary_message(msg)
|
|
assert isinstance(bmsg, bytes)
|
|
|
|
def test_deserialize_binary():
|
|
s = Session()
|
|
msg = s.msg('data_pub', content={'a': 'b'})
|
|
msg['buffers'] = [ memoryview(os.urandom(2)) for i in range(3) ]
|
|
bmsg = serialize_binary_message(msg)
|
|
msg2 = deserialize_binary_message(bmsg)
|
|
assert msg2 == msg
|