| import unittest |
| import ctypes |
| from ctypes.test import need_symbol |
| |
| import _ctypes_test |
| |
| @need_symbol('c_wchar') |
| class UnicodeTestCase(unittest.TestCase): |
| def test_wcslen(self): |
| dll = ctypes.CDLL(_ctypes_test.__file__) |
| wcslen = dll.my_wcslen |
| wcslen.argtypes = [ctypes.c_wchar_p] |
| |
| self.assertEqual(wcslen("abc"), 3) |
| self.assertEqual(wcslen("ab\u2070"), 3) |
| self.assertRaises(ctypes.ArgumentError, wcslen, b"ab\xe4") |
| |
| def test_buffers(self): |
| buf = ctypes.create_unicode_buffer("abc") |
| self.assertEqual(len(buf), 3+1) |
| |
| buf = ctypes.create_unicode_buffer("ab\xe4\xf6\xfc") |
| self.assertEqual(buf[:], "ab\xe4\xf6\xfc\0") |
| self.assertEqual(buf[::], "ab\xe4\xf6\xfc\0") |
| self.assertEqual(buf[::-1], '\x00\xfc\xf6\xe4ba') |
| self.assertEqual(buf[::2], 'a\xe4\xfc') |
| self.assertEqual(buf[6:5:-1], "") |
| |
| func = ctypes.CDLL(_ctypes_test.__file__)._testfunc_p_p |
| |
| class StringTestCase(UnicodeTestCase): |
| def setUp(self): |
| func.argtypes = [ctypes.c_char_p] |
| func.restype = ctypes.c_char_p |
| |
| def tearDown(self): |
| func.argtypes = None |
| func.restype = ctypes.c_int |
| |
| def test_func(self): |
| self.assertEqual(func(b"abc\xe4"), b"abc\xe4") |
| |
| def test_buffers(self): |
| buf = ctypes.create_string_buffer(b"abc") |
| self.assertEqual(len(buf), 3+1) |
| |
| buf = ctypes.create_string_buffer(b"ab\xe4\xf6\xfc") |
| self.assertEqual(buf[:], b"ab\xe4\xf6\xfc\0") |
| self.assertEqual(buf[::], b"ab\xe4\xf6\xfc\0") |
| self.assertEqual(buf[::-1], b'\x00\xfc\xf6\xe4ba') |
| self.assertEqual(buf[::2], b'a\xe4\xfc') |
| self.assertEqual(buf[6:5:-1], b"") |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |