| Jeremy Hylton | be467e5 | 2000-09-15 15:14:51 +0000 | [diff] [blame] | 1 | import pickle |
| Jeremy Hylton | 6642653 | 2001-10-15 21:38:56 +0000 | [diff] [blame] | 2 | from cStringIO import StringIO |
| 3 | from pickletester import AbstractPickleTests, AbstractPickleModuleTests |
| 4 | from test_support import run_unittest |
| 5 | |
| 6 | class PickleTests(AbstractPickleTests, AbstractPickleModuleTests): |
| 7 | |
| 8 | def setUp(self): |
| 9 | self.dumps = pickle.dumps |
| 10 | self.loads = pickle.loads |
| Tim Peters | e0c446b | 2001-10-18 21:57:37 +0000 | [diff] [blame^] | 11 | |
| Jeremy Hylton | 6642653 | 2001-10-15 21:38:56 +0000 | [diff] [blame] | 12 | module = pickle |
| 13 | error = KeyError |
| 14 | |
| 15 | class PicklerTests(AbstractPickleTests): |
| 16 | |
| 17 | error = KeyError |
| 18 | |
| 19 | def dumps(self, arg, bin=0): |
| 20 | f = StringIO() |
| 21 | p = pickle.Pickler(f, bin) |
| 22 | p.dump(arg) |
| 23 | f.seek(0) |
| 24 | return f.read() |
| 25 | |
| 26 | def loads(self, buf): |
| 27 | f = StringIO(buf) |
| 28 | u = pickle.Unpickler(f) |
| 29 | return u.load() |
| 30 | |
| 31 | if __name__ == "__main__": |
| 32 | run_unittest(PickleTests) |
| 33 | run_unittest(PicklerTests) |