Author: mattip <[email protected]>
Branch:
Changeset: r58110:f41eea130430
Date: 2012-10-13 21:12 +0200
http://bitbucket.org/pypy/pypy/changeset/f41eea130430/
Log: test, fix c_div for nan in denom
diff --git a/pypy/rlib/rcomplex.py b/pypy/rlib/rcomplex.py
--- a/pypy/rlib/rcomplex.py
+++ b/pypy/rlib/rcomplex.py
@@ -57,6 +57,9 @@
denom = r2 + i2 * ratio
rr = (r1 + i1 * ratio) / denom
ir = (i1 - r1 * ratio) / denom
+ elif isnan(r2):
+ rr = NAN
+ ir = NAN
else:
ratio = r2 / i2
denom = r2 * ratio + i2
diff --git a/pypy/rlib/test/test_rcomplex.py b/pypy/rlib/test/test_rcomplex.py
--- a/pypy/rlib/test/test_rcomplex.py
+++ b/pypy/rlib/test/test_rcomplex.py
@@ -33,6 +33,9 @@
]:
assert c.c_mul(c1, c2) == result
+def test_div():
+ c.c_div((2., 3.), (float('nan'), 0.)) == (float('nan'), float('nan'))
+
def parse_testfile2(fname):
"""Parse a file with test values
_______________________________________________
pypy-commit mailing list
[email protected]
http://mail.python.org/mailman/listinfo/pypy-commit
