Convert test_math to unittest.

This commit is contained in:
Georg Brandl 2006-10-28 13:51:49 +00:00
parent 856b446793
commit 2f03760a98
2 changed files with 174 additions and 196 deletions

View File

@ -1,28 +0,0 @@
test_math
math module, testing with eps 1e-05
constants
acos
asin
atan
atan2
ceil
cos
cosh
degrees
exp
fabs
floor
fmod
frexp
hypot
ldexp
log
log10
modf
pow
radians
sin
sinh
sqrt
tan
tanh

View File

@ -1,208 +1,214 @@
# Python test set -- math module # Python test set -- math module
# XXXX Should not do tests around zero only # XXXX Should not do tests around zero only
from test.test_support import TestFailed, verbose from test.test_support import run_unittest, verbose
import unittest
import math
seps='1e-05' seps='1e-05'
eps = eval(seps) eps = eval(seps)
print 'math module, testing with eps', seps
import math
def testit(name, value, expected): class MathTests(unittest.TestCase):
if abs(value-expected) > eps:
raise TestFailed, '%s returned %f, expected %f'%\
(name, value, expected)
print 'constants' def ftest(self, name, value, expected):
testit('pi', math.pi, 3.1415926) if abs(value-expected) > eps:
testit('e', math.e, 2.7182818) self.fail('%s returned %f, expected %f'%\
(name, value, expected))
print 'acos' def testConstants(self):
testit('acos(-1)', math.acos(-1), math.pi) self.ftest('pi', math.pi, 3.1415926)
testit('acos(0)', math.acos(0), math.pi/2) self.ftest('e', math.e, 2.7182818)
testit('acos(1)', math.acos(1), 0)
print 'asin' def testAcos(self):
testit('asin(-1)', math.asin(-1), -math.pi/2) self.ftest('acos(-1)', math.acos(-1), math.pi)
testit('asin(0)', math.asin(0), 0) self.ftest('acos(0)', math.acos(0), math.pi/2)
testit('asin(1)', math.asin(1), math.pi/2) self.ftest('acos(1)', math.acos(1), 0)
print 'atan' def testAsin(self):
testit('atan(-1)', math.atan(-1), -math.pi/4) self.ftest('asin(-1)', math.asin(-1), -math.pi/2)
testit('atan(0)', math.atan(0), 0) self.ftest('asin(0)', math.asin(0), 0)
testit('atan(1)', math.atan(1), math.pi/4) self.ftest('asin(1)', math.asin(1), math.pi/2)
print 'atan2' def testAtan(self):
testit('atan2(-1, 0)', math.atan2(-1, 0), -math.pi/2) self.ftest('atan(-1)', math.atan(-1), -math.pi/4)
testit('atan2(-1, 1)', math.atan2(-1, 1), -math.pi/4) self.ftest('atan(0)', math.atan(0), 0)
testit('atan2(0, 1)', math.atan2(0, 1), 0) self.ftest('atan(1)', math.atan(1), math.pi/4)
testit('atan2(1, 1)', math.atan2(1, 1), math.pi/4)
testit('atan2(1, 0)', math.atan2(1, 0), math.pi/2)
print 'ceil' def testAtan2(self):
testit('ceil(0.5)', math.ceil(0.5), 1) self.ftest('atan2(-1, 0)', math.atan2(-1, 0), -math.pi/2)
testit('ceil(1.0)', math.ceil(1.0), 1) self.ftest('atan2(-1, 1)', math.atan2(-1, 1), -math.pi/4)
testit('ceil(1.5)', math.ceil(1.5), 2) self.ftest('atan2(0, 1)', math.atan2(0, 1), 0)
testit('ceil(-0.5)', math.ceil(-0.5), 0) self.ftest('atan2(1, 1)', math.atan2(1, 1), math.pi/4)
testit('ceil(-1.0)', math.ceil(-1.0), -1) self.ftest('atan2(1, 0)', math.atan2(1, 0), math.pi/2)
testit('ceil(-1.5)', math.ceil(-1.5), -1)
print 'cos' def testCeil(self):
testit('cos(-pi/2)', math.cos(-math.pi/2), 0) self.ftest('ceil(0.5)', math.ceil(0.5), 1)
testit('cos(0)', math.cos(0), 1) self.ftest('ceil(1.0)', math.ceil(1.0), 1)
testit('cos(pi/2)', math.cos(math.pi/2), 0) self.ftest('ceil(1.5)', math.ceil(1.5), 2)
testit('cos(pi)', math.cos(math.pi), -1) self.ftest('ceil(-0.5)', math.ceil(-0.5), 0)
self.ftest('ceil(-1.0)', math.ceil(-1.0), -1)
self.ftest('ceil(-1.5)', math.ceil(-1.5), -1)
print 'cosh' def testCos(self):
testit('cosh(0)', math.cosh(0), 1) self.ftest('cos(-pi/2)', math.cos(-math.pi/2), 0)
testit('cosh(2)-2*cosh(1)**2', math.cosh(2)-2*math.cosh(1)**2, -1) # Thanks to Lambert self.ftest('cos(0)', math.cos(0), 1)
self.ftest('cos(pi/2)', math.cos(math.pi/2), 0)
self.ftest('cos(pi)', math.cos(math.pi), -1)
print 'degrees' def testCosh(self):
testit('degrees(pi)', math.degrees(math.pi), 180.0) self.ftest('cosh(0)', math.cosh(0), 1)
testit('degrees(pi/2)', math.degrees(math.pi/2), 90.0) self.ftest('cosh(2)-2*cosh(1)**2', math.cosh(2)-2*math.cosh(1)**2, -1) # Thanks to Lambert
testit('degrees(-pi/4)', math.degrees(-math.pi/4), -45.0)
print 'exp' def testDegrees(self):
testit('exp(-1)', math.exp(-1), 1/math.e) self.ftest('degrees(pi)', math.degrees(math.pi), 180.0)
testit('exp(0)', math.exp(0), 1) self.ftest('degrees(pi/2)', math.degrees(math.pi/2), 90.0)
testit('exp(1)', math.exp(1), math.e) self.ftest('degrees(-pi/4)', math.degrees(-math.pi/4), -45.0)
print 'fabs' def testExp(self):
testit('fabs(-1)', math.fabs(-1), 1) self.ftest('exp(-1)', math.exp(-1), 1/math.e)
testit('fabs(0)', math.fabs(0), 0) self.ftest('exp(0)', math.exp(0), 1)
testit('fabs(1)', math.fabs(1), 1) self.ftest('exp(1)', math.exp(1), math.e)
print 'floor' def testFabs(self):
testit('floor(0.5)', math.floor(0.5), 0) self.ftest('fabs(-1)', math.fabs(-1), 1)
testit('floor(1.0)', math.floor(1.0), 1) self.ftest('fabs(0)', math.fabs(0), 0)
testit('floor(1.5)', math.floor(1.5), 1) self.ftest('fabs(1)', math.fabs(1), 1)
testit('floor(-0.5)', math.floor(-0.5), -1)
testit('floor(-1.0)', math.floor(-1.0), -1)
testit('floor(-1.5)', math.floor(-1.5), -2)
print 'fmod' def testFloor(self):
testit('fmod(10,1)', math.fmod(10,1), 0) self.ftest('floor(0.5)', math.floor(0.5), 0)
testit('fmod(10,0.5)', math.fmod(10,0.5), 0) self.ftest('floor(1.0)', math.floor(1.0), 1)
testit('fmod(10,1.5)', math.fmod(10,1.5), 1) self.ftest('floor(1.5)', math.floor(1.5), 1)
testit('fmod(-10,1)', math.fmod(-10,1), 0) self.ftest('floor(-0.5)', math.floor(-0.5), -1)
testit('fmod(-10,0.5)', math.fmod(-10,0.5), 0) self.ftest('floor(-1.0)', math.floor(-1.0), -1)
testit('fmod(-10,1.5)', math.fmod(-10,1.5), -1) self.ftest('floor(-1.5)', math.floor(-1.5), -2)
print 'frexp' def testFmod(self):
def testfrexp(name, (mant, exp), (emant, eexp)): self.ftest('fmod(10,1)', math.fmod(10,1), 0)
if abs(mant-emant) > eps or exp != eexp: self.ftest('fmod(10,0.5)', math.fmod(10,0.5), 0)
raise TestFailed, '%s returned %r, expected %r'%\ self.ftest('fmod(10,1.5)', math.fmod(10,1.5), 1)
(name, (mant, exp), (emant,eexp)) self.ftest('fmod(-10,1)', math.fmod(-10,1), 0)
self.ftest('fmod(-10,0.5)', math.fmod(-10,0.5), 0)
self.ftest('fmod(-10,1.5)', math.fmod(-10,1.5), -1)
testfrexp('frexp(-1)', math.frexp(-1), (-0.5, 1)) def testFrexp(self):
testfrexp('frexp(0)', math.frexp(0), (0, 0)) def testfrexp(name, (mant, exp), (emant, eexp)):
testfrexp('frexp(1)', math.frexp(1), (0.5, 1)) if abs(mant-emant) > eps or exp != eexp:
testfrexp('frexp(2)', math.frexp(2), (0.5, 2)) self.fail('%s returned %r, expected %r'%\
(name, (mant, exp), (emant,eexp)))
print 'hypot' testfrexp('frexp(-1)', math.frexp(-1), (-0.5, 1))
testit('hypot(0,0)', math.hypot(0,0), 0) testfrexp('frexp(0)', math.frexp(0), (0, 0))
testit('hypot(3,4)', math.hypot(3,4), 5) testfrexp('frexp(1)', math.frexp(1), (0.5, 1))
testfrexp('frexp(2)', math.frexp(2), (0.5, 2))
print 'ldexp' def testHypot(self):
testit('ldexp(0,1)', math.ldexp(0,1), 0) self.ftest('hypot(0,0)', math.hypot(0,0), 0)
testit('ldexp(1,1)', math.ldexp(1,1), 2) self.ftest('hypot(3,4)', math.hypot(3,4), 5)
testit('ldexp(1,-1)', math.ldexp(1,-1), 0.5)
testit('ldexp(-1,1)', math.ldexp(-1,1), -2)
print 'log' def testLdexp(self):
testit('log(1/e)', math.log(1/math.e), -1) self.ftest('ldexp(0,1)', math.ldexp(0,1), 0)
testit('log(1)', math.log(1), 0) self.ftest('ldexp(1,1)', math.ldexp(1,1), 2)
testit('log(e)', math.log(math.e), 1) self.ftest('ldexp(1,-1)', math.ldexp(1,-1), 0.5)
testit('log(32,2)', math.log(32,2), 5) self.ftest('ldexp(-1,1)', math.ldexp(-1,1), -2)
testit('log(10**40, 10)', math.log(10**40, 10), 40)
testit('log(10**40, 10**20)', math.log(10**40, 10**20), 2)
print 'log10' def testLog(self):
testit('log10(0.1)', math.log10(0.1), -1) self.ftest('log(1/e)', math.log(1/math.e), -1)
testit('log10(1)', math.log10(1), 0) self.ftest('log(1)', math.log(1), 0)
testit('log10(10)', math.log10(10), 1) self.ftest('log(e)', math.log(math.e), 1)
self.ftest('log(32,2)', math.log(32,2), 5)
self.ftest('log(10**40, 10)', math.log(10**40, 10), 40)
self.ftest('log(10**40, 10**20)', math.log(10**40, 10**20), 2)
print 'modf' def testLog10(self):
def testmodf(name, (v1, v2), (e1, e2)): self.ftest('log10(0.1)', math.log10(0.1), -1)
if abs(v1-e1) > eps or abs(v2-e2): self.ftest('log10(1)', math.log10(1), 0)
raise TestFailed, '%s returned %r, expected %r'%\ self.ftest('log10(10)', math.log10(10), 1)
(name, (v1,v2), (e1,e2))
testmodf('modf(1.5)', math.modf(1.5), (0.5, 1.0)) def testModf(self):
testmodf('modf(-1.5)', math.modf(-1.5), (-0.5, -1.0)) def testmodf(name, (v1, v2), (e1, e2)):
if abs(v1-e1) > eps or abs(v2-e2):
self.fail('%s returned %r, expected %r'%\
(name, (v1,v2), (e1,e2)))
print 'pow' testmodf('modf(1.5)', math.modf(1.5), (0.5, 1.0))
testit('pow(0,1)', math.pow(0,1), 0) testmodf('modf(-1.5)', math.modf(-1.5), (-0.5, -1.0))
testit('pow(1,0)', math.pow(1,0), 1)
testit('pow(2,1)', math.pow(2,1), 2) def testPow(self):
testit('pow(2,-1)', math.pow(2,-1), 0.5) self.ftest('pow(0,1)', math.pow(0,1), 0)
self.ftest('pow(1,0)', math.pow(1,0), 1)
self.ftest('pow(2,1)', math.pow(2,1), 2)
self.ftest('pow(2,-1)', math.pow(2,-1), 0.5)
print 'radians' def testRadians(self):
testit('radians(180)', math.radians(180), math.pi) self.ftest('radians(180)', math.radians(180), math.pi)
testit('radians(90)', math.radians(90), math.pi/2) self.ftest('radians(90)', math.radians(90), math.pi/2)
testit('radians(-45)', math.radians(-45), -math.pi/4) self.ftest('radians(-45)', math.radians(-45), -math.pi/4)
print 'sin' def testSin(self):
testit('sin(0)', math.sin(0), 0) self.ftest('sin(0)', math.sin(0), 0)
testit('sin(pi/2)', math.sin(math.pi/2), 1) self.ftest('sin(pi/2)', math.sin(math.pi/2), 1)
testit('sin(-pi/2)', math.sin(-math.pi/2), -1) self.ftest('sin(-pi/2)', math.sin(-math.pi/2), -1)
print 'sinh' def testSinh(self):
testit('sinh(0)', math.sinh(0), 0) self.ftest('sinh(0)', math.sinh(0), 0)
testit('sinh(1)**2-cosh(1)**2', math.sinh(1)**2-math.cosh(1)**2, -1) self.ftest('sinh(1)**2-cosh(1)**2', math.sinh(1)**2-math.cosh(1)**2, -1)
testit('sinh(1)+sinh(-1)', math.sinh(1)+math.sinh(-1), 0) self.ftest('sinh(1)+sinh(-1)', math.sinh(1)+math.sinh(-1), 0)
print 'sqrt' def testSqrt(self):
testit('sqrt(0)', math.sqrt(0), 0) self.ftest('sqrt(0)', math.sqrt(0), 0)
testit('sqrt(1)', math.sqrt(1), 1) self.ftest('sqrt(1)', math.sqrt(1), 1)
testit('sqrt(4)', math.sqrt(4), 2) self.ftest('sqrt(4)', math.sqrt(4), 2)
print 'tan' def testTan(self):
testit('tan(0)', math.tan(0), 0) self.ftest('tan(0)', math.tan(0), 0)
testit('tan(pi/4)', math.tan(math.pi/4), 1) self.ftest('tan(pi/4)', math.tan(math.pi/4), 1)
testit('tan(-pi/4)', math.tan(-math.pi/4), -1) self.ftest('tan(-pi/4)', math.tan(-math.pi/4), -1)
print 'tanh' def testTanh(self):
testit('tanh(0)', math.tanh(0), 0) self.ftest('tanh(0)', math.tanh(0), 0)
testit('tanh(1)+tanh(-1)', math.tanh(1)+math.tanh(-1), 0) self.ftest('tanh(1)+tanh(-1)', math.tanh(1)+math.tanh(-1), 0)
# RED_FLAG 16-Oct-2000 Tim # RED_FLAG 16-Oct-2000 Tim
# While 2.0 is more consistent about exceptions than previous releases, it # While 2.0 is more consistent about exceptions than previous releases, it
# still fails this part of the test on some platforms. For now, we only # still fails this part of the test on some platforms. For now, we only
# *run* test_exceptions() in verbose mode, so that this isn't normally # *run* test_exceptions() in verbose mode, so that this isn't normally
# tested. # tested.
def test_exceptions(): if verbose:
print 'exceptions' def test_exceptions(self):
try: try:
x = math.exp(-1000000000) x = math.exp(-1000000000)
except: except:
# mathmodule.c is failing to weed out underflows from libm, or # mathmodule.c is failing to weed out underflows from libm, or
# we've got an fp format with huge dynamic range # we've got an fp format with huge dynamic range
raise TestFailed("underflowing exp() should not have raised " self.fail("underflowing exp() should not have raised "
"an exception") "an exception")
if x != 0: if x != 0:
raise TestFailed("underflowing exp() should have returned 0") self.fail("underflowing exp() should have returned 0")
# If this fails, probably using a strict IEEE-754 conforming libm, and x # If this fails, probably using a strict IEEE-754 conforming libm, and x
# is +Inf afterwards. But Python wants overflows detected by default. # is +Inf afterwards. But Python wants overflows detected by default.
try: try:
x = math.exp(1000000000) x = math.exp(1000000000)
except OverflowError: except OverflowError:
pass pass
else: else:
raise TestFailed("overflowing exp() didn't trigger OverflowError") self.fail("overflowing exp() didn't trigger OverflowError")
# If this fails, it could be a puzzle. One odd possibility is that # If this fails, it could be a puzzle. One odd possibility is that
# mathmodule.c's macros are getting confused while comparing # mathmodule.c's macros are getting confused while comparing
# Inf (HUGE_VAL) to a NaN, and artificially setting errno to ERANGE # Inf (HUGE_VAL) to a NaN, and artificially setting errno to ERANGE
# as a result (and so raising OverflowError instead). # as a result (and so raising OverflowError instead).
try: try:
x = math.sqrt(-1.0) x = math.sqrt(-1.0)
except ValueError: except ValueError:
pass pass
else: else:
raise TestFailed("sqrt(-1) didn't raise ValueError") self.fail("sqrt(-1) didn't raise ValueError")
if verbose:
test_exceptions() def test_main():
run_unittest(MathTests)
if __name__ == '__main__':
test_main()