test_unicode.py 82 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869
  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
  6. import struct
  7. import codecs
  8. import unittest
  9. from test import test_support, string_tests
  10. # decorator to skip tests on narrow builds
  11. requires_wide_build = unittest.skipIf(sys.maxunicode == 65535,
  12. 'requires wide build')
  13. # Error handling (bad decoder return)
  14. def search_function(encoding):
  15. def decode1(input, errors="strict"):
  16. return 42 # not a tuple
  17. def encode1(input, errors="strict"):
  18. return 42 # not a tuple
  19. def encode2(input, errors="strict"):
  20. return (42, 42) # no unicode
  21. def decode2(input, errors="strict"):
  22. return (42, 42) # no unicode
  23. if encoding=="test.unicode1":
  24. return (encode1, decode1, None, None)
  25. elif encoding=="test.unicode2":
  26. return (encode2, decode2, None, None)
  27. else:
  28. return None
  29. codecs.register(search_function)
  30. class UnicodeSubclass(unicode):
  31. pass
  32. class UnicodeTest(
  33. string_tests.CommonTest,
  34. string_tests.MixinStrUnicodeUserStringTest,
  35. string_tests.MixinStrUnicodeTest,
  36. ):
  37. type2test = unicode
  38. def assertEqual(self, first, second, msg=None):
  39. # strict assertEqual method: reject implicit bytes/unicode equality
  40. super(UnicodeTest, self).assertEqual(first, second, msg)
  41. if isinstance(first, unicode) or isinstance(second, unicode):
  42. self.assertIsInstance(first, unicode)
  43. self.assertIsInstance(second, unicode)
  44. elif isinstance(first, str) or isinstance(second, str):
  45. self.assertIsInstance(first, str)
  46. self.assertIsInstance(second, str)
  47. def checkequalnofix(self, result, object, methodname, *args):
  48. method = getattr(object, methodname)
  49. realresult = method(*args)
  50. self.assertEqual(realresult, result)
  51. self.assertTrue(type(realresult) is type(result))
  52. # if the original is returned make sure that
  53. # this doesn't happen with subclasses
  54. if realresult is object:
  55. class usub(unicode):
  56. def __repr__(self):
  57. return 'usub(%r)' % unicode.__repr__(self)
  58. object = usub(object)
  59. method = getattr(object, methodname)
  60. realresult = method(*args)
  61. self.assertEqual(realresult, result)
  62. self.assertTrue(object is not realresult)
  63. def test_literals(self):
  64. self.assertEqual(u'\xff', u'\u00ff')
  65. self.assertEqual(u'\uffff', u'\U0000ffff')
  66. self.assertRaises(SyntaxError, eval, 'u\'\\Ufffffffe\'')
  67. self.assertRaises(SyntaxError, eval, 'u\'\\Uffffffff\'')
  68. self.assertRaises(SyntaxError, eval, 'u\'\\U%08x\'' % 0x110000)
  69. def test_repr(self):
  70. if not sys.platform.startswith('java'):
  71. # Test basic sanity of repr()
  72. self.assertEqual(repr(u'abc'), "u'abc'")
  73. self.assertEqual(repr(u'ab\\c'), "u'ab\\\\c'")
  74. self.assertEqual(repr(u'ab\\'), "u'ab\\\\'")
  75. self.assertEqual(repr(u'\\c'), "u'\\\\c'")
  76. self.assertEqual(repr(u'\\'), "u'\\\\'")
  77. self.assertEqual(repr(u'\n'), "u'\\n'")
  78. self.assertEqual(repr(u'\r'), "u'\\r'")
  79. self.assertEqual(repr(u'\t'), "u'\\t'")
  80. self.assertEqual(repr(u'\b'), "u'\\x08'")
  81. self.assertEqual(repr(u"'\""), """u'\\'"'""")
  82. self.assertEqual(repr(u"'\""), """u'\\'"'""")
  83. self.assertEqual(repr(u"'"), '''u"'"''')
  84. self.assertEqual(repr(u'"'), """u'"'""")
  85. latin1repr = (
  86. "u'\\x00\\x01\\x02\\x03\\x04\\x05\\x06\\x07\\x08\\t\\n\\x0b\\x0c\\r"
  87. "\\x0e\\x0f\\x10\\x11\\x12\\x13\\x14\\x15\\x16\\x17\\x18\\x19\\x1a"
  88. "\\x1b\\x1c\\x1d\\x1e\\x1f !\"#$%&\\'()*+,-./0123456789:;<=>?@ABCDEFGHI"
  89. "JKLMNOPQRSTUVWXYZ[\\\\]^_`abcdefghijklmnopqrstuvwxyz{|}~\\x7f"
  90. "\\x80\\x81\\x82\\x83\\x84\\x85\\x86\\x87\\x88\\x89\\x8a\\x8b\\x8c\\x8d"
  91. "\\x8e\\x8f\\x90\\x91\\x92\\x93\\x94\\x95\\x96\\x97\\x98\\x99\\x9a\\x9b"
  92. "\\x9c\\x9d\\x9e\\x9f\\xa0\\xa1\\xa2\\xa3\\xa4\\xa5\\xa6\\xa7\\xa8\\xa9"
  93. "\\xaa\\xab\\xac\\xad\\xae\\xaf\\xb0\\xb1\\xb2\\xb3\\xb4\\xb5\\xb6\\xb7"
  94. "\\xb8\\xb9\\xba\\xbb\\xbc\\xbd\\xbe\\xbf\\xc0\\xc1\\xc2\\xc3\\xc4\\xc5"
  95. "\\xc6\\xc7\\xc8\\xc9\\xca\\xcb\\xcc\\xcd\\xce\\xcf\\xd0\\xd1\\xd2\\xd3"
  96. "\\xd4\\xd5\\xd6\\xd7\\xd8\\xd9\\xda\\xdb\\xdc\\xdd\\xde\\xdf\\xe0\\xe1"
  97. "\\xe2\\xe3\\xe4\\xe5\\xe6\\xe7\\xe8\\xe9\\xea\\xeb\\xec\\xed\\xee\\xef"
  98. "\\xf0\\xf1\\xf2\\xf3\\xf4\\xf5\\xf6\\xf7\\xf8\\xf9\\xfa\\xfb\\xfc\\xfd"
  99. "\\xfe\\xff'")
  100. testrepr = repr(u''.join(map(unichr, xrange(256))))
  101. self.assertEqual(testrepr, latin1repr)
  102. # Test repr works on wide unicode escapes without overflow.
  103. self.assertEqual(repr(u"\U00010000" * 39 + u"\uffff" * 4096),
  104. repr(u"\U00010000" * 39 + u"\uffff" * 4096))
  105. def test_count(self):
  106. string_tests.CommonTest.test_count(self)
  107. # check mixed argument types
  108. self.checkequalnofix(3, 'aaa', 'count', u'a')
  109. self.checkequalnofix(0, 'aaa', 'count', u'b')
  110. self.checkequalnofix(3, u'aaa', 'count', 'a')
  111. self.checkequalnofix(0, u'aaa', 'count', 'b')
  112. self.checkequalnofix(0, u'aaa', 'count', 'b')
  113. self.checkequalnofix(1, u'aaa', 'count', 'a', -1)
  114. self.checkequalnofix(3, u'aaa', 'count', 'a', -10)
  115. self.checkequalnofix(2, u'aaa', 'count', 'a', 0, -1)
  116. self.checkequalnofix(0, u'aaa', 'count', 'a', 0, -10)
  117. def test_find(self):
  118. self.checkequalnofix(0, u'abcdefghiabc', 'find', u'abc')
  119. self.checkequalnofix(9, u'abcdefghiabc', 'find', u'abc', 1)
  120. self.checkequalnofix(-1, u'abcdefghiabc', 'find', u'def', 4)
  121. self.assertRaises(TypeError, u'hello'.find)
  122. self.assertRaises(TypeError, u'hello'.find, 42)
  123. def test_rfind(self):
  124. string_tests.CommonTest.test_rfind(self)
  125. # check mixed argument types
  126. self.checkequalnofix(9, 'abcdefghiabc', 'rfind', u'abc')
  127. self.checkequalnofix(12, 'abcdefghiabc', 'rfind', u'')
  128. self.checkequalnofix(12, u'abcdefghiabc', 'rfind', '')
  129. def test_index(self):
  130. string_tests.CommonTest.test_index(self)
  131. # check mixed argument types
  132. for (t1, t2) in ((str, unicode), (unicode, str)):
  133. self.checkequalnofix(0, t1('abcdefghiabc'), 'index', t2(''))
  134. self.checkequalnofix(3, t1('abcdefghiabc'), 'index', t2('def'))
  135. self.checkequalnofix(0, t1('abcdefghiabc'), 'index', t2('abc'))
  136. self.checkequalnofix(9, t1('abcdefghiabc'), 'index', t2('abc'), 1)
  137. self.assertRaises(ValueError, t1('abcdefghiabc').index, t2('hib'))
  138. self.assertRaises(ValueError, t1('abcdefghiab').index, t2('abc'), 1)
  139. self.assertRaises(ValueError, t1('abcdefghi').index, t2('ghi'), 8)
  140. self.assertRaises(ValueError, t1('abcdefghi').index, t2('ghi'), -1)
  141. def test_rindex(self):
  142. string_tests.CommonTest.test_rindex(self)
  143. # check mixed argument types
  144. for (t1, t2) in ((str, unicode), (unicode, str)):
  145. self.checkequalnofix(12, t1('abcdefghiabc'), 'rindex', t2(''))
  146. self.checkequalnofix(3, t1('abcdefghiabc'), 'rindex', t2('def'))
  147. self.checkequalnofix(9, t1('abcdefghiabc'), 'rindex', t2('abc'))
  148. self.checkequalnofix(0, t1('abcdefghiabc'), 'rindex', t2('abc'), 0, -1)
  149. self.assertRaises(ValueError, t1('abcdefghiabc').rindex, t2('hib'))
  150. self.assertRaises(ValueError, t1('defghiabc').rindex, t2('def'), 1)
  151. self.assertRaises(ValueError, t1('defghiabc').rindex, t2('abc'), 0, -1)
  152. self.assertRaises(ValueError, t1('abcdefghi').rindex, t2('ghi'), 0, 8)
  153. self.assertRaises(ValueError, t1('abcdefghi').rindex, t2('ghi'), 0, -1)
  154. def test_translate(self):
  155. self.checkequalnofix(u'bbbc', u'abababc', 'translate', {ord('a'):None})
  156. self.checkequalnofix(u'iiic', u'abababc', 'translate', {ord('a'):None, ord('b'):ord('i')})
  157. self.checkequalnofix(u'iiix', u'abababc', 'translate', {ord('a'):None, ord('b'):ord('i'), ord('c'):u'x'})
  158. self.checkequalnofix(u'<i><i><i>c', u'abababc', 'translate', {ord('a'):None, ord('b'):u'<i>'})
  159. self.checkequalnofix(u'c', u'abababc', 'translate', {ord('a'):None, ord('b'):u''})
  160. self.checkequalnofix(u'xyyx', u'xzx', 'translate', {ord('z'):u'yy'})
  161. self.assertRaises(TypeError, u'hello'.translate)
  162. self.assertRaises(TypeError, u'abababc'.translate, {ord('a'):''})
  163. def test_split(self):
  164. string_tests.CommonTest.test_split(self)
  165. # Mixed arguments
  166. self.checkequalnofix([u'a', u'b', u'c', u'd'], u'a//b//c//d', 'split', '//')
  167. self.checkequalnofix([u'a', u'b', u'c', u'd'], 'a//b//c//d', 'split', u'//')
  168. self.checkequalnofix([u'endcase ', u''], u'endcase test', 'split', 'test')
  169. def test_join(self):
  170. string_tests.MixinStrUnicodeUserStringTest.test_join(self)
  171. # mixed arguments
  172. self.checkequalnofix(u'a b c d', u' ', 'join', ['a', 'b', u'c', u'd'])
  173. self.checkequalnofix(u'abcd', u'', 'join', (u'a', u'b', u'c', u'd'))
  174. self.checkequalnofix(u'w x y z', u' ', 'join', string_tests.Sequence('wxyz'))
  175. self.checkequalnofix(u'a b c d', ' ', 'join', [u'a', u'b', u'c', u'd'])
  176. self.checkequalnofix(u'a b c d', ' ', 'join', ['a', 'b', u'c', u'd'])
  177. self.checkequalnofix(u'abcd', '', 'join', (u'a', u'b', u'c', u'd'))
  178. self.checkequalnofix(u'w x y z', ' ', 'join', string_tests.Sequence(u'wxyz'))
  179. def test_strip(self):
  180. string_tests.CommonTest.test_strip(self)
  181. self.assertRaises(UnicodeError, u"hello".strip, "\xff")
  182. def test_replace(self):
  183. string_tests.CommonTest.test_replace(self)
  184. # method call forwarded from str implementation because of unicode argument
  185. self.checkequalnofix(u'one@two!three!', 'one!two!three!', 'replace', u'!', u'@', 1)
  186. self.assertRaises(TypeError, 'replace'.replace, u"r", 42)
  187. def test_comparison(self):
  188. # Comparisons:
  189. self.assertTrue(u'abc' == 'abc')
  190. self.assertTrue('abc' == u'abc')
  191. self.assertTrue(u'abc' == u'abc')
  192. self.assertTrue(u'abcd' > 'abc')
  193. self.assertTrue('abcd' > u'abc')
  194. self.assertTrue(u'abcd' > u'abc')
  195. self.assertTrue(u'abc' < 'abcd')
  196. self.assertTrue('abc' < u'abcd')
  197. self.assertTrue(u'abc' < u'abcd')
  198. if 0:
  199. # Move these tests to a Unicode collation module test...
  200. # Testing UTF-16 code point order comparisons...
  201. # No surrogates, no fixup required.
  202. self.assertTrue(u'\u0061' < u'\u20ac')
  203. # Non surrogate below surrogate value, no fixup required
  204. self.assertTrue(u'\u0061' < u'\ud800\udc02')
  205. # Non surrogate above surrogate value, fixup required
  206. def test_lecmp(s, s2):
  207. self.assertTrue(s < s2)
  208. def test_fixup(s):
  209. s2 = u'\ud800\udc01'
  210. test_lecmp(s, s2)
  211. s2 = u'\ud900\udc01'
  212. test_lecmp(s, s2)
  213. s2 = u'\uda00\udc01'
  214. test_lecmp(s, s2)
  215. s2 = u'\udb00\udc01'
  216. test_lecmp(s, s2)
  217. s2 = u'\ud800\udd01'
  218. test_lecmp(s, s2)
  219. s2 = u'\ud900\udd01'
  220. test_lecmp(s, s2)
  221. s2 = u'\uda00\udd01'
  222. test_lecmp(s, s2)
  223. s2 = u'\udb00\udd01'
  224. test_lecmp(s, s2)
  225. s2 = u'\ud800\ude01'
  226. test_lecmp(s, s2)
  227. s2 = u'\ud900\ude01'
  228. test_lecmp(s, s2)
  229. s2 = u'\uda00\ude01'
  230. test_lecmp(s, s2)
  231. s2 = u'\udb00\ude01'
  232. test_lecmp(s, s2)
  233. s2 = u'\ud800\udfff'
  234. test_lecmp(s, s2)
  235. s2 = u'\ud900\udfff'
  236. test_lecmp(s, s2)
  237. s2 = u'\uda00\udfff'
  238. test_lecmp(s, s2)
  239. s2 = u'\udb00\udfff'
  240. test_lecmp(s, s2)
  241. test_fixup(u'\ue000')
  242. test_fixup(u'\uff61')
  243. # Surrogates on both sides, no fixup required
  244. self.assertTrue(u'\ud800\udc02' < u'\ud84d\udc56')
  245. def test_capitalize(self):
  246. string_tests.CommonTest.test_capitalize(self)
  247. # check that titlecased chars are lowered correctly
  248. # \u1ffc is the titlecased char
  249. self.checkequal(u'\u1ffc\u1ff3\u1ff3\u1ff3',
  250. u'\u1ff3\u1ff3\u1ffc\u1ffc', 'capitalize')
  251. # check with cased non-letter chars
  252. self.checkequal(u'\u24c5\u24e8\u24e3\u24d7\u24de\u24dd',
  253. u'\u24c5\u24ce\u24c9\u24bd\u24c4\u24c3', 'capitalize')
  254. self.checkequal(u'\u24c5\u24e8\u24e3\u24d7\u24de\u24dd',
  255. u'\u24df\u24e8\u24e3\u24d7\u24de\u24dd', 'capitalize')
  256. self.checkequal(u'\u2160\u2171\u2172',
  257. u'\u2160\u2161\u2162', 'capitalize')
  258. self.checkequal(u'\u2160\u2171\u2172',
  259. u'\u2170\u2171\u2172', 'capitalize')
  260. # check with Ll chars with no upper - nothing changes here
  261. self.checkequal(u'\u019b\u1d00\u1d86\u0221\u1fb7',
  262. u'\u019b\u1d00\u1d86\u0221\u1fb7', 'capitalize')
  263. def test_islower(self):
  264. string_tests.MixinStrUnicodeUserStringTest.test_islower(self)
  265. self.checkequalnofix(False, u'\u1FFc', 'islower')
  266. @requires_wide_build
  267. def test_islower_non_bmp(self):
  268. # non-BMP, uppercase
  269. self.assertFalse(u'\U00010401'.islower())
  270. self.assertFalse(u'\U00010427'.islower())
  271. # non-BMP, lowercase
  272. self.assertTrue(u'\U00010429'.islower())
  273. self.assertTrue(u'\U0001044E'.islower())
  274. # non-BMP, non-cased
  275. self.assertFalse(u'\U0001F40D'.islower())
  276. self.assertFalse(u'\U0001F46F'.islower())
  277. def test_isupper(self):
  278. string_tests.MixinStrUnicodeUserStringTest.test_isupper(self)
  279. if not sys.platform.startswith('java'):
  280. self.checkequalnofix(False, u'\u1FFc', 'isupper')
  281. @requires_wide_build
  282. def test_isupper_non_bmp(self):
  283. # non-BMP, uppercase
  284. self.assertTrue(u'\U00010401'.isupper())
  285. self.assertTrue(u'\U00010427'.isupper())
  286. # non-BMP, lowercase
  287. self.assertFalse(u'\U00010429'.isupper())
  288. self.assertFalse(u'\U0001044E'.isupper())
  289. # non-BMP, non-cased
  290. self.assertFalse(u'\U0001F40D'.isupper())
  291. self.assertFalse(u'\U0001F46F'.isupper())
  292. def test_istitle(self):
  293. string_tests.MixinStrUnicodeUserStringTest.test_istitle(self)
  294. self.checkequalnofix(True, u'\u1FFc', 'istitle')
  295. self.checkequalnofix(True, u'Greek \u1FFcitlecases ...', 'istitle')
  296. @requires_wide_build
  297. def test_istitle_non_bmp(self):
  298. # non-BMP, uppercase + lowercase
  299. self.assertTrue(u'\U00010401\U00010429'.istitle())
  300. self.assertTrue(u'\U00010427\U0001044E'.istitle())
  301. # apparently there are no titlecased (Lt) non-BMP chars in Unicode 6
  302. for ch in [u'\U00010429', u'\U0001044E', u'\U0001F40D', u'\U0001F46F']:
  303. self.assertFalse(ch.istitle(), '{!r} is not title'.format(ch))
  304. def test_isspace(self):
  305. string_tests.MixinStrUnicodeUserStringTest.test_isspace(self)
  306. self.checkequalnofix(True, u'\u2000', 'isspace')
  307. self.checkequalnofix(True, u'\u200a', 'isspace')
  308. self.checkequalnofix(False, u'\u2014', 'isspace')
  309. @requires_wide_build
  310. def test_isspace_non_bmp(self):
  311. # apparently there are no non-BMP spaces chars in Unicode 6
  312. for ch in [u'\U00010401', u'\U00010427', u'\U00010429', u'\U0001044E',
  313. u'\U0001F40D', u'\U0001F46F']:
  314. self.assertFalse(ch.isspace(), '{!r} is not space.'.format(ch))
  315. @requires_wide_build
  316. def test_isalnum_non_bmp(self):
  317. for ch in [u'\U00010401', u'\U00010427', u'\U00010429', u'\U0001044E',
  318. u'\U0001D7F6', u'\U000104A0', u'\U000104A0', u'\U0001F107']:
  319. self.assertTrue(ch.isalnum(), '{!r} is alnum.'.format(ch))
  320. def test_isalpha(self):
  321. string_tests.MixinStrUnicodeUserStringTest.test_isalpha(self)
  322. self.checkequalnofix(True, u'\u1FFc', 'isalpha')
  323. @requires_wide_build
  324. def test_isalpha_non_bmp(self):
  325. # non-BMP, cased
  326. self.assertTrue(u'\U00010401'.isalpha())
  327. self.assertTrue(u'\U00010427'.isalpha())
  328. self.assertTrue(u'\U00010429'.isalpha())
  329. self.assertTrue(u'\U0001044E'.isalpha())
  330. # non-BMP, non-cased
  331. self.assertFalse(u'\U0001F40D'.isalpha())
  332. self.assertFalse(u'\U0001F46F'.isalpha())
  333. def test_isdecimal(self):
  334. self.checkequalnofix(False, u'', 'isdecimal')
  335. self.checkequalnofix(False, u'a', 'isdecimal')
  336. self.checkequalnofix(True, u'0', 'isdecimal')
  337. self.checkequalnofix(False, u'\u2460', 'isdecimal') # CIRCLED DIGIT ONE
  338. self.checkequalnofix(False, u'\xbc', 'isdecimal') # VULGAR FRACTION ONE QUARTER
  339. self.checkequalnofix(True, u'\u0660', 'isdecimal') # ARABIC-INDIC DIGIT ZERO
  340. self.checkequalnofix(True, u'0123456789', 'isdecimal')
  341. self.checkequalnofix(False, u'0123456789a', 'isdecimal')
  342. self.checkraises(TypeError, 'abc', 'isdecimal', 42)
  343. @requires_wide_build
  344. def test_isdecimal_non_bmp(self):
  345. for ch in [u'\U00010401', u'\U00010427', u'\U00010429', u'\U0001044E',
  346. u'\U0001F40D', u'\U0001F46F', u'\U00011065', u'\U0001F107']:
  347. self.assertFalse(ch.isdecimal(), '{!r} is not decimal.'.format(ch))
  348. for ch in [u'\U0001D7F6', u'\U000104A0', u'\U000104A0']:
  349. self.assertTrue(ch.isdecimal(), '{!r} is decimal.'.format(ch))
  350. def test_isdigit(self):
  351. string_tests.MixinStrUnicodeUserStringTest.test_isdigit(self)
  352. self.checkequalnofix(True, u'\u2460', 'isdigit')
  353. self.checkequalnofix(False, u'\xbc', 'isdigit')
  354. self.checkequalnofix(True, u'\u0660', 'isdigit')
  355. @requires_wide_build
  356. def test_isdigit_non_bmp(self):
  357. for ch in [u'\U00010401', u'\U00010427', u'\U00010429', u'\U0001044E',
  358. u'\U0001F40D', u'\U0001F46F', u'\U00011065']:
  359. self.assertFalse(ch.isdigit(), '{!r} is not a digit.'.format(ch))
  360. for ch in [u'\U0001D7F6', u'\U000104A0', u'\U000104A0', u'\U0001F107']:
  361. self.assertTrue(ch.isdigit(), '{!r} is a digit.'.format(ch))
  362. def test_isnumeric(self):
  363. self.checkequalnofix(False, u'', 'isnumeric')
  364. self.checkequalnofix(False, u'a', 'isnumeric')
  365. self.checkequalnofix(True, u'0', 'isnumeric')
  366. self.checkequalnofix(True, u'\u2460', 'isnumeric')
  367. self.checkequalnofix(True, u'\xbc', 'isnumeric')
  368. self.checkequalnofix(True, u'\u0660', 'isnumeric')
  369. self.checkequalnofix(True, u'0123456789', 'isnumeric')
  370. self.checkequalnofix(False, u'0123456789a', 'isnumeric')
  371. self.assertRaises(TypeError, u"abc".isnumeric, 42)
  372. @requires_wide_build
  373. def test_isnumeric_non_bmp(self):
  374. for ch in [u'\U00010401', u'\U00010427', u'\U00010429', u'\U0001044E',
  375. u'\U0001F40D', u'\U0001F46F']:
  376. self.assertFalse(ch.isnumeric(), '{!r} is not numeric.'.format(ch))
  377. for ch in [u'\U00010107', u'\U0001D7F6', u'\U00023b1b',
  378. u'\U000104A0', u'\U0001F107']:
  379. self.assertTrue(ch.isnumeric(), '{!r} is numeric.'.format(ch))
  380. @requires_wide_build
  381. def test_surrogates(self):
  382. # this test actually passes on narrow too, but it's just by accident.
  383. # Surrogates are seen as non-cased chars, so u'X\uD800X' is as
  384. # uppercase as 'X X'
  385. for s in (u'a\uD800b\uDFFF', u'a\uDFFFb\uD800',
  386. u'a\uD800b\uDFFFa', u'a\uDFFFb\uD800a'):
  387. self.assertTrue(s.islower())
  388. self.assertFalse(s.isupper())
  389. self.assertFalse(s.istitle())
  390. for s in (u'A\uD800B\uDFFF', u'A\uDFFFB\uD800',
  391. u'A\uD800B\uDFFFA', u'A\uDFFFB\uD800A'):
  392. self.assertFalse(s.islower())
  393. self.assertTrue(s.isupper())
  394. self.assertTrue(s.istitle())
  395. for meth_name in ('islower', 'isupper', 'istitle'):
  396. meth = getattr(unicode, meth_name)
  397. for s in (u'\uD800', u'\uDFFF', u'\uD800\uD800', u'\uDFFF\uDFFF'):
  398. self.assertFalse(meth(s), '%r.%s() is False' % (s, meth_name))
  399. for meth_name in ('isalpha', 'isalnum', 'isdigit', 'isspace',
  400. 'isdecimal', 'isnumeric'):
  401. meth = getattr(unicode, meth_name)
  402. for s in (u'\uD800', u'\uDFFF', u'\uD800\uD800', u'\uDFFF\uDFFF',
  403. u'a\uD800b\uDFFF', u'a\uDFFFb\uD800',
  404. u'a\uD800b\uDFFFa', u'a\uDFFFb\uD800a'):
  405. self.assertFalse(meth(s), '%r.%s() is False' % (s, meth_name))
  406. @requires_wide_build
  407. def test_lower(self):
  408. string_tests.CommonTest.test_lower(self)
  409. self.assertEqual(u'\U00010427'.lower(), u'\U0001044F')
  410. self.assertEqual(u'\U00010427\U00010427'.lower(),
  411. u'\U0001044F\U0001044F')
  412. self.assertEqual(u'\U00010427\U0001044F'.lower(),
  413. u'\U0001044F\U0001044F')
  414. self.assertEqual(u'X\U00010427x\U0001044F'.lower(),
  415. u'x\U0001044Fx\U0001044F')
  416. @requires_wide_build
  417. def test_upper(self):
  418. string_tests.CommonTest.test_upper(self)
  419. self.assertEqual(u'\U0001044F'.upper(), u'\U00010427')
  420. self.assertEqual(u'\U0001044F\U0001044F'.upper(),
  421. u'\U00010427\U00010427')
  422. self.assertEqual(u'\U00010427\U0001044F'.upper(),
  423. u'\U00010427\U00010427')
  424. self.assertEqual(u'X\U00010427x\U0001044F'.upper(),
  425. u'X\U00010427X\U00010427')
  426. @requires_wide_build
  427. def test_capitalize_wide_build(self):
  428. string_tests.CommonTest.test_capitalize(self)
  429. self.assertEqual(u'\U0001044F'.capitalize(), u'\U00010427')
  430. self.assertEqual(u'\U0001044F\U0001044F'.capitalize(),
  431. u'\U00010427\U0001044F')
  432. self.assertEqual(u'\U00010427\U0001044F'.capitalize(),
  433. u'\U00010427\U0001044F')
  434. self.assertEqual(u'\U0001044F\U00010427'.capitalize(),
  435. u'\U00010427\U0001044F')
  436. self.assertEqual(u'X\U00010427x\U0001044F'.capitalize(),
  437. u'X\U0001044Fx\U0001044F')
  438. @requires_wide_build
  439. def test_title(self):
  440. string_tests.MixinStrUnicodeUserStringTest.test_title(self)
  441. self.assertEqual(u'\U0001044F'.title(), u'\U00010427')
  442. self.assertEqual(u'\U0001044F\U0001044F'.title(),
  443. u'\U00010427\U0001044F')
  444. self.assertEqual(u'\U0001044F\U0001044F \U0001044F\U0001044F'.title(),
  445. u'\U00010427\U0001044F \U00010427\U0001044F')
  446. self.assertEqual(u'\U00010427\U0001044F \U00010427\U0001044F'.title(),
  447. u'\U00010427\U0001044F \U00010427\U0001044F')
  448. self.assertEqual(u'\U0001044F\U00010427 \U0001044F\U00010427'.title(),
  449. u'\U00010427\U0001044F \U00010427\U0001044F')
  450. self.assertEqual(u'X\U00010427x\U0001044F X\U00010427x\U0001044F'.title(),
  451. u'X\U0001044Fx\U0001044F X\U0001044Fx\U0001044F')
  452. @requires_wide_build
  453. def test_swapcase(self):
  454. string_tests.CommonTest.test_swapcase(self)
  455. self.assertEqual(u'\U0001044F'.swapcase(), u'\U00010427')
  456. self.assertEqual(u'\U00010427'.swapcase(), u'\U0001044F')
  457. self.assertEqual(u'\U0001044F\U0001044F'.swapcase(),
  458. u'\U00010427\U00010427')
  459. self.assertEqual(u'\U00010427\U0001044F'.swapcase(),
  460. u'\U0001044F\U00010427')
  461. self.assertEqual(u'\U0001044F\U00010427'.swapcase(),
  462. u'\U00010427\U0001044F')
  463. self.assertEqual(u'X\U00010427x\U0001044F'.swapcase(),
  464. u'x\U0001044FX\U00010427')
  465. def test_contains(self):
  466. # Testing Unicode contains method
  467. self.assertIn('a', u'abdb')
  468. self.assertIn('a', u'bdab')
  469. self.assertIn('a', u'bdaba')
  470. self.assertIn('a', u'bdba')
  471. self.assertIn('a', u'bdba')
  472. self.assertIn(u'a', u'bdba')
  473. self.assertNotIn(u'a', u'bdb')
  474. self.assertNotIn(u'a', 'bdb')
  475. self.assertIn(u'a', 'bdba')
  476. self.assertIn(u'a', ('a',1,None))
  477. self.assertIn(u'a', (1,None,'a'))
  478. self.assertIn(u'a', (1,None,u'a'))
  479. self.assertIn('a', ('a',1,None))
  480. self.assertIn('a', (1,None,'a'))
  481. self.assertIn('a', (1,None,u'a'))
  482. self.assertNotIn('a', ('x',1,u'y'))
  483. self.assertNotIn('a', ('x',1,None))
  484. self.assertNotIn(u'abcd', u'abcxxxx')
  485. self.assertIn(u'ab', u'abcd')
  486. self.assertIn('ab', u'abc')
  487. self.assertIn(u'ab', 'abc')
  488. self.assertIn(u'ab', (1,None,u'ab'))
  489. self.assertIn(u'', u'abc')
  490. self.assertIn('', u'abc')
  491. # If the following fails either
  492. # the contains operator does not propagate UnicodeErrors or
  493. # someone has changed the default encoding
  494. self.assertRaises(UnicodeDecodeError, 'g\xe2teau'.__contains__, u'\xe2')
  495. self.assertRaises(UnicodeDecodeError, u'g\xe2teau'.__contains__, '\xe2')
  496. self.assertIn(u'', '')
  497. self.assertIn('', u'')
  498. self.assertIn(u'', u'')
  499. self.assertIn(u'', 'abc')
  500. self.assertIn('', u'abc')
  501. self.assertIn(u'', u'abc')
  502. self.assertNotIn(u'\0', 'abc')
  503. self.assertNotIn('\0', u'abc')
  504. self.assertNotIn(u'\0', u'abc')
  505. self.assertIn(u'\0', '\0abc')
  506. self.assertIn('\0', u'\0abc')
  507. self.assertIn(u'\0', u'\0abc')
  508. self.assertIn(u'\0', 'abc\0')
  509. self.assertIn('\0', u'abc\0')
  510. self.assertIn(u'\0', u'abc\0')
  511. self.assertIn(u'a', '\0abc')
  512. self.assertIn('a', u'\0abc')
  513. self.assertIn(u'a', u'\0abc')
  514. self.assertIn(u'asdf', 'asdf')
  515. self.assertIn('asdf', u'asdf')
  516. self.assertIn(u'asdf', u'asdf')
  517. self.assertNotIn(u'asdf', 'asd')
  518. self.assertNotIn('asdf', u'asd')
  519. self.assertNotIn(u'asdf', u'asd')
  520. self.assertNotIn(u'asdf', '')
  521. self.assertNotIn('asdf', u'')
  522. self.assertNotIn(u'asdf', u'')
  523. self.assertRaises(TypeError, u"abc".__contains__)
  524. self.assertRaises(TypeError, u"abc".__contains__, object())
  525. def test_formatting(self):
  526. string_tests.MixinStrUnicodeUserStringTest.test_formatting(self)
  527. # Testing Unicode formatting strings...
  528. self.assertEqual(u"%s, %s" % (u"abc", "abc"), u'abc, abc')
  529. self.assertEqual(u"%s, %s, %i, %f, %5.2f" % (u"abc", "abc", 1, 2, 3), u'abc, abc, 1, 2.000000, 3.00')
  530. self.assertEqual(u"%s, %s, %i, %f, %5.2f" % (u"abc", "abc", 1, -2, 3), u'abc, abc, 1, -2.000000, 3.00')
  531. self.assertEqual(u"%s, %s, %i, %f, %5.2f" % (u"abc", "abc", -1, -2, 3.5), u'abc, abc, -1, -2.000000, 3.50')
  532. self.assertEqual(u"%s, %s, %i, %f, %5.2f" % (u"abc", "abc", -1, -2, 3.57), u'abc, abc, -1, -2.000000, 3.57')
  533. self.assertEqual(u"%s, %s, %i, %f, %5.2f" % (u"abc", "abc", -1, -2, 1003.57), u'abc, abc, -1, -2.000000, 1003.57')
  534. if not sys.platform.startswith('java'):
  535. self.assertEqual(u"%r, %r" % (u"abc", "abc"), u"u'abc', 'abc'")
  536. self.assertEqual(u"%(x)s, %(y)s" % {'x':u"abc", 'y':"def"}, u'abc, def')
  537. self.assertEqual(u"%(x)s, %(\xfc)s" % {'x':u"abc", u'\xfc':"def"}, u'abc, def')
  538. self.assertEqual(u'%c' % 0x1234, u'\u1234')
  539. self.assertRaises(OverflowError, u"%c".__mod__, (sys.maxunicode+1,))
  540. self.assertRaises(ValueError, u"%.1\u1032f".__mod__, (1.0/3))
  541. for num in range(0x00,0x80):
  542. char = chr(num)
  543. self.assertEqual(u"%c" % char, unicode(char))
  544. self.assertEqual(u"%c" % num, unicode(char))
  545. self.assertTrue(char == u"%c" % char)
  546. self.assertTrue(char == u"%c" % num)
  547. # Issue 7649
  548. for num in range(0x80,0x100):
  549. uchar = unichr(num)
  550. self.assertEqual(uchar, u"%c" % num) # works only with ints
  551. self.assertEqual(uchar, u"%c" % uchar) # and unicode chars
  552. # the implicit decoding should fail for non-ascii chars
  553. self.assertRaises(UnicodeDecodeError, u"%c".__mod__, chr(num))
  554. self.assertRaises(UnicodeDecodeError, u"%s".__mod__, chr(num))
  555. # formatting jobs delegated from the string implementation:
  556. self.assertEqual('...%(foo)s...' % {'foo':u"abc"}, u'...abc...')
  557. self.assertEqual('...%(foo)s...' % {'foo':"abc"}, '...abc...')
  558. self.assertEqual('...%(foo)s...' % {u'foo':"abc"}, '...abc...')
  559. self.assertEqual('...%(foo)s...' % {u'foo':u"abc"}, u'...abc...')
  560. self.assertEqual('...%(foo)s...' % {u'foo':u"abc",'def':123}, u'...abc...')
  561. self.assertEqual('...%(foo)s...' % {u'foo':u"abc",u'def':123}, u'...abc...')
  562. self.assertEqual('...%s...%s...%s...%s...' % (1,2,3,u"abc"), u'...1...2...3...abc...')
  563. self.assertEqual('...%%...%%s...%s...%s...%s...%s...' % (1,2,3,u"abc"), u'...%...%s...1...2...3...abc...')
  564. self.assertEqual('...%s...' % u"abc", u'...abc...')
  565. self.assertEqual('%*s' % (5,u'abc',), u' abc')
  566. self.assertEqual('%*s' % (-5,u'abc',), u'abc ')
  567. self.assertEqual('%*.*s' % (5,2,u'abc',), u' ab')
  568. self.assertEqual('%*.*s' % (5,3,u'abc',), u' abc')
  569. self.assertEqual('%i %*.*s' % (10, 5,3,u'abc',), u'10 abc')
  570. self.assertEqual('%i%s %*.*s' % (10, 3, 5, 3, u'abc',), u'103 abc')
  571. self.assertEqual('%c' % u'a', u'a')
  572. class Wrapper:
  573. def __str__(self):
  574. return u'\u1234'
  575. self.assertEqual('%s' % Wrapper(), u'\u1234')
  576. def test_formatting_huge_precision(self):
  577. format_string = u"%.{}f".format(sys.maxsize + 1)
  578. with self.assertRaises(ValueError):
  579. result = format_string % 2.34
  580. @test_support.cpython_only
  581. def test_formatting_huge_precision_c_limits(self):
  582. from _testcapi import INT_MAX
  583. format_string = u"%.{}f".format(INT_MAX + 1)
  584. with self.assertRaises(ValueError):
  585. result = format_string % 2.34
  586. def test_formatting_huge_width(self):
  587. format_string = u"%{}f".format(sys.maxsize + 1)
  588. with self.assertRaises(ValueError):
  589. result = format_string % 2.34
  590. def test_startswith_endswith_errors(self):
  591. for meth in (u'foo'.startswith, u'foo'.endswith):
  592. with self.assertRaises(UnicodeDecodeError):
  593. meth('\xff')
  594. with self.assertRaises(TypeError) as cm:
  595. meth(['f'])
  596. exc = str(cm.exception)
  597. self.assertIn('unicode', exc)
  598. self.assertIn('str', exc)
  599. self.assertIn('tuple', exc)
  600. @test_support.run_with_locale('LC_ALL', 'de_DE', 'fr_FR')
  601. def test_format_float(self):
  602. # should not format with a comma, but always with C locale
  603. self.assertEqual(u'1.0', u'%.1f' % 1.0)
  604. def test_constructor(self):
  605. # unicode(obj) tests (this maps to PyObject_Unicode() at C level)
  606. self.assertEqual(
  607. unicode(u'unicode remains unicode'),
  608. u'unicode remains unicode'
  609. )
  610. self.assertEqual(
  611. unicode(UnicodeSubclass('unicode subclass becomes unicode')),
  612. u'unicode subclass becomes unicode'
  613. )
  614. self.assertEqual(
  615. unicode('strings are converted to unicode'),
  616. u'strings are converted to unicode'
  617. )
  618. class UnicodeCompat:
  619. def __init__(self, x):
  620. self.x = x
  621. def __unicode__(self):
  622. return self.x
  623. self.assertEqual(
  624. unicode(UnicodeCompat('__unicode__ compatible objects are recognized')),
  625. u'__unicode__ compatible objects are recognized')
  626. class StringCompat:
  627. def __init__(self, x):
  628. self.x = x
  629. def __str__(self):
  630. return self.x
  631. self.assertEqual(
  632. unicode(StringCompat('__str__ compatible objects are recognized')),
  633. u'__str__ compatible objects are recognized'
  634. )
  635. # unicode(obj) is compatible to str():
  636. o = StringCompat('unicode(obj) is compatible to str()')
  637. self.assertEqual(unicode(o), u'unicode(obj) is compatible to str()')
  638. self.assertEqual(str(o), 'unicode(obj) is compatible to str()')
  639. # %-formatting and .__unicode__()
  640. self.assertEqual(u'%s' %
  641. UnicodeCompat(u"u'%s' % obj uses obj.__unicode__()"),
  642. u"u'%s' % obj uses obj.__unicode__()")
  643. self.assertEqual(u'%s' %
  644. UnicodeCompat(u"u'%s' % obj falls back to obj.__str__()"),
  645. u"u'%s' % obj falls back to obj.__str__()")
  646. for obj in (123, 123.45, 123L):
  647. self.assertEqual(unicode(obj), unicode(str(obj)))
  648. # unicode(obj, encoding, error) tests (this maps to
  649. # PyUnicode_FromEncodedObject() at C level)
  650. if not sys.platform.startswith('java'):
  651. self.assertRaises(
  652. TypeError,
  653. unicode,
  654. u'decoding unicode is not supported',
  655. 'utf-8',
  656. 'strict'
  657. )
  658. self.assertEqual(
  659. unicode('strings are decoded to unicode', 'utf-8', 'strict'),
  660. u'strings are decoded to unicode'
  661. )
  662. if not sys.platform.startswith('java'):
  663. with test_support.check_py3k_warnings():
  664. buf = buffer('character buffers are decoded to unicode')
  665. self.assertEqual(
  666. unicode(
  667. buf,
  668. 'utf-8',
  669. 'strict'
  670. ),
  671. u'character buffers are decoded to unicode'
  672. )
  673. self.assertRaises(TypeError, unicode, 42, 42, 42)
  674. def test_codecs_utf7(self):
  675. utfTests = [
  676. (u'A\u2262\u0391.', 'A+ImIDkQ.'), # RFC2152 example
  677. (u'Hi Mom -\u263a-!', 'Hi Mom -+Jjo--!'), # RFC2152 example
  678. (u'\u65E5\u672C\u8A9E', '+ZeVnLIqe-'), # RFC2152 example
  679. (u'Item 3 is \u00a31.', 'Item 3 is +AKM-1.'), # RFC2152 example
  680. (u'+', '+-'),
  681. (u'+-', '+--'),
  682. (u'+?', '+-?'),
  683. (u'\?', '+AFw?'),
  684. (u'+?', '+-?'),
  685. (ur'\\?', '+AFwAXA?'),
  686. (ur'\\\?', '+AFwAXABc?'),
  687. (ur'++--', '+-+---'),
  688. (u'\U000abcde', '+2m/c3g-'), # surrogate pairs
  689. (u'/', '/'),
  690. ]
  691. for (x, y) in utfTests:
  692. self.assertEqual(x.encode('utf-7'), y)
  693. # Unpaired surrogates are passed through
  694. self.assertEqual(u'\uD801'.encode('utf-7'), '+2AE-')
  695. self.assertEqual(u'\uD801x'.encode('utf-7'), '+2AE-x')
  696. self.assertEqual(u'\uDC01'.encode('utf-7'), '+3AE-')
  697. self.assertEqual(u'\uDC01x'.encode('utf-7'), '+3AE-x')
  698. self.assertEqual('+2AE-'.decode('utf-7'), u'\uD801')
  699. self.assertEqual('+2AE-x'.decode('utf-7'), u'\uD801x')
  700. self.assertEqual('+3AE-'.decode('utf-7'), u'\uDC01')
  701. self.assertEqual('+3AE-x'.decode('utf-7'), u'\uDC01x')
  702. self.assertEqual(u'\uD801\U000abcde'.encode('utf-7'), '+2AHab9ze-')
  703. self.assertEqual('+2AHab9ze-'.decode('utf-7'), u'\uD801\U000abcde')
  704. # Direct encoded characters
  705. set_d = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'(),-./:?"
  706. # Optional direct characters
  707. set_o = '!"#$%&*;<=>@[]^_`{|}'
  708. for c in set_d:
  709. self.assertEqual(c.encode('utf7'), c.encode('ascii'))
  710. self.assertEqual(c.encode('ascii').decode('utf7'), unicode(c))
  711. self.assertTrue(c == c.encode('ascii').decode('utf7'))
  712. for c in set_o:
  713. self.assertEqual(c.encode('ascii').decode('utf7'), unicode(c))
  714. self.assertTrue(c == c.encode('ascii').decode('utf7'))
  715. def test_codecs_utf8(self):
  716. self.assertEqual(u''.encode('utf-8'), '')
  717. self.assertEqual(u'\u20ac'.encode('utf-8'), '\xe2\x82\xac')
  718. self.assertEqual(u'\ud800\udc02'.encode('utf-8'), '\xf0\x90\x80\x82')
  719. self.assertEqual(u'\ud84d\udc56'.encode('utf-8'), '\xf0\xa3\x91\x96')
  720. self.assertEqual(u'\ud800'.encode('utf-8'), '\xed\xa0\x80')
  721. self.assertEqual(u'\udc00'.encode('utf-8'), '\xed\xb0\x80')
  722. self.assertEqual(
  723. (u'\ud800\udc02'*1000).encode('utf-8'),
  724. '\xf0\x90\x80\x82'*1000
  725. )
  726. self.assertEqual(
  727. u'\u6b63\u78ba\u306b\u8a00\u3046\u3068\u7ffb\u8a33\u306f'
  728. u'\u3055\u308c\u3066\u3044\u307e\u305b\u3093\u3002\u4e00'
  729. u'\u90e8\u306f\u30c9\u30a4\u30c4\u8a9e\u3067\u3059\u304c'
  730. u'\u3001\u3042\u3068\u306f\u3067\u305f\u3089\u3081\u3067'
  731. u'\u3059\u3002\u5b9f\u969b\u306b\u306f\u300cWenn ist das'
  732. u' Nunstuck git und'.encode('utf-8'),
  733. '\xe6\xad\xa3\xe7\xa2\xba\xe3\x81\xab\xe8\xa8\x80\xe3\x81'
  734. '\x86\xe3\x81\xa8\xe7\xbf\xbb\xe8\xa8\xb3\xe3\x81\xaf\xe3'
  735. '\x81\x95\xe3\x82\x8c\xe3\x81\xa6\xe3\x81\x84\xe3\x81\xbe'
  736. '\xe3\x81\x9b\xe3\x82\x93\xe3\x80\x82\xe4\xb8\x80\xe9\x83'
  737. '\xa8\xe3\x81\xaf\xe3\x83\x89\xe3\x82\xa4\xe3\x83\x84\xe8'
  738. '\xaa\x9e\xe3\x81\xa7\xe3\x81\x99\xe3\x81\x8c\xe3\x80\x81'
  739. '\xe3\x81\x82\xe3\x81\xa8\xe3\x81\xaf\xe3\x81\xa7\xe3\x81'
  740. '\x9f\xe3\x82\x89\xe3\x82\x81\xe3\x81\xa7\xe3\x81\x99\xe3'
  741. '\x80\x82\xe5\xae\x9f\xe9\x9a\x9b\xe3\x81\xab\xe3\x81\xaf'
  742. '\xe3\x80\x8cWenn ist das Nunstuck git und'
  743. )
  744. # UTF-8 specific decoding tests
  745. self.assertEqual(unicode('\xf0\xa3\x91\x96', 'utf-8'), u'\U00023456')
  746. self.assertEqual(unicode('\xf0\x90\x80\x82', 'utf-8'), u'\U00010002')
  747. self.assertEqual(unicode('\xe2\x82\xac', 'utf-8'), u'\u20ac')
  748. # Other possible utf-8 test cases:
  749. # * strict decoding testing for all of the
  750. # UTF8_ERROR cases in PyUnicode_DecodeUTF8
  751. def test_utf8_decode_valid_sequences(self):
  752. sequences = [
  753. # single byte
  754. ('\x00', u'\x00'), ('a', u'a'), ('\x7f', u'\x7f'),
  755. # 2 bytes
  756. ('\xc2\x80', u'\x80'), ('\xdf\xbf', u'\u07ff'),
  757. # 3 bytes
  758. ('\xe0\xa0\x80', u'\u0800'), ('\xed\x9f\xbf', u'\ud7ff'),
  759. ('\xee\x80\x80', u'\uE000'), ('\xef\xbf\xbf', u'\uffff'),
  760. # 4 bytes
  761. ('\xF0\x90\x80\x80', u'\U00010000'),
  762. ('\xf4\x8f\xbf\xbf', u'\U0010FFFF')
  763. ]
  764. for seq, res in sequences:
  765. self.assertEqual(seq.decode('utf-8'), res)
  766. for ch in map(unichr, range(0, sys.maxunicode)):
  767. self.assertEqual(ch, ch.encode('utf-8').decode('utf-8'))
  768. def test_utf8_decode_invalid_sequences(self):
  769. # continuation bytes in a sequence of 2, 3, or 4 bytes
  770. continuation_bytes = map(chr, range(0x80, 0xC0))
  771. # start bytes of a 2-byte sequence equivalent to code points < 0x7F
  772. invalid_2B_seq_start_bytes = map(chr, range(0xC0, 0xC2))
  773. # start bytes of a 4-byte sequence equivalent to code points > 0x10FFFF
  774. invalid_4B_seq_start_bytes = map(chr, range(0xF5, 0xF8))
  775. invalid_start_bytes = (
  776. continuation_bytes + invalid_2B_seq_start_bytes +
  777. invalid_4B_seq_start_bytes + map(chr, range(0xF7, 0x100))
  778. )
  779. for byte in invalid_start_bytes:
  780. self.assertRaises(UnicodeDecodeError, byte.decode, 'utf-8')
  781. for sb in invalid_2B_seq_start_bytes:
  782. for cb in continuation_bytes:
  783. self.assertRaises(UnicodeDecodeError, (sb+cb).decode, 'utf-8')
  784. for sb in invalid_4B_seq_start_bytes:
  785. for cb1 in continuation_bytes[:3]:
  786. for cb3 in continuation_bytes[:3]:
  787. self.assertRaises(UnicodeDecodeError,
  788. (sb+cb1+'\x80'+cb3).decode, 'utf-8')
  789. for cb in map(chr, range(0x80, 0xA0)):
  790. self.assertRaises(UnicodeDecodeError,
  791. ('\xE0'+cb+'\x80').decode, 'utf-8')
  792. self.assertRaises(UnicodeDecodeError,
  793. ('\xE0'+cb+'\xBF').decode, 'utf-8')
  794. # XXX: surrogates shouldn't be valid UTF-8!
  795. # see http://www.unicode.org/versions/Unicode5.2.0/ch03.pdf
  796. # (table 3-7) and http://www.rfc-editor.org/rfc/rfc3629.txt
  797. #for cb in map(chr, range(0xA0, 0xC0)):
  798. #self.assertRaises(UnicodeDecodeError,
  799. #('\xED'+cb+'\x80').decode, 'utf-8')
  800. #self.assertRaises(UnicodeDecodeError,
  801. #('\xED'+cb+'\xBF').decode, 'utf-8')
  802. # but since they are valid on Python 2 add a test for that:
  803. for cb, surrogate in zip(map(chr, range(0xA0, 0xC0)),
  804. map(unichr, range(0xd800, 0xe000, 64))):
  805. encoded = '\xED'+cb+'\x80'
  806. self.assertEqual(encoded.decode('utf-8'), surrogate)
  807. self.assertEqual(surrogate.encode('utf-8'), encoded)
  808. for cb in map(chr, range(0x80, 0x90)):
  809. self.assertRaises(UnicodeDecodeError,
  810. ('\xF0'+cb+'\x80\x80').decode, 'utf-8')
  811. self.assertRaises(UnicodeDecodeError,
  812. ('\xF0'+cb+'\xBF\xBF').decode, 'utf-8')
  813. for cb in map(chr, range(0x90, 0xC0)):
  814. self.assertRaises(UnicodeDecodeError,
  815. ('\xF4'+cb+'\x80\x80').decode, 'utf-8')
  816. self.assertRaises(UnicodeDecodeError,
  817. ('\xF4'+cb+'\xBF\xBF').decode, 'utf-8')
  818. def test_issue8271(self):
  819. # Issue #8271: during the decoding of an invalid UTF-8 byte sequence,
  820. # only the start byte and the continuation byte(s) are now considered
  821. # invalid, instead of the number of bytes specified by the start byte.
  822. # See http://www.unicode.org/versions/Unicode5.2.0/ch03.pdf (page 95,
  823. # table 3-8, Row 2) for more information about the algorithm used.
  824. FFFD = u'\ufffd'
  825. sequences = [
  826. # invalid start bytes
  827. ('\x80', FFFD), # continuation byte
  828. ('\x80\x80', FFFD*2), # 2 continuation bytes
  829. ('\xc0', FFFD),
  830. ('\xc0\xc0', FFFD*2),
  831. ('\xc1', FFFD),
  832. ('\xc1\xc0', FFFD*2),
  833. ('\xc0\xc1', FFFD*2),
  834. # with start byte of a 2-byte sequence
  835. ('\xc2', FFFD), # only the start byte
  836. ('\xc2\xc2', FFFD*2), # 2 start bytes
  837. ('\xc2\xc2\xc2', FFFD*3), # 2 start bytes
  838. ('\xc2\x41', FFFD+'A'), # invalid continuation byte
  839. # with start byte of a 3-byte sequence
  840. ('\xe1', FFFD), # only the start byte
  841. ('\xe1\xe1', FFFD*2), # 2 start bytes
  842. ('\xe1\xe1\xe1', FFFD*3), # 3 start bytes
  843. ('\xe1\xe1\xe1\xe1', FFFD*4), # 4 start bytes
  844. ('\xe1\x80', FFFD), # only 1 continuation byte
  845. ('\xe1\x41', FFFD+'A'), # invalid continuation byte
  846. ('\xe1\x41\x80', FFFD+'A'+FFFD), # invalid cb followed by valid cb
  847. ('\xe1\x41\x41', FFFD+'AA'), # 2 invalid continuation bytes
  848. ('\xe1\x80\x41', FFFD+'A'), # only 1 valid continuation byte
  849. ('\xe1\x80\xe1\x41', FFFD*2+'A'), # 1 valid and the other invalid
  850. ('\xe1\x41\xe1\x80', FFFD+'A'+FFFD), # 1 invalid and the other valid
  851. # with start byte of a 4-byte sequence
  852. ('\xf1', FFFD), # only the start byte
  853. ('\xf1\xf1', FFFD*2), # 2 start bytes
  854. ('\xf1\xf1\xf1', FFFD*3), # 3 start bytes
  855. ('\xf1\xf1\xf1\xf1', FFFD*4), # 4 start bytes
  856. ('\xf1\xf1\xf1\xf1\xf1', FFFD*5), # 5 start bytes
  857. ('\xf1\x80', FFFD), # only 1 continuation bytes
  858. ('\xf1\x80\x80', FFFD), # only 2 continuation bytes
  859. ('\xf1\x80\x41', FFFD+'A'), # 1 valid cb and 1 invalid
  860. ('\xf1\x80\x41\x41', FFFD+'AA'), # 1 valid cb and 1 invalid
  861. ('\xf1\x80\x80\x41', FFFD+'A'), # 2 valid cb and 1 invalid
  862. ('\xf1\x41\x80', FFFD+'A'+FFFD), # 1 invalid cv and 1 valid
  863. ('\xf1\x41\x80\x80', FFFD+'A'+FFFD*2), # 1 invalid cb and 2 invalid
  864. ('\xf1\x41\x80\x41', FFFD+'A'+FFFD+'A'), # 2 invalid cb and 1 invalid
  865. ('\xf1\x41\x41\x80', FFFD+'AA'+FFFD), # 1 valid cb and 1 invalid
  866. ('\xf1\x41\xf1\x80', FFFD+'A'+FFFD),
  867. ('\xf1\x41\x80\xf1', FFFD+'A'+FFFD*2),
  868. ('\xf1\xf1\x80\x41', FFFD*2+'A'),
  869. ('\xf1\x41\xf1\xf1', FFFD+'A'+FFFD*2),
  870. # with invalid start byte of a 4-byte sequence (rfc2279)
  871. ('\xf5', FFFD), # only the start byte
  872. ('\xf5\xf5', FFFD*2), # 2 start bytes
  873. ('\xf5\x80', FFFD*2), # only 1 continuation byte
  874. ('\xf5\x80\x80', FFFD*3), # only 2 continuation byte
  875. ('\xf5\x80\x80\x80', FFFD*4), # 3 continuation bytes
  876. ('\xf5\x80\x41', FFFD*2+'A'), # 1 valid cb and 1 invalid
  877. ('\xf5\x80\x41\xf5', FFFD*2+'A'+FFFD),
  878. ('\xf5\x41\x80\x80\x41', FFFD+'A'+FFFD*2+'A'),
  879. # with invalid start byte of a 5-byte sequence (rfc2279)
  880. ('\xf8', FFFD), # only the start byte
  881. ('\xf8\xf8', FFFD*2), # 2 start bytes
  882. ('\xf8\x80', FFFD*2), # only one continuation byte
  883. ('\xf8\x80\x41', FFFD*2 + 'A'), # 1 valid cb and 1 invalid
  884. ('\xf8\x80\x80\x80\x80', FFFD*5), # invalid 5 bytes seq with 5 bytes
  885. # with invalid start byte of a 6-byte sequence (rfc2279)
  886. ('\xfc', FFFD), # only the start byte
  887. ('\xfc\xfc', FFFD*2), # 2 start bytes
  888. ('\xfc\x80\x80', FFFD*3), # only 2 continuation bytes
  889. ('\xfc\x80\x80\x80\x80\x80', FFFD*6), # 6 continuation bytes
  890. # invalid start byte
  891. ('\xfe', FFFD),
  892. ('\xfe\x80\x80', FFFD*3),
  893. # other sequences
  894. ('\xf1\x80\x41\x42\x43', u'\ufffd\x41\x42\x43'),
  895. ('\xf1\x80\xff\x42\x43', u'\ufffd\ufffd\x42\x43'),
  896. ('\xf1\x80\xc2\x81\x43', u'\ufffd\x81\x43'),
  897. ('\x61\xF1\x80\x80\xE1\x80\xC2\x62\x80\x63\x80\xBF\x64',
  898. u'\x61\uFFFD\uFFFD\uFFFD\x62\uFFFD\x63\uFFFD\uFFFD\x64'),
  899. ]
  900. for n, (seq, res) in enumerate(sequences):
  901. self.assertRaises(UnicodeDecodeError, seq.decode, 'utf-8', 'strict')
  902. self.assertEqual(seq.decode('utf-8', 'replace'), res)
  903. self.assertEqual((seq+'b').decode('utf-8', 'replace'), res+'b')
  904. self.assertEqual(seq.decode('utf-8', 'ignore'),
  905. res.replace(u'\uFFFD', ''))
  906. def test_codecs_idna(self):
  907. # Test whether trailing dot is preserved
  908. self.assertEqual(u"www.python.org.".encode("idna"), "www.python.org.")
  909. def test_codecs_errors(self):
  910. # Error handling (encoding)
  911. self.assertRaises(UnicodeError, u'Andr\202 x'.encode, 'ascii')
  912. self.assertRaises(UnicodeError, u'Andr\202 x'.encode, 'ascii','strict')
  913. self.assertEqual(u'Andr\202 x'.encode('ascii','ignore'), "Andr x")
  914. self.assertEqual(u'Andr\202 x'.encode('ascii','replace'), "Andr? x")
  915. self.assertEqual(u'Andr\202 x'.encode('ascii', 'replace'),
  916. u'Andr\202 x'.encode('ascii', errors='replace'))
  917. self.assertEqual(u'Andr\202 x'.encode('ascii', 'ignore'),
  918. u'Andr\202 x'.encode(encoding='ascii', errors='ignore'))
  919. # Error handling (decoding)
  920. self.assertRaises(UnicodeError, unicode, 'Andr\202 x', 'ascii')
  921. self.assertRaises(UnicodeError, unicode, 'Andr\202 x', 'ascii','strict')
  922. self.assertEqual(unicode('Andr\202 x','ascii','ignore'), u"Andr x")
  923. self.assertEqual(unicode('Andr\202 x','ascii','replace'), u'Andr\uFFFD x')
  924. self.assertEqual(unicode('\202 x', 'ascii', 'replace'), u'\uFFFD x')
  925. with test_support.check_py3k_warnings():
  926. self.assertEqual(u'abcde'.decode('ascii', 'ignore'),
  927. u'abcde'.decode('ascii', errors='ignore'))
  928. with test_support.check_py3k_warnings():
  929. self.assertEqual(u'abcde'.decode('ascii', 'replace'),
  930. u'abcde'.decode(encoding='ascii', errors='replace'))
  931. # Error handling (unknown character names)
  932. self.assertEqual("\\N{foo}xx".decode("unicode-escape", "ignore"), u"xx")
  933. # Error handling (truncated escape sequence)
  934. self.assertRaises(UnicodeError, "\\".decode, "unicode-escape")
  935. self.assertRaises(TypeError, "hello".decode, "test.unicode1")
  936. self.assertRaises(TypeError, unicode, "hello", "test.unicode2")
  937. self.assertRaises(TypeError, u"hello".encode, "test.unicode1")
  938. self.assertRaises(TypeError, u"hello".encode, "test.unicode2")
  939. # executes PyUnicode_Encode()
  940. import imp
  941. self.assertRaises(
  942. ImportError,
  943. imp.find_module,
  944. "non-existing module",
  945. [u"non-existing dir"]
  946. )
  947. # Error handling (wrong arguments)
  948. self.assertRaises(TypeError, u"hello".encode, 42, 42, 42)
  949. # Error handling (PyUnicode_EncodeDecimal())
  950. self.assertRaises(UnicodeError, int, u"\u0200")
  951. def test_codecs(self):
  952. # Encoding
  953. self.assertEqual(u'hello'.encode('ascii'), 'hello')
  954. self.assertEqual(u'hello'.encode('utf-7'), 'hello')
  955. self.assertEqual(u'hello'.encode('utf-8'), 'hello')
  956. self.assertEqual(u'hello'.encode('utf8'), 'hello')
  957. self.assertEqual(u'hello'.encode('utf-16-le'), 'h\000e\000l\000l\000o\000')
  958. self.assertEqual(u'hello'.encode('utf-16-be'), '\000h\000e\000l\000l\000o')
  959. self.assertEqual(u'hello'.encode('latin-1'), 'hello')
  960. # Roundtrip safety for BMP (just the first 1024 chars)
  961. for c in xrange(1024):
  962. u = unichr(c)
  963. for encoding in ('utf-7', 'utf-8', 'utf-16', 'utf-16-le',
  964. 'utf-16-be', 'raw_unicode_escape',
  965. 'unicode_escape', 'unicode_internal'):
  966. self.assertEqual(unicode(u.encode(encoding),encoding), u)
  967. # Roundtrip safety for BMP (just the first 256 chars)
  968. for c in xrange(256):
  969. u = unichr(c)
  970. for encoding in ('latin-1',):
  971. self.assertEqual(unicode(u.encode(encoding),encoding), u)
  972. # Roundtrip safety for BMP (just the first 128 chars)
  973. for c in xrange(128):
  974. u = unichr(c)
  975. for encoding in ('ascii',):
  976. self.assertEqual(unicode(u.encode(encoding),encoding), u)
  977. # Roundtrip safety for non-BMP (just a few chars)
  978. u = u'\U00010001\U00020002\U00030003\U00040004\U00050005'
  979. for encoding in ('utf-8', 'utf-16', 'utf-16-le', 'utf-16-be',
  980. #'raw_unicode_escape',
  981. 'unicode_escape', 'unicode_internal'):
  982. self.assertEqual(unicode(u.encode(encoding),encoding), u)
  983. # UTF-8 must be roundtrip safe for all UCS-2 code points
  984. # This excludes surrogates: in the full range, there would be
  985. # a surrogate pair (\udbff\udc00), which gets converted back
  986. # to a non-BMP character (\U0010fc00)
  987. u = u''.join(map(unichr, range(0,0xd800)+range(0xe000,0x10000)))
  988. for encoding in ('utf-8',):
  989. self.assertEqual(unicode(u.encode(encoding),encoding), u)
  990. def test_codecs_charmap(self):
  991. # 0-127
  992. s = ''.join(map(chr, xrange(128)))
  993. for encoding in (
  994. 'cp037', 'cp1026',
  995. 'cp437', 'cp500', 'cp720', 'cp737', 'cp775', 'cp850',
  996. 'cp852', 'cp855', 'cp858', 'cp860', 'cp861', 'cp862',
  997. 'cp863', 'cp865', 'cp866',
  998. 'iso8859_10', 'iso8859_13', 'iso8859_14', 'iso8859_15',
  999. 'iso8859_2', 'iso8859_3', 'iso8859_4', 'iso8859_5', 'iso8859_6',
  1000. 'iso8859_7', 'iso8859_9', 'koi8_r', 'latin_1',
  1001. 'mac_cyrillic', 'mac_latin2',
  1002. 'cp1250', 'cp1251', 'cp1252', 'cp1253', 'cp1254', 'cp1255',
  1003. 'cp1256', 'cp1257', 'cp1258',
  1004. 'cp856', 'cp857', 'cp864', 'cp869', 'cp874',
  1005. 'mac_greek', 'mac_iceland','mac_roman', 'mac_turkish',
  1006. 'cp1006', 'iso8859_8',
  1007. ### These have undefined mappings:
  1008. #'cp424',
  1009. ### These fail the round-trip:
  1010. #'cp875'
  1011. ):
  1012. self.assertEqual(unicode(s, encoding).encode(encoding), s)
  1013. # 128-255
  1014. s = ''.join(map(chr, xrange(128, 256)))
  1015. for encoding in (
  1016. 'cp037', 'cp1026',
  1017. 'cp437', 'cp500', 'cp720', 'cp737', 'cp775', 'cp850',
  1018. 'cp852', 'cp855', 'cp858', 'cp860', 'cp861', 'cp862',
  1019. 'cp863', 'cp865', 'cp866',
  1020. 'iso8859_10', 'iso8859_13', 'iso8859_14', 'iso8859_15',
  1021. 'iso8859_2', 'iso8859_4', 'iso8859_5',
  1022. 'iso8859_9', 'koi8_r', 'latin_1',
  1023. 'mac_cyrillic', 'mac_latin2',
  1024. ### These have undefined mappings:
  1025. #'cp1250', 'cp1251', 'cp1252', 'cp1253', 'cp1254', 'cp1255',
  1026. #'cp1256', 'cp1257', 'cp1258',
  1027. #'cp424', 'cp856', 'cp857', 'cp864', 'cp869', 'cp874',
  1028. #'iso8859_3', 'iso8859_6', 'iso8859_7',
  1029. #'mac_greek', 'mac_iceland','mac_roman', 'mac_turkish',
  1030. ### These fail the round-trip:
  1031. #'cp1006', 'cp875', 'iso8859_8',
  1032. ):
  1033. self.assertEqual(unicode(s, encoding).encode(encoding), s)
  1034. def test_concatenation(self):
  1035. self.assertEqual((u"abc" u"def"), u"abcdef")
  1036. self.assertEqual(("abc" u"def"), u"abcdef")
  1037. self.assertEqual((u"abc" "def"), u"abcdef")
  1038. self.assertEqual((u"abc" u"def" "ghi"), u"abcdefghi")
  1039. self.assertEqual(("abc" "def" u"ghi"), u"abcdefghi")
  1040. def test_printing(self):
  1041. class BitBucket:
  1042. def write(self, text):
  1043. pass
  1044. out = BitBucket()
  1045. print >>out, u'abc'
  1046. print >>out, u'abc', u'def'
  1047. print >>out, u'abc', 'def'
  1048. print >>out, 'abc', u'def'
  1049. print >>out, u'abc\n'
  1050. print >>out, u'abc\n',
  1051. print >>out, u'abc\n',
  1052. print >>out, u'def\n'
  1053. print >>out, u'def\n'
  1054. def test_ucs4(self):
  1055. x = u'\U00100000'
  1056. y = x.encode("raw-unicode-escape").decode("raw-unicode-escape")
  1057. self.assertEqual(x, y)
  1058. y = r'\U00100000'
  1059. x = y.decode("raw-unicode-escape").encode("raw-unicode-escape")
  1060. self.assertEqual(x, y)
  1061. y = r'\U00010000'
  1062. x = y.decode("raw-unicode-escape").encode("raw-unicode-escape")
  1063. self.assertEqual(x, y)
  1064. try:
  1065. '\U11111111'.decode("raw-unicode-escape")
  1066. except UnicodeDecodeError as e:
  1067. self.assertEqual(e.start, 0)
  1068. self.assertEqual(e.end, 10)
  1069. else:
  1070. self.fail("Should have raised UnicodeDecodeError")
  1071. def test_conversion(self):
  1072. # Make sure __unicode__() works properly
  1073. class Foo0:
  1074. def __str__(self):
  1075. return "foo"
  1076. class Foo1:
  1077. def __unicode__(self):
  1078. return u"foo"
  1079. class Foo2(object):
  1080. def __unicode__(self):
  1081. return u"foo"
  1082. class Foo3(object):
  1083. def __unicode__(self):
  1084. return "foo"
  1085. class Foo4(str):
  1086. def __unicode__(self):
  1087. return "foo"
  1088. class Foo5(unicode):
  1089. def __unicode__(self):
  1090. return "foo"
  1091. class Foo6(str):
  1092. def __str__(self):
  1093. return "foos"
  1094. def __unicode__(self):
  1095. return u"foou"
  1096. class Foo7(unicode):
  1097. def __str__(self):
  1098. return "foos"
  1099. def __unicode__(self):
  1100. return u"foou"
  1101. class Foo8(unicode):
  1102. def __new__(cls, content=""):
  1103. return unicode.__new__(cls, 2*content)
  1104. def __unicode__(self):
  1105. return self
  1106. class Foo9(unicode):
  1107. def __str__(self):
  1108. return "string"
  1109. def __unicode__(self):
  1110. return "not unicode"
  1111. self.assertEqual(unicode(Foo0()), u"foo")
  1112. self.assertEqual(unicode(Foo1()), u"foo")
  1113. self.assertEqual(unicode(Foo2()), u"foo")
  1114. self.assertEqual(unicode(Foo3()), u"foo")
  1115. self.assertEqual(unicode(Foo4("bar")), u"foo")
  1116. self.assertEqual(unicode(Foo5("bar")), u"foo")
  1117. self.assertEqual(unicode(Foo6("bar")), u"foou")
  1118. self.assertEqual(unicode(Foo7("bar")), u"foou")
  1119. self.assertEqual(unicode(Foo8("foo")), u"foofoo")
  1120. self.assertIs(type(unicode(Foo8("foo"))), Foo8)
  1121. self.assertEqual(UnicodeSubclass(Foo8("foo")), u"foofoo")
  1122. self.assertIs(type(UnicodeSubclass(Foo8("foo"))), UnicodeSubclass)
  1123. self.assertEqual(str(Foo9("foo")), "string")
  1124. self.assertEqual(unicode(Foo9("foo")), u"not unicode")
  1125. def test_unicode_repr(self):
  1126. class s1:
  1127. def __repr__(self):
  1128. return '\\n'
  1129. class s2:
  1130. def __repr__(self):
  1131. return u'\\n'
  1132. self.assertEqual(repr(s1()), '\\n')
  1133. self.assertEqual(repr(s2()), '\\n')
  1134. # This test only affects 32-bit platforms because expandtabs can only take
  1135. # an int as the max value, not a 64-bit C long. If expandtabs is changed
  1136. # to take a 64-bit long, this test should apply to all platforms.
  1137. @unittest.skipIf(sys.maxint > (1 << 32) or struct.calcsize('P') != 4,
  1138. 'only applies to 32-bit platforms')
  1139. def test_expandtabs_overflows_gracefully(self):
  1140. self.assertRaises(OverflowError, u't\tt\t'.expandtabs, sys.maxint)
  1141. def test__format__(self):
  1142. def test(value, format, expected):
  1143. # test both with and without the trailing 's'
  1144. self.assertEqual(value.__format__(format), expected)
  1145. self.assertEqual(value.__format__(format + u's'), expected)
  1146. test(u'', u'', u'')
  1147. test(u'abc', u'', u'abc')
  1148. test(u'abc', u'.3', u'abc')
  1149. test(u'ab', u'.3', u'ab')
  1150. test(u'abcdef', u'.3', u'abc')
  1151. test(u'abcdef', u'.0', u'')
  1152. test(u'abc', u'3.3', u'abc')
  1153. test(u'abc', u'2.3', u'abc')
  1154. test(u'abc', u'2.2', u'ab')
  1155. test(u'abc', u'3.2', u'ab ')
  1156. test(u'result', u'x<0', u'result')
  1157. test(u'result', u'x<5', u'result')
  1158. test(u'result', u'x<6', u'result')
  1159. test(u'result', u'x<7', u'resultx')
  1160. test(u'result', u'x<8', u'resultxx')
  1161. test(u'result', u' <7', u'result ')
  1162. test(u'result', u'<7', u'result ')
  1163. test(u'result', u'>7', u' result')
  1164. test(u'result', u'>8', u' result')
  1165. test(u'result', u'^8', u' result ')
  1166. test(u'result', u'^9', u' result ')
  1167. test(u'result', u'^10', u' result ')
  1168. test(u'a', u'10000', u'a' + u' ' * 9999)
  1169. test(u'', u'10000', u' ' * 10000)
  1170. test(u'', u'10000000', u' ' * 10000000)
  1171. # test mixing unicode and str
  1172. self.assertEqual(u'abc'.__format__('s'), u'abc')
  1173. self.assertEqual(u'abc'.__format__('->10s'), u'-------abc')
  1174. def test_format(self):
  1175. self.assertEqual(u''.format(), u'')
  1176. self.assertEqual(u'a'.format(), u'a')
  1177. self.assertEqual(u'ab'.format(), u'ab')
  1178. self.assertEqual(u'a{{'.format(), u'a{')
  1179. self.assertEqual(u'a}}'.format(), u'a}')
  1180. self.assertEqual(u'{{b'.format(), u'{b')
  1181. self.assertEqual(u'}}b'.format(), u'}b')
  1182. self.assertEqual(u'a{{b'.format(), u'a{b')
  1183. # examples from the PEP:
  1184. import datetime
  1185. self.assertEqual(u"My name is {0}".format(u'Fred'), u"My name is Fred")
  1186. self.assertEqual(u"My name is {0[name]}".format(dict(name=u'Fred')),
  1187. u"My name is Fred")
  1188. self.assertEqual(u"My name is {0} :-{{}}".format(u'Fred'),
  1189. u"My name is Fred :-{}")
  1190. # datetime.__format__ doesn't work with unicode
  1191. #d = datetime.date(2007, 8, 18)
  1192. #self.assertEqual("The year is {0.year}".format(d),
  1193. # "The year is 2007")
  1194. # classes we'll use for testing
  1195. class C:
  1196. def __init__(self, x=100):
  1197. self._x = x
  1198. def __format__(self, spec):
  1199. return spec
  1200. class D:
  1201. def __init__(self, x):
  1202. self.x = x
  1203. def __format__(self, spec):
  1204. return str(self.x)
  1205. # class with __str__, but no __format__
  1206. class E:
  1207. def __init__(self, x):
  1208. self.x = x
  1209. def __str__(self):
  1210. return u'E(' + self.x + u')'
  1211. # class with __repr__, but no __format__ or __str__
  1212. class F:
  1213. def __init__(self, x):
  1214. self.x = x
  1215. def __repr__(self):
  1216. return u'F(' + self.x + u')'
  1217. # class with __format__ that forwards to string, for some format_spec's
  1218. class G:
  1219. def __init__(self, x):
  1220. self.x = x
  1221. def __str__(self):
  1222. return u"string is " + self.x
  1223. def __format__(self, format_spec):
  1224. if format_spec == 'd':
  1225. return u'G(' + self.x + u')'
  1226. return object.__format__(self, format_spec)
  1227. # class that returns a bad type from __format__
  1228. class H:
  1229. def __format__(self, format_spec):
  1230. return 1.0
  1231. class I(datetime.date):
  1232. def __format__(self, format_spec):
  1233. return self.strftime(format_spec)
  1234. class J(int):
  1235. def __format__(self, format_spec):
  1236. return int.__format__(self * 2, format_spec)
  1237. self.assertEqual(u''.format(), u'')
  1238. self.assertEqual(u'abc'.format(), u'abc')
  1239. self.assertEqual(u'{0}'.format(u'abc'), u'abc')
  1240. self.assertEqual(u'{0:}'.format(u'abc'), u'abc')
  1241. self.assertEqual(u'X{0}'.format(u'abc'), u'Xabc')
  1242. self.assertEqual(u'{0}X'.format(u'abc'), u'abcX')
  1243. self.assertEqual(u'X{0}Y'.format(u'abc'), u'XabcY')
  1244. self.assertEqual(u'{1}'.format(1, u'abc'), u'abc')
  1245. self.assertEqual(u'X{1}'.format(1, u'abc'), u'Xabc')
  1246. self.assertEqual(u'{1}X'.format(1, u'abc'), u'abcX')
  1247. self.assertEqual(u'X{1}Y'.format(1, u'abc'), u'XabcY')
  1248. self.assertEqual(u'{0}'.format(-15), u'-15')
  1249. self.assertEqual(u'{0}{1}'.format(-15, u'abc'), u'-15abc')
  1250. self.assertEqual(u'{0}X{1}'.format(-15, u'abc'), u'-15Xabc')
  1251. self.assertEqual(u'{{'.format(), u'{')
  1252. self.assertEqual(u'}}'.format(), u'}')
  1253. self.assertEqual(u'{{}}'.format(), u'{}')
  1254. self.assertEqual(u'{{x}}'.format(), u'{x}')
  1255. self.assertEqual(u'{{{0}}}'.format(123), u'{123}')
  1256. self.assertEqual(u'{{{{0}}}}'.format(), u'{{0}}')
  1257. self.assertEqual(u'}}{{'.format(), u'}{')
  1258. self.assertEqual(u'}}x{{'.format(), u'}x{')
  1259. # weird field names
  1260. self.assertEqual(u"{0[foo-bar]}".format({u'foo-bar':u'baz'}), u'baz')
  1261. self.assertEqual(u"{0[foo bar]}".format({u'foo bar':u'baz'}), u'baz')
  1262. self.assertEqual(u"{0[ ]}".format({u' ':3}), u'3')
  1263. self.assertEqual(u'{foo._x}'.format(foo=C(20)), u'20')
  1264. self.assertEqual(u'{1}{0}'.format(D(10), D(20)), u'2010')
  1265. self.assertEqual(u'{0._x.x}'.format(C(D(u'abc'))), u'abc')
  1266. self.assertEqual(u'{0[0]}'.format([u'abc', u'def']), u'abc')
  1267. self.assertEqual(u'{0[1]}'.format([u'abc', u'def']), u'def')
  1268. self.assertEqual(u'{0[1][0]}'.format([u'abc', [u'def']]), u'def')
  1269. self.assertEqual(u'{0[1][0].x}'.format(['abc', [D(u'def')]]), u'def')
  1270. # strings
  1271. self.assertEqual(u'{0:.3s}'.format(u'abc'), u'abc')
  1272. self.assertEqual(u'{0:.3s}'.format(u'ab'), u'ab')
  1273. self.assertEqual(u'{0:.3s}'.format(u'abcdef'), u'abc')
  1274. self.assertEqual(u'{0:.0s}'.format(u'abcdef'), u'')
  1275. self.assertEqual(u'{0:3.3s}'.format(u'abc'), u'abc')
  1276. self.assertEqual(u'{0:2.3s}'.format(u'abc'), u'abc')
  1277. self.assertEqual(u'{0:2.2s}'.format(u'abc'), u'ab')
  1278. self.assertEqual(u'{0:3.2s}'.format(u'abc'), u'ab ')
  1279. self.assertEqual(u'{0:x<0s}'.format(u'result'), u'result')
  1280. self.assertEqual(u'{0:x<5s}'.format(u'result'), u'result')
  1281. self.assertEqual(u'{0:x<6s}'.format(u'result'), u'result')
  1282. self.assertEqual(u'{0:x<7s}'.format(u'result'), u'resultx')
  1283. self.assertEqual(u'{0:x<8s}'.format(u'result'), u'resultxx')
  1284. self.assertEqual(u'{0: <7s}'.format(u'result'), u'result ')
  1285. self.assertEqual(u'{0:<7s}'.format(u'result'), u'result ')
  1286. self.assertEqual(u'{0:>7s}'.format(u'result'), u' result')
  1287. self.assertEqual(u'{0:>8s}'.format(u'result'), u' result')
  1288. self.assertEqual(u'{0:^8s}'.format(u'result'), u' result ')
  1289. self.assertEqual(u'{0:^9s}'.format(u'result'), u' result ')
  1290. self.assertEqual(u'{0:^10s}'.format(u'result'), u' result ')
  1291. self.assertEqual(u'{0:10000}'.format(u'a'), u'a' + u' ' * 9999)
  1292. self.assertEqual(u'{0:10000}'.format(u''), u' ' * 10000)
  1293. self.assertEqual(u'{0:10000000}'.format(u''), u' ' * 10000000)
  1294. # issue 12546: use \x00 as a fill character
  1295. self.assertEqual('{0:\x00<6s}'.format('foo'), 'foo\x00\x00\x00')
  1296. self.assertEqual('{0:\x01<6s}'.format('foo'), 'foo\x01\x01\x01')
  1297. self.assertEqual('{0:\x00^6s}'.format('foo'), '\x00foo\x00\x00')
  1298. self.assertEqual('{0:^6s}'.format('foo'), ' foo ')
  1299. self.assertEqual('{0:\x00<6}'.format(3), '3\x00\x00\x00\x00\x00')
  1300. self.assertEqual('{0:\x01<6}'.format(3), '3\x01\x01\x01\x01\x01')
  1301. self.assertEqual('{0:\x00^6}'.format(3), '\x00\x003\x00\x00\x00')
  1302. self.assertEqual('{0:<6}'.format(3), '3 ')
  1303. self.assertEqual('{0:\x00<6}'.format(3.14), '3.14\x00\x00')
  1304. self.assertEqual('{0:\x01<6}'.format(3.14), '3.14\x01\x01')
  1305. self.assertEqual('{0:\x00^6}'.format(3.14), '\x003.14\x00')
  1306. self.assertEqual('{0:^6}'.format(3.14), ' 3.14 ')
  1307. self.assertEqual('{0:\x00<12}'.format(3+2.0j), '(3+2j)\x00\x00\x00\x00\x00\x00')
  1308. self.assertEqual('{0:\x01<12}'.format(3+2.0j), '(3+2j)\x01\x01\x01\x01\x01\x01')
  1309. self.assertEqual('{0:\x00^12}'.format(3+2.0j), '\x00\x00\x00(3+2j)\x00\x00\x00')
  1310. self.assertEqual('{0:^12}'.format(3+2.0j), ' (3+2j) ')
  1311. # format specifiers for user defined type
  1312. self.assertEqual(u'{0:abc}'.format(C()), u'abc')
  1313. # !r and !s coercions
  1314. self.assertEqual(u'{0!s}'.format(u'Hello'), u'Hello')
  1315. self.assertEqual(u'{0!s:}'.format(u'Hello'), u'Hello')
  1316. self.assertEqual(u'{0!s:15}'.format(u'Hello'), u'Hello ')
  1317. self.assertEqual(u'{0!s:15s}'.format(u'Hello'), u'Hello ')
  1318. self.assertEqual(u'{0!r}'.format(u'Hello'), u"u'Hello'")
  1319. self.assertEqual(u'{0!r:}'.format(u'Hello'), u"u'Hello'")
  1320. self.assertEqual(u'{0!r}'.format(F(u'Hello')), u'F(Hello)')
  1321. # test fallback to object.__format__
  1322. self.assertEqual(u'{0}'.format({}), u'{}')
  1323. self.assertEqual(u'{0}'.format([]), u'[]')
  1324. self.assertEqual(u'{0}'.format([1]), u'[1]')
  1325. self.assertEqual(u'{0}'.format(E(u'data')), u'E(data)')
  1326. self.assertEqual(u'{0:d}'.format(G(u'data')), u'G(data)')
  1327. self.assertEqual(u'{0!s}'.format(G(u'data')), u'string is data')
  1328. msg = 'object.__format__ with a non-empty format string is deprecated'
  1329. with test_support.check_warnings((msg, PendingDeprecationWarning)):
  1330. self.assertEqual(u'{0:^10}'.format(E(u'data')), u' E(data) ')
  1331. self.assertEqual(u'{0:^10s}'.format(E(u'data')), u' E(data) ')
  1332. self.assertEqual(u'{0:>15s}'.format(G(u'data')), u' string is data')
  1333. self.assertEqual(u"{0:date: %Y-%m-%d}".format(I(year=2007,
  1334. month=8,
  1335. day=27)),
  1336. u"date: 2007-08-27")
  1337. # test deriving from a builtin type and overriding __format__
  1338. self.assertEqual(u"{0}".format(J(10)), u"20")
  1339. # string format specifiers
  1340. self.assertEqual(u'{0:}'.format('a'), u'a')
  1341. # computed format specifiers
  1342. self.assertEqual(u"{0:.{1}}".format(u'hello world', 5), u'hello')
  1343. self.assertEqual(u"{0:.{1}s}".format(u'hello world', 5), u'hello')
  1344. self.assertEqual(u"{0:.{precision}s}".format('hello world', precision=5), u'hello')
  1345. self.assertEqual(u"{0:{width}.{precision}s}".format('hello world', width=10, precision=5), u'hello ')
  1346. self.assertEqual(u"{0:{width}.{precision}s}".format('hello world', width='10', precision='5'), u'hello ')
  1347. # test various errors
  1348. self.assertRaises(ValueError, u'{'.format)
  1349. self.assertRaises(ValueError, u'}'.format)
  1350. self.assertRaises(ValueError, u'a{'.format)
  1351. self.assertRaises(ValueError, u'a}'.format)
  1352. self.assertRaises(ValueError, u'{a'.format)
  1353. self.assertRaises(ValueError, u'}a'.format)
  1354. self.assertRaises(IndexError, u'{0}'.format)
  1355. self.assertRaises(IndexError, u'{1}'.format, u'abc')
  1356. self.assertRaises(KeyError, u'{x}'.format)
  1357. self.assertRaises(ValueError, u"}{".format)
  1358. self.assertRaises(ValueError, u"{".format)
  1359. self.assertRaises(ValueError, u"}".format)
  1360. self.assertRaises(ValueError, u"abc{0:{}".format)
  1361. self.assertRaises(ValueError, u"{0".format)
  1362. self.assertRaises(IndexError, u"{0.}".format)
  1363. self.assertRaises(ValueError, u"{0.}".format, 0)
  1364. self.assertRaises(IndexError, u"{0[}".format)
  1365. self.assertRaises(ValueError, u"{0[}".format, [])
  1366. self.assertRaises(KeyError, u"{0]}".format)
  1367. self.assertRaises(ValueError, u"{0.[]}".format, 0)
  1368. self.assertRaises(ValueError, u"{0..foo}".format, 0)
  1369. self.assertRaises(ValueError, u"{0[0}".format, 0)
  1370. self.assertRaises(ValueError, u"{0[0:foo}".format, 0)
  1371. self.assertRaises(KeyError, u"{c]}".format)
  1372. self.assertRaises(ValueError, u"{{ {{{0}}".format, 0)
  1373. self.assertRaises(ValueError, u"{0}}".format, 0)
  1374. self.assertRaises(KeyError, u"{foo}".format, bar=3)
  1375. self.assertRaises(ValueError, u"{0!x}".format, 3)
  1376. self.assertRaises(ValueError, u"{0!}".format, 0)
  1377. self.assertRaises(ValueError, u"{0!rs}".format, 0)
  1378. self.assertRaises(ValueError, u"{!}".format)
  1379. self.assertRaises(IndexError, u"{:}".format)
  1380. self.assertRaises(IndexError, u"{:s}".format)
  1381. self.assertRaises(IndexError, u"{}".format)
  1382. big = u"23098475029384702983476098230754973209482573"
  1383. self.assertRaises(ValueError, (u"{" + big + u"}").format)
  1384. self.assertRaises(ValueError, (u"{[" + big + u"]}").format, [0])
  1385. # issue 6089
  1386. self.assertRaises(ValueError, u"{0[0]x}".format, [None])
  1387. self.assertRaises(ValueError, u"{0[0](10)}".format, [None])
  1388. # can't have a replacement on the field name portion
  1389. self.assertRaises(TypeError, u'{0[{1}]}'.format, u'abcdefg', 4)
  1390. # exceed maximum recursion depth
  1391. self.assertRaises(ValueError, u"{0:{1:{2}}}".format, u'abc', u's', u'')
  1392. self.assertRaises(ValueError, u"{0:{1:{2:{3:{4:{5:{6}}}}}}}".format,
  1393. 0, 1, 2, 3, 4, 5, 6, 7)
  1394. # string format spec errors
  1395. self.assertRaises(ValueError, u"{0:-s}".format, u'')
  1396. self.assertRaises(ValueError, format, u"", u"-")
  1397. self.assertRaises(ValueError, u"{0:=s}".format, u'')
  1398. # test combining string and unicode
  1399. self.assertEqual(u"foo{0}".format('bar'), u'foobar')
  1400. # This will try to convert the argument from unicode to str, which
  1401. # will succeed
  1402. self.assertEqual("foo{0}".format(u'bar'), 'foobar')
  1403. # This will try to convert the argument from unicode to str, which
  1404. # will fail
  1405. self.assertRaises(UnicodeEncodeError, "foo{0}".format, u'\u1000bar')
  1406. def test_format_huge_precision(self):
  1407. format_string = u".{}f".format(sys.maxsize + 1)
  1408. with self.assertRaises(ValueError):
  1409. result = format(2.34, format_string)
  1410. def test_format_huge_width(self):
  1411. format_string = u"{}f".format(sys.maxsize + 1)
  1412. with self.assertRaises(ValueError):
  1413. result = format(2.34, format_string)
  1414. def test_format_huge_item_number(self):
  1415. format_string = u"{{{}:.6f}}".format(sys.maxsize + 1)
  1416. with self.assertRaises(ValueError):
  1417. result = format_string.format(2.34)
  1418. def test_format_auto_numbering(self):
  1419. class C:
  1420. def __init__(self, x=100):
  1421. self._x = x
  1422. def __format__(self, spec):
  1423. return spec
  1424. self.assertEqual(u'{}'.format(10), u'10')
  1425. self.assertEqual(u'{:5}'.format('s'), u's ')
  1426. self.assertEqual(u'{!r}'.format('s'), u"'s'")
  1427. self.assertEqual(u'{._x}'.format(C(10)), u'10')
  1428. self.assertEqual(u'{[1]}'.format([1, 2]), u'2')
  1429. self.assertEqual(u'{[a]}'.format({'a':4, 'b':2}), u'4')
  1430. self.assertEqual(u'a{}b{}c'.format(0, 1), u'a0b1c')
  1431. self.assertEqual(u'a{:{}}b'.format('x', '^10'), u'a x b')
  1432. self.assertEqual(u'a{:{}x}b'.format(20, '#'), u'a0x14b')
  1433. # can't mix and match numbering and auto-numbering
  1434. self.assertRaises(ValueError, u'{}{1}'.format, 1, 2)
  1435. self.assertRaises(ValueError, u'{1}{}'.format, 1, 2)
  1436. self.assertRaises(ValueError, u'{:{1}}'.format, 1, 2)
  1437. self.assertRaises(ValueError, u'{0:{}}'.format, 1, 2)
  1438. # can mix and match auto-numbering and named
  1439. self.assertEqual(u'{f}{}'.format(4, f='test'), u'test4')
  1440. self.assertEqual(u'{}{f}'.format(4, f='test'), u'4test')
  1441. self.assertEqual(u'{:{f}}{g}{}'.format(1, 3, g='g', f=2), u' 1g3')
  1442. self.assertEqual(u'{f:{}}{}{g}'.format(2, 4, f=1, g='g'), u' 14g')
  1443. def test_raiseMemError(self):
  1444. # Ensure that the freelist contains a consistent object, even
  1445. # when a string allocation fails with a MemoryError.
  1446. # This used to crash the interpreter,
  1447. # or leak references when the number was smaller.
  1448. charwidth = 4 if sys.maxunicode >= 0x10000 else 2
  1449. # Note: sys.maxsize is half of the actual max allocation because of
  1450. # the signedness of Py_ssize_t.
  1451. alloc = lambda: u"a" * (sys.maxsize // charwidth * 2)
  1452. self.assertRaises(MemoryError, alloc)
  1453. self.assertRaises(MemoryError, alloc)
  1454. def test_format_subclass(self):
  1455. class U(unicode):
  1456. def __unicode__(self):
  1457. return u'__unicode__ overridden'
  1458. u = U(u'xxx')
  1459. self.assertEqual("%s" % u, u'__unicode__ overridden')
  1460. self.assertEqual("{}".format(u), '__unicode__ overridden')
  1461. # Test PyUnicode_FromFormat()
  1462. def test_from_format(self):
  1463. test_support.import_module('ctypes')
  1464. from ctypes import (
  1465. pythonapi, py_object, sizeof,
  1466. c_int, c_long, c_longlong, c_ssize_t,
  1467. c_uint, c_ulong, c_ulonglong, c_size_t, c_void_p)
  1468. if sys.maxunicode == 0xffff:
  1469. name = "PyUnicodeUCS2_FromFormat"
  1470. else:
  1471. name = "PyUnicodeUCS4_FromFormat"
  1472. _PyUnicode_FromFormat = getattr(pythonapi, name)
  1473. _PyUnicode_FromFormat.restype = py_object
  1474. def PyUnicode_FromFormat(format, *args):
  1475. cargs = tuple(
  1476. py_object(arg) if isinstance(arg, unicode) else arg
  1477. for arg in args)
  1478. return _PyUnicode_FromFormat(format, *cargs)
  1479. def check_format(expected, format, *args):
  1480. text = PyUnicode_FromFormat(format, *args)
  1481. self.assertEqual(expected, text)
  1482. # ascii format, non-ascii argument
  1483. check_format(u'ascii\x7f=unicode\xe9',
  1484. b'ascii\x7f=%U', u'unicode\xe9')
  1485. # non-ascii format, ascii argument: ensure that PyUnicode_FromFormatV()
  1486. # raises an error
  1487. #self.assertRaisesRegex(ValueError,
  1488. # '^PyUnicode_FromFormatV\(\) expects an ASCII-encoded format '
  1489. # 'string, got a non-ASCII byte: 0xe9$',
  1490. # PyUnicode_FromFormat, b'unicode\xe9=%s', u'ascii')
  1491. # test "%c"
  1492. check_format(u'\uabcd',
  1493. b'%c', c_int(0xabcd))
  1494. if sys.maxunicode > 0xffff:
  1495. check_format(u'\U0010ffff',
  1496. b'%c', c_int(0x10ffff))
  1497. else:
  1498. with self.assertRaises(OverflowError):
  1499. PyUnicode_FromFormat(b'%c', c_int(0x10000))
  1500. with self.assertRaises(OverflowError):
  1501. PyUnicode_FromFormat(b'%c', c_int(0x110000))
  1502. # Issue #18183
  1503. if sys.maxunicode > 0xffff:
  1504. check_format(u'\U00010000\U00100000',
  1505. b'%c%c', c_int(0x10000), c_int(0x100000))
  1506. # test "%"
  1507. check_format(u'%',
  1508. b'%')
  1509. check_format(u'%',
  1510. b'%%')
  1511. check_format(u'%s',
  1512. b'%%s')
  1513. check_format(u'[%]',
  1514. b'[%%]')
  1515. check_format(u'%abc',
  1516. b'%%%s', b'abc')
  1517. # test %S
  1518. check_format(u"repr=abc",
  1519. b'repr=%S', u'abc')
  1520. # test %R
  1521. check_format(u"repr=u'abc'",
  1522. b'repr=%R', u'abc')
  1523. # test integer formats (%i, %d, %u)
  1524. check_format(u'010',
  1525. b'%03i', c_int(10))
  1526. check_format(u'0010',
  1527. b'%0.4i', c_int(10))
  1528. check_format(u'-123',
  1529. b'%i', c_int(-123))
  1530. check_format(u'-123',
  1531. b'%d', c_int(-123))
  1532. check_format(u'-123',
  1533. b'%ld', c_long(-123))
  1534. check_format(u'-123',
  1535. b'%zd', c_ssize_t(-123))
  1536. check_format(u'123',
  1537. b'%u', c_uint(123))
  1538. check_format(u'123',
  1539. b'%lu', c_ulong(123))
  1540. check_format(u'123',
  1541. b'%zu', c_size_t(123))
  1542. # test long output
  1543. min_long = -(2 ** (8 * sizeof(c_long) - 1))
  1544. max_long = -min_long - 1
  1545. check_format(unicode(min_long),
  1546. b'%ld', c_long(min_long))
  1547. check_format(unicode(max_long),
  1548. b'%ld', c_long(max_long))
  1549. max_ulong = 2 ** (8 * sizeof(c_ulong)) - 1
  1550. check_format(unicode(max_ulong),
  1551. b'%lu', c_ulong(max_ulong))
  1552. PyUnicode_FromFormat(b'%p', c_void_p(-1))
  1553. # test padding (width and/or precision)
  1554. check_format(u'123'.rjust(10, u'0'),
  1555. b'%010i', c_int(123))
  1556. check_format(u'123'.rjust(100),
  1557. b'%100i', c_int(123))
  1558. check_format(u'123'.rjust(100, u'0'),
  1559. b'%.100i', c_int(123))
  1560. check_format(u'123'.rjust(80, u'0').rjust(100),
  1561. b'%100.80i', c_int(123))
  1562. check_format(u'123'.rjust(10, u'0'),
  1563. b'%010u', c_uint(123))
  1564. check_format(u'123'.rjust(100),
  1565. b'%100u', c_uint(123))
  1566. check_format(u'123'.rjust(100, u'0'),
  1567. b'%.100u', c_uint(123))
  1568. check_format(u'123'.rjust(80, u'0').rjust(100),
  1569. b'%100.80u', c_uint(123))
  1570. check_format(u'123'.rjust(10, u'0'),
  1571. b'%010x', c_int(0x123))
  1572. check_format(u'123'.rjust(100),
  1573. b'%100x', c_int(0x123))
  1574. check_format(u'123'.rjust(100, u'0'),
  1575. b'%.100x', c_int(0x123))
  1576. check_format(u'123'.rjust(80, u'0').rjust(100),
  1577. b'%100.80x', c_int(0x123))
  1578. # test %V
  1579. check_format(u'repr=abc',
  1580. b'repr=%V', u'abc', b'xyz')
  1581. check_format(u'repr=\xe4\xba\xba\xe6\xb0\x91',
  1582. b'repr=%V', None, b'\xe4\xba\xba\xe6\xb0\x91')
  1583. check_format(u'repr=abc\xff',
  1584. b'repr=%V', None, b'abc\xff')
  1585. # not supported: copy the raw format string. these tests are just here
  1586. # to check for crashs and should not be considered as specifications
  1587. check_format(u'%s',
  1588. b'%1%s', b'abc')
  1589. check_format(u'%1abc',
  1590. b'%1abc')
  1591. check_format(u'%+i',
  1592. b'%+i', c_int(10))
  1593. check_format(u'%s',
  1594. b'%.%s', b'abc')
  1595. @test_support.cpython_only
  1596. def test_encode_decimal(self):
  1597. from _testcapi import unicode_encodedecimal
  1598. self.assertEqual(unicode_encodedecimal(u'123'),
  1599. b'123')
  1600. self.assertEqual(unicode_encodedecimal(u'\u0663.\u0661\u0664'),
  1601. b'3.14')
  1602. self.assertEqual(unicode_encodedecimal(u"\N{EM SPACE}3.14\N{EN SPACE}"),
  1603. b' 3.14 ')
  1604. self.assertRaises(UnicodeEncodeError,
  1605. unicode_encodedecimal, u"123\u20ac", "strict")
  1606. self.assertEqual(unicode_encodedecimal(u"123\u20ac", "replace"),
  1607. b'123?')
  1608. self.assertEqual(unicode_encodedecimal(u"123\u20ac", "ignore"),
  1609. b'123')
  1610. self.assertEqual(unicode_encodedecimal(u"123\u20ac", "xmlcharrefreplace"),
  1611. b'123&#8364;')
  1612. self.assertEqual(unicode_encodedecimal(u"123\u20ac", "backslashreplace"),
  1613. b'123\\u20ac')
  1614. self.assertEqual(unicode_encodedecimal(u"123\u20ac\N{EM SPACE}", "replace"),
  1615. b'123? ')
  1616. self.assertEqual(unicode_encodedecimal(u"123\u20ac\u20ac", "replace"),
  1617. b'123??')
  1618. self.assertEqual(unicode_encodedecimal(u"123\u20ac\u0660", "replace"),
  1619. b'123?0')
  1620. @test_support.cpython_only
  1621. def test_encode_decimal_with_surrogates(self):
  1622. from _testcapi import unicode_encodedecimal
  1623. tests = [(u'\U0001f49d', '&#128157;'),
  1624. (u'\ud83d', '&#55357;'),
  1625. (u'\udc9d', '&#56477;'),
  1626. ]
  1627. if u'\ud83d\udc9d' != u'\U0001f49d':
  1628. tests += [(u'\ud83d\udc9d', '&#55357;&#56477;')]
  1629. for s, exp in tests:
  1630. self.assertEqual(
  1631. unicode_encodedecimal(u"123" + s, "xmlcharrefreplace"),
  1632. '123' + exp)
  1633. def test_free_after_iterating(self):
  1634. test_support.check_free_after_iterating(self, iter, unicode)
  1635. test_support.check_free_after_iterating(self, reversed, unicode)
  1636. def test_main():
  1637. test_support.run_unittest(__name__)
  1638. if __name__ == "__main__":
  1639. test_main()