Author: Vincent Legoll <[email protected]>
Branch: fix-1674
Changeset: r81490:cc2bb5dd979c
Date: 2015-12-29 11:06 +0100
http://bitbucket.org/pypy/pypy/changeset/cc2bb5dd979c/
Log: Add even 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
@@ -3263,14 +3263,23 @@
assert (data[m] == array([[ 0., 1., 2., 3., 4.],
[ 10., 11., 12., 13., 14.]])).all()
assert (data[m, 0] == array([ 0., 10.])).all()
+ assert (data[(m, 0, Ellipsis)] == array([ 0., 10.])).all()
+ assert (data[m, 0, ...] == array([ 0., 10.])).all()
+ assert (data[m, ..., 0] == array([ 0., 10.])).all()
+ assert (data[..., m, 0] == array([ 0., 10.])).all()
+
# More of the same type of cases
assert (data[m, 0:1] == array([[0.], [10.]])).all()
assert (data[m, 0:1, ...] == array([[0.], [10.]])).all()
+ assert (data[m, ..., 0:1] == array([[0.], [10.]])).all()
assert (data[..., m, 0:1] == array([[0.], [10.]])).all()
# Assume False for missing elements of the bool index array
assert (data[0, m] == array([ 0., 2.])).all()
assert (data[0:1, m] == array([[ 0., 2.]])).all()
assert (data[0, m, None] == array([[0.], [2.]])).all()
+ assert (data[0:1, m, None] == array([[[0.], [2.]]])).all()
+ assert (data[None, m, 0:1] == array([[[0.], [10.]]])).all()
+ assert (data[m, m] == array([ 0., 12.])).all()
def test_dual_indexing_selecting(self):
from numpy import arange, array
_______________________________________________
pypy-commit mailing list
[email protected]
https://mail.python.org/mailman/listinfo/pypy-commit
