[pypy-commit] pypy refactor-signature: Fixes.

[fijal]

Author: Maciej Fijalkowski <fijall at gmail.com>
Branch: refactor-signature
Changeset: r50660:0e374fa5eab3
Date: 2011-12-18 18:02 +0200
http://bitbucket.org/pypy/pypy/changeset/0e374fa5eab3/

Log:	Fixes.

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
@@ -970,7 +970,7 @@
         return array
 
     def create_sig(self):
-        return signature.ViewSignature(self.parent.create_sig())
+        return signature.ViewSignature(self.dtype)
 
     def setshape(self, space, new_shape):
         if len(self.shape) < 1:
diff --git a/pypy/module/micronumpy/signature.py b/pypy/module/micronumpy/signature.py
--- a/pypy/module/micronumpy/signature.py
+++ b/pypy/module/micronumpy/signature.py
@@ -203,6 +203,14 @@
         allnumbers.append(no)
         self.iter_no = no
 
+    def _create_iter(self, iterlist, arraylist, arr):
+        from pypy.module.micronumpy.interp_numarray import ConcreteArray
+        assert isinstance(arr, ConcreteArray)
+        if self.iter_no >= len(iterlist):
+            iterlist.append(ViewIterator(arr))
+        if self.array_no >= len(arraylist):
+            arraylist.append(arr.storage)
+
 class FlatiterSignature(ViewSignature):
     def debug_repr(self):
         return 'FlatIter(%s)' % self.child.debug_repr()