[pypy-commit] pypy numpy-refactor: nbytes & itemsize

[fijal]

Author: Maciej Fijalkowski <fijall at gmail.com>
Branch: numpy-refactor
Changeset: r57163:6ff55e1ac59b
Date: 2012-09-05 22:12 +0200
http://bitbucket.org/pypy/pypy/changeset/6ff55e1ac59b/

Log:	nbytes & itemsize

diff --git a/pypy/module/micronumpy/interp_numarray.py b/pypy/module/micronumpy/interp_numarray.py
--- a/pypy/module/micronumpy/interp_numarray.py
+++ b/pypy/module/micronumpy/interp_numarray.py
@@ -44,6 +44,12 @@
     def descr_get_ndim(self, space):
         return space.wrap(len(self.get_shape()))
 
+    def descr_get_itemsize(self, space):
+        return space.wrap(self.get_dtype().itemtype.get_element_size())
+
+    def descr_get_nbytes(self, space):
+        return space.wrap(self.get_size() * self.get_dtype().itemtype.get_element_size())
+
     def descr_getitem(self, space, w_idx):
         if (isinstance(w_idx, W_NDimArray) and w_idx.get_shape() == self.get_shape() and
             w_idx.get_dtype().is_bool_type()):
@@ -405,6 +411,8 @@
                            W_NDimArray.descr_set_shape),
     ndim = GetSetProperty(W_NDimArray.descr_get_ndim),
     size = GetSetProperty(W_NDimArray.descr_get_size),
+    itemsize = GetSetProperty(W_NDimArray.descr_get_itemsize),
+    nbytes = GetSetProperty(W_NDimArray.descr_get_nbytes),
 
     mean = interp2app(W_NDimArray.descr_mean),
     sum = interp2app(W_NDimArray.descr_sum),