Author: Vincent Legoll <[email protected]>
Branch: fix-1674
Changeset: r81481:905da6af2d4c
Date: 2015-12-23 15:36 +0100
http://bitbucket.org/pypy/pypy/changeset/905da6af2d4c/
Log: Add more test cases
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
@@ -3406,8 +3406,14 @@
assert (array([])[[]] == []).all()
a = array([[1, 2], [3, 4], [5, 6]])
+ assert (a[[0, 0]] == [[1, 2], [1, 2]]).all()
assert (a[array([0, 0])] == [[1, 2], [1, 2]]).all()
+ assert (a[array([0, 0]), 0] == [[1, 1]]).all()
+ assert (a[array([0, 0]), [0]] == [[1, 1]]).all()
+ assert (a[array([0, 0]), array([0])] == [[1, 1]]).all()
+
assert (a[[[0, 1], [0, 0]]] == array([1, 3])).all()
+
assert (a[array([0, 2])] == [[1, 2], [5, 6]]).all()
assert (a[array([0, 2]), 1] == [2, 6]).all()
assert (a[array([0, 2]), array([1])] == [2, 6]).all()
_______________________________________________
pypy-commit mailing list
[email protected]
https://mail.python.org/mailman/listinfo/pypy-commit
