Author: Vincent Legoll <vincent.leg...@idgrilles.fr> Branch: fix-1674 Changeset: r81484:63607714b6bf Date: 2015-12-23 17:55 +0100 http://bitbucket.org/pypy/pypy/changeset/63607714b6bf/
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 @@ -3263,11 +3263,13 @@ assert (data[m] == array([[ 0., 1., 2., 3., 4.], [ 10., 11., 12., 13., 14.]])).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() def test_dual_indexing_selecting(self): _______________________________________________ pypy-commit mailing list pypy-commit@python.org https://mail.python.org/mailman/listinfo/pypy-commit