[pypy-commit] pypy newindex: add simple (failing) tests for newindex

[MichaelBlume]

Author: Mike Blume <mike at loggly.com>
Branch: newindex
Changeset: r53751:b3eff16f15be
Date: 2012-03-14 15:22 -0700
http://bitbucket.org/pypy/pypy/changeset/b3eff16f15be/

Log:	add simple (failing) tests for newindex

diff --git a/pypy/module/micronumpy/test/test_numarray.py b/pypy/module/micronumpy/test/test_numarray.py
--- a/pypy/module/micronumpy/test/test_numarray.py
+++ b/pypy/module/micronumpy/test/test_numarray.py
@@ -374,6 +374,26 @@
         assert a[1] == 0.
         assert a[3] == 0.
 
+    def test_newindex(self):
+        from _numpypy import array
+        newindex = None
+        a = array(range(5))
+        b = array([range(5)])
+        assert (a[newindex] == b).all()
+
+    def test_newindex_slice(self):
+        from _numpypy import array
+        newindex = None
+
+        a = array(range(5))
+        b = array(range(1,5))
+        c = array([range(1,5)])
+        d = array([[x] for x in range(1,5)])
+
+        assert (a[1:] == b).all()
+        assert (a[1:,newindex] == d).all()
+        assert (a[newindex,1:] == c).all()
+
     def test_scalar(self):
         from _numpypy import array, dtype
         a = array(3)