1import unittest
2
3from ctypes import *
4import _ctypes_test
5
6lib = CDLL(_ctypes_test.__file__)
7
8class LibTest(unittest.TestCase):
9    def test_sqrt(self):
10        lib.my_sqrt.argtypes = c_double,
11        lib.my_sqrt.restype = c_double
12        self.assertEqual(lib.my_sqrt(4.0), 2.0)
13        import math
14        self.assertEqual(lib.my_sqrt(2.0), math.sqrt(2.0))
15
16    def test_qsort(self):
17        comparefunc = CFUNCTYPE(c_int, POINTER(c_char), POINTER(c_char))
18        lib.my_qsort.argtypes = c_void_p, c_size_t, c_size_t, comparefunc
19        lib.my_qsort.restype = None
20
21        def sort(a, b):
22            return cmp(a[0], b[0])
23
24        chars = create_string_buffer("spam, spam, and spam")
25        lib.my_qsort(chars, len(chars)-1, sizeof(c_char), comparefunc(sort))
26        self.assertEqual(chars.raw, "   ,,aaaadmmmnpppsss\x00")
27
28if __name__ == "__main__":
29    unittest.main()
30