[pypy-commit] pypy default: Export numpy.power and a basic test for it
nbm
noreply at buildbot.pypy.org
Tue Mar 13 02:09:01 CET 2012
Author: Neil Blakey-Milner <nbm at nxsy.org>
Branch:
Changeset: r53404:2242a6bc4d52
Date: 2012-03-12 17:55 -0700
http://bitbucket.org/pypy/pypy/changeset/2242a6bc4d52/
Log: Export numpy.power and a basic test for it
diff --git a/pypy/module/micronumpy/__init__.py b/pypy/module/micronumpy/__init__.py
--- a/pypy/module/micronumpy/__init__.py
+++ b/pypy/module/micronumpy/__init__.py
@@ -114,6 +114,7 @@
('log2', 'log2'),
('log10', 'log10'),
('log1p', 'log1p'),
+ ('power', 'power'),
]:
interpleveldefs[exposed] = "interp_ufuncs.get(space).%s" % impl
diff --git a/pypy/module/micronumpy/test/test_ufuncs.py b/pypy/module/micronumpy/test/test_ufuncs.py
--- a/pypy/module/micronumpy/test/test_ufuncs.py
+++ b/pypy/module/micronumpy/test/test_ufuncs.py
@@ -543,3 +543,17 @@
assert log1p(v) == float("-inf")
assert log1p(float('inf')) == float('inf')
assert (log1p([0, 1e-50, math.e - 1]) == [0, 1e-50, 1]).all()
+
+ def test_power(self):
+ import math
+ from _numpypy import power, array
+ a = array([1., 2., 3.])
+ b = power(a, 3)
+ for i in range(len(a)):
+ assert b[i] == a[i] ** 3
+
+ a = array([1., 2., 3.])
+ b = array([1., 2., 3.])
+ c = power(a, b)
+ for i in range(len(a)):
+ assert c[i] == a[i] ** b[i]
More information about the pypy-commit
mailing list