diff options
author | Mark Dickinson <dickinsm@gmail.com> | 2008-07-31 14:48:32 (GMT) |
---|---|---|
committer | Mark Dickinson <dickinsm@gmail.com> | 2008-07-31 14:48:32 (GMT) |
commit | 0f6414a0d6c4f09ec2ab88cef801b2475f6c84a3 (patch) | |
tree | 9d02b81d473c214903932b3014473aa940caac23 | |
parent | cda5ce24ed20a7e9f68282ac55b0d6949fe0c7db (diff) | |
download | cpython-0f6414a0d6c4f09ec2ab88cef801b2475f6c84a3.zip cpython-0f6414a0d6c4f09ec2ab88cef801b2475f6c84a3.tar.gz cpython-0f6414a0d6c4f09ec2ab88cef801b2475f6c84a3.tar.bz2 |
Rename testSum to testFsum and move it to proper place in test_math.py
-rw-r--r-- | Lib/test/test_math.py | 193 |
1 files changed, 96 insertions, 97 deletions
diff --git a/Lib/test/test_math.py b/Lib/test/test_math.py index 9f5fdba..c82c775 100644 --- a/Lib/test/test_math.py +++ b/Lib/test/test_math.py @@ -364,6 +364,102 @@ class MathTests(unittest.TestCase): self.assertEquals(math.frexp(NINF)[0], NINF) self.assert_(math.isnan(math.frexp(NAN)[0])) + def testFsum(self): + # math.fsum relies on exact rounding for correct operation. + # There's a known problem with IA32 floating-point that causes + # inexact rounding in some situations, and will cause the + # math.fsum tests below to fail; see issue #2937. On non IEEE + # 754 platforms, and on IEEE 754 platforms that exhibit the + # problem described in issue #2937, we simply skip the whole + # test. + + if not float.__getformat__("double").startswith("IEEE"): + return + + # on IEEE 754 compliant machines, both of the expressions + # below should round to 10000000000000002.0. + if 1e16+2.0 != 1e16+2.9999: + return + + # Python version of math.fsum, for comparison. Uses a + # different algorithm based on frexp, ldexp and integer + # arithmetic. + from sys import float_info + mant_dig = float_info.mant_dig + etiny = float_info.min_exp - mant_dig + + def msum(iterable): + """Full precision summation. Compute sum(iterable) without any + intermediate accumulation of error. Based on the 'lsum' function + at http://code.activestate.com/recipes/393090/ + + """ + tmant, texp = 0, 0 + for x in iterable: + mant, exp = math.frexp(x) + mant, exp = int(math.ldexp(mant, mant_dig)), exp - mant_dig + if texp > exp: + tmant <<= texp-exp + texp = exp + else: + mant <<= exp-texp + tmant += mant + # Round tmant * 2**texp to a float. The original recipe + # used float(str(tmant)) * 2.0**texp for this, but that's + # a little unsafe because str -> float conversion can't be + # relied upon to do correct rounding on all platforms. + tail = max(len(bin(abs(tmant)))-2 - mant_dig, etiny - texp) + if tail > 0: + h = 1 << (tail-1) + tmant = tmant // (2*h) + bool(tmant & h and tmant & 3*h-1) + texp += tail + return math.ldexp(tmant, texp) + + test_values = [ + ([], 0.0), + ([0.0], 0.0), + ([1e100, 1.0, -1e100, 1e-100, 1e50, -1.0, -1e50], 1e-100), + ([2.0**53, -0.5, -2.0**-54], 2.0**53-1.0), + ([2.0**53, 1.0, 2.0**-100], 2.0**53+2.0), + ([2.0**53+10.0, 1.0, 2.0**-100], 2.0**53+12.0), + ([2.0**53-4.0, 0.5, 2.0**-54], 2.0**53-3.0), + ([1./n for n in range(1, 1001)], + float.fromhex('0x1.df11f45f4e61ap+2')), + ([(-1.)**n/n for n in range(1, 1001)], + float.fromhex('-0x1.62a2af1bd3624p-1')), + ([1.7**(i+1)-1.7**i for i in range(1000)] + [-1.7**1000], -1.0), + ([1e16, 1., 1e-16], 10000000000000002.0), + ([1e16-2., 1.-2.**-53, -(1e16-2.), -(1.-2.**-53)], 0.0), + # exercise code for resizing partials array + ([2.**n - 2.**(n+50) + 2.**(n+52) for n in range(-1074, 972, 2)] + + [-2.**1022], + float.fromhex('0x1.5555555555555p+970')), + ] + + for i, (vals, expected) in enumerate(test_values): + try: + actual = math.fsum(vals) + except OverflowError: + self.fail("test %d failed: got OverflowError, expected %r " + "for math.fsum(%.100r)" % (i, expected, vals)) + except ValueError: + self.fail("test %d failed: got ValueError, expected %r " + "for math.fsum(%.100r)" % (i, expected, vals)) + self.assertEqual(actual, expected) + + from random import random, gauss, shuffle + for j in xrange(1000): + vals = [7, 1e100, -7, -1e100, -9e-20, 8e-20] * 10 + s = 0 + for i in xrange(200): + v = gauss(0, random()) ** 7 - s + s += v + vals.append(v) + shuffle(vals) + + s = msum(vals) + self.assertEqual(msum(vals), math.fsum(vals)) + def testHypot(self): self.assertRaises(TypeError, math.hypot) self.ftest('hypot(0,0)', math.hypot(0,0), 0) @@ -645,103 +741,6 @@ class MathTests(unittest.TestCase): self.assertRaises(ValueError, math.sqrt, NINF) self.assert_(math.isnan(math.sqrt(NAN))) - def testSum(self): - # math.fsum relies on exact rounding for correct operation. - # There's a known problem with IA32 floating-point that causes - # inexact rounding in some situations, and will cause the - # math.fsum tests below to fail; see issue #2937. On non IEEE - # 754 platforms, and on IEEE 754 platforms that exhibit the - # problem described in issue #2937, we simply skip the whole - # test. - - if not float.__getformat__("double").startswith("IEEE"): - return - - # on IEEE 754 compliant machines, both of the expressions - # below should round to 10000000000000002.0. - if 1e16+2.0 != 1e16+2.9999: - return - - # Python version of math.fsum, for comparison. Uses a - # different algorithm based on frexp, ldexp and integer - # arithmetic. - from sys import float_info - mant_dig = float_info.mant_dig - etiny = float_info.min_exp - mant_dig - - def msum(iterable): - """Full precision summation. Compute sum(iterable) without any - intermediate accumulation of error. Based on the 'lsum' function - at http://code.activestate.com/recipes/393090/ - - """ - tmant, texp = 0, 0 - for x in iterable: - mant, exp = math.frexp(x) - mant, exp = int(math.ldexp(mant, mant_dig)), exp - mant_dig - if texp > exp: - tmant <<= texp-exp - texp = exp - else: - mant <<= exp-texp - tmant += mant - # Round tmant * 2**texp to a float. The original recipe - # used float(str(tmant)) * 2.0**texp for this, but that's - # a little unsafe because str -> float conversion can't be - # relied upon to do correct rounding on all platforms. - tail = max(len(bin(abs(tmant)))-2 - mant_dig, etiny - texp) - if tail > 0: - h = 1 << (tail-1) - tmant = tmant // (2*h) + bool(tmant & h and tmant & 3*h-1) - texp += tail - return math.ldexp(tmant, texp) - - test_values = [ - ([], 0.0), - ([0.0], 0.0), - ([1e100, 1.0, -1e100, 1e-100, 1e50, -1.0, -1e50], 1e-100), - ([2.0**53, -0.5, -2.0**-54], 2.0**53-1.0), - ([2.0**53, 1.0, 2.0**-100], 2.0**53+2.0), - ([2.0**53+10.0, 1.0, 2.0**-100], 2.0**53+12.0), - ([2.0**53-4.0, 0.5, 2.0**-54], 2.0**53-3.0), - ([1./n for n in range(1, 1001)], - float.fromhex('0x1.df11f45f4e61ap+2')), - ([(-1.)**n/n for n in range(1, 1001)], - float.fromhex('-0x1.62a2af1bd3624p-1')), - ([1.7**(i+1)-1.7**i for i in range(1000)] + [-1.7**1000], -1.0), - ([1e16, 1., 1e-16], 10000000000000002.0), - ([1e16-2., 1.-2.**-53, -(1e16-2.), -(1.-2.**-53)], 0.0), - # exercise code for resizing partials array - ([2.**n - 2.**(n+50) + 2.**(n+52) for n in range(-1074, 972, 2)] + - [-2.**1022], - float.fromhex('0x1.5555555555555p+970')), - ] - - for i, (vals, expected) in enumerate(test_values): - try: - actual = math.fsum(vals) - except OverflowError: - self.fail("test %d failed: got OverflowError, expected %r " - "for math.fsum(%.100r)" % (i, expected, vals)) - except ValueError: - self.fail("test %d failed: got ValueError, expected %r " - "for math.fsum(%.100r)" % (i, expected, vals)) - self.assertEqual(actual, expected) - - from random import random, gauss, shuffle - for j in xrange(1000): - vals = [7, 1e100, -7, -1e100, -9e-20, 8e-20] * 10 - s = 0 - for i in xrange(200): - v = gauss(0, random()) ** 7 - s - s += v - vals.append(v) - shuffle(vals) - - s = msum(vals) - self.assertEqual(msum(vals), math.fsum(vals)) - - def testTan(self): self.assertRaises(TypeError, math.tan) self.ftest('tan(0)', math.tan(0), 0) |