/Lib/test/test_int.py

http://unladen-swallow.googlecode.com/ · Python · 331 lines · 265 code · 42 blank · 24 comment · 16 complexity · 4639ae84b039077b0a2796732fcd7e30 MD5 · raw file

  1. import sys
  2. import unittest
  3. from test.test_support import run_unittest, have_unicode
  4. L = [
  5. ('0', 0),
  6. ('1', 1),
  7. ('9', 9),
  8. ('10', 10),
  9. ('99', 99),
  10. ('100', 100),
  11. ('314', 314),
  12. (' 314', 314),
  13. ('314 ', 314),
  14. (' \t\t 314 \t\t ', 314),
  15. (repr(sys.maxint), sys.maxint),
  16. (' 1x', ValueError),
  17. (' 1 ', 1),
  18. (' 1\02 ', ValueError),
  19. ('', ValueError),
  20. (' ', ValueError),
  21. (' \t\t ', ValueError)
  22. ]
  23. if have_unicode:
  24. L += [
  25. (unicode('0'), 0),
  26. (unicode('1'), 1),
  27. (unicode('9'), 9),
  28. (unicode('10'), 10),
  29. (unicode('99'), 99),
  30. (unicode('100'), 100),
  31. (unicode('314'), 314),
  32. (unicode(' 314'), 314),
  33. (unicode('\u0663\u0661\u0664 ','raw-unicode-escape'), 314),
  34. (unicode(' \t\t 314 \t\t '), 314),
  35. (unicode(' 1x'), ValueError),
  36. (unicode(' 1 '), 1),
  37. (unicode(' 1\02 '), ValueError),
  38. (unicode(''), ValueError),
  39. (unicode(' '), ValueError),
  40. (unicode(' \t\t '), ValueError),
  41. (unichr(0x200), ValueError),
  42. ]
  43. class IntTestCases(unittest.TestCase):
  44. def test_basic(self):
  45. self.assertEqual(int(314), 314)
  46. self.assertEqual(int(3.14), 3)
  47. self.assertEqual(int(314L), 314)
  48. # Check that conversion from float truncates towards zero
  49. self.assertEqual(int(-3.14), -3)
  50. self.assertEqual(int(3.9), 3)
  51. self.assertEqual(int(-3.9), -3)
  52. self.assertEqual(int(3.5), 3)
  53. self.assertEqual(int(-3.5), -3)
  54. # Different base:
  55. self.assertEqual(int("10",16), 16L)
  56. if have_unicode:
  57. self.assertEqual(int(unicode("10"),16), 16L)
  58. # Test conversion from strings and various anomalies
  59. for s, v in L:
  60. for sign in "", "+", "-":
  61. for prefix in "", " ", "\t", " \t\t ":
  62. ss = prefix + sign + s
  63. vv = v
  64. if sign == "-" and v is not ValueError:
  65. vv = -v
  66. try:
  67. self.assertEqual(int(ss), vv)
  68. except v:
  69. pass
  70. s = repr(-1-sys.maxint)
  71. x = int(s)
  72. self.assertEqual(x+1, -sys.maxint)
  73. self.assert_(isinstance(x, int))
  74. # should return long
  75. self.assertEqual(int(s[1:]), sys.maxint+1)
  76. # should return long
  77. x = int(1e100)
  78. self.assert_(isinstance(x, long))
  79. x = int(-1e100)
  80. self.assert_(isinstance(x, long))
  81. # SF bug 434186: 0x80000000/2 != 0x80000000>>1.
  82. # Worked by accident in Windows release build, but failed in debug build.
  83. # Failed in all Linux builds.
  84. x = -1-sys.maxint
  85. self.assertEqual(x >> 1, x//2)
  86. self.assertRaises(ValueError, int, '123\0')
  87. self.assertRaises(ValueError, int, '53', 40)
  88. # SF bug 1545497: embedded NULs were not detected with
  89. # explicit base
  90. self.assertRaises(ValueError, int, '123\0', 10)
  91. self.assertRaises(ValueError, int, '123\x00 245', 20)
  92. x = int('1' * 600)
  93. self.assert_(isinstance(x, long))
  94. if have_unicode:
  95. x = int(unichr(0x661) * 600)
  96. self.assert_(isinstance(x, long))
  97. self.assertRaises(TypeError, int, 1, 12)
  98. self.assertEqual(int('0123', 0), 83)
  99. self.assertEqual(int('0x123', 16), 291)
  100. # Bug 1679: "0x" is not a valid hex literal
  101. self.assertRaises(ValueError, int, "0x", 16)
  102. self.assertRaises(ValueError, int, "0x", 0)
  103. self.assertRaises(ValueError, int, "0o", 8)
  104. self.assertRaises(ValueError, int, "0o", 0)
  105. self.assertRaises(ValueError, int, "0b", 2)
  106. self.assertRaises(ValueError, int, "0b", 0)
  107. # SF bug 1334662: int(string, base) wrong answers
  108. # Various representations of 2**32 evaluated to 0
  109. # rather than 2**32 in previous versions
  110. self.assertEqual(int('100000000000000000000000000000000', 2), 4294967296L)
  111. self.assertEqual(int('102002022201221111211', 3), 4294967296L)
  112. self.assertEqual(int('10000000000000000', 4), 4294967296L)
  113. self.assertEqual(int('32244002423141', 5), 4294967296L)
  114. self.assertEqual(int('1550104015504', 6), 4294967296L)
  115. self.assertEqual(int('211301422354', 7), 4294967296L)
  116. self.assertEqual(int('40000000000', 8), 4294967296L)
  117. self.assertEqual(int('12068657454', 9), 4294967296L)
  118. self.assertEqual(int('4294967296', 10), 4294967296L)
  119. self.assertEqual(int('1904440554', 11), 4294967296L)
  120. self.assertEqual(int('9ba461594', 12), 4294967296L)
  121. self.assertEqual(int('535a79889', 13), 4294967296L)
  122. self.assertEqual(int('2ca5b7464', 14), 4294967296L)
  123. self.assertEqual(int('1a20dcd81', 15), 4294967296L)
  124. self.assertEqual(int('100000000', 16), 4294967296L)
  125. self.assertEqual(int('a7ffda91', 17), 4294967296L)
  126. self.assertEqual(int('704he7g4', 18), 4294967296L)
  127. self.assertEqual(int('4f5aff66', 19), 4294967296L)
  128. self.assertEqual(int('3723ai4g', 20), 4294967296L)
  129. self.assertEqual(int('281d55i4', 21), 4294967296L)
  130. self.assertEqual(int('1fj8b184', 22), 4294967296L)
  131. self.assertEqual(int('1606k7ic', 23), 4294967296L)
  132. self.assertEqual(int('mb994ag', 24), 4294967296L)
  133. self.assertEqual(int('hek2mgl', 25), 4294967296L)
  134. self.assertEqual(int('dnchbnm', 26), 4294967296L)
  135. self.assertEqual(int('b28jpdm', 27), 4294967296L)
  136. self.assertEqual(int('8pfgih4', 28), 4294967296L)
  137. self.assertEqual(int('76beigg', 29), 4294967296L)
  138. self.assertEqual(int('5qmcpqg', 30), 4294967296L)
  139. self.assertEqual(int('4q0jto4', 31), 4294967296L)
  140. self.assertEqual(int('4000000', 32), 4294967296L)
  141. self.assertEqual(int('3aokq94', 33), 4294967296L)
  142. self.assertEqual(int('2qhxjli', 34), 4294967296L)
  143. self.assertEqual(int('2br45qb', 35), 4294967296L)
  144. self.assertEqual(int('1z141z4', 36), 4294967296L)
  145. # tests with base 0
  146. # this fails on 3.0, but in 2.x the old octal syntax is allowed
  147. self.assertEqual(int(' 0123 ', 0), 83)
  148. self.assertEqual(int(' 0123 ', 0), 83)
  149. self.assertEqual(int('000', 0), 0)
  150. self.assertEqual(int('0o123', 0), 83)
  151. self.assertEqual(int('0x123', 0), 291)
  152. self.assertEqual(int('0b100', 0), 4)
  153. self.assertEqual(int(' 0O123 ', 0), 83)
  154. self.assertEqual(int(' 0X123 ', 0), 291)
  155. self.assertEqual(int(' 0B100 ', 0), 4)
  156. # without base still base 10
  157. self.assertEqual(int('0123'), 123)
  158. self.assertEqual(int('0123', 10), 123)
  159. # tests with prefix and base != 0
  160. self.assertEqual(int('0x123', 16), 291)
  161. self.assertEqual(int('0o123', 8), 83)
  162. self.assertEqual(int('0b100', 2), 4)
  163. self.assertEqual(int('0X123', 16), 291)
  164. self.assertEqual(int('0O123', 8), 83)
  165. self.assertEqual(int('0B100', 2), 4)
  166. # the code has special checks for the first character after the
  167. # type prefix
  168. self.assertRaises(ValueError, int, '0b2', 2)
  169. self.assertRaises(ValueError, int, '0b02', 2)
  170. self.assertRaises(ValueError, int, '0B2', 2)
  171. self.assertRaises(ValueError, int, '0B02', 2)
  172. self.assertRaises(ValueError, int, '0o8', 8)
  173. self.assertRaises(ValueError, int, '0o08', 8)
  174. self.assertRaises(ValueError, int, '0O8', 8)
  175. self.assertRaises(ValueError, int, '0O08', 8)
  176. self.assertRaises(ValueError, int, '0xg', 16)
  177. self.assertRaises(ValueError, int, '0x0g', 16)
  178. self.assertRaises(ValueError, int, '0Xg', 16)
  179. self.assertRaises(ValueError, int, '0X0g', 16)
  180. # SF bug 1334662: int(string, base) wrong answers
  181. # Checks for proper evaluation of 2**32 + 1
  182. self.assertEqual(int('100000000000000000000000000000001', 2), 4294967297L)
  183. self.assertEqual(int('102002022201221111212', 3), 4294967297L)
  184. self.assertEqual(int('10000000000000001', 4), 4294967297L)
  185. self.assertEqual(int('32244002423142', 5), 4294967297L)
  186. self.assertEqual(int('1550104015505', 6), 4294967297L)
  187. self.assertEqual(int('211301422355', 7), 4294967297L)
  188. self.assertEqual(int('40000000001', 8), 4294967297L)
  189. self.assertEqual(int('12068657455', 9), 4294967297L)
  190. self.assertEqual(int('4294967297', 10), 4294967297L)
  191. self.assertEqual(int('1904440555', 11), 4294967297L)
  192. self.assertEqual(int('9ba461595', 12), 4294967297L)
  193. self.assertEqual(int('535a7988a', 13), 4294967297L)
  194. self.assertEqual(int('2ca5b7465', 14), 4294967297L)
  195. self.assertEqual(int('1a20dcd82', 15), 4294967297L)
  196. self.assertEqual(int('100000001', 16), 4294967297L)
  197. self.assertEqual(int('a7ffda92', 17), 4294967297L)
  198. self.assertEqual(int('704he7g5', 18), 4294967297L)
  199. self.assertEqual(int('4f5aff67', 19), 4294967297L)
  200. self.assertEqual(int('3723ai4h', 20), 4294967297L)
  201. self.assertEqual(int('281d55i5', 21), 4294967297L)
  202. self.assertEqual(int('1fj8b185', 22), 4294967297L)
  203. self.assertEqual(int('1606k7id', 23), 4294967297L)
  204. self.assertEqual(int('mb994ah', 24), 4294967297L)
  205. self.assertEqual(int('hek2mgm', 25), 4294967297L)
  206. self.assertEqual(int('dnchbnn', 26), 4294967297L)
  207. self.assertEqual(int('b28jpdn', 27), 4294967297L)
  208. self.assertEqual(int('8pfgih5', 28), 4294967297L)
  209. self.assertEqual(int('76beigh', 29), 4294967297L)
  210. self.assertEqual(int('5qmcpqh', 30), 4294967297L)
  211. self.assertEqual(int('4q0jto5', 31), 4294967297L)
  212. self.assertEqual(int('4000001', 32), 4294967297L)
  213. self.assertEqual(int('3aokq95', 33), 4294967297L)
  214. self.assertEqual(int('2qhxjlj', 34), 4294967297L)
  215. self.assertEqual(int('2br45qc', 35), 4294967297L)
  216. self.assertEqual(int('1z141z5', 36), 4294967297L)
  217. def test_intconversion(self):
  218. # Test __int__()
  219. class ClassicMissingMethods:
  220. pass
  221. self.assertRaises(AttributeError, int, ClassicMissingMethods())
  222. class MissingMethods(object):
  223. pass
  224. self.assertRaises(TypeError, int, MissingMethods())
  225. class Foo0:
  226. def __int__(self):
  227. return 42
  228. class Foo1(object):
  229. def __int__(self):
  230. return 42
  231. class Foo2(int):
  232. def __int__(self):
  233. return 42
  234. class Foo3(int):
  235. def __int__(self):
  236. return self
  237. class Foo4(int):
  238. def __int__(self):
  239. return 42L
  240. class Foo5(int):
  241. def __int__(self):
  242. return 42.
  243. self.assertEqual(int(Foo0()), 42)
  244. self.assertEqual(int(Foo1()), 42)
  245. self.assertEqual(int(Foo2()), 42)
  246. self.assertEqual(int(Foo3()), 0)
  247. self.assertEqual(int(Foo4()), 42L)
  248. self.assertRaises(TypeError, int, Foo5())
  249. class Classic:
  250. pass
  251. for base in (object, Classic):
  252. class IntOverridesTrunc(base):
  253. def __int__(self):
  254. return 42
  255. def __trunc__(self):
  256. return -12
  257. self.assertEqual(int(IntOverridesTrunc()), 42)
  258. class JustTrunc(base):
  259. def __trunc__(self):
  260. return 42
  261. self.assertEqual(int(JustTrunc()), 42)
  262. for trunc_result_base in (object, Classic):
  263. class Integral(trunc_result_base):
  264. def __int__(self):
  265. return 42
  266. class TruncReturnsNonInt(base):
  267. def __trunc__(self):
  268. return Integral()
  269. self.assertEqual(int(TruncReturnsNonInt()), 42)
  270. class NonIntegral(trunc_result_base):
  271. def __trunc__(self):
  272. # Check that we avoid infinite recursion.
  273. return NonIntegral()
  274. class TruncReturnsNonIntegral(base):
  275. def __trunc__(self):
  276. return NonIntegral()
  277. try:
  278. int(TruncReturnsNonIntegral())
  279. except TypeError as e:
  280. self.assertEquals(str(e),
  281. "__trunc__ returned non-Integral"
  282. " (type NonIntegral)")
  283. else:
  284. self.fail("Failed to raise TypeError with %s" %
  285. ((base, trunc_result_base),))
  286. def test_main():
  287. run_unittest(IntTestCases)
  288. if __name__ == "__main__":
  289. test_main()