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

/lib-python/2.7/test/test_unicode.py

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