Maciej Fijalkowski avatar Maciej Fijalkowski committed b5668c7

more tests and a fix

Comments (0)

Files changed (2)

pypy/module/micronumpy/interp_iter.py

         for i in range(shapelen - 1, -1, -1):
             if self.indices[i] < self.shape[i] - 1:
                 self.indices[i] += 1
-                self.chunk_iters[i] = self.chunk_iters[i].next()
+                self.chunk_iters[i] = self.chunk_iters[i].next(shapelen)
                 break
             else:
                 self.indices[i] = 0

pypy/module/micronumpy/test/test_numarray.py

         raises(IndexError,'arange(3)[array([15])]')
         assert arange(3)[array([-3])] == 0
         raises(IndexError,'arange(3)[array([-15])]')
+        a = arange(10)
+        assert (a[array([3, 2, 1])] == [3, 2, 1]).all()
+        raises(IndexError, 'a[array([3, 2, 1, 15])]')
+        assert (a[array([3, 2, 5, 3])] == [3, 2, 5, 3]).all()
 
     def test_fill(self):
         from _numpypy import array
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.