B
    3\                 @   sH   d Z ddlZddlmZ ddlmZ ddlmZm	Z	 dd Z
dd	 ZdS )
z0Test serialize/deserialize messages with buffers    N)Session   )serialize_binary_messagedeserialize_binary_messagec              C   sF   t  } | jdddid}dd tdD |d< t|}t|t d S )	Ndata_pubab)contentc             S   s   g | ]}t td qS )   )
memoryviewosurandom).0i r   <lib/python3.7/site-packages/notebook/tests/test_serialize.py
<listcomp>   s    z)test_serialize_binary.<locals>.<listcomp>r
   buffers)r   msgranger   ntZassert_is_instancebytes)sr   bmsgr   r   r   test_serialize_binary   s
    r   c              C   sN   t  } | jdddid}dd tdD |d< t|}t|}t|| d S )	Nr   r   r   )r	   c             S   s   g | ]}t td qS )r   )r   r   r   )r   r   r   r   r   r      s    z+test_deserialize_binary.<locals>.<listcomp>r
   r   )r   r   r   r   r   r   Zassert_equal)r   r   r   Zmsg2r   r   r   test_deserialize_binary   s    r   )__doc__r   Z
nose.toolsZtoolsr   Zjupyter_client.sessionr   Zbase.zmqhandlersr   r   r   r   r   r   r   r   <module>   s   