31 lines
820 B
Python
31 lines
820 B
Python
"Test rpc, coverage 20%."
|
|
|
|
from idlelib import rpc
|
|
import unittest
|
|
|
|
import marshal
|
|
|
|
|
|
class CodePicklerTest(unittest.TestCase):
|
|
|
|
def test_pickle_unpickle(self):
|
|
def f(): return a + 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'))
|
|
|
|
def test_code_pickler(self):
|
|
self.assertIn(type((lambda:None).__code__),
|
|
rpc.CodePickler.dispatch_table)
|
|
|
|
def test_dumps(self):
|
|
def f(): 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)
|