Author: mattip <[email protected]>
Branch: fix_indexing_by_numpy_int
Changeset: r83047:31bb9ad05cae
Date: 2016-03-14 20:38 +0200
http://bitbucket.org/pypy/pypy/changeset/31bb9ad05cae/
Log: add passing test
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
@@ -3397,22 +3397,21 @@
a.itemset(1, 2, 100)
assert a[1, 2] == 100
- def test_index_int64(self):
+ def test_index_int(self):
import numpy as np
- res = np.array([10, 20, 30])[np.int64(1)]
-
+ a = np.array([10, 20, 30])
+ res = a[np.int64(1)]
assert isinstance(res, np.int64)
- assert not isinstance(res, np.ndarray)
assert res == 20
-
- def test_index_int32(self):
- import numpy as np
- res = np.array([10, 20, 30])[np.int32(0)]
-
+ res = a[np.int32(0)]
assert isinstance(res, np.int64)
- assert not isinstance(res, np.ndarray)
assert res == 10
+ b = a.astype(float)
+ res = b[np.int64(1)]
+ assert res == 20.0
+ assert isinstance(res, np.float64)
+
def test_index(self):
import numpy as np
a = np.array([1], np.uint16)
_______________________________________________
pypy-commit mailing list
[email protected]
https://mail.python.org/mailman/listinfo/pypy-commit
