PageRenderTime 57ms CodeModel.GetById 19ms RepoModel.GetById 1ms app.codeStats 0ms

/Lib/test/test_builtin.py

http://unladen-swallow.googlecode.com/
Python | 1542 lines | 1422 code | 88 blank | 32 comment | 46 complexity | addc20e6276091a619d68970ac8a0cf7 MD5 | raw file
Possible License(s): 0BSD, BSD-3-Clause
  1. # Python test set -- built-in functions
  2. import test.test_support, unittest
  3. from test.test_support import fcmp, have_unicode, TESTFN, unlink, \
  4. run_unittest, run_with_locale
  5. from operator import neg
  6. import sys, warnings, cStringIO, random, fractions, UserDict
  7. warnings.filterwarnings("ignore", "hex../oct.. of negative int",
  8. FutureWarning, __name__)
  9. warnings.filterwarnings("ignore", "integer argument expected",
  10. DeprecationWarning, "unittest")
  11. # count the number of test runs.
  12. # used to skip running test_execfile() multiple times
  13. numruns = 0
  14. class Squares:
  15. def __init__(self, max):
  16. self.max = max
  17. self.sofar = []
  18. def __len__(self): return len(self.sofar)
  19. def __getitem__(self, i):
  20. if not 0 <= i < self.max: raise IndexError
  21. n = len(self.sofar)
  22. while n <= i:
  23. self.sofar.append(n*n)
  24. n += 1
  25. return self.sofar[i]
  26. class StrSquares:
  27. def __init__(self, max):
  28. self.max = max
  29. self.sofar = []
  30. def __len__(self):
  31. return len(self.sofar)
  32. def __getitem__(self, i):
  33. if not 0 <= i < self.max:
  34. raise IndexError
  35. n = len(self.sofar)
  36. while n <= i:
  37. self.sofar.append(str(n*n))
  38. n += 1
  39. return self.sofar[i]
  40. class BitBucket:
  41. def write(self, line):
  42. pass
  43. class TestFailingBool:
  44. def __nonzero__(self):
  45. raise RuntimeError
  46. class TestFailingIter:
  47. def __iter__(self):
  48. raise RuntimeError
  49. class BuiltinTest(unittest.TestCase):
  50. def test_import(self):
  51. __import__('sys')
  52. __import__('time')
  53. __import__('string')
  54. __import__(name='sys')
  55. __import__(name='time', level=0)
  56. self.assertRaises(ImportError, __import__, 'spamspam')
  57. self.assertRaises(TypeError, __import__, 1, 2, 3, 4)
  58. self.assertRaises(ValueError, __import__, '')
  59. self.assertRaises(TypeError, __import__, 'sys', name='sys')
  60. def test_abs(self):
  61. # int
  62. self.assertEqual(abs(0), 0)
  63. self.assertEqual(abs(1234), 1234)
  64. self.assertEqual(abs(-1234), 1234)
  65. self.assertTrue(abs(-sys.maxint-1) > 0)
  66. # float
  67. self.assertEqual(abs(0.0), 0.0)
  68. self.assertEqual(abs(3.14), 3.14)
  69. self.assertEqual(abs(-3.14), 3.14)
  70. # long
  71. self.assertEqual(abs(0L), 0L)
  72. self.assertEqual(abs(1234L), 1234L)
  73. self.assertEqual(abs(-1234L), 1234L)
  74. # str
  75. self.assertRaises(TypeError, abs, 'a')
  76. def test_all(self):
  77. self.assertEqual(all([2, 4, 6]), True)
  78. self.assertEqual(all([2, None, 6]), False)
  79. self.assertRaises(RuntimeError, all, [2, TestFailingBool(), 6])
  80. self.assertRaises(RuntimeError, all, TestFailingIter())
  81. self.assertRaises(TypeError, all, 10) # Non-iterable
  82. self.assertRaises(TypeError, all) # No args
  83. self.assertRaises(TypeError, all, [2, 4, 6], []) # Too many args
  84. self.assertEqual(all([]), True) # Empty iterator
  85. S = [50, 60]
  86. self.assertEqual(all(x > 42 for x in S), True)
  87. S = [50, 40, 60]
  88. self.assertEqual(all(x > 42 for x in S), False)
  89. def test_any(self):
  90. self.assertEqual(any([None, None, None]), False)
  91. self.assertEqual(any([None, 4, None]), True)
  92. self.assertRaises(RuntimeError, any, [None, TestFailingBool(), 6])
  93. self.assertRaises(RuntimeError, all, TestFailingIter())
  94. self.assertRaises(TypeError, any, 10) # Non-iterable
  95. self.assertRaises(TypeError, any) # No args
  96. self.assertRaises(TypeError, any, [2, 4, 6], []) # Too many args
  97. self.assertEqual(any([]), False) # Empty iterator
  98. S = [40, 60, 30]
  99. self.assertEqual(any(x > 42 for x in S), True)
  100. S = [10, 20, 30]
  101. self.assertEqual(any(x > 42 for x in S), False)
  102. def test_neg(self):
  103. x = -sys.maxint-1
  104. self.assert_(isinstance(x, int))
  105. self.assertEqual(-x, sys.maxint+1)
  106. def test_apply(self):
  107. def f0(*args):
  108. self.assertEqual(args, ())
  109. def f1(a1):
  110. self.assertEqual(a1, 1)
  111. def f2(a1, a2):
  112. self.assertEqual(a1, 1)
  113. self.assertEqual(a2, 2)
  114. def f3(a1, a2, a3):
  115. self.assertEqual(a1, 1)
  116. self.assertEqual(a2, 2)
  117. self.assertEqual(a3, 3)
  118. apply(f0, ())
  119. apply(f1, (1,))
  120. apply(f2, (1, 2))
  121. apply(f3, (1, 2, 3))
  122. # A PyCFunction that takes only positional parameters should allow an
  123. # empty keyword dictionary to pass without a complaint, but raise a
  124. # TypeError if the dictionary is non-empty.
  125. apply(id, (1,), {})
  126. self.assertRaises(TypeError, apply, id, (1,), {"foo": 1})
  127. self.assertRaises(TypeError, apply)
  128. self.assertRaises(TypeError, apply, id, 42)
  129. self.assertRaises(TypeError, apply, id, (42,), 42)
  130. def test_callable(self):
  131. self.assert_(callable(len))
  132. def f(): pass
  133. self.assert_(callable(f))
  134. class C:
  135. def meth(self): pass
  136. self.assert_(callable(C))
  137. x = C()
  138. self.assert_(callable(x.meth))
  139. self.assert_(not callable(x))
  140. class D(C):
  141. def __call__(self): pass
  142. y = D()
  143. self.assert_(callable(y))
  144. y()
  145. def test_chr(self):
  146. self.assertEqual(chr(32), ' ')
  147. self.assertEqual(chr(65), 'A')
  148. self.assertEqual(chr(97), 'a')
  149. self.assertEqual(chr(0xff), '\xff')
  150. self.assertRaises(ValueError, chr, 256)
  151. self.assertRaises(TypeError, chr)
  152. def test_cmp(self):
  153. self.assertEqual(cmp(-1, 1), -1)
  154. self.assertEqual(cmp(1, -1), 1)
  155. self.assertEqual(cmp(1, 1), 0)
  156. # verify that circular objects are not handled
  157. a = []; a.append(a)
  158. b = []; b.append(b)
  159. from UserList import UserList
  160. c = UserList(); c.append(c)
  161. self.assertRaises(RuntimeError, cmp, a, b)
  162. self.assertRaises(RuntimeError, cmp, b, c)
  163. self.assertRaises(RuntimeError, cmp, c, a)
  164. self.assertRaises(RuntimeError, cmp, a, c)
  165. # okay, now break the cycles
  166. a.pop(); b.pop(); c.pop()
  167. self.assertRaises(TypeError, cmp)
  168. def test_coerce(self):
  169. self.assert_(not fcmp(coerce(1, 1.1), (1.0, 1.1)))
  170. self.assertEqual(coerce(1, 1L), (1L, 1L))
  171. self.assert_(not fcmp(coerce(1L, 1.1), (1.0, 1.1)))
  172. self.assertRaises(TypeError, coerce)
  173. class BadNumber:
  174. def __coerce__(self, other):
  175. raise ValueError
  176. self.assertRaises(ValueError, coerce, 42, BadNumber())
  177. self.assertRaises(OverflowError, coerce, 0.5, int("12345" * 1000))
  178. def test_compile(self):
  179. compile('print 1\n', '', 'exec')
  180. bom = '\xef\xbb\xbf'
  181. compile(bom + 'print 1\n', '', 'exec')
  182. compile(source='pass', filename='?', mode='exec')
  183. compile(dont_inherit=0, filename='tmp', source='0', mode='eval')
  184. compile('pass', '?', dont_inherit=1, mode='exec')
  185. self.assertRaises(TypeError, compile)
  186. self.assertRaises(ValueError, compile, 'print 42\n', '<string>', 'badmode')
  187. self.assertRaises(ValueError, compile, 'print 42\n', '<string>', 'single', 0xff)
  188. self.assertRaises(TypeError, compile, chr(0), 'f', 'exec')
  189. self.assertRaises(TypeError, compile, 'pass', '?', 'exec',
  190. mode='eval', source='0', filename='tmp')
  191. if have_unicode:
  192. compile(unicode('print u"\xc3\xa5"\n', 'utf8'), '', 'exec')
  193. self.assertRaises(TypeError, compile, unichr(0), 'f', 'exec')
  194. self.assertRaises(ValueError, compile, unicode('a = 1'), 'f', 'bad')
  195. def test_delattr(self):
  196. import sys
  197. sys.spam = 1
  198. delattr(sys, 'spam')
  199. self.assertRaises(TypeError, delattr)
  200. def test_dir(self):
  201. # dir(wrong number of arguments)
  202. self.assertRaises(TypeError, dir, 42, 42)
  203. # dir() - local scope
  204. local_var = 1
  205. self.assert_('local_var' in dir())
  206. # dir(module)
  207. import sys
  208. self.assert_('exit' in dir(sys))
  209. # dir(module_with_invalid__dict__)
  210. import types
  211. class Foo(types.ModuleType):
  212. __dict__ = 8
  213. f = Foo("foo")
  214. self.assertRaises(TypeError, dir, f)
  215. # dir(type)
  216. self.assert_("strip" in dir(str))
  217. self.assert_("__mro__" not in dir(str))
  218. # dir(obj)
  219. class Foo(object):
  220. def __init__(self):
  221. self.x = 7
  222. self.y = 8
  223. self.z = 9
  224. f = Foo()
  225. self.assert_("y" in dir(f))
  226. # dir(obj_no__dict__)
  227. class Foo(object):
  228. __slots__ = []
  229. f = Foo()
  230. self.assert_("__repr__" in dir(f))
  231. # dir(obj_no__class__with__dict__)
  232. # (an ugly trick to cause getattr(f, "__class__") to fail)
  233. class Foo(object):
  234. __slots__ = ["__class__", "__dict__"]
  235. def __init__(self):
  236. self.bar = "wow"
  237. f = Foo()
  238. self.assert_("__repr__" not in dir(f))
  239. self.assert_("bar" in dir(f))
  240. # dir(obj_using __dir__)
  241. class Foo(object):
  242. def __dir__(self):
  243. return ["kan", "ga", "roo"]
  244. f = Foo()
  245. self.assert_(dir(f) == ["ga", "kan", "roo"])
  246. # dir(obj__dir__not_list)
  247. class Foo(object):
  248. def __dir__(self):
  249. return 7
  250. f = Foo()
  251. self.assertRaises(TypeError, dir, f)
  252. def test_divmod(self):
  253. self.assertEqual(divmod(12, 7), (1, 5))
  254. self.assertEqual(divmod(-12, 7), (-2, 2))
  255. self.assertEqual(divmod(12, -7), (-2, -2))
  256. self.assertEqual(divmod(-12, -7), (1, -5))
  257. self.assertEqual(divmod(12L, 7L), (1L, 5L))
  258. self.assertEqual(divmod(-12L, 7L), (-2L, 2L))
  259. self.assertEqual(divmod(12L, -7L), (-2L, -2L))
  260. self.assertEqual(divmod(-12L, -7L), (1L, -5L))
  261. self.assertEqual(divmod(12, 7L), (1, 5L))
  262. self.assertEqual(divmod(-12, 7L), (-2, 2L))
  263. self.assertEqual(divmod(12L, -7), (-2L, -2))
  264. self.assertEqual(divmod(-12L, -7), (1L, -5))
  265. self.assertEqual(divmod(-sys.maxint-1, -1),
  266. (sys.maxint+1, 0))
  267. self.assert_(not fcmp(divmod(3.25, 1.0), (3.0, 0.25)))
  268. self.assert_(not fcmp(divmod(-3.25, 1.0), (-4.0, 0.75)))
  269. self.assert_(not fcmp(divmod(3.25, -1.0), (-4.0, -0.75)))
  270. self.assert_(not fcmp(divmod(-3.25, -1.0), (3.0, -0.25)))
  271. self.assertRaises(TypeError, divmod)
  272. def test_eval(self):
  273. self.assertEqual(eval('1+1'), 2)
  274. self.assertEqual(eval(' 1+1\n'), 2)
  275. globals = {'a': 1, 'b': 2}
  276. locals = {'b': 200, 'c': 300}
  277. self.assertEqual(eval('a', globals) , 1)
  278. self.assertEqual(eval('a', globals, locals), 1)
  279. self.assertEqual(eval('b', globals, locals), 200)
  280. self.assertEqual(eval('c', globals, locals), 300)
  281. if have_unicode:
  282. self.assertEqual(eval(unicode('1+1')), 2)
  283. self.assertEqual(eval(unicode(' 1+1\n')), 2)
  284. globals = {'a': 1, 'b': 2}
  285. locals = {'b': 200, 'c': 300}
  286. if have_unicode:
  287. self.assertEqual(eval(unicode('a'), globals), 1)
  288. self.assertEqual(eval(unicode('a'), globals, locals), 1)
  289. self.assertEqual(eval(unicode('b'), globals, locals), 200)
  290. self.assertEqual(eval(unicode('c'), globals, locals), 300)
  291. bom = '\xef\xbb\xbf'
  292. self.assertEqual(eval(bom + 'a', globals, locals), 1)
  293. self.assertEqual(eval(unicode('u"\xc3\xa5"', 'utf8'), globals),
  294. unicode('\xc3\xa5', 'utf8'))
  295. self.assertRaises(TypeError, eval)
  296. self.assertRaises(TypeError, eval, ())
  297. def test_general_eval(self):
  298. # Tests that general mappings can be used for the locals argument
  299. class M:
  300. "Test mapping interface versus possible calls from eval()."
  301. def __getitem__(self, key):
  302. if key == 'a':
  303. return 12
  304. raise KeyError
  305. def keys(self):
  306. return list('xyz')
  307. m = M()
  308. g = globals()
  309. self.assertEqual(eval('a', g, m), 12)
  310. self.assertRaises(NameError, eval, 'b', g, m)
  311. self.assertEqual(eval('dir()', g, m), list('xyz'))
  312. self.assertEqual(eval('globals()', g, m), g)
  313. self.assertEqual(eval('locals()', g, m), m)
  314. self.assertRaises(TypeError, eval, 'a', m)
  315. class A:
  316. "Non-mapping"
  317. pass
  318. m = A()
  319. self.assertRaises(TypeError, eval, 'a', g, m)
  320. # Verify that dict subclasses work as well
  321. class D(dict):
  322. def __getitem__(self, key):
  323. if key == 'a':
  324. return 12
  325. return dict.__getitem__(self, key)
  326. def keys(self):
  327. return list('xyz')
  328. d = D()
  329. self.assertEqual(eval('a', g, d), 12)
  330. self.assertRaises(NameError, eval, 'b', g, d)
  331. self.assertEqual(eval('dir()', g, d), list('xyz'))
  332. self.assertEqual(eval('globals()', g, d), g)
  333. self.assertEqual(eval('locals()', g, d), d)
  334. # Verify locals stores (used by list comps)
  335. eval('[locals() for i in (2,3)]', g, d)
  336. eval('[locals() for i in (2,3)]', g, UserDict.UserDict())
  337. class SpreadSheet:
  338. "Sample application showing nested, calculated lookups."
  339. _cells = {}
  340. def __setitem__(self, key, formula):
  341. self._cells[key] = formula
  342. def __getitem__(self, key):
  343. return eval(self._cells[key], globals(), self)
  344. ss = SpreadSheet()
  345. ss['a1'] = '5'
  346. ss['a2'] = 'a1*6'
  347. ss['a3'] = 'a2*7'
  348. self.assertEqual(ss['a3'], 210)
  349. # Verify that dir() catches a non-list returned by eval
  350. # SF bug #1004669
  351. class C:
  352. def __getitem__(self, item):
  353. raise KeyError(item)
  354. def keys(self):
  355. return 'a'
  356. self.assertRaises(TypeError, eval, 'dir()', globals(), C())
  357. # Done outside of the method test_z to get the correct scope
  358. z = 0
  359. f = open(TESTFN, 'w')
  360. f.write('z = z+1\n')
  361. f.write('z = z*2\n')
  362. f.close()
  363. execfile(TESTFN)
  364. def test_execfile(self):
  365. global numruns
  366. if numruns:
  367. return
  368. numruns += 1
  369. globals = {'a': 1, 'b': 2}
  370. locals = {'b': 200, 'c': 300}
  371. self.assertEqual(self.__class__.z, 2)
  372. globals['z'] = 0
  373. execfile(TESTFN, globals)
  374. self.assertEqual(globals['z'], 2)
  375. locals['z'] = 0
  376. execfile(TESTFN, globals, locals)
  377. self.assertEqual(locals['z'], 2)
  378. class M:
  379. "Test mapping interface versus possible calls from execfile()."
  380. def __init__(self):
  381. self.z = 10
  382. def __getitem__(self, key):
  383. if key == 'z':
  384. return self.z
  385. raise KeyError
  386. def __setitem__(self, key, value):
  387. if key == 'z':
  388. self.z = value
  389. return
  390. raise KeyError
  391. locals = M()
  392. locals['z'] = 0
  393. execfile(TESTFN, globals, locals)
  394. self.assertEqual(locals['z'], 2)
  395. unlink(TESTFN)
  396. self.assertRaises(TypeError, execfile)
  397. self.assertRaises(TypeError, execfile, TESTFN, {}, ())
  398. import os
  399. self.assertRaises(IOError, execfile, os.curdir)
  400. self.assertRaises(IOError, execfile, "I_dont_exist")
  401. def test_filter(self):
  402. self.assertEqual(filter(lambda c: 'a' <= c <= 'z', 'Hello World'), 'elloorld')
  403. self.assertEqual(filter(None, [1, 'hello', [], [3], '', None, 9, 0]), [1, 'hello', [3], 9])
  404. self.assertEqual(filter(lambda x: x > 0, [1, -3, 9, 0, 2]), [1, 9, 2])
  405. self.assertEqual(filter(None, Squares(10)), [1, 4, 9, 16, 25, 36, 49, 64, 81])
  406. self.assertEqual(filter(lambda x: x%2, Squares(10)), [1, 9, 25, 49, 81])
  407. def identity(item):
  408. return 1
  409. filter(identity, Squares(5))
  410. self.assertRaises(TypeError, filter)
  411. class BadSeq(object):
  412. def __getitem__(self, index):
  413. if index<4:
  414. return 42
  415. raise ValueError
  416. self.assertRaises(ValueError, filter, lambda x: x, BadSeq())
  417. def badfunc():
  418. pass
  419. self.assertRaises(TypeError, filter, badfunc, range(5))
  420. # test bltinmodule.c::filtertuple()
  421. self.assertEqual(filter(None, (1, 2)), (1, 2))
  422. self.assertEqual(filter(lambda x: x>=3, (1, 2, 3, 4)), (3, 4))
  423. self.assertRaises(TypeError, filter, 42, (1, 2))
  424. # test bltinmodule.c::filterstring()
  425. self.assertEqual(filter(None, "12"), "12")
  426. self.assertEqual(filter(lambda x: x>="3", "1234"), "34")
  427. self.assertRaises(TypeError, filter, 42, "12")
  428. class badstr(str):
  429. def __getitem__(self, index):
  430. raise ValueError
  431. self.assertRaises(ValueError, filter, lambda x: x >="3", badstr("1234"))
  432. class badstr2(str):
  433. def __getitem__(self, index):
  434. return 42
  435. self.assertRaises(TypeError, filter, lambda x: x >=42, badstr2("1234"))
  436. class weirdstr(str):
  437. def __getitem__(self, index):
  438. return weirdstr(2*str.__getitem__(self, index))
  439. self.assertEqual(filter(lambda x: x>="33", weirdstr("1234")), "3344")
  440. class shiftstr(str):
  441. def __getitem__(self, index):
  442. return chr(ord(str.__getitem__(self, index))+1)
  443. self.assertEqual(filter(lambda x: x>="3", shiftstr("1234")), "345")
  444. if have_unicode:
  445. # test bltinmodule.c::filterunicode()
  446. self.assertEqual(filter(None, unicode("12")), unicode("12"))
  447. self.assertEqual(filter(lambda x: x>="3", unicode("1234")), unicode("34"))
  448. self.assertRaises(TypeError, filter, 42, unicode("12"))
  449. self.assertRaises(ValueError, filter, lambda x: x >="3", badstr(unicode("1234")))
  450. class badunicode(unicode):
  451. def __getitem__(self, index):
  452. return 42
  453. self.assertRaises(TypeError, filter, lambda x: x >=42, badunicode("1234"))
  454. class weirdunicode(unicode):
  455. def __getitem__(self, index):
  456. return weirdunicode(2*unicode.__getitem__(self, index))
  457. self.assertEqual(
  458. filter(lambda x: x>=unicode("33"), weirdunicode("1234")), unicode("3344"))
  459. class shiftunicode(unicode):
  460. def __getitem__(self, index):
  461. return unichr(ord(unicode.__getitem__(self, index))+1)
  462. self.assertEqual(
  463. filter(lambda x: x>=unicode("3"), shiftunicode("1234")),
  464. unicode("345")
  465. )
  466. def test_filter_subclasses(self):
  467. # test that filter() never returns tuple, str or unicode subclasses
  468. # and that the result always goes through __getitem__
  469. funcs = (None, bool, lambda x: True)
  470. class tuple2(tuple):
  471. def __getitem__(self, index):
  472. return 2*tuple.__getitem__(self, index)
  473. class str2(str):
  474. def __getitem__(self, index):
  475. return 2*str.__getitem__(self, index)
  476. inputs = {
  477. tuple2: {(): (), (1, 2, 3): (2, 4, 6)},
  478. str2: {"": "", "123": "112233"}
  479. }
  480. if have_unicode:
  481. class unicode2(unicode):
  482. def __getitem__(self, index):
  483. return 2*unicode.__getitem__(self, index)
  484. inputs[unicode2] = {
  485. unicode(): unicode(),
  486. unicode("123"): unicode("112233")
  487. }
  488. for (cls, inps) in inputs.iteritems():
  489. for (inp, exp) in inps.iteritems():
  490. # make sure the output goes through __getitem__
  491. # even if func is None
  492. self.assertEqual(
  493. filter(funcs[0], cls(inp)),
  494. filter(funcs[1], cls(inp))
  495. )
  496. for func in funcs:
  497. outp = filter(func, cls(inp))
  498. self.assertEqual(outp, exp)
  499. self.assert_(not isinstance(outp, cls))
  500. def test_getattr(self):
  501. import sys
  502. self.assert_(getattr(sys, 'stdout') is sys.stdout)
  503. self.assertRaises(TypeError, getattr, sys, 1)
  504. self.assertRaises(TypeError, getattr, sys, 1, "foo")
  505. self.assertRaises(TypeError, getattr)
  506. if have_unicode:
  507. self.assertRaises(UnicodeError, getattr, sys, unichr(sys.maxunicode))
  508. def test_hasattr(self):
  509. import sys
  510. self.assert_(hasattr(sys, 'stdout'))
  511. self.assertRaises(TypeError, hasattr, sys, 1)
  512. self.assertRaises(TypeError, hasattr)
  513. if have_unicode:
  514. self.assertRaises(UnicodeError, hasattr, sys, unichr(sys.maxunicode))
  515. # Check that hasattr allows SystemExit and KeyboardInterrupts by
  516. class A:
  517. def __getattr__(self, what):
  518. raise KeyboardInterrupt
  519. self.assertRaises(KeyboardInterrupt, hasattr, A(), "b")
  520. class B:
  521. def __getattr__(self, what):
  522. raise SystemExit
  523. self.assertRaises(SystemExit, hasattr, B(), "b")
  524. def test_hash(self):
  525. hash(None)
  526. self.assertEqual(hash(1), hash(1L))
  527. self.assertEqual(hash(1), hash(1.0))
  528. hash('spam')
  529. if have_unicode:
  530. self.assertEqual(hash('spam'), hash(unicode('spam')))
  531. hash((0,1,2,3))
  532. def f(): pass
  533. self.assertRaises(TypeError, hash, [])
  534. self.assertRaises(TypeError, hash, {})
  535. # Bug 1536021: Allow hash to return long objects
  536. class X:
  537. def __hash__(self):
  538. return 2**100
  539. self.assertEquals(type(hash(X())), int)
  540. class Y(object):
  541. def __hash__(self):
  542. return 2**100
  543. self.assertEquals(type(hash(Y())), int)
  544. class Z(long):
  545. def __hash__(self):
  546. return self
  547. self.assertEquals(hash(Z(42)), hash(42L))
  548. def test_hex(self):
  549. self.assertEqual(hex(16), '0x10')
  550. self.assertEqual(hex(16L), '0x10L')
  551. self.assertEqual(hex(-16), '-0x10')
  552. self.assertEqual(hex(-16L), '-0x10L')
  553. self.assertRaises(TypeError, hex, {})
  554. def test_id(self):
  555. id(None)
  556. id(1)
  557. id(1L)
  558. id(1.0)
  559. id('spam')
  560. id((0,1,2,3))
  561. id([0,1,2,3])
  562. id({'spam': 1, 'eggs': 2, 'ham': 3})
  563. # Test input() later, together with raw_input
  564. def test_intern(self):
  565. self.assertRaises(TypeError, intern)
  566. s = "never interned before"
  567. self.assert_(intern(s) is s)
  568. s2 = s.swapcase().swapcase()
  569. self.assert_(intern(s2) is s)
  570. # Verify the error message we get when interning a non-string.
  571. try:
  572. intern(object())
  573. except TypeError, e:
  574. self.assertEqual(str(e),
  575. "intern() argument 1 must be string, not object")
  576. else:
  577. self.fail("Failed to raise exception")
  578. # Subclasses of string can't be interned, because they
  579. # provide too much opportunity for insane things to happen.
  580. # We don't want them in the interned dict and if they aren't
  581. # actually interned, we don't want to create the appearance
  582. # that they are by allowing intern() to succeeed.
  583. class S(str):
  584. def __hash__(self):
  585. return 123
  586. self.assertRaises(TypeError, intern, S("abc"))
  587. # It's still safe to pass these strings to routines that
  588. # call intern internally, e.g. PyObject_SetAttr().
  589. s = S("abc")
  590. setattr(s, s, s)
  591. self.assertEqual(getattr(s, s), s)
  592. def test_iter(self):
  593. self.assertRaises(TypeError, iter)
  594. self.assertRaises(TypeError, iter, 42, 42)
  595. lists = [("1", "2"), ["1", "2"], "12"]
  596. if have_unicode:
  597. lists.append(unicode("12"))
  598. for l in lists:
  599. i = iter(l)
  600. self.assertEqual(i.next(), '1')
  601. self.assertEqual(i.next(), '2')
  602. self.assertRaises(StopIteration, i.next)
  603. def test_isinstance(self):
  604. class C:
  605. pass
  606. class D(C):
  607. pass
  608. class E:
  609. pass
  610. c = C()
  611. d = D()
  612. e = E()
  613. self.assert_(isinstance(c, C))
  614. self.assert_(isinstance(d, C))
  615. self.assert_(not isinstance(e, C))
  616. self.assert_(not isinstance(c, D))
  617. self.assert_(not isinstance('foo', E))
  618. self.assertRaises(TypeError, isinstance, E, 'foo')
  619. self.assertRaises(TypeError, isinstance)
  620. def test_issubclass(self):
  621. class C:
  622. pass
  623. class D(C):
  624. pass
  625. class E:
  626. pass
  627. c = C()
  628. d = D()
  629. e = E()
  630. self.assert_(issubclass(D, C))
  631. self.assert_(issubclass(C, C))
  632. self.assert_(not issubclass(C, D))
  633. self.assertRaises(TypeError, issubclass, 'foo', E)
  634. self.assertRaises(TypeError, issubclass, E, 'foo')
  635. self.assertRaises(TypeError, issubclass)
  636. def test_len(self):
  637. self.assertEqual(len('123'), 3)
  638. self.assertEqual(len(()), 0)
  639. self.assertEqual(len((1, 2, 3, 4)), 4)
  640. self.assertEqual(len([1, 2, 3, 4]), 4)
  641. self.assertEqual(len({}), 0)
  642. self.assertEqual(len({'a':1, 'b': 2}), 2)
  643. class BadSeq:
  644. def __len__(self):
  645. raise ValueError
  646. self.assertRaises(ValueError, len, BadSeq())
  647. def test_map(self):
  648. self.assertEqual(
  649. map(None, 'hello world'),
  650. ['h','e','l','l','o',' ','w','o','r','l','d']
  651. )
  652. self.assertEqual(
  653. map(None, 'abcd', 'efg'),
  654. [('a', 'e'), ('b', 'f'), ('c', 'g'), ('d', None)]
  655. )
  656. self.assertEqual(
  657. map(None, range(10)),
  658. [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
  659. )
  660. self.assertEqual(
  661. map(lambda x: x*x, range(1,4)),
  662. [1, 4, 9]
  663. )
  664. try:
  665. from math import sqrt
  666. except ImportError:
  667. def sqrt(x):
  668. return pow(x, 0.5)
  669. self.assertEqual(
  670. map(lambda x: map(sqrt,x), [[16, 4], [81, 9]]),
  671. [[4.0, 2.0], [9.0, 3.0]]
  672. )
  673. self.assertEqual(
  674. map(lambda x, y: x+y, [1,3,2], [9,1,4]),
  675. [10, 4, 6]
  676. )
  677. def plus(*v):
  678. accu = 0
  679. for i in v: accu = accu + i
  680. return accu
  681. self.assertEqual(
  682. map(plus, [1, 3, 7]),
  683. [1, 3, 7]
  684. )
  685. self.assertEqual(
  686. map(plus, [1, 3, 7], [4, 9, 2]),
  687. [1+4, 3+9, 7+2]
  688. )
  689. self.assertEqual(
  690. map(plus, [1, 3, 7], [4, 9, 2], [1, 1, 0]),
  691. [1+4+1, 3+9+1, 7+2+0]
  692. )
  693. self.assertEqual(
  694. map(None, Squares(10)),
  695. [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
  696. )
  697. self.assertEqual(
  698. map(int, Squares(10)),
  699. [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
  700. )
  701. self.assertEqual(
  702. map(None, Squares(3), Squares(2)),
  703. [(0,0), (1,1), (4,None)]
  704. )
  705. self.assertEqual(
  706. map(max, Squares(3), Squares(2)),
  707. [0, 1, 4]
  708. )
  709. self.assertRaises(TypeError, map)
  710. self.assertRaises(TypeError, map, lambda x: x, 42)
  711. self.assertEqual(map(None, [42]), [42])
  712. class BadSeq:
  713. def __getitem__(self, index):
  714. raise ValueError
  715. self.assertRaises(ValueError, map, lambda x: x, BadSeq())
  716. def badfunc(x):
  717. raise RuntimeError
  718. self.assertRaises(RuntimeError, map, badfunc, range(5))
  719. def test_max(self):
  720. self.assertEqual(max('123123'), '3')
  721. self.assertEqual(max(1, 2, 3), 3)
  722. self.assertEqual(max((1, 2, 3, 1, 2, 3)), 3)
  723. self.assertEqual(max([1, 2, 3, 1, 2, 3]), 3)
  724. self.assertEqual(max(1, 2L, 3.0), 3.0)
  725. self.assertEqual(max(1L, 2.0, 3), 3)
  726. self.assertEqual(max(1.0, 2, 3L), 3L)
  727. for stmt in (
  728. "max(key=int)", # no args
  729. "max(1, key=int)", # single arg not iterable
  730. "max(1, 2, keystone=int)", # wrong keyword
  731. "max(1, 2, key=int, abc=int)", # two many keywords
  732. "max(1, 2, key=1)", # keyfunc is not callable
  733. ):
  734. try:
  735. exec(stmt) in globals()
  736. except TypeError:
  737. pass
  738. else:
  739. self.fail(stmt)
  740. self.assertEqual(max((1,), key=neg), 1) # one elem iterable
  741. self.assertEqual(max((1,2), key=neg), 1) # two elem iterable
  742. self.assertEqual(max(1, 2, key=neg), 1) # two elems
  743. data = [random.randrange(200) for i in range(100)]
  744. keys = dict((elem, random.randrange(50)) for elem in data)
  745. f = keys.__getitem__
  746. self.assertEqual(max(data, key=f),
  747. sorted(reversed(data), key=f)[-1])
  748. def test_min(self):
  749. self.assertEqual(min('123123'), '1')
  750. self.assertEqual(min(1, 2, 3), 1)
  751. self.assertEqual(min((1, 2, 3, 1, 2, 3)), 1)
  752. self.assertEqual(min([1, 2, 3, 1, 2, 3]), 1)
  753. self.assertEqual(min(1, 2L, 3.0), 1)
  754. self.assertEqual(min(1L, 2.0, 3), 1L)
  755. self.assertEqual(min(1.0, 2, 3L), 1.0)
  756. self.assertRaises(TypeError, min)
  757. self.assertRaises(TypeError, min, 42)
  758. self.assertRaises(ValueError, min, ())
  759. class BadSeq:
  760. def __getitem__(self, index):
  761. raise ValueError
  762. self.assertRaises(ValueError, min, BadSeq())
  763. class BadNumber:
  764. def __cmp__(self, other):
  765. raise ValueError
  766. self.assertRaises(ValueError, min, (42, BadNumber()))
  767. for stmt in (
  768. "min(key=int)", # no args
  769. "min(1, key=int)", # single arg not iterable
  770. "min(1, 2, keystone=int)", # wrong keyword
  771. "min(1, 2, key=int, abc=int)", # two many keywords
  772. "min(1, 2, key=1)", # keyfunc is not callable
  773. ):
  774. try:
  775. exec(stmt) in globals()
  776. except TypeError:
  777. pass
  778. else:
  779. self.fail(stmt)
  780. self.assertEqual(min((1,), key=neg), 1) # one elem iterable
  781. self.assertEqual(min((1,2), key=neg), 2) # two elem iterable
  782. self.assertEqual(min(1, 2, key=neg), 2) # two elems
  783. data = [random.randrange(200) for i in range(100)]
  784. keys = dict((elem, random.randrange(50)) for elem in data)
  785. f = keys.__getitem__
  786. self.assertEqual(min(data, key=f),
  787. sorted(data, key=f)[0])
  788. def test_next(self):
  789. it = iter(range(2))
  790. self.assertEqual(next(it), 0)
  791. self.assertEqual(next(it), 1)
  792. self.assertRaises(StopIteration, next, it)
  793. self.assertRaises(StopIteration, next, it)
  794. self.assertEquals(next(it, 42), 42)
  795. class Iter(object):
  796. def __iter__(self):
  797. return self
  798. def next(self):
  799. raise StopIteration
  800. it = iter(Iter())
  801. self.assertEquals(next(it, 42), 42)
  802. self.assertRaises(StopIteration, next, it)
  803. def gen():
  804. yield 1
  805. return
  806. it = gen()
  807. self.assertEquals(next(it), 1)
  808. self.assertRaises(StopIteration, next, it)
  809. self.assertEquals(next(it, 42), 42)
  810. def test_oct(self):
  811. self.assertEqual(oct(100), '0144')
  812. self.assertEqual(oct(100L), '0144L')
  813. self.assertEqual(oct(-100), '-0144')
  814. self.assertEqual(oct(-100L), '-0144L')
  815. self.assertRaises(TypeError, oct, ())
  816. def write_testfile(self):
  817. # NB the first 4 lines are also used to test input and raw_input, below
  818. fp = open(TESTFN, 'w')
  819. try:
  820. fp.write('1+1\n')
  821. fp.write('1+1\n')
  822. fp.write('The quick brown fox jumps over the lazy dog')
  823. fp.write('.\n')
  824. fp.write('Dear John\n')
  825. fp.write('XXX'*100)
  826. fp.write('YYY'*100)
  827. finally:
  828. fp.close()
  829. def test_open(self):
  830. self.write_testfile()
  831. fp = open(TESTFN, 'r')
  832. try:
  833. self.assertEqual(fp.readline(4), '1+1\n')
  834. self.assertEqual(fp.readline(4), '1+1\n')
  835. self.assertEqual(fp.readline(), 'The quick brown fox jumps over the lazy dog.\n')
  836. self.assertEqual(fp.readline(4), 'Dear')
  837. self.assertEqual(fp.readline(100), ' John\n')
  838. self.assertEqual(fp.read(300), 'XXX'*100)
  839. self.assertEqual(fp.read(1000), 'YYY'*100)
  840. finally:
  841. fp.close()
  842. unlink(TESTFN)
  843. def test_ord(self):
  844. self.assertEqual(ord(' '), 32)
  845. self.assertEqual(ord('A'), 65)
  846. self.assertEqual(ord('a'), 97)
  847. if have_unicode:
  848. self.assertEqual(ord(unichr(sys.maxunicode)), sys.maxunicode)
  849. self.assertRaises(TypeError, ord, 42)
  850. if have_unicode:
  851. self.assertRaises(TypeError, ord, unicode("12"))
  852. def test_pow(self):
  853. self.assertEqual(pow(0,0), 1)
  854. self.assertEqual(pow(0,1), 0)
  855. self.assertEqual(pow(1,0), 1)
  856. self.assertEqual(pow(1,1), 1)
  857. self.assertEqual(pow(2,0), 1)
  858. self.assertEqual(pow(2,10), 1024)
  859. self.assertEqual(pow(2,20), 1024*1024)
  860. self.assertEqual(pow(2,30), 1024*1024*1024)
  861. self.assertEqual(pow(-2,0), 1)
  862. self.assertEqual(pow(-2,1), -2)
  863. self.assertEqual(pow(-2,2), 4)
  864. self.assertEqual(pow(-2,3), -8)
  865. self.assertEqual(pow(0L,0), 1)
  866. self.assertEqual(pow(0L,1), 0)
  867. self.assertEqual(pow(1L,0), 1)
  868. self.assertEqual(pow(1L,1), 1)
  869. self.assertEqual(pow(2L,0), 1)
  870. self.assertEqual(pow(2L,10), 1024)
  871. self.assertEqual(pow(2L,20), 1024*1024)
  872. self.assertEqual(pow(2L,30), 1024*1024*1024)
  873. self.assertEqual(pow(-2L,0), 1)
  874. self.assertEqual(pow(-2L,1), -2)
  875. self.assertEqual(pow(-2L,2), 4)
  876. self.assertEqual(pow(-2L,3), -8)
  877. self.assertAlmostEqual(pow(0.,0), 1.)
  878. self.assertAlmostEqual(pow(0.,1), 0.)
  879. self.assertAlmostEqual(pow(1.,0), 1.)
  880. self.assertAlmostEqual(pow(1.,1), 1.)
  881. self.assertAlmostEqual(pow(2.,0), 1.)
  882. self.assertAlmostEqual(pow(2.,10), 1024.)
  883. self.assertAlmostEqual(pow(2.,20), 1024.*1024.)
  884. self.assertAlmostEqual(pow(2.,30), 1024.*1024.*1024.)
  885. self.assertAlmostEqual(pow(-2.,0), 1.)
  886. self.assertAlmostEqual(pow(-2.,1), -2.)
  887. self.assertAlmostEqual(pow(-2.,2), 4.)
  888. self.assertAlmostEqual(pow(-2.,3), -8.)
  889. for x in 2, 2L, 2.0:
  890. for y in 10, 10L, 10.0:
  891. for z in 1000, 1000L, 1000.0:
  892. if isinstance(x, float) or \
  893. isinstance(y, float) or \
  894. isinstance(z, float):
  895. self.assertRaises(TypeError, pow, x, y, z)
  896. else:
  897. self.assertAlmostEqual(pow(x, y, z), 24.0)
  898. self.assertRaises(TypeError, pow, -1, -2, 3)
  899. self.assertRaises(ValueError, pow, 1, 2, 0)
  900. self.assertRaises(TypeError, pow, -1L, -2L, 3L)
  901. self.assertRaises(ValueError, pow, 1L, 2L, 0L)
  902. # Will return complex in 3.0:
  903. self.assertRaises(ValueError, pow, -342.43, 0.234)
  904. self.assertRaises(TypeError, pow)
  905. def test_range(self):
  906. self.assertEqual(range(3), [0, 1, 2])
  907. self.assertEqual(range(1, 5), [1, 2, 3, 4])
  908. self.assertEqual(range(0), [])
  909. self.assertEqual(range(-3), [])
  910. self.assertEqual(range(1, 10, 3), [1, 4, 7])
  911. self.assertEqual(range(5, -5, -3), [5, 2, -1, -4])
  912. # Now test range() with longs
  913. self.assertEqual(range(-2**100), [])
  914. self.assertEqual(range(0, -2**100), [])
  915. self.assertEqual(range(0, 2**100, -1), [])
  916. self.assertEqual(range(0, 2**100, -1), [])
  917. a = long(10 * sys.maxint)
  918. b = long(100 * sys.maxint)
  919. c = long(50 * sys.maxint)
  920. self.assertEqual(range(a, a+2), [a, a+1])
  921. self.assertEqual(range(a+2, a, -1L), [a+2, a+1])
  922. self.assertEqual(range(a+4, a, -2), [a+4, a+2])
  923. seq = range(a, b, c)
  924. self.assert_(a in seq)
  925. self.assert_(b not in seq)
  926. self.assertEqual(len(seq), 2)
  927. seq = range(b, a, -c)
  928. self.assert_(b in seq)
  929. self.assert_(a not in seq)
  930. self.assertEqual(len(seq), 2)
  931. seq = range(-a, -b, -c)
  932. self.assert_(-a in seq)
  933. self.assert_(-b not in seq)
  934. self.assertEqual(len(seq), 2)
  935. self.assertRaises(TypeError, range)
  936. self.assertRaises(TypeError, range, 1, 2, 3, 4)
  937. self.assertRaises(ValueError, range, 1, 2, 0)
  938. self.assertRaises(ValueError, range, a, a + 1, long(0))
  939. class badzero(int):
  940. def __cmp__(self, other):
  941. raise RuntimeError
  942. __hash__ = None # Invalid cmp makes this unhashable
  943. self.assertRaises(RuntimeError, range, a, a + 1, badzero(1))
  944. # Reject floats when it would require PyLongs to represent.
  945. # (smaller floats still accepted, but deprecated)
  946. self.assertRaises(TypeError, range, 1e100, 1e101, 1e101)
  947. self.assertRaises(TypeError, range, 0, "spam")
  948. self.assertRaises(TypeError, range, 0, 42, "spam")
  949. self.assertRaises(OverflowError, range, -sys.maxint, sys.maxint)
  950. self.assertRaises(OverflowError, range, 0, 2*sys.maxint)
  951. def test_input_and_raw_input(self):
  952. self.write_testfile()
  953. fp = open(TESTFN, 'r')
  954. savestdin = sys.stdin
  955. savestdout = sys.stdout # Eats the echo
  956. try:
  957. sys.stdin = fp
  958. sys.stdout = BitBucket()
  959. self.assertEqual(input(), 2)
  960. self.assertEqual(input('testing\n'), 2)
  961. self.assertEqual(raw_input(), 'The quick brown fox jumps over the lazy dog.')
  962. self.assertEqual(raw_input('testing\n'), 'Dear John')
  963. # SF 1535165: don't segfault on closed stdin
  964. # sys.stdout must be a regular file for triggering
  965. sys.stdout = savestdout
  966. sys.stdin.close()
  967. self.assertRaises(ValueError, input)
  968. sys.stdout = BitBucket()
  969. sys.stdin = cStringIO.StringIO("NULL\0")
  970. self.assertRaises(TypeError, input, 42, 42)
  971. sys.stdin = cStringIO.StringIO(" 'whitespace'")
  972. self.assertEqual(input(), 'whitespace')
  973. sys.stdin = cStringIO.StringIO()
  974. self.assertRaises(EOFError, input)
  975. # SF 876178: make sure input() respect future options.
  976. sys.stdin = cStringIO.StringIO('1/2')
  977. sys.stdout = cStringIO.StringIO()
  978. exec compile('print input()', 'test_builtin_tmp', 'exec')
  979. sys.stdin.seek(0, 0)
  980. exec compile('from __future__ import division;print input()',
  981. 'test_builtin_tmp', 'exec')
  982. sys.stdin.seek(0, 0)
  983. exec compile('print input()', 'test_builtin_tmp', 'exec')
  984. # The result we expect depends on whether new division semantics
  985. # are already in effect.
  986. if 1/2 == 0:
  987. # This test was compiled with old semantics.
  988. expected = ['0', '0.5', '0']
  989. else:
  990. # This test was compiled with new semantics (e.g., -Qnew
  991. # was given on the command line.
  992. expected = ['0.5', '0.5', '0.5']
  993. self.assertEqual(sys.stdout.getvalue().splitlines(), expected)
  994. del sys.stdout
  995. self.assertRaises(RuntimeError, input, 'prompt')
  996. del sys.stdin
  997. self.assertRaises(RuntimeError, input, 'prompt')
  998. finally:
  999. sys.stdin = savestdin
  1000. sys.stdout = savestdout
  1001. fp.close()
  1002. unlink(TESTFN)
  1003. def test_reduce(self):
  1004. self.assertEqual(reduce(lambda x, y: x+y, ['a', 'b', 'c'], ''), 'abc')
  1005. self.assertEqual(
  1006. reduce(lambda x, y: x+y, [['a', 'c'], [], ['d', 'w']], []),
  1007. ['a','c','d','w']
  1008. )
  1009. self.assertEqual(reduce(lambda x, y: x*y, range(2,8), 1), 5040)
  1010. self.assertEqual(
  1011. reduce(lambda x, y: x*y, range(2,21), 1L),
  1012. 2432902008176640000L
  1013. )
  1014. self.assertEqual(reduce(lambda x, y: x+y, Squares(10)), 285)
  1015. self.assertEqual(reduce(lambda x, y: x+y, Squares(10), 0), 285)
  1016. self.assertEqual(reduce(lambda x, y: x+y, Squares(0), 0), 0)
  1017. self.assertRaises(TypeError, reduce)
  1018. self.assertRaises(TypeError, reduce, 42, 42)
  1019. self.assertRaises(TypeError, reduce, 42, 42, 42)
  1020. self.assertEqual(reduce(42, "1"), "1") # func is never called with one item
  1021. self.assertEqual(reduce(42, "", "1"), "1") # func is never called with one item
  1022. self.assertRaises(TypeError, reduce, 42, (42, 42))
  1023. class BadSeq:
  1024. def __getitem__(self, index):
  1025. raise ValueError
  1026. self.assertRaises(ValueError, reduce, 42, BadSeq())
  1027. def test_reload(self):
  1028. import marshal
  1029. reload(marshal)
  1030. import string
  1031. reload(string)
  1032. ## import sys
  1033. ## self.assertRaises(ImportError, reload, sys)
  1034. def test_repr(self):
  1035. self.assertEqual(repr(''), '\'\'')
  1036. self.assertEqual(repr(0), '0')
  1037. self.assertEqual(repr(0L), '0L')
  1038. self.assertEqual(repr(()), '()')
  1039. self.assertEqual(repr([]), '[]')
  1040. self.assertEqual(repr({}), '{}')
  1041. a = []
  1042. a.append(a)
  1043. self.assertEqual(repr(a), '[[...]]')
  1044. a = {}
  1045. a[0] = a
  1046. self.assertEqual(repr(a), '{0: {...}}')
  1047. def test_round(self):
  1048. self.assertEqual(round(0.0), 0.0)
  1049. self.assertEqual(type(round(0.0)), float) # Will be int in 3.0.
  1050. self.assertEqual(round(1.0), 1.0)
  1051. self.assertEqual(round(10.0), 10.0)
  1052. self.assertEqual(round(1000000000.0), 1000000000.0)
  1053. self.assertEqual(round(1e20), 1e20)
  1054. self.assertEqual(round(-1.0), -1.0)
  1055. self.assertEqual(round(-10.0), -10.0)
  1056. self.assertEqual(round(-1000000000.0), -1000000000.0)
  1057. self.assertEqual(round(-1e20), -1e20)
  1058. self.assertEqual(round(0.1), 0.0)
  1059. self.assertEqual(round(1.1), 1.0)
  1060. self.assertEqual(round(10.1), 10.0)
  1061. self.assertEqual(round(1000000000.1), 1000000000.0)
  1062. self.assertEqual(round(-1.1), -1.0)
  1063. self.assertEqual(round(-10.1), -10.0)
  1064. self.assertEqual(round(-1000000000.1), -1000000000.0)
  1065. self.assertEqual(round(0.9), 1.0)
  1066. self.assertEqual(round(9.9), 10.0)
  1067. self.assertEqual(round(999999999.9), 1000000000.0)
  1068. self.assertEqual(round(-0.9), -1.0)
  1069. self.assertEqual(round(-9.9), -10.0)
  1070. self.assertEqual(round(-999999999.9), -1000000000.0)
  1071. self.assertEqual(round(-8.0, -1), -10.0)
  1072. self.assertEqual(type(round(-8.0, -1)), float)
  1073. self.assertEqual(type(round(-8.0, 0)), float)
  1074. self.assertEqual(type(round(-8.0, 1)), float)
  1075. # Check half rounding behaviour.
  1076. self.assertEqual(round(5.5), 6)
  1077. self.assertEqual(round(6.5), 7)
  1078. self.assertEqual(round(-5.5), -6)
  1079. self.assertEqual(round(-6.5), -7)
  1080. # Check behavior on ints
  1081. self.assertEqual(round(0), 0)
  1082. self.assertEqual(round(8), 8)
  1083. self.assertEqual(round(-8), -8)
  1084. self.assertEqual(type(round(0)), float) # Will be int in 3.0.
  1085. self.assertEqual(type(round(-8, -1)), float)
  1086. self.assertEqual(type(round(-8, 0)), float)
  1087. self.assertEqual(type(round(-8, 1)), float)
  1088. # test new kwargs
  1089. self.assertEqual(round(number=-8.0, ndigits=-1), -10.0)
  1090. self.assertRaises(TypeError, round)
  1091. # test generic rounding delegation for reals
  1092. class TestRound(object):
  1093. def __float__(self):
  1094. return 23.0
  1095. class TestNoRound(object):
  1096. pass
  1097. self.assertEqual(round(TestRound()), 23)
  1098. self.assertRaises(TypeError, round, 1, 2, 3)
  1099. self.assertRaises(TypeError, round, TestNoRound())
  1100. t = TestNoRound()
  1101. t.__float__ = lambda *args: args
  1102. self.assertRaises(TypeError, round, t)
  1103. self.assertRaises(TypeError, round, t, 0)
  1104. def test_setattr(self):
  1105. setattr(sys, 'spam', 1)
  1106. self.assertEqual(sys.spam, 1)
  1107. self.assertRaises(TypeError, setattr, sys, 1, 'spam')
  1108. self.assertRaises(TypeError, setattr)
  1109. def test_sum(self):
  1110. self.assertEqual(sum([]), 0)
  1111. self.assertEqual(sum(range(2,8)), 27)
  1112. self.assertEqual(sum(iter(range(2,8))), 27)
  1113. self.assertEqual(sum(Squares(10)), 285)
  1114. self.assertEqual(sum(iter(Squares(10))), 285)
  1115. self.assertEqual(sum([[1], [2], [3]], []), [1, 2, 3])
  1116. self.assertRaises(TypeError, sum)
  1117. self.assertRaises(TypeError, sum, 42)
  1118. self.assertRaises(TypeError, sum, ['a', 'b', 'c'])
  1119. self.assertRaises(TypeError, sum, ['a', 'b', 'c'], '')
  1120. self.assertRaises(TypeError, sum, [[1], [2], [3]])
  1121. self.assertRaises(TypeError, sum, [{2:3}])
  1122. self.assertRaises(TypeError, sum, [{2:3}]*2, {2:3})
  1123. class BadSeq:
  1124. def __getitem__(self, index):
  1125. raise ValueError
  1126. self.assertRaises(ValueError, sum, BadSeq())
  1127. def test_type(self):
  1128. self.assertEqual(type(''), type('123'))
  1129. self.assertNotEqual(type(''), type(()))
  1130. def test_unichr(self):
  1131. if have_unicode:
  1132. self.assertEqual(unichr(32), unicode(' '))
  1133. self.assertEqual(unichr(65), unicode('A'))
  1134. self.assertEqual(unichr(97), unicode('a'))
  1135. self.assertEqual(
  1136. unichr(sys.maxunicode),
  1137. unicode('\\U%08x' % (sys.maxunicode), 'unicode-escape')
  1138. )
  1139. self.assertRaises(ValueError, unichr, sys.maxunicode+1)
  1140. self.assertRaises(TypeError, unichr)
  1141. self.assertRaises((OverflowError, ValueError), unichr, 2**32)
  1142. # We don't want self in vars(), so these are static methods
  1143. @staticmethod
  1144. def get_vars_f0():
  1145. return vars()
  1146. @staticmethod
  1147. def get_vars_f2():
  1148. BuiltinTest.get_vars_f0()
  1149. a = 1
  1150. b = 2
  1151. return vars()
  1152. def test_vars(self):
  1153. self.assertEqual(set(vars()), set(dir()))
  1154. import sys
  1155. self.assertEqual(set(vars(sys)), set(dir(sys)))
  1156. self.assertEqual(self.get_vars_f0(), {})
  1157. self.assertEqual(self.get_vars_f2(), {'a': 1, 'b': 2})
  1158. self.assertRaises(TypeError, vars, 42, 42)
  1159. self.assertRaises(TypeError, vars, 42)
  1160. def test_zip(self):
  1161. a = (1, 2, 3)
  1162. b = (4, 5, 6)
  1163. t = [(1, 4), (2, 5), (3, 6)]
  1164. self.assertEqual(zip(a, b), t)
  1165. b = [4, 5, 6]
  1166. self.assertEqual(zip(a, b), t)
  1167. b = (4, 5, 6, 7)
  1168. self.assertEqual(zip(a, b), t)
  1169. class I:
  1170. def __getitem__(self, i):
  1171. if i < 0 or i > 2: raise IndexError
  1172. return i + 4
  1173. self.assertEqual(zip(a, I()), t)
  1174. self.assertEqual(zip(), [])
  1175. self.assertEqual(zip(*[]), [])
  1176. self.assertRaises(TypeError, zip, None)
  1177. class G:
  1178. pass
  1179. self.assertRaises(TypeError, zip, a, G())
  1180. # Make sure zip doesn't try to allocate a billion elements for the
  1181. # result list when one of its arguments doesn't say how long it is.
  1182. # A MemoryError is the most likely failure mode.
  1183. class SequenceWithoutALength:
  1184. def __getitem__(self, i):
  1185. if i == 5:
  1186. raise IndexError
  1187. else:
  1188. return i
  1189. self.assertEqual(
  1190. zip(SequenceWithoutALength(), xrange(2**30)),
  1191. list(enumerate(range(5)))
  1192. )
  1193. class BadSeq:
  1194. def __getitem__(self, i):
  1195. if i == 5:
  1196. raise ValueError
  1197. else:
  1198. return i
  1199. self.assertRaises(ValueError, zip, BadSeq(), BadSeq())
  1200. def test_format(self):
  1201. # Test the basic machinery of the format() builtin. Don't test
  1202. # the specifics of the various formatters
  1203. self.assertEqual(format(3, ''), '3')
  1204. # Returns some classes to use for various tests. There's
  1205. # an old-style version, and a new-style version
  1206. def classes_new():
  1207. class A(object):
  1208. def __init__(self, x):
  1209. self.x = x
  1210. def __format__(self, format_spec):
  1211. return str(self.x) + format_spec
  1212. class DerivedFromA(A):
  1213. pass
  1214. class Simple(object): pass
  1215. class DerivedFromSimple(Simple):
  1216. def __init__(self, x):
  1217. self.x = x
  1218. def __format__(self, format_spec):
  1219. return str(self.x) + format_spec
  1220. class DerivedFromSimple2(DerivedFromSimple): pass
  1221. return A, DerivedFromA, DerivedFromSimple, DerivedFromSimple2
  1222. # In 3.0, classes_classic has the same meaning as classes_new
  1223. def classes_classic():
  1224. class A:
  1225. def __init__(self, x):
  1226. self.x = x
  1227. def __format__(self, format_spec):
  1228. return str(self.x) + format_spec
  1229. class DerivedFromA(A):
  1230. pass
  1231. class Simple: pass
  1232. class DerivedFromSimple(Simple):
  1233. def __init__(self, x):
  1234. self.x = x
  1235. def __format__(self, format_spec):
  1236. return str(self.x) + format_spec
  1237. class DerivedFromSimple2(DerivedFromSimple): pass
  1238. return A, DerivedFromA, DerivedFromSimple, DerivedFromSimple2
  1239. def class_test(A, DerivedFromA, DerivedFromSimple, DerivedFromSimple2):
  1240. self.assertEqual(format(A(3), 'spec'), '3spec')
  1241. self.assertEqual(format(DerivedFromA(4), 'spec'), '4spec')
  1242. self.assertEqual(format(DerivedFromSimple(5), 'abc'), '5abc')
  1243. self.assertEqual(format(DerivedFromSimple2(10), 'abcdef'),
  1244. '10abcdef')
  1245. class_test(*classes_new())
  1246. class_test(*classes_classic())
  1247. def empty_format_spec(value):
  1248. # test that:
  1249. # format(x, '') == str(x)
  1250. # format(x) == str(x)
  1251. self.assertEqual(format(value, ""), str(value))
  1252. self.assertEqual(format(value), str(value))
  1253. # for builtin types, format(x, "") == str(x)
  1254. empty_format_spec(17**13)
  1255. empty_format_spec(1.0)
  1256. empty_format_spec(3.1415e104)
  1257. empty_format_spec(-3.1415e104)
  1258. empty_format_spec(3.1415e-104)
  1259. empty_format_spec(-3.1415e-104)
  1260. empty_format_spec(object)
  1261. empty_format_spec(None)
  1262. # TypeError because self.__format__ returns the wrong type
  1263. class BadFormatResult:
  1264. def __format__(self, format_spec):
  1265. return 1.0
  1266. self.assertRaises(TypeError, format, BadFormatResult(), "")
  1267. # TypeError because format_spec is not unicode or str
  1268. self.assertRaises(TypeError, format, object(), 4)
  1269. self.assertRaises(TypeError, format, object(), object())
  1270. # tests for object.__format__ really belong elsewhere, but
  1271. # there's no good place to put them
  1272. x = object().__format__('')
  1273. self.assert_(x.startswith('<object object at'))
  1274. # first argument to object.__format__ must be string
  1275. self.assertRaises(TypeError, object().__format__, 3)
  1276. self.assertRaises(TypeError, object().__format__, object())
  1277. self.assertRaises(TypeError, object().__format__, None)
  1278. # make sure we can take a subclass of str as a format spec
  1279. class DerivedFromStr(str): pass
  1280. self.assertEqual(format(0, DerivedFromStr('10')), ' 0')
  1281. def test_bin(self):
  1282. self.assertEqual(bin(0), '0b0')
  1283. self.assertEqual(bin(1), '0b1')
  1284. self.assertEqual(bin(-1), '-0b1')
  1285. self.assertEqual(bin(2**65), '0b1' + '0' * 65)
  1286. self.assertEqual(bin(2**65-1), '0b' + '1' * 65)
  1287. self.assertEqual(bin(-(2**65)), '-0b1' + '0' * 65)
  1288. self.assertEqual(bin(-(2**65-1)), '-0b' + '1' * 65)
  1289. def test_bytearray_translate(self):
  1290. x = bytearray("abc")
  1291. self.assertRaises(ValueError, x.translate, "1", 1)
  1292. self.assertRaises(TypeError, x.translate, "1"*256, 1)
  1293. class TestSorted(unittest.TestCase):
  1294. def test_basic(self):
  1295. data = range(100)
  1296. copy = data[:]
  1297. random.shuffle(copy)
  1298. self.assertEqual(data, sorted(copy))
  1299. self.assertNotEqual(data, copy)
  1300. data.reverse()
  1301. random.shuffle(copy)
  1302. self.assertEqual(data, sorted(copy, cmp=lambda x, y: cmp(y,x)))
  1303. self.assertNotEqual(data, copy)
  1304. random.shuffle(copy)
  1305. self.assertEqual(data, sorted(copy, key=lambda x: -x))
  1306. self.assertNotEqual(data, copy)
  1307. random.shuffle(copy)
  1308. self.assertEqual(data, sorted(copy, reverse=1))
  1309. self.assertNotEqual(data, copy)
  1310. def test_inputtypes(self):
  1311. s = 'abracadabra'
  1312. types = [list, tuple]
  1313. if have_unicode:
  1314. types.insert(0, unicode)
  1315. for T in types:
  1316. self.assertEqual(sorted(s), sorted(T(s)))
  1317. s = ''.join(dict.fromkeys(s).keys()) # unique letters only
  1318. types = [set, frozenset, list, tuple, dict.fromkeys]
  1319. if have_unicode:
  1320. types.insert(0, unicode)
  1321. for T in types:
  1322. self.assertEqual(sorted(s), sorted(T(s)))
  1323. def test_baddecorator(self):
  1324. data = 'The quick Brown fox Jumped over The lazy Dog'.split()
  1325. self.assertRaises(TypeError, sorted, data, None, lambda x,y: 0)
  1326. def test_main(verbose=None):
  1327. test_classes = (BuiltinTest, TestSorted)
  1328. run_unittest(*test_classes)
  1329. # verify reference counting
  1330. if verbose and hasattr(sys, "gettotalrefcount"):
  1331. import gc
  1332. counts = [None] * 5
  1333. for i in xrange(len(counts)):
  1334. run_unittest(*test_classes)
  1335. gc.collect()
  1336. counts[i] = sys.gettotalrefcount()
  1337. print counts
  1338. if __name__ == "__main__":
  1339. test_main(verbose=True)