def test_on_msg(self): def raise_error(msg): if msg == 'Error': raise Exception() comm = JupyterComm(id='Test', on_msg=raise_error) with self.assertRaises(Exception): comm._handle_msg({'content': {'data': 'Error'}})
def test_decode(self): msg = {'content': {'data': 'Test'}} decoded = JupyterComm.decode(msg) self.assertEqual(decoded, 'Test')
def test_init_comm_id(self): comm = JupyterComm(id='Test') self.assertEqual(comm.id, 'Test')
def test_init_comm(self): JupyterComm()