blob: 5b9e37aa8d169d5d32b8196499b58e68baec0824 [file] [log] [blame]
Walter Dörwalda0021592005-06-13 21:44:48 +00001import unittest
Benjamin Petersonee8712c2008-05-20 21:35:26 +00002from test import support
Eric Smith3ab08ca2010-12-04 15:17:38 +00003
Christian Heimes81ee3ef2008-05-04 22:42:01 +00004import sys
Walter Dörwalda0021592005-06-13 21:44:48 +00005
6import random
Mark Dickinson54bc1ec2008-12-17 16:19:07 +00007import math
Alexandre Vassalottic36c3782010-01-09 20:35:09 +00008import array
Walter Dörwalda0021592005-06-13 21:44:48 +00009
Guido van Rossum4365cab1998-08-13 14:20:17 +000010# SHIFT should match the value in longintrepr.h for best testing.
Mark Dickinsonbd792642009-03-18 20:06:12 +000011SHIFT = sys.int_info.bits_per_digit
Guido van Rossum4365cab1998-08-13 14:20:17 +000012BASE = 2 ** SHIFT
13MASK = BASE - 1
Tim Petersdaec9612004-08-30 23:18:23 +000014KARATSUBA_CUTOFF = 70 # from longobject.c
Guido van Rossum4365cab1998-08-13 14:20:17 +000015
16# Max number of base BASE digits to use in test cases. Doubling
Tim Peters28b0e2a2002-08-13 02:17:11 +000017# this will more than double the runtime.
18MAXDIGITS = 15
Guido van Rossum4365cab1998-08-13 14:20:17 +000019
Guido van Rossum4581a0c1998-10-02 01:19:48 +000020# build some special values
Guido van Rossumc1f779c2007-07-03 08:25:58 +000021special = [0, 1, 2, BASE, BASE >> 1, 0x5555555555555555, 0xaaaaaaaaaaaaaaaa]
Guido van Rossum4581a0c1998-10-02 01:19:48 +000022# some solid strings of one bits
Guido van Rossume2a383d2007-01-15 16:59:06 +000023p2 = 4 # 0 and 1 already added
Guido van Rossum4581a0c1998-10-02 01:19:48 +000024for i in range(2*SHIFT):
25 special.append(p2 - 1)
26 p2 = p2 << 1
27del p2
28# add complements & negations
Guido van Rossumc1f779c2007-07-03 08:25:58 +000029special += [~x for x in special] + [-x for x in special]
Guido van Rossum4581a0c1998-10-02 01:19:48 +000030
Mark Dickinsoncbb62742009-12-27 15:09:50 +000031DBL_MAX = sys.float_info.max
32DBL_MAX_EXP = sys.float_info.max_exp
33DBL_MIN_EXP = sys.float_info.min_exp
34DBL_MANT_DIG = sys.float_info.mant_dig
35DBL_MIN_OVERFLOW = 2**DBL_MAX_EXP - 2**(DBL_MAX_EXP - DBL_MANT_DIG - 1)
36
Mark Dickinson30970e92011-10-23 20:07:13 +010037
38# Pure Python version of correctly-rounded integer-to-float conversion.
39def int_to_float(n):
40 """
41 Correctly-rounded integer-to-float conversion.
42
43 """
44 # Constants, depending only on the floating-point format in use.
45 # We use an extra 2 bits of precision for rounding purposes.
46 PRECISION = sys.float_info.mant_dig + 2
47 SHIFT_MAX = sys.float_info.max_exp - PRECISION
48 Q_MAX = 1 << PRECISION
49 ROUND_HALF_TO_EVEN_CORRECTION = [0, -1, -2, 1, 0, -1, 2, 1]
50
51 # Reduce to the case where n is positive.
52 if n == 0:
53 return 0.0
54 elif n < 0:
55 return -int_to_float(-n)
56
57 # Convert n to a 'floating-point' number q * 2**shift, where q is an
58 # integer with 'PRECISION' significant bits. When shifting n to create q,
59 # the least significant bit of q is treated as 'sticky'. That is, the
60 # least significant bit of q is set if either the corresponding bit of n
61 # was already set, or any one of the bits of n lost in the shift was set.
62 shift = n.bit_length() - PRECISION
63 q = n << -shift if shift < 0 else (n >> shift) | bool(n & ~(-1 << shift))
64
65 # Round half to even (actually rounds to the nearest multiple of 4,
66 # rounding ties to a multiple of 8).
67 q += ROUND_HALF_TO_EVEN_CORRECTION[q & 7]
68
69 # Detect overflow.
70 if shift + (q == Q_MAX) > SHIFT_MAX:
71 raise OverflowError("integer too large to convert to float")
72
73 # Checks: q is exactly representable, and q**2**shift doesn't overflow.
74 assert q % 4 == 0 and q // 4 <= 2**(sys.float_info.mant_dig)
75 assert q * 2**shift <= sys.float_info.max
76
77 # Some circularity here, since float(q) is doing an int-to-float
78 # conversion. But here q is of bounded size, and is exactly representable
79 # as a float. In a low-level C-like language, this operation would be a
80 # simple cast (e.g., from unsigned long long to double).
81 return math.ldexp(float(q), shift)
82
83
Mark Dickinsoncbb62742009-12-27 15:09:50 +000084# pure Python version of correctly-rounded true division
85def truediv(a, b):
86 """Correctly-rounded true division for integers."""
87 negative = a^b < 0
88 a, b = abs(a), abs(b)
89
90 # exceptions: division by zero, overflow
91 if not b:
92 raise ZeroDivisionError("division by zero")
93 if a >= DBL_MIN_OVERFLOW * b:
94 raise OverflowError("int/int too large to represent as a float")
95
96 # find integer d satisfying 2**(d - 1) <= a/b < 2**d
97 d = a.bit_length() - b.bit_length()
98 if d >= 0 and a >= 2**d * b or d < 0 and a * 2**-d >= b:
99 d += 1
100
101 # compute 2**-exp * a / b for suitable exp
102 exp = max(d, DBL_MIN_EXP) - DBL_MANT_DIG
103 a, b = a << max(-exp, 0), b << max(exp, 0)
104 q, r = divmod(a, b)
105
106 # round-half-to-even: fractional part is r/b, which is > 0.5 iff
107 # 2*r > b, and == 0.5 iff 2*r == b.
108 if 2*r > b or 2*r == b and q % 2 == 1:
109 q += 1
110
Mark Dickinsona4e15062009-12-27 19:03:31 +0000111 result = math.ldexp(q, exp)
Mark Dickinsoncbb62742009-12-27 15:09:50 +0000112 return -result if negative else result
113
114
Walter Dörwalda0021592005-06-13 21:44:48 +0000115class LongTest(unittest.TestCase):
Guido van Rossum4365cab1998-08-13 14:20:17 +0000116
Walter Dörwalda0021592005-06-13 21:44:48 +0000117 # Get quasi-random long consisting of ndigits digits (in base BASE).
118 # quasi == the most-significant digit will not be 0, and the number
119 # is constructed to contain long strings of 0 and 1 bits. These are
120 # more likely than random bits to provoke digit-boundary errors.
121 # The sign of the number is also random.
Guido van Rossum4365cab1998-08-13 14:20:17 +0000122
Walter Dörwalda0021592005-06-13 21:44:48 +0000123 def getran(self, ndigits):
Serhiy Storchaka3a20a5d2014-02-08 14:28:33 +0200124 self.assertGreater(ndigits, 0)
Walter Dörwalda0021592005-06-13 21:44:48 +0000125 nbits_hi = ndigits * SHIFT
126 nbits_lo = nbits_hi - SHIFT + 1
Guido van Rossume2a383d2007-01-15 16:59:06 +0000127 answer = 0
Walter Dörwalda0021592005-06-13 21:44:48 +0000128 nbits = 0
129 r = int(random.random() * (SHIFT * 2)) | 1 # force 1 bits to start
130 while nbits < nbits_lo:
131 bits = (r >> 1) + 1
132 bits = min(bits, nbits_hi - nbits)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000133 self.assertTrue(1 <= bits <= SHIFT)
Walter Dörwalda0021592005-06-13 21:44:48 +0000134 nbits = nbits + bits
135 answer = answer << bits
136 if r & 1:
137 answer = answer | ((1 << bits) - 1)
138 r = int(random.random() * (SHIFT * 2))
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000139 self.assertTrue(nbits_lo <= nbits <= nbits_hi)
Walter Dörwalda0021592005-06-13 21:44:48 +0000140 if random.random() < 0.5:
141 answer = -answer
142 return answer
Guido van Rossum4581a0c1998-10-02 01:19:48 +0000143
Walter Dörwalda0021592005-06-13 21:44:48 +0000144 # Get random long consisting of ndigits random digits (relative to base
145 # BASE). The sign bit is also random.
Guido van Rossum4581a0c1998-10-02 01:19:48 +0000146
Walter Dörwalda0021592005-06-13 21:44:48 +0000147 def getran2(ndigits):
Guido van Rossume2a383d2007-01-15 16:59:06 +0000148 answer = 0
Guido van Rossum805365e2007-05-07 22:24:25 +0000149 for i in range(ndigits):
Walter Dörwalda0021592005-06-13 21:44:48 +0000150 answer = (answer << SHIFT) | random.randint(0, MASK)
151 if random.random() < 0.5:
152 answer = -answer
153 return answer
Guido van Rossum4365cab1998-08-13 14:20:17 +0000154
Walter Dörwalda0021592005-06-13 21:44:48 +0000155 def check_division(self, x, y):
156 eq = self.assertEqual
Martin Pantercbe16ae2015-09-25 23:50:47 +0000157 with self.subTest(x=x, y=y):
158 q, r = divmod(x, y)
159 q2, r2 = x//y, x%y
160 pab, pba = x*y, y*x
161 eq(pab, pba, "multiplication does not commute")
162 eq(q, q2, "divmod returns different quotient than /")
163 eq(r, r2, "divmod returns different mod than %")
164 eq(x, q*y + r, "x != q*y + r after divmod")
165 if y > 0:
166 self.assertTrue(0 <= r < y, "bad mod from divmod")
167 else:
168 self.assertTrue(y < r <= 0, "bad mod from divmod")
Guido van Rossum4365cab1998-08-13 14:20:17 +0000169
Walter Dörwalda0021592005-06-13 21:44:48 +0000170 def test_division(self):
Guido van Rossum805365e2007-05-07 22:24:25 +0000171 digits = list(range(1, MAXDIGITS+1)) + list(range(KARATSUBA_CUTOFF,
172 KARATSUBA_CUTOFF + 14))
Walter Dörwalda0021592005-06-13 21:44:48 +0000173 digits.append(KARATSUBA_CUTOFF * 3)
174 for lenx in digits:
175 x = self.getran(lenx)
176 for leny in digits:
Guido van Rossume2a383d2007-01-15 16:59:06 +0000177 y = self.getran(leny) or 1
Walter Dörwalda0021592005-06-13 21:44:48 +0000178 self.check_division(x, y)
Guido van Rossum4365cab1998-08-13 14:20:17 +0000179
Mark Dickinsonbd792642009-03-18 20:06:12 +0000180 # specific numbers chosen to exercise corner cases of the
181 # current long division implementation
182
183 # 30-bit cases involving a quotient digit estimate of BASE+1
184 self.check_division(1231948412290879395966702881,
185 1147341367131428698)
186 self.check_division(815427756481275430342312021515587883,
187 707270836069027745)
188 self.check_division(627976073697012820849443363563599041,
189 643588798496057020)
190 self.check_division(1115141373653752303710932756325578065,
191 1038556335171453937726882627)
192 # 30-bit cases that require the post-subtraction correction step
193 self.check_division(922498905405436751940989320930368494,
194 949985870686786135626943396)
195 self.check_division(768235853328091167204009652174031844,
196 1091555541180371554426545266)
197
198 # 15-bit cases involving a quotient digit estimate of BASE+1
199 self.check_division(20172188947443, 615611397)
200 self.check_division(1020908530270155025, 950795710)
201 self.check_division(128589565723112408, 736393718)
202 self.check_division(609919780285761575, 18613274546784)
203 # 15-bit cases that require the post-subtraction correction step
204 self.check_division(710031681576388032, 26769404391308)
205 self.check_division(1933622614268221, 30212853348836)
206
207
208
Walter Dörwalda0021592005-06-13 21:44:48 +0000209 def test_karatsuba(self):
Guido van Rossum805365e2007-05-07 22:24:25 +0000210 digits = list(range(1, 5)) + list(range(KARATSUBA_CUTOFF,
211 KARATSUBA_CUTOFF + 10))
Walter Dörwalda0021592005-06-13 21:44:48 +0000212 digits.extend([KARATSUBA_CUTOFF * 10, KARATSUBA_CUTOFF * 100])
Guido van Rossum4365cab1998-08-13 14:20:17 +0000213
Walter Dörwalda0021592005-06-13 21:44:48 +0000214 bits = [digit * SHIFT for digit in digits]
Guido van Rossum4365cab1998-08-13 14:20:17 +0000215
Walter Dörwalda0021592005-06-13 21:44:48 +0000216 # Test products of long strings of 1 bits -- (2**x-1)*(2**y-1) ==
217 # 2**(x+y) - 2**x - 2**y + 1, so the proper result is easy to check.
218 for abits in bits:
Guido van Rossume2a383d2007-01-15 16:59:06 +0000219 a = (1 << abits) - 1
Walter Dörwalda0021592005-06-13 21:44:48 +0000220 for bbits in bits:
221 if bbits < abits:
222 continue
Martin Pantercbe16ae2015-09-25 23:50:47 +0000223 with self.subTest(abits=abits, bbits=bbits):
224 b = (1 << bbits) - 1
225 x = a * b
226 y = ((1 << (abits + bbits)) -
227 (1 << abits) -
228 (1 << bbits) +
229 1)
230 self.assertEqual(x, y)
Tim Peters7f270ba2002-08-13 21:06:55 +0000231
Walter Dörwalda0021592005-06-13 21:44:48 +0000232 def check_bitop_identities_1(self, x):
233 eq = self.assertEqual
Martin Pantercbe16ae2015-09-25 23:50:47 +0000234 with self.subTest(x=x):
235 eq(x & 0, 0)
236 eq(x | 0, x)
237 eq(x ^ 0, x)
238 eq(x & -1, x)
239 eq(x | -1, -1)
240 eq(x ^ -1, ~x)
241 eq(x, ~~x)
242 eq(x & x, x)
243 eq(x | x, x)
244 eq(x ^ x, 0)
245 eq(x & ~x, 0)
246 eq(x | ~x, -1)
247 eq(x ^ ~x, -1)
248 eq(-x, 1 + ~x)
249 eq(-x, ~(x-1))
Guido van Rossum805365e2007-05-07 22:24:25 +0000250 for n in range(2*SHIFT):
Guido van Rossume2a383d2007-01-15 16:59:06 +0000251 p2 = 2 ** n
Martin Pantercbe16ae2015-09-25 23:50:47 +0000252 with self.subTest(x=x, n=n, p2=p2):
253 eq(x << n >> n, x)
254 eq(x // p2, x >> n)
255 eq(x * p2, x << n)
256 eq(x & -p2, x >> n << n)
257 eq(x & -p2, x & ~(p2 - 1))
Tim Peters7f270ba2002-08-13 21:06:55 +0000258
Walter Dörwalda0021592005-06-13 21:44:48 +0000259 def check_bitop_identities_2(self, x, y):
260 eq = self.assertEqual
Martin Pantercbe16ae2015-09-25 23:50:47 +0000261 with self.subTest(x=x, y=y):
262 eq(x & y, y & x)
263 eq(x | y, y | x)
264 eq(x ^ y, y ^ x)
265 eq(x ^ y ^ x, y)
266 eq(x & y, ~(~x | ~y))
267 eq(x | y, ~(~x & ~y))
268 eq(x ^ y, (x | y) & ~(x & y))
269 eq(x ^ y, (x & ~y) | (~x & y))
270 eq(x ^ y, (x | y) & (~x | ~y))
Tim Peters7f270ba2002-08-13 21:06:55 +0000271
Walter Dörwalda0021592005-06-13 21:44:48 +0000272 def check_bitop_identities_3(self, x, y, z):
273 eq = self.assertEqual
Martin Pantercbe16ae2015-09-25 23:50:47 +0000274 with self.subTest(x=x, y=y, z=z):
275 eq((x & y) & z, x & (y & z))
276 eq((x | y) | z, x | (y | z))
277 eq((x ^ y) ^ z, x ^ (y ^ z))
278 eq(x & (y | z), (x & y) | (x & z))
279 eq(x | (y & z), (x | y) & (x | z))
Tim Peters7f270ba2002-08-13 21:06:55 +0000280
Walter Dörwalda0021592005-06-13 21:44:48 +0000281 def test_bitop_identities(self):
282 for x in special:
283 self.check_bitop_identities_1(x)
Guido van Rossum805365e2007-05-07 22:24:25 +0000284 digits = range(1, MAXDIGITS+1)
Walter Dörwalda0021592005-06-13 21:44:48 +0000285 for lenx in digits:
286 x = self.getran(lenx)
287 self.check_bitop_identities_1(x)
288 for leny in digits:
289 y = self.getran(leny)
290 self.check_bitop_identities_2(x, y)
291 self.check_bitop_identities_3(x, y, self.getran((lenx + leny)//2))
Guido van Rossum4365cab1998-08-13 14:20:17 +0000292
Walter Dörwalda0021592005-06-13 21:44:48 +0000293 def slow_format(self, x, base):
Walter Dörwalda0021592005-06-13 21:44:48 +0000294 digits = []
295 sign = 0
296 if x < 0:
297 sign, x = 1, -x
298 while x:
299 x, r = divmod(x, base)
300 digits.append(int(r))
301 digits.reverse()
302 digits = digits or [0]
303 return '-'[:sign] + \
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000304 {2: '0b', 8: '0o', 10: '', 16: '0x'}[base] + \
Georg Brandlcbd2ab12010-12-04 10:39:14 +0000305 "".join("0123456789abcdef"[i] for i in digits)
Guido van Rossum4365cab1998-08-13 14:20:17 +0000306
Walter Dörwalda0021592005-06-13 21:44:48 +0000307 def check_format_1(self, x):
Serhiy Storchaka95949422013-08-27 19:40:23 +0300308 for base, mapper in (2, bin), (8, oct), (10, str), (10, repr), (16, hex):
Walter Dörwalda0021592005-06-13 21:44:48 +0000309 got = mapper(x)
Martin Pantercbe16ae2015-09-25 23:50:47 +0000310 with self.subTest(x=x, mapper=mapper.__name__):
311 expected = self.slow_format(x, base)
312 self.assertEqual(got, expected)
313 with self.subTest(got=got):
314 self.assertEqual(int(got, 0), x)
Guido van Rossum4365cab1998-08-13 14:20:17 +0000315
Walter Dörwalda0021592005-06-13 21:44:48 +0000316 def test_format(self):
317 for x in special:
318 self.check_format_1(x)
Guido van Rossum805365e2007-05-07 22:24:25 +0000319 for i in range(10):
320 for lenx in range(1, MAXDIGITS+1):
Walter Dörwalda0021592005-06-13 21:44:48 +0000321 x = self.getran(lenx)
322 self.check_format_1(x)
Guido van Rossum4365cab1998-08-13 14:20:17 +0000323
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000324 def test_long(self):
Mark Dickinson5c2db372009-12-05 20:28:34 +0000325 # Check conversions from string
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000326 LL = [
327 ('1' + '0'*20, 10**20),
328 ('1' + '0'*100, 10**100)
329 ]
Mark Dickinson5c2db372009-12-05 20:28:34 +0000330 for s, v in LL:
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000331 for sign in "", "+", "-":
332 for prefix in "", " ", "\t", " \t\t ":
333 ss = prefix + sign + s
334 vv = v
335 if sign == "-" and v is not ValueError:
336 vv = -v
337 try:
Mark Dickinson5c2db372009-12-05 20:28:34 +0000338 self.assertEqual(int(ss), vv)
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000339 except ValueError:
340 pass
341
Mark Dickinson9ffc0202009-01-20 20:45:53 +0000342 # trailing L should no longer be accepted...
343 self.assertRaises(ValueError, int, '123L')
344 self.assertRaises(ValueError, int, '123l')
345 self.assertRaises(ValueError, int, '0L')
346 self.assertRaises(ValueError, int, '-37L')
347 self.assertRaises(ValueError, int, '0x32L', 16)
348 self.assertRaises(ValueError, int, '1L', 21)
349 # ... but it's just a normal digit if base >= 22
350 self.assertEqual(int('1L', 22), 43)
351
Mark Dickinson56544db2010-05-26 19:14:01 +0000352 # tests with base 0
353 self.assertEqual(int('000', 0), 0)
354 self.assertEqual(int('0o123', 0), 83)
355 self.assertEqual(int('0x123', 0), 291)
356 self.assertEqual(int('0b100', 0), 4)
357 self.assertEqual(int(' 0O123 ', 0), 83)
358 self.assertEqual(int(' 0X123 ', 0), 291)
359 self.assertEqual(int(' 0B100 ', 0), 4)
360 self.assertEqual(int('0', 0), 0)
361 self.assertEqual(int('+0', 0), 0)
362 self.assertEqual(int('-0', 0), 0)
363 self.assertEqual(int('00', 0), 0)
364 self.assertRaises(ValueError, int, '08', 0)
365 self.assertRaises(ValueError, int, '-012395', 0)
366
Mark Dickinsonf9a5a8e2010-05-26 20:07:58 +0000367 # invalid bases
368 invalid_bases = [-909,
369 2**31-1, 2**31, -2**31, -2**31-1,
370 2**63-1, 2**63, -2**63, -2**63-1,
371 2**100, -2**100,
372 ]
373 for base in invalid_bases:
374 self.assertRaises(ValueError, int, '42', base)
375
376
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000377 def test_conversion(self):
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000378
Mark Dickinson5c2db372009-12-05 20:28:34 +0000379 class JustLong:
380 # test that __long__ no longer used in 3.x
381 def __long__(self):
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000382 return 42
Mark Dickinson5c2db372009-12-05 20:28:34 +0000383 self.assertRaises(TypeError, int, JustLong())
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000384
Mark Dickinson5c2db372009-12-05 20:28:34 +0000385 class LongTrunc:
386 # __long__ should be ignored in 3.x
387 def __long__(self):
Christian Heimes81ee3ef2008-05-04 22:42:01 +0000388 return 42
Mark Dickinson5c2db372009-12-05 20:28:34 +0000389 def __trunc__(self):
390 return 1729
391 self.assertEqual(int(LongTrunc()), 1729)
Tim Peters26c7fa32001-08-23 22:56:21 +0000392
Mark Dickinson30970e92011-10-23 20:07:13 +0100393 def check_float_conversion(self, n):
394 # Check that int -> float conversion behaviour matches
395 # that of the pure Python version above.
396 try:
397 actual = float(n)
398 except OverflowError:
399 actual = 'overflow'
400
401 try:
402 expected = int_to_float(n)
403 except OverflowError:
404 expected = 'overflow'
405
406 msg = ("Error in conversion of integer {} to float. "
407 "Got {}, expected {}.".format(n, actual, expected))
408 self.assertEqual(actual, expected, msg)
409
Eric Smith3ab08ca2010-12-04 15:17:38 +0000410 @support.requires_IEEE_754
Mark Dickinsonc6300392009-04-20 21:38:00 +0000411 def test_float_conversion(self):
Mark Dickinsonc6300392009-04-20 21:38:00 +0000412
413 exact_values = [0, 1, 2,
414 2**53-3,
415 2**53-2,
416 2**53-1,
417 2**53,
418 2**53+2,
419 2**54-4,
420 2**54-2,
421 2**54,
422 2**54+4]
423 for x in exact_values:
424 self.assertEqual(float(x), x)
425 self.assertEqual(float(-x), -x)
426
427 # test round-half-even
428 for x, y in [(1, 0), (2, 2), (3, 4), (4, 4), (5, 4), (6, 6), (7, 8)]:
429 for p in range(15):
430 self.assertEqual(int(float(2**p*(2**53+x))), 2**p*(2**53+y))
431
432 for x, y in [(0, 0), (1, 0), (2, 0), (3, 4), (4, 4), (5, 4), (6, 8),
433 (7, 8), (8, 8), (9, 8), (10, 8), (11, 12), (12, 12),
434 (13, 12), (14, 16), (15, 16)]:
435 for p in range(15):
436 self.assertEqual(int(float(2**p*(2**54+x))), 2**p*(2**54+y))
437
438 # behaviour near extremes of floating-point range
439 int_dbl_max = int(DBL_MAX)
440 top_power = 2**DBL_MAX_EXP
441 halfway = (int_dbl_max + top_power)//2
442 self.assertEqual(float(int_dbl_max), DBL_MAX)
443 self.assertEqual(float(int_dbl_max+1), DBL_MAX)
444 self.assertEqual(float(halfway-1), DBL_MAX)
445 self.assertRaises(OverflowError, float, halfway)
446 self.assertEqual(float(1-halfway), -DBL_MAX)
447 self.assertRaises(OverflowError, float, -halfway)
448 self.assertRaises(OverflowError, float, top_power-1)
449 self.assertRaises(OverflowError, float, top_power)
450 self.assertRaises(OverflowError, float, top_power+1)
451 self.assertRaises(OverflowError, float, 2*top_power-1)
452 self.assertRaises(OverflowError, float, 2*top_power)
453 self.assertRaises(OverflowError, float, top_power*top_power)
454
455 for p in range(100):
456 x = 2**p * (2**53 + 1) + 1
457 y = 2**p * (2**53 + 2)
458 self.assertEqual(int(float(x)), y)
459
460 x = 2**p * (2**53 + 1)
461 y = 2**p * 2**53
462 self.assertEqual(int(float(x)), y)
463
Mark Dickinson30970e92011-10-23 20:07:13 +0100464 # Compare builtin float conversion with pure Python int_to_float
465 # function above.
466 test_values = [
467 int_dbl_max-1, int_dbl_max, int_dbl_max+1,
468 halfway-1, halfway, halfway + 1,
469 top_power-1, top_power, top_power+1,
470 2*top_power-1, 2*top_power, top_power*top_power,
471 ]
472 test_values.extend(exact_values)
473 for p in range(-4, 8):
474 for x in range(-128, 128):
475 test_values.append(2**(p+53) + x)
476 for value in test_values:
477 self.check_float_conversion(value)
478 self.check_float_conversion(-value)
479
Walter Dörwalda0021592005-06-13 21:44:48 +0000480 def test_float_overflow(self):
Walter Dörwalda0021592005-06-13 21:44:48 +0000481 for x in -2.0, -1.0, 0.0, 1.0, 2.0:
Guido van Rossume2a383d2007-01-15 16:59:06 +0000482 self.assertEqual(float(int(x)), x)
Tim Peters9fffa3e2001-09-04 05:14:19 +0000483
Walter Dörwalda0021592005-06-13 21:44:48 +0000484 shuge = '12345' * 120
Guido van Rossume2a383d2007-01-15 16:59:06 +0000485 huge = 1 << 30000
Walter Dörwalda0021592005-06-13 21:44:48 +0000486 mhuge = -huge
487 namespace = {'huge': huge, 'mhuge': mhuge, 'shuge': shuge, 'math': math}
488 for test in ["float(huge)", "float(mhuge)",
489 "complex(huge)", "complex(mhuge)",
490 "complex(huge, 1)", "complex(mhuge, 1)",
491 "complex(1, huge)", "complex(1, mhuge)",
492 "1. + huge", "huge + 1.", "1. + mhuge", "mhuge + 1.",
493 "1. - huge", "huge - 1.", "1. - mhuge", "mhuge - 1.",
494 "1. * huge", "huge * 1.", "1. * mhuge", "mhuge * 1.",
495 "1. // huge", "huge // 1.", "1. // mhuge", "mhuge // 1.",
496 "1. / huge", "huge / 1.", "1. / mhuge", "mhuge / 1.",
497 "1. ** huge", "huge ** 1.", "1. ** mhuge", "mhuge ** 1.",
498 "math.sin(huge)", "math.sin(mhuge)",
499 "math.sqrt(huge)", "math.sqrt(mhuge)", # should do better
Guido van Rossum28bbe422007-08-24 03:46:30 +0000500 # math.floor() of an int returns an int now
501 ##"math.floor(huge)", "math.floor(mhuge)",
502 ]:
Tim Peters9fffa3e2001-09-04 05:14:19 +0000503
Walter Dörwalda0021592005-06-13 21:44:48 +0000504 self.assertRaises(OverflowError, eval, test, namespace)
Tim Peters9fffa3e2001-09-04 05:14:19 +0000505
Mark Dickinson5c2db372009-12-05 20:28:34 +0000506 # XXX Perhaps float(shuge) can raise OverflowError on some box?
507 # The comparison should not.
508 self.assertNotEqual(float(shuge), int(shuge),
509 "float(shuge) should not equal int(shuge)")
Tim Peters83e7ccc2001-09-04 06:37:28 +0000510
Walter Dörwalda0021592005-06-13 21:44:48 +0000511 def test_logs(self):
Walter Dörwalda0021592005-06-13 21:44:48 +0000512 LOG10E = math.log10(math.e)
Tim Peters307fa782004-09-23 08:06:40 +0000513
Guido van Rossum805365e2007-05-07 22:24:25 +0000514 for exp in list(range(10)) + [100, 1000, 10000]:
Walter Dörwalda0021592005-06-13 21:44:48 +0000515 value = 10 ** exp
516 log10 = math.log10(value)
517 self.assertAlmostEqual(log10, exp)
Tim Peters78526162001-09-05 00:53:45 +0000518
Walter Dörwalda0021592005-06-13 21:44:48 +0000519 # log10(value) == exp, so log(value) == log10(value)/log10(e) ==
520 # exp/LOG10E
521 expected = exp / LOG10E
522 log = math.log(value)
523 self.assertAlmostEqual(log, expected)
Tim Peters78526162001-09-05 00:53:45 +0000524
Guido van Rossume2a383d2007-01-15 16:59:06 +0000525 for bad in -(1 << 10000), -2, 0:
Walter Dörwalda0021592005-06-13 21:44:48 +0000526 self.assertRaises(ValueError, math.log, bad)
527 self.assertRaises(ValueError, math.log10, bad)
Tim Peters78526162001-09-05 00:53:45 +0000528
Walter Dörwalda0021592005-06-13 21:44:48 +0000529 def test_mixed_compares(self):
530 eq = self.assertEqual
Tim Peters78526162001-09-05 00:53:45 +0000531
Serhiy Storchaka95949422013-08-27 19:40:23 +0300532 # We're mostly concerned with that mixing floats and ints does the
533 # right stuff, even when ints are too large to fit in a float.
Walter Dörwalda0021592005-06-13 21:44:48 +0000534 # The safest way to check the results is to use an entirely different
535 # method, which we do here via a skeletal rational class (which
Serhiy Storchaka95949422013-08-27 19:40:23 +0300536 # represents all Python ints and floats exactly).
Walter Dörwalda0021592005-06-13 21:44:48 +0000537 class Rat:
538 def __init__(self, value):
Walter Dörwaldaa97f042007-05-03 21:05:51 +0000539 if isinstance(value, int):
Walter Dörwalda0021592005-06-13 21:44:48 +0000540 self.n = value
541 self.d = 1
542 elif isinstance(value, float):
543 # Convert to exact rational equivalent.
544 f, e = math.frexp(abs(value))
545 assert f == 0 or 0.5 <= f < 1.0
546 # |value| = f * 2**e exactly
Tim Peters78526162001-09-05 00:53:45 +0000547
Walter Dörwalda0021592005-06-13 21:44:48 +0000548 # Suck up CHUNK bits at a time; 28 is enough so that we suck
549 # up all bits in 2 iterations for all known binary double-
550 # precision formats, and small enough to fit in an int.
551 CHUNK = 28
552 top = 0
553 # invariant: |value| = (top + f) * 2**e exactly
554 while f:
555 f = math.ldexp(f, CHUNK)
556 digit = int(f)
557 assert digit >> CHUNK == 0
558 top = (top << CHUNK) | digit
559 f -= digit
560 assert 0.0 <= f < 1.0
561 e -= CHUNK
Tim Peters78526162001-09-05 00:53:45 +0000562
Walter Dörwalda0021592005-06-13 21:44:48 +0000563 # Now |value| = top * 2**e exactly.
564 if e >= 0:
565 n = top << e
566 d = 1
567 else:
568 n = top
569 d = 1 << -e
570 if value < 0:
571 n = -n
572 self.n = n
573 self.d = d
574 assert float(n) / float(d) == value
Tim Peters307fa782004-09-23 08:06:40 +0000575 else:
Georg Brandl89fad142010-03-14 10:23:39 +0000576 raise TypeError("can't deal with %r" % value)
Tim Peters307fa782004-09-23 08:06:40 +0000577
Benjamin Peterson60192082008-10-16 19:34:46 +0000578 def _cmp__(self, other):
Walter Dörwalda0021592005-06-13 21:44:48 +0000579 if not isinstance(other, Rat):
580 other = Rat(other)
Mark Dickinsona56c4672009-01-27 18:17:45 +0000581 x, y = self.n * other.d, self.d * other.n
582 return (x > y) - (x < y)
Benjamin Peterson60192082008-10-16 19:34:46 +0000583 def __eq__(self, other):
584 return self._cmp__(other) == 0
585 def __ne__(self, other):
586 return self._cmp__(other) != 0
587 def __ge__(self, other):
588 return self._cmp__(other) >= 0
589 def __gt__(self, other):
590 return self._cmp__(other) > 0
591 def __le__(self, other):
592 return self._cmp__(other) <= 0
593 def __lt__(self, other):
594 return self._cmp__(other) < 0
Tim Peters307fa782004-09-23 08:06:40 +0000595
Walter Dörwalda0021592005-06-13 21:44:48 +0000596 cases = [0, 0.001, 0.99, 1.0, 1.5, 1e20, 1e200]
597 # 2**48 is an important boundary in the internals. 2**53 is an
598 # important boundary for IEEE double precision.
599 for t in 2.0**48, 2.0**50, 2.0**53:
600 cases.extend([t - 1.0, t - 0.3, t, t + 0.3, t + 1.0,
Guido van Rossume2a383d2007-01-15 16:59:06 +0000601 int(t-1), int(t), int(t+1)])
Christian Heimesa37d4c62007-12-04 23:02:19 +0000602 cases.extend([0, 1, 2, sys.maxsize, float(sys.maxsize)])
Mark Dickinson5c2db372009-12-05 20:28:34 +0000603 # 1 << 20000 should exceed all double formats. int(1e200) is to
Walter Dörwalda0021592005-06-13 21:44:48 +0000604 # check that we get equality with 1e200 above.
Guido van Rossume2a383d2007-01-15 16:59:06 +0000605 t = int(1e200)
606 cases.extend([0, 1, 2, 1 << 20000, t-1, t, t+1])
Walter Dörwalda0021592005-06-13 21:44:48 +0000607 cases.extend([-x for x in cases])
608 for x in cases:
609 Rx = Rat(x)
610 for y in cases:
611 Ry = Rat(y)
Mark Dickinsona56c4672009-01-27 18:17:45 +0000612 Rcmp = (Rx > Ry) - (Rx < Ry)
Martin Pantercbe16ae2015-09-25 23:50:47 +0000613 with self.subTest(x=x, y=y, Rcmp=Rcmp):
614 xycmp = (x > y) - (x < y)
615 eq(Rcmp, xycmp)
616 eq(x == y, Rcmp == 0)
617 eq(x != y, Rcmp != 0)
618 eq(x < y, Rcmp < 0)
619 eq(x <= y, Rcmp <= 0)
620 eq(x > y, Rcmp > 0)
621 eq(x >= y, Rcmp >= 0)
Tim Peters307fa782004-09-23 08:06:40 +0000622
Eric Smith0dd1b632008-02-11 17:55:01 +0000623 def test__format__(self):
Eric Smith8c663262007-08-25 02:26:07 +0000624 self.assertEqual(format(123456789, 'd'), '123456789')
625 self.assertEqual(format(123456789, 'd'), '123456789')
626
Eric Smith185e30c2007-08-30 22:23:08 +0000627 # sign and aligning are interdependent
628 self.assertEqual(format(1, "-"), '1')
629 self.assertEqual(format(-1, "-"), '-1')
630 self.assertEqual(format(1, "-3"), ' 1')
631 self.assertEqual(format(-1, "-3"), ' -1')
632 self.assertEqual(format(1, "+3"), ' +1')
633 self.assertEqual(format(-1, "+3"), ' -1')
634 self.assertEqual(format(1, " 3"), ' 1')
635 self.assertEqual(format(-1, " 3"), ' -1')
636 self.assertEqual(format(1, " "), ' 1')
637 self.assertEqual(format(-1, " "), '-1')
638
Eric Smith8c663262007-08-25 02:26:07 +0000639 # hex
640 self.assertEqual(format(3, "x"), "3")
641 self.assertEqual(format(3, "X"), "3")
642 self.assertEqual(format(1234, "x"), "4d2")
643 self.assertEqual(format(-1234, "x"), "-4d2")
644 self.assertEqual(format(1234, "8x"), " 4d2")
Eric Smith185e30c2007-08-30 22:23:08 +0000645 self.assertEqual(format(-1234, "8x"), " -4d2")
Eric Smith8c663262007-08-25 02:26:07 +0000646 self.assertEqual(format(1234, "x"), "4d2")
647 self.assertEqual(format(-1234, "x"), "-4d2")
648 self.assertEqual(format(-3, "x"), "-3")
649 self.assertEqual(format(-3, "X"), "-3")
650 self.assertEqual(format(int('be', 16), "x"), "be")
651 self.assertEqual(format(int('be', 16), "X"), "BE")
652 self.assertEqual(format(-int('be', 16), "x"), "-be")
653 self.assertEqual(format(-int('be', 16), "X"), "-BE")
654
655 # octal
656 self.assertEqual(format(3, "b"), "11")
657 self.assertEqual(format(-3, "b"), "-11")
658 self.assertEqual(format(1234, "b"), "10011010010")
659 self.assertEqual(format(-1234, "b"), "-10011010010")
660 self.assertEqual(format(1234, "-b"), "10011010010")
661 self.assertEqual(format(-1234, "-b"), "-10011010010")
662 self.assertEqual(format(1234, " b"), " 10011010010")
663 self.assertEqual(format(-1234, " b"), "-10011010010")
664 self.assertEqual(format(1234, "+b"), "+10011010010")
665 self.assertEqual(format(-1234, "+b"), "-10011010010")
666
Eric Smith8c663262007-08-25 02:26:07 +0000667 # make sure these are errors
668 self.assertRaises(ValueError, format, 3, "1.3") # precision disallowed
Eric Smith8c663262007-08-25 02:26:07 +0000669 self.assertRaises(ValueError, format, 3, "+c") # sign not allowed
670 # with 'c'
Eric Smithfa767ef2008-01-28 10:59:27 +0000671
672 # ensure that only int and float type specifiers work
Eric Smith7b69c6c2008-01-27 21:07:59 +0000673 for format_spec in ([chr(x) for x in range(ord('a'), ord('z')+1)] +
674 [chr(x) for x in range(ord('A'), ord('Z')+1)]):
Eric Smithfa767ef2008-01-28 10:59:27 +0000675 if not format_spec in 'bcdoxXeEfFgGn%':
Eric Smith7b69c6c2008-01-27 21:07:59 +0000676 self.assertRaises(ValueError, format, 0, format_spec)
677 self.assertRaises(ValueError, format, 1, format_spec)
678 self.assertRaises(ValueError, format, -1, format_spec)
679 self.assertRaises(ValueError, format, 2**100, format_spec)
680 self.assertRaises(ValueError, format, -(2**100), format_spec)
681
Eric Smithfa767ef2008-01-28 10:59:27 +0000682 # ensure that float type specifiers work; format converts
683 # the int to a float
Eric Smith5807c412008-05-11 21:00:57 +0000684 for format_spec in 'eEfFgG%':
Eric Smithfa767ef2008-01-28 10:59:27 +0000685 for value in [0, 1, -1, 100, -100, 1234567890, -1234567890]:
686 self.assertEqual(format(value, format_spec),
687 format(float(value), format_spec))
Eric Smith8c663262007-08-25 02:26:07 +0000688
Christian Heimesa34706f2008-01-04 03:06:10 +0000689 def test_nan_inf(self):
Christian Heimes1aa7b302008-01-04 03:22:53 +0000690 self.assertRaises(OverflowError, int, float('inf'))
Georg Brandl6aa2d1f2008-08-12 08:35:52 +0000691 self.assertRaises(OverflowError, int, float('-inf'))
692 self.assertRaises(ValueError, int, float('nan'))
Christian Heimesa34706f2008-01-04 03:06:10 +0000693
Benjamin Peterson875d4c02008-07-13 17:44:16 +0000694 def test_true_division(self):
695 huge = 1 << 40000
696 mhuge = -huge
697 self.assertEqual(huge / huge, 1.0)
698 self.assertEqual(mhuge / mhuge, 1.0)
699 self.assertEqual(huge / mhuge, -1.0)
700 self.assertEqual(mhuge / huge, -1.0)
701 self.assertEqual(1 / huge, 0.0)
702 self.assertEqual(1 / huge, 0.0)
703 self.assertEqual(1 / mhuge, 0.0)
704 self.assertEqual(1 / mhuge, 0.0)
705 self.assertEqual((666 * huge + (huge >> 1)) / huge, 666.5)
706 self.assertEqual((666 * mhuge + (mhuge >> 1)) / mhuge, 666.5)
707 self.assertEqual((666 * huge + (huge >> 1)) / mhuge, -666.5)
708 self.assertEqual((666 * mhuge + (mhuge >> 1)) / huge, -666.5)
709 self.assertEqual(huge / (huge << 1), 0.5)
710 self.assertEqual((1000000 * huge) / huge, 1000000)
711
712 namespace = {'huge': huge, 'mhuge': mhuge}
713
714 for overflow in ["float(huge)", "float(mhuge)",
715 "huge / 1", "huge / 2", "huge / -1", "huge / -2",
716 "mhuge / 100", "mhuge / 200"]:
717 self.assertRaises(OverflowError, eval, overflow, namespace)
718
719 for underflow in ["1 / huge", "2 / huge", "-1 / huge", "-2 / huge",
720 "100 / mhuge", "200 / mhuge"]:
721 result = eval(underflow, namespace)
722 self.assertEqual(result, 0.0,
723 "expected underflow to 0 from %r" % underflow)
724
725 for zero in ["huge / 0", "mhuge / 0"]:
726 self.assertRaises(ZeroDivisionError, eval, zero, namespace)
727
Mark Dickinsoncbb62742009-12-27 15:09:50 +0000728 def check_truediv(self, a, b, skip_small=True):
729 """Verify that the result of a/b is correctly rounded, by
730 comparing it with a pure Python implementation of correctly
731 rounded division. b should be nonzero."""
732
733 # skip check for small a and b: in this case, the current
734 # implementation converts the arguments to float directly and
735 # then applies a float division. This can give doubly-rounded
736 # results on x87-using machines (particularly 32-bit Linux).
737 if skip_small and max(abs(a), abs(b)) < 2**DBL_MANT_DIG:
738 return
739
740 try:
741 # use repr so that we can distinguish between -0.0 and 0.0
742 expected = repr(truediv(a, b))
743 except OverflowError:
744 expected = 'overflow'
745 except ZeroDivisionError:
746 expected = 'zerodivision'
747
748 try:
749 got = repr(a / b)
750 except OverflowError:
751 got = 'overflow'
752 except ZeroDivisionError:
753 got = 'zerodivision'
754
Mark Dickinson2cfda802009-12-27 21:34:05 +0000755 self.assertEqual(expected, got, "Incorrectly rounded division {}/{}: "
756 "expected {}, got {}".format(a, b, expected, got))
Mark Dickinsoncbb62742009-12-27 15:09:50 +0000757
Eric Smith3ab08ca2010-12-04 15:17:38 +0000758 @support.requires_IEEE_754
Mark Dickinsoncbb62742009-12-27 15:09:50 +0000759 def test_correctly_rounded_true_division(self):
760 # more stringent tests than those above, checking that the
761 # result of true division of ints is always correctly rounded.
762 # This test should probably be considered CPython-specific.
763
764 # Exercise all the code paths not involving Gb-sized ints.
765 # ... divisions involving zero
766 self.check_truediv(123, 0)
767 self.check_truediv(-456, 0)
768 self.check_truediv(0, 3)
769 self.check_truediv(0, -3)
770 self.check_truediv(0, 0)
771 # ... overflow or underflow by large margin
772 self.check_truediv(671 * 12345 * 2**DBL_MAX_EXP, 12345)
773 self.check_truediv(12345, 345678 * 2**(DBL_MANT_DIG - DBL_MIN_EXP))
774 # ... a much larger or smaller than b
775 self.check_truediv(12345*2**100, 98765)
776 self.check_truediv(12345*2**30, 98765*7**81)
777 # ... a / b near a boundary: one of 1, 2**DBL_MANT_DIG, 2**DBL_MIN_EXP,
778 # 2**DBL_MAX_EXP, 2**(DBL_MIN_EXP-DBL_MANT_DIG)
779 bases = (0, DBL_MANT_DIG, DBL_MIN_EXP,
780 DBL_MAX_EXP, DBL_MIN_EXP - DBL_MANT_DIG)
781 for base in bases:
782 for exp in range(base - 15, base + 15):
783 self.check_truediv(75312*2**max(exp, 0), 69187*2**max(-exp, 0))
784 self.check_truediv(69187*2**max(exp, 0), 75312*2**max(-exp, 0))
785
786 # overflow corner case
787 for m in [1, 2, 7, 17, 12345, 7**100,
788 -1, -2, -5, -23, -67891, -41**50]:
789 for n in range(-10, 10):
790 self.check_truediv(m*DBL_MIN_OVERFLOW + n, m)
791 self.check_truediv(m*DBL_MIN_OVERFLOW + n, -m)
792
793 # check detection of inexactness in shifting stage
794 for n in range(250):
795 # (2**DBL_MANT_DIG+1)/(2**DBL_MANT_DIG) lies halfway
796 # between two representable floats, and would usually be
797 # rounded down under round-half-to-even. The tiniest of
798 # additions to the numerator should cause it to be rounded
799 # up instead.
800 self.check_truediv((2**DBL_MANT_DIG + 1)*12345*2**200 + 2**n,
801 2**DBL_MANT_DIG*12345)
802
803 # 1/2731 is one of the smallest division cases that's subject
804 # to double rounding on IEEE 754 machines working internally with
805 # 64-bit precision. On such machines, the next check would fail,
806 # were it not explicitly skipped in check_truediv.
807 self.check_truediv(1, 2731)
808
809 # a particularly bad case for the old algorithm: gives an
810 # error of close to 3.5 ulps.
811 self.check_truediv(295147931372582273023, 295147932265116303360)
812 for i in range(1000):
813 self.check_truediv(10**(i+1), 10**i)
814 self.check_truediv(10**i, 10**(i+1))
815
816 # test round-half-to-even behaviour, normal result
817 for m in [1, 2, 4, 7, 8, 16, 17, 32, 12345, 7**100,
818 -1, -2, -5, -23, -67891, -41**50]:
819 for n in range(-10, 10):
820 self.check_truediv(2**DBL_MANT_DIG*m + n, m)
821
822 # test round-half-to-even, subnormal result
823 for n in range(-20, 20):
824 self.check_truediv(n, 2**1076)
825
826 # largeish random divisions: a/b where |a| <= |b| <=
827 # 2*|a|; |ans| is between 0.5 and 1.0, so error should
828 # always be bounded by 2**-54 with equality possible only
829 # if the least significant bit of q=ans*2**53 is zero.
830 for M in [10**10, 10**100, 10**1000]:
831 for i in range(1000):
832 a = random.randrange(1, M)
833 b = random.randrange(a, 2*a+1)
834 self.check_truediv(a, b)
835 self.check_truediv(-a, b)
836 self.check_truediv(a, -b)
837 self.check_truediv(-a, -b)
838
839 # and some (genuinely) random tests
840 for _ in range(10000):
841 a_bits = random.randrange(1000)
842 b_bits = random.randrange(1, 1000)
843 x = random.randrange(2**a_bits)
844 y = random.randrange(1, 2**b_bits)
845 self.check_truediv(x, y)
846 self.check_truediv(x, -y)
847 self.check_truediv(-x, y)
848 self.check_truediv(-x, -y)
Benjamin Peterson875d4c02008-07-13 17:44:16 +0000849
Facundo Batista6e6f59b2008-07-24 18:57:11 +0000850 def test_small_ints(self):
851 for i in range(-5, 257):
Serhiy Storchaka3a20a5d2014-02-08 14:28:33 +0200852 self.assertIs(i, i + 0)
853 self.assertIs(i, i * 1)
854 self.assertIs(i, i - 0)
855 self.assertIs(i, i // 1)
856 self.assertIs(i, i & -1)
857 self.assertIs(i, i | 0)
858 self.assertIs(i, i ^ 0)
859 self.assertIs(i, ~~i)
860 self.assertIs(i, i**1)
861 self.assertIs(i, int(str(i)))
862 self.assertIs(i, i<<2>>2, str(i))
Facundo Batista6e6f59b2008-07-24 18:57:11 +0000863 # corner cases
864 i = 1 << 70
Serhiy Storchaka3a20a5d2014-02-08 14:28:33 +0200865 self.assertIs(i - i, 0)
866 self.assertIs(0 * i, 0)
Facundo Batista6e6f59b2008-07-24 18:57:11 +0000867
Mark Dickinson54bc1ec2008-12-17 16:19:07 +0000868 def test_bit_length(self):
869 tiny = 1e-10
870 for x in range(-65000, 65000):
871 k = x.bit_length()
872 # Check equivalence with Python version
873 self.assertEqual(k, len(bin(x).lstrip('-0b')))
874 # Behaviour as specified in the docs
875 if x != 0:
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000876 self.assertTrue(2**(k-1) <= abs(x) < 2**k)
Mark Dickinson54bc1ec2008-12-17 16:19:07 +0000877 else:
878 self.assertEqual(k, 0)
879 # Alternative definition: x.bit_length() == 1 + floor(log_2(x))
880 if x != 0:
881 # When x is an exact power of 2, numeric errors can
882 # cause floor(log(x)/log(2)) to be one too small; for
883 # small x this can be fixed by adding a small quantity
884 # to the quotient before taking the floor.
885 self.assertEqual(k, 1 + math.floor(
886 math.log(abs(x))/math.log(2) + tiny))
887
888 self.assertEqual((0).bit_length(), 0)
889 self.assertEqual((1).bit_length(), 1)
890 self.assertEqual((-1).bit_length(), 1)
891 self.assertEqual((2).bit_length(), 2)
892 self.assertEqual((-2).bit_length(), 2)
893 for i in [2, 3, 15, 16, 17, 31, 32, 33, 63, 64, 234]:
894 a = 2**i
895 self.assertEqual((a-1).bit_length(), i)
896 self.assertEqual((1-a).bit_length(), i)
897 self.assertEqual((a).bit_length(), i+1)
898 self.assertEqual((-a).bit_length(), i+1)
899 self.assertEqual((a+1).bit_length(), i+1)
900 self.assertEqual((-a-1).bit_length(), i+1)
901
Mark Dickinson1124e712009-01-28 21:25:58 +0000902 def test_round(self):
903 # check round-half-even algorithm. For round to nearest ten;
904 # rounding map is invariant under adding multiples of 20
905 test_dict = {0:0, 1:0, 2:0, 3:0, 4:0, 5:0,
906 6:10, 7:10, 8:10, 9:10, 10:10, 11:10, 12:10, 13:10, 14:10,
907 15:20, 16:20, 17:20, 18:20, 19:20}
908 for offset in range(-520, 520, 20):
909 for k, v in test_dict.items():
910 got = round(k+offset, -1)
911 expected = v+offset
912 self.assertEqual(got, expected)
Serhiy Storchaka3a20a5d2014-02-08 14:28:33 +0200913 self.assertIs(type(got), int)
Mark Dickinson1124e712009-01-28 21:25:58 +0000914
915 # larger second argument
916 self.assertEqual(round(-150, -2), -200)
917 self.assertEqual(round(-149, -2), -100)
918 self.assertEqual(round(-51, -2), -100)
919 self.assertEqual(round(-50, -2), 0)
920 self.assertEqual(round(-49, -2), 0)
921 self.assertEqual(round(-1, -2), 0)
922 self.assertEqual(round(0, -2), 0)
923 self.assertEqual(round(1, -2), 0)
924 self.assertEqual(round(49, -2), 0)
925 self.assertEqual(round(50, -2), 0)
926 self.assertEqual(round(51, -2), 100)
927 self.assertEqual(round(149, -2), 100)
928 self.assertEqual(round(150, -2), 200)
929 self.assertEqual(round(250, -2), 200)
930 self.assertEqual(round(251, -2), 300)
931 self.assertEqual(round(172500, -3), 172000)
932 self.assertEqual(round(173500, -3), 174000)
933 self.assertEqual(round(31415926535, -1), 31415926540)
934 self.assertEqual(round(31415926535, -2), 31415926500)
935 self.assertEqual(round(31415926535, -3), 31415927000)
936 self.assertEqual(round(31415926535, -4), 31415930000)
937 self.assertEqual(round(31415926535, -5), 31415900000)
938 self.assertEqual(round(31415926535, -6), 31416000000)
939 self.assertEqual(round(31415926535, -7), 31420000000)
940 self.assertEqual(round(31415926535, -8), 31400000000)
941 self.assertEqual(round(31415926535, -9), 31000000000)
942 self.assertEqual(round(31415926535, -10), 30000000000)
943 self.assertEqual(round(31415926535, -11), 0)
944 self.assertEqual(round(31415926535, -12), 0)
945 self.assertEqual(round(31415926535, -999), 0)
946
947 # should get correct results even for huge inputs
948 for k in range(10, 100):
949 got = round(10**k + 324678, -3)
950 expect = 10**k + 325000
951 self.assertEqual(got, expect)
Serhiy Storchaka3a20a5d2014-02-08 14:28:33 +0200952 self.assertIs(type(got), int)
Mark Dickinson1124e712009-01-28 21:25:58 +0000953
954 # nonnegative second argument: round(x, n) should just return x
955 for n in range(5):
956 for i in range(100):
957 x = random.randrange(-10000, 10000)
958 got = round(x, n)
959 self.assertEqual(got, x)
Serhiy Storchaka3a20a5d2014-02-08 14:28:33 +0200960 self.assertIs(type(got), int)
Mark Dickinson1124e712009-01-28 21:25:58 +0000961 for huge_n in 2**31-1, 2**31, 2**63-1, 2**63, 2**100, 10**100:
962 self.assertEqual(round(8979323, huge_n), 8979323)
963
964 # omitted second argument
965 for i in range(100):
966 x = random.randrange(-10000, 10000)
967 got = round(x)
968 self.assertEqual(got, x)
Serhiy Storchaka3a20a5d2014-02-08 14:28:33 +0200969 self.assertIs(type(got), int)
Mark Dickinson1124e712009-01-28 21:25:58 +0000970
971 # bad second argument
972 bad_exponents = ('brian', 2.0, 0j, None)
973 for e in bad_exponents:
974 self.assertRaises(TypeError, round, 3, e)
975
Alexandre Vassalottic36c3782010-01-09 20:35:09 +0000976 def test_to_bytes(self):
977 def check(tests, byteorder, signed=False):
978 for test, expected in tests.items():
979 try:
980 self.assertEqual(
981 test.to_bytes(len(expected), byteorder, signed=signed),
982 expected)
983 except Exception as err:
984 raise AssertionError(
985 "failed to convert {0} with byteorder={1} and signed={2}"
986 .format(test, byteorder, signed)) from err
987
988 # Convert integers to signed big-endian byte arrays.
989 tests1 = {
990 0: b'\x00',
991 1: b'\x01',
992 -1: b'\xff',
993 -127: b'\x81',
994 -128: b'\x80',
995 -129: b'\xff\x7f',
996 127: b'\x7f',
997 129: b'\x00\x81',
998 -255: b'\xff\x01',
999 -256: b'\xff\x00',
1000 255: b'\x00\xff',
1001 256: b'\x01\x00',
1002 32767: b'\x7f\xff',
1003 -32768: b'\xff\x80\x00',
1004 65535: b'\x00\xff\xff',
1005 -65536: b'\xff\x00\x00',
1006 -8388608: b'\x80\x00\x00'
1007 }
1008 check(tests1, 'big', signed=True)
1009
1010 # Convert integers to signed little-endian byte arrays.
1011 tests2 = {
1012 0: b'\x00',
1013 1: b'\x01',
1014 -1: b'\xff',
1015 -127: b'\x81',
1016 -128: b'\x80',
1017 -129: b'\x7f\xff',
1018 127: b'\x7f',
1019 129: b'\x81\x00',
1020 -255: b'\x01\xff',
1021 -256: b'\x00\xff',
1022 255: b'\xff\x00',
1023 256: b'\x00\x01',
1024 32767: b'\xff\x7f',
1025 -32768: b'\x00\x80',
1026 65535: b'\xff\xff\x00',
1027 -65536: b'\x00\x00\xff',
1028 -8388608: b'\x00\x00\x80'
1029 }
1030 check(tests2, 'little', signed=True)
1031
1032 # Convert integers to unsigned big-endian byte arrays.
1033 tests3 = {
1034 0: b'\x00',
1035 1: b'\x01',
1036 127: b'\x7f',
1037 128: b'\x80',
1038 255: b'\xff',
1039 256: b'\x01\x00',
1040 32767: b'\x7f\xff',
1041 32768: b'\x80\x00',
1042 65535: b'\xff\xff',
1043 65536: b'\x01\x00\x00'
1044 }
1045 check(tests3, 'big', signed=False)
1046
1047 # Convert integers to unsigned little-endian byte arrays.
1048 tests4 = {
1049 0: b'\x00',
1050 1: b'\x01',
1051 127: b'\x7f',
1052 128: b'\x80',
1053 255: b'\xff',
1054 256: b'\x00\x01',
1055 32767: b'\xff\x7f',
1056 32768: b'\x00\x80',
1057 65535: b'\xff\xff',
1058 65536: b'\x00\x00\x01'
1059 }
1060 check(tests4, 'little', signed=False)
1061
1062 self.assertRaises(OverflowError, (256).to_bytes, 1, 'big', signed=False)
1063 self.assertRaises(OverflowError, (256).to_bytes, 1, 'big', signed=True)
1064 self.assertRaises(OverflowError, (256).to_bytes, 1, 'little', signed=False)
1065 self.assertRaises(OverflowError, (256).to_bytes, 1, 'little', signed=True)
Victor Stinner3fa1aae2013-03-26 01:14:08 +01001066 self.assertRaises(OverflowError, (-1).to_bytes, 2, 'big', signed=False)
Alexandre Vassalottic36c3782010-01-09 20:35:09 +00001067 self.assertRaises(OverflowError, (-1).to_bytes, 2, 'little', signed=False)
1068 self.assertEqual((0).to_bytes(0, 'big'), b'')
1069 self.assertEqual((1).to_bytes(5, 'big'), b'\x00\x00\x00\x00\x01')
1070 self.assertEqual((0).to_bytes(5, 'big'), b'\x00\x00\x00\x00\x00')
1071 self.assertEqual((-1).to_bytes(5, 'big', signed=True),
1072 b'\xff\xff\xff\xff\xff')
1073 self.assertRaises(OverflowError, (1).to_bytes, 0, 'big')
1074
1075 def test_from_bytes(self):
1076 def check(tests, byteorder, signed=False):
1077 for test, expected in tests.items():
1078 try:
1079 self.assertEqual(
1080 int.from_bytes(test, byteorder, signed=signed),
1081 expected)
1082 except Exception as err:
1083 raise AssertionError(
1084 "failed to convert {0} with byteorder={1!r} and signed={2}"
1085 .format(test, byteorder, signed)) from err
1086
1087 # Convert signed big-endian byte arrays to integers.
1088 tests1 = {
1089 b'': 0,
1090 b'\x00': 0,
1091 b'\x00\x00': 0,
1092 b'\x01': 1,
1093 b'\x00\x01': 1,
1094 b'\xff': -1,
1095 b'\xff\xff': -1,
1096 b'\x81': -127,
1097 b'\x80': -128,
1098 b'\xff\x7f': -129,
1099 b'\x7f': 127,
1100 b'\x00\x81': 129,
1101 b'\xff\x01': -255,
1102 b'\xff\x00': -256,
1103 b'\x00\xff': 255,
1104 b'\x01\x00': 256,
1105 b'\x7f\xff': 32767,
1106 b'\x80\x00': -32768,
1107 b'\x00\xff\xff': 65535,
1108 b'\xff\x00\x00': -65536,
1109 b'\x80\x00\x00': -8388608
1110 }
1111 check(tests1, 'big', signed=True)
1112
1113 # Convert signed little-endian byte arrays to integers.
1114 tests2 = {
1115 b'': 0,
1116 b'\x00': 0,
1117 b'\x00\x00': 0,
1118 b'\x01': 1,
1119 b'\x00\x01': 256,
1120 b'\xff': -1,
1121 b'\xff\xff': -1,
1122 b'\x81': -127,
1123 b'\x80': -128,
1124 b'\x7f\xff': -129,
1125 b'\x7f': 127,
1126 b'\x81\x00': 129,
1127 b'\x01\xff': -255,
1128 b'\x00\xff': -256,
1129 b'\xff\x00': 255,
1130 b'\x00\x01': 256,
1131 b'\xff\x7f': 32767,
1132 b'\x00\x80': -32768,
1133 b'\xff\xff\x00': 65535,
1134 b'\x00\x00\xff': -65536,
1135 b'\x00\x00\x80': -8388608
1136 }
1137 check(tests2, 'little', signed=True)
1138
1139 # Convert unsigned big-endian byte arrays to integers.
1140 tests3 = {
1141 b'': 0,
1142 b'\x00': 0,
1143 b'\x01': 1,
1144 b'\x7f': 127,
1145 b'\x80': 128,
1146 b'\xff': 255,
1147 b'\x01\x00': 256,
1148 b'\x7f\xff': 32767,
1149 b'\x80\x00': 32768,
1150 b'\xff\xff': 65535,
1151 b'\x01\x00\x00': 65536,
1152 }
1153 check(tests3, 'big', signed=False)
1154
1155 # Convert integers to unsigned little-endian byte arrays.
1156 tests4 = {
1157 b'': 0,
1158 b'\x00': 0,
1159 b'\x01': 1,
1160 b'\x7f': 127,
1161 b'\x80': 128,
1162 b'\xff': 255,
1163 b'\x00\x01': 256,
1164 b'\xff\x7f': 32767,
1165 b'\x00\x80': 32768,
1166 b'\xff\xff': 65535,
1167 b'\x00\x00\x01': 65536,
1168 }
1169 check(tests4, 'little', signed=False)
1170
1171 class myint(int):
1172 pass
1173
Serhiy Storchaka3a20a5d2014-02-08 14:28:33 +02001174 self.assertIs(type(myint.from_bytes(b'\x00', 'big')), myint)
Alexandre Vassalottic36c3782010-01-09 20:35:09 +00001175 self.assertEqual(myint.from_bytes(b'\x01', 'big'), 1)
Serhiy Storchaka3a20a5d2014-02-08 14:28:33 +02001176 self.assertIs(
1177 type(myint.from_bytes(b'\x00', 'big', signed=False)), myint)
Alexandre Vassalottic36c3782010-01-09 20:35:09 +00001178 self.assertEqual(myint.from_bytes(b'\x01', 'big', signed=False), 1)
Serhiy Storchaka3a20a5d2014-02-08 14:28:33 +02001179 self.assertIs(type(myint.from_bytes(b'\x00', 'little')), myint)
Alexandre Vassalottic36c3782010-01-09 20:35:09 +00001180 self.assertEqual(myint.from_bytes(b'\x01', 'little'), 1)
Serhiy Storchaka3a20a5d2014-02-08 14:28:33 +02001181 self.assertIs(type(myint.from_bytes(
1182 b'\x00', 'little', signed=False)), myint)
Alexandre Vassalottic36c3782010-01-09 20:35:09 +00001183 self.assertEqual(myint.from_bytes(b'\x01', 'little', signed=False), 1)
1184 self.assertEqual(
1185 int.from_bytes([255, 0, 0], 'big', signed=True), -65536)
1186 self.assertEqual(
1187 int.from_bytes((255, 0, 0), 'big', signed=True), -65536)
1188 self.assertEqual(int.from_bytes(
1189 bytearray(b'\xff\x00\x00'), 'big', signed=True), -65536)
1190 self.assertEqual(int.from_bytes(
1191 bytearray(b'\xff\x00\x00'), 'big', signed=True), -65536)
1192 self.assertEqual(int.from_bytes(
1193 array.array('B', b'\xff\x00\x00'), 'big', signed=True), -65536)
1194 self.assertEqual(int.from_bytes(
1195 memoryview(b'\xff\x00\x00'), 'big', signed=True), -65536)
1196 self.assertRaises(ValueError, int.from_bytes, [256], 'big')
1197 self.assertRaises(ValueError, int.from_bytes, [0], 'big\x00')
1198 self.assertRaises(ValueError, int.from_bytes, [0], 'little\x00')
1199 self.assertRaises(TypeError, int.from_bytes, "", 'big')
1200 self.assertRaises(TypeError, int.from_bytes, "\x00", 'big')
1201 self.assertRaises(TypeError, int.from_bytes, 0, 'big')
1202 self.assertRaises(TypeError, int.from_bytes, 0, 'big', True)
1203 self.assertRaises(TypeError, myint.from_bytes, "", 'big')
1204 self.assertRaises(TypeError, myint.from_bytes, "\x00", 'big')
1205 self.assertRaises(TypeError, myint.from_bytes, 0, 'big')
1206 self.assertRaises(TypeError, int.from_bytes, 0, 'big', True)
Mark Dickinson1124e712009-01-28 21:25:58 +00001207
Mark Dickinsonbcc17ee2012-04-20 21:42:49 +01001208 def test_access_to_nonexistent_digit_0(self):
1209 # http://bugs.python.org/issue14630: A bug in _PyLong_Copy meant that
1210 # ob_digit[0] was being incorrectly accessed for instances of a
1211 # subclass of int, with value 0.
1212 class Integer(int):
1213 def __new__(cls, value=0):
1214 self = int.__new__(cls, value)
1215 self.foo = 'foo'
1216 return self
1217
1218 integers = [Integer(0) for i in range(1000)]
1219 for n in map(int, integers):
1220 self.assertEqual(n, 0)
1221
Victor Stinner7fe10492014-05-12 22:35:40 +02001222 def test_shift_bool(self):
1223 # Issue #21422: ensure that bool << int and bool >> int return int
1224 for value in (True, False):
1225 for shift in (0, 2):
1226 self.assertEqual(type(value << shift), int)
1227 self.assertEqual(type(value >> shift), int)
1228
Mark Dickinson54bc1ec2008-12-17 16:19:07 +00001229
Walter Dörwalda0021592005-06-13 21:44:48 +00001230def test_main():
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001231 support.run_unittest(LongTest)
Tim Peters307fa782004-09-23 08:06:40 +00001232
Walter Dörwalda0021592005-06-13 21:44:48 +00001233if __name__ == "__main__":
1234 test_main()