[pypy-commit] pypy py3k: Fix shape of array buffers

[rlamy]

Author: Ronan Lamy <ronan.lamy at gmail.com>
Branch: py3k
Changeset: r86784:00ac4fb4c63b
Date: 2016-08-31 16:21 +0100
http://bitbucket.org/pypy/pypy/changeset/00ac4fb4c63b/

Log:	Fix shape of array buffers

diff --git a/pypy/module/array/interp_array.py b/pypy/module/array/interp_array.py
--- a/pypy/module/array/interp_array.py
+++ b/pypy/module/array/interp_array.py
@@ -645,6 +645,9 @@
     def getndim(self):
         return 1
 
+    def getshape(self):
+        return [self.array.len]
+
     def getstrides(self):
         return [self.getitemsize()]
 
diff --git a/pypy/objspace/std/test/test_memoryobject.py b/pypy/objspace/std/test/test_memoryobject.py
--- a/pypy/objspace/std/test/test_memoryobject.py
+++ b/pypy/objspace/std/test/test_memoryobject.py
@@ -153,6 +153,7 @@
         assert m.format == 'i'
         assert m.itemsize == 4
         assert len(m) == 10
+        assert m.shape == (10,)
         assert len(m.tobytes()) == 40
         assert m[0] == 0
         m[0] = 1