Commit 8941b9fd authored by Walter Dörwald's avatar Walter Dörwald

Port test_pow.py to PyUnit. From SF patch #662807

parent 91c79c45
test_pow
Testing integer mode...
Testing 2-argument pow() function...
Testing 3-argument pow() function...
Testing long integer mode...
Testing 2-argument pow() function...
Testing 3-argument pow() function...
Testing floating point mode...
Testing 3-argument pow() function...
The number in both columns should match.
3 3
-5 -5
-1 -1
5 5
-3 -3
-7 -7
3L 3L
-5L -5L
-1L -1L
5L 5L
-3L -3L
-7L -7L
import sys import test.test_support, unittest
from test import test_support
class PowTest(unittest.TestCase):
def powtest(type): def powtest(self, type):
if type != float: if type != float:
print " Testing 2-argument pow() function..." for i in range(-1000, 1000):
for i in range(-1000, 1000): self.assertEquals(pow(type(i), 0), 1)
if pow(type(i), 0) != 1: self.assertEquals(pow(type(i), 1), type(i))
raise ValueError, 'pow('+str(i)+',0) != 1' self.assertEquals(pow(type(0), 1), type(0))
if pow(type(i), 1) != type(i): self.assertEquals(pow(type(1), 1), type(1))
raise ValueError, 'pow('+str(i)+',1) != '+str(i)
if pow(type(0), 1) != type(0): for i in range(-100, 100):
raise ValueError, 'pow(0,'+str(i)+') != 0' self.assertEquals(pow(type(i), 3), i*i*i)
if pow(type(1), 1) != type(1):
raise ValueError, 'pow(1,'+str(i)+') != 1' pow2 = 1
for i in range(0,31):
for i in range(-100, 100): self.assertEquals(pow(2, i), pow2)
if pow(type(i), 3) != i*i*i: if i != 30 : pow2 = pow2*2
raise ValueError, 'pow('+str(i)+',3) != '+str(i*i*i)
for othertype in int, long:
pow2 = 1 for i in range(-10, 0) + range(1, 10):
for i in range(0,31): ii = type(i)
if pow(2, i) != pow2: for j in range(1, 11):
raise ValueError, 'pow(2,'+str(i)+') != '+str(pow2) jj = -othertype(j)
if i != 30 : pow2 = pow2*2
for othertype in int, long:
for i in range(-10, 0) + range(1, 10):
ii = type(i)
for j in range(1, 11):
jj = -othertype(j)
try:
pow(ii, jj) pow(ii, jj)
except ValueError:
raise ValueError, "pow(%s, %s) failed" % (ii, jj) for othertype in int, long, float:
for i in range(1, 100):
for othertype in int, long, float: zero = type(0)
for i in range(1, 100): exp = -othertype(i/10.0)
zero = type(0) if exp == 0:
exp = -othertype(i/10.0) continue
if exp == 0: self.assertRaises(ZeroDivisionError, pow, zero, exp)
continue
try: il, ih = -20, 20
pow(zero, exp) jl, jh = -5, 5
except ZeroDivisionError: kl, kh = -10, 10
pass # taking zero to any negative exponent should fail asseq = self.assertEqual
else: if type == float:
raise ValueError, "pow(%s, %s) did not fail" % (zero, exp) il = 1
asseq = self.assertAlmostEqual
print " Testing 3-argument pow() function..." elif type == int:
il, ih = -20, 20 jl = 0
jl, jh = -5, 5 elif type == long:
kl, kh = -10, 10 jl, jh = 0, 15
compare = cmp for i in range(il, ih+1):
if type == float: for j in range(jl, jh+1):
il = 1 for k in range(kl, kh+1):
compare = test_support.fcmp if k != 0:
elif type == int: if type == float or j < 0:
jl = 0 self.assertRaises(TypeError, pow, type(i), j, k)
elif type == long: continue
jl, jh = 0, 15 asseq(
for i in range(il, ih+1): pow(type(i),j,k),
for j in range(jl, jh+1): pow(type(i),j)% type(k)
for k in range(kl, kh+1): )
if k != 0:
if type == float or j < 0: def test_powint(self):
try: self.powtest(int)
pow(type(i),j,k)
except TypeError: def test_powlong(self):
pass self.powtest(long)
else:
raise ValueError, "expected TypeError from " + \ def test_powfloat(self):
"pow%r" % ((type(i), j, k),) self.powtest(float)
continue
if compare(pow(type(i),j,k), pow(type(i),j)% type(k)): def test_other(self):
raise ValueError, "pow(" +str(i)+ "," +str(j)+ \ # Other tests-- not very systematic
"," +str(k)+ ") != pow(" +str(i)+ "," + \ self.assertEquals(pow(3,3) % 8, pow(3,3,8))
str(j)+ ") % " +str(k) self.assertEquals(pow(3,3) % -8, pow(3,3,-8))
self.assertEquals(pow(3,2) % -2, pow(3,2,-2))
self.assertEquals(pow(-3,3) % 8, pow(-3,3,8))
print 'Testing integer mode...' self.assertEquals(pow(-3,3) % -8, pow(-3,3,-8))
powtest(int) self.assertEquals(pow(5,2) % -8, pow(5,2,-8))
print 'Testing long integer mode...'
powtest(long) self.assertEquals(pow(3L,3L) % 8, pow(3L,3L,8))
print 'Testing floating point mode...' self.assertEquals(pow(3L,3L) % -8, pow(3L,3L,-8))
powtest(float) self.assertEquals(pow(3L,2) % -2, pow(3L,2,-2))
self.assertEquals(pow(-3L,3L) % 8, pow(-3L,3L,8))
# Other tests-- not very systematic self.assertEquals(pow(-3L,3L) % -8, pow(-3L,3L,-8))
self.assertEquals(pow(5L,2) % -8, pow(5L,2,-8))
print 'The number in both columns should match.'
print `pow(3,3) % 8`, `pow(3,3,8)` for i in range(-10, 11):
print `pow(3,3) % -8`, `pow(3,3,-8)` for j in range(0, 6):
print `pow(3,2) % -2`, `pow(3,2,-2)` for k in range(-7, 11):
print `pow(-3,3) % 8`, `pow(-3,3,8)` if j >= 0 and k != 0:
print `pow(-3,3) % -8`, `pow(-3,3,-8)` self.assertEquals(
print `pow(5,2) % -8`, `pow(5,2,-8)` pow(i,j) % k,
print pow(i,j,k)
)
print `pow(3L,3L) % 8`, `pow(3L,3L,8)` if j >= 0 and k != 0:
print `pow(3L,3L) % -8`, `pow(3L,3L,-8)` self.assertEquals(
print `pow(3L,2) % -2`, `pow(3L,2,-2)` pow(long(i),j) % k,
print `pow(-3L,3L) % 8`, `pow(-3L,3L,8)` pow(long(i),j,k)
print `pow(-3L,3L) % -8`, `pow(-3L,3L,-8)` )
print `pow(5L,2) % -8`, `pow(5L,2,-8)`
print def test_bug643260(self):
class TestRpow:
print def __rpow__(self, other):
return None
for i in range(-10, 11): None ** TestRpow() # Won't fail when __rpow__ invoked. SF bug #643260.
for j in range(0, 6):
for k in range(-7, 11):
if j >= 0 and k != 0: def test_main():
o = pow(i,j) % k suite = unittest.TestSuite()
n = pow(i,j,k) suite.addTest(unittest.makeSuite(PowTest))
if o != n: print 'Integer mismatch:', i,j,k test.test_support.run_suite(suite)
if j >= 0 and k != 0:
o = pow(long(i),j) % k if __name__ == "__main__":
n = pow(long(i),j,k) test_main()
if o != n: print 'Integer mismatch:', i,j,k
class TestRpow:
def __rpow__(self, other):
return None
None ** TestRpow() # Won't fail when __rpow__ invoked. SF bug #643260.
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment