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.
57 lines
1.7 KiB
57 lines
1.7 KiB
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()
|