22 lines
599 B
Python
22 lines
599 B
Python
import unittest
|
|
|
|
from ipython_shell.transport import KernelMessage
|
|
|
|
|
|
class TransportTests(unittest.TestCase):
|
|
def test_kernel_message_keeps_content_and_buffers(self):
|
|
message = KernelMessage(
|
|
msg_type="display_data",
|
|
parent_id="call-1",
|
|
content={"data": {"text/plain": "42"}},
|
|
buffers=[b"binary"],
|
|
)
|
|
|
|
self.assertEqual(message.parent_id, "call-1")
|
|
self.assertEqual(message.content["data"], {"text/plain": "42"})
|
|
self.assertEqual(message.buffers, [b"binary"])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|