Author: Mike Blume <[email protected]>
Branch: newindex
Changeset: r53757:cb80a924de4c
Date: 2012-03-14 17:25 -0700
http://bitbucket.org/pypy/pypy/changeset/cb80a924de4c/
Log: still more tests
slice then newaxis, newaxis then slice
diff --git a/pypy/module/micronumpy/test/test_numarray.py
b/pypy/module/micronumpy/test/test_numarray.py
--- a/pypy/module/micronumpy/test/test_numarray.py
+++ b/pypy/module/micronumpy/test/test_numarray.py
@@ -407,10 +407,24 @@
from numpypy.core.numeric import newaxis
a = array(range(5))
- b = (a + a)[None]
+ b = (a + a)[newaxis]
c = array([[0, 2, 4, 6, 8]])
assert (b == c).all()
+ def test_newaxis_then_slice(self):
+ from _numpypy import array
+ from numpypy.core.numeric import newaxis
+ a = array(range(5))
+ b = a[newaxis]
+ assert (b[0,1:] == a[1:]).all()
+
+ def test_slice_then_newaxis(self):
+ from _numpypy import array
+ from numpypy.core.numeric import newaxis
+ a = array(range(5))
+ b = a[2:]
+ assert (b[newaxis] == [[2, 3, 4]]).all()
+
def test_scalar(self):
from _numpypy import array, dtype
a = array(3)
_______________________________________________
pypy-commit mailing list
[email protected]
http://mail.python.org/mailman/listinfo/pypy-commit
