1 # A lot of failures in these tests on Mac OS X. 2 # Byte order related? 3 4 import unittest 5 from ctypes import * 6 from ctypes.test import need_symbol 7 8 import _ctypes_test 9 10 class CFunctions(unittest.TestCase): 11 _dll = CDLL(_ctypes_test.__file__) 12 13 def S(self): 14 return c_longlong.in_dll(self._dll, "last_tf_arg_s").value 15 def U(self): 16 return c_ulonglong.in_dll(self._dll, "last_tf_arg_u").value 17 18 def test_byte(self): 19 self._dll.tf_b.restype = c_byte 20 self._dll.tf_b.argtypes = (c_byte,) 21 self.assertEqual(self._dll.tf_b(-126), -42) 22 self.assertEqual(self.S(), -126) 23 24 def test_byte_plus(self): 25 self._dll.tf_bb.restype = c_byte 26 self._dll.tf_bb.argtypes = (c_byte, c_byte) 27 self.assertEqual(self._dll.tf_bb(0, -126), -42) 28 self.assertEqual(self.S(), -126) 29 30 def test_ubyte(self): 31 self._dll.tf_B.restype = c_ubyte 32 self._dll.tf_B.argtypes = (c_ubyte,) 33 self.assertEqual(self._dll.tf_B(255), 85) 34 self.assertEqual(self.U(), 255) 35 36 def test_ubyte_plus(self): 37 self._dll.tf_bB.restype = c_ubyte 38 self._dll.tf_bB.argtypes = (c_byte, c_ubyte) 39 self.assertEqual(self._dll.tf_bB(0, 255), 85) 40 self.assertEqual(self.U(), 255) 41 42 def test_short(self): 43 self._dll.tf_h.restype = c_short 44 self._dll.tf_h.argtypes = (c_short,) 45 self.assertEqual(self._dll.tf_h(-32766), -10922) 46 self.assertEqual(self.S(), -32766) 47 48 def test_short_plus(self): 49 self._dll.tf_bh.restype = c_short 50 self._dll.tf_bh.argtypes = (c_byte, c_short) 51 self.assertEqual(self._dll.tf_bh(0, -32766), -10922) 52 self.assertEqual(self.S(), -32766) 53 54 def test_ushort(self): 55 self._dll.tf_H.restype = c_ushort 56 self._dll.tf_H.argtypes = (c_ushort,) 57 self.assertEqual(self._dll.tf_H(65535), 21845) 58 self.assertEqual(self.U(), 65535) 59 60 def test_ushort_plus(self): 61 self._dll.tf_bH.restype = c_ushort 62 self._dll.tf_bH.argtypes = (c_byte, c_ushort) 63 self.assertEqual(self._dll.tf_bH(0, 65535), 21845) 64 self.assertEqual(self.U(), 65535) 65 66 def test_int(self): 67 self._dll.tf_i.restype = c_int 68 self._dll.tf_i.argtypes = (c_int,) 69 self.assertEqual(self._dll.tf_i(-2147483646), -715827882) 70 self.assertEqual(self.S(), -2147483646) 71 72 def test_int_plus(self): 73 self._dll.tf_bi.restype = c_int 74 self._dll.tf_bi.argtypes = (c_byte, c_int) 75 self.assertEqual(self._dll.tf_bi(0, -2147483646), -715827882) 76 self.assertEqual(self.S(), -2147483646) 77 78 def test_uint(self): 79 self._dll.tf_I.restype = c_uint 80 self._dll.tf_I.argtypes = (c_uint,) 81 self.assertEqual(self._dll.tf_I(4294967295), 1431655765) 82 self.assertEqual(self.U(), 4294967295) 83 84 def test_uint_plus(self): 85 self._dll.tf_bI.restype = c_uint 86 self._dll.tf_bI.argtypes = (c_byte, c_uint) 87 self.assertEqual(self._dll.tf_bI(0, 4294967295), 1431655765) 88 self.assertEqual(self.U(), 4294967295) 89 90 def test_long(self): 91 self._dll.tf_l.restype = c_long 92 self._dll.tf_l.argtypes = (c_long,) 93 self.assertEqual(self._dll.tf_l(-2147483646), -715827882) 94 self.assertEqual(self.S(), -2147483646) 95 96 def test_long_plus(self): 97 self._dll.tf_bl.restype = c_long 98 self._dll.tf_bl.argtypes = (c_byte, c_long) 99 self.assertEqual(self._dll.tf_bl(0, -2147483646), -715827882) 100 self.assertEqual(self.S(), -2147483646) 101 102 def test_ulong(self): 103 self._dll.tf_L.restype = c_ulong 104 self._dll.tf_L.argtypes = (c_ulong,) 105 self.assertEqual(self._dll.tf_L(4294967295), 1431655765) 106 self.assertEqual(self.U(), 4294967295) 107 108 def test_ulong_plus(self): 109 self._dll.tf_bL.restype = c_ulong 110 self._dll.tf_bL.argtypes = (c_char, c_ulong) 111 self.assertEqual(self._dll.tf_bL(b' ', 4294967295), 1431655765) 112 self.assertEqual(self.U(), 4294967295) 113 114 def test_longlong(self): 115 self._dll.tf_q.restype = c_longlong 116 self._dll.tf_q.argtypes = (c_longlong, ) 117 self.assertEqual(self._dll.tf_q(-9223372036854775806), -3074457345618258602) 118 self.assertEqual(self.S(), -9223372036854775806) 119 120 def test_longlong_plus(self): 121 self._dll.tf_bq.restype = c_longlong 122 self._dll.tf_bq.argtypes = (c_byte, c_longlong) 123 self.assertEqual(self._dll.tf_bq(0, -9223372036854775806), -3074457345618258602) 124 self.assertEqual(self.S(), -9223372036854775806) 125 126 def test_ulonglong(self): 127 self._dll.tf_Q.restype = c_ulonglong 128 self._dll.tf_Q.argtypes = (c_ulonglong, ) 129 self.assertEqual(self._dll.tf_Q(18446744073709551615), 6148914691236517205) 130 self.assertEqual(self.U(), 18446744073709551615) 131 132 def test_ulonglong_plus(self): 133 self._dll.tf_bQ.restype = c_ulonglong 134 self._dll.tf_bQ.argtypes = (c_byte, c_ulonglong) 135 self.assertEqual(self._dll.tf_bQ(0, 18446744073709551615), 6148914691236517205) 136 self.assertEqual(self.U(), 18446744073709551615) 137 138 def test_float(self): 139 self._dll.tf_f.restype = c_float 140 self._dll.tf_f.argtypes = (c_float,) 141 self.assertEqual(self._dll.tf_f(-42.), -14.) 142 self.assertEqual(self.S(), -42) 143 144 def test_float_plus(self): 145 self._dll.tf_bf.restype = c_float 146 self._dll.tf_bf.argtypes = (c_byte, c_float) 147 self.assertEqual(self._dll.tf_bf(0, -42.), -14.) 148 self.assertEqual(self.S(), -42) 149 150 def test_double(self): 151 self._dll.tf_d.restype = c_double 152 self._dll.tf_d.argtypes = (c_double,) 153 self.assertEqual(self._dll.tf_d(42.), 14.) 154 self.assertEqual(self.S(), 42) 155 156 def test_double_plus(self): 157 self._dll.tf_bd.restype = c_double 158 self._dll.tf_bd.argtypes = (c_byte, c_double) 159 self.assertEqual(self._dll.tf_bd(0, 42.), 14.) 160 self.assertEqual(self.S(), 42) 161 162 def test_longdouble(self): 163 self._dll.tf_D.restype = c_longdouble 164 self._dll.tf_D.argtypes = (c_longdouble,) 165 self.assertEqual(self._dll.tf_D(42.), 14.) 166 self.assertEqual(self.S(), 42) 167 168 def test_longdouble_plus(self): 169 self._dll.tf_bD.restype = c_longdouble 170 self._dll.tf_bD.argtypes = (c_byte, c_longdouble) 171 self.assertEqual(self._dll.tf_bD(0, 42.), 14.) 172 self.assertEqual(self.S(), 42) 173 174 def test_callwithresult(self): 175 def process_result(result): 176 return result * 2 177 self._dll.tf_i.restype = process_result 178 self._dll.tf_i.argtypes = (c_int,) 179 self.assertEqual(self._dll.tf_i(42), 28) 180 self.assertEqual(self.S(), 42) 181 self.assertEqual(self._dll.tf_i(-42), -28) 182 self.assertEqual(self.S(), -42) 183 184 def test_void(self): 185 self._dll.tv_i.restype = None 186 self._dll.tv_i.argtypes = (c_int,) 187 self.assertEqual(self._dll.tv_i(42), None) 188 self.assertEqual(self.S(), 42) 189 self.assertEqual(self._dll.tv_i(-42), None) 190 self.assertEqual(self.S(), -42) 191 192 # The following repeats the above tests with stdcall functions (where 193 # they are available) 194 try: 195 WinDLL 196 except NameError: 197 def stdcall_dll(*_): pass 198 else: 199 class stdcall_dll(WinDLL): 200 def __getattr__(self, name): 201 if name[:2] == '__' and name[-2:] == '__': 202 raise AttributeError(name) 203 func = self._FuncPtr(("s_" + name, self)) 204 setattr(self, name, func) 205 return func 206 207 @need_symbol('WinDLL') 208 class stdcallCFunctions(CFunctions): 209 _dll = stdcall_dll(_ctypes_test.__file__) 210 211 if __name__ == '__main__': 212 unittest.main() 213
