[pypy-commit] pypy virtual-arguments: one more

[fijal]

Author: Maciej Fijalkowski <fijall at gmail.com>
Branch: virtual-arguments
Changeset: r56228:3c234f36e008
Date: 2012-07-19 18:27 +0200
http://bitbucket.org/pypy/pypy/changeset/3c234f36e008/

Log:	one more

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
@@ -1271,7 +1271,9 @@
     if w_ndmin is not None and not space.is_w(w_ndmin, space.w_None):
         ndmin = space.int_w(w_ndmin)
         if ndmin > shapelen:
-            shape = [1] * (ndmin - shapelen) + shape
+            lgt = (ndmin - shapelen) + shape
+            assert lgt >= 0
+            shape = [1] * lgt
             shapelen = ndmin
     arr = W_NDimArray(shape[:], dtype=dtype, order=order)
     arr_iter = arr.create_iter()