PageRenderTime 54ms CodeModel.GetById 20ms RepoModel.GetById 0ms app.codeStats 0ms

/IronPython_Main/Languages/IronPython/Tests/modules/io_related/re_test.py

#
Python | 808 lines | 734 code | 38 blank | 36 comment | 11 complexity | 5aa86e63ff74d5633a52da6de0413221 MD5 | raw file
Possible License(s): GPL-2.0, MPL-2.0-no-copyleft-exception, CPL-1.0, CC-BY-SA-3.0, BSD-3-Clause, ISC, AGPL-3.0, LGPL-2.1, Apache-2.0
  1. #####################################################################################
  2. #
  3. # Copyright (c) Microsoft Corporation. All rights reserved.
  4. #
  5. # This source code is subject to terms and conditions of the Apache License, Version 2.0. A
  6. # copy of the license can be found in the License.html file at the root of this distribution. If
  7. # you cannot locate the Apache License, Version 2.0, please send an email to
  8. # ironpy@microsoft.com. By using this source code in any fashion, you are agreeing to be bound
  9. # by the terms of the Apache License, Version 2.0.
  10. #
  11. # You must not remove this notice, or any other, from this software.
  12. #
  13. #
  14. #####################################################################################
  15. from iptest.assert_util import *
  16. import re
  17. def test_none():
  18. for x in 'compile search match split findall finditer'.split():
  19. y = getattr(re, x)
  20. AssertError(TypeError, y, None)
  21. AssertError(TypeError, y, None, None)
  22. AssertError(TypeError, y, None, 'abc')
  23. AssertError(TypeError, y, 'abc', None)
  24. # Other exceptional input tests
  25. for x in (re.sub, re.subn):
  26. AssertError(TypeError, x, 'abc', None, 'abc')
  27. AssertError(TypeError, x, 'abc', None, None)
  28. AssertError(TypeError, x, None, 'abc', 'abc')
  29. AssertError(TypeError, x, 'abc', 'abc', None)
  30. AssertError(TypeError, re.escape, None)
  31. def test_sanity_re():
  32. '''
  33. Basic sanity tests for the re module. Each module member is
  34. used at least once.
  35. '''
  36. #compile
  37. Assert(hasattr(re.compile("(abc){1}"), "pattern"))
  38. Assert(hasattr(re.compile("(abc){1}", re.L), "pattern"))
  39. Assert(hasattr(re.compile("(abc){1}", flags=re.L), "pattern"))
  40. #I IGNORECASE L LOCAL MMULTILINE S DOTALL U UNICODE X VERBOSE
  41. flags = ["I", "IGNORECASE",
  42. "L", "LOCALE",
  43. "M", "MULTILINE",
  44. "S", "DOTALL",
  45. "U", "UNICODE",
  46. "X", "VERBOSE"]
  47. for f in flags:
  48. Assert(hasattr(re, f))
  49. #search
  50. AreEqual(re.search("(abc){1}", ""), None)
  51. AreEqual(re.search("(abc){1}", "abcxyz").span(), (0,3))
  52. AreEqual(re.search("(abc){1}", "abcxyz", re.L).span(), (0,3))
  53. AreEqual(re.search("(abc){1}", "abcxyz", flags=re.L).span(), (0,3))
  54. AreEqual(re.search("(abc){1}", "xyzabc").span(), (3,6))
  55. AreEqual(re.search("(abc){1}", buffer("")), None)
  56. AreEqual(re.search("(abc){1}", buffer("abcxyz")).span(), (0,3))
  57. AreEqual(re.search("(abc){1}", buffer("abcxyz"), re.L).span(), (0,3))
  58. AreEqual(re.search("(abc){1}", buffer("abcxyz"), flags=re.L).span(), (0,3))
  59. AreEqual(re.search("(abc){1}", buffer("xyzabc")).span(), (3,6))
  60. #match
  61. AreEqual(re.match("(abc){1}", ""), None)
  62. AreEqual(re.match("(abc){1}", "abcxyz").span(), (0,3))
  63. AreEqual(re.match("(abc){1}", "abcxyz", re.L).span(), (0,3))
  64. AreEqual(re.match("(abc){1}", "abcxyz", flags=re.L).span(), (0,3))
  65. #split
  66. AreEqual(re.split("(abc){1}", ""), [''])
  67. AreEqual(re.split("(abc){1}", "abcxyz"), ['', 'abc', 'xyz'])
  68. #maxsplit
  69. AreEqual(re.split("(abc){1}", "abc", 0), ['', 'abc', ''])
  70. for i in xrange(3):
  71. AreEqual(re.split("(abc){1}", "abc", maxsplit=i), ['', 'abc', ''])
  72. AreEqual(re.split("(abc){1}", "", maxsplit=i), [''])
  73. AreEqual(re.split("(abc){1}", "abcxyz", maxsplit=i), ['', 'abc', 'xyz'])
  74. AreEqual(re.split("(abc){1}", "abcxyzabc", maxsplit=0), ['', 'abc', 'xyz', 'abc', ''])
  75. AreEqual(re.split("(abc){1}", "abcxyzabc", maxsplit=1), ['', 'abc', 'xyzabc'])
  76. AreEqual(re.split("(abc){1}", "abcxyzabc", maxsplit=2), ['', 'abc', 'xyz', 'abc', ''])
  77. #findall
  78. AreEqual(re.findall("(abc){1}", ""), [])
  79. AreEqual(re.findall("(abc){1}", "abcxyz"), ['abc'])
  80. AreEqual(re.findall("(abc){1}", "abcxyz", re.L), ['abc'])
  81. AreEqual(re.findall("(abc){1}", "abcxyz", flags=re.L), ['abc'])
  82. AreEqual(re.findall("(abc){1}", "xyzabcabc"), ['abc', 'abc'])
  83. #finditer
  84. AreEqual([x.group() for x in re.finditer("(abc){1}", "")], [])
  85. AreEqual([x.group() for x in re.finditer("(abc){1}", "abcxyz")], ['abc'])
  86. AreEqual([x.group() for x in re.finditer("(abc){1}", "abcxyz", re.L)], ['abc'])
  87. AreEqual([x.group() for x in re.finditer("(abc){1}", "abcxyz", flags=re.L)], ['abc'])
  88. AreEqual([x.group() for x in re.finditer("(abc){1}", "xyzabcabc")], ['abc', 'abc'])
  89. rex = re.compile("foo")
  90. for m in rex.finditer("this is a foo and a foo bar"):
  91. AreEqual((m.pos, m.endpos), (0, 27))
  92. for m in rex.finditer(""):
  93. AreEqual((m.pos, m.endpos), (0, 1))
  94. for m in rex.finditer("abc"):
  95. AreEqual((m.pos, m.endpos), (0, 4))
  96. for m in rex.finditer("foo foo foo foo foo"):
  97. AreEqual((m.pos, m.endpos), (0, 19))
  98. #sub
  99. AreEqual(re.sub("(abc){1}", "9", "abcd"), "9d")
  100. AreEqual(re.sub("(abc){1}", "abcxyz",'abcd'), "abcxyzd")
  101. AreEqual(re.sub("(abc){1}", "1", "abcd", 0), "1d")
  102. AreEqual(re.sub("(abc){1}", "1", "abcd", count=0), "1d")
  103. AreEqual(re.sub("(abc){1}", "1", "abcdabcd", 1), "1dabcd")
  104. AreEqual(re.sub("(abc){1}", "1", "abcdabcd", 2), "1d1d")
  105. #subn
  106. AreEqual(re.subn("(abc){1}", "9", "abcd"), ("9d", 1))
  107. AreEqual(re.subn("(abc){1}", "abcxyz",'abcd'), ("abcxyzd",1))
  108. AreEqual(re.subn("(abc){1}", "1", "abcd", 0), ("1d",1))
  109. AreEqual(re.subn("(abc){1}", "1", "abcd", count=0), ("1d",1))
  110. AreEqual(re.subn("(abc){1}", "1", "abcdabcd", 1), ("1dabcd",1))
  111. AreEqual(re.subn("(abc){1}", "1", "abcdabcd", 2), ("1d1d",2))
  112. #escape
  113. AreEqual(re.escape("abc"), "abc")
  114. AreEqual(re.escape(""), "")
  115. AreEqual(re.escape("_"), "\\_")
  116. AreEqual(re.escape("a_c"), "a\\_c")
  117. #error
  118. exc = re.error()
  119. exc = re.error("some args")
  120. #purge
  121. re.purge()
  122. def test_sanity_re_pattern():
  123. '''
  124. Basic sanity tests for the re module's Regular Expression
  125. objects (i.e., Pattern in CPython). Each method/member is
  126. utilized at least once.
  127. '''
  128. pattern = re.compile("(abc){1}")
  129. #match
  130. AreEqual(pattern.match(""), None)
  131. AreEqual(pattern.match("abcxyz").span(), (0,3))
  132. AreEqual(pattern.match("abc", 0).span(), (0,3))
  133. AreEqual(pattern.match("abc", 0, 3).span(), (0,3))
  134. AreEqual(pattern.match("abc", pos=0, endpos=3).span(), (0,3))
  135. for i in [-1, -2, -5, -7, -8, -65536]:
  136. for j in [3, 4, 5, 7, 8, 65536]:
  137. AreEqual(pattern.match("abc", i, j).span(), (0,3))
  138. AssertError(OverflowError, lambda: pattern.match("abc", 0, 2**32).span())
  139. AssertError(OverflowError, lambda: pattern.match("abc", -(2**32), 3).span())
  140. #search
  141. AreEqual(pattern.search(""), None)
  142. AreEqual(pattern.search("abcxyz").span(), (0,3))
  143. AreEqual(pattern.search("abc", 0).span(), (0,3))
  144. AreEqual(pattern.search("abc", 0, 3).span(), (0,3))
  145. AreEqual(pattern.search("abc", pos=0, endpos=3).span(), (0,3))
  146. AreEqual(pattern.search("xyzabc").span(), (3,6))
  147. #split
  148. AreEqual(pattern.split(""), [''])
  149. AreEqual(pattern.split("abcxyz"), ['', 'abc', 'xyz'])
  150. AreEqual(pattern.split("abc", 0), ['', 'abc', ''])
  151. AreEqual(pattern.split("abc", maxsplit=0), ['', 'abc', ''])
  152. AreEqual(pattern.split("abcxyzabc", maxsplit=1), ['', 'abc', 'xyzabc'])
  153. #findall
  154. AreEqual(pattern.findall(""), [])
  155. AreEqual(pattern.findall("abcxyz"), ['abc'])
  156. AreEqual(pattern.findall("abc", 0), ['abc'])
  157. AreEqual(pattern.findall("abc", 0, 3), ['abc'])
  158. AreEqual(pattern.findall("abc", pos=0, endpos=3), ['abc'])
  159. AreEqual(pattern.findall("xyzabcabc"), ['abc', 'abc'])
  160. #sub
  161. AreEqual(pattern.sub("9", "abcd"), "9d")
  162. AreEqual(pattern.sub("abcxyz",'abcd'), "abcxyzd")
  163. AreEqual(pattern.sub("1", "abcd", 0), "1d")
  164. AreEqual(pattern.sub("1", "abcd", count=0), "1d")
  165. AreEqual(pattern.sub("1", "abcdabcd", 1), "1dabcd")
  166. AreEqual(pattern.sub("1", "abcdabcd", 2), "1d1d")
  167. #subn
  168. AreEqual(pattern.subn("9", "abcd"), ("9d", 1))
  169. AreEqual(pattern.subn("abcxyz",'abcd'), ("abcxyzd",1))
  170. AreEqual(pattern.subn("1", "abcd", 0), ("1d",1))
  171. AreEqual(pattern.subn("1", "abcd", count=0), ("1d",1))
  172. AreEqual(pattern.subn("1", "abcdabcd", 1), ("1dabcd",1))
  173. AreEqual(pattern.subn("1", "abcdabcd", 2), ("1d1d",2))
  174. #flags
  175. AreEqual(pattern.flags, 0)
  176. AreEqual(re.compile("(abc){1}", re.L).flags, re.L)
  177. #groupindex
  178. AreEqual(pattern.groupindex, {})
  179. AreEqual(re.compile("(?P<abc>)(?P<bcd>)").groupindex, {'bcd': 2, 'abc': 1})
  180. #pattern
  181. AreEqual(pattern.pattern, "(abc){1}")
  182. AreEqual(re.compile("").pattern, "")
  183. def test_groupindex_empty():
  184. test_list = [ ".", "^", "$", "1*", "2+", "3?", "4*?", "5+?", "6??", "7{1}", "8{1,2}",
  185. "9{1,2}?", "[a-z]", "|", "(...)", "(?:abc)",
  186. "\(\?P\<Blah\>abc\)", "(?#...)", "(?=...)", "(?!...)", "(?<=...)",
  187. "(?<!...)", "\1", "\A", "\d"
  188. ]
  189. for x in test_list:
  190. AreEqual(re.compile(x).groupindex, {})
  191. def test_sanity_re_match():
  192. '''
  193. Basic sanity tests for the re module's Match objects. Each method/member
  194. is utilized at least once.
  195. '''
  196. pattern = re.compile("(abc){1}")
  197. match_obj = pattern.match("abcxyzabc123 and some other words...")
  198. #expand
  199. AreEqual(match_obj.expand("\1\g<1>.nt"), '\x01abc.nt')
  200. #group
  201. AreEqual(match_obj.group(), 'abc')
  202. AreEqual(match_obj.group(1), 'abc')
  203. #groups
  204. AreEqual(match_obj.groups(), ('abc',))
  205. AreEqual(match_obj.groups(1), ('abc',))
  206. AreEqual(match_obj.groups(99), ('abc',))
  207. #groupdict
  208. AreEqual(match_obj.groupdict(), {})
  209. AreEqual(match_obj.groupdict(None), {})
  210. AreEqual(re.compile("(abc)").match("abcxyzabc123 and...").groupdict(), {})
  211. #start
  212. AreEqual(match_obj.start(), 0)
  213. AreEqual(match_obj.start(1), 0)
  214. #end
  215. AreEqual(match_obj.end(), 3)
  216. AreEqual(match_obj.end(1), 3)
  217. #span
  218. AreEqual(match_obj.span(), (0,3))
  219. AreEqual(match_obj.span(1), (0,3))
  220. #pos
  221. AreEqual(match_obj.pos, 0)
  222. #endpos
  223. AreEqual(match_obj.endpos, 36)
  224. #lastindex
  225. AreEqual(match_obj.lastindex, 1)
  226. #lastgroup
  227. #CodePlex Work Item 5518
  228. #AreEqual(match_obj.lastgroup, None)
  229. #re
  230. Assert(match_obj.re==pattern)
  231. #string
  232. AreEqual(match_obj.string, "abcxyzabc123 and some other words...")
  233. def test_comment():
  234. '''
  235. (?#...)
  236. '''
  237. pattern = "a(?#foo)bc"
  238. c = re.compile(pattern)
  239. AreEqual(c.findall("abc"), ['abc'])
  240. pattern = "a(?#)bc"
  241. c = re.compile(pattern)
  242. AreEqual(c.findall("abc"), ['abc'])
  243. pattern = "a(?#foo)bdc"
  244. c = re.compile(pattern)
  245. AreEqual(len(c.findall("abc")), 0)
  246. def test_optional_paren():
  247. pattern = r"""\(?\w+\)?"""
  248. c = re.compile(pattern, re.X)
  249. AreEqual(c.findall('abc'), ['abc'])
  250. def test_back_match():
  251. p = re.compile('(?P<grp>.+?)(?P=grp)')
  252. AreEqual(p.match('abcabc').groupdict(), {'grp':'abc'})
  253. p = re.compile(r'(?P<delim>[%$])(?P<escaped>(?P=delim))')
  254. AreEqual(p.match('$$').groupdict(), {'escaped': '$', 'delim': '$'})
  255. AreEqual(p.match('$%'), None)
  256. p = re.compile(r'(?P<grp>ab)(a(?P=grp)b)')
  257. AreEqual(p.match('abaabb').groups(), ('ab', 'aabb'))
  258. def test_expand():
  259. AreEqual(re.match("(a)(b)", "ab").expand("blah\g<1>\g<2>"), "blahab")
  260. AreEqual(re.match("(a)()", "ab").expand("blah\g<1>\g<2>\n\r\t\\\\"),'blaha\n\r\t\\')
  261. AreEqual(re.match("(a)()", "ab").expand(""),'')
  262. def test_sub():
  263. x = '\n #region Generated Foo\nblah\nblah#end region'
  264. a = re.compile("^([ \t]+)#region Generated Foo.*?#end region", re.MULTILINE|re.DOTALL)
  265. AreEqual(a.sub("xx", x), "\nxx") # should match successfully
  266. AreEqual(a.sub("\\x12", x), "\n\\x12") # should match, but shouldn't un-escape for \x
  267. #if optional count arg is 0 then all occurrences should be replaced
  268. AreEqual('bbbb', re.sub("a","b","abab", 0))
  269. AreEqual(re.sub(r'(?P<id>b)', '\g<id>\g<id>yadayada', 'bb'), 'bbyadayadabbyadayada')
  270. AreEqual(re.sub(r'(?P<id>b)', '\g<1>\g<id>yadayada', 'bb'), 'bbyadayadabbyadayada')
  271. AssertError(IndexError, re.sub, r'(?P<id>b)', '\g<1>\g<i2>yadayada', 'bb')
  272. # the native implementation just gives a sre_constants.error instead indicating an invalid
  273. # group reference
  274. if is_cli:
  275. AssertError(IndexError, re.sub, r'(?P<id>b)', '\g<1>\g<30>yadayada', 'bb')
  276. AreEqual(re.sub('x*', '-', 'abc'), '-a-b-c-')
  277. AreEqual(re.subn('x*', '-', 'abc'), ('-a-b-c-', 4))
  278. AreEqual(re.sub('a*', '-', 'abc'), '-b-c-')
  279. AreEqual(re.subn('a*', '-', 'abc'), ('-b-c-', 3))
  280. AreEqual(re.sub('a*', '-', 'a'), '-')
  281. AreEqual(re.subn('a*', '-', 'a'), ('-', 1))
  282. AreEqual(re.sub("a*", "-", "abaabb"), '-b-b-b-')
  283. AreEqual(re.subn("a*", "-", "abaabb"), ('-b-b-b-', 4))
  284. AreEqual(re.sub("(a*)b", "-", "abaabb"), '---')
  285. AreEqual(re.subn("(a*)b", "-", "abaabb"), ('---', 3))
  286. AreEqual(re.subn("(ab)*", "cd", "abababababab", 10), ('cd', 1))
  287. AreEqual(re.sub('x*', '-', 'abxd'), '-a-b-d-')
  288. AreEqual(re.subn('x*', '-', 'abxd'), ('-a-b-d-', 4))
  289. Assert(re.sub('([^aeiou])y$', r'\lies', 'vacancy') == 'vacan\\lies')
  290. Assert(re.sub('([^aeiou])y$', r'\1ies', 'vacancy') == 'vacancies')
  291. AreEqual(re.sub("a+", "\n\t\\\?\"\b", "abc"), '\n\t\\?"\x08bc')
  292. AreEqual(re.sub("a+", r"\n\t\\\?\"\b", "abc"), '\n\t\\\\?\\"\x08bc')
  293. AreEqual(re.sub("a+", "\n\t\\\\\\?\"\b", "abc"), '\n\t\\\\?"\x08bc')
  294. def test_dot():
  295. a = re.compile('.')
  296. AreEqual(a.groupindex, {})
  297. p = re.compile('.')
  298. z = []
  299. for c in p.finditer('abc'): z.append((c.start(), c.end()))
  300. z.sort()
  301. AreEqual(z, [(0,1), (1,2), (2,3)])
  302. def test_x():
  303. nonmatchingp = re.compile('x')
  304. AreEqual(nonmatchingp.search('ecks', 1, 4), None)
  305. def test_match():
  306. p = re.compile('.')
  307. AreEqual(p.match('bazbar', 1,2).span(), (1,2))
  308. def test_span():
  309. AreEqual(re.match('(baz)(bar)(m)', "bazbarmxyz").span(2),(3, 6))
  310. def test_regs():
  311. AreEqual(re.match('(baz)(bar)(m)', "bazbarmxyz").regs,
  312. ((0, 7), (0, 3), (3, 6), (6, 7)))
  313. AreEqual(re.match('bazbar(mm)+(abc)(xyz)', "bazbarmmmmabcxyz123456abc").regs,
  314. ((0, 16), (8, 10), (10, 13), (13, 16)))
  315. def test_endpos():
  316. AreEqual(re.match('(baz)(bar)(m)', "bazbarmx").endpos, 8)
  317. pass
  318. def test_re():
  319. #Just ensure it's there for now
  320. stuff = re.match('a(baz)(bar)(m)', "abazbarmx")
  321. Assert(hasattr(stuff, "re"))
  322. Assert(hasattr(stuff.re, "sub"))
  323. def test_pos():
  324. AreEqual(re.match('(baz)(bar)(m)', "bazbarmx").pos, 0)
  325. def test_startandend():
  326. m = re.match(r'(a)|(b)', 'b')
  327. AreEqual(m.groups(), (None, 'b'))
  328. AreEqual(m.group(0), "b")
  329. AreEqual(m.start(0), 0)
  330. AreEqual(m.end(0), 1)
  331. AreEqual(m.start(1), -1)
  332. AreEqual(m.end(1), -1)
  333. m = re.match(".*", '')
  334. AreEqual(m.groups(), ())
  335. AreEqual(m.start(0), 0)
  336. AreEqual(m.end(0), 0)
  337. AssertError(IndexError, m.group, "112")
  338. AssertError(IndexError, m.group, 112)
  339. AssertError(IndexError, m.group, "-1")
  340. AssertError(IndexError, m.group, -1)
  341. AssertError(IndexError, m.start, 112)
  342. AssertError(IndexError, m.start, -1)
  343. AssertError(IndexError, m.end, "112")
  344. AssertError(IndexError, m.end, 112)
  345. AssertError(IndexError, m.end, "-1")
  346. AssertError(IndexError, m.end, -1)
  347. match = re.match(r'(?P<test>test)', 'test')
  348. AreEqual(match.start('test'), 0)
  349. AreEqual(match.end('test'), 4)
  350. def test_start_of_str():
  351. startOfStr = re.compile('^')
  352. AreEqual(startOfStr.match('bazbar', 1), None)
  353. AreEqual(startOfStr.match('bazbar', 0,0).span(), (0,0))
  354. AreEqual(startOfStr.match('bazbar', 1,2), None)
  355. AreEqual(startOfStr.match('bazbar', endpos=3).span(), (0,0))
  356. AreEqual(re.sub('^', 'x', ''), 'x')
  357. AreEqual(re.sub('^', 'x', ' '), 'x ')
  358. AreEqual(re.sub('^', 'x', 'abc'), 'xabc')
  359. # check that groups in split RE are added properly
  360. def test_split():
  361. AreEqual(re.split('{(,)?}', '1 {} 2 {,} 3 {} 4'), ['1 ', None, ' 2 ', ',', ' 3 ', None, ' 4'])
  362. pnogrp = ','
  363. ptwogrp = '((,))'
  364. csv = '0,1,1,2,3,5,8,13,21,44'
  365. AreEqual(re.split(pnogrp, csv, 1), ['0', csv[2:]])
  366. AreEqual(re.split(pnogrp, csv, 2), ['0','1', csv[4:]])
  367. AreEqual(re.split(pnogrp, csv, 1000), re.split(pnogrp, csv))
  368. AreEqual(re.split(pnogrp, csv, 0), re.split(pnogrp, csv))
  369. AreEqual(re.split(pnogrp, csv, -1), [csv])
  370. ponegrp = '(,)'
  371. AreEqual(re.split(ponegrp, csv, 1), ['0', ',', csv[2:]])
  372. def test_escape():
  373. compiled = re.compile(re.escape("hi_"))
  374. all = re.compile('(.*)')
  375. AreEqual(all.search('abcdef', 3).group(0), 'def')
  376. AssertError(IndexError, re.match("a[bcd]*b", 'abcbd').group, 1)
  377. AreEqual(re.match('(a[bcd]*b)', 'abcbd').group(1), 'abcb')
  378. s = ''
  379. for i in range(32, 128):
  380. if not chr(i).isalnum():
  381. s = s + chr(i)
  382. x = re.escape(s)
  383. Assert(x == '\\ \\!\\"\\#\\$\\%\\&\\\'\\(\\)\\*\\+\\,\\-\\.\\/\\:\\;\\<\\=\\>\\?\\@\\[\\\\\\]\\^\\_\\`\\{\\|\\}\\~\\\x7f')
  384. x = re.compile(r'[\\A-Z\.\+]')
  385. Assert(x.search('aaaA\\B\\Caaa'))
  386. # From the docs: "^" matches only at the start of the string, or in MULTILINE mode also immediately
  387. # following a newline.
  388. m = re.compile("a").match("ba", 1) # succeed
  389. AreEqual('a', m.group(0))
  390. # bug 23668
  391. #AreEqual(re.compile("^a").search("ba", 1), None) # fails; 'a' not at start
  392. #AreEqual(re.compile("^a").search("\na", 1), None) # fails; 'a' not at start
  393. m = re.compile("^a", re.M).search("\na", 1) # succeed (multiline)
  394. AreEqual('a', m.group(0))
  395. # bug 938
  396. #AreEqual(re.compile("^a", re.M).search("ba", 1), None) # fails; no preceding \n
  397. # findall
  398. def test_findall():
  399. for (x, y, z) in (
  400. ('\d+', '99 blahblahblah 183 blah 12 blah 7777 yada yada', ['99', '183', '12', '7777']),
  401. ('^\d+', '0blahblahblah blah blah yada yada1', ['0']),
  402. ('^\d+', 'blahblahblah blah blah yada yada1', []),
  403. ("(\d+)|(\w+)", "x = 999y + 23", [('', 'x'), ('999', ''), ('', 'y'), ('23', '')]),
  404. ("(\d)(\d\d)(\d\d\d)", "123456789123456789", [('1', '23', '456'), ('7', '89', '123'), ('4', '56', '789')]),
  405. (r"(?i)(\w+)\s+fish\b", "green fish black fish red fish blue fish", ['green', 'black', 'red', 'blue']),
  406. ('(a)(b)', 'abab', [('a', 'b'), ('a', 'b')]),
  407. ):
  408. AreEqual(re.findall(x, y), z)
  409. AreEqual(re.compile(x).findall(y), z)
  410. def test_match_groups():
  411. m = re.match('(?P<test>a)(b)', 'ab')
  412. Assert(m.groups() == ('a', 'b'))
  413. m = re.match('(u)(?P<test>v)(b)(?P<Named2>w)(x)(y)', 'uvbwxy')
  414. Assert(m.groups() == ('u', 'v', 'b', 'w', 'x', 'y'))
  415. def test_options():
  416. # coverage for ?iLmsux options in re.compile path
  417. tests = [ ("t(?=s)", "atreftsadbeatwttta", ['t']),
  418. ("t(?!s)", "atreftsadbeatststs", ['t']) ]
  419. # native implementation does not handle extensions specified in this way
  420. if is_cli:
  421. tests.extend([
  422. ("(?i:foo)", "fooFoo FOO fOo fo oFO O\n\t\nFo ofO O", ['foo', 'Foo', 'FOO', 'fOo']),
  423. ("(?im:^foo)", "fooFoo FOO fOo\n\t\nFoo\nFOO", ['foo', 'Foo', 'FOO']), # ignorecase, multiline (matches at beginning of string and at each newline)
  424. ("(?s:foo.*bar)", "foo yadayadayada\nyadayadayada bar", ['foo yadayadayada\nyadayadayada bar']), # dotall (make "." match any chr, including a newline)
  425. ("(?x:baz bar)", "bazbar foo bar bazbar \n\n\tbazbar", ['bazbar', 'bazbar', 'bazbar']), #verbose (ignore whitespace)
  426. ])
  427. for (x, y, z) in tests:
  428. AreEqual(re.findall(x, y), z)
  429. AreEqual(re.compile(x).findall(y), z)
  430. def test_bug858():
  431. pattern = r"""\(? #optional paren
  432. \)? #optional paren
  433. \d+ """
  434. c = re.compile(pattern, re.X)
  435. l = c.findall("989")
  436. Assert(l == ['989'])
  437. def test_finditer():
  438. # finditer
  439. matches = re.finditer("baz","barbazbarbazbar")
  440. num = 0
  441. for m in matches:
  442. num = num + 1
  443. AreEqual("baz", m.group(0))
  444. Assert(num == 2)
  445. matches = re.finditer("baz","barbazbarbazbar", re.L)
  446. num = 0
  447. for m in matches:
  448. num = num + 1
  449. AreEqual("baz", m.group(0))
  450. Assert(num == 2)
  451. matches = re.compile("baz").finditer("barbazbarbazbar", 0)
  452. num = 0
  453. for m in matches:
  454. num = num + 1
  455. AreEqual("baz", m.group(0))
  456. Assert(num == 2)
  457. matches = re.compile("baz").finditer("barbazbarbazbar", 14)
  458. num = 0
  459. for m in matches:
  460. num = num + 1
  461. AreEqual("baz", m.group(0))
  462. Assert(num == 0)
  463. matches = re.compile("baz").finditer("barbazbarbazbar", 0, 14)
  464. num = 0
  465. for m in matches:
  466. num = num + 1
  467. AreEqual("baz", m.group(0))
  468. Assert(num == 2)
  469. matches = re.compile("baz").finditer("barbazbarbazbar", 9, 12)
  470. num = 0
  471. for m in matches:
  472. num = num + 1
  473. AreEqual("baz", m.group(0))
  474. AreEqual(num, 1)
  475. matches = re.compile("baz").finditer("barbazbarbazbar", 9, 11)
  476. num = 0
  477. for m in matches:
  478. num = num + 1
  479. AreEqual("baz", m.group(0))
  480. AreEqual(num, 0)
  481. matches = re.compile("baz").finditer("barbazbarbazbar", 10, 12)
  482. num = 0
  483. for m in matches:
  484. num = num + 1
  485. AreEqual("baz", m.group(0))
  486. AreEqual(num, 0)
  487. def test_search():
  488. # search
  489. sp = re.search('super', 'blahsupersuper').span()
  490. Assert(sp == (4, 9))
  491. sp = re.search('super', 'superblahsuper').span()
  492. Assert(sp == (0, 5))
  493. #re.search.group() index error
  494. AreEqual(re.search("z.*z", "az123za").group(),'z123z')
  495. AreEqual(re.search("z.*z", "az12za").group(),'z12z')
  496. AreEqual(re.search("z.*z", "azza").group(),'zz')
  497. AreEqual(re.search("z123p+z", "az123ppppppppppza").group(),'z123ppppppppppz')
  498. AreEqual(re.search("z123p+z", "az123pza").group(),'z123pz')
  499. AreEqual(re.search("z123p?z", "az123pza").group(),'z123pz')
  500. AreEqual(re.search("z123p?z", "az123za").group(),'z123z')
  501. AreEqual(re.search('b', 'abc').string, 'abc')
  502. def test_subn():
  503. # subn
  504. tup = re.subn("ab", "cd", "abababababab")
  505. Assert(tup == ('cdcdcdcdcdcd', 6))
  506. tup = re.subn("ab", "cd", "abababababab", 0)
  507. Assert(tup == ('cdcdcdcdcdcd', 6))
  508. tup = re.subn("ab", "cd", "abababababab", 1)
  509. Assert(tup == ('cdababababab', 1))
  510. tup = re.subn("ab", "cd", "abababababab", 10)
  511. Assert(tup == ('cdcdcdcdcdcd', 6))
  512. tup = re.subn("ababab", "cd", "ab", 10)
  513. Assert(tup == ('ab', 0))
  514. tup = re.subn("ababab", "cd", "ab")
  515. Assert(tup == ('ab', 0))
  516. tup = re.subn("(ab)*", "cd", "abababababab", 10)
  517. Assert(tup == ('cd', 1))
  518. tup = re.subn("(ab)?", "cd", "abababababab", 10)
  519. Assert(tup == ('cdcdcdcdcdcd', 6))
  520. def test_groups():
  521. reg = re.compile("\[(?P<header>.*?)\]")
  522. m = reg.search("[DEFAULT]")
  523. Assert( m.groups() == ('DEFAULT',))
  524. Assert( m.group('header') == 'DEFAULT' )
  525. reg2 = re.compile("(?P<grp>\S+)?")
  526. m2 = reg2.search("")
  527. Assert ( m2.groups() == (None,))
  528. Assert ( m2.groups('Default') == ('Default',))
  529. def test_locale_flags():
  530. AreEqual(re.compile(r"^\#[ \t]*(\w[\d\w]*)[ \t](.*)").flags, 0)
  531. AreEqual(re.compile(r"^\#[ \t]*(\w[\d\w]*)[ \t](.*)", re.L).flags, re.L)
  532. AreEqual(re.compile(r"(?L)^\#[ \t]*(\w[\d\w]*)[ \t](.*)").flags, re.L)
  533. def test_end():
  534. ex = re.compile(r'\s+')
  535. m = ex.match('(object Petal', 7)
  536. Assert (m.end(0) == 8)
  537. def test_lone_hat():
  538. """Single ^ reg-ex shouldn't match w/ a sub-set of a string"""
  539. sol = re.compile('^')
  540. AreEqual(sol.match('bazbar', 1, 2), None)
  541. AreEqual(sol.match('foobar', 1, 2), None)
  542. def test_escape_backslash():
  543. x = re.compile (r"[\\A-Z\.\+]")
  544. AreEqual(x.search('aaaA\\B\\Caaa').span(), (3,4))
  545. def test_eol():
  546. r = re.compile(r'<(/|\Z)')
  547. s = r.search("<", 0)
  548. Assert(s != None)
  549. AreEqual(s.span(), (0, 1))
  550. AreEqual(s.group(0), '<')
  551. AreEqual(r.search("<Z", 0), None)
  552. def test_lastindex():
  553. for (pat, index) in [
  554. ('(a)b', 1), ('((a)(b))', 1), ('((ab))', 1),
  555. ('(a)(b)', 2),
  556. ('(a)?ab', None),
  557. ('(a)?b', 1),
  558. ]:
  559. AreEqual(re.match(pat, 'ab').lastindex, index)
  560. for (pat, index) in [
  561. ('(a)ab', 1),
  562. ('(a)(a)b', 2),
  563. ('(a)(a)(b)', 3),
  564. ('((a)a(b))', 1),
  565. ('((a)(a)(b))', 1),
  566. ('(a(a)(b))', 1),
  567. ('(a(a)?(b))', 1),
  568. ('(aa(a)?(b))', 1),
  569. ('(aa(b))', 1),
  570. ('(a(ab))', 1),
  571. ('(a)?ab', 1),
  572. ('a(a)?ab', None),
  573. ('a(a)?(a)?b', 1),
  574. ('a(a)?(a)?(b)', 3),
  575. ('a(a)b', 1),
  576. ('(a(a))(b)', 3),
  577. ('(a(a))b', 1),
  578. ('((a)(a))(b)', 4),
  579. ('((a)(a))b', 1),
  580. ]:
  581. AreEqual(re.match(pat, 'aab').lastindex, index)
  582. def test_empty_split():
  583. cases =[
  584. ('', ['']),
  585. ('*', ['*']),
  586. (':', ['', '']),
  587. ('::', ['', '']),
  588. ('a::', ['a', '']),
  589. ('::b', ['', 'b']),
  590. (':c:', ['', 'c', '']),
  591. (':\t: ', ['', '\t', ' ']),
  592. ('a:b::c', ['a', 'b', 'c']),
  593. (':a:b::c', ['', 'a', 'b', 'c']),
  594. ('::a:b::c:', ['', 'a', 'b', 'c', '']),
  595. ]
  596. for expr, result in cases:
  597. AreEqual(re.split(":*", expr), result)
  598. @skip("silverlight")
  599. def test_cp15298():
  600. regex = "^" + "\d\.\d\.\d \(IronPython \d\.\d(\.\d)? ((Alpha )|(Beta )|())\(\d\.\d\.\d\.\d{3,4}\) on \.NET \d(\.\d{1,5}){3}\)" * 15 + "$"
  601. match_str = "2.5.0 (IronPython 2.0 Beta (2.0.0.1000) on .NET 2.0.50727.1433)" * 15
  602. compiled_regex = re.compile(regex)
  603. retval = compiled_regex.match(match_str)
  604. Assert(retval != None)
  605. retval = re.match(regex, match_str)
  606. Assert(retval != None)
  607. def test_cp11136():
  608. regex = re.compile(r"^(?P<msg>NMAKE[A-Za-z0-9]*)'\"?(?P<file>[\\A-Za-z0-9/:_\.\+]+)" )
  609. Assert(regex.search(r"NMAKE0119'adirectory\afile.txt")!=None)
  610. def test_cp17111():
  611. test_cases = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789~!@#%&_+-=]{};':,.//<>" + '"'
  612. for x in test_cases:
  613. regex = re.compile(r".*\\%s" % x)
  614. Assert(regex.search(r"\\%s" % x)!=None)
  615. Assert(regex.search(r"")==None)
  616. def test_cp1089():
  617. test_cases = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789~!@#%&_+-=]{};':,.//<>" + '"'
  618. for x in test_cases:
  619. #Just make sure they don't throw
  620. temp = re.compile('\\\\' + x)
  621. def test_cp16657():
  622. Assert(re.compile(r'^bar', re.M).search('foo\nbar') != None)
  623. Assert(re.compile(r'^bar(?m)').search('foo\nbar') != None)
  624. Assert(re.compile(r'^bar', re.M).search('foo\nbaar') == None)
  625. Assert(re.compile(r'^bar(?m)').search('foo\nbaar') == None)
  626. Assert(re.compile(r'^bar', re.U).search('bar') != None)
  627. Assert(re.compile(r'^bar(?u)').search('bar') != None)
  628. Assert(re.compile(r'^bar', re.U).search('baar') == None)
  629. Assert(re.compile(r'^bar(?u)').search('baar') == None)
  630. Assert(re.compile(r' b ar ', re.X).search('bar') != None)
  631. Assert(re.compile(r'b ar(?x)').search('bar') != None)
  632. Assert(re.compile(r' b ar ', re.X).search('baar') == None)
  633. Assert(re.compile(r'b ar(?x)').search('baar') == None)
  634. Assert(re.compile(r'b ar').search('bar') == None)
  635. def test_n_m_quantifier():
  636. AreEqual(re.search('ab{,2}a', 'abba').span(), (0, 4))
  637. AreEqual(re.search('ab{,2}a', 'aba').span(), (0, 3))
  638. AreEqual(re.search('ab{,2}a', 'abbba'), None)
  639. AreEqual(re.search('ab{,2}a', 'abba').span(), re.search('ab{0,2}a', 'abba').span())
  640. AreEqual(re.search('ab{0,2}a', 'abbba'), None)
  641. AreEqual(re.search('ab{2,}a', 'abba').span(), (0,4))
  642. AreEqual(re.search('ab{2,}a', 'abbba').span(), (0,5))
  643. AreEqual(re.search('ab{2,}a', 'aba'), None)
  644. def test_mixed_named_and_unnamed_groups():
  645. example1=r"(?P<one>Blah)"
  646. example2=r"(?P<one>(Blah))"
  647. RegExsToTest=[example1,example2]
  648. for regString in RegExsToTest:
  649. g=re.compile(regString)
  650. AreEqual(g.groupindex, {'one' : 1})
  651. def test__pickle():
  652. '''
  653. TODO: just a sanity test for now. Needs far more testing.
  654. '''
  655. regex = re.compile(r"^(?P<msg>NMAKE[A-Za-z0-9]*)'\"?(?P<file>[\\A-Za-z0-9/:_\.\+]+)" )
  656. pickled_regex = re._pickle(regex)
  657. AreEqual(len(pickled_regex), 2)
  658. AreEqual(pickled_regex[1],
  659. ('^(?P<msg>NMAKE[A-Za-z0-9]*)\'\\"?(?P<file>[\\\\A-Za-z0-9/:_\\.\\+]+)', 0))
  660. def test_conditional():
  661. p = re.compile(r'(a)?(b)((?(1)c))')
  662. AreEqual(p.match('abc').groups(), ('a', 'b', 'c'))
  663. p = re.compile(r'(?P<first>a)?(b)((?(first)c))')
  664. AreEqual(p.match('abc').groups(), ('a', 'b', 'c'))
  665. s = r'((?(a)ab|cd))'
  666. if is_cli or is_silverlight:
  667. p = re.compile(s)
  668. AreEqual(p.match('ab').groups(), ('ab',))
  669. else:
  670. AssertError(re.error, re.compile, s)
  671. #--MAIN------------------------------------------------------------------------
  672. run_test(__name__)