1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
"""Misc cPickle tests.
Made for Jython.
"""
import __builtin__
import sys
import cPickle
import pickle
import unittest
from StringIO import StringIO
from test import test_support
class MyClass(object):
pass
class CPickleTestCase(unittest.TestCase):
def test_zero_long(self):
self.assertEqual(cPickle.loads(cPickle.dumps(0L, 2)), 0L)
self.assertEqual(cPickle.dumps(0L, 2), pickle.dumps(0L, 2))
def test_cyclic_memoize(self):
# http://bugs.python.org/issue998998 - cPickle shouldn't fail
# this, though pickle.py still does
m = MyClass()
m2 = MyClass()
s = set([m])
m.foo = set([m2])
m2.foo = s
s2 = cPickle.loads(cPickle.dumps(s))
self.assertEqual(len(s2), 1)
m3 = iter(s2).next()
self.assertEqual(len(m3.foo), 1)
m4 = iter(m3.foo).next()
self.assertEqual(m4.foo, s2)
def test_find_global(self):
class A(object):
def __init__(self, x, y):
self.x = x
self.y = y
def __eq__(self, other):
if isinstance(other, A) and self.x == other.x and self.y == other.y:
return True
return False
class B(object):
def __init__(self, x, y):
self.x = x
self.y = y
def restrictive_find_global(module, clsname):
if clsname == 'A':
return A
else:
raise pickle.UnpicklingError("Cannot load class", module, clsname)
a = A("python", "C")
a_pickled = cPickle.dumps(a, 2)
a_unpickler = cPickle.Unpickler(StringIO(a_pickled))
a_unpickler.find_global = restrictive_find_global
self.assertEqual(a_unpickler.load(), a)
b_pickled = cPickle.dumps(B("jython", "java"), 2)
b_unpickler = cPickle.Unpickler(StringIO(b_pickled))
b_unpickler.find_global = restrictive_find_global
self.assertRaises(pickle.UnpicklingError, b_unpickler.load)
def testWithUserDefinedImport(self):
"""test cPickle calling a user defined import function."""
# This tests the fix for http://bugs.jython.org/issue1665
# setup
original_import = __builtin__.__import__
def import_hook(name, _globals=None, locals=None, fromlist=None, level= -1):
return original_import(name, _globals, locals, fromlist, level)
# test
__builtin__.__import__ = import_hook
try:
if "no_such_module" in sys.modules:
del sys.modules["no_such_module"] # force cPickle to call __import__
self.assertRaises(ImportError, cPickle.loads, pickle.GLOBAL + "no_such_module\n" + "no_such_class\n")
finally:
__builtin__.__import__ = original_import
def test_main():
test_support.run_unittest(CPickleTestCase)
if __name__ == '__main__':
test_main()
|
