Current Path : /usr/local/lib/python2.5/ctypes/test/ |
FreeBSD hs32.drive.ne.jp 9.1-RELEASE FreeBSD 9.1-RELEASE #1: Wed Jan 14 12:18:08 JST 2015 root@hs32.drive.ne.jp:/sys/amd64/compile/hs32 amd64 |
Current File : //usr/local/lib/python2.5/ctypes/test/test_strings.py |
import unittest from ctypes import * class StringArrayTestCase(unittest.TestCase): def test(self): BUF = c_char * 4 buf = BUF("a", "b", "c") self.failUnlessEqual(buf.value, "abc") self.failUnlessEqual(buf.raw, "abc\000") buf.value = "ABCD" self.failUnlessEqual(buf.value, "ABCD") self.failUnlessEqual(buf.raw, "ABCD") buf.value = "x" self.failUnlessEqual(buf.value, "x") self.failUnlessEqual(buf.raw, "x\000CD") buf[1] = "Z" self.failUnlessEqual(buf.value, "xZCD") self.failUnlessEqual(buf.raw, "xZCD") self.assertRaises(ValueError, setattr, buf, "value", "aaaaaaaa") self.assertRaises(TypeError, setattr, buf, "value", 42) def test_c_buffer_value(self): buf = c_buffer(32) buf.value = "Hello, World" self.failUnlessEqual(buf.value, "Hello, World") self.failUnlessRaises(TypeError, setattr, buf, "value", buffer("Hello, World")) self.assertRaises(TypeError, setattr, buf, "value", buffer("abc")) self.assertRaises(ValueError, setattr, buf, "raw", buffer("x" * 100)) def test_c_buffer_raw(self): buf = c_buffer(32) buf.raw = buffer("Hello, World") self.failUnlessEqual(buf.value, "Hello, World") self.assertRaises(TypeError, setattr, buf, "value", buffer("abc")) self.assertRaises(ValueError, setattr, buf, "raw", buffer("x" * 100)) def test_param_1(self): BUF = c_char * 4 buf = BUF() ## print c_char_p.from_param(buf) def test_param_2(self): BUF = c_char * 4 buf = BUF() ## print BUF.from_param(c_char_p("python")) ## print BUF.from_param(BUF(*"pyth")) try: c_wchar except NameError: pass else: class WStringArrayTestCase(unittest.TestCase): def test(self): BUF = c_wchar * 4 buf = BUF(u"a", u"b", u"c") self.failUnlessEqual(buf.value, u"abc") buf.value = u"ABCD" self.failUnlessEqual(buf.value, u"ABCD") buf.value = u"x" self.failUnlessEqual(buf.value, u"x") buf[1] = u"Z" self.failUnlessEqual(buf.value, u"xZCD") class StringTestCase(unittest.TestCase): def XX_test_basic_strings(self): cs = c_string("abcdef") # Cannot call len on a c_string any longer self.assertRaises(TypeError, len, cs) self.failUnlessEqual(sizeof(cs), 7) # The value property is the string up to the first terminating NUL. self.failUnlessEqual(cs.value, "abcdef") self.failUnlessEqual(c_string("abc\000def").value, "abc") # The raw property is the total buffer contents: self.failUnlessEqual(cs.raw, "abcdef\000") self.failUnlessEqual(c_string("abc\000def").raw, "abc\000def\000") # We can change the value: cs.value = "ab" self.failUnlessEqual(cs.value, "ab") self.failUnlessEqual(cs.raw, "ab\000\000\000\000\000") cs.raw = "XY" self.failUnlessEqual(cs.value, "XY") self.failUnlessEqual(cs.raw, "XY\000\000\000\000\000") self.assertRaises(TypeError, c_string, u"123") def XX_test_sized_strings(self): # New in releases later than 0.4.0: self.assertRaises(TypeError, c_string, None) # New in releases later than 0.4.0: # c_string(number) returns an empty string of size number self.failUnless(len(c_string(32).raw) == 32) self.assertRaises(ValueError, c_string, -1) self.assertRaises(ValueError, c_string, 0) # These tests fail, because it is no longer initialized ## self.failUnless(c_string(2).value == "") ## self.failUnless(c_string(2).raw == "\000\000") self.failUnless(c_string(2).raw[-1] == "\000") self.failUnless(len(c_string(2).raw) == 2) def XX_test_initialized_strings(self): self.failUnless(c_string("ab", 4).raw[:2] == "ab") self.failUnless(c_string("ab", 4).raw[-1] == "\000") self.failUnless(c_string("ab", 2).raw == "a\000") def XX_test_toolong(self): cs = c_string("abcdef") # Much too long string: self.assertRaises(ValueError, setattr, cs, "value", "123456789012345") # One char too long values: self.assertRaises(ValueError, setattr, cs, "value", "1234567") ## def test_perf(self): ## check_perf() try: c_wchar except NameError: pass else: class WStringTestCase(unittest.TestCase): def test_wchar(self): c_wchar(u"x") repr(byref(c_wchar(u"x"))) c_wchar("x") def X_test_basic_wstrings(self): cs = c_wstring(u"abcdef") # XXX This behaviour is about to change: # len returns the size of the internal buffer in bytes. # This includes the terminating NUL character. self.failUnless(sizeof(cs) == 14) # The value property is the string up to the first terminating NUL. self.failUnless(cs.value == u"abcdef") self.failUnless(c_wstring(u"abc\000def").value == u"abc") self.failUnless(c_wstring(u"abc\000def").value == u"abc") # The raw property is the total buffer contents: self.failUnless(cs.raw == u"abcdef\000") self.failUnless(c_wstring(u"abc\000def").raw == u"abc\000def\000") # We can change the value: cs.value = u"ab" self.failUnless(cs.value == u"ab") self.failUnless(cs.raw == u"ab\000\000\000\000\000") self.assertRaises(TypeError, c_wstring, "123") self.assertRaises(ValueError, c_wstring, 0) def X_test_toolong(self): cs = c_wstring(u"abcdef") # Much too long string: self.assertRaises(ValueError, setattr, cs, "value", u"123456789012345") # One char too long values: self.assertRaises(ValueError, setattr, cs, "value", u"1234567") def run_test(rep, msg, func, arg): items = range(rep) from time import clock start = clock() for i in items: func(arg); func(arg); func(arg); func(arg); func(arg) stop = clock() print "%20s: %.2f us" % (msg, ((stop-start)*1e6/5/rep)) def check_perf(): # Construct 5 objects REP = 200000 run_test(REP, "c_string(None)", c_string, None) run_test(REP, "c_string('abc')", c_string, 'abc') # Python 2.3 -OO, win2k, P4 700 MHz: # # c_string(None): 1.75 us # c_string('abc'): 2.74 us # Python 2.2 -OO, win2k, P4 700 MHz: # # c_string(None): 2.95 us # c_string('abc'): 3.67 us if __name__ == '__main__': ## check_perf() unittest.main()