Author: Vincent Legoll <[email protected]>
Branch: fix-1674
Changeset: r81483:bd0b3e017721
Date: 2015-12-23 17:45 +0100
http://bitbucket.org/pypy/pypy/changeset/bd0b3e017721/
Log: Add more test cases from numpy doc 1 fail 1 pass
diff --git a/pypy/module/micronumpy/test/test_ndarray.py
b/pypy/module/micronumpy/test/test_ndarray.py
--- a/pypy/module/micronumpy/test/test_ndarray.py
+++ b/pypy/module/micronumpy/test/test_ndarray.py
@@ -3285,6 +3285,40 @@
assert (data[m, 0] == array([ 9., 12., 3.])).all()
assert (data[array([1, 3, 4, 1]), 1] == array([4., 10., 13.,
4.])).all()
+ def test_multiple_advanced_integer_indexing(self):
+ # Taken from:
+ #
http://docs.scipy.org/doc/numpy-1.10.0/reference/arrays.indexing.html#advanced-indexing
+ from numpy import array, intp, arange
+ x = arange(12).reshape(4, 3)
+ assert (x == array([[ 0, 1, 2],
+ [ 3, 4, 5],
+ [ 6, 7, 8],
+ [ 9, 10, 11]])).all()
+ rows = array([[0, 0],
+ [3, 3]], dtype=intp)
+ columns = array([[0, 2],
+ [0, 2]], dtype=intp)
+ assert (x[(rows, columns)] == array([[ 0, 2], [ 9, 11]])).all()
+ assert (x[1:2, 1:3] == array([[4, 5]])).all()
+ assert (x[1:2, [1, 2]] == array([[4, 5]])).all()
+
+ def test_multiple_advanced_indexing(self):
+ # Taken from:
+ #
http://docs.scipy.org/doc/numpy-1.10.0/reference/arrays.indexing.html#advanced-indexing
+ from numpy import array, newaxis
+ x = array([[ 0, 1, 2],
+ [ 3, 4, 5],
+ [ 6, 7, 8],
+ [ 9, 10, 11]])
+ rows = ((x.sum(-1) % 2) == 0)
+ assert (rows == array([False, True, False, True], dtype=bool)).all()
+ rows = rows.nonzero()[0]
+ assert (rows == array([1, 3])).all()
+ columns = [0, 2]
+ assert (x[rows[:, newaxis], columns] == array([[ 3, 5], [ 9,
11]])).all()
+ columns = array([0, 2])
+ assert (x[rows[:, newaxis], columns] == array([[ 3, 5], [ 9,
11]])).all()
+
def test_ravel(self):
from numpy import arange
assert (arange(3).ravel() == arange(3)).all()
_______________________________________________
pypy-commit mailing list
[email protected]
https://mail.python.org/mailman/listinfo/pypy-commit
