| Jeremy Hylton | be467e5 | 2000-09-15 15:14:51 +0000 | [diff] [blame] | 1 | import pickle |
| Fred Drake | 694ed09 | 2001-12-19 16:42:15 +0000 | [diff] [blame] | 2 | import unittest |
| Jeremy Hylton | 6642653 | 2001-10-15 21:38:56 +0000 | [diff] [blame] | 3 | from cStringIO import StringIO |
| Barry Warsaw | 408b6d3 | 2002-07-30 23:27:12 +0000 | [diff] [blame] | 4 | from test.pickletester import AbstractPickleTests, AbstractPickleModuleTests |
| Barry Warsaw | 04f357c | 2002-07-23 19:04:11 +0000 | [diff] [blame] | 5 | from test import test_support |
| Jeremy Hylton | 6642653 | 2001-10-15 21:38:56 +0000 | [diff] [blame] | 6 | |
| 7 | class PickleTests(AbstractPickleTests, AbstractPickleModuleTests): |
| 8 | |
| 9 | def setUp(self): |
| 10 | self.dumps = pickle.dumps |
| 11 | self.loads = pickle.loads |
| Tim Peters | e0c446b | 2001-10-18 21:57:37 +0000 | [diff] [blame] | 12 | |
| Jeremy Hylton | 6642653 | 2001-10-15 21:38:56 +0000 | [diff] [blame] | 13 | module = pickle |
| 14 | error = KeyError |
| 15 | |
| 16 | class PicklerTests(AbstractPickleTests): |
| 17 | |
| 18 | error = KeyError |
| 19 | |
| 20 | def dumps(self, arg, bin=0): |
| 21 | f = StringIO() |
| 22 | p = pickle.Pickler(f, bin) |
| 23 | p.dump(arg) |
| 24 | f.seek(0) |
| 25 | return f.read() |
| 26 | |
| 27 | def loads(self, buf): |
| 28 | f = StringIO(buf) |
| 29 | u = pickle.Unpickler(f) |
| 30 | return u.load() |
| 31 | |
| Fred Drake | 694ed09 | 2001-12-19 16:42:15 +0000 | [diff] [blame] | 32 | def test_main(): |
| 33 | loader = unittest.TestLoader() |
| 34 | suite = unittest.TestSuite() |
| 35 | suite.addTest(loader.loadTestsFromTestCase(PickleTests)) |
| 36 | suite.addTest(loader.loadTestsFromTestCase(PicklerTests)) |
| 37 | test_support.run_suite(suite) |
| 38 | |
| Jeremy Hylton | 6642653 | 2001-10-15 21:38:56 +0000 | [diff] [blame] | 39 | if __name__ == "__main__": |
| Fred Drake | 694ed09 | 2001-12-19 16:42:15 +0000 | [diff] [blame] | 40 | test_main() |