Blame - Lib/test/test_pickle.py - platform/external/python/cpython2

blob: cbcf4912df6d5a88ef8e648b5723fa3b3cb6a577 [file] [log] [blame]

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
Tim Peters	47a6b13	2003-01-28 22:34:11 +0000	[diff] [blame]	3
Barry Warsaw	04f357c	2002-07-23 19:04:11 +0000	[diff] [blame]	4	from test import test_support
Jeremy Hylton	6642653	2001-10-15 21:38:56 +0000	[diff] [blame]	5
Serhiy Storchaka	4d2cf55	2015-09-29 15:36:28 +0300	[diff] [blame]	6	from test.pickletester import (AbstractUnpickleTests,
				7	AbstractPickleTests,
Serhiy Storchaka	cdc7a91	2013-02-12 21:36:47 +0200	[diff] [blame]	8	AbstractPickleModuleTests,
				9	AbstractPersistentPicklerTests,
				10	AbstractPicklerUnpicklerObjectTests,
				11	BigmemPickleTests)
Tim Peters	47a6b13	2003-01-28 22:34:11 +0000	[diff] [blame]	12
Serhiy Storchaka	4d2cf55	2015-09-29 15:36:28 +0300	[diff] [blame]	13	class PickleTests(AbstractUnpickleTests, AbstractPickleTests,
				14	AbstractPickleModuleTests):
Jeremy Hylton	6642653	2001-10-15 21:38:56 +0000	[diff] [blame]	15
Guido van Rossum	5d9113d	2003-01-29 17:58:45 +0000	[diff] [blame]	16	def dumps(self, arg, proto=0, fast=0):
				17	# Ignore fast
				18	return pickle.dumps(arg, proto)
				19
				20	def loads(self, buf):
				21	# Ignore fast
				22	return pickle.loads(buf)
Tim Peters	e0c446b	2001-10-18 21:57:37 +0000	[diff] [blame]	23
Jeremy Hylton	6642653	2001-10-15 21:38:56 +0000	[diff] [blame]	24	module = pickle
				25	error = KeyError
Serhiy Storchaka	5c13766	2015-11-23 15:20:43 +0200	[diff] [blame]	26	bad_stack_errors = (IndexError,)
Jeremy Hylton	6642653	2001-10-15 21:38:56 +0000	[diff] [blame]	27
Serhiy Storchaka	4d2cf55	2015-09-29 15:36:28 +0300	[diff] [blame]	28	class UnpicklerTests(AbstractUnpickleTests):
Jeremy Hylton	6642653	2001-10-15 21:38:56 +0000	[diff] [blame]	29
				30	error = KeyError
Serhiy Storchaka	5c13766	2015-11-23 15:20:43 +0200	[diff] [blame]	31	bad_stack_errors = (IndexError,)
Jeremy Hylton	6642653	2001-10-15 21:38:56 +0000	[diff] [blame]	32
Serhiy Storchaka	4d2cf55	2015-09-29 15:36:28 +0300	[diff] [blame]	33	def loads(self, buf):
				34	f = StringIO(buf)
				35	u = pickle.Unpickler(f)
				36	return u.load()
				37
Serhiy Storchaka	4d2cf55	2015-09-29 15:36:28 +0300	[diff] [blame]	38	class PicklerTests(AbstractPickleTests):
				39
Guido van Rossum	5d9113d	2003-01-29 17:58:45 +0000	[diff] [blame]	40	def dumps(self, arg, proto=0, fast=0):
Jeremy Hylton	6642653	2001-10-15 21:38:56 +0000	[diff] [blame]	41	f = StringIO()
Guido van Rossum	9d32bb1	2003-01-28 03:51:53 +0000	[diff] [blame]	42	p = pickle.Pickler(f, proto)
Guido van Rossum	5d9113d	2003-01-29 17:58:45 +0000	[diff] [blame]	43	if fast:
				44	p.fast = fast
Jeremy Hylton	6642653	2001-10-15 21:38:56 +0000	[diff] [blame]	45	p.dump(arg)
				46	f.seek(0)
				47	return f.read()
				48
				49	def loads(self, buf):
				50	f = StringIO(buf)
				51	u = pickle.Unpickler(f)
				52	return u.load()
				53
Jeremy Hylton	5e0f4e7	2002-11-13 22:01:27 +0000	[diff] [blame]	54	class PersPicklerTests(AbstractPersistentPicklerTests):
				55
Guido van Rossum	5d9113d	2003-01-29 17:58:45 +0000	[diff] [blame]	56	def dumps(self, arg, proto=0, fast=0):
Jeremy Hylton	5e0f4e7	2002-11-13 22:01:27 +0000	[diff] [blame]	57	class PersPickler(pickle.Pickler):
				58	def persistent_id(subself, obj):
				59	return self.persistent_id(obj)
				60	f = StringIO()
Guido van Rossum	9d32bb1	2003-01-28 03:51:53 +0000	[diff] [blame]	61	p = PersPickler(f, proto)
Guido van Rossum	5d9113d	2003-01-29 17:58:45 +0000	[diff] [blame]	62	if fast:
				63	p.fast = fast
Jeremy Hylton	5e0f4e7	2002-11-13 22:01:27 +0000	[diff] [blame]	64	p.dump(arg)
				65	f.seek(0)
				66	return f.read()
				67
				68	def loads(self, buf):
				69	class PersUnpickler(pickle.Unpickler):
				70	def persistent_load(subself, obj):
				71	return self.persistent_load(obj)
				72	f = StringIO(buf)
				73	u = PersUnpickler(f)
				74	return u.load()
				75
Collin Winter	f8089c7	2009-04-09 16:46:46 +0000	[diff] [blame]	76	class PicklerUnpicklerObjectTests(AbstractPicklerUnpicklerObjectTests):
				77
				78	pickler_class = pickle.Pickler
				79	unpickler_class = pickle.Unpickler
				80
Serhiy Storchaka	cdc7a91	2013-02-12 21:36:47 +0200	[diff] [blame]	81	class PickleBigmemPickleTests(BigmemPickleTests):
				82
				83	def dumps(self, arg, proto=0, fast=0):
				84	# Ignore fast
				85	return pickle.dumps(arg, proto)
				86
				87	def loads(self, buf):
				88	# Ignore fast
				89	return pickle.loads(buf)
				90
Collin Winter	f8089c7	2009-04-09 16:46:46 +0000	[diff] [blame]	91
Fred Drake	694ed09	2001-12-19 16:42:15 +0000	[diff] [blame]	92	def test_main():
Walter Dörwald	21d3a32	2003-05-01 17:45:56 +0000	[diff] [blame]	93	test_support.run_unittest(
				94	PickleTests,
Serhiy Storchaka	4d2cf55	2015-09-29 15:36:28 +0300	[diff] [blame]	95	UnpicklerTests,
Walter Dörwald	21d3a32	2003-05-01 17:45:56 +0000	[diff] [blame]	96	PicklerTests,
Collin Winter	f8089c7	2009-04-09 16:46:46 +0000	[diff] [blame]	97	PersPicklerTests,
				98	PicklerUnpicklerObjectTests,
Serhiy Storchaka	cdc7a91	2013-02-12 21:36:47 +0200	[diff] [blame]	99	PickleBigmemPickleTests,
Walter Dörwald	21d3a32	2003-05-01 17:45:56 +0000	[diff] [blame]	100	)
Tim Peters	e14295c	2003-01-30 21:27:37 +0000	[diff] [blame]	101	test_support.run_doctest(pickle)
Fred Drake	694ed09	2001-12-19 16:42:15 +0000	[diff] [blame]	102
Jeremy Hylton	6642653	2001-10-15 21:38:56 +0000	[diff] [blame]	103	if __name__ == "__main__":
Fred Drake	694ed09	2001-12-19 16:42:15 +0000	[diff] [blame]	104	test_main()