| import unittest |
| from ctypes import * |
| import _ctypes_test |
| |
| dll = CDLL(_ctypes_test.__file__) |
| |
| try: |
| CALLBACK_FUNCTYPE = WINFUNCTYPE |
| except NameError: |
| # fake to enable this test on Linux |
| CALLBACK_FUNCTYPE = CFUNCTYPE |
| |
| class POINT(Structure): |
| _fields_ = [("x", c_int), ("y", c_int)] |
| |
| class BasicWrapTestCase(unittest.TestCase): |
| def wrap(self, param): |
| return param |
| |
| def test_wchar_parm(self): |
| try: |
| c_wchar |
| except NameError: |
| return |
| f = dll._testfunc_i_bhilfd |
| f.argtypes = [c_byte, c_wchar, c_int, c_long, c_float, c_double] |
| result = f(self.wrap(1), self.wrap("x"), self.wrap(3), self.wrap(4), self.wrap(5.0), self.wrap(6.0)) |
| self.failUnlessEqual(result, 139) |
| self.failUnless(type(result), int) |
| |
| def test_pointers(self): |
| f = dll._testfunc_p_p |
| f.restype = POINTER(c_int) |
| f.argtypes = [POINTER(c_int)] |
| |
| # This only works if the value c_int(42) passed to the |
| # function is still alive while the pointer (the result) is |
| # used. |
| |
| v = c_int(42) |
| |
| self.failUnlessEqual(pointer(v).contents.value, 42) |
| result = f(self.wrap(pointer(v))) |
| self.failUnlessEqual(type(result), POINTER(c_int)) |
| self.failUnlessEqual(result.contents.value, 42) |
| |
| # This on works... |
| result = f(self.wrap(pointer(v))) |
| self.failUnlessEqual(result.contents.value, v.value) |
| |
| p = pointer(c_int(99)) |
| result = f(self.wrap(p)) |
| self.failUnlessEqual(result.contents.value, 99) |
| |
| def test_shorts(self): |
| f = dll._testfunc_callback_i_if |
| |
| args = [] |
| expected = [262144, 131072, 65536, 32768, 16384, 8192, 4096, 2048, |
| 1024, 512, 256, 128, 64, 32, 16, 8, 4, 2, 1] |
| |
| def callback(v): |
| args.append(v) |
| return v |
| |
| CallBack = CFUNCTYPE(c_int, c_int) |
| |
| cb = CallBack(callback) |
| f(self.wrap(2**18), self.wrap(cb)) |
| self.failUnlessEqual(args, expected) |
| |
| ################################################################ |
| |
| def test_callbacks(self): |
| f = dll._testfunc_callback_i_if |
| f.restype = c_int |
| |
| MyCallback = CFUNCTYPE(c_int, c_int) |
| |
| def callback(value): |
| #print "called back with", value |
| return value |
| |
| cb = MyCallback(callback) |
| |
| result = f(self.wrap(-10), self.wrap(cb)) |
| self.failUnlessEqual(result, -18) |
| |
| # test with prototype |
| f.argtypes = [c_int, MyCallback] |
| cb = MyCallback(callback) |
| |
| result = f(self.wrap(-10), self.wrap(cb)) |
| self.failUnlessEqual(result, -18) |
| |
| result = f(self.wrap(-10), self.wrap(cb)) |
| self.failUnlessEqual(result, -18) |
| |
| AnotherCallback = CALLBACK_FUNCTYPE(c_int, c_int, c_int, c_int, c_int) |
| |
| # check that the prototype works: we call f with wrong |
| # argument types |
| cb = AnotherCallback(callback) |
| self.assertRaises(ArgumentError, f, self.wrap(-10), self.wrap(cb)) |
| |
| def test_callbacks_2(self): |
| # Can also use simple datatypes as argument type specifiers |
| # for the callback function. |
| # In this case the call receives an instance of that type |
| f = dll._testfunc_callback_i_if |
| f.restype = c_int |
| |
| MyCallback = CFUNCTYPE(c_int, c_int) |
| |
| f.argtypes = [c_int, MyCallback] |
| |
| def callback(value): |
| #print "called back with", value |
| self.failUnlessEqual(type(value), int) |
| return value |
| |
| cb = MyCallback(callback) |
| result = f(self.wrap(-10), self.wrap(cb)) |
| self.failUnlessEqual(result, -18) |
| |
| def test_longlong_callbacks(self): |
| |
| f = dll._testfunc_callback_q_qf |
| f.restype = c_longlong |
| |
| MyCallback = CFUNCTYPE(c_longlong, c_longlong) |
| |
| f.argtypes = [c_longlong, MyCallback] |
| |
| def callback(value): |
| self.failUnless(isinstance(value, (int, int))) |
| return value & 0x7FFFFFFF |
| |
| cb = MyCallback(callback) |
| |
| self.failUnlessEqual(13577625587, int(f(self.wrap(1000000000000), self.wrap(cb)))) |
| |
| def test_byval(self): |
| # without prototype |
| ptin = POINT(1, 2) |
| ptout = POINT() |
| # EXPORT int _testfunc_byval(point in, point *pout) |
| result = dll._testfunc_byval(ptin, byref(ptout)) |
| got = result, ptout.x, ptout.y |
| expected = 3, 1, 2 |
| self.failUnlessEqual(got, expected) |
| |
| # with prototype |
| ptin = POINT(101, 102) |
| ptout = POINT() |
| dll._testfunc_byval.argtypes = (POINT, POINTER(POINT)) |
| dll._testfunc_byval.restype = c_int |
| result = dll._testfunc_byval(self.wrap(ptin), byref(ptout)) |
| got = result, ptout.x, ptout.y |
| expected = 203, 101, 102 |
| self.failUnlessEqual(got, expected) |
| |
| def test_struct_return_2H(self): |
| class S2H(Structure): |
| _fields_ = [("x", c_short), |
| ("y", c_short)] |
| dll.ret_2h_func.restype = S2H |
| dll.ret_2h_func.argtypes = [S2H] |
| inp = S2H(99, 88) |
| s2h = dll.ret_2h_func(self.wrap(inp)) |
| self.failUnlessEqual((s2h.x, s2h.y), (99*2, 88*3)) |
| |
| def test_struct_return_8H(self): |
| class S8I(Structure): |
| _fields_ = [("a", c_int), |
| ("b", c_int), |
| ("c", c_int), |
| ("d", c_int), |
| ("e", c_int), |
| ("f", c_int), |
| ("g", c_int), |
| ("h", c_int)] |
| dll.ret_8i_func.restype = S8I |
| dll.ret_8i_func.argtypes = [S8I] |
| inp = S8I(9, 8, 7, 6, 5, 4, 3, 2) |
| s8i = dll.ret_8i_func(self.wrap(inp)) |
| self.failUnlessEqual((s8i.a, s8i.b, s8i.c, s8i.d, s8i.e, s8i.f, s8i.g, s8i.h), |
| (9*2, 8*3, 7*4, 6*5, 5*6, 4*7, 3*8, 2*9)) |
| |
| #~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
| |
| class AsParamWrapper(object): |
| def __init__(self, param): |
| self._as_parameter_ = param |
| |
| class AsParamWrapperTestCase(BasicWrapTestCase): |
| wrap = AsParamWrapper |
| |
| #~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
| |
| class AsParamPropertyWrapper(object): |
| def __init__(self, param): |
| self._param = param |
| |
| def getParameter(self): |
| return self._param |
| _as_parameter_ = property(getParameter) |
| |
| class AsParamPropertyWrapperTestCase(BasicWrapTestCase): |
| wrap = AsParamPropertyWrapper |
| |
| #~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
| |
| if __name__ == '__main__': |
| unittest.main() |