← Back to branch summary

~malept/ubuntu/lucid/python2.6/dev-dependency-fix

« back to all changes in this revision

Viewing changes to Lib/ctypes/test/test_array_in_pointer.py

  • Committer: Bazaar Package Importer
  • Author(s): Matthias Klose
  • Date: 2009-02-13 12:51:00 UTC
  • Revision ID: james.westby@ubuntu.com-20090213125100-uufgcb9yeqzujpqw
Tags: upstream-2.6.1
Import upstream version 2.6.1

Show diffs side-by-side

added added

removed removed

Lines of Context:
Lib/ctypes/test/test_array_in_pointer.py
 
1
import unittest
 
2
from ctypes import *
 
3
from binascii import hexlify
 
4
import re
 
5
 
 
6
def dump(obj):
 
7
    # helper function to dump memory contents in hex, with a hyphen
 
8
    # between the bytes.
 
9
    h = hexlify(buffer(obj))
 
10
    return re.sub(r"(..)", r"\1-", h)[:-1]
 
11
 
 
12
 
 
13
class Value(Structure):
 
14
    _fields_ = [("val", c_byte)]
 
15
 
 
16
class Container(Structure):
 
17
    _fields_ = [("pvalues", POINTER(Value))]
 
18
 
 
19
class Test(unittest.TestCase):
 
20
    def test(self):
 
21
        # create an array of 4 values
 
22
        val_array = (Value * 4)()
 
23
 
 
24
        # create a container, which holds a pointer to the pvalues array.
 
25
        c = Container()
 
26
        c.pvalues = val_array
 
27
 
 
28
        # memory contains 4 NUL bytes now, that's correct
 
29
        self.failUnlessEqual("00-00-00-00", dump(val_array))
 
30
 
 
31
        # set the values of the array through the pointer:
 
32
        for i in range(4):
 
33
            c.pvalues[i].val = i + 1
 
34
 
 
35
        values = [c.pvalues[i].val for i in range(4)]
 
36
 
 
37
        # These are the expected results: here s the bug!
 
38
        self.failUnlessEqual(
 
39
            (values, dump(val_array)),
 
40
            ([1, 2, 3, 4], "01-02-03-04")
 
41
        )
 
42
 
 
43
    def test_2(self):
 
44
 
 
45
        val_array = (Value * 4)()
 
46
 
 
47
        # memory contains 4 NUL bytes now, that's correct
 
48
        self.failUnlessEqual("00-00-00-00", dump(val_array))
 
49
 
 
50
        ptr = cast(val_array, POINTER(Value))
 
51
        # set the values of the array through the pointer:
 
52
        for i in range(4):
 
53
            ptr[i].val = i + 1
 
54
 
 
55
        values = [ptr[i].val for i in range(4)]
 
56
 
 
57
        # These are the expected results: here s the bug!
 
58
        self.failUnlessEqual(
 
59
            (values, dump(val_array)),
 
60
            ([1, 2, 3, 4], "01-02-03-04")
 
61
        )
 
62
 
 
63
if __name__ == "__main__":
 
64
    unittest.main()