test_complex.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656
  1. import unittest
  2. from test import test_support
  3. from random import random
  4. from math import atan2, isnan, copysign
  5. INF = float("inf")
  6. NAN = float("nan")
  7. # These tests ensure that complex math does the right thing
  8. class ComplexTest(unittest.TestCase):
  9. def assertAlmostEqual(self, a, b):
  10. if isinstance(a, complex):
  11. if isinstance(b, complex):
  12. unittest.TestCase.assertAlmostEqual(self, a.real, b.real)
  13. unittest.TestCase.assertAlmostEqual(self, a.imag, b.imag)
  14. else:
  15. unittest.TestCase.assertAlmostEqual(self, a.real, b)
  16. unittest.TestCase.assertAlmostEqual(self, a.imag, 0.)
  17. else:
  18. if isinstance(b, complex):
  19. unittest.TestCase.assertAlmostEqual(self, a, b.real)
  20. unittest.TestCase.assertAlmostEqual(self, 0., b.imag)
  21. else:
  22. unittest.TestCase.assertAlmostEqual(self, a, b)
  23. def assertCloseAbs(self, x, y, eps=1e-9):
  24. """Return true iff floats x and y "are close"."""
  25. # put the one with larger magnitude second
  26. if abs(x) > abs(y):
  27. x, y = y, x
  28. if y == 0:
  29. return abs(x) < eps
  30. if x == 0:
  31. return abs(y) < eps
  32. # check that relative difference < eps
  33. self.assertTrue(abs((x-y)/y) < eps)
  34. def assertFloatsAreIdentical(self, x, y):
  35. """assert that floats x and y are identical, in the sense that:
  36. (1) both x and y are nans, or
  37. (2) both x and y are infinities, with the same sign, or
  38. (3) both x and y are zeros, with the same sign, or
  39. (4) x and y are both finite and nonzero, and x == y
  40. """
  41. msg = 'floats {!r} and {!r} are not identical'
  42. if isnan(x) or isnan(y):
  43. if isnan(x) and isnan(y):
  44. return
  45. elif x == y:
  46. if x != 0.0:
  47. return
  48. # both zero; check that signs match
  49. elif copysign(1.0, x) == copysign(1.0, y):
  50. return
  51. else:
  52. msg += ': zeros have different signs'
  53. self.fail(msg.format(x, y))
  54. def assertClose(self, x, y, eps=1e-9):
  55. """Return true iff complexes x and y "are close"."""
  56. self.assertCloseAbs(x.real, y.real, eps)
  57. self.assertCloseAbs(x.imag, y.imag, eps)
  58. def check_div(self, x, y):
  59. """Compute complex z=x*y, and check that z/x==y and z/y==x."""
  60. z = x * y
  61. if x != 0:
  62. q = z / x
  63. self.assertClose(q, y)
  64. q = z.__div__(x)
  65. self.assertClose(q, y)
  66. q = z.__truediv__(x)
  67. self.assertClose(q, y)
  68. if y != 0:
  69. q = z / y
  70. self.assertClose(q, x)
  71. q = z.__div__(y)
  72. self.assertClose(q, x)
  73. q = z.__truediv__(y)
  74. self.assertClose(q, x)
  75. def test_div(self):
  76. simple_real = [float(i) for i in xrange(-5, 6)]
  77. simple_complex = [complex(x, y) for x in simple_real for y in simple_real]
  78. for x in simple_complex:
  79. for y in simple_complex:
  80. self.check_div(x, y)
  81. # A naive complex division algorithm (such as in 2.0) is very prone to
  82. # nonsense errors for these (overflows and underflows).
  83. self.check_div(complex(1e200, 1e200), 1+0j)
  84. self.check_div(complex(1e-200, 1e-200), 1+0j)
  85. # Just for fun.
  86. for i in xrange(100):
  87. self.check_div(complex(random(), random()),
  88. complex(random(), random()))
  89. self.assertRaises(ZeroDivisionError, complex.__div__, 1+1j, 0+0j)
  90. # FIXME: The following currently crashes on Alpha
  91. # self.assertRaises(OverflowError, pow, 1e200+1j, 1e200+1j)
  92. def test_truediv(self):
  93. self.assertAlmostEqual(complex.__truediv__(2+0j, 1+1j), 1-1j)
  94. self.assertRaises(ZeroDivisionError, complex.__truediv__, 1+1j, 0+0j)
  95. for denom_real, denom_imag in [(0, NAN), (NAN, 0), (NAN, NAN)]:
  96. z = complex(0, 0) / complex(denom_real, denom_imag)
  97. self.assertTrue(isnan(z.real))
  98. self.assertTrue(isnan(z.imag))
  99. def test_floordiv(self):
  100. self.assertAlmostEqual(complex.__floordiv__(3+0j, 1.5+0j), 2)
  101. self.assertRaises(ZeroDivisionError, complex.__floordiv__, 3+0j, 0+0j)
  102. def test_coerce(self):
  103. self.assertRaises(OverflowError, complex.__coerce__, 1+1j, 1L<<10000)
  104. def test_no_implicit_coerce(self):
  105. # Python 2.7 removed implicit coercion from the complex type
  106. class A(object):
  107. def __coerce__(self, other):
  108. raise RuntimeError
  109. __hash__ = None
  110. def __cmp__(self, other):
  111. return -1
  112. a = A()
  113. self.assertRaises(TypeError, lambda: a + 2.0j)
  114. self.assertTrue(a < 2.0j)
  115. def test_richcompare(self):
  116. self.assertEqual(complex.__eq__(1+1j, 1L<<10000), False)
  117. self.assertEqual(complex.__lt__(1+1j, None), NotImplemented)
  118. self.assertIs(complex.__eq__(1+1j, 1+1j), True)
  119. self.assertIs(complex.__eq__(1+1j, 2+2j), False)
  120. self.assertIs(complex.__ne__(1+1j, 1+1j), False)
  121. self.assertIs(complex.__ne__(1+1j, 2+2j), True)
  122. self.assertRaises(TypeError, complex.__lt__, 1+1j, 2+2j)
  123. self.assertRaises(TypeError, complex.__le__, 1+1j, 2+2j)
  124. self.assertRaises(TypeError, complex.__gt__, 1+1j, 2+2j)
  125. self.assertRaises(TypeError, complex.__ge__, 1+1j, 2+2j)
  126. def test_richcompare_boundaries(self):
  127. def check(n, deltas, is_equal, imag = 0.0):
  128. for delta in deltas:
  129. i = n + delta
  130. z = complex(i, imag)
  131. self.assertIs(complex.__eq__(z, i), is_equal(delta))
  132. self.assertIs(complex.__ne__(z, i), not is_equal(delta))
  133. # For IEEE-754 doubles the following should hold:
  134. # x in [2 ** (52 + i), 2 ** (53 + i + 1)] -> x mod 2 ** i == 0
  135. # where the interval is representable, of course.
  136. for i in range(1, 10):
  137. pow = 52 + i
  138. mult = 2 ** i
  139. check(2 ** pow, range(1, 101), lambda delta: delta % mult == 0)
  140. check(2 ** pow, range(1, 101), lambda delta: False, float(i))
  141. check(2 ** 53, range(-100, 0), lambda delta: True)
  142. def test_mod(self):
  143. self.assertRaises(ZeroDivisionError, (1+1j).__mod__, 0+0j)
  144. a = 3.33+4.43j
  145. try:
  146. a % 0
  147. except ZeroDivisionError:
  148. pass
  149. else:
  150. self.fail("modulo parama can't be 0")
  151. def test_divmod(self):
  152. self.assertRaises(ZeroDivisionError, divmod, 1+1j, 0+0j)
  153. def test_pow(self):
  154. self.assertAlmostEqual(pow(1+1j, 0+0j), 1.0)
  155. self.assertAlmostEqual(pow(0+0j, 2+0j), 0.0)
  156. self.assertRaises(ZeroDivisionError, pow, 0+0j, 1j)
  157. self.assertAlmostEqual(pow(1j, -1), 1/1j)
  158. self.assertAlmostEqual(pow(1j, 200), 1)
  159. self.assertRaises(ValueError, pow, 1+1j, 1+1j, 1+1j)
  160. a = 3.33+4.43j
  161. self.assertEqual(a ** 0j, 1)
  162. self.assertEqual(a ** 0.+0.j, 1)
  163. self.assertEqual(3j ** 0j, 1)
  164. self.assertEqual(3j ** 0, 1)
  165. try:
  166. 0j ** a
  167. except ZeroDivisionError:
  168. pass
  169. else:
  170. self.fail("should fail 0.0 to negative or complex power")
  171. try:
  172. 0j ** (3-2j)
  173. except ZeroDivisionError:
  174. pass
  175. else:
  176. self.fail("should fail 0.0 to negative or complex power")
  177. # The following is used to exercise certain code paths
  178. self.assertEqual(a ** 105, a ** 105)
  179. self.assertEqual(a ** -105, a ** -105)
  180. self.assertEqual(a ** -30, a ** -30)
  181. self.assertEqual(0.0j ** 0, 1)
  182. b = 5.1+2.3j
  183. self.assertRaises(ValueError, pow, a, b, 0)
  184. def test_boolcontext(self):
  185. for i in xrange(100):
  186. self.assertTrue(complex(random() + 1e-6, random() + 1e-6))
  187. self.assertTrue(not complex(0.0, 0.0))
  188. def test_conjugate(self):
  189. self.assertClose(complex(5.3, 9.8).conjugate(), 5.3-9.8j)
  190. def test_constructor(self):
  191. class OS:
  192. def __init__(self, value): self.value = value
  193. def __complex__(self): return self.value
  194. class NS(object):
  195. def __init__(self, value): self.value = value
  196. def __complex__(self): return self.value
  197. self.assertEqual(complex(OS(1+10j)), 1+10j)
  198. self.assertEqual(complex(NS(1+10j)), 1+10j)
  199. self.assertRaises(TypeError, complex, OS(None))
  200. self.assertRaises(TypeError, complex, NS(None))
  201. self.assertAlmostEqual(complex("1+10j"), 1+10j)
  202. self.assertAlmostEqual(complex(10), 10+0j)
  203. self.assertAlmostEqual(complex(10.0), 10+0j)
  204. self.assertAlmostEqual(complex(10L), 10+0j)
  205. self.assertAlmostEqual(complex(10+0j), 10+0j)
  206. self.assertAlmostEqual(complex(1,10), 1+10j)
  207. self.assertAlmostEqual(complex(1,10L), 1+10j)
  208. self.assertAlmostEqual(complex(1,10.0), 1+10j)
  209. self.assertAlmostEqual(complex(1L,10), 1+10j)
  210. self.assertAlmostEqual(complex(1L,10L), 1+10j)
  211. self.assertAlmostEqual(complex(1L,10.0), 1+10j)
  212. self.assertAlmostEqual(complex(1.0,10), 1+10j)
  213. self.assertAlmostEqual(complex(1.0,10L), 1+10j)
  214. self.assertAlmostEqual(complex(1.0,10.0), 1+10j)
  215. self.assertAlmostEqual(complex(3.14+0j), 3.14+0j)
  216. self.assertAlmostEqual(complex(3.14), 3.14+0j)
  217. self.assertAlmostEqual(complex(314), 314.0+0j)
  218. self.assertAlmostEqual(complex(314L), 314.0+0j)
  219. self.assertAlmostEqual(complex(3.14+0j, 0j), 3.14+0j)
  220. self.assertAlmostEqual(complex(3.14, 0.0), 3.14+0j)
  221. self.assertAlmostEqual(complex(314, 0), 314.0+0j)
  222. self.assertAlmostEqual(complex(314L, 0L), 314.0+0j)
  223. self.assertAlmostEqual(complex(0j, 3.14j), -3.14+0j)
  224. self.assertAlmostEqual(complex(0.0, 3.14j), -3.14+0j)
  225. self.assertAlmostEqual(complex(0j, 3.14), 3.14j)
  226. self.assertAlmostEqual(complex(0.0, 3.14), 3.14j)
  227. self.assertAlmostEqual(complex("1"), 1+0j)
  228. self.assertAlmostEqual(complex("1j"), 1j)
  229. self.assertAlmostEqual(complex(), 0)
  230. self.assertAlmostEqual(complex("-1"), -1)
  231. self.assertAlmostEqual(complex("+1"), +1)
  232. self.assertAlmostEqual(complex("(1+2j)"), 1+2j)
  233. self.assertAlmostEqual(complex("(1.3+2.2j)"), 1.3+2.2j)
  234. self.assertAlmostEqual(complex("3.14+1J"), 3.14+1j)
  235. self.assertAlmostEqual(complex(" ( +3.14-6J )"), 3.14-6j)
  236. self.assertAlmostEqual(complex(" ( +3.14-J )"), 3.14-1j)
  237. self.assertAlmostEqual(complex(" ( +3.14+j )"), 3.14+1j)
  238. self.assertAlmostEqual(complex("J"), 1j)
  239. self.assertAlmostEqual(complex("( j )"), 1j)
  240. self.assertAlmostEqual(complex("+J"), 1j)
  241. self.assertAlmostEqual(complex("( -j)"), -1j)
  242. self.assertAlmostEqual(complex('1e-500'), 0.0 + 0.0j)
  243. self.assertAlmostEqual(complex('-1e-500j'), 0.0 - 0.0j)
  244. self.assertAlmostEqual(complex('-1e-500+1e-500j'), -0.0 + 0.0j)
  245. class complex2(complex): pass
  246. self.assertAlmostEqual(complex(complex2(1+1j)), 1+1j)
  247. self.assertAlmostEqual(complex(real=17, imag=23), 17+23j)
  248. self.assertAlmostEqual(complex(real=17+23j), 17+23j)
  249. self.assertAlmostEqual(complex(real=17+23j, imag=23), 17+46j)
  250. self.assertAlmostEqual(complex(real=1+2j, imag=3+4j), -3+5j)
  251. # check that the sign of a zero in the real or imaginary part
  252. # is preserved when constructing from two floats. (These checks
  253. # are harmless on systems without support for signed zeros.)
  254. def split_zeros(x):
  255. """Function that produces different results for 0. and -0."""
  256. return atan2(x, -1.)
  257. self.assertEqual(split_zeros(complex(1., 0.).imag), split_zeros(0.))
  258. self.assertEqual(split_zeros(complex(1., -0.).imag), split_zeros(-0.))
  259. self.assertEqual(split_zeros(complex(0., 1.).real), split_zeros(0.))
  260. self.assertEqual(split_zeros(complex(-0., 1.).real), split_zeros(-0.))
  261. c = 3.14 + 1j
  262. self.assertTrue(complex(c) is c)
  263. del c
  264. self.assertRaises(TypeError, complex, "1", "1")
  265. self.assertRaises(TypeError, complex, 1, "1")
  266. if test_support.have_unicode:
  267. self.assertEqual(complex(unicode(" 3.14+J ")), 3.14+1j)
  268. # SF bug 543840: complex(string) accepts strings with \0
  269. # Fixed in 2.3.
  270. self.assertRaises(ValueError, complex, '1+1j\0j')
  271. self.assertRaises(TypeError, int, 5+3j)
  272. self.assertRaises(TypeError, long, 5+3j)
  273. self.assertRaises(TypeError, float, 5+3j)
  274. self.assertRaises(ValueError, complex, "")
  275. self.assertRaises(TypeError, complex, None)
  276. self.assertRaises(ValueError, complex, "\0")
  277. self.assertRaises(ValueError, complex, "3\09")
  278. self.assertRaises(TypeError, complex, "1", "2")
  279. self.assertRaises(TypeError, complex, "1", 42)
  280. self.assertRaises(TypeError, complex, 1, "2")
  281. self.assertRaises(ValueError, complex, "1+")
  282. self.assertRaises(ValueError, complex, "1+1j+1j")
  283. self.assertRaises(ValueError, complex, "--")
  284. self.assertRaises(ValueError, complex, "(1+2j")
  285. self.assertRaises(ValueError, complex, "1+2j)")
  286. self.assertRaises(ValueError, complex, "1+(2j)")
  287. self.assertRaises(ValueError, complex, "(1+2j)123")
  288. if test_support.have_unicode:
  289. self.assertRaises(ValueError, complex, unicode("x"))
  290. self.assertRaises(ValueError, complex, "1j+2")
  291. self.assertRaises(ValueError, complex, "1e1ej")
  292. self.assertRaises(ValueError, complex, "1e++1ej")
  293. self.assertRaises(ValueError, complex, ")1+2j(")
  294. # the following three are accepted by Python 2.6
  295. self.assertRaises(ValueError, complex, "1..1j")
  296. self.assertRaises(ValueError, complex, "1.11.1j")
  297. self.assertRaises(ValueError, complex, "1e1.1j")
  298. if test_support.have_unicode:
  299. # check that complex accepts long unicode strings
  300. self.assertEqual(type(complex(unicode("1"*500))), complex)
  301. class EvilExc(Exception):
  302. pass
  303. class evilcomplex:
  304. def __complex__(self):
  305. raise EvilExc
  306. self.assertRaises(EvilExc, complex, evilcomplex())
  307. class float2:
  308. def __init__(self, value):
  309. self.value = value
  310. def __float__(self):
  311. return self.value
  312. self.assertAlmostEqual(complex(float2(42.)), 42)
  313. self.assertAlmostEqual(complex(real=float2(17.), imag=float2(23.)), 17+23j)
  314. self.assertRaises(TypeError, complex, float2(None))
  315. class complex0(complex):
  316. """Test usage of __complex__() when inheriting from 'complex'"""
  317. def __complex__(self):
  318. return 42j
  319. class complex1(complex):
  320. """Test usage of __complex__() with a __new__() method"""
  321. def __new__(self, value=0j):
  322. return complex.__new__(self, 2*value)
  323. def __complex__(self):
  324. return self
  325. class complex2(complex):
  326. """Make sure that __complex__() calls fail if anything other than a
  327. complex is returned"""
  328. def __complex__(self):
  329. return None
  330. self.assertAlmostEqual(complex(complex0(1j)), 42j)
  331. self.assertAlmostEqual(complex(complex1(1j)), 2j)
  332. self.assertRaises(TypeError, complex, complex2(1j))
  333. def test_subclass(self):
  334. class xcomplex(complex):
  335. def __add__(self,other):
  336. return xcomplex(complex(self) + other)
  337. __radd__ = __add__
  338. def __sub__(self,other):
  339. return xcomplex(complex(self) + other)
  340. __rsub__ = __sub__
  341. def __mul__(self,other):
  342. return xcomplex(complex(self) * other)
  343. __rmul__ = __mul__
  344. def __div__(self,other):
  345. return xcomplex(complex(self) / other)
  346. def __rdiv__(self,other):
  347. return xcomplex(other / complex(self))
  348. __truediv__ = __div__
  349. __rtruediv__ = __rdiv__
  350. def __floordiv__(self,other):
  351. return xcomplex(complex(self) // other)
  352. def __rfloordiv__(self,other):
  353. return xcomplex(other // complex(self))
  354. def __pow__(self,other):
  355. return xcomplex(complex(self) ** other)
  356. def __rpow__(self,other):
  357. return xcomplex(other ** complex(self) )
  358. def __mod__(self,other):
  359. return xcomplex(complex(self) % other)
  360. def __rmod__(self,other):
  361. return xcomplex(other % complex(self))
  362. infix_binops = ('+', '-', '*', '**', '%', '//', '/')
  363. xcomplex_values = (xcomplex(1), xcomplex(123.0),
  364. xcomplex(-10+2j), xcomplex(3+187j),
  365. xcomplex(3-78j))
  366. test_values = (1, 123.0, 10-19j, xcomplex(1+2j),
  367. xcomplex(1+87j), xcomplex(10+90j))
  368. for op in infix_binops:
  369. for x in xcomplex_values:
  370. for y in test_values:
  371. a = 'x %s y' % op
  372. b = 'y %s x' % op
  373. self.assertTrue(type(eval(a)) is type(eval(b)) is xcomplex)
  374. def test_hash(self):
  375. for x in xrange(-30, 30):
  376. self.assertEqual(hash(x), hash(complex(x, 0)))
  377. x /= 3.0 # now check against floating point
  378. self.assertEqual(hash(x), hash(complex(x, 0.)))
  379. def test_abs(self):
  380. nums = [complex(x/3., y/7.) for x in xrange(-9,9) for y in xrange(-9,9)]
  381. for num in nums:
  382. self.assertAlmostEqual((num.real**2 + num.imag**2) ** 0.5, abs(num))
  383. def test_repr(self):
  384. self.assertEqual(repr(1+6j), '(1+6j)')
  385. self.assertEqual(repr(1-6j), '(1-6j)')
  386. self.assertNotEqual(repr(-(1+0j)), '(-1+-0j)')
  387. self.assertEqual(1-6j,complex(repr(1-6j)))
  388. self.assertEqual(1+6j,complex(repr(1+6j)))
  389. self.assertEqual(-6j,complex(repr(-6j)))
  390. self.assertEqual(6j,complex(repr(6j)))
  391. self.assertEqual(repr(complex(1., INF)), "(1+infj)")
  392. self.assertEqual(repr(complex(1., -INF)), "(1-infj)")
  393. self.assertEqual(repr(complex(INF, 1)), "(inf+1j)")
  394. self.assertEqual(repr(complex(-INF, INF)), "(-inf+infj)")
  395. self.assertEqual(repr(complex(NAN, 1)), "(nan+1j)")
  396. self.assertEqual(repr(complex(1, NAN)), "(1+nanj)")
  397. self.assertEqual(repr(complex(NAN, NAN)), "(nan+nanj)")
  398. self.assertEqual(repr(complex(0, INF)), "infj")
  399. self.assertEqual(repr(complex(0, -INF)), "-infj")
  400. self.assertEqual(repr(complex(0, NAN)), "nanj")
  401. def test_neg(self):
  402. self.assertEqual(-(1+6j), -1-6j)
  403. def test_file(self):
  404. a = 3.33+4.43j
  405. b = 5.1+2.3j
  406. fo = None
  407. try:
  408. fo = open(test_support.TESTFN, "wb")
  409. print >>fo, a, b
  410. fo.close()
  411. fo = open(test_support.TESTFN, "rb")
  412. self.assertEqual(fo.read(), "%s %s\n" % (a, b))
  413. finally:
  414. if (fo is not None) and (not fo.closed):
  415. fo.close()
  416. test_support.unlink(test_support.TESTFN)
  417. def test_getnewargs(self):
  418. self.assertEqual((1+2j).__getnewargs__(), (1.0, 2.0))
  419. self.assertEqual((1-2j).__getnewargs__(), (1.0, -2.0))
  420. self.assertEqual((2j).__getnewargs__(), (0.0, 2.0))
  421. self.assertEqual((-0j).__getnewargs__(), (0.0, -0.0))
  422. self.assertEqual(complex(0, INF).__getnewargs__(), (0.0, INF))
  423. self.assertEqual(complex(INF, 0).__getnewargs__(), (INF, 0.0))
  424. if float.__getformat__("double").startswith("IEEE"):
  425. def test_plus_minus_0j(self):
  426. # test that -0j and 0j literals are not identified
  427. z1, z2 = 0j, -0j
  428. self.assertEqual(atan2(z1.imag, -1.), atan2(0., -1.))
  429. self.assertEqual(atan2(z2.imag, -1.), atan2(-0., -1.))
  430. @unittest.skipUnless(float.__getformat__("double").startswith("IEEE"),
  431. "test requires IEEE 754 doubles")
  432. def test_overflow(self):
  433. self.assertEqual(complex("1e500"), complex(INF, 0.0))
  434. self.assertEqual(complex("-1e500j"), complex(0.0, -INF))
  435. self.assertEqual(complex("-1e500+1.8e308j"), complex(-INF, INF))
  436. @unittest.skipUnless(float.__getformat__("double").startswith("IEEE"),
  437. "test requires IEEE 754 doubles")
  438. def test_repr_roundtrip(self):
  439. vals = [0.0, 1e-500, 1e-315, 1e-200, 0.0123, 3.1415, 1e50, INF, NAN]
  440. vals += [-v for v in vals]
  441. # complex(repr(z)) should recover z exactly, even for complex
  442. # numbers involving an infinity, nan, or negative zero
  443. for x in vals:
  444. for y in vals:
  445. z = complex(x, y)
  446. roundtrip = complex(repr(z))
  447. self.assertFloatsAreIdentical(z.real, roundtrip.real)
  448. self.assertFloatsAreIdentical(z.imag, roundtrip.imag)
  449. # if we predefine some constants, then eval(repr(z)) should
  450. # also work, except that it might change the sign of zeros
  451. inf, nan = float('inf'), float('nan')
  452. infj, nanj = complex(0.0, inf), complex(0.0, nan)
  453. for x in vals:
  454. for y in vals:
  455. z = complex(x, y)
  456. roundtrip = eval(repr(z))
  457. # adding 0.0 has no effect beside changing -0.0 to 0.0
  458. self.assertFloatsAreIdentical(0.0 + z.real,
  459. 0.0 + roundtrip.real)
  460. self.assertFloatsAreIdentical(0.0 + z.imag,
  461. 0.0 + roundtrip.imag)
  462. def test_format(self):
  463. # empty format string is same as str()
  464. self.assertEqual(format(1+3j, ''), str(1+3j))
  465. self.assertEqual(format(1.5+3.5j, ''), str(1.5+3.5j))
  466. self.assertEqual(format(3j, ''), str(3j))
  467. self.assertEqual(format(3.2j, ''), str(3.2j))
  468. self.assertEqual(format(3+0j, ''), str(3+0j))
  469. self.assertEqual(format(3.2+0j, ''), str(3.2+0j))
  470. # empty presentation type should still be analogous to str,
  471. # even when format string is nonempty (issue #5920).
  472. self.assertEqual(format(3.2+0j, '-'), str(3.2+0j))
  473. self.assertEqual(format(3.2+0j, '<'), str(3.2+0j))
  474. z = 4/7. - 100j/7.
  475. self.assertEqual(format(z, ''), str(z))
  476. self.assertEqual(format(z, '-'), str(z))
  477. self.assertEqual(format(z, '<'), str(z))
  478. self.assertEqual(format(z, '10'), str(z))
  479. z = complex(0.0, 3.0)
  480. self.assertEqual(format(z, ''), str(z))
  481. self.assertEqual(format(z, '-'), str(z))
  482. self.assertEqual(format(z, '<'), str(z))
  483. self.assertEqual(format(z, '2'), str(z))
  484. z = complex(-0.0, 2.0)
  485. self.assertEqual(format(z, ''), str(z))
  486. self.assertEqual(format(z, '-'), str(z))
  487. self.assertEqual(format(z, '<'), str(z))
  488. self.assertEqual(format(z, '3'), str(z))
  489. self.assertEqual(format(1+3j, 'g'), '1+3j')
  490. self.assertEqual(format(3j, 'g'), '0+3j')
  491. self.assertEqual(format(1.5+3.5j, 'g'), '1.5+3.5j')
  492. self.assertEqual(format(1.5+3.5j, '+g'), '+1.5+3.5j')
  493. self.assertEqual(format(1.5-3.5j, '+g'), '+1.5-3.5j')
  494. self.assertEqual(format(1.5-3.5j, '-g'), '1.5-3.5j')
  495. self.assertEqual(format(1.5+3.5j, ' g'), ' 1.5+3.5j')
  496. self.assertEqual(format(1.5-3.5j, ' g'), ' 1.5-3.5j')
  497. self.assertEqual(format(-1.5+3.5j, ' g'), '-1.5+3.5j')
  498. self.assertEqual(format(-1.5-3.5j, ' g'), '-1.5-3.5j')
  499. self.assertEqual(format(-1.5-3.5e-20j, 'g'), '-1.5-3.5e-20j')
  500. self.assertEqual(format(-1.5-3.5j, 'f'), '-1.500000-3.500000j')
  501. self.assertEqual(format(-1.5-3.5j, 'F'), '-1.500000-3.500000j')
  502. self.assertEqual(format(-1.5-3.5j, 'e'), '-1.500000e+00-3.500000e+00j')
  503. self.assertEqual(format(-1.5-3.5j, '.2e'), '-1.50e+00-3.50e+00j')
  504. self.assertEqual(format(-1.5-3.5j, '.2E'), '-1.50E+00-3.50E+00j')
  505. self.assertEqual(format(-1.5e10-3.5e5j, '.2G'), '-1.5E+10-3.5E+05j')
  506. self.assertEqual(format(1.5+3j, '<20g'), '1.5+3j ')
  507. self.assertEqual(format(1.5+3j, '*<20g'), '1.5+3j**************')
  508. self.assertEqual(format(1.5+3j, '>20g'), ' 1.5+3j')
  509. self.assertEqual(format(1.5+3j, '^20g'), ' 1.5+3j ')
  510. self.assertEqual(format(1.5+3j, '<20'), '(1.5+3j) ')
  511. self.assertEqual(format(1.5+3j, '>20'), ' (1.5+3j)')
  512. self.assertEqual(format(1.5+3j, '^20'), ' (1.5+3j) ')
  513. self.assertEqual(format(1.123-3.123j, '^20.2'), ' (1.1-3.1j) ')
  514. self.assertEqual(format(1.5+3j, '20.2f'), ' 1.50+3.00j')
  515. self.assertEqual(format(1.5+3j, '>20.2f'), ' 1.50+3.00j')
  516. self.assertEqual(format(1.5+3j, '<20.2f'), '1.50+3.00j ')
  517. self.assertEqual(format(1.5e20+3j, '<20.2f'), '150000000000000000000.00+3.00j')
  518. self.assertEqual(format(1.5e20+3j, '>40.2f'), ' 150000000000000000000.00+3.00j')
  519. self.assertEqual(format(1.5e20+3j, '^40,.2f'), ' 150,000,000,000,000,000,000.00+3.00j ')
  520. self.assertEqual(format(1.5e21+3j, '^40,.2f'), ' 1,500,000,000,000,000,000,000.00+3.00j ')
  521. self.assertEqual(format(1.5e21+3000j, ',.2f'), '1,500,000,000,000,000,000,000.00+3,000.00j')
  522. # alternate is invalid
  523. self.assertRaises(ValueError, (1.5+0.5j).__format__, '#f')
  524. # zero padding is invalid
  525. self.assertRaises(ValueError, (1.5+0.5j).__format__, '010f')
  526. # '=' alignment is invalid
  527. self.assertRaises(ValueError, (1.5+3j).__format__, '=20')
  528. # integer presentation types are an error
  529. for t in 'bcdoxX':
  530. self.assertRaises(ValueError, (1.5+0.5j).__format__, t)
  531. # make sure everything works in ''.format()
  532. self.assertEqual('*{0:.3f}*'.format(3.14159+2.71828j), '*3.142+2.718j*')
  533. # issue 3382: 'f' and 'F' with inf's and nan's
  534. self.assertEqual('{0:f}'.format(INF+0j), 'inf+0.000000j')
  535. self.assertEqual('{0:F}'.format(INF+0j), 'INF+0.000000j')
  536. self.assertEqual('{0:f}'.format(-INF+0j), '-inf+0.000000j')
  537. self.assertEqual('{0:F}'.format(-INF+0j), '-INF+0.000000j')
  538. self.assertEqual('{0:f}'.format(complex(INF, INF)), 'inf+infj')
  539. self.assertEqual('{0:F}'.format(complex(INF, INF)), 'INF+INFj')
  540. self.assertEqual('{0:f}'.format(complex(INF, -INF)), 'inf-infj')
  541. self.assertEqual('{0:F}'.format(complex(INF, -INF)), 'INF-INFj')
  542. self.assertEqual('{0:f}'.format(complex(-INF, INF)), '-inf+infj')
  543. self.assertEqual('{0:F}'.format(complex(-INF, INF)), '-INF+INFj')
  544. self.assertEqual('{0:f}'.format(complex(-INF, -INF)), '-inf-infj')
  545. self.assertEqual('{0:F}'.format(complex(-INF, -INF)), '-INF-INFj')
  546. self.assertEqual('{0:f}'.format(complex(NAN, 0)), 'nan+0.000000j')
  547. self.assertEqual('{0:F}'.format(complex(NAN, 0)), 'NAN+0.000000j')
  548. self.assertEqual('{0:f}'.format(complex(NAN, NAN)), 'nan+nanj')
  549. self.assertEqual('{0:F}'.format(complex(NAN, NAN)), 'NAN+NANj')
  550. def test_main():
  551. with test_support.check_warnings(("complex divmod.., // and % are "
  552. "deprecated", DeprecationWarning)):
  553. test_support.run_unittest(ComplexTest)
  554. if __name__ == "__main__":
  555. test_main()