test_rpc.py

"Test rpc, coverage 20%."

fromidlelibimportrpc
importunittest



classCodePicklerTest(unittest.TestCase):

deftest_pickle_unpickle(self):
deff(): returna+b+c
func, (cbytes,) =rpc.pickle_code(f.__code__)
self.assertIs(func, rpc.unpickle_code)
self.assertIn(b'test_rpc.py', cbytes)
code=rpc.unpickle_code(cbytes)
self.assertEqual(code.co_names, ('a', 'b', 'c'))

deftest_code_pickler(self):
self.assertIn(type((lambda:None).__code__),
rpc.CodePickler.dispatch_table)

deftest_dumps(self):
deff(): pass
# The main test here is that pickling code does not raise.
self.assertIn(b'test_rpc.py', rpc.dumps(f.__code__))


if__name__=='__main__':
unittest.main(verbosity=2)