[pypy-commit] pypy refactor-signature: minor cleanup

[fijal]

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

Log:	minor cleanup

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
@@ -140,10 +140,13 @@
         from pypy.module.micronumpy.interp_numarray import ConcreteArray
         assert isinstance(arr, ConcreteArray)
         if self.iter_no >= len(iterlist):
-            iterlist.append(ArrayIterator(arr.size))
+            iterlist.append(self.allocate_iter(arr))
         if self.array_no >= len(arraylist):
             arraylist.append(arr.storage)
 
+    def allocate_iter(self, arr):
+        return ArrayIterator(arr.size)
+
     def eval(self, frame, arr):
         from pypy.module.micronumpy.interp_numarray import ConcreteArray
         assert isinstance(arr, ConcreteArray)
@@ -203,13 +206,8 @@
         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)
+    def allocate_iter(self, arr):
+        return ViewIterator(arr)
 
 class FlatiterSignature(ViewSignature):
     def debug_repr(self):