You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
2.2 KiB
82 lines
2.2 KiB
6 years ago
|
import unittest
|
||
|
import pickle
|
||
|
from ctypes import *
|
||
|
import _ctypes_test
|
||
|
dll = CDLL(_ctypes_test.__file__)
|
||
|
|
||
|
class X(Structure):
|
||
|
_fields_ = [("a", c_int), ("b", c_double)]
|
||
|
init_called = 0
|
||
|
def __init__(self, *args, **kw):
|
||
|
X.init_called += 1
|
||
|
self.x = 42
|
||
|
|
||
|
class Y(X):
|
||
|
_fields_ = [("str", c_char_p)]
|
||
|
|
||
|
class PickleTest:
|
||
|
def dumps(self, item):
|
||
|
return pickle.dumps(item, self.proto)
|
||
|
|
||
|
def loads(self, item):
|
||
|
return pickle.loads(item)
|
||
|
|
||
|
def test_simple(self):
|
||
|
for src in [
|
||
|
c_int(42),
|
||
|
c_double(3.14),
|
||
|
]:
|
||
|
dst = self.loads(self.dumps(src))
|
||
|
self.assertEqual(src.__dict__, dst.__dict__)
|
||
|
self.assertEqual(memoryview(src).tobytes(),
|
||
|
memoryview(dst).tobytes())
|
||
|
|
||
|
def test_struct(self):
|
||
|
X.init_called = 0
|
||
|
|
||
|
x = X()
|
||
|
x.a = 42
|
||
|
self.assertEqual(X.init_called, 1)
|
||
|
|
||
|
y = self.loads(self.dumps(x))
|
||
|
|
||
|
# loads must NOT call __init__
|
||
|
self.assertEqual(X.init_called, 1)
|
||
|
|
||
|
# ctypes instances are identical when the instance __dict__
|
||
|
# and the memory buffer are identical
|
||
|
self.assertEqual(y.__dict__, x.__dict__)
|
||
|
self.assertEqual(memoryview(y).tobytes(),
|
||
|
memoryview(x).tobytes())
|
||
|
|
||
|
def test_unpickable(self):
|
||
|
# ctypes objects that are pointers or contain pointers are
|
||
|
# unpickable.
|
||
|
self.assertRaises(ValueError, lambda: self.dumps(Y()))
|
||
|
|
||
|
prototype = CFUNCTYPE(c_int)
|
||
|
|
||
|
for item in [
|
||
|
c_char_p(),
|
||
|
c_wchar_p(),
|
||
|
c_void_p(),
|
||
|
pointer(c_int(42)),
|
||
|
dll._testfunc_p_p,
|
||
|
prototype(lambda: 42),
|
||
|
]:
|
||
|
self.assertRaises(ValueError, lambda: self.dumps(item))
|
||
|
|
||
|
def test_wchar(self):
|
||
|
self.dumps(c_char(b"x"))
|
||
|
# Issue 5049
|
||
|
self.dumps(c_wchar("x"))
|
||
|
|
||
|
for proto in range(pickle.HIGHEST_PROTOCOL + 1):
|
||
|
name = 'PickleTest_%s' % proto
|
||
|
globals()[name] = type(name,
|
||
|
(PickleTest, unittest.TestCase),
|
||
|
{'proto': proto})
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
unittest.main()
|