summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJulian Taylor <juliantaylor108@gmail.com>2014-01-02 15:00:27 -0800
committerJulian Taylor <juliantaylor108@gmail.com>2014-01-02 15:00:27 -0800
commitb91d3f456040a39a37cc1281ba499cb4316d15ec (patch)
tree0a7582479e05f48598e0ce2569e69e6e1c94c429
parent7213b97ae2c16dcf6fdd692030614e102abecda8 (diff)
parentbd6d287dd73c483654cfb10f4cfab4dcc7548dd2 (diff)
downloadnumpy-b91d3f456040a39a37cc1281ba499cb4316d15ec.tar.gz
Merge pull request #4163 from juliantaylor/complex-norm-fix
BUG: fix complex norm of higher order
-rw-r--r--numpy/linalg/linalg.py2
-rw-r--r--numpy/linalg/tests/test_linalg.py12
2 files changed, 13 insertions, 1 deletions
diff --git a/numpy/linalg/linalg.py b/numpy/linalg/linalg.py
index dd586827e..a7e12aa08 100644
--- a/numpy/linalg/linalg.py
+++ b/numpy/linalg/linalg.py
@@ -2088,7 +2088,7 @@ def norm(x, ord=None, axis=None):
# because it will downcast to float64.
absx = abs(x)
else:
- absx = asfarray(x)
+ absx = x if isComplexType(x.dtype.type) else asfarray(x)
if absx.dtype is x.dtype:
absx = abs(absx)
else:
diff --git a/numpy/linalg/tests/test_linalg.py b/numpy/linalg/tests/test_linalg.py
index feb4c8224..0c21a4229 100644
--- a/numpy/linalg/tests/test_linalg.py
+++ b/numpy/linalg/tests/test_linalg.py
@@ -921,6 +921,18 @@ class _TestNorm(object):
x = np.array([-2 ** 31], dtype=np.int32)
old_assert_almost_equal(norm(x, ord=3), 2 ** 31, decimal=5)
+ def test_complex_high_ord(self):
+ # gh-4156
+ d = np.empty((2,), dtype=np.clongdouble)
+ d[0] = 6+7j
+ d[1] = -6+7j
+ res = 11.615898132184
+ old_assert_almost_equal(np.linalg.norm(d, ord=3), res, decimal=10)
+ d = d.astype(np.complex128)
+ old_assert_almost_equal(np.linalg.norm(d, ord=3), res, decimal=9)
+ d = d.astype(np.complex64)
+ old_assert_almost_equal(np.linalg.norm(d, ord=3), res, decimal=5)
+
class TestNormDouble(_TestNorm):
dt = np.double