Mercurial > cpython-lru_cache-weakref

annotate Lib/test/test_memoryview.py @ 65035:87f4b32905df 2.7

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Cosmetic fix to use print function.
author Hirokazu Yamamoto <ocean-city@m2.ccsnet.ne.jp>
date Fri, 01 Oct 2010 10:48:47 +0000
parents fa4d78b9e3a2
children 614c84670c63
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
52749
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
1 """Unit tests for the memoryview
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
2
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
3 XXX We need more tests! Some tests are in test_bytes
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
4 """
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
5
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
6 import unittest
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
7 import sys
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
8 import gc
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
9 import weakref
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
10 import array
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
11 from test import test_support
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
12
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
13
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
14 class AbstractMemoryTests:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
15 source_bytes = b"abcdef"
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
16
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
17 @property
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
18 def _source(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
19 return self.source_bytes
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
20
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
21 @property
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
22 def _types(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
23 return filter(None, [self.ro_type, self.rw_type])
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
24
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
25 def check_getitem_with_type(self, tp):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
26 item = self.getitem_type
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
27 b = tp(self._source)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
28 oldrefcount = sys.getrefcount(b)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
29 m = self._view(b)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
30 self.assertEquals(m[0], item(b"a"))
58472
6ffcdafa0118 use assert[Not]IsInstance where appropriate
Ezio Melotti <ezio.melotti@gmail.com>
parents: 56543
diff changeset
31 self.assertIsInstance(m[0], bytes)
52749
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
32 self.assertEquals(m[5], item(b"f"))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
33 self.assertEquals(m[-1], item(b"f"))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
34 self.assertEquals(m[-6], item(b"a"))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
35 # Bounds checking
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
36 self.assertRaises(IndexError, lambda: m[6])
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
37 self.assertRaises(IndexError, lambda: m[-7])
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
38 self.assertRaises(IndexError, lambda: m[sys.maxsize])
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
39 self.assertRaises(IndexError, lambda: m[-sys.maxsize])
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
40 # Type checking
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
41 self.assertRaises(TypeError, lambda: m[None])
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
42 self.assertRaises(TypeError, lambda: m[0.0])
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
43 self.assertRaises(TypeError, lambda: m["a"])
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
44 m = None
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
45 self.assertEquals(sys.getrefcount(b), oldrefcount)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
46
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
47 def test_getitem(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
48 for tp in self._types:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
49 self.check_getitem_with_type(tp)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
50
54829
f65adad3988d Issue 6329: Fix iteration for memoryviews.
Raymond Hettinger <python@rcn.com>
parents: 52749
diff changeset
51 def test_iter(self):
f65adad3988d Issue 6329: Fix iteration for memoryviews.
Raymond Hettinger <python@rcn.com>
parents: 52749
diff changeset
52 for tp in self._types:
f65adad3988d Issue 6329: Fix iteration for memoryviews.
Raymond Hettinger <python@rcn.com>
parents: 52749
diff changeset
53 b = tp(self._source)
f65adad3988d Issue 6329: Fix iteration for memoryviews.
Raymond Hettinger <python@rcn.com>
parents: 52749
diff changeset
54 m = self._view(b)
f65adad3988d Issue 6329: Fix iteration for memoryviews.
Raymond Hettinger <python@rcn.com>
parents: 52749
diff changeset
55 self.assertEqual(list(m), [m[i] for i in range(len(m))])
f65adad3988d Issue 6329: Fix iteration for memoryviews.
Raymond Hettinger <python@rcn.com>
parents: 52749
diff changeset
56
56543
1617c11556af Issue #7142: Fix uses of unicode in memoryview objects
Mark Dickinson <dickinsm@gmail.com>
parents: 55002
diff changeset
57 def test_repr(self):
1617c11556af Issue #7142: Fix uses of unicode in memoryview objects
Mark Dickinson <dickinsm@gmail.com>
parents: 55002
diff changeset
58 for tp in self._types:
1617c11556af Issue #7142: Fix uses of unicode in memoryview objects
Mark Dickinson <dickinsm@gmail.com>
parents: 55002
diff changeset
59 b = tp(self._source)
1617c11556af Issue #7142: Fix uses of unicode in memoryview objects
Mark Dickinson <dickinsm@gmail.com>
parents: 55002
diff changeset
60 m = self._view(b)
1617c11556af Issue #7142: Fix uses of unicode in memoryview objects
Mark Dickinson <dickinsm@gmail.com>
parents: 55002
diff changeset
61 self.assertIsInstance(m.__repr__(), str)
1617c11556af Issue #7142: Fix uses of unicode in memoryview objects
Mark Dickinson <dickinsm@gmail.com>
parents: 55002
diff changeset
62
52749
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
63 def test_setitem_readonly(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
64 if not self.ro_type:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
65 return
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
66 b = self.ro_type(self._source)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
67 oldrefcount = sys.getrefcount(b)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
68 m = self._view(b)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
69 def setitem(value):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
70 m[0] = value
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
71 self.assertRaises(TypeError, setitem, b"a")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
72 self.assertRaises(TypeError, setitem, 65)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
73 self.assertRaises(TypeError, setitem, memoryview(b"a"))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
74 m = None
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
75 self.assertEquals(sys.getrefcount(b), oldrefcount)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
76
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
77 def test_setitem_writable(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
78 if not self.rw_type:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
79 return
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
80 tp = self.rw_type
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
81 b = self.rw_type(self._source)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
82 oldrefcount = sys.getrefcount(b)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
83 m = self._view(b)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
84 m[0] = tp(b"0")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
85 self._check_contents(tp, b, b"0bcdef")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
86 m[1:3] = tp(b"12")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
87 self._check_contents(tp, b, b"012def")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
88 m[1:1] = tp(b"")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
89 self._check_contents(tp, b, b"012def")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
90 m[:] = tp(b"abcdef")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
91 self._check_contents(tp, b, b"abcdef")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
92
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
93 # Overlapping copies of a view into itself
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
94 m[0:3] = m[2:5]
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
95 self._check_contents(tp, b, b"cdedef")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
96 m[:] = tp(b"abcdef")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
97 m[2:5] = m[0:3]
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
98 self._check_contents(tp, b, b"ababcf")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
99
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
100 def setitem(key, value):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
101 m[key] = tp(value)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
102 # Bounds checking
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
103 self.assertRaises(IndexError, setitem, 6, b"a")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
104 self.assertRaises(IndexError, setitem, -7, b"a")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
105 self.assertRaises(IndexError, setitem, sys.maxsize, b"a")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
106 self.assertRaises(IndexError, setitem, -sys.maxsize, b"a")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
107 # Wrong index/slice types
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
108 self.assertRaises(TypeError, setitem, 0.0, b"a")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
109 self.assertRaises(TypeError, setitem, (0,), b"a")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
110 self.assertRaises(TypeError, setitem, "a", b"a")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
111 # Trying to resize the memory object
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
112 self.assertRaises(ValueError, setitem, 0, b"")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
113 self.assertRaises(ValueError, setitem, 0, b"ab")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
114 self.assertRaises(ValueError, setitem, slice(1,1), b"a")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
115 self.assertRaises(ValueError, setitem, slice(0,2), b"a")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
116
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
117 m = None
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
118 self.assertEquals(sys.getrefcount(b), oldrefcount)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
119
64363
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58472
diff changeset
120 def test_delitem(self):
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58472
diff changeset
121 for tp in self._types:
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58472
diff changeset
122 b = tp(self._source)
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58472
diff changeset
123 m = self._view(b)
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58472
diff changeset
124 with self.assertRaises(TypeError):
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58472
diff changeset
125 del m[1]
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58472
diff changeset
126 with self.assertRaises(TypeError):
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58472
diff changeset
127 del m[1:4]
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58472
diff changeset
128
52749
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
129 def test_tobytes(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
130 for tp in self._types:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
131 m = self._view(tp(self._source))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
132 b = m.tobytes()
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
133 # This calls self.getitem_type() on each separate byte of b"abcdef"
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
134 expected = b"".join(
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
135 self.getitem_type(c) for c in b"abcdef")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
136 self.assertEquals(b, expected)
58472
6ffcdafa0118 use assert[Not]IsInstance where appropriate
Ezio Melotti <ezio.melotti@gmail.com>
parents: 56543
diff changeset
137 self.assertIsInstance(b, bytes)
52749
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
138
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
139 def test_tolist(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
140 for tp in self._types:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
141 m = self._view(tp(self._source))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
142 l = m.tolist()
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
143 self.assertEquals(l, map(ord, b"abcdef"))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
144
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
145 def test_compare(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
146 # memoryviews can compare for equality with other objects
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
147 # having the buffer interface.
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
148 for tp in self._types:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
149 m = self._view(tp(self._source))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
150 for tp_comp in self._types:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
151 self.assertTrue(m == tp_comp(b"abcdef"))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
152 self.assertFalse(m != tp_comp(b"abcdef"))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
153 self.assertFalse(m == tp_comp(b"abcde"))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
154 self.assertTrue(m != tp_comp(b"abcde"))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
155 self.assertFalse(m == tp_comp(b"abcde1"))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
156 self.assertTrue(m != tp_comp(b"abcde1"))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
157 self.assertTrue(m == m)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
158 self.assertTrue(m == m[:])
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
159 self.assertTrue(m[0:6] == m[:])
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
160 self.assertFalse(m[0:5] == m)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
161
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
162 # Comparison with objects which don't support the buffer API
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
163 self.assertFalse(m == u"abcdef")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
164 self.assertTrue(m != u"abcdef")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
165 self.assertFalse(u"abcdef" == m)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
166 self.assertTrue(u"abcdef" != m)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
167
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
168 # Unordered comparisons are unimplemented, and therefore give
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
169 # arbitrary results (they raise a TypeError in py3k)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
170
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
171 def check_attributes_with_type(self, tp):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
172 m = self._view(tp(self._source))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
173 self.assertEquals(m.format, self.format)
56543
1617c11556af Issue #7142: Fix uses of unicode in memoryview objects
Mark Dickinson <dickinsm@gmail.com>
parents: 55002
diff changeset
174 self.assertIsInstance(m.format, str)
52749
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
175 self.assertEquals(m.itemsize, self.itemsize)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
176 self.assertEquals(m.ndim, 1)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
177 self.assertEquals(m.shape, (6,))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
178 self.assertEquals(len(m), 6)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
179 self.assertEquals(m.strides, (self.itemsize,))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
180 self.assertEquals(m.suboffsets, None)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
181 return m
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
182
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
183 def test_attributes_readonly(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
184 if not self.ro_type:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
185 return
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
186 m = self.check_attributes_with_type(self.ro_type)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
187 self.assertEquals(m.readonly, True)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
188
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
189 def test_attributes_writable(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
190 if not self.rw_type:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
191 return
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
192 m = self.check_attributes_with_type(self.rw_type)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
193 self.assertEquals(m.readonly, False)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
194
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
195 # Disabled: unicode uses the old buffer API in 2.x
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
196
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
197 #def test_getbuffer(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
198 ## Test PyObject_GetBuffer() on a memoryview object.
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
199 #for tp in self._types:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
200 #b = tp(self._source)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
201 #oldrefcount = sys.getrefcount(b)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
202 #m = self._view(b)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
203 #oldviewrefcount = sys.getrefcount(m)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
204 #s = unicode(m, "utf-8")
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
205 #self._check_contents(tp, b, s.encode("utf-8"))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
206 #self.assertEquals(sys.getrefcount(m), oldviewrefcount)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
207 #m = None
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
208 #self.assertEquals(sys.getrefcount(b), oldrefcount)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
209
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
210 def test_gc(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
211 for tp in self._types:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
212 if not isinstance(tp, type):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
213 # If tp is a factory rather than a plain type, skip
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
214 continue
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
215
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
216 class MySource(tp):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
217 pass
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
218 class MyObject:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
219 pass
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
220
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
221 # Create a reference cycle through a memoryview object
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
222 b = MySource(tp(b'abc'))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
223 m = self._view(b)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
224 o = MyObject()
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
225 b.m = m
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
226 b.o = o
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
227 wr = weakref.ref(o)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
228 b = m = o = None
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
229 # The cycle must be broken
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
230 gc.collect()
55002
6768b2350c57 convert usage of fail* to assert*
Benjamin Peterson <benjamin@python.org>
parents: 54829
diff changeset
231 self.assertTrue(wr() is None, wr())
52749
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
232
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
233
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
234 # Variations on source objects for the buffer: bytes-like objects, then arrays
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
235 # with itemsize > 1.
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
236 # NOTE: support for multi-dimensional objects is unimplemented.
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
237
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
238 class BaseBytesMemoryTests(AbstractMemoryTests):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
239 ro_type = bytes
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
240 rw_type = bytearray
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
241 getitem_type = bytes
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
242 itemsize = 1
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
243 format = 'B'
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
244
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
245 # Disabled: array.array() does not support the new buffer API in 2.x
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
246
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
247 #class BaseArrayMemoryTests(AbstractMemoryTests):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
248 #ro_type = None
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
249 #rw_type = lambda self, b: array.array('i', map(ord, b))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
250 #getitem_type = lambda self, b: array.array('i', map(ord, b)).tostring()
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
251 #itemsize = array.array('i').itemsize
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
252 #format = 'i'
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
253
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
254 #def test_getbuffer(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
255 ## XXX Test should be adapted for non-byte buffers
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
256 #pass
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
257
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
258 #def test_tolist(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
259 ## XXX NotImplementedError: tolist() only supports byte views
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
260 #pass
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
261
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
262
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
263 # Variations on indirection levels: memoryview, slice of memoryview,
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
264 # slice of slice of memoryview.
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
265 # This is important to test allocation subtleties.
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
266
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
267 class BaseMemoryviewTests:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
268 def _view(self, obj):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
269 return memoryview(obj)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
270
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
271 def _check_contents(self, tp, obj, contents):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
272 self.assertEquals(obj, tp(contents))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
273
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
274 class BaseMemorySliceTests:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
275 source_bytes = b"XabcdefY"
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
276
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
277 def _view(self, obj):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
278 m = memoryview(obj)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
279 return m[1:7]
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
280
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
281 def _check_contents(self, tp, obj, contents):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
282 self.assertEquals(obj[1:7], tp(contents))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
283
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
284 def test_refs(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
285 for tp in self._types:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
286 m = memoryview(tp(self._source))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
287 oldrefcount = sys.getrefcount(m)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
288 m[1:2]
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
289 self.assertEquals(sys.getrefcount(m), oldrefcount)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
290
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
291 class BaseMemorySliceSliceTests:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
292 source_bytes = b"XabcdefY"
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
293
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
294 def _view(self, obj):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
295 m = memoryview(obj)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
296 return m[:7][1:]
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
297
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
298 def _check_contents(self, tp, obj, contents):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
299 self.assertEquals(obj[1:7], tp(contents))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
300
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
301
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
302 # Concrete test classes
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
303
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
304 class BytesMemoryviewTest(unittest.TestCase,
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
305 BaseMemoryviewTests, BaseBytesMemoryTests):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
306
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
307 def test_constructor(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
308 for tp in self._types:
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
309 ob = tp(self._source)
55002
6768b2350c57 convert usage of fail* to assert*
Benjamin Peterson <benjamin@python.org>
parents: 54829
diff changeset
310 self.assertTrue(memoryview(ob))
6768b2350c57 convert usage of fail* to assert*
Benjamin Peterson <benjamin@python.org>
parents: 54829
diff changeset
311 self.assertTrue(memoryview(object=ob))
52749
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
312 self.assertRaises(TypeError, memoryview)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
313 self.assertRaises(TypeError, memoryview, ob, ob)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
314 self.assertRaises(TypeError, memoryview, argument=ob)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
315 self.assertRaises(TypeError, memoryview, ob, argument=True)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
316
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
317 #class ArrayMemoryviewTest(unittest.TestCase,
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
318 #BaseMemoryviewTests, BaseArrayMemoryTests):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
319
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
320 #def test_array_assign(self):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
321 ## Issue #4569: segfault when mutating a memoryview with itemsize != 1
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
322 #a = array.array('i', range(10))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
323 #m = memoryview(a)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
324 #new_a = array.array('i', range(9, -1, -1))
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
325 #m[:] = new_a
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
326 #self.assertEquals(a, new_a)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
327
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
328
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
329 class BytesMemorySliceTest(unittest.TestCase,
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
330 BaseMemorySliceTests, BaseBytesMemoryTests):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
331 pass
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
332
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
333 #class ArrayMemorySliceTest(unittest.TestCase,
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
334 #BaseMemorySliceTests, BaseArrayMemoryTests):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
335 #pass
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
336
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
337 class BytesMemorySliceSliceTest(unittest.TestCase,
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
338 BaseMemorySliceSliceTests, BaseBytesMemoryTests):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
339 pass
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
340
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
341 #class ArrayMemorySliceSliceTest(unittest.TestCase,
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
342 #BaseMemorySliceSliceTests, BaseArrayMemoryTests):
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
343 #pass
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
344
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
345
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
346 def test_main():
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
347 test_support.run_unittest(__name__)
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
348
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
349 if __name__ == "__main__":
75efd757434b Issue #2396: backport the memoryview object.
Antoine Pitrou <solipsis@pitrou.net>
parents:
diff changeset
350 test_main()