/Lib/test/test_unicode.py

http://unladen-swallow.googlecode.com/ · Python · 1136 lines · 864 code · 156 blank · 116 comment · 33 complexity · eb73381ad196cf6f4ca903146e6c5208 MD5 · raw file

  1. # -*- coding: iso-8859-1 -*-
  2. """ Test script for the Unicode implementation.
  3. Written by Marc-Andre Lemburg (mal@lemburg.com).
  4. (c) Copyright CNRI, All Rights Reserved. NO WARRANTY.
  5. """#"
  6. import sys, struct, codecs
  7. from test import test_support, string_tests
  8. # Error handling (bad decoder return)
  9. def search_function(encoding):
  10. def decode1(input, errors="strict"):
  11. return 42 # not a tuple
  12. def encode1(input, errors="strict"):
  13. return 42 # not a tuple
  14. def encode2(input, errors="strict"):
  15. return (42, 42) # no unicode
  16. def decode2(input, errors="strict"):
  17. return (42, 42) # no unicode
  18. if encoding=="test.unicode1":
  19. return (encode1, decode1, None, None)
  20. elif encoding=="test.unicode2":
  21. return (encode2, decode2, None, None)
  22. else:
  23. return None
  24. codecs.register(search_function)
  25. class UnicodeTest(
  26. string_tests.CommonTest,
  27. string_tests.MixinStrUnicodeUserStringTest,
  28. string_tests.MixinStrUnicodeTest,
  29. ):
  30. type2test = unicode
  31. def checkequalnofix(self, result, object, methodname, *args):
  32. method = getattr(object, methodname)
  33. realresult = method(*args)
  34. self.assertEqual(realresult, result)
  35. self.assert_(type(realresult) is type(result))
  36. # if the original is returned make sure that
  37. # this doesn't happen with subclasses
  38. if realresult is object:
  39. class usub(unicode):
  40. def __repr__(self):
  41. return 'usub(%r)' % unicode.__repr__(self)
  42. object = usub(object)
  43. method = getattr(object, methodname)
  44. realresult = method(*args)
  45. self.assertEqual(realresult, result)
  46. self.assert_(object is not realresult)
  47. def test_literals(self):
  48. self.assertEqual(u'\xff', u'\u00ff')
  49. self.assertEqual(u'\uffff', u'\U0000ffff')
  50. self.assertRaises(SyntaxError, eval, 'u\'\\Ufffffffe\'')
  51. self.assertRaises(SyntaxError, eval, 'u\'\\Uffffffff\'')
  52. self.assertRaises(SyntaxError, eval, 'u\'\\U%08x\'' % 0x110000)
  53. def test_repr(self):
  54. if not sys.platform.startswith('java'):
  55. # Test basic sanity of repr()
  56. self.assertEqual(repr(u'abc'), "u'abc'")
  57. self.assertEqual(repr(u'ab\\c'), "u'ab\\\\c'")
  58. self.assertEqual(repr(u'ab\\'), "u'ab\\\\'")
  59. self.assertEqual(repr(u'\\c'), "u'\\\\c'")
  60. self.assertEqual(repr(u'\\'), "u'\\\\'")
  61. self.assertEqual(repr(u'\n'), "u'\\n'")
  62. self.assertEqual(repr(u'\r'), "u'\\r'")
  63. self.assertEqual(repr(u'\t'), "u'\\t'")
  64. self.assertEqual(repr(u'\b'), "u'\\x08'")
  65. self.assertEqual(repr(u"'\""), """u'\\'"'""")
  66. self.assertEqual(repr(u"'\""), """u'\\'"'""")
  67. self.assertEqual(repr(u"'"), '''u"'"''')
  68. self.assertEqual(repr(u'"'), """u'"'""")
  69. latin1repr = (
  70. "u'\\x00\\x01\\x02\\x03\\x04\\x05\\x06\\x07\\x08\\t\\n\\x0b\\x0c\\r"
  71. "\\x0e\\x0f\\x10\\x11\\x12\\x13\\x14\\x15\\x16\\x17\\x18\\x19\\x1a"
  72. "\\x1b\\x1c\\x1d\\x1e\\x1f !\"#$%&\\'()*+,-./0123456789:;<=>?@ABCDEFGHI"
  73. "JKLMNOPQRSTUVWXYZ[\\\\]^_`abcdefghijklmnopqrstuvwxyz{|}~\\x7f"
  74. "\\x80\\x81\\x82\\x83\\x84\\x85\\x86\\x87\\x88\\x89\\x8a\\x8b\\x8c\\x8d"
  75. "\\x8e\\x8f\\x90\\x91\\x92\\x93\\x94\\x95\\x96\\x97\\x98\\x99\\x9a\\x9b"
  76. "\\x9c\\x9d\\x9e\\x9f\\xa0\\xa1\\xa2\\xa3\\xa4\\xa5\\xa6\\xa7\\xa8\\xa9"
  77. "\\xaa\\xab\\xac\\xad\\xae\\xaf\\xb0\\xb1\\xb2\\xb3\\xb4\\xb5\\xb6\\xb7"
  78. "\\xb8\\xb9\\xba\\xbb\\xbc\\xbd\\xbe\\xbf\\xc0\\xc1\\xc2\\xc3\\xc4\\xc5"
  79. "\\xc6\\xc7\\xc8\\xc9\\xca\\xcb\\xcc\\xcd\\xce\\xcf\\xd0\\xd1\\xd2\\xd3"
  80. "\\xd4\\xd5\\xd6\\xd7\\xd8\\xd9\\xda\\xdb\\xdc\\xdd\\xde\\xdf\\xe0\\xe1"
  81. "\\xe2\\xe3\\xe4\\xe5\\xe6\\xe7\\xe8\\xe9\\xea\\xeb\\xec\\xed\\xee\\xef"
  82. "\\xf0\\xf1\\xf2\\xf3\\xf4\\xf5\\xf6\\xf7\\xf8\\xf9\\xfa\\xfb\\xfc\\xfd"
  83. "\\xfe\\xff'")
  84. testrepr = repr(u''.join(map(unichr, xrange(256))))
  85. self.assertEqual(testrepr, latin1repr)
  86. # Test repr works on wide unicode escapes without overflow.
  87. self.assertEqual(repr(u"\U00010000" * 39 + u"\uffff" * 4096),
  88. repr(u"\U00010000" * 39 + u"\uffff" * 4096))
  89. def test_count(self):
  90. string_tests.CommonTest.test_count(self)
  91. # check mixed argument types
  92. self.checkequalnofix(3, 'aaa', 'count', u'a')
  93. self.checkequalnofix(0, 'aaa', 'count', u'b')
  94. self.checkequalnofix(3, u'aaa', 'count', 'a')
  95. self.checkequalnofix(0, u'aaa', 'count', 'b')
  96. self.checkequalnofix(0, u'aaa', 'count', 'b')
  97. self.checkequalnofix(1, u'aaa', 'count', 'a', -1)
  98. self.checkequalnofix(3, u'aaa', 'count', 'a', -10)
  99. self.checkequalnofix(2, u'aaa', 'count', 'a', 0, -1)
  100. self.checkequalnofix(0, u'aaa', 'count', 'a', 0, -10)
  101. def test_find(self):
  102. self.checkequalnofix(0, u'abcdefghiabc', 'find', u'abc')
  103. self.checkequalnofix(9, u'abcdefghiabc', 'find', u'abc', 1)
  104. self.checkequalnofix(-1, u'abcdefghiabc', 'find', u'def', 4)
  105. self.assertRaises(TypeError, u'hello'.find)
  106. self.assertRaises(TypeError, u'hello'.find, 42)
  107. def test_rfind(self):
  108. string_tests.CommonTest.test_rfind(self)
  109. # check mixed argument types
  110. self.checkequalnofix(9, 'abcdefghiabc', 'rfind', u'abc')
  111. self.checkequalnofix(12, 'abcdefghiabc', 'rfind', u'')
  112. self.checkequalnofix(12, u'abcdefghiabc', 'rfind', '')
  113. def test_index(self):
  114. string_tests.CommonTest.test_index(self)
  115. # check mixed argument types
  116. for (t1, t2) in ((str, unicode), (unicode, str)):
  117. self.checkequalnofix(0, t1('abcdefghiabc'), 'index', t2(''))
  118. self.checkequalnofix(3, t1('abcdefghiabc'), 'index', t2('def'))
  119. self.checkequalnofix(0, t1('abcdefghiabc'), 'index', t2('abc'))
  120. self.checkequalnofix(9, t1('abcdefghiabc'), 'index', t2('abc'), 1)
  121. self.assertRaises(ValueError, t1('abcdefghiabc').index, t2('hib'))
  122. self.assertRaises(ValueError, t1('abcdefghiab').index, t2('abc'), 1)
  123. self.assertRaises(ValueError, t1('abcdefghi').index, t2('ghi'), 8)
  124. self.assertRaises(ValueError, t1('abcdefghi').index, t2('ghi'), -1)
  125. def test_rindex(self):
  126. string_tests.CommonTest.test_rindex(self)
  127. # check mixed argument types
  128. for (t1, t2) in ((str, unicode), (unicode, str)):
  129. self.checkequalnofix(12, t1('abcdefghiabc'), 'rindex', t2(''))
  130. self.checkequalnofix(3, t1('abcdefghiabc'), 'rindex', t2('def'))
  131. self.checkequalnofix(9, t1('abcdefghiabc'), 'rindex', t2('abc'))
  132. self.checkequalnofix(0, t1('abcdefghiabc'), 'rindex', t2('abc'), 0, -1)
  133. self.assertRaises(ValueError, t1('abcdefghiabc').rindex, t2('hib'))
  134. self.assertRaises(ValueError, t1('defghiabc').rindex, t2('def'), 1)
  135. self.assertRaises(ValueError, t1('defghiabc').rindex, t2('abc'), 0, -1)
  136. self.assertRaises(ValueError, t1('abcdefghi').rindex, t2('ghi'), 0, 8)
  137. self.assertRaises(ValueError, t1('abcdefghi').rindex, t2('ghi'), 0, -1)
  138. def test_translate(self):
  139. self.checkequalnofix(u'bbbc', u'abababc', 'translate', {ord('a'):None})
  140. self.checkequalnofix(u'iiic', u'abababc', 'translate', {ord('a'):None, ord('b'):ord('i')})
  141. self.checkequalnofix(u'iiix', u'abababc', 'translate', {ord('a'):None, ord('b'):ord('i'), ord('c'):u'x'})
  142. self.checkequalnofix(u'<i><i><i>c', u'abababc', 'translate', {ord('a'):None, ord('b'):u'<i>'})
  143. self.checkequalnofix(u'c', u'abababc', 'translate', {ord('a'):None, ord('b'):u''})
  144. self.checkequalnofix(u'xyyx', u'xzx', 'translate', {ord('z'):u'yy'})
  145. self.assertRaises(TypeError, u'hello'.translate)
  146. self.assertRaises(TypeError, u'abababc'.translate, {ord('a'):''})
  147. def test_split(self):
  148. string_tests.CommonTest.test_split(self)
  149. # Mixed arguments
  150. self.checkequalnofix([u'a', u'b', u'c', u'd'], u'a//b//c//d', 'split', '//')
  151. self.checkequalnofix([u'a', u'b', u'c', u'd'], 'a//b//c//d', 'split', u'//')
  152. self.checkequalnofix([u'endcase ', u''], u'endcase test', 'split', 'test')
  153. def test_join(self):
  154. string_tests.MixinStrUnicodeUserStringTest.test_join(self)
  155. # mixed arguments
  156. self.checkequalnofix(u'a b c d', u' ', 'join', ['a', 'b', u'c', u'd'])
  157. self.checkequalnofix(u'abcd', u'', 'join', (u'a', u'b', u'c', u'd'))
  158. self.checkequalnofix(u'w x y z', u' ', 'join', string_tests.Sequence('wxyz'))
  159. self.checkequalnofix(u'a b c d', ' ', 'join', [u'a', u'b', u'c', u'd'])
  160. self.checkequalnofix(u'a b c d', ' ', 'join', ['a', 'b', u'c', u'd'])
  161. self.checkequalnofix(u'abcd', '', 'join', (u'a', u'b', u'c', u'd'))
  162. self.checkequalnofix(u'w x y z', ' ', 'join', string_tests.Sequence(u'wxyz'))
  163. def test_strip(self):
  164. string_tests.CommonTest.test_strip(self)
  165. self.assertRaises(UnicodeError, u"hello".strip, "\xff")
  166. def test_replace(self):
  167. string_tests.CommonTest.test_replace(self)
  168. # method call forwarded from str implementation because of unicode argument
  169. self.checkequalnofix(u'one@two!three!', 'one!two!three!', 'replace', u'!', u'@', 1)
  170. self.assertRaises(TypeError, 'replace'.replace, u"r", 42)
  171. def test_comparison(self):
  172. # Comparisons:
  173. self.assertEqual(u'abc', 'abc')
  174. self.assertEqual('abc', u'abc')
  175. self.assertEqual(u'abc', u'abc')
  176. self.assert_(u'abcd' > 'abc')
  177. self.assert_('abcd' > u'abc')
  178. self.assert_(u'abcd' > u'abc')
  179. self.assert_(u'abc' < 'abcd')
  180. self.assert_('abc' < u'abcd')
  181. self.assert_(u'abc' < u'abcd')
  182. if 0:
  183. # Move these tests to a Unicode collation module test...
  184. # Testing UTF-16 code point order comparisons...
  185. # No surrogates, no fixup required.
  186. self.assert_(u'\u0061' < u'\u20ac')
  187. # Non surrogate below surrogate value, no fixup required
  188. self.assert_(u'\u0061' < u'\ud800\udc02')
  189. # Non surrogate above surrogate value, fixup required
  190. def test_lecmp(s, s2):
  191. self.assert_(s < s2)
  192. def test_fixup(s):
  193. s2 = u'\ud800\udc01'
  194. test_lecmp(s, s2)
  195. s2 = u'\ud900\udc01'
  196. test_lecmp(s, s2)
  197. s2 = u'\uda00\udc01'
  198. test_lecmp(s, s2)
  199. s2 = u'\udb00\udc01'
  200. test_lecmp(s, s2)
  201. s2 = u'\ud800\udd01'
  202. test_lecmp(s, s2)
  203. s2 = u'\ud900\udd01'
  204. test_lecmp(s, s2)
  205. s2 = u'\uda00\udd01'
  206. test_lecmp(s, s2)
  207. s2 = u'\udb00\udd01'
  208. test_lecmp(s, s2)
  209. s2 = u'\ud800\ude01'
  210. test_lecmp(s, s2)
  211. s2 = u'\ud900\ude01'
  212. test_lecmp(s, s2)
  213. s2 = u'\uda00\ude01'
  214. test_lecmp(s, s2)
  215. s2 = u'\udb00\ude01'
  216. test_lecmp(s, s2)
  217. s2 = u'\ud800\udfff'
  218. test_lecmp(s, s2)
  219. s2 = u'\ud900\udfff'
  220. test_lecmp(s, s2)
  221. s2 = u'\uda00\udfff'
  222. test_lecmp(s, s2)
  223. s2 = u'\udb00\udfff'
  224. test_lecmp(s, s2)
  225. test_fixup(u'\ue000')
  226. test_fixup(u'\uff61')
  227. # Surrogates on both sides, no fixup required
  228. self.assert_(u'\ud800\udc02' < u'\ud84d\udc56')
  229. def test_islower(self):
  230. string_tests.MixinStrUnicodeUserStringTest.test_islower(self)
  231. self.checkequalnofix(False, u'\u1FFc', 'islower')
  232. def test_isupper(self):
  233. string_tests.MixinStrUnicodeUserStringTest.test_isupper(self)
  234. if not sys.platform.startswith('java'):
  235. self.checkequalnofix(False, u'\u1FFc', 'isupper')
  236. def test_istitle(self):
  237. string_tests.MixinStrUnicodeUserStringTest.test_title(self)
  238. self.checkequalnofix(True, u'\u1FFc', 'istitle')
  239. self.checkequalnofix(True, u'Greek \u1FFcitlecases ...', 'istitle')
  240. def test_isspace(self):
  241. string_tests.MixinStrUnicodeUserStringTest.test_isspace(self)
  242. self.checkequalnofix(True, u'\u2000', 'isspace')
  243. self.checkequalnofix(True, u'\u200a', 'isspace')
  244. self.checkequalnofix(False, u'\u2014', 'isspace')
  245. def test_isalpha(self):
  246. string_tests.MixinStrUnicodeUserStringTest.test_isalpha(self)
  247. self.checkequalnofix(True, u'\u1FFc', 'isalpha')
  248. def test_isdecimal(self):
  249. self.checkequalnofix(False, u'', 'isdecimal')
  250. self.checkequalnofix(False, u'a', 'isdecimal')
  251. self.checkequalnofix(True, u'0', 'isdecimal')
  252. self.checkequalnofix(False, u'\u2460', 'isdecimal') # CIRCLED DIGIT ONE
  253. self.checkequalnofix(False, u'\xbc', 'isdecimal') # VULGAR FRACTION ONE QUARTER
  254. self.checkequalnofix(True, u'\u0660', 'isdecimal') # ARABIC-INDIC DIGIT ZERO
  255. self.checkequalnofix(True, u'0123456789', 'isdecimal')
  256. self.checkequalnofix(False, u'0123456789a', 'isdecimal')
  257. self.checkraises(TypeError, 'abc', 'isdecimal', 42)
  258. def test_isdigit(self):
  259. string_tests.MixinStrUnicodeUserStringTest.test_isdigit(self)
  260. self.checkequalnofix(True, u'\u2460', 'isdigit')
  261. self.checkequalnofix(False, u'\xbc', 'isdigit')
  262. self.checkequalnofix(True, u'\u0660', 'isdigit')
  263. def test_isnumeric(self):
  264. self.checkequalnofix(False, u'', 'isnumeric')
  265. self.checkequalnofix(False, u'a', 'isnumeric')
  266. self.checkequalnofix(True, u'0', 'isnumeric')
  267. self.checkequalnofix(True, u'\u2460', 'isnumeric')
  268. self.checkequalnofix(True, u'\xbc', 'isnumeric')
  269. self.checkequalnofix(True, u'\u0660', 'isnumeric')
  270. self.checkequalnofix(True, u'0123456789', 'isnumeric')
  271. self.checkequalnofix(False, u'0123456789a', 'isnumeric')
  272. self.assertRaises(TypeError, u"abc".isnumeric, 42)
  273. def test_contains(self):
  274. # Testing Unicode contains method
  275. self.assert_('a' in u'abdb')
  276. self.assert_('a' in u'bdab')
  277. self.assert_('a' in u'bdaba')
  278. self.assert_('a' in u'bdba')
  279. self.assert_('a' in u'bdba')
  280. self.assert_(u'a' in u'bdba')
  281. self.assert_(u'a' not in u'bdb')
  282. self.assert_(u'a' not in 'bdb')
  283. self.assert_(u'a' in 'bdba')
  284. self.assert_(u'a' in ('a',1,None))
  285. self.assert_(u'a' in (1,None,'a'))
  286. self.assert_(u'a' in (1,None,u'a'))
  287. self.assert_('a' in ('a',1,None))
  288. self.assert_('a' in (1,None,'a'))
  289. self.assert_('a' in (1,None,u'a'))
  290. self.assert_('a' not in ('x',1,u'y'))
  291. self.assert_('a' not in ('x',1,None))
  292. self.assert_(u'abcd' not in u'abcxxxx')
  293. self.assert_(u'ab' in u'abcd')
  294. self.assert_('ab' in u'abc')
  295. self.assert_(u'ab' in 'abc')
  296. self.assert_(u'ab' in (1,None,u'ab'))
  297. self.assert_(u'' in u'abc')
  298. self.assert_('' in u'abc')
  299. # If the following fails either
  300. # the contains operator does not propagate UnicodeErrors or
  301. # someone has changed the default encoding
  302. self.assertRaises(UnicodeError, 'g\xe2teau'.__contains__, u'\xe2')
  303. self.assert_(u'' in '')
  304. self.assert_('' in u'')
  305. self.assert_(u'' in u'')
  306. self.assert_(u'' in 'abc')
  307. self.assert_('' in u'abc')
  308. self.assert_(u'' in u'abc')
  309. self.assert_(u'\0' not in 'abc')
  310. self.assert_('\0' not in u'abc')
  311. self.assert_(u'\0' not in u'abc')
  312. self.assert_(u'\0' in '\0abc')
  313. self.assert_('\0' in u'\0abc')
  314. self.assert_(u'\0' in u'\0abc')
  315. self.assert_(u'\0' in 'abc\0')
  316. self.assert_('\0' in u'abc\0')
  317. self.assert_(u'\0' in u'abc\0')
  318. self.assert_(u'a' in '\0abc')
  319. self.assert_('a' in u'\0abc')
  320. self.assert_(u'a' in u'\0abc')
  321. self.assert_(u'asdf' in 'asdf')
  322. self.assert_('asdf' in u'asdf')
  323. self.assert_(u'asdf' in u'asdf')
  324. self.assert_(u'asdf' not in 'asd')
  325. self.assert_('asdf' not in u'asd')
  326. self.assert_(u'asdf' not in u'asd')
  327. self.assert_(u'asdf' not in '')
  328. self.assert_('asdf' not in u'')
  329. self.assert_(u'asdf' not in u'')
  330. self.assertRaises(TypeError, u"abc".__contains__)
  331. def test_formatting(self):
  332. string_tests.MixinStrUnicodeUserStringTest.test_formatting(self)
  333. # Testing Unicode formatting strings...
  334. self.assertEqual(u"%s, %s" % (u"abc", "abc"), u'abc, abc')
  335. self.assertEqual(u"%s, %s, %i, %f, %5.2f" % (u"abc", "abc", 1, 2, 3), u'abc, abc, 1, 2.000000, 3.00')
  336. self.assertEqual(u"%s, %s, %i, %f, %5.2f" % (u"abc", "abc", 1, -2, 3), u'abc, abc, 1, -2.000000, 3.00')
  337. self.assertEqual(u"%s, %s, %i, %f, %5.2f" % (u"abc", "abc", -1, -2, 3.5), u'abc, abc, -1, -2.000000, 3.50')
  338. self.assertEqual(u"%s, %s, %i, %f, %5.2f" % (u"abc", "abc", -1, -2, 3.57), u'abc, abc, -1, -2.000000, 3.57')
  339. self.assertEqual(u"%s, %s, %i, %f, %5.2f" % (u"abc", "abc", -1, -2, 1003.57), u'abc, abc, -1, -2.000000, 1003.57')
  340. if not sys.platform.startswith('java'):
  341. self.assertEqual(u"%r, %r" % (u"abc", "abc"), u"u'abc', 'abc'")
  342. self.assertEqual(u"%(x)s, %(y)s" % {'x':u"abc", 'y':"def"}, u'abc, def')
  343. self.assertEqual(u"%(x)s, %(\xfc)s" % {'x':u"abc", u'\xfc':"def"}, u'abc, def')
  344. self.assertEqual(u'%c' % 0x1234, u'\u1234')
  345. self.assertRaises(OverflowError, u"%c".__mod__, (sys.maxunicode+1,))
  346. # formatting jobs delegated from the string implementation:
  347. self.assertEqual('...%(foo)s...' % {'foo':u"abc"}, u'...abc...')
  348. self.assertEqual('...%(foo)s...' % {'foo':"abc"}, '...abc...')
  349. self.assertEqual('...%(foo)s...' % {u'foo':"abc"}, '...abc...')
  350. self.assertEqual('...%(foo)s...' % {u'foo':u"abc"}, u'...abc...')
  351. self.assertEqual('...%(foo)s...' % {u'foo':u"abc",'def':123}, u'...abc...')
  352. self.assertEqual('...%(foo)s...' % {u'foo':u"abc",u'def':123}, u'...abc...')
  353. self.assertEqual('...%s...%s...%s...%s...' % (1,2,3,u"abc"), u'...1...2...3...abc...')
  354. self.assertEqual('...%%...%%s...%s...%s...%s...%s...' % (1,2,3,u"abc"), u'...%...%s...1...2...3...abc...')
  355. self.assertEqual('...%s...' % u"abc", u'...abc...')
  356. self.assertEqual('%*s' % (5,u'abc',), u' abc')
  357. self.assertEqual('%*s' % (-5,u'abc',), u'abc ')
  358. self.assertEqual('%*.*s' % (5,2,u'abc',), u' ab')
  359. self.assertEqual('%*.*s' % (5,3,u'abc',), u' abc')
  360. self.assertEqual('%i %*.*s' % (10, 5,3,u'abc',), u'10 abc')
  361. self.assertEqual('%i%s %*.*s' % (10, 3, 5, 3, u'abc',), u'103 abc')
  362. self.assertEqual('%c' % u'a', u'a')
  363. class Wrapper:
  364. def __str__(self):
  365. return u'\u1234'
  366. self.assertEqual('%s' % Wrapper(), u'\u1234')
  367. @test_support.run_with_locale('LC_ALL', 'de_DE', 'fr_FR')
  368. def test_format_float(self):
  369. # should not format with a comma, but always with C locale
  370. self.assertEqual(u'1.0', u'%.1f' % 1.0)
  371. def test_constructor(self):
  372. # unicode(obj) tests (this maps to PyObject_Unicode() at C level)
  373. self.assertEqual(
  374. unicode(u'unicode remains unicode'),
  375. u'unicode remains unicode'
  376. )
  377. class UnicodeSubclass(unicode):
  378. pass
  379. self.assertEqual(
  380. unicode(UnicodeSubclass('unicode subclass becomes unicode')),
  381. u'unicode subclass becomes unicode'
  382. )
  383. self.assertEqual(
  384. unicode('strings are converted to unicode'),
  385. u'strings are converted to unicode'
  386. )
  387. class UnicodeCompat:
  388. def __init__(self, x):
  389. self.x = x
  390. def __unicode__(self):
  391. return self.x
  392. self.assertEqual(
  393. unicode(UnicodeCompat('__unicode__ compatible objects are recognized')),
  394. u'__unicode__ compatible objects are recognized')
  395. class StringCompat:
  396. def __init__(self, x):
  397. self.x = x
  398. def __str__(self):
  399. return self.x
  400. self.assertEqual(
  401. unicode(StringCompat('__str__ compatible objects are recognized')),
  402. u'__str__ compatible objects are recognized'
  403. )
  404. # unicode(obj) is compatible to str():
  405. o = StringCompat('unicode(obj) is compatible to str()')
  406. self.assertEqual(unicode(o), u'unicode(obj) is compatible to str()')
  407. self.assertEqual(str(o), 'unicode(obj) is compatible to str()')
  408. # %-formatting and .__unicode__()
  409. self.assertEqual(u'%s' %
  410. UnicodeCompat(u"u'%s' % obj uses obj.__unicode__()"),
  411. u"u'%s' % obj uses obj.__unicode__()")
  412. self.assertEqual(u'%s' %
  413. UnicodeCompat(u"u'%s' % obj falls back to obj.__str__()"),
  414. u"u'%s' % obj falls back to obj.__str__()")
  415. for obj in (123, 123.45, 123L):
  416. self.assertEqual(unicode(obj), unicode(str(obj)))
  417. # unicode(obj, encoding, error) tests (this maps to
  418. # PyUnicode_FromEncodedObject() at C level)
  419. if not sys.platform.startswith('java'):
  420. self.assertRaises(
  421. TypeError,
  422. unicode,
  423. u'decoding unicode is not supported',
  424. 'utf-8',
  425. 'strict'
  426. )
  427. self.assertEqual(
  428. unicode('strings are decoded to unicode', 'utf-8', 'strict'),
  429. u'strings are decoded to unicode'
  430. )
  431. if not sys.platform.startswith('java'):
  432. self.assertEqual(
  433. unicode(
  434. buffer('character buffers are decoded to unicode'),
  435. 'utf-8',
  436. 'strict'
  437. ),
  438. u'character buffers are decoded to unicode'
  439. )
  440. self.assertRaises(TypeError, unicode, 42, 42, 42)
  441. def test_codecs_utf7(self):
  442. utfTests = [
  443. (u'A\u2262\u0391.', 'A+ImIDkQ.'), # RFC2152 example
  444. (u'Hi Mom -\u263a-!', 'Hi Mom -+Jjo--!'), # RFC2152 example
  445. (u'\u65E5\u672C\u8A9E', '+ZeVnLIqe-'), # RFC2152 example
  446. (u'Item 3 is \u00a31.', 'Item 3 is +AKM-1.'), # RFC2152 example
  447. (u'+', '+-'),
  448. (u'+-', '+--'),
  449. (u'+?', '+-?'),
  450. (u'\?', '+AFw?'),
  451. (u'+?', '+-?'),
  452. (ur'\\?', '+AFwAXA?'),
  453. (ur'\\\?', '+AFwAXABc?'),
  454. (ur'++--', '+-+---')
  455. ]
  456. for (x, y) in utfTests:
  457. self.assertEqual(x.encode('utf-7'), y)
  458. # surrogates not supported
  459. self.assertRaises(UnicodeError, unicode, '+3ADYAA-', 'utf-7')
  460. self.assertEqual(unicode('+3ADYAA-', 'utf-7', 'replace'), u'\ufffd')
  461. # Issue #2242: crash on some Windows/MSVC versions
  462. self.assertRaises(UnicodeDecodeError, '+\xc1'.decode, 'utf-7')
  463. def test_codecs_utf8(self):
  464. self.assertEqual(u''.encode('utf-8'), '')
  465. self.assertEqual(u'\u20ac'.encode('utf-8'), '\xe2\x82\xac')
  466. self.assertEqual(u'\ud800\udc02'.encode('utf-8'), '\xf0\x90\x80\x82')
  467. self.assertEqual(u'\ud84d\udc56'.encode('utf-8'), '\xf0\xa3\x91\x96')
  468. self.assertEqual(u'\ud800'.encode('utf-8'), '\xed\xa0\x80')
  469. self.assertEqual(u'\udc00'.encode('utf-8'), '\xed\xb0\x80')
  470. self.assertEqual(
  471. (u'\ud800\udc02'*1000).encode('utf-8'),
  472. '\xf0\x90\x80\x82'*1000
  473. )
  474. self.assertEqual(
  475. u'\u6b63\u78ba\u306b\u8a00\u3046\u3068\u7ffb\u8a33\u306f'
  476. u'\u3055\u308c\u3066\u3044\u307e\u305b\u3093\u3002\u4e00'
  477. u'\u90e8\u306f\u30c9\u30a4\u30c4\u8a9e\u3067\u3059\u304c'
  478. u'\u3001\u3042\u3068\u306f\u3067\u305f\u3089\u3081\u3067'
  479. u'\u3059\u3002\u5b9f\u969b\u306b\u306f\u300cWenn ist das'
  480. u' Nunstuck git und'.encode('utf-8'),
  481. '\xe6\xad\xa3\xe7\xa2\xba\xe3\x81\xab\xe8\xa8\x80\xe3\x81'
  482. '\x86\xe3\x81\xa8\xe7\xbf\xbb\xe8\xa8\xb3\xe3\x81\xaf\xe3'
  483. '\x81\x95\xe3\x82\x8c\xe3\x81\xa6\xe3\x81\x84\xe3\x81\xbe'
  484. '\xe3\x81\x9b\xe3\x82\x93\xe3\x80\x82\xe4\xb8\x80\xe9\x83'
  485. '\xa8\xe3\x81\xaf\xe3\x83\x89\xe3\x82\xa4\xe3\x83\x84\xe8'
  486. '\xaa\x9e\xe3\x81\xa7\xe3\x81\x99\xe3\x81\x8c\xe3\x80\x81'
  487. '\xe3\x81\x82\xe3\x81\xa8\xe3\x81\xaf\xe3\x81\xa7\xe3\x81'
  488. '\x9f\xe3\x82\x89\xe3\x82\x81\xe3\x81\xa7\xe3\x81\x99\xe3'
  489. '\x80\x82\xe5\xae\x9f\xe9\x9a\x9b\xe3\x81\xab\xe3\x81\xaf'
  490. '\xe3\x80\x8cWenn ist das Nunstuck git und'
  491. )
  492. # UTF-8 specific decoding tests
  493. self.assertEqual(unicode('\xf0\xa3\x91\x96', 'utf-8'), u'\U00023456' )
  494. self.assertEqual(unicode('\xf0\x90\x80\x82', 'utf-8'), u'\U00010002' )
  495. self.assertEqual(unicode('\xe2\x82\xac', 'utf-8'), u'\u20ac' )
  496. # Other possible utf-8 test cases:
  497. # * strict decoding testing for all of the
  498. # UTF8_ERROR cases in PyUnicode_DecodeUTF8
  499. def test_codecs_idna(self):
  500. # Test whether trailing dot is preserved
  501. self.assertEqual(u"www.python.org.".encode("idna"), "www.python.org.")
  502. def test_codecs_errors(self):
  503. # Error handling (encoding)
  504. self.assertRaises(UnicodeError, u'Andr\202 x'.encode, 'ascii')
  505. self.assertRaises(UnicodeError, u'Andr\202 x'.encode, 'ascii','strict')
  506. self.assertEqual(u'Andr\202 x'.encode('ascii','ignore'), "Andr x")
  507. self.assertEqual(u'Andr\202 x'.encode('ascii','replace'), "Andr? x")
  508. # Error handling (decoding)
  509. self.assertRaises(UnicodeError, unicode, 'Andr\202 x', 'ascii')
  510. self.assertRaises(UnicodeError, unicode, 'Andr\202 x', 'ascii','strict')
  511. self.assertEqual(unicode('Andr\202 x','ascii','ignore'), u"Andr x")
  512. self.assertEqual(unicode('Andr\202 x','ascii','replace'), u'Andr\uFFFD x')
  513. # Error handling (unknown character names)
  514. self.assertEqual("\\N{foo}xx".decode("unicode-escape", "ignore"), u"xx")
  515. # Error handling (truncated escape sequence)
  516. self.assertRaises(UnicodeError, "\\".decode, "unicode-escape")
  517. self.assertRaises(TypeError, "hello".decode, "test.unicode1")
  518. self.assertRaises(TypeError, unicode, "hello", "test.unicode2")
  519. self.assertRaises(TypeError, u"hello".encode, "test.unicode1")
  520. self.assertRaises(TypeError, u"hello".encode, "test.unicode2")
  521. # executes PyUnicode_Encode()
  522. import imp
  523. self.assertRaises(
  524. ImportError,
  525. imp.find_module,
  526. "non-existing module",
  527. [u"non-existing dir"]
  528. )
  529. # Error handling (wrong arguments)
  530. self.assertRaises(TypeError, u"hello".encode, 42, 42, 42)
  531. # Error handling (PyUnicode_EncodeDecimal())
  532. self.assertRaises(UnicodeError, int, u"\u0200")
  533. def test_codecs(self):
  534. # Encoding
  535. self.assertEqual(u'hello'.encode('ascii'), 'hello')
  536. self.assertEqual(u'hello'.encode('utf-7'), 'hello')
  537. self.assertEqual(u'hello'.encode('utf-8'), 'hello')
  538. self.assertEqual(u'hello'.encode('utf8'), 'hello')
  539. self.assertEqual(u'hello'.encode('utf-16-le'), 'h\000e\000l\000l\000o\000')
  540. self.assertEqual(u'hello'.encode('utf-16-be'), '\000h\000e\000l\000l\000o')
  541. self.assertEqual(u'hello'.encode('latin-1'), 'hello')
  542. # Roundtrip safety for BMP (just the first 1024 chars)
  543. for c in xrange(1024):
  544. u = unichr(c)
  545. for encoding in ('utf-7', 'utf-8', 'utf-16', 'utf-16-le',
  546. 'utf-16-be', 'raw_unicode_escape',
  547. 'unicode_escape', 'unicode_internal'):
  548. self.assertEqual(unicode(u.encode(encoding),encoding), u)
  549. # Roundtrip safety for BMP (just the first 256 chars)
  550. for c in xrange(256):
  551. u = unichr(c)
  552. for encoding in ('latin-1',):
  553. self.assertEqual(unicode(u.encode(encoding),encoding), u)
  554. # Roundtrip safety for BMP (just the first 128 chars)
  555. for c in xrange(128):
  556. u = unichr(c)
  557. for encoding in ('ascii',):
  558. self.assertEqual(unicode(u.encode(encoding),encoding), u)
  559. # Roundtrip safety for non-BMP (just a few chars)
  560. u = u'\U00010001\U00020002\U00030003\U00040004\U00050005'
  561. for encoding in ('utf-8', 'utf-16', 'utf-16-le', 'utf-16-be',
  562. #'raw_unicode_escape',
  563. 'unicode_escape', 'unicode_internal'):
  564. self.assertEqual(unicode(u.encode(encoding),encoding), u)
  565. # UTF-8 must be roundtrip safe for all UCS-2 code points
  566. # This excludes surrogates: in the full range, there would be
  567. # a surrogate pair (\udbff\udc00), which gets converted back
  568. # to a non-BMP character (\U0010fc00)
  569. u = u''.join(map(unichr, range(0,0xd800)+range(0xe000,0x10000)))
  570. for encoding in ('utf-8',):
  571. self.assertEqual(unicode(u.encode(encoding),encoding), u)
  572. def test_codecs_charmap(self):
  573. # 0-127
  574. s = ''.join(map(chr, xrange(128)))
  575. for encoding in (
  576. 'cp037', 'cp1026',
  577. 'cp437', 'cp500', 'cp737', 'cp775', 'cp850',
  578. 'cp852', 'cp855', 'cp860', 'cp861', 'cp862',
  579. 'cp863', 'cp865', 'cp866',
  580. 'iso8859_10', 'iso8859_13', 'iso8859_14', 'iso8859_15',
  581. 'iso8859_2', 'iso8859_3', 'iso8859_4', 'iso8859_5', 'iso8859_6',
  582. 'iso8859_7', 'iso8859_9', 'koi8_r', 'latin_1',
  583. 'mac_cyrillic', 'mac_latin2',
  584. 'cp1250', 'cp1251', 'cp1252', 'cp1253', 'cp1254', 'cp1255',
  585. 'cp1256', 'cp1257', 'cp1258',
  586. 'cp856', 'cp857', 'cp864', 'cp869', 'cp874',
  587. 'mac_greek', 'mac_iceland','mac_roman', 'mac_turkish',
  588. 'cp1006', 'iso8859_8',
  589. ### These have undefined mappings:
  590. #'cp424',
  591. ### These fail the round-trip:
  592. #'cp875'
  593. ):
  594. self.assertEqual(unicode(s, encoding).encode(encoding), s)
  595. # 128-255
  596. s = ''.join(map(chr, xrange(128, 256)))
  597. for encoding in (
  598. 'cp037', 'cp1026',
  599. 'cp437', 'cp500', 'cp737', 'cp775', 'cp850',
  600. 'cp852', 'cp855', 'cp860', 'cp861', 'cp862',
  601. 'cp863', 'cp865', 'cp866',
  602. 'iso8859_10', 'iso8859_13', 'iso8859_14', 'iso8859_15',
  603. 'iso8859_2', 'iso8859_4', 'iso8859_5',
  604. 'iso8859_9', 'koi8_r', 'latin_1',
  605. 'mac_cyrillic', 'mac_latin2',
  606. ### These have undefined mappings:
  607. #'cp1250', 'cp1251', 'cp1252', 'cp1253', 'cp1254', 'cp1255',
  608. #'cp1256', 'cp1257', 'cp1258',
  609. #'cp424', 'cp856', 'cp857', 'cp864', 'cp869', 'cp874',
  610. #'iso8859_3', 'iso8859_6', 'iso8859_7',
  611. #'mac_greek', 'mac_iceland','mac_roman', 'mac_turkish',
  612. ### These fail the round-trip:
  613. #'cp1006', 'cp875', 'iso8859_8',
  614. ):
  615. self.assertEqual(unicode(s, encoding).encode(encoding), s)
  616. def test_concatenation(self):
  617. self.assertEqual((u"abc" u"def"), u"abcdef")
  618. self.assertEqual(("abc" u"def"), u"abcdef")
  619. self.assertEqual((u"abc" "def"), u"abcdef")
  620. self.assertEqual((u"abc" u"def" "ghi"), u"abcdefghi")
  621. self.assertEqual(("abc" "def" u"ghi"), u"abcdefghi")
  622. def test_printing(self):
  623. class BitBucket:
  624. def write(self, text):
  625. pass
  626. out = BitBucket()
  627. print >>out, u'abc'
  628. print >>out, u'abc', u'def'
  629. print >>out, u'abc', 'def'
  630. print >>out, 'abc', u'def'
  631. print >>out, u'abc\n'
  632. print >>out, u'abc\n',
  633. print >>out, u'abc\n',
  634. print >>out, u'def\n'
  635. print >>out, u'def\n'
  636. def test_ucs4(self):
  637. x = u'\U00100000'
  638. y = x.encode("raw-unicode-escape").decode("raw-unicode-escape")
  639. self.assertEqual(x, y)
  640. y = r'\U00100000'
  641. x = y.decode("raw-unicode-escape").encode("raw-unicode-escape")
  642. self.assertEqual(x, y)
  643. y = r'\U00010000'
  644. x = y.decode("raw-unicode-escape").encode("raw-unicode-escape")
  645. self.assertEqual(x, y)
  646. try:
  647. '\U11111111'.decode("raw-unicode-escape")
  648. except UnicodeDecodeError as e:
  649. self.assertEqual(e.start, 0)
  650. self.assertEqual(e.end, 10)
  651. else:
  652. self.fail("Should have raised UnicodeDecodeError")
  653. def test_conversion(self):
  654. # Make sure __unicode__() works properly
  655. class Foo0:
  656. def __str__(self):
  657. return "foo"
  658. class Foo1:
  659. def __unicode__(self):
  660. return u"foo"
  661. class Foo2(object):
  662. def __unicode__(self):
  663. return u"foo"
  664. class Foo3(object):
  665. def __unicode__(self):
  666. return "foo"
  667. class Foo4(str):
  668. def __unicode__(self):
  669. return "foo"
  670. class Foo5(unicode):
  671. def __unicode__(self):
  672. return "foo"
  673. class Foo6(str):
  674. def __str__(self):
  675. return "foos"
  676. def __unicode__(self):
  677. return u"foou"
  678. class Foo7(unicode):
  679. def __str__(self):
  680. return "foos"
  681. def __unicode__(self):
  682. return u"foou"
  683. class Foo8(unicode):
  684. def __new__(cls, content=""):
  685. return unicode.__new__(cls, 2*content)
  686. def __unicode__(self):
  687. return self
  688. class Foo9(unicode):
  689. def __str__(self):
  690. return "string"
  691. def __unicode__(self):
  692. return "not unicode"
  693. self.assertEqual(unicode(Foo0()), u"foo")
  694. self.assertEqual(unicode(Foo1()), u"foo")
  695. self.assertEqual(unicode(Foo2()), u"foo")
  696. self.assertEqual(unicode(Foo3()), u"foo")
  697. self.assertEqual(unicode(Foo4("bar")), u"foo")
  698. self.assertEqual(unicode(Foo5("bar")), u"foo")
  699. self.assertEqual(unicode(Foo6("bar")), u"foou")
  700. self.assertEqual(unicode(Foo7("bar")), u"foou")
  701. self.assertEqual(unicode(Foo8("foo")), u"foofoo")
  702. self.assertEqual(str(Foo9("foo")), "string")
  703. self.assertEqual(unicode(Foo9("foo")), u"not unicode")
  704. def test_unicode_repr(self):
  705. class s1:
  706. def __repr__(self):
  707. return '\\n'
  708. class s2:
  709. def __repr__(self):
  710. return u'\\n'
  711. self.assertEqual(repr(s1()), '\\n')
  712. self.assertEqual(repr(s2()), '\\n')
  713. def test_expandtabs_overflows_gracefully(self):
  714. # This test only affects 32-bit platforms because expandtabs can only take
  715. # an int as the max value, not a 64-bit C long. If expandtabs is changed
  716. # to take a 64-bit long, this test should apply to all platforms.
  717. if sys.maxint > (1 << 32) or struct.calcsize('P') != 4:
  718. return
  719. self.assertRaises(OverflowError, u't\tt\t'.expandtabs, sys.maxint)
  720. def test__format__(self):
  721. def test(value, format, expected):
  722. # test both with and without the trailing 's'
  723. self.assertEqual(value.__format__(format), expected)
  724. self.assertEqual(value.__format__(format + u's'), expected)
  725. test(u'', u'', u'')
  726. test(u'abc', u'', u'abc')
  727. test(u'abc', u'.3', u'abc')
  728. test(u'ab', u'.3', u'ab')
  729. test(u'abcdef', u'.3', u'abc')
  730. test(u'abcdef', u'.0', u'')
  731. test(u'abc', u'3.3', u'abc')
  732. test(u'abc', u'2.3', u'abc')
  733. test(u'abc', u'2.2', u'ab')
  734. test(u'abc', u'3.2', u'ab ')
  735. test(u'result', u'x<0', u'result')
  736. test(u'result', u'x<5', u'result')
  737. test(u'result', u'x<6', u'result')
  738. test(u'result', u'x<7', u'resultx')
  739. test(u'result', u'x<8', u'resultxx')
  740. test(u'result', u' <7', u'result ')
  741. test(u'result', u'<7', u'result ')
  742. test(u'result', u'>7', u' result')
  743. test(u'result', u'>8', u' result')
  744. test(u'result', u'^8', u' result ')
  745. test(u'result', u'^9', u' result ')
  746. test(u'result', u'^10', u' result ')
  747. test(u'a', u'10000', u'a' + u' ' * 9999)
  748. test(u'', u'10000', u' ' * 10000)
  749. test(u'', u'10000000', u' ' * 10000000)
  750. # test mixing unicode and str
  751. self.assertEqual(u'abc'.__format__('s'), u'abc')
  752. self.assertEqual(u'abc'.__format__('->10s'), u'-------abc')
  753. def test_format(self):
  754. self.assertEqual(u''.format(), u'')
  755. self.assertEqual(u'a'.format(), u'a')
  756. self.assertEqual(u'ab'.format(), u'ab')
  757. self.assertEqual(u'a{{'.format(), u'a{')
  758. self.assertEqual(u'a}}'.format(), u'a}')
  759. self.assertEqual(u'{{b'.format(), u'{b')
  760. self.assertEqual(u'}}b'.format(), u'}b')
  761. self.assertEqual(u'a{{b'.format(), u'a{b')
  762. # examples from the PEP:
  763. import datetime
  764. self.assertEqual(u"My name is {0}".format(u'Fred'), u"My name is Fred")
  765. self.assertEqual(u"My name is {0[name]}".format(dict(name=u'Fred')),
  766. u"My name is Fred")
  767. self.assertEqual(u"My name is {0} :-{{}}".format(u'Fred'),
  768. u"My name is Fred :-{}")
  769. # datetime.__format__ doesn't work with unicode
  770. #d = datetime.date(2007, 8, 18)
  771. #self.assertEqual("The year is {0.year}".format(d),
  772. # "The year is 2007")
  773. # classes we'll use for testing
  774. class C:
  775. def __init__(self, x=100):
  776. self._x = x
  777. def __format__(self, spec):
  778. return spec
  779. class D:
  780. def __init__(self, x):
  781. self.x = x
  782. def __format__(self, spec):
  783. return str(self.x)
  784. # class with __str__, but no __format__
  785. class E:
  786. def __init__(self, x):
  787. self.x = x
  788. def __str__(self):
  789. return u'E(' + self.x + u')'
  790. # class with __repr__, but no __format__ or __str__
  791. class F:
  792. def __init__(self, x):
  793. self.x = x
  794. def __repr__(self):
  795. return u'F(' + self.x + u')'
  796. # class with __format__ that forwards to string, for some format_spec's
  797. class G:
  798. def __init__(self, x):
  799. self.x = x
  800. def __str__(self):
  801. return u"string is " + self.x
  802. def __format__(self, format_spec):
  803. if format_spec == 'd':
  804. return u'G(' + self.x + u')'
  805. return object.__format__(self, format_spec)
  806. # class that returns a bad type from __format__
  807. class H:
  808. def __format__(self, format_spec):
  809. return 1.0
  810. class I(datetime.date):
  811. def __format__(self, format_spec):
  812. return self.strftime(format_spec)
  813. class J(int):
  814. def __format__(self, format_spec):
  815. return int.__format__(self * 2, format_spec)
  816. self.assertEqual(u''.format(), u'')
  817. self.assertEqual(u'abc'.format(), u'abc')
  818. self.assertEqual(u'{0}'.format(u'abc'), u'abc')
  819. self.assertEqual(u'{0:}'.format(u'abc'), u'abc')
  820. self.assertEqual(u'X{0}'.format(u'abc'), u'Xabc')
  821. self.assertEqual(u'{0}X'.format(u'abc'), u'abcX')
  822. self.assertEqual(u'X{0}Y'.format(u'abc'), u'XabcY')
  823. self.assertEqual(u'{1}'.format(1, u'abc'), u'abc')
  824. self.assertEqual(u'X{1}'.format(1, u'abc'), u'Xabc')
  825. self.assertEqual(u'{1}X'.format(1, u'abc'), u'abcX')
  826. self.assertEqual(u'X{1}Y'.format(1, u'abc'), u'XabcY')
  827. self.assertEqual(u'{0}'.format(-15), u'-15')
  828. self.assertEqual(u'{0}{1}'.format(-15, u'abc'), u'-15abc')
  829. self.assertEqual(u'{0}X{1}'.format(-15, u'abc'), u'-15Xabc')
  830. self.assertEqual(u'{{'.format(), u'{')
  831. self.assertEqual(u'}}'.format(), u'}')
  832. self.assertEqual(u'{{}}'.format(), u'{}')
  833. self.assertEqual(u'{{x}}'.format(), u'{x}')
  834. self.assertEqual(u'{{{0}}}'.format(123), u'{123}')
  835. self.assertEqual(u'{{{{0}}}}'.format(), u'{{0}}')
  836. self.assertEqual(u'}}{{'.format(), u'}{')
  837. self.assertEqual(u'}}x{{'.format(), u'}x{')
  838. # weird field names
  839. self.assertEqual(u"{0[foo-bar]}".format({u'foo-bar':u'baz'}), u'baz')
  840. self.assertEqual(u"{0[foo bar]}".format({u'foo bar':u'baz'}), u'baz')
  841. self.assertEqual(u"{0[ ]}".format({u' ':3}), u'3')
  842. self.assertEqual(u'{foo._x}'.format(foo=C(20)), u'20')
  843. self.assertEqual(u'{1}{0}'.format(D(10), D(20)), u'2010')
  844. self.assertEqual(u'{0._x.x}'.format(C(D(u'abc'))), u'abc')
  845. self.assertEqual(u'{0[0]}'.format([u'abc', u'def']), u'abc')
  846. self.assertEqual(u'{0[1]}'.format([u'abc', u'def']), u'def')
  847. self.assertEqual(u'{0[1][0]}'.format([u'abc', [u'def']]), u'def')
  848. self.assertEqual(u'{0[1][0].x}'.format(['abc', [D(u'def')]]), u'def')
  849. # strings
  850. self.assertEqual(u'{0:.3s}'.format(u'abc'), u'abc')
  851. self.assertEqual(u'{0:.3s}'.format(u'ab'), u'ab')
  852. self.assertEqual(u'{0:.3s}'.format(u'abcdef'), u'abc')
  853. self.assertEqual(u'{0:.0s}'.format(u'abcdef'), u'')
  854. self.assertEqual(u'{0:3.3s}'.format(u'abc'), u'abc')
  855. self.assertEqual(u'{0:2.3s}'.format(u'abc'), u'abc')
  856. self.assertEqual(u'{0:2.2s}'.format(u'abc'), u'ab')
  857. self.assertEqual(u'{0:3.2s}'.format(u'abc'), u'ab ')
  858. self.assertEqual(u'{0:x<0s}'.format(u'result'), u'result')
  859. self.assertEqual(u'{0:x<5s}'.format(u'result'), u'result')
  860. self.assertEqual(u'{0:x<6s}'.format(u'result'), u'result')
  861. self.assertEqual(u'{0:x<7s}'.format(u'result'), u'resultx')
  862. self.assertEqual(u'{0:x<8s}'.format(u'result'), u'resultxx')
  863. self.assertEqual(u'{0: <7s}'.format(u'result'), u'result ')
  864. self.assertEqual(u'{0:<7s}'.format(u'result'), u'result ')
  865. self.assertEqual(u'{0:>7s}'.format(u'result'), u' result')
  866. self.assertEqual(u'{0:>8s}'.format(u'result'), u' result')
  867. self.assertEqual(u'{0:^8s}'.format(u'result'), u' result ')
  868. self.assertEqual(u'{0:^9s}'.format(u'result'), u' result ')
  869. self.assertEqual(u'{0:^10s}'.format(u'result'), u' result ')
  870. self.assertEqual(u'{0:10000}'.format(u'a'), u'a' + u' ' * 9999)
  871. self.assertEqual(u'{0:10000}'.format(u''), u' ' * 10000)
  872. self.assertEqual(u'{0:10000000}'.format(u''), u' ' * 10000000)
  873. # format specifiers for user defined type
  874. self.assertEqual(u'{0:abc}'.format(C()), u'abc')
  875. # !r and !s coersions
  876. self.assertEqual(u'{0!s}'.format(u'Hello'), u'Hello')
  877. self.assertEqual(u'{0!s:}'.format(u'Hello'), u'Hello')
  878. self.assertEqual(u'{0!s:15}'.format(u'Hello'), u'Hello ')
  879. self.assertEqual(u'{0!s:15s}'.format(u'Hello'), u'Hello ')
  880. self.assertEqual(u'{0!r}'.format(u'Hello'), u"u'Hello'")
  881. self.assertEqual(u'{0!r:}'.format(u'Hello'), u"u'Hello'")
  882. self.assertEqual(u'{0!r}'.format(F(u'Hello')), u'F(Hello)')
  883. # test fallback to object.__format__
  884. self.assertEqual(u'{0}'.format({}), u'{}')
  885. self.assertEqual(u'{0}'.format([]), u'[]')
  886. self.assertEqual(u'{0}'.format([1]), u'[1]')
  887. self.assertEqual(u'{0}'.format(E(u'data')), u'E(data)')
  888. self.assertEqual(u'{0:^10}'.format(E(u'data')), u' E(data) ')
  889. self.assertEqual(u'{0:^10s}'.format(E(u'data')), u' E(data) ')
  890. self.assertEqual(u'{0:d}'.format(G(u'data')), u'G(data)')
  891. self.assertEqual(u'{0:>15s}'.format(G(u'data')), u' string is data')
  892. self.assertEqual(u'{0!s}'.format(G(u'data')), u'string is data')
  893. self.assertEqual(u"{0:date: %Y-%m-%d}".format(I(year=2007,
  894. month=8,
  895. day=27)),
  896. u"date: 2007-08-27")
  897. # test deriving from a builtin type and overriding __format__
  898. self.assertEqual(u"{0}".format(J(10)), u"20")
  899. # string format specifiers
  900. self.assertEqual(u'{0:}'.format('a'), u'a')
  901. # computed format specifiers
  902. self.assertEqual(u"{0:.{1}}".format(u'hello world', 5), u'hello')
  903. self.assertEqual(u"{0:.{1}s}".format(u'hello world', 5), u'hello')
  904. self.assertEqual(u"{0:.{precision}s}".format('hello world', precision=5), u'hello')
  905. self.assertEqual(u"{0:{width}.{precision}s}".format('hello world', width=10, precision=5), u'hello ')
  906. self.assertEqual(u"{0:{width}.{precision}s}".format('hello world', width='10', precision='5'), u'hello ')
  907. # test various errors
  908. self.assertRaises(ValueError, u'{'.format)
  909. self.assertRaises(ValueError, u'}'.format)
  910. self.assertRaises(ValueError, u'a{'.format)
  911. self.assertRaises(ValueError, u'a}'.format)
  912. self.assertRaises(ValueError, u'{a'.format)
  913. self.assertRaises(ValueError, u'}a'.format)
  914. self.assertRaises(IndexError, u'{0}'.format)
  915. self.assertRaises(IndexError, u'{1}'.format, u'abc')
  916. self.assertRaises(KeyError, u'{x}'.format)
  917. self.assertRaises(ValueError, u"}{".format)
  918. self.assertRaises(ValueError, u"{".format)
  919. self.assertRaises(ValueError, u"}".format)
  920. self.assertRaises(ValueError, u"abc{0:{}".format)
  921. self.assertRaises(ValueError, u"{0".format)
  922. self.assertRaises(IndexError, u"{0.}".format)
  923. self.assertRaises(ValueError, u"{0.}".format, 0)
  924. self.assertRaises(IndexError, u"{0[}".format)
  925. self.assertRaises(ValueError, u"{0[}".format, [])
  926. self.assertRaises(KeyError, u"{0]}".format)
  927. self.assertRaises(ValueError, u"{0.[]}".format, 0)
  928. self.assertRaises(ValueError, u"{0..foo}".format, 0)
  929. self.assertRaises(ValueError, u"{0[0}".format, 0)
  930. self.assertRaises(ValueError, u"{0[0:foo}".format, 0)
  931. self.assertRaises(KeyError, u"{c]}".format)
  932. self.assertRaises(ValueError, u"{{ {{{0}}".format, 0)
  933. self.assertRaises(ValueError, u"{0}}".format, 0)
  934. self.assertRaises(KeyError, u"{foo}".format, bar=3)
  935. self.assertRaises(ValueError, u"{0!x}".format, 3)
  936. self.assertRaises(ValueError, u"{0!}".format, 0)
  937. self.assertRaises(ValueError, u"{0!rs}".format, 0)
  938. self.assertRaises(ValueError, u"{!}".format)
  939. self.assertRaises(ValueError, u"{:}".format)
  940. self.assertRaises(ValueError, u"{:s}".format)
  941. self.assertRaises(ValueError, u"{}".format)
  942. # issue 6089
  943. self.assertRaises(ValueError, u"{0[0]x}".format, [None])
  944. self.assertRaises(ValueError, u"{0[0](10)}".format, [None])
  945. # can't have a replacement on the field name portion
  946. self.assertRaises(TypeError, u'{0[{1}]}'.format, u'abcdefg', 4)
  947. # exceed maximum recursion depth
  948. self.assertRaises(ValueError, u"{0:{1:{2}}}".format, u'abc', u's', u'')
  949. self.assertRaises(ValueError, u"{0:{1:{2:{3:{4:{5:{6}}}}}}}".format,
  950. 0, 1, 2, 3, 4, 5, 6, 7)
  951. # string format spec errors
  952. self.assertRaises(ValueError, u"{0:-s}".format, u'')
  953. self.assertRaises(ValueError, format, u"", u"-")
  954. self.assertRaises(ValueError, u"{0:=s}".format, u'')
  955. # test combining string and unicode
  956. self.assertEqual(u"foo{0}".format('bar'), u'foobar')
  957. # This will try to convert the argument from unicode to str, which
  958. # will succeed
  959. self.assertEqual("foo{0}".format(u'bar'), 'foobar')
  960. # This will try to convert the argument from unicode to str, which
  961. # will fail
  962. self.assertRaises(UnicodeEncodeError, "foo{0}".format, u'\u1000bar')
  963. def test_raiseMemError(self):
  964. # Ensure that the freelist contains a consistent object, even
  965. # when a string allocation fails with a MemoryError.
  966. # This used to crash the interpreter,
  967. # or leak references when the number was smaller.
  968. charwidth = 4 if sys.maxunicode >= 0x10000 else 2
  969. # Note: sys.maxsize is half of the actual max allocation because of
  970. # the signedness of Py_ssize_t.
  971. alloc = lambda: u"a" * (sys.maxsize // charwidth * 2)
  972. self.assertRaises(MemoryError, alloc)
  973. self.assertRaises(MemoryError, alloc)
  974. def test_main():
  975. test_support.run_unittest(__name__)
  976. if __name__ == "__main__":
  977. test_main()