1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
"""Misc complex tests
Made for Jython.
"""
import math
import operator
import os
import re
import unittest
from test import test_support
INF, NINF, NAN = map(float, ("inf", "-inf", "nan"))
class ComplexTest(unittest.TestCase):
def test_dunder_coerce(self):
self.assertEqual(complex.__coerce__(1+1j, None), NotImplemented)
self.assertRaises(TypeError, complex.__coerce__, None, 1+2j)
def test_pow(self):
class Foo(object):
def __rpow__(self, other):
return other ** 2
# regression in 2.5 alphas
self.assertEqual((4+0j) ** Foo(), (16+0j))
def test___nonzero__(self):
self.assertTrue(0.25+0j)
self.assertTrue(25j)
def test_abs_big(self):
# These are close to overflow but don't
close = [ complex( 1.794e+308, 0.000e+00),
complex( 1.119e+308, 1.403e+308),
complex(-3.992e+307, 1.749e+308),
complex(-1.617e+308, 7.785e+307),
complex(-1.617e+308,-7.785e+307),
complex(-3.992e+307,-1.749e+308) ]
# These are a little bigger and do overflow
over = [ complex( 1.130e+308, 1.417e+308),
complex(-4.032e+307, 1.767e+308),
complex(-1.633e+308, 7.863e+307),
complex(-1.633e+308,-7.863e+307),
complex(-4.032e+307,-1.767e+308) ]
# If you start with infinity, the return is infinity, no overflow
infinities = [ complex(INF, 1), complex(NINF, 2), complex(3, INF), complex(4, NINF) ]
for z in close :
self.assertAlmostEquals(abs(z), 1.794e+308, delta=0.01e+308)
for z in over :
self.assertRaises(OverflowError, abs, z)
for z in infinities :
self.assertEqual(abs(z), INF)
def data_file(*name):
return os.path.join(os.path.dirname(__file__), *name)
class ComplexArithmeticTest(unittest.TestCase):
def almostEqual(self, a, b):
if a == b: # also accounts for infinities
return True
return abs(a - b) < 0.0000000001
def assertComplexEqual(self, x, y):
self.assertTrue(
(self.almostEqual(x.real, y.real) or (math.isnan(x.real) and math.isnan(y.real))) and \
(self.almostEqual(x.imag, y.imag) or (math.isnan(x.imag) and math.isnan(y.imag))),
"expected %r != actual %r" % (x, y))
def test_complex_arithmetic(self):
"""Verify *, /, +, - on representative complex numbers results in the same values as in CPython"""
# Verifies fix for http://bugs.jython.org/issue2460
re_comment = re.compile("^(\s*)\#")
with open(data_file("complex_arithmetic.txt")) as f:
for line in f:
if re_comment.match(line):
continue
op, a, b, result = line.split()
if result == "ZeroDivisionError":
self.assertRaises(ZeroDivisionError, getattr(operator, op), complex(a), complex(b))
else:
self.assertComplexEqual(complex(result), getattr(operator, op)(complex(a), complex(b)))
def test_main():
test_support.run_unittest(ComplexTest, ComplexArithmeticTest)
if __name__ == "__main__":
test_main()
|
