[pypy-commit] pypy default: add size attribute for scalars

bdkearns noreply at buildbot.pypy.org
Thu Nov 14 03:36:05 CET 2013


Author: Brian Kearns <bdkearns at gmail.com>
Branch: 
Changeset: r68010:e14d70573095
Date: 2013-11-13 20:52 -0500
http://bitbucket.org/pypy/pypy/changeset/e14d70573095/

Log:	add size attribute for scalars

diff --git a/pypy/module/micronumpy/interp_boxes.py b/pypy/module/micronumpy/interp_boxes.py
--- a/pypy/module/micronumpy/interp_boxes.py
+++ b/pypy/module/micronumpy/interp_boxes.py
@@ -284,6 +284,9 @@
     def descr_get_dtype(self, space):
         return self.get_dtype(space)
 
+    def descr_get_size(self, space):
+        return space.wrap(1)
+
     def descr_get_itemsize(self, space):
         return self.get_dtype(space).descr_get_itemsize(space)
 
@@ -554,6 +557,7 @@
     copy = interp2app(W_GenericBox.descr_copy),
 
     dtype = GetSetProperty(W_GenericBox.descr_get_dtype),
+    size = GetSetProperty(W_GenericBox.descr_get_size),
     itemsize = GetSetProperty(W_GenericBox.descr_get_itemsize),
     nbytes = GetSetProperty(W_GenericBox.descr_get_itemsize),
     shape = GetSetProperty(W_GenericBox.descr_get_shape),
diff --git a/pypy/module/micronumpy/test/test_scalar.py b/pypy/module/micronumpy/test/test_scalar.py
--- a/pypy/module/micronumpy/test/test_scalar.py
+++ b/pypy/module/micronumpy/test/test_scalar.py
@@ -85,6 +85,7 @@
         import numpypy as np
         value = np.dtype('int64').type(12345)
         assert value.dtype == np.dtype('int64')
+        assert value.size == 1
         assert value.itemsize == 8
         assert value.nbytes == 8
         assert value.shape == ()


More information about the pypy-commit mailing list