You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

4057 lines
192 KiB

4 years ago
  1. import regex
  2. import string
  3. from weakref import proxy
  4. import unittest
  5. import copy
  6. import pickle
  7. from test.support import run_unittest
  8. import sys
  9. # String subclasses for issue 18468.
  10. class StrSubclass(str):
  11. def __getitem__(self, index):
  12. return StrSubclass(super().__getitem__(index))
  13. class BytesSubclass(bytes):
  14. def __getitem__(self, index):
  15. return BytesSubclass(super().__getitem__(index))
  16. class RegexTests(unittest.TestCase):
  17. PATTERN_CLASS = "<class '_regex.Pattern'>"
  18. FLAGS_WITH_COMPILED_PAT = "cannot process flags argument with a compiled pattern"
  19. INVALID_GROUP_REF = "invalid group reference"
  20. MISSING_GT = "missing >"
  21. BAD_GROUP_NAME = "bad character in group name"
  22. MISSING_GROUP_NAME = "missing group name"
  23. MISSING_LT = "missing <"
  24. UNKNOWN_GROUP_I = "unknown group"
  25. UNKNOWN_GROUP = "unknown group"
  26. BAD_ESCAPE = r"bad escape \(end of pattern\)"
  27. BAD_OCTAL_ESCAPE = r"bad escape \\"
  28. BAD_SET = "unterminated character set"
  29. STR_PAT_ON_BYTES = "cannot use a string pattern on a bytes-like object"
  30. BYTES_PAT_ON_STR = "cannot use a bytes pattern on a string-like object"
  31. STR_PAT_BYTES_TEMPL = "expected str instance, bytes found"
  32. BYTES_PAT_STR_TEMPL = "expected a bytes-like object, str found"
  33. BYTES_PAT_UNI_FLAG = "cannot use UNICODE flag with a bytes pattern"
  34. MIXED_FLAGS = "ASCII, LOCALE and UNICODE flags are mutually incompatible"
  35. MISSING_RPAREN = "missing \\)"
  36. TRAILING_CHARS = "unbalanced parenthesis"
  37. BAD_CHAR_RANGE = "bad character range"
  38. NOTHING_TO_REPEAT = "nothing to repeat"
  39. MULTIPLE_REPEAT = "multiple repeat"
  40. OPEN_GROUP = "cannot refer to an open group"
  41. DUPLICATE_GROUP = "duplicate group"
  42. CANT_TURN_OFF = "bad inline flags: cannot turn flags off"
  43. UNDEF_CHAR_NAME = "undefined character name"
  44. def assertTypedEqual(self, actual, expect, msg=None):
  45. self.assertEqual(actual, expect, msg)
  46. def recurse(actual, expect):
  47. if isinstance(expect, (tuple, list)):
  48. for x, y in zip(actual, expect):
  49. recurse(x, y)
  50. else:
  51. self.assertIs(type(actual), type(expect), msg)
  52. recurse(actual, expect)
  53. def test_weakref(self):
  54. s = 'QabbbcR'
  55. x = regex.compile('ab+c')
  56. y = proxy(x)
  57. if x.findall('QabbbcR') != y.findall('QabbbcR'):
  58. self.fail()
  59. def test_search_star_plus(self):
  60. self.assertEqual(regex.search('a*', 'xxx').span(0), (0, 0))
  61. self.assertEqual(regex.search('x*', 'axx').span(), (0, 0))
  62. self.assertEqual(regex.search('x+', 'axx').span(0), (1, 3))
  63. self.assertEqual(regex.search('x+', 'axx').span(), (1, 3))
  64. self.assertEqual(regex.search('x', 'aaa'), None)
  65. self.assertEqual(regex.match('a*', 'xxx').span(0), (0, 0))
  66. self.assertEqual(regex.match('a*', 'xxx').span(), (0, 0))
  67. self.assertEqual(regex.match('x*', 'xxxa').span(0), (0, 3))
  68. self.assertEqual(regex.match('x*', 'xxxa').span(), (0, 3))
  69. self.assertEqual(regex.match('a+', 'xxx'), None)
  70. def bump_num(self, matchobj):
  71. int_value = int(matchobj[0])
  72. return str(int_value + 1)
  73. def test_basic_regex_sub(self):
  74. self.assertEqual(regex.sub("(?i)b+", "x", "bbbb BBBB"), 'x x')
  75. self.assertEqual(regex.sub(r'\d+', self.bump_num, '08.2 -2 23x99y'),
  76. '9.3 -3 24x100y')
  77. self.assertEqual(regex.sub(r'\d+', self.bump_num, '08.2 -2 23x99y', 3),
  78. '9.3 -3 23x99y')
  79. self.assertEqual(regex.sub('.', lambda m: r"\n", 'x'), "\\n")
  80. self.assertEqual(regex.sub('.', r"\n", 'x'), "\n")
  81. self.assertEqual(regex.sub('(?P<a>x)', r'\g<a>\g<a>', 'xx'), 'xxxx')
  82. self.assertEqual(regex.sub('(?P<a>x)', r'\g<a>\g<1>', 'xx'), 'xxxx')
  83. self.assertEqual(regex.sub('(?P<unk>x)', r'\g<unk>\g<unk>', 'xx'),
  84. 'xxxx')
  85. self.assertEqual(regex.sub('(?P<unk>x)', r'\g<1>\g<1>', 'xx'), 'xxxx')
  86. self.assertEqual(regex.sub('a', r'\t\n\v\r\f\a\b\B\Z\a\A\w\W\s\S\d\D',
  87. 'a'), "\t\n\v\r\f\a\b\\B\\Z\a\\A\\w\\W\\s\\S\\d\\D")
  88. self.assertEqual(regex.sub('a', '\t\n\v\r\f\a', 'a'), "\t\n\v\r\f\a")
  89. self.assertEqual(regex.sub('a', '\t\n\v\r\f\a', 'a'), chr(9) + chr(10)
  90. + chr(11) + chr(13) + chr(12) + chr(7))
  91. self.assertEqual(regex.sub(r'^\s*', 'X', 'test'), 'Xtest')
  92. self.assertEqual(regex.sub(r"x", r"\x0A", "x"), "\n")
  93. self.assertEqual(regex.sub(r"x", r"\u000A", "x"), "\n")
  94. self.assertEqual(regex.sub(r"x", r"\U0000000A", "x"), "\n")
  95. self.assertEqual(regex.sub(r"x", r"\N{LATIN CAPITAL LETTER A}",
  96. "x"), "A")
  97. self.assertEqual(regex.sub(br"x", br"\x0A", b"x"), b"\n")
  98. self.assertEqual(regex.sub(br"x", br"\u000A", b"x"), b"\\u000A")
  99. self.assertEqual(regex.sub(br"x", br"\U0000000A", b"x"),
  100. b"\\U0000000A")
  101. self.assertEqual(regex.sub(br"x", br"\N{LATIN CAPITAL LETTER A}",
  102. b"x"), b"\\N{LATIN CAPITAL LETTER A}")
  103. def test_bug_449964(self):
  104. # Fails for group followed by other escape.
  105. self.assertEqual(regex.sub(r'(?P<unk>x)', r'\g<1>\g<1>\b', 'xx'),
  106. "xx\bxx\b")
  107. def test_bug_449000(self):
  108. # Test for sub() on escaped characters.
  109. self.assertEqual(regex.sub(r'\r\n', r'\n', 'abc\r\ndef\r\n'),
  110. "abc\ndef\n")
  111. self.assertEqual(regex.sub('\r\n', r'\n', 'abc\r\ndef\r\n'),
  112. "abc\ndef\n")
  113. self.assertEqual(regex.sub(r'\r\n', '\n', 'abc\r\ndef\r\n'),
  114. "abc\ndef\n")
  115. self.assertEqual(regex.sub('\r\n', '\n', 'abc\r\ndef\r\n'),
  116. "abc\ndef\n")
  117. def test_bug_1661(self):
  118. # Verify that flags do not get silently ignored with compiled patterns
  119. pattern = regex.compile('.')
  120. self.assertRaisesRegex(ValueError, self.FLAGS_WITH_COMPILED_PAT,
  121. lambda: regex.match(pattern, 'A', regex.I))
  122. self.assertRaisesRegex(ValueError, self.FLAGS_WITH_COMPILED_PAT,
  123. lambda: regex.search(pattern, 'A', regex.I))
  124. self.assertRaisesRegex(ValueError, self.FLAGS_WITH_COMPILED_PAT,
  125. lambda: regex.findall(pattern, 'A', regex.I))
  126. self.assertRaisesRegex(ValueError, self.FLAGS_WITH_COMPILED_PAT,
  127. lambda: regex.compile(pattern, regex.I))
  128. def test_bug_3629(self):
  129. # A regex that triggered a bug in the sre-code validator
  130. self.assertEqual(repr(type(regex.compile("(?P<quote>)(?(quote))"))),
  131. self.PATTERN_CLASS)
  132. def test_sub_template_numeric_escape(self):
  133. # Bug 776311 and friends.
  134. self.assertEqual(regex.sub('x', r'\0', 'x'), "\0")
  135. self.assertEqual(regex.sub('x', r'\000', 'x'), "\000")
  136. self.assertEqual(regex.sub('x', r'\001', 'x'), "\001")
  137. self.assertEqual(regex.sub('x', r'\008', 'x'), "\0" + "8")
  138. self.assertEqual(regex.sub('x', r'\009', 'x'), "\0" + "9")
  139. self.assertEqual(regex.sub('x', r'\111', 'x'), "\111")
  140. self.assertEqual(regex.sub('x', r'\117', 'x'), "\117")
  141. self.assertEqual(regex.sub('x', r'\1111', 'x'), "\1111")
  142. self.assertEqual(regex.sub('x', r'\1111', 'x'), "\111" + "1")
  143. self.assertEqual(regex.sub('x', r'\00', 'x'), '\x00')
  144. self.assertEqual(regex.sub('x', r'\07', 'x'), '\x07')
  145. self.assertEqual(regex.sub('x', r'\08', 'x'), "\0" + "8")
  146. self.assertEqual(regex.sub('x', r'\09', 'x'), "\0" + "9")
  147. self.assertEqual(regex.sub('x', r'\0a', 'x'), "\0" + "a")
  148. self.assertEqual(regex.sub('x', r'\400', 'x'), "\u0100")
  149. self.assertEqual(regex.sub('x', r'\777', 'x'), "\u01FF")
  150. self.assertEqual(regex.sub(b'x', br'\400', b'x'), b"\x00")
  151. self.assertEqual(regex.sub(b'x', br'\777', b'x'), b"\xFF")
  152. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  153. regex.sub('x', r'\1', 'x'))
  154. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  155. regex.sub('x', r'\8', 'x'))
  156. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  157. regex.sub('x', r'\9', 'x'))
  158. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  159. regex.sub('x', r'\11', 'x'))
  160. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  161. regex.sub('x', r'\18', 'x'))
  162. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  163. regex.sub('x', r'\1a', 'x'))
  164. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  165. regex.sub('x', r'\90', 'x'))
  166. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  167. regex.sub('x', r'\99', 'x'))
  168. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  169. regex.sub('x', r'\118', 'x')) # r'\11' + '8'
  170. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  171. regex.sub('x', r'\11a', 'x'))
  172. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  173. regex.sub('x', r'\181', 'x')) # r'\18' + '1'
  174. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  175. regex.sub('x', r'\800', 'x')) # r'\80' + '0'
  176. # In Python 2.3 (etc), these loop endlessly in sre_parser.py.
  177. self.assertEqual(regex.sub('(((((((((((x)))))))))))', r'\11', 'x'),
  178. 'x')
  179. self.assertEqual(regex.sub('((((((((((y))))))))))(.)', r'\118', 'xyz'),
  180. 'xz8')
  181. self.assertEqual(regex.sub('((((((((((y))))))))))(.)', r'\11a', 'xyz'),
  182. 'xza')
  183. def test_qualified_re_sub(self):
  184. self.assertEqual(regex.sub('a', 'b', 'aaaaa'), 'bbbbb')
  185. self.assertEqual(regex.sub('a', 'b', 'aaaaa', 1), 'baaaa')
  186. def test_bug_114660(self):
  187. self.assertEqual(regex.sub(r'(\S)\s+(\S)', r'\1 \2', 'hello there'),
  188. 'hello there')
  189. def test_bug_462270(self):
  190. # Test for empty sub() behaviour, see SF bug #462270
  191. if sys.version_info >= (3, 7, 0):
  192. self.assertEqual(regex.sub('(?V0)x*', '-', 'abxd'), '-a-b--d-')
  193. else:
  194. self.assertEqual(regex.sub('(?V0)x*', '-', 'abxd'), '-a-b-d-')
  195. self.assertEqual(regex.sub('(?V1)x*', '-', 'abxd'), '-a-b--d-')
  196. self.assertEqual(regex.sub('x+', '-', 'abxd'), 'ab-d')
  197. def test_bug_14462(self):
  198. # chr(255) is a valid identifier in Python 3.
  199. group_name = '\xFF'
  200. self.assertEqual(regex.search(r'(?P<' + group_name + '>a)',
  201. 'abc').group(group_name), 'a')
  202. def test_symbolic_refs(self):
  203. self.assertRaisesRegex(regex.error, self.MISSING_GT, lambda:
  204. regex.sub('(?P<a>x)', r'\g<a', 'xx'))
  205. self.assertRaisesRegex(regex.error, self.MISSING_GROUP_NAME, lambda:
  206. regex.sub('(?P<a>x)', r'\g<', 'xx'))
  207. self.assertRaisesRegex(regex.error, self.MISSING_LT, lambda:
  208. regex.sub('(?P<a>x)', r'\g', 'xx'))
  209. self.assertRaisesRegex(regex.error, self.BAD_GROUP_NAME, lambda:
  210. regex.sub('(?P<a>x)', r'\g<a a>', 'xx'))
  211. self.assertRaisesRegex(regex.error, self.BAD_GROUP_NAME, lambda:
  212. regex.sub('(?P<a>x)', r'\g<1a1>', 'xx'))
  213. self.assertRaisesRegex(IndexError, self.UNKNOWN_GROUP_I, lambda:
  214. regex.sub('(?P<a>x)', r'\g<ab>', 'xx'))
  215. # The new behaviour of unmatched but valid groups is to treat them like
  216. # empty matches in the replacement template, like in Perl.
  217. self.assertEqual(regex.sub('(?P<a>x)|(?P<b>y)', r'\g<b>', 'xx'), '')
  218. self.assertEqual(regex.sub('(?P<a>x)|(?P<b>y)', r'\2', 'xx'), '')
  219. # The old behaviour was to raise it as an IndexError.
  220. self.assertRaisesRegex(regex.error, self.BAD_GROUP_NAME, lambda:
  221. regex.sub('(?P<a>x)', r'\g<-1>', 'xx'))
  222. def test_re_subn(self):
  223. self.assertEqual(regex.subn("(?i)b+", "x", "bbbb BBBB"), ('x x', 2))
  224. self.assertEqual(regex.subn("b+", "x", "bbbb BBBB"), ('x BBBB', 1))
  225. self.assertEqual(regex.subn("b+", "x", "xyz"), ('xyz', 0))
  226. self.assertEqual(regex.subn("b*", "x", "xyz"), ('xxxyxzx', 4))
  227. self.assertEqual(regex.subn("b*", "x", "xyz", 2), ('xxxyz', 2))
  228. def test_re_split(self):
  229. self.assertEqual(regex.split(":", ":a:b::c"), ['', 'a', 'b', '', 'c'])
  230. if sys.version_info >= (3, 7, 0):
  231. self.assertEqual(regex.split(":*", ":a:b::c"), ['', '', 'a', '',
  232. 'b', '', 'c', ''])
  233. self.assertEqual(regex.split("(:*)", ":a:b::c"), ['', ':', '', '',
  234. 'a', ':', '', '', 'b', '::', '', '', 'c', '', ''])
  235. self.assertEqual(regex.split("(?::*)", ":a:b::c"), ['', '', 'a',
  236. '', 'b', '', 'c', ''])
  237. self.assertEqual(regex.split("(:)*", ":a:b::c"), ['', ':', '',
  238. None, 'a', ':', '', None, 'b', ':', '', None, 'c', None, ''])
  239. else:
  240. self.assertEqual(regex.split(":*", ":a:b::c"), ['', 'a', 'b', 'c'])
  241. self.assertEqual(regex.split("(:*)", ":a:b::c"), ['', ':', 'a',
  242. ':', 'b', '::', 'c'])
  243. self.assertEqual(regex.split("(?::*)", ":a:b::c"), ['', 'a', 'b',
  244. 'c'])
  245. self.assertEqual(regex.split("(:)*", ":a:b::c"), ['', ':', 'a',
  246. ':', 'b', ':', 'c'])
  247. self.assertEqual(regex.split("([b:]+)", ":a:b::c"), ['', ':', 'a',
  248. ':b::', 'c'])
  249. self.assertEqual(regex.split("(b)|(:+)", ":a:b::c"), ['', None, ':',
  250. 'a', None, ':', '', 'b', None, '', None, '::', 'c'])
  251. self.assertEqual(regex.split("(?:b)|(?::+)", ":a:b::c"), ['', 'a', '',
  252. '', 'c'])
  253. self.assertEqual(regex.split("x", "xaxbxc"), ['', 'a', 'b', 'c'])
  254. self.assertEqual([m for m in regex.splititer("x", "xaxbxc")], ['', 'a',
  255. 'b', 'c'])
  256. self.assertEqual(regex.split("(?r)x", "xaxbxc"), ['c', 'b', 'a', ''])
  257. self.assertEqual([m for m in regex.splititer("(?r)x", "xaxbxc")], ['c',
  258. 'b', 'a', ''])
  259. self.assertEqual(regex.split("(x)|(y)", "xaxbxc"), ['', 'x', None, 'a',
  260. 'x', None, 'b', 'x', None, 'c'])
  261. self.assertEqual([m for m in regex.splititer("(x)|(y)", "xaxbxc")],
  262. ['', 'x', None, 'a', 'x', None, 'b', 'x', None, 'c'])
  263. self.assertEqual(regex.split("(?r)(x)|(y)", "xaxbxc"), ['c', 'x', None,
  264. 'b', 'x', None, 'a', 'x', None, ''])
  265. self.assertEqual([m for m in regex.splititer("(?r)(x)|(y)", "xaxbxc")],
  266. ['c', 'x', None, 'b', 'x', None, 'a', 'x', None, ''])
  267. self.assertEqual(regex.split(r"(?V1)\b", "a b c"), ['', 'a', ' ', 'b',
  268. ' ', 'c', ''])
  269. self.assertEqual(regex.split(r"(?V1)\m", "a b c"), ['', 'a ', 'b ',
  270. 'c'])
  271. self.assertEqual(regex.split(r"(?V1)\M", "a b c"), ['a', ' b', ' c',
  272. ''])
  273. def test_qualified_re_split(self):
  274. self.assertEqual(regex.split(":", ":a:b::c", 2), ['', 'a', 'b::c'])
  275. self.assertEqual(regex.split(':', 'a:b:c:d', 2), ['a', 'b', 'c:d'])
  276. self.assertEqual(regex.split("(:)", ":a:b::c", 2), ['', ':', 'a', ':',
  277. 'b::c'])
  278. if sys.version_info >= (3, 7, 0):
  279. self.assertEqual(regex.split("(:*)", ":a:b::c", 2), ['', ':', '',
  280. '', 'a:b::c'])
  281. else:
  282. self.assertEqual(regex.split("(:*)", ":a:b::c", 2), ['', ':', 'a',
  283. ':', 'b::c'])
  284. def test_re_findall(self):
  285. self.assertEqual(regex.findall(":+", "abc"), [])
  286. self.assertEqual(regex.findall(":+", "a:b::c:::d"), [':', '::', ':::'])
  287. self.assertEqual(regex.findall("(:+)", "a:b::c:::d"), [':', '::',
  288. ':::'])
  289. self.assertEqual(regex.findall("(:)(:*)", "a:b::c:::d"), [(':', ''),
  290. (':', ':'), (':', '::')])
  291. self.assertEqual(regex.findall(r"\((?P<test>.{0,5}?TEST)\)",
  292. "(MY TEST)"), ["MY TEST"])
  293. self.assertEqual(regex.findall(r"\((?P<test>.{0,3}?TEST)\)",
  294. "(MY TEST)"), ["MY TEST"])
  295. self.assertEqual(regex.findall(r"\((?P<test>.{0,3}?T)\)", "(MY T)"),
  296. ["MY T"])
  297. self.assertEqual(regex.findall(r"[^a]{2}[A-Z]", "\n S"), [' S'])
  298. self.assertEqual(regex.findall(r"[^a]{2,3}[A-Z]", "\n S"), ['\n S'])
  299. self.assertEqual(regex.findall(r"[^a]{2,3}[A-Z]", "\n S"), [' S'])
  300. self.assertEqual(regex.findall(r"X(Y[^Y]+?){1,2}( |Q)+DEF",
  301. "XYABCYPPQ\nQ DEF"), [('YPPQ\n', ' ')])
  302. self.assertEqual(regex.findall(r"(\nTest(\n+.+?){0,2}?)?\n+End",
  303. "\nTest\nxyz\nxyz\nEnd"), [('\nTest\nxyz\nxyz', '\nxyz')])
  304. def test_bug_117612(self):
  305. self.assertEqual(regex.findall(r"(a|(b))", "aba"), [('a', ''), ('b',
  306. 'b'), ('a', '')])
  307. def test_re_match(self):
  308. self.assertEqual(regex.match('a', 'a')[:], ('a',))
  309. self.assertEqual(regex.match('(a)', 'a')[:], ('a', 'a'))
  310. self.assertEqual(regex.match(r'(a)', 'a')[0], 'a')
  311. self.assertEqual(regex.match(r'(a)', 'a')[1], 'a')
  312. self.assertEqual(regex.match(r'(a)', 'a').group(1, 1), ('a', 'a'))
  313. pat = regex.compile('((a)|(b))(c)?')
  314. self.assertEqual(pat.match('a')[:], ('a', 'a', 'a', None, None))
  315. self.assertEqual(pat.match('b')[:], ('b', 'b', None, 'b', None))
  316. self.assertEqual(pat.match('ac')[:], ('ac', 'a', 'a', None, 'c'))
  317. self.assertEqual(pat.match('bc')[:], ('bc', 'b', None, 'b', 'c'))
  318. self.assertEqual(pat.match('bc')[:], ('bc', 'b', None, 'b', 'c'))
  319. # A single group.
  320. m = regex.match('(a)', 'a')
  321. self.assertEqual(m.group(), 'a')
  322. self.assertEqual(m.group(0), 'a')
  323. self.assertEqual(m.group(1), 'a')
  324. self.assertEqual(m.group(1, 1), ('a', 'a'))
  325. pat = regex.compile('(?:(?P<a1>a)|(?P<b2>b))(?P<c3>c)?')
  326. self.assertEqual(pat.match('a').group(1, 2, 3), ('a', None, None))
  327. self.assertEqual(pat.match('b').group('a1', 'b2', 'c3'), (None, 'b',
  328. None))
  329. self.assertEqual(pat.match('ac').group(1, 'b2', 3), ('a', None, 'c'))
  330. def test_re_groupref_exists(self):
  331. self.assertEqual(regex.match(r'^(\()?([^()]+)(?(1)\))$', '(a)')[:],
  332. ('(a)', '(', 'a'))
  333. self.assertEqual(regex.match(r'^(\()?([^()]+)(?(1)\))$', 'a')[:], ('a',
  334. None, 'a'))
  335. self.assertEqual(regex.match(r'^(\()?([^()]+)(?(1)\))$', 'a)'), None)
  336. self.assertEqual(regex.match(r'^(\()?([^()]+)(?(1)\))$', '(a'), None)
  337. self.assertEqual(regex.match('^(?:(a)|c)((?(1)b|d))$', 'ab')[:], ('ab',
  338. 'a', 'b'))
  339. self.assertEqual(regex.match('^(?:(a)|c)((?(1)b|d))$', 'cd')[:], ('cd',
  340. None, 'd'))
  341. self.assertEqual(regex.match('^(?:(a)|c)((?(1)|d))$', 'cd')[:], ('cd',
  342. None, 'd'))
  343. self.assertEqual(regex.match('^(?:(a)|c)((?(1)|d))$', 'a')[:], ('a',
  344. 'a', ''))
  345. # Tests for bug #1177831: exercise groups other than the first group.
  346. p = regex.compile('(?P<g1>a)(?P<g2>b)?((?(g2)c|d))')
  347. self.assertEqual(p.match('abc')[:], ('abc', 'a', 'b', 'c'))
  348. self.assertEqual(p.match('ad')[:], ('ad', 'a', None, 'd'))
  349. self.assertEqual(p.match('abd'), None)
  350. self.assertEqual(p.match('ac'), None)
  351. def test_re_groupref(self):
  352. self.assertEqual(regex.match(r'^(\|)?([^()]+)\1$', '|a|')[:], ('|a|',
  353. '|', 'a'))
  354. self.assertEqual(regex.match(r'^(\|)?([^()]+)\1?$', 'a')[:], ('a',
  355. None, 'a'))
  356. self.assertEqual(regex.match(r'^(\|)?([^()]+)\1$', 'a|'), None)
  357. self.assertEqual(regex.match(r'^(\|)?([^()]+)\1$', '|a'), None)
  358. self.assertEqual(regex.match(r'^(?:(a)|c)(\1)$', 'aa')[:], ('aa', 'a',
  359. 'a'))
  360. self.assertEqual(regex.match(r'^(?:(a)|c)(\1)?$', 'c')[:], ('c', None,
  361. None))
  362. self.assertEqual(regex.findall(r"(?i)(.{1,40}?),(.{1,40}?)(?:;)+(.{1,80}).{1,40}?\3(\ |;)+(.{1,80}?)\1",
  363. "TEST, BEST; LEST ; Lest 123 Test, Best"), [('TEST', ' BEST',
  364. ' LEST', ' ', '123 ')])
  365. def test_groupdict(self):
  366. self.assertEqual(regex.match('(?P<first>first) (?P<second>second)',
  367. 'first second').groupdict(), {'first': 'first', 'second': 'second'})
  368. def test_expand(self):
  369. self.assertEqual(regex.match("(?P<first>first) (?P<second>second)",
  370. "first second").expand(r"\2 \1 \g<second> \g<first>"),
  371. 'second first second first')
  372. def test_repeat_minmax(self):
  373. self.assertEqual(regex.match(r"^(\w){1}$", "abc"), None)
  374. self.assertEqual(regex.match(r"^(\w){1}?$", "abc"), None)
  375. self.assertEqual(regex.match(r"^(\w){1,2}$", "abc"), None)
  376. self.assertEqual(regex.match(r"^(\w){1,2}?$", "abc"), None)
  377. self.assertEqual(regex.match(r"^(\w){3}$", "abc")[1], 'c')
  378. self.assertEqual(regex.match(r"^(\w){1,3}$", "abc")[1], 'c')
  379. self.assertEqual(regex.match(r"^(\w){1,4}$", "abc")[1], 'c')
  380. self.assertEqual(regex.match(r"^(\w){3,4}?$", "abc")[1], 'c')
  381. self.assertEqual(regex.match(r"^(\w){3}?$", "abc")[1], 'c')
  382. self.assertEqual(regex.match(r"^(\w){1,3}?$", "abc")[1], 'c')
  383. self.assertEqual(regex.match(r"^(\w){1,4}?$", "abc")[1], 'c')
  384. self.assertEqual(regex.match(r"^(\w){3,4}?$", "abc")[1], 'c')
  385. self.assertEqual(regex.match("^x{1}$", "xxx"), None)
  386. self.assertEqual(regex.match("^x{1}?$", "xxx"), None)
  387. self.assertEqual(regex.match("^x{1,2}$", "xxx"), None)
  388. self.assertEqual(regex.match("^x{1,2}?$", "xxx"), None)
  389. self.assertEqual(regex.match("^x{1}", "xxx")[0], 'x')
  390. self.assertEqual(regex.match("^x{1}?", "xxx")[0], 'x')
  391. self.assertEqual(regex.match("^x{0,1}", "xxx")[0], 'x')
  392. self.assertEqual(regex.match("^x{0,1}?", "xxx")[0], '')
  393. self.assertEqual(bool(regex.match("^x{3}$", "xxx")), True)
  394. self.assertEqual(bool(regex.match("^x{1,3}$", "xxx")), True)
  395. self.assertEqual(bool(regex.match("^x{1,4}$", "xxx")), True)
  396. self.assertEqual(bool(regex.match("^x{3,4}?$", "xxx")), True)
  397. self.assertEqual(bool(regex.match("^x{3}?$", "xxx")), True)
  398. self.assertEqual(bool(regex.match("^x{1,3}?$", "xxx")), True)
  399. self.assertEqual(bool(regex.match("^x{1,4}?$", "xxx")), True)
  400. self.assertEqual(bool(regex.match("^x{3,4}?$", "xxx")), True)
  401. self.assertEqual(regex.match("^x{}$", "xxx"), None)
  402. self.assertEqual(bool(regex.match("^x{}$", "x{}")), True)
  403. def test_getattr(self):
  404. self.assertEqual(regex.compile("(?i)(a)(b)").pattern, '(?i)(a)(b)')
  405. self.assertEqual(regex.compile("(?i)(a)(b)").flags, regex.I | regex.U |
  406. regex.DEFAULT_VERSION)
  407. self.assertEqual(regex.compile(b"(?i)(a)(b)").flags, regex.A | regex.I
  408. | regex.DEFAULT_VERSION)
  409. self.assertEqual(regex.compile("(?i)(a)(b)").groups, 2)
  410. self.assertEqual(regex.compile("(?i)(a)(b)").groupindex, {})
  411. self.assertEqual(regex.compile("(?i)(?P<first>a)(?P<other>b)").groupindex,
  412. {'first': 1, 'other': 2})
  413. self.assertEqual(regex.match("(a)", "a").pos, 0)
  414. self.assertEqual(regex.match("(a)", "a").endpos, 1)
  415. self.assertEqual(regex.search("b(c)", "abcdef").pos, 0)
  416. self.assertEqual(regex.search("b(c)", "abcdef").endpos, 6)
  417. self.assertEqual(regex.search("b(c)", "abcdef").span(), (1, 3))
  418. self.assertEqual(regex.search("b(c)", "abcdef").span(1), (2, 3))
  419. self.assertEqual(regex.match("(a)", "a").string, 'a')
  420. self.assertEqual(regex.match("(a)", "a").regs, ((0, 1), (0, 1)))
  421. self.assertEqual(repr(type(regex.match("(a)", "a").re)),
  422. self.PATTERN_CLASS)
  423. # Issue 14260.
  424. p = regex.compile(r'abc(?P<n>def)')
  425. p.groupindex["n"] = 0
  426. self.assertEqual(p.groupindex["n"], 1)
  427. def test_special_escapes(self):
  428. self.assertEqual(regex.search(r"\b(b.)\b", "abcd abc bcd bx")[1], 'bx')
  429. self.assertEqual(regex.search(r"\B(b.)\B", "abc bcd bc abxd")[1], 'bx')
  430. self.assertEqual(regex.search(br"\b(b.)\b", b"abcd abc bcd bx",
  431. regex.LOCALE)[1], b'bx')
  432. self.assertEqual(regex.search(br"\B(b.)\B", b"abc bcd bc abxd",
  433. regex.LOCALE)[1], b'bx')
  434. self.assertEqual(regex.search(r"\b(b.)\b", "abcd abc bcd bx",
  435. regex.UNICODE)[1], 'bx')
  436. self.assertEqual(regex.search(r"\B(b.)\B", "abc bcd bc abxd",
  437. regex.UNICODE)[1], 'bx')
  438. self.assertEqual(regex.search(r"^abc$", "\nabc\n", regex.M)[0], 'abc')
  439. self.assertEqual(regex.search(r"^\Aabc\Z$", "abc", regex.M)[0], 'abc')
  440. self.assertEqual(regex.search(r"^\Aabc\Z$", "\nabc\n", regex.M), None)
  441. self.assertEqual(regex.search(br"\b(b.)\b", b"abcd abc bcd bx")[1],
  442. b'bx')
  443. self.assertEqual(regex.search(br"\B(b.)\B", b"abc bcd bc abxd")[1],
  444. b'bx')
  445. self.assertEqual(regex.search(br"^abc$", b"\nabc\n", regex.M)[0],
  446. b'abc')
  447. self.assertEqual(regex.search(br"^\Aabc\Z$", b"abc", regex.M)[0],
  448. b'abc')
  449. self.assertEqual(regex.search(br"^\Aabc\Z$", b"\nabc\n", regex.M),
  450. None)
  451. self.assertEqual(regex.search(r"\d\D\w\W\s\S", "1aa! a")[0], '1aa! a')
  452. self.assertEqual(regex.search(br"\d\D\w\W\s\S", b"1aa! a",
  453. regex.LOCALE)[0], b'1aa! a')
  454. self.assertEqual(regex.search(r"\d\D\w\W\s\S", "1aa! a",
  455. regex.UNICODE)[0], '1aa! a')
  456. def test_bigcharset(self):
  457. self.assertEqual(regex.match(r"([\u2222\u2223])", "\u2222")[1],
  458. '\u2222')
  459. self.assertEqual(regex.match(r"([\u2222\u2223])", "\u2222",
  460. regex.UNICODE)[1], '\u2222')
  461. self.assertEqual("".join(regex.findall(".",
  462. "e\xe8\xe9\xea\xeb\u0113\u011b\u0117", flags=regex.UNICODE)),
  463. 'e\xe8\xe9\xea\xeb\u0113\u011b\u0117')
  464. self.assertEqual("".join(regex.findall(r"[e\xe8\xe9\xea\xeb\u0113\u011b\u0117]",
  465. "e\xe8\xe9\xea\xeb\u0113\u011b\u0117", flags=regex.UNICODE)),
  466. 'e\xe8\xe9\xea\xeb\u0113\u011b\u0117')
  467. self.assertEqual("".join(regex.findall(r"e|\xe8|\xe9|\xea|\xeb|\u0113|\u011b|\u0117",
  468. "e\xe8\xe9\xea\xeb\u0113\u011b\u0117", flags=regex.UNICODE)),
  469. 'e\xe8\xe9\xea\xeb\u0113\u011b\u0117')
  470. def test_anyall(self):
  471. self.assertEqual(regex.match("a.b", "a\nb", regex.DOTALL)[0], "a\nb")
  472. self.assertEqual(regex.match("a.*b", "a\n\nb", regex.DOTALL)[0],
  473. "a\n\nb")
  474. def test_non_consuming(self):
  475. self.assertEqual(regex.match(r"(a(?=\s[^a]))", "a b")[1], 'a')
  476. self.assertEqual(regex.match(r"(a(?=\s[^a]*))", "a b")[1], 'a')
  477. self.assertEqual(regex.match(r"(a(?=\s[abc]))", "a b")[1], 'a')
  478. self.assertEqual(regex.match(r"(a(?=\s[abc]*))", "a bc")[1], 'a')
  479. self.assertEqual(regex.match(r"(a)(?=\s\1)", "a a")[1], 'a')
  480. self.assertEqual(regex.match(r"(a)(?=\s\1*)", "a aa")[1], 'a')
  481. self.assertEqual(regex.match(r"(a)(?=\s(abc|a))", "a a")[1], 'a')
  482. self.assertEqual(regex.match(r"(a(?!\s[^a]))", "a a")[1], 'a')
  483. self.assertEqual(regex.match(r"(a(?!\s[abc]))", "a d")[1], 'a')
  484. self.assertEqual(regex.match(r"(a)(?!\s\1)", "a b")[1], 'a')
  485. self.assertEqual(regex.match(r"(a)(?!\s(abc|a))", "a b")[1], 'a')
  486. def test_ignore_case(self):
  487. self.assertEqual(regex.match("abc", "ABC", regex.I)[0], 'ABC')
  488. self.assertEqual(regex.match(b"abc", b"ABC", regex.I)[0], b'ABC')
  489. self.assertEqual(regex.match(r"(a\s[^a]*)", "a bb", regex.I)[1],
  490. 'a bb')
  491. self.assertEqual(regex.match(r"(a\s[abc])", "a b", regex.I)[1], 'a b')
  492. self.assertEqual(regex.match(r"(a\s[abc]*)", "a bb", regex.I)[1],
  493. 'a bb')
  494. self.assertEqual(regex.match(r"((a)\s\2)", "a a", regex.I)[1], 'a a')
  495. self.assertEqual(regex.match(r"((a)\s\2*)", "a aa", regex.I)[1],
  496. 'a aa')
  497. self.assertEqual(regex.match(r"((a)\s(abc|a))", "a a", regex.I)[1],
  498. 'a a')
  499. self.assertEqual(regex.match(r"((a)\s(abc|a)*)", "a aa", regex.I)[1],
  500. 'a aa')
  501. # Issue 3511.
  502. self.assertEqual(regex.match(r"[Z-a]", "_").span(), (0, 1))
  503. self.assertEqual(regex.match(r"(?i)[Z-a]", "_").span(), (0, 1))
  504. self.assertEqual(bool(regex.match(r"(?i)nao", "nAo")), True)
  505. self.assertEqual(bool(regex.match(r"(?i)n\xE3o", "n\xC3o")), True)
  506. self.assertEqual(bool(regex.match(r"(?i)n\xE3o", "N\xC3O")), True)
  507. self.assertEqual(bool(regex.match(r"(?i)s", "\u017F")), True)
  508. def test_case_folding(self):
  509. self.assertEqual(regex.search(r"(?fi)ss", "SS").span(), (0, 2))
  510. self.assertEqual(regex.search(r"(?fi)SS", "ss").span(), (0, 2))
  511. self.assertEqual(regex.search(r"(?fi)SS",
  512. "\N{LATIN SMALL LETTER SHARP S}").span(), (0, 1))
  513. self.assertEqual(regex.search(r"(?fi)\N{LATIN SMALL LETTER SHARP S}",
  514. "SS").span(), (0, 2))
  515. self.assertEqual(regex.search(r"(?fi)\N{LATIN SMALL LIGATURE ST}",
  516. "ST").span(), (0, 2))
  517. self.assertEqual(regex.search(r"(?fi)ST",
  518. "\N{LATIN SMALL LIGATURE ST}").span(), (0, 1))
  519. self.assertEqual(regex.search(r"(?fi)ST",
  520. "\N{LATIN SMALL LIGATURE LONG S T}").span(), (0, 1))
  521. self.assertEqual(regex.search(r"(?fi)SST",
  522. "\N{LATIN SMALL LETTER SHARP S}t").span(), (0, 2))
  523. self.assertEqual(regex.search(r"(?fi)SST",
  524. "s\N{LATIN SMALL LIGATURE LONG S T}").span(), (0, 2))
  525. self.assertEqual(regex.search(r"(?fi)SST",
  526. "s\N{LATIN SMALL LIGATURE ST}").span(), (0, 2))
  527. self.assertEqual(regex.search(r"(?fi)\N{LATIN SMALL LIGATURE ST}",
  528. "SST").span(), (1, 3))
  529. self.assertEqual(regex.search(r"(?fi)SST",
  530. "s\N{LATIN SMALL LIGATURE ST}").span(), (0, 2))
  531. self.assertEqual(regex.search(r"(?fi)FFI",
  532. "\N{LATIN SMALL LIGATURE FFI}").span(), (0, 1))
  533. self.assertEqual(regex.search(r"(?fi)FFI",
  534. "\N{LATIN SMALL LIGATURE FF}i").span(), (0, 2))
  535. self.assertEqual(regex.search(r"(?fi)FFI",
  536. "f\N{LATIN SMALL LIGATURE FI}").span(), (0, 2))
  537. self.assertEqual(regex.search(r"(?fi)\N{LATIN SMALL LIGATURE FFI}",
  538. "FFI").span(), (0, 3))
  539. self.assertEqual(regex.search(r"(?fi)\N{LATIN SMALL LIGATURE FF}i",
  540. "FFI").span(), (0, 3))
  541. self.assertEqual(regex.search(r"(?fi)f\N{LATIN SMALL LIGATURE FI}",
  542. "FFI").span(), (0, 3))
  543. sigma = "\u03A3\u03C3\u03C2"
  544. for ch1 in sigma:
  545. for ch2 in sigma:
  546. if not regex.match(r"(?fi)" + ch1, ch2):
  547. self.fail()
  548. self.assertEqual(bool(regex.search(r"(?iV1)ff", "\uFB00\uFB01")),
  549. True)
  550. self.assertEqual(bool(regex.search(r"(?iV1)ff", "\uFB01\uFB00")),
  551. True)
  552. self.assertEqual(bool(regex.search(r"(?iV1)fi", "\uFB00\uFB01")),
  553. True)
  554. self.assertEqual(bool(regex.search(r"(?iV1)fi", "\uFB01\uFB00")),
  555. True)
  556. self.assertEqual(bool(regex.search(r"(?iV1)fffi", "\uFB00\uFB01")),
  557. True)
  558. self.assertEqual(bool(regex.search(r"(?iV1)f\uFB03",
  559. "\uFB00\uFB01")), True)
  560. self.assertEqual(bool(regex.search(r"(?iV1)ff", "\uFB00\uFB01")),
  561. True)
  562. self.assertEqual(bool(regex.search(r"(?iV1)fi", "\uFB00\uFB01")),
  563. True)
  564. self.assertEqual(bool(regex.search(r"(?iV1)fffi", "\uFB00\uFB01")),
  565. True)
  566. self.assertEqual(bool(regex.search(r"(?iV1)f\uFB03",
  567. "\uFB00\uFB01")), True)
  568. self.assertEqual(bool(regex.search(r"(?iV1)f\uFB01", "\uFB00i")),
  569. True)
  570. self.assertEqual(bool(regex.search(r"(?iV1)f\uFB01", "\uFB00i")),
  571. True)
  572. self.assertEqual(regex.findall(r"(?iV0)\m(?:word){e<=3}\M(?<!\m(?:word){e<=1}\M)",
  573. "word word2 word word3 word word234 word23 word"), ["word234",
  574. "word23"])
  575. self.assertEqual(regex.findall(r"(?iV1)\m(?:word){e<=3}\M(?<!\m(?:word){e<=1}\M)",
  576. "word word2 word word3 word word234 word23 word"), ["word234",
  577. "word23"])
  578. self.assertEqual(regex.search(r"(?fi)a\N{LATIN SMALL LIGATURE FFI}ne",
  579. " affine ").span(), (2, 8))
  580. self.assertEqual(regex.search(r"(?fi)a(?:\N{LATIN SMALL LIGATURE FFI}|x)ne",
  581. " affine ").span(), (2, 8))
  582. self.assertEqual(regex.search(r"(?fi)a(?:\N{LATIN SMALL LIGATURE FFI}|xy)ne",
  583. " affine ").span(), (2, 8))
  584. self.assertEqual(regex.search(r"(?fi)a\L<options>ne", "affine",
  585. options=["\N{LATIN SMALL LIGATURE FFI}"]).span(), (0, 6))
  586. self.assertEqual(regex.search(r"(?fi)a\L<options>ne",
  587. "a\N{LATIN SMALL LIGATURE FFI}ne", options=["ffi"]).span(), (0, 4))
  588. def test_category(self):
  589. self.assertEqual(regex.match(r"(\s)", " ")[1], ' ')
  590. def test_not_literal(self):
  591. self.assertEqual(regex.search(r"\s([^a])", " b")[1], 'b')
  592. self.assertEqual(regex.search(r"\s([^a]*)", " bb")[1], 'bb')
  593. def test_search_coverage(self):
  594. self.assertEqual(regex.search(r"\s(b)", " b")[1], 'b')
  595. self.assertEqual(regex.search(r"a\s", "a ")[0], 'a ')
  596. def test_re_escape(self):
  597. p = ""
  598. self.assertEqual(regex.escape(p), p)
  599. for i in range(0, 256):
  600. p += chr(i)
  601. self.assertEqual(bool(regex.match(regex.escape(chr(i)), chr(i))),
  602. True)
  603. self.assertEqual(regex.match(regex.escape(chr(i)), chr(i)).span(),
  604. (0, 1))
  605. pat = regex.compile(regex.escape(p))
  606. self.assertEqual(pat.match(p).span(), (0, 256))
  607. def test_re_escape_byte(self):
  608. p = b""
  609. self.assertEqual(regex.escape(p), p)
  610. for i in range(0, 256):
  611. b = bytes([i])
  612. p += b
  613. self.assertEqual(bool(regex.match(regex.escape(b), b)), True)
  614. self.assertEqual(regex.match(regex.escape(b), b).span(), (0, 1))
  615. pat = regex.compile(regex.escape(p))
  616. self.assertEqual(pat.match(p).span(), (0, 256))
  617. def test_constants(self):
  618. if regex.I != regex.IGNORECASE:
  619. self.fail()
  620. if regex.L != regex.LOCALE:
  621. self.fail()
  622. if regex.M != regex.MULTILINE:
  623. self.fail()
  624. if regex.S != regex.DOTALL:
  625. self.fail()
  626. if regex.X != regex.VERBOSE:
  627. self.fail()
  628. def test_flags(self):
  629. for flag in [regex.I, regex.M, regex.X, regex.S, regex.L]:
  630. self.assertEqual(repr(type(regex.compile('^pattern$', flag))),
  631. self.PATTERN_CLASS)
  632. def test_sre_character_literals(self):
  633. for i in [0, 8, 16, 32, 64, 127, 128, 255]:
  634. self.assertEqual(bool(regex.match(r"\%03o" % i, chr(i))), True)
  635. self.assertEqual(bool(regex.match(r"\%03o0" % i, chr(i) + "0")),
  636. True)
  637. self.assertEqual(bool(regex.match(r"\%03o8" % i, chr(i) + "8")),
  638. True)
  639. self.assertEqual(bool(regex.match(r"\x%02x" % i, chr(i))), True)
  640. self.assertEqual(bool(regex.match(r"\x%02x0" % i, chr(i) + "0")),
  641. True)
  642. self.assertEqual(bool(regex.match(r"\x%02xz" % i, chr(i) + "z")),
  643. True)
  644. self.assertRaisesRegex(regex.error, self.INVALID_GROUP_REF, lambda:
  645. regex.match(r"\911", ""))
  646. def test_sre_character_class_literals(self):
  647. for i in [0, 8, 16, 32, 64, 127, 128, 255]:
  648. self.assertEqual(bool(regex.match(r"[\%03o]" % i, chr(i))), True)
  649. self.assertEqual(bool(regex.match(r"[\%03o0]" % i, chr(i))), True)
  650. self.assertEqual(bool(regex.match(r"[\%03o8]" % i, chr(i))), True)
  651. self.assertEqual(bool(regex.match(r"[\x%02x]" % i, chr(i))), True)
  652. self.assertEqual(bool(regex.match(r"[\x%02x0]" % i, chr(i))), True)
  653. self.assertEqual(bool(regex.match(r"[\x%02xz]" % i, chr(i))), True)
  654. self.assertRaisesRegex(regex.error, self.BAD_OCTAL_ESCAPE, lambda:
  655. regex.match(r"[\911]", ""))
  656. def test_bug_113254(self):
  657. self.assertEqual(regex.match(r'(a)|(b)', 'b').start(1), -1)
  658. self.assertEqual(regex.match(r'(a)|(b)', 'b').end(1), -1)
  659. self.assertEqual(regex.match(r'(a)|(b)', 'b').span(1), (-1, -1))
  660. def test_bug_527371(self):
  661. # Bug described in patches 527371/672491.
  662. self.assertEqual(regex.match(r'(a)?a','a').lastindex, None)
  663. self.assertEqual(regex.match(r'(a)(b)?b','ab').lastindex, 1)
  664. self.assertEqual(regex.match(r'(?P<a>a)(?P<b>b)?b','ab').lastgroup,
  665. 'a')
  666. self.assertEqual(regex.match("(?P<a>a(b))", "ab").lastgroup, 'a')
  667. self.assertEqual(regex.match("((a))", "a").lastindex, 1)
  668. def test_bug_545855(self):
  669. # Bug 545855 -- This pattern failed to cause a compile error as it
  670. # should, instead provoking a TypeError.
  671. self.assertRaisesRegex(regex.error, self.BAD_SET, lambda:
  672. regex.compile('foo[a-'))
  673. def test_bug_418626(self):
  674. # Bugs 418626 at al. -- Testing Greg Chapman's addition of op code
  675. # SRE_OP_MIN_REPEAT_ONE for eliminating recursion on simple uses of
  676. # pattern '*?' on a long string.
  677. self.assertEqual(regex.match('.*?c', 10000 * 'ab' + 'cd').end(0),
  678. 20001)
  679. self.assertEqual(regex.match('.*?cd', 5000 * 'ab' + 'c' + 5000 * 'ab' +
  680. 'cde').end(0), 20003)
  681. self.assertEqual(regex.match('.*?cd', 20000 * 'abc' + 'de').end(0),
  682. 60001)
  683. # Non-simple '*?' still used to hit the recursion limit, before the
  684. # non-recursive scheme was implemented.
  685. self.assertEqual(regex.search('(a|b)*?c', 10000 * 'ab' + 'cd').end(0),
  686. 20001)
  687. def test_bug_612074(self):
  688. pat = "[" + regex.escape("\u2039") + "]"
  689. self.assertEqual(regex.compile(pat) and 1, 1)
  690. def test_stack_overflow(self):
  691. # Nasty cases that used to overflow the straightforward recursive
  692. # implementation of repeated groups.
  693. self.assertEqual(regex.match('(x)*', 50000 * 'x')[1], 'x')
  694. self.assertEqual(regex.match('(x)*y', 50000 * 'x' + 'y')[1], 'x')
  695. self.assertEqual(regex.match('(x)*?y', 50000 * 'x' + 'y')[1], 'x')
  696. def test_scanner(self):
  697. def s_ident(scanner, token): return token
  698. def s_operator(scanner, token): return "op%s" % token
  699. def s_float(scanner, token): return float(token)
  700. def s_int(scanner, token): return int(token)
  701. scanner = regex.Scanner([(r"[a-zA-Z_]\w*", s_ident), (r"\d+\.\d*",
  702. s_float), (r"\d+", s_int), (r"=|\+|-|\*|/", s_operator), (r"\s+",
  703. None), ])
  704. self.assertEqual(repr(type(scanner.scanner.scanner("").pattern)),
  705. self.PATTERN_CLASS)
  706. self.assertEqual(scanner.scan("sum = 3*foo + 312.50 + bar"), (['sum',
  707. 'op=', 3, 'op*', 'foo', 'op+', 312.5, 'op+', 'bar'], ''))
  708. def test_bug_448951(self):
  709. # Bug 448951 (similar to 429357, but with single char match).
  710. # (Also test greedy matches.)
  711. for op in '', '?', '*':
  712. self.assertEqual(regex.match(r'((.%s):)?z' % op, 'z')[:], ('z',
  713. None, None))
  714. self.assertEqual(regex.match(r'((.%s):)?z' % op, 'a:z')[:], ('a:z',
  715. 'a:', 'a'))
  716. def test_bug_725106(self):
  717. # Capturing groups in alternatives in repeats.
  718. self.assertEqual(regex.match('^((a)|b)*', 'abc')[:], ('ab', 'b', 'a'))
  719. self.assertEqual(regex.match('^(([ab])|c)*', 'abc')[:], ('abc', 'c',
  720. 'b'))
  721. self.assertEqual(regex.match('^((d)|[ab])*', 'abc')[:], ('ab', 'b',
  722. None))
  723. self.assertEqual(regex.match('^((a)c|[ab])*', 'abc')[:], ('ab', 'b',
  724. None))
  725. self.assertEqual(regex.match('^((a)|b)*?c', 'abc')[:], ('abc', 'b',
  726. 'a'))
  727. self.assertEqual(regex.match('^(([ab])|c)*?d', 'abcd')[:], ('abcd',
  728. 'c', 'b'))
  729. self.assertEqual(regex.match('^((d)|[ab])*?c', 'abc')[:], ('abc', 'b',
  730. None))
  731. self.assertEqual(regex.match('^((a)c|[ab])*?c', 'abc')[:], ('abc', 'b',
  732. None))
  733. def test_bug_725149(self):
  734. # Mark_stack_base restoring before restoring marks.
  735. self.assertEqual(regex.match('(a)(?:(?=(b)*)c)*', 'abb')[:], ('a', 'a',
  736. None))
  737. self.assertEqual(regex.match('(a)((?!(b)*))*', 'abb')[:], ('a', 'a',
  738. None, None))
  739. def test_bug_764548(self):
  740. # Bug 764548, regex.compile() barfs on str/unicode subclasses.
  741. class my_unicode(str): pass
  742. pat = regex.compile(my_unicode("abc"))
  743. self.assertEqual(pat.match("xyz"), None)
  744. def test_finditer(self):
  745. it = regex.finditer(r":+", "a:b::c:::d")
  746. self.assertEqual([item[0] for item in it], [':', '::', ':::'])
  747. def test_bug_926075(self):
  748. if regex.compile('bug_926075') is regex.compile(b'bug_926075'):
  749. self.fail()
  750. def test_bug_931848(self):
  751. pattern = "[\u002E\u3002\uFF0E\uFF61]"
  752. self.assertEqual(regex.compile(pattern).split("a.b.c"), ['a', 'b',
  753. 'c'])
  754. def test_bug_581080(self):
  755. it = regex.finditer(r"\s", "a b")
  756. self.assertEqual(next(it).span(), (1, 2))
  757. self.assertRaises(StopIteration, lambda: next(it))
  758. scanner = regex.compile(r"\s").scanner("a b")
  759. self.assertEqual(scanner.search().span(), (1, 2))
  760. self.assertEqual(scanner.search(), None)
  761. def test_bug_817234(self):
  762. it = regex.finditer(r".*", "asdf")
  763. self.assertEqual(next(it).span(), (0, 4))
  764. self.assertEqual(next(it).span(), (4, 4))
  765. self.assertRaises(StopIteration, lambda: next(it))
  766. def test_empty_array(self):
  767. # SF buf 1647541.
  768. import array
  769. for typecode in 'bBuhHiIlLfd':
  770. a = array.array(typecode)
  771. self.assertEqual(regex.compile(b"bla").match(a), None)
  772. self.assertEqual(regex.compile(b"").match(a)[1 : ], ())
  773. def test_inline_flags(self):
  774. # Bug #1700.
  775. upper_char = chr(0x1ea0) # Latin Capital Letter A with Dot Below
  776. lower_char = chr(0x1ea1) # Latin Small Letter A with Dot Below
  777. p = regex.compile(upper_char, regex.I | regex.U)
  778. self.assertEqual(bool(p.match(lower_char)), True)
  779. p = regex.compile(lower_char, regex.I | regex.U)
  780. self.assertEqual(bool(p.match(upper_char)), True)
  781. p = regex.compile('(?i)' + upper_char, regex.U)
  782. self.assertEqual(bool(p.match(lower_char)), True)
  783. p = regex.compile('(?i)' + lower_char, regex.U)
  784. self.assertEqual(bool(p.match(upper_char)), True)
  785. p = regex.compile('(?iu)' + upper_char)
  786. self.assertEqual(bool(p.match(lower_char)), True)
  787. p = regex.compile('(?iu)' + lower_char)
  788. self.assertEqual(bool(p.match(upper_char)), True)
  789. self.assertEqual(bool(regex.match(r"(?i)a", "A")), True)
  790. self.assertEqual(bool(regex.match(r"a(?i)", "A")), True)
  791. self.assertEqual(bool(regex.match(r"(?iV1)a", "A")), True)
  792. self.assertEqual(regex.match(r"a(?iV1)", "A"), None)
  793. def test_dollar_matches_twice(self):
  794. # $ matches the end of string, and just before the terminating \n.
  795. pattern = regex.compile('$')
  796. self.assertEqual(pattern.sub('#', 'a\nb\n'), 'a\nb#\n#')
  797. self.assertEqual(pattern.sub('#', 'a\nb\nc'), 'a\nb\nc#')
  798. self.assertEqual(pattern.sub('#', '\n'), '#\n#')
  799. pattern = regex.compile('$', regex.MULTILINE)
  800. self.assertEqual(pattern.sub('#', 'a\nb\n' ), 'a#\nb#\n#')
  801. self.assertEqual(pattern.sub('#', 'a\nb\nc'), 'a#\nb#\nc#')
  802. self.assertEqual(pattern.sub('#', '\n'), '#\n#')
  803. def test_bytes_str_mixing(self):
  804. # Mixing str and bytes is disallowed.
  805. pat = regex.compile('.')
  806. bpat = regex.compile(b'.')
  807. self.assertRaisesRegex(TypeError, self.STR_PAT_ON_BYTES, lambda:
  808. pat.match(b'b'))
  809. self.assertRaisesRegex(TypeError, self.BYTES_PAT_ON_STR, lambda:
  810. bpat.match('b'))
  811. self.assertRaisesRegex(TypeError, self.STR_PAT_BYTES_TEMPL, lambda:
  812. pat.sub(b'b', 'c'))
  813. self.assertRaisesRegex(TypeError, self.STR_PAT_ON_BYTES, lambda:
  814. pat.sub('b', b'c'))
  815. self.assertRaisesRegex(TypeError, self.STR_PAT_ON_BYTES, lambda:
  816. pat.sub(b'b', b'c'))
  817. self.assertRaisesRegex(TypeError, self.BYTES_PAT_ON_STR, lambda:
  818. bpat.sub(b'b', 'c'))
  819. self.assertRaisesRegex(TypeError, self.BYTES_PAT_STR_TEMPL, lambda:
  820. bpat.sub('b', b'c'))
  821. self.assertRaisesRegex(TypeError, self.BYTES_PAT_ON_STR, lambda:
  822. bpat.sub('b', 'c'))
  823. self.assertRaisesRegex(ValueError, self.BYTES_PAT_UNI_FLAG, lambda:
  824. regex.compile(br'\w', regex.UNICODE))
  825. self.assertRaisesRegex(ValueError, self.BYTES_PAT_UNI_FLAG, lambda:
  826. regex.compile(br'(?u)\w'))
  827. self.assertRaisesRegex(ValueError, self.MIXED_FLAGS, lambda:
  828. regex.compile(r'\w', regex.UNICODE | regex.ASCII))
  829. self.assertRaisesRegex(ValueError, self.MIXED_FLAGS, lambda:
  830. regex.compile(r'(?u)\w', regex.ASCII))
  831. self.assertRaisesRegex(ValueError, self.MIXED_FLAGS, lambda:
  832. regex.compile(r'(?a)\w', regex.UNICODE))
  833. self.assertRaisesRegex(ValueError, self.MIXED_FLAGS, lambda:
  834. regex.compile(r'(?au)\w'))
  835. def test_ascii_and_unicode_flag(self):
  836. # String patterns.
  837. for flags in (0, regex.UNICODE):
  838. pat = regex.compile('\xc0', flags | regex.IGNORECASE)
  839. self.assertEqual(bool(pat.match('\xe0')), True)
  840. pat = regex.compile(r'\w', flags)
  841. self.assertEqual(bool(pat.match('\xe0')), True)
  842. pat = regex.compile('\xc0', regex.ASCII | regex.IGNORECASE)
  843. self.assertEqual(pat.match('\xe0'), None)
  844. pat = regex.compile('(?a)\xc0', regex.IGNORECASE)
  845. self.assertEqual(pat.match('\xe0'), None)
  846. pat = regex.compile(r'\w', regex.ASCII)
  847. self.assertEqual(pat.match('\xe0'), None)
  848. pat = regex.compile(r'(?a)\w')
  849. self.assertEqual(pat.match('\xe0'), None)
  850. # Bytes patterns.
  851. for flags in (0, regex.ASCII):
  852. pat = regex.compile(b'\xc0', flags | regex.IGNORECASE)
  853. self.assertEqual(pat.match(b'\xe0'), None)
  854. pat = regex.compile(br'\w')
  855. self.assertEqual(pat.match(b'\xe0'), None)
  856. self.assertRaisesRegex(ValueError, self.MIXED_FLAGS, lambda:
  857. regex.compile(r'(?au)\w'))
  858. def test_subscripting_match(self):
  859. m = regex.match(r'(?<a>\w)', 'xy')
  860. if not m:
  861. self.fail("Failed: expected match but returned None")
  862. elif not m or m[0] != m.group(0) or m[1] != m.group(1):
  863. self.fail("Failed")
  864. if not m:
  865. self.fail("Failed: expected match but returned None")
  866. elif m[:] != ('x', 'x'):
  867. self.fail("Failed: expected \"('x', 'x')\" but got {} instead".format(ascii(m[:])))
  868. def test_new_named_groups(self):
  869. m0 = regex.match(r'(?P<a>\w)', 'x')
  870. m1 = regex.match(r'(?<a>\w)', 'x')
  871. if not (m0 and m1 and m0[:] == m1[:]):
  872. self.fail("Failed")
  873. def test_properties(self):
  874. self.assertEqual(regex.match(b'(?ai)\xC0', b'\xE0'), None)
  875. self.assertEqual(regex.match(br'(?ai)\xC0', b'\xE0'), None)
  876. self.assertEqual(regex.match(br'(?a)\w', b'\xE0'), None)
  877. self.assertEqual(bool(regex.match(r'\w', '\xE0')), True)
  878. # Dropped the following test. It's not possible to determine what the
  879. # correct result should be in the general case.
  880. # self.assertEqual(bool(regex.match(br'(?L)\w', b'\xE0')),
  881. # b'\xE0'.isalnum())
  882. self.assertEqual(bool(regex.match(br'(?L)\d', b'0')), True)
  883. self.assertEqual(bool(regex.match(br'(?L)\s', b' ')), True)
  884. self.assertEqual(bool(regex.match(br'(?L)\w', b'a')), True)
  885. self.assertEqual(regex.match(br'(?L)\d', b'?'), None)
  886. self.assertEqual(regex.match(br'(?L)\s', b'?'), None)
  887. self.assertEqual(regex.match(br'(?L)\w', b'?'), None)
  888. self.assertEqual(regex.match(br'(?L)\D', b'0'), None)
  889. self.assertEqual(regex.match(br'(?L)\S', b' '), None)
  890. self.assertEqual(regex.match(br'(?L)\W', b'a'), None)
  891. self.assertEqual(bool(regex.match(br'(?L)\D', b'?')), True)
  892. self.assertEqual(bool(regex.match(br'(?L)\S', b'?')), True)
  893. self.assertEqual(bool(regex.match(br'(?L)\W', b'?')), True)
  894. self.assertEqual(bool(regex.match(r'\p{Cyrillic}',
  895. '\N{CYRILLIC CAPITAL LETTER A}')), True)
  896. self.assertEqual(bool(regex.match(r'(?i)\p{Cyrillic}',
  897. '\N{CYRILLIC CAPITAL LETTER A}')), True)
  898. self.assertEqual(bool(regex.match(r'\p{IsCyrillic}',
  899. '\N{CYRILLIC CAPITAL LETTER A}')), True)
  900. self.assertEqual(bool(regex.match(r'\p{Script=Cyrillic}',
  901. '\N{CYRILLIC CAPITAL LETTER A}')), True)
  902. self.assertEqual(bool(regex.match(r'\p{InCyrillic}',
  903. '\N{CYRILLIC CAPITAL LETTER A}')), True)
  904. self.assertEqual(bool(regex.match(r'\p{Block=Cyrillic}',
  905. '\N{CYRILLIC CAPITAL LETTER A}')), True)
  906. self.assertEqual(bool(regex.match(r'[[:Cyrillic:]]',
  907. '\N{CYRILLIC CAPITAL LETTER A}')), True)
  908. self.assertEqual(bool(regex.match(r'[[:IsCyrillic:]]',
  909. '\N{CYRILLIC CAPITAL LETTER A}')), True)
  910. self.assertEqual(bool(regex.match(r'[[:Script=Cyrillic:]]',
  911. '\N{CYRILLIC CAPITAL LETTER A}')), True)
  912. self.assertEqual(bool(regex.match(r'[[:InCyrillic:]]',
  913. '\N{CYRILLIC CAPITAL LETTER A}')), True)
  914. self.assertEqual(bool(regex.match(r'[[:Block=Cyrillic:]]',
  915. '\N{CYRILLIC CAPITAL LETTER A}')), True)
  916. self.assertEqual(bool(regex.match(r'\P{Cyrillic}',
  917. '\N{LATIN CAPITAL LETTER A}')), True)
  918. self.assertEqual(bool(regex.match(r'\P{IsCyrillic}',
  919. '\N{LATIN CAPITAL LETTER A}')), True)
  920. self.assertEqual(bool(regex.match(r'\P{Script=Cyrillic}',
  921. '\N{LATIN CAPITAL LETTER A}')), True)
  922. self.assertEqual(bool(regex.match(r'\P{InCyrillic}',
  923. '\N{LATIN CAPITAL LETTER A}')), True)
  924. self.assertEqual(bool(regex.match(r'\P{Block=Cyrillic}',
  925. '\N{LATIN CAPITAL LETTER A}')), True)
  926. self.assertEqual(bool(regex.match(r'\p{^Cyrillic}',
  927. '\N{LATIN CAPITAL LETTER A}')), True)
  928. self.assertEqual(bool(regex.match(r'\p{^IsCyrillic}',
  929. '\N{LATIN CAPITAL LETTER A}')), True)
  930. self.assertEqual(bool(regex.match(r'\p{^Script=Cyrillic}',
  931. '\N{LATIN CAPITAL LETTER A}')), True)
  932. self.assertEqual(bool(regex.match(r'\p{^InCyrillic}',
  933. '\N{LATIN CAPITAL LETTER A}')), True)
  934. self.assertEqual(bool(regex.match(r'\p{^Block=Cyrillic}',
  935. '\N{LATIN CAPITAL LETTER A}')), True)
  936. self.assertEqual(bool(regex.match(r'[[:^Cyrillic:]]',
  937. '\N{LATIN CAPITAL LETTER A}')), True)
  938. self.assertEqual(bool(regex.match(r'[[:^IsCyrillic:]]',
  939. '\N{LATIN CAPITAL LETTER A}')), True)
  940. self.assertEqual(bool(regex.match(r'[[:^Script=Cyrillic:]]',
  941. '\N{LATIN CAPITAL LETTER A}')), True)
  942. self.assertEqual(bool(regex.match(r'[[:^InCyrillic:]]',
  943. '\N{LATIN CAPITAL LETTER A}')), True)
  944. self.assertEqual(bool(regex.match(r'[[:^Block=Cyrillic:]]',
  945. '\N{LATIN CAPITAL LETTER A}')), True)
  946. self.assertEqual(bool(regex.match(r'\d', '0')), True)
  947. self.assertEqual(bool(regex.match(r'\s', ' ')), True)
  948. self.assertEqual(bool(regex.match(r'\w', 'A')), True)
  949. self.assertEqual(regex.match(r"\d", "?"), None)
  950. self.assertEqual(regex.match(r"\s", "?"), None)
  951. self.assertEqual(regex.match(r"\w", "?"), None)
  952. self.assertEqual(regex.match(r"\D", "0"), None)
  953. self.assertEqual(regex.match(r"\S", " "), None)
  954. self.assertEqual(regex.match(r"\W", "A"), None)
  955. self.assertEqual(bool(regex.match(r'\D', '?')), True)
  956. self.assertEqual(bool(regex.match(r'\S', '?')), True)
  957. self.assertEqual(bool(regex.match(r'\W', '?')), True)
  958. self.assertEqual(bool(regex.match(r'\p{L}', 'A')), True)
  959. self.assertEqual(bool(regex.match(r'\p{L}', 'a')), True)
  960. self.assertEqual(bool(regex.match(r'\p{Lu}', 'A')), True)
  961. self.assertEqual(bool(regex.match(r'\p{Ll}', 'a')), True)
  962. self.assertEqual(bool(regex.match(r'(?i)a', 'a')), True)
  963. self.assertEqual(bool(regex.match(r'(?i)a', 'A')), True)
  964. self.assertEqual(bool(regex.match(r'\w', '0')), True)
  965. self.assertEqual(bool(regex.match(r'\w', 'a')), True)
  966. self.assertEqual(bool(regex.match(r'\w', '_')), True)
  967. self.assertEqual(regex.match(r"\X", "\xE0").span(), (0, 1))
  968. self.assertEqual(regex.match(r"\X", "a\u0300").span(), (0, 2))
  969. self.assertEqual(regex.findall(r"\X",
  970. "a\xE0a\u0300e\xE9e\u0301"), ['a', '\xe0', 'a\u0300', 'e',
  971. '\xe9', 'e\u0301'])
  972. self.assertEqual(regex.findall(r"\X{3}",
  973. "a\xE0a\u0300e\xE9e\u0301"), ['a\xe0a\u0300', 'e\xe9e\u0301'])
  974. self.assertEqual(regex.findall(r"\X", "\r\r\n\u0301A\u0301"),
  975. ['\r', '\r\n', '\u0301', 'A\u0301'])
  976. self.assertEqual(bool(regex.match(r'\p{Ll}', 'a')), True)
  977. chars_u = "-09AZaz_\u0393\u03b3"
  978. chars_b = b"-09AZaz_"
  979. word_set = set("Ll Lm Lo Lt Lu Mc Me Mn Nd Nl No Pc".split())
  980. tests = [
  981. (r"\w", chars_u, "09AZaz_\u0393\u03b3"),
  982. (r"[[:word:]]", chars_u, "09AZaz_\u0393\u03b3"),
  983. (r"\W", chars_u, "-"),
  984. (r"[[:^word:]]", chars_u, "-"),
  985. (r"\d", chars_u, "09"),
  986. (r"[[:digit:]]", chars_u, "09"),
  987. (r"\D", chars_u, "-AZaz_\u0393\u03b3"),
  988. (r"[[:^digit:]]", chars_u, "-AZaz_\u0393\u03b3"),
  989. (r"[[:alpha:]]", chars_u, "AZaz\u0393\u03b3"),
  990. (r"[[:^alpha:]]", chars_u, "-09_"),
  991. (r"[[:alnum:]]", chars_u, "09AZaz\u0393\u03b3"),
  992. (r"[[:^alnum:]]", chars_u, "-_"),
  993. (r"[[:xdigit:]]", chars_u, "09Aa"),
  994. (r"[[:^xdigit:]]", chars_u, "-Zz_\u0393\u03b3"),
  995. (r"\p{InBasicLatin}", "a\xE1", "a"),
  996. (r"\P{InBasicLatin}", "a\xE1", "\xE1"),
  997. (r"(?i)\p{InBasicLatin}", "a\xE1", "a"),
  998. (r"(?i)\P{InBasicLatin}", "a\xE1", "\xE1"),
  999. (br"(?L)\w", chars_b, b"09AZaz_"),
  1000. (br"(?L)[[:word:]]", chars_b, b"09AZaz_"),
  1001. (br"(?L)\W", chars_b, b"-"),
  1002. (br"(?L)[[:^word:]]", chars_b, b"-"),
  1003. (br"(?L)\d", chars_b, b"09"),
  1004. (br"(?L)[[:digit:]]", chars_b, b"09"),
  1005. (br"(?L)\D", chars_b, b"-AZaz_"),
  1006. (br"(?L)[[:^digit:]]", chars_b, b"-AZaz_"),
  1007. (br"(?L)[[:alpha:]]", chars_b, b"AZaz"),
  1008. (br"(?L)[[:^alpha:]]", chars_b, b"-09_"),
  1009. (br"(?L)[[:alnum:]]", chars_b, b"09AZaz"),
  1010. (br"(?L)[[:^alnum:]]", chars_b, b"-_"),
  1011. (br"(?L)[[:xdigit:]]", chars_b, b"09Aa"),
  1012. (br"(?L)[[:^xdigit:]]", chars_b, b"-Zz_"),
  1013. (br"(?a)\w", chars_b, b"09AZaz_"),
  1014. (br"(?a)[[:word:]]", chars_b, b"09AZaz_"),
  1015. (br"(?a)\W", chars_b, b"-"),
  1016. (br"(?a)[[:^word:]]", chars_b, b"-"),
  1017. (br"(?a)\d", chars_b, b"09"),
  1018. (br"(?a)[[:digit:]]", chars_b, b"09"),
  1019. (br"(?a)\D", chars_b, b"-AZaz_"),
  1020. (br"(?a)[[:^digit:]]", chars_b, b"-AZaz_"),
  1021. (br"(?a)[[:alpha:]]", chars_b, b"AZaz"),
  1022. (br"(?a)[[:^alpha:]]", chars_b, b"-09_"),
  1023. (br"(?a)[[:alnum:]]", chars_b, b"09AZaz"),
  1024. (br"(?a)[[:^alnum:]]", chars_b, b"-_"),
  1025. (br"(?a)[[:xdigit:]]", chars_b, b"09Aa"),
  1026. (br"(?a)[[:^xdigit:]]", chars_b, b"-Zz_"),
  1027. ]
  1028. for pattern, chars, expected in tests:
  1029. try:
  1030. if chars[ : 0].join(regex.findall(pattern, chars)) != expected:
  1031. self.fail("Failed: {}".format(pattern))
  1032. except Exception as e:
  1033. self.fail("Failed: {} raised {}".format(pattern, ascii(e)))
  1034. self.assertEqual(bool(regex.match(r"\p{NumericValue=0}", "0")),
  1035. True)
  1036. self.assertEqual(bool(regex.match(r"\p{NumericValue=1/2}",
  1037. "\N{VULGAR FRACTION ONE HALF}")), True)
  1038. self.assertEqual(bool(regex.match(r"\p{NumericValue=0.5}",
  1039. "\N{VULGAR FRACTION ONE HALF}")), True)
  1040. def test_word_class(self):
  1041. self.assertEqual(regex.findall(r"\w+",
  1042. " \u0939\u093f\u0928\u094d\u0926\u0940,"),
  1043. ['\u0939\u093f\u0928\u094d\u0926\u0940'])
  1044. self.assertEqual(regex.findall(r"\W+",
  1045. " \u0939\u093f\u0928\u094d\u0926\u0940,"), [' ', ','])
  1046. self.assertEqual(regex.split(r"(?V1)\b",
  1047. " \u0939\u093f\u0928\u094d\u0926\u0940,"), [' ',
  1048. '\u0939\u093f\u0928\u094d\u0926\u0940', ','])
  1049. self.assertEqual(regex.split(r"(?V1)\B",
  1050. " \u0939\u093f\u0928\u094d\u0926\u0940,"), ['', ' \u0939',
  1051. '\u093f', '\u0928', '\u094d', '\u0926', '\u0940,', ''])
  1052. def test_search_anchor(self):
  1053. self.assertEqual(regex.findall(r"\G\w{2}", "abcd ef"), ['ab', 'cd'])
  1054. def test_search_reverse(self):
  1055. self.assertEqual(regex.findall(r"(?r).", "abc"), ['c', 'b', 'a'])
  1056. self.assertEqual(regex.findall(r"(?r).", "abc", overlapped=True), ['c',
  1057. 'b', 'a'])
  1058. self.assertEqual(regex.findall(r"(?r)..", "abcde"), ['de', 'bc'])
  1059. self.assertEqual(regex.findall(r"(?r)..", "abcde", overlapped=True),
  1060. ['de', 'cd', 'bc', 'ab'])
  1061. self.assertEqual(regex.findall(r"(?r)(.)(-)(.)", "a-b-c",
  1062. overlapped=True), [("b", "-", "c"), ("a", "-", "b")])
  1063. self.assertEqual([m[0] for m in regex.finditer(r"(?r).", "abc")], ['c',
  1064. 'b', 'a'])
  1065. self.assertEqual([m[0] for m in regex.finditer(r"(?r)..", "abcde",
  1066. overlapped=True)], ['de', 'cd', 'bc', 'ab'])
  1067. self.assertEqual([m[0] for m in regex.finditer(r"(?r).", "abc")], ['c',
  1068. 'b', 'a'])
  1069. self.assertEqual([m[0] for m in regex.finditer(r"(?r)..", "abcde",
  1070. overlapped=True)], ['de', 'cd', 'bc', 'ab'])
  1071. self.assertEqual(regex.findall(r"^|\w+", "foo bar"), ['', 'foo',
  1072. 'bar'])
  1073. self.assertEqual(regex.findall(r"(?V1)^|\w+", "foo bar"), ['', 'foo',
  1074. 'bar'])
  1075. self.assertEqual(regex.findall(r"(?r)^|\w+", "foo bar"), ['bar', 'foo',
  1076. ''])
  1077. self.assertEqual(regex.findall(r"(?rV1)^|\w+", "foo bar"), ['bar',
  1078. 'foo', ''])
  1079. self.assertEqual([m[0] for m in regex.finditer(r"^|\w+", "foo bar")],
  1080. ['', 'foo', 'bar'])
  1081. self.assertEqual([m[0] for m in regex.finditer(r"(?V1)^|\w+",
  1082. "foo bar")], ['', 'foo', 'bar'])
  1083. self.assertEqual([m[0] for m in regex.finditer(r"(?r)^|\w+",
  1084. "foo bar")], ['bar', 'foo', ''])
  1085. self.assertEqual([m[0] for m in regex.finditer(r"(?rV1)^|\w+",
  1086. "foo bar")], ['bar', 'foo', ''])
  1087. self.assertEqual(regex.findall(r"\G\w{2}", "abcd ef"), ['ab', 'cd'])
  1088. self.assertEqual(regex.findall(r".{2}(?<=\G.*)", "abcd"), ['ab', 'cd'])
  1089. self.assertEqual(regex.findall(r"(?r)\G\w{2}", "abcd ef"), [])
  1090. self.assertEqual(regex.findall(r"(?r)\w{2}\G", "abcd ef"), ['ef'])
  1091. self.assertEqual(regex.findall(r"q*", "qqwe"), ['qq', '', '', ''])
  1092. self.assertEqual(regex.findall(r"(?V1)q*", "qqwe"), ['qq', '', '', ''])
  1093. self.assertEqual(regex.findall(r"(?r)q*", "qqwe"), ['', '', 'qq', ''])
  1094. self.assertEqual(regex.findall(r"(?rV1)q*", "qqwe"), ['', '', 'qq',
  1095. ''])
  1096. self.assertEqual(regex.findall(".", "abcd", pos=1, endpos=3), ['b',
  1097. 'c'])
  1098. self.assertEqual(regex.findall(".", "abcd", pos=1, endpos=-1), ['b',
  1099. 'c'])
  1100. self.assertEqual([m[0] for m in regex.finditer(".", "abcd", pos=1,
  1101. endpos=3)], ['b', 'c'])
  1102. self.assertEqual([m[0] for m in regex.finditer(".", "abcd", pos=1,
  1103. endpos=-1)], ['b', 'c'])
  1104. self.assertEqual([m[0] for m in regex.finditer("(?r).", "abcd", pos=1,
  1105. endpos=3)], ['c', 'b'])
  1106. self.assertEqual([m[0] for m in regex.finditer("(?r).", "abcd", pos=1,
  1107. endpos=-1)], ['c', 'b'])
  1108. self.assertEqual(regex.findall("(?r).", "abcd", pos=1, endpos=3), ['c',
  1109. 'b'])
  1110. self.assertEqual(regex.findall("(?r).", "abcd", pos=1, endpos=-1),
  1111. ['c', 'b'])
  1112. self.assertEqual(regex.findall(r"[ab]", "aB", regex.I), ['a', 'B'])
  1113. self.assertEqual(regex.findall(r"(?r)[ab]", "aB", regex.I), ['B', 'a'])
  1114. self.assertEqual(regex.findall(r"(?r).{2}", "abc"), ['bc'])
  1115. self.assertEqual(regex.findall(r"(?r).{2}", "abc", overlapped=True),
  1116. ['bc', 'ab'])
  1117. self.assertEqual(regex.findall(r"(\w+) (\w+)",
  1118. "first second third fourth fifth"), [('first', 'second'), ('third',
  1119. 'fourth')])
  1120. self.assertEqual(regex.findall(r"(?r)(\w+) (\w+)",
  1121. "first second third fourth fifth"), [('fourth', 'fifth'), ('second',
  1122. 'third')])
  1123. self.assertEqual([m[0] for m in regex.finditer(r"(?r).{2}", "abc")],
  1124. ['bc'])
  1125. self.assertEqual([m[0] for m in regex.finditer(r"(?r).{2}", "abc",
  1126. overlapped=True)], ['bc', 'ab'])
  1127. self.assertEqual([m[0] for m in regex.finditer(r"(\w+) (\w+)",
  1128. "first second third fourth fifth")], ['first second',
  1129. 'third fourth'])
  1130. self.assertEqual([m[0] for m in regex.finditer(r"(?r)(\w+) (\w+)",
  1131. "first second third fourth fifth")], ['fourth fifth',
  1132. 'second third'])
  1133. self.assertEqual(regex.search("abcdef", "abcdef").span(), (0, 6))
  1134. self.assertEqual(regex.search("(?r)abcdef", "abcdef").span(), (0, 6))
  1135. self.assertEqual(regex.search("(?i)abcdef", "ABCDEF").span(), (0, 6))
  1136. self.assertEqual(regex.search("(?ir)abcdef", "ABCDEF").span(), (0, 6))
  1137. self.assertEqual(regex.sub(r"(.)", r"\1", "abc"), 'abc')
  1138. self.assertEqual(regex.sub(r"(?r)(.)", r"\1", "abc"), 'abc')
  1139. def test_atomic(self):
  1140. # Issue 433030.
  1141. self.assertEqual(regex.search(r"(?>a*)a", "aa"), None)
  1142. def test_possessive(self):
  1143. # Single-character non-possessive.
  1144. self.assertEqual(regex.search(r"a?a", "a").span(), (0, 1))
  1145. self.assertEqual(regex.search(r"a*a", "aaa").span(), (0, 3))
  1146. self.assertEqual(regex.search(r"a+a", "aaa").span(), (0, 3))
  1147. self.assertEqual(regex.search(r"a{1,3}a", "aaa").span(), (0, 3))
  1148. # Multiple-character non-possessive.
  1149. self.assertEqual(regex.search(r"(?:ab)?ab", "ab").span(), (0, 2))
  1150. self.assertEqual(regex.search(r"(?:ab)*ab", "ababab").span(), (0, 6))
  1151. self.assertEqual(regex.search(r"(?:ab)+ab", "ababab").span(), (0, 6))
  1152. self.assertEqual(regex.search(r"(?:ab){1,3}ab", "ababab").span(), (0,
  1153. 6))
  1154. # Single-character possessive.
  1155. self.assertEqual(regex.search(r"a?+a", "a"), None)
  1156. self.assertEqual(regex.search(r"a*+a", "aaa"), None)
  1157. self.assertEqual(regex.search(r"a++a", "aaa"), None)
  1158. self.assertEqual(regex.search(r"a{1,3}+a", "aaa"), None)
  1159. # Multiple-character possessive.
  1160. self.assertEqual(regex.search(r"(?:ab)?+ab", "ab"), None)
  1161. self.assertEqual(regex.search(r"(?:ab)*+ab", "ababab"), None)
  1162. self.assertEqual(regex.search(r"(?:ab)++ab", "ababab"), None)
  1163. self.assertEqual(regex.search(r"(?:ab){1,3}+ab", "ababab"), None)
  1164. def test_zerowidth(self):
  1165. # Issue 3262.
  1166. if sys.version_info >= (3, 7, 0):
  1167. self.assertEqual(regex.split(r"\b", "a b"), ['', 'a', ' ', 'b',
  1168. ''])
  1169. else:
  1170. self.assertEqual(regex.split(r"\b", "a b"), ['a b'])
  1171. self.assertEqual(regex.split(r"(?V1)\b", "a b"), ['', 'a', ' ', 'b',
  1172. ''])
  1173. # Issue 1647489.
  1174. self.assertEqual(regex.findall(r"^|\w+", "foo bar"), ['', 'foo',
  1175. 'bar'])
  1176. self.assertEqual([m[0] for m in regex.finditer(r"^|\w+", "foo bar")],
  1177. ['', 'foo', 'bar'])
  1178. self.assertEqual(regex.findall(r"(?r)^|\w+", "foo bar"), ['bar',
  1179. 'foo', ''])
  1180. self.assertEqual([m[0] for m in regex.finditer(r"(?r)^|\w+",
  1181. "foo bar")], ['bar', 'foo', ''])
  1182. self.assertEqual(regex.findall(r"(?V1)^|\w+", "foo bar"), ['', 'foo',
  1183. 'bar'])
  1184. self.assertEqual([m[0] for m in regex.finditer(r"(?V1)^|\w+",
  1185. "foo bar")], ['', 'foo', 'bar'])
  1186. self.assertEqual(regex.findall(r"(?rV1)^|\w+", "foo bar"), ['bar',
  1187. 'foo', ''])
  1188. self.assertEqual([m[0] for m in regex.finditer(r"(?rV1)^|\w+",
  1189. "foo bar")], ['bar', 'foo', ''])
  1190. if sys.version_info >= (3, 7, 0):
  1191. self.assertEqual(regex.split("", "xaxbxc"), ['', 'x', 'a', 'x',
  1192. 'b', 'x', 'c', ''])
  1193. self.assertEqual([m for m in regex.splititer("", "xaxbxc")], ['',
  1194. 'x', 'a', 'x', 'b', 'x', 'c', ''])
  1195. else:
  1196. self.assertEqual(regex.split("", "xaxbxc"), ['xaxbxc'])
  1197. self.assertEqual([m for m in regex.splititer("", "xaxbxc")],
  1198. ['xaxbxc'])
  1199. if sys.version_info >= (3, 7, 0):
  1200. self.assertEqual(regex.split("(?r)", "xaxbxc"), ['', 'c', 'x', 'b',
  1201. 'x', 'a', 'x', ''])
  1202. self.assertEqual([m for m in regex.splititer("(?r)", "xaxbxc")],
  1203. ['', 'c', 'x', 'b', 'x', 'a', 'x', ''])
  1204. else:
  1205. self.assertEqual(regex.split("(?r)", "xaxbxc"), ['xaxbxc'])
  1206. self.assertEqual([m for m in regex.splititer("(?r)", "xaxbxc")],
  1207. ['xaxbxc'])
  1208. self.assertEqual(regex.split("(?V1)", "xaxbxc"), ['', 'x', 'a', 'x',
  1209. 'b', 'x', 'c', ''])
  1210. self.assertEqual([m for m in regex.splititer("(?V1)", "xaxbxc")], ['',
  1211. 'x', 'a', 'x', 'b', 'x', 'c', ''])
  1212. self.assertEqual(regex.split("(?rV1)", "xaxbxc"), ['', 'c', 'x', 'b',
  1213. 'x', 'a', 'x', ''])
  1214. self.assertEqual([m for m in regex.splititer("(?rV1)", "xaxbxc")], ['',
  1215. 'c', 'x', 'b', 'x', 'a', 'x', ''])
  1216. def test_scoped_and_inline_flags(self):
  1217. # Issues 433028, 433024, 433027.
  1218. self.assertEqual(regex.search(r"(?i)Ab", "ab").span(), (0, 2))
  1219. self.assertEqual(regex.search(r"(?i:A)b", "ab").span(), (0, 2))
  1220. self.assertEqual(regex.search(r"A(?i)b", "ab").span(), (0, 2))
  1221. self.assertEqual(regex.search(r"A(?iV1)b", "ab"), None)
  1222. self.assertRaisesRegex(regex.error, self.CANT_TURN_OFF, lambda:
  1223. regex.search(r"(?V0-i)Ab", "ab", flags=regex.I))
  1224. self.assertEqual(regex.search(r"(?V0)Ab", "ab"), None)
  1225. self.assertEqual(regex.search(r"(?V1)Ab", "ab"), None)
  1226. self.assertEqual(regex.search(r"(?V1-i)Ab", "ab", flags=regex.I), None)
  1227. self.assertEqual(regex.search(r"(?-i:A)b", "ab", flags=regex.I), None)
  1228. self.assertEqual(regex.search(r"A(?V1-i)b", "ab",
  1229. flags=regex.I).span(), (0, 2))
  1230. def test_repeated_repeats(self):
  1231. # Issue 2537.
  1232. self.assertEqual(regex.search(r"(?:a+)+", "aaa").span(), (0, 3))
  1233. self.assertEqual(regex.search(r"(?:(?:ab)+c)+", "abcabc").span(), (0,
  1234. 6))
  1235. def test_lookbehind(self):
  1236. self.assertEqual(regex.search(r"123(?<=a\d+)", "a123").span(), (1, 4))
  1237. self.assertEqual(regex.search(r"123(?<=a\d+)", "b123"), None)
  1238. self.assertEqual(regex.search(r"123(?<!a\d+)", "a123"), None)
  1239. self.assertEqual(regex.search(r"123(?<!a\d+)", "b123").span(), (1, 4))
  1240. self.assertEqual(bool(regex.match("(a)b(?<=b)(c)", "abc")), True)
  1241. self.assertEqual(regex.match("(a)b(?<=c)(c)", "abc"), None)
  1242. self.assertEqual(bool(regex.match("(a)b(?=c)(c)", "abc")), True)
  1243. self.assertEqual(regex.match("(a)b(?=b)(c)", "abc"), None)
  1244. self.assertEqual(regex.match("(?:(a)|(x))b(?<=(?(2)x|c))c", "abc"),
  1245. None)
  1246. self.assertEqual(regex.match("(?:(a)|(x))b(?<=(?(2)b|x))c", "abc"),
  1247. None)
  1248. self.assertEqual(bool(regex.match("(?:(a)|(x))b(?<=(?(2)x|b))c",
  1249. "abc")), True)
  1250. self.assertEqual(regex.match("(?:(a)|(x))b(?<=(?(1)c|x))c", "abc"),
  1251. None)
  1252. self.assertEqual(bool(regex.match("(?:(a)|(x))b(?<=(?(1)b|x))c",
  1253. "abc")), True)
  1254. self.assertEqual(bool(regex.match("(?:(a)|(x))b(?=(?(2)x|c))c",
  1255. "abc")), True)
  1256. self.assertEqual(regex.match("(?:(a)|(x))b(?=(?(2)c|x))c", "abc"),
  1257. None)
  1258. self.assertEqual(bool(regex.match("(?:(a)|(x))b(?=(?(2)x|c))c",
  1259. "abc")), True)
  1260. self.assertEqual(regex.match("(?:(a)|(x))b(?=(?(1)b|x))c", "abc"),
  1261. None)
  1262. self.assertEqual(bool(regex.match("(?:(a)|(x))b(?=(?(1)c|x))c",
  1263. "abc")), True)
  1264. self.assertEqual(regex.match("(a)b(?<=(?(2)x|c))(c)", "abc"), None)
  1265. self.assertEqual(regex.match("(a)b(?<=(?(2)b|x))(c)", "abc"), None)
  1266. self.assertEqual(regex.match("(a)b(?<=(?(1)c|x))(c)", "abc"), None)
  1267. self.assertEqual(bool(regex.match("(a)b(?<=(?(1)b|x))(c)", "abc")),
  1268. True)
  1269. self.assertEqual(bool(regex.match("(a)b(?=(?(2)x|c))(c)", "abc")),
  1270. True)
  1271. self.assertEqual(regex.match("(a)b(?=(?(2)b|x))(c)", "abc"), None)
  1272. self.assertEqual(bool(regex.match("(a)b(?=(?(1)c|x))(c)", "abc")),
  1273. True)
  1274. self.assertEqual(repr(type(regex.compile(r"(a)\2(b)"))),
  1275. self.PATTERN_CLASS)
  1276. def test_unmatched_in_sub(self):
  1277. # Issue 1519638.
  1278. if sys.version_info >= (3, 7, 0):
  1279. self.assertEqual(regex.sub(r"(?V0)(x)?(y)?", r"\2-\1", "xy"),
  1280. 'y-x-')
  1281. else:
  1282. self.assertEqual(regex.sub(r"(?V0)(x)?(y)?", r"\2-\1", "xy"),
  1283. 'y-x')
  1284. self.assertEqual(regex.sub(r"(?V1)(x)?(y)?", r"\2-\1", "xy"), 'y-x-')
  1285. if sys.version_info >= (3, 7, 0):
  1286. self.assertEqual(regex.sub(r"(?V0)(x)?(y)?", r"\2-\1", "x"), '-x-')
  1287. else:
  1288. self.assertEqual(regex.sub(r"(?V0)(x)?(y)?", r"\2-\1", "x"), '-x')
  1289. self.assertEqual(regex.sub(r"(?V1)(x)?(y)?", r"\2-\1", "x"), '-x-')
  1290. if sys.version_info >= (3, 7, 0):
  1291. self.assertEqual(regex.sub(r"(?V0)(x)?(y)?", r"\2-\1", "y"), 'y--')
  1292. else:
  1293. self.assertEqual(regex.sub(r"(?V0)(x)?(y)?", r"\2-\1", "y"), 'y-')
  1294. self.assertEqual(regex.sub(r"(?V1)(x)?(y)?", r"\2-\1", "y"), 'y--')
  1295. def test_bug_10328 (self):
  1296. # Issue 10328.
  1297. pat = regex.compile(r'(?mV0)(?P<trailing_ws>[ \t]+\r*$)|(?P<no_final_newline>(?<=[^\n])\Z)')
  1298. if sys.version_info >= (3, 7, 0):
  1299. self.assertEqual(pat.subn(lambda m: '<' + m.lastgroup + '>',
  1300. 'foobar '), ('foobar<trailing_ws><no_final_newline>', 2))
  1301. else:
  1302. self.assertEqual(pat.subn(lambda m: '<' + m.lastgroup + '>',
  1303. 'foobar '), ('foobar<trailing_ws>', 1))
  1304. self.assertEqual([m.group() for m in pat.finditer('foobar ')], [' ',
  1305. ''])
  1306. pat = regex.compile(r'(?mV1)(?P<trailing_ws>[ \t]+\r*$)|(?P<no_final_newline>(?<=[^\n])\Z)')
  1307. self.assertEqual(pat.subn(lambda m: '<' + m.lastgroup + '>',
  1308. 'foobar '), ('foobar<trailing_ws><no_final_newline>', 2))
  1309. self.assertEqual([m.group() for m in pat.finditer('foobar ')], [' ',
  1310. ''])
  1311. def test_overlapped(self):
  1312. self.assertEqual(regex.findall(r"..", "abcde"), ['ab', 'cd'])
  1313. self.assertEqual(regex.findall(r"..", "abcde", overlapped=True), ['ab',
  1314. 'bc', 'cd', 'de'])
  1315. self.assertEqual(regex.findall(r"(?r)..", "abcde"), ['de', 'bc'])
  1316. self.assertEqual(regex.findall(r"(?r)..", "abcde", overlapped=True),
  1317. ['de', 'cd', 'bc', 'ab'])
  1318. self.assertEqual(regex.findall(r"(.)(-)(.)", "a-b-c", overlapped=True),
  1319. [("a", "-", "b"), ("b", "-", "c")])
  1320. self.assertEqual([m[0] for m in regex.finditer(r"..", "abcde")], ['ab',
  1321. 'cd'])
  1322. self.assertEqual([m[0] for m in regex.finditer(r"..", "abcde",
  1323. overlapped=True)], ['ab', 'bc', 'cd', 'de'])
  1324. self.assertEqual([m[0] for m in regex.finditer(r"(?r)..", "abcde")],
  1325. ['de', 'bc'])
  1326. self.assertEqual([m[0] for m in regex.finditer(r"(?r)..", "abcde",
  1327. overlapped=True)], ['de', 'cd', 'bc', 'ab'])
  1328. self.assertEqual([m.groups() for m in regex.finditer(r"(.)(-)(.)",
  1329. "a-b-c", overlapped=True)], [("a", "-", "b"), ("b", "-", "c")])
  1330. self.assertEqual([m.groups() for m in regex.finditer(r"(?r)(.)(-)(.)",
  1331. "a-b-c", overlapped=True)], [("b", "-", "c"), ("a", "-", "b")])
  1332. def test_splititer(self):
  1333. self.assertEqual(regex.split(r",", "a,b,,c,"), ['a', 'b', '', 'c', ''])
  1334. self.assertEqual([m for m in regex.splititer(r",", "a,b,,c,")], ['a',
  1335. 'b', '', 'c', ''])
  1336. def test_grapheme(self):
  1337. self.assertEqual(regex.match(r"\X", "\xE0").span(), (0, 1))
  1338. self.assertEqual(regex.match(r"\X", "a\u0300").span(), (0, 2))
  1339. self.assertEqual(regex.findall(r"\X",
  1340. "a\xE0a\u0300e\xE9e\u0301"), ['a', '\xe0', 'a\u0300', 'e',
  1341. '\xe9', 'e\u0301'])
  1342. self.assertEqual(regex.findall(r"\X{3}",
  1343. "a\xE0a\u0300e\xE9e\u0301"), ['a\xe0a\u0300', 'e\xe9e\u0301'])
  1344. self.assertEqual(regex.findall(r"\X", "\r\r\n\u0301A\u0301"),
  1345. ['\r', '\r\n', '\u0301', 'A\u0301'])
  1346. def test_word_boundary(self):
  1347. text = 'The quick ("brown") fox can\'t jump 32.3 feet, right?'
  1348. self.assertEqual(regex.split(r'(?V1)\b', text), ['', 'The', ' ',
  1349. 'quick', ' ("', 'brown', '") ', 'fox', ' ', 'can', "'", 't',
  1350. ' ', 'jump', ' ', '32', '.', '3', ' ', 'feet', ', ',
  1351. 'right', '?'])
  1352. self.assertEqual(regex.split(r'(?V1w)\b', text), ['', 'The', ' ',
  1353. 'quick', ' ', '(', '"', 'brown', '"', ')', ' ', 'fox', ' ',
  1354. "can't", ' ', 'jump', ' ', '32.3', ' ', 'feet', ',', ' ',
  1355. 'right', '?', ''])
  1356. text = "The fox"
  1357. self.assertEqual(regex.split(r'(?V1)\b', text), ['', 'The', ' ',
  1358. 'fox', ''])
  1359. self.assertEqual(regex.split(r'(?V1w)\b', text), ['', 'The', ' ',
  1360. ' ', 'fox', ''])
  1361. text = "can't aujourd'hui l'objectif"
  1362. self.assertEqual(regex.split(r'(?V1)\b', text), ['', 'can', "'",
  1363. 't', ' ', 'aujourd', "'", 'hui', ' ', 'l', "'", 'objectif',
  1364. ''])
  1365. self.assertEqual(regex.split(r'(?V1w)\b', text), ['', "can't", ' ',
  1366. "aujourd'hui", ' ', "l'", 'objectif', ''])
  1367. def test_line_boundary(self):
  1368. self.assertEqual(regex.findall(r".+", "Line 1\nLine 2\n"), ["Line 1",
  1369. "Line 2"])
  1370. self.assertEqual(regex.findall(r".+", "Line 1\rLine 2\r"),
  1371. ["Line 1\rLine 2\r"])
  1372. self.assertEqual(regex.findall(r".+", "Line 1\r\nLine 2\r\n"),
  1373. ["Line 1\r", "Line 2\r"])
  1374. self.assertEqual(regex.findall(r"(?w).+", "Line 1\nLine 2\n"),
  1375. ["Line 1", "Line 2"])
  1376. self.assertEqual(regex.findall(r"(?w).+", "Line 1\rLine 2\r"),
  1377. ["Line 1", "Line 2"])
  1378. self.assertEqual(regex.findall(r"(?w).+", "Line 1\r\nLine 2\r\n"),
  1379. ["Line 1", "Line 2"])
  1380. self.assertEqual(regex.search(r"^abc", "abc").start(), 0)
  1381. self.assertEqual(regex.search(r"^abc", "\nabc"), None)
  1382. self.assertEqual(regex.search(r"^abc", "\rabc"), None)
  1383. self.assertEqual(regex.search(r"(?w)^abc", "abc").start(), 0)
  1384. self.assertEqual(regex.search(r"(?w)^abc", "\nabc"), None)
  1385. self.assertEqual(regex.search(r"(?w)^abc", "\rabc"), None)
  1386. self.assertEqual(regex.search(r"abc$", "abc").start(), 0)
  1387. self.assertEqual(regex.search(r"abc$", "abc\n").start(), 0)
  1388. self.assertEqual(regex.search(r"abc$", "abc\r"), None)
  1389. self.assertEqual(regex.search(r"(?w)abc$", "abc").start(), 0)
  1390. self.assertEqual(regex.search(r"(?w)abc$", "abc\n").start(), 0)
  1391. self.assertEqual(regex.search(r"(?w)abc$", "abc\r").start(), 0)
  1392. self.assertEqual(regex.search(r"(?m)^abc", "abc").start(), 0)
  1393. self.assertEqual(regex.search(r"(?m)^abc", "\nabc").start(), 1)
  1394. self.assertEqual(regex.search(r"(?m)^abc", "\rabc"), None)
  1395. self.assertEqual(regex.search(r"(?mw)^abc", "abc").start(), 0)
  1396. self.assertEqual(regex.search(r"(?mw)^abc", "\nabc").start(), 1)
  1397. self.assertEqual(regex.search(r"(?mw)^abc", "\rabc").start(), 1)
  1398. self.assertEqual(regex.search(r"(?m)abc$", "abc").start(), 0)
  1399. self.assertEqual(regex.search(r"(?m)abc$", "abc\n").start(), 0)
  1400. self.assertEqual(regex.search(r"(?m)abc$", "abc\r"), None)
  1401. self.assertEqual(regex.search(r"(?mw)abc$", "abc").start(), 0)
  1402. self.assertEqual(regex.search(r"(?mw)abc$", "abc\n").start(), 0)
  1403. self.assertEqual(regex.search(r"(?mw)abc$", "abc\r").start(), 0)
  1404. def test_branch_reset(self):
  1405. self.assertEqual(regex.match(r"(?:(a)|(b))(c)", "ac").groups(), ('a',
  1406. None, 'c'))
  1407. self.assertEqual(regex.match(r"(?:(a)|(b))(c)", "bc").groups(), (None,
  1408. 'b', 'c'))
  1409. self.assertEqual(regex.match(r"(?:(?<a>a)|(?<b>b))(?<c>c)",
  1410. "ac").groups(), ('a', None, 'c'))
  1411. self.assertEqual(regex.match(r"(?:(?<a>a)|(?<b>b))(?<c>c)",
  1412. "bc").groups(), (None, 'b', 'c'))
  1413. self.assertEqual(regex.match(r"(?<a>a)(?:(?<b>b)|(?<c>c))(?<d>d)",
  1414. "abd").groups(), ('a', 'b', None, 'd'))
  1415. self.assertEqual(regex.match(r"(?<a>a)(?:(?<b>b)|(?<c>c))(?<d>d)",
  1416. "acd").groups(), ('a', None, 'c', 'd'))
  1417. self.assertEqual(regex.match(r"(a)(?:(b)|(c))(d)", "abd").groups(),
  1418. ('a', 'b', None, 'd'))
  1419. self.assertEqual(regex.match(r"(a)(?:(b)|(c))(d)", "acd").groups(),
  1420. ('a', None, 'c', 'd'))
  1421. self.assertEqual(regex.match(r"(a)(?|(b)|(b))(d)", "abd").groups(),
  1422. ('a', 'b', 'd'))
  1423. self.assertEqual(regex.match(r"(?|(?<a>a)|(?<b>b))(c)", "ac").groups(),
  1424. ('a', None, 'c'))
  1425. self.assertEqual(regex.match(r"(?|(?<a>a)|(?<b>b))(c)", "bc").groups(),
  1426. (None, 'b', 'c'))
  1427. self.assertEqual(regex.match(r"(?|(?<a>a)|(?<a>b))(c)", "ac").groups(),
  1428. ('a', 'c'))
  1429. self.assertEqual(regex.match(r"(?|(?<a>a)|(?<a>b))(c)", "bc").groups(),
  1430. ('b', 'c'))
  1431. self.assertEqual(regex.match(r"(?|(?<a>a)(?<b>b)|(?<b>c)(?<a>d))(e)",
  1432. "abe").groups(), ('a', 'b', 'e'))
  1433. self.assertEqual(regex.match(r"(?|(?<a>a)(?<b>b)|(?<b>c)(?<a>d))(e)",
  1434. "cde").groups(), ('d', 'c', 'e'))
  1435. self.assertEqual(regex.match(r"(?|(?<a>a)(?<b>b)|(?<b>c)(d))(e)",
  1436. "abe").groups(), ('a', 'b', 'e'))
  1437. self.assertEqual(regex.match(r"(?|(?<a>a)(?<b>b)|(?<b>c)(d))(e)",
  1438. "cde").groups(), ('d', 'c', 'e'))
  1439. self.assertEqual(regex.match(r"(?|(?<a>a)(?<b>b)|(c)(d))(e)",
  1440. "abe").groups(), ('a', 'b', 'e'))
  1441. self.assertEqual(regex.match(r"(?|(?<a>a)(?<b>b)|(c)(d))(e)",
  1442. "cde").groups(), ('c', 'd', 'e'))
  1443. # Hg issue 87: Allow duplicate names of groups
  1444. self.assertEqual(regex.match(r"(?|(?<a>a)(?<b>b)|(c)(?<a>d))(e)",
  1445. "abe").groups(), ("a", "b", "e"))
  1446. self.assertEqual(regex.match(r"(?|(?<a>a)(?<b>b)|(c)(?<a>d))(e)",
  1447. "abe").capturesdict(), {"a": ["a"], "b": ["b"]})
  1448. self.assertEqual(regex.match(r"(?|(?<a>a)(?<b>b)|(c)(?<a>d))(e)",
  1449. "cde").groups(), ("d", None, "e"))
  1450. self.assertEqual(regex.match(r"(?|(?<a>a)(?<b>b)|(c)(?<a>d))(e)",
  1451. "cde").capturesdict(), {"a": ["c", "d"], "b": []})
  1452. def test_set(self):
  1453. self.assertEqual(regex.match(r"[a]", "a").span(), (0, 1))
  1454. self.assertEqual(regex.match(r"(?i)[a]", "A").span(), (0, 1))
  1455. self.assertEqual(regex.match(r"[a-b]", r"a").span(), (0, 1))
  1456. self.assertEqual(regex.match(r"(?i)[a-b]", r"A").span(), (0, 1))
  1457. self.assertEqual(regex.sub(r"(?V0)([][])", r"-", "a[b]c"), "a-b-c")
  1458. self.assertEqual(regex.findall(r"[\p{Alpha}]", "a0"), ["a"])
  1459. self.assertEqual(regex.findall(r"(?i)[\p{Alpha}]", "A0"), ["A"])
  1460. self.assertEqual(regex.findall(r"[a\p{Alpha}]", "ab0"), ["a", "b"])
  1461. self.assertEqual(regex.findall(r"[a\P{Alpha}]", "ab0"), ["a", "0"])
  1462. self.assertEqual(regex.findall(r"(?i)[a\p{Alpha}]", "ab0"), ["a",
  1463. "b"])
  1464. self.assertEqual(regex.findall(r"(?i)[a\P{Alpha}]", "ab0"), ["a",
  1465. "0"])
  1466. self.assertEqual(regex.findall(r"[a-b\p{Alpha}]", "abC0"), ["a",
  1467. "b", "C"])
  1468. self.assertEqual(regex.findall(r"(?i)[a-b\p{Alpha}]", "AbC0"), ["A",
  1469. "b", "C"])
  1470. self.assertEqual(regex.findall(r"[\p{Alpha}]", "a0"), ["a"])
  1471. self.assertEqual(regex.findall(r"[\P{Alpha}]", "a0"), ["0"])
  1472. self.assertEqual(regex.findall(r"[^\p{Alpha}]", "a0"), ["0"])
  1473. self.assertEqual(regex.findall(r"[^\P{Alpha}]", "a0"), ["a"])
  1474. self.assertEqual("".join(regex.findall(r"[^\d-h]", "a^b12c-h")),
  1475. 'a^bc')
  1476. self.assertEqual("".join(regex.findall(r"[^\dh]", "a^b12c-h")),
  1477. 'a^bc-')
  1478. self.assertEqual("".join(regex.findall(r"[^h\s\db]", "a^b 12c-h")),
  1479. 'a^c-')
  1480. self.assertEqual("".join(regex.findall(r"[^b\w]", "a b")), ' ')
  1481. self.assertEqual("".join(regex.findall(r"[^b\S]", "a b")), ' ')
  1482. self.assertEqual("".join(regex.findall(r"[^8\d]", "a 1b2")), 'a b')
  1483. all_chars = "".join(chr(c) for c in range(0x100))
  1484. self.assertEqual(len(regex.findall(r"\p{ASCII}", all_chars)), 128)
  1485. self.assertEqual(len(regex.findall(r"\p{Letter}", all_chars)),
  1486. 117)
  1487. self.assertEqual(len(regex.findall(r"\p{Digit}", all_chars)), 10)
  1488. # Set operators
  1489. self.assertEqual(len(regex.findall(r"(?V1)[\p{ASCII}&&\p{Letter}]",
  1490. all_chars)), 52)
  1491. self.assertEqual(len(regex.findall(r"(?V1)[\p{ASCII}&&\p{Alnum}&&\p{Letter}]",
  1492. all_chars)), 52)
  1493. self.assertEqual(len(regex.findall(r"(?V1)[\p{ASCII}&&\p{Alnum}&&\p{Digit}]",
  1494. all_chars)), 10)
  1495. self.assertEqual(len(regex.findall(r"(?V1)[\p{ASCII}&&\p{Cc}]",
  1496. all_chars)), 33)
  1497. self.assertEqual(len(regex.findall(r"(?V1)[\p{ASCII}&&\p{Graph}]",
  1498. all_chars)), 94)
  1499. self.assertEqual(len(regex.findall(r"(?V1)[\p{ASCII}--\p{Cc}]",
  1500. all_chars)), 95)
  1501. self.assertEqual(len(regex.findall(r"[\p{Letter}\p{Digit}]",
  1502. all_chars)), 127)
  1503. self.assertEqual(len(regex.findall(r"(?V1)[\p{Letter}||\p{Digit}]",
  1504. all_chars)), 127)
  1505. self.assertEqual(len(regex.findall(r"\p{HexDigit}", all_chars)),
  1506. 22)
  1507. self.assertEqual(len(regex.findall(r"(?V1)[\p{HexDigit}~~\p{Digit}]",
  1508. all_chars)), 12)
  1509. self.assertEqual(len(regex.findall(r"(?V1)[\p{Digit}~~\p{HexDigit}]",
  1510. all_chars)), 12)
  1511. self.assertEqual(repr(type(regex.compile(r"(?V0)([][-])"))),
  1512. self.PATTERN_CLASS)
  1513. self.assertEqual(regex.findall(r"(?V1)[[a-z]--[aei]]", "abc"), ["b",
  1514. "c"])
  1515. self.assertEqual(regex.findall(r"(?iV1)[[a-z]--[aei]]", "abc"), ["b",
  1516. "c"])
  1517. self.assertEqual(regex.findall(r"(?V1)[\w--a]","abc"), ["b", "c"])
  1518. self.assertEqual(regex.findall(r"(?iV1)[\w--a]","abc"), ["b", "c"])
  1519. def test_various(self):
  1520. tests = [
  1521. # Test ?P< and ?P= extensions.
  1522. ('(?P<foo_123', '', '', regex.error, self.MISSING_GT), # Unterminated group identifier.
  1523. ('(?P<1>a)', '', '', regex.error, self.BAD_GROUP_NAME), # Begins with a digit.
  1524. ('(?P<!>a)', '', '', regex.error, self.BAD_GROUP_NAME), # Begins with an illegal char.
  1525. ('(?P<foo!>a)', '', '', regex.error, self.BAD_GROUP_NAME), # Begins with an illegal char.
  1526. # Same tests, for the ?P= form.
  1527. ('(?P<foo_123>a)(?P=foo_123', 'aa', '', regex.error,
  1528. self.MISSING_RPAREN),
  1529. ('(?P<foo_123>a)(?P=1)', 'aa', '1', ascii('a')),
  1530. ('(?P<foo_123>a)(?P=0)', 'aa', '', regex.error,
  1531. self.BAD_GROUP_NAME),
  1532. ('(?P<foo_123>a)(?P=-1)', 'aa', '', regex.error,
  1533. self.BAD_GROUP_NAME),
  1534. ('(?P<foo_123>a)(?P=!)', 'aa', '', regex.error,
  1535. self.BAD_GROUP_NAME),
  1536. ('(?P<foo_123>a)(?P=foo_124)', 'aa', '', regex.error,
  1537. self.UNKNOWN_GROUP), # Backref to undefined group.
  1538. ('(?P<foo_123>a)', 'a', '1', ascii('a')),
  1539. ('(?P<foo_123>a)(?P=foo_123)', 'aa', '1', ascii('a')),
  1540. # Mal-formed \g in pattern treated as literal for compatibility.
  1541. (r'(?<foo_123>a)\g<foo_123', 'aa', '', ascii(None)),
  1542. (r'(?<foo_123>a)\g<1>', 'aa', '1', ascii('a')),
  1543. (r'(?<foo_123>a)\g<!>', 'aa', '', ascii(None)),
  1544. (r'(?<foo_123>a)\g<foo_124>', 'aa', '', regex.error,
  1545. self.UNKNOWN_GROUP), # Backref to undefined group.
  1546. ('(?<foo_123>a)', 'a', '1', ascii('a')),
  1547. (r'(?<foo_123>a)\g<foo_123>', 'aa', '1', ascii('a')),
  1548. # Test octal escapes.
  1549. ('\\1', 'a', '', regex.error, self.INVALID_GROUP_REF), # Backreference.
  1550. ('[\\1]', '\1', '0', "'\\x01'"), # Character.
  1551. ('\\09', chr(0) + '9', '0', ascii(chr(0) + '9')),
  1552. ('\\141', 'a', '0', ascii('a')),
  1553. ('(a)(b)(c)(d)(e)(f)(g)(h)(i)(j)(k)(l)\\119', 'abcdefghijklk9',
  1554. '0,11', ascii(('abcdefghijklk9', 'k'))),
  1555. # Test \0 is handled everywhere.
  1556. (r'\0', '\0', '0', ascii('\0')),
  1557. (r'[\0a]', '\0', '0', ascii('\0')),
  1558. (r'[a\0]', '\0', '0', ascii('\0')),
  1559. (r'[^a\0]', '\0', '', ascii(None)),
  1560. # Test various letter escapes.
  1561. (r'\a[\b]\f\n\r\t\v', '\a\b\f\n\r\t\v', '0',
  1562. ascii('\a\b\f\n\r\t\v')),
  1563. (r'[\a][\b][\f][\n][\r][\t][\v]', '\a\b\f\n\r\t\v', '0',
  1564. ascii('\a\b\f\n\r\t\v')),
  1565. (r'\c\e\g\h\i\j\k\o\p\q\y\z', 'ceghijkopqyz', '0',
  1566. ascii('ceghijkopqyz')),
  1567. (r'\xff', '\377', '0', ascii(chr(255))),
  1568. # New \x semantics.
  1569. (r'\x00ffffffffffffff', '\377', '', ascii(None)),
  1570. (r'\x00f', '\017', '', ascii(None)),
  1571. (r'\x00fe', '\376', '', ascii(None)),
  1572. (r'\x00ff', '\377', '', ascii(None)),
  1573. (r'\t\n\v\r\f\a\g', '\t\n\v\r\f\ag', '0', ascii('\t\n\v\r\f\ag')),
  1574. ('\t\n\v\r\f\a\\g', '\t\n\v\r\f\ag', '0', ascii('\t\n\v\r\f\ag')),
  1575. (r'\t\n\v\r\f\a', '\t\n\v\r\f\a', '0', ascii(chr(9) + chr(10) +
  1576. chr(11) + chr(13) + chr(12) + chr(7))),
  1577. (r'[\t][\n][\v][\r][\f][\b]', '\t\n\v\r\f\b', '0',
  1578. ascii('\t\n\v\r\f\b')),
  1579. (r"^\w+=(\\[\000-\277]|[^\n\\])*",
  1580. "SRC=eval.c g.c blah blah blah \\\\\n\tapes.c", '0',
  1581. ascii("SRC=eval.c g.c blah blah blah \\\\")),
  1582. # Test that . only matches \n in DOTALL mode.
  1583. ('a.b', 'acb', '0', ascii('acb')),
  1584. ('a.b', 'a\nb', '', ascii(None)),
  1585. ('a.*b', 'acc\nccb', '', ascii(None)),
  1586. ('a.{4,5}b', 'acc\nccb', '', ascii(None)),
  1587. ('a.b', 'a\rb', '0', ascii('a\rb')),
  1588. # The new behaviour is that the inline flag affects only what follows.
  1589. ('a.b(?s)', 'a\nb', '0', ascii('a\nb')),
  1590. ('a.b(?sV1)', 'a\nb', '', ascii(None)),
  1591. ('(?s)a.b', 'a\nb', '0', ascii('a\nb')),
  1592. ('a.*(?s)b', 'acc\nccb', '0', ascii('acc\nccb')),
  1593. ('a.*(?sV1)b', 'acc\nccb', '', ascii(None)),
  1594. ('(?s)a.*b', 'acc\nccb', '0', ascii('acc\nccb')),
  1595. ('(?s)a.{4,5}b', 'acc\nccb', '0', ascii('acc\nccb')),
  1596. (')', '', '', regex.error, self.TRAILING_CHARS), # Unmatched right bracket.
  1597. ('', '', '0', "''"), # Empty pattern.
  1598. ('abc', 'abc', '0', ascii('abc')),
  1599. ('abc', 'xbc', '', ascii(None)),
  1600. ('abc', 'axc', '', ascii(None)),
  1601. ('abc', 'abx', '', ascii(None)),
  1602. ('abc', 'xabcy', '0', ascii('abc')),
  1603. ('abc', 'ababc', '0', ascii('abc')),
  1604. ('ab*c', 'abc', '0', ascii('abc')),
  1605. ('ab*bc', 'abc', '0', ascii('abc')),
  1606. ('ab*bc', 'abbc', '0', ascii('abbc')),
  1607. ('ab*bc', 'abbbbc', '0', ascii('abbbbc')),
  1608. ('ab+bc', 'abbc', '0', ascii('abbc')),
  1609. ('ab+bc', 'abc', '', ascii(None)),
  1610. ('ab+bc', 'abq', '', ascii(None)),
  1611. ('ab+bc', 'abbbbc', '0', ascii('abbbbc')),
  1612. ('ab?bc', 'abbc', '0', ascii('abbc')),
  1613. ('ab?bc', 'abc', '0', ascii('abc')),
  1614. ('ab?bc', 'abbbbc', '', ascii(None)),
  1615. ('ab?c', 'abc', '0', ascii('abc')),
  1616. ('^abc$', 'abc', '0', ascii('abc')),
  1617. ('^abc$', 'abcc', '', ascii(None)),
  1618. ('^abc', 'abcc', '0', ascii('abc')),
  1619. ('^abc$', 'aabc', '', ascii(None)),
  1620. ('abc$', 'aabc', '0', ascii('abc')),
  1621. ('^', 'abc', '0', ascii('')),
  1622. ('$', 'abc', '0', ascii('')),
  1623. ('a.c', 'abc', '0', ascii('abc')),
  1624. ('a.c', 'axc', '0', ascii('axc')),
  1625. ('a.*c', 'axyzc', '0', ascii('axyzc')),
  1626. ('a.*c', 'axyzd', '', ascii(None)),
  1627. ('a[bc]d', 'abc', '', ascii(None)),
  1628. ('a[bc]d', 'abd', '0', ascii('abd')),
  1629. ('a[b-d]e', 'abd', '', ascii(None)),
  1630. ('a[b-d]e', 'ace', '0', ascii('ace')),
  1631. ('a[b-d]', 'aac', '0', ascii('ac')),
  1632. ('a[-b]', 'a-', '0', ascii('a-')),
  1633. ('a[\\-b]', 'a-', '0', ascii('a-')),
  1634. ('a[b-]', 'a-', '0', ascii('a-')),
  1635. ('a[]b', '-', '', regex.error, self.BAD_SET),
  1636. ('a[', '-', '', regex.error, self.BAD_SET),
  1637. ('a\\', '-', '', regex.error, self.BAD_ESCAPE),
  1638. ('abc)', '-', '', regex.error, self.TRAILING_CHARS),
  1639. ('(abc', '-', '', regex.error, self.MISSING_RPAREN),
  1640. ('a]', 'a]', '0', ascii('a]')),
  1641. ('a[]]b', 'a]b', '0', ascii('a]b')),
  1642. ('a[]]b', 'a]b', '0', ascii('a]b')),
  1643. ('a[^bc]d', 'aed', '0', ascii('aed')),
  1644. ('a[^bc]d', 'abd', '', ascii(None)),
  1645. ('a[^-b]c', 'adc', '0', ascii('adc')),
  1646. ('a[^-b]c', 'a-c', '', ascii(None)),
  1647. ('a[^]b]c', 'a]c', '', ascii(None)),
  1648. ('a[^]b]c', 'adc', '0', ascii('adc')),
  1649. ('\\ba\\b', 'a-', '0', ascii('a')),
  1650. ('\\ba\\b', '-a', '0', ascii('a')),
  1651. ('\\ba\\b', '-a-', '0', ascii('a')),
  1652. ('\\by\\b', 'xy', '', ascii(None)),
  1653. ('\\by\\b', 'yz', '', ascii(None)),
  1654. ('\\by\\b', 'xyz', '', ascii(None)),
  1655. ('x\\b', 'xyz', '', ascii(None)),
  1656. ('x\\B', 'xyz', '0', ascii('x')),
  1657. ('\\Bz', 'xyz', '0', ascii('z')),
  1658. ('z\\B', 'xyz', '', ascii(None)),
  1659. ('\\Bx', 'xyz', '', ascii(None)),
  1660. ('\\Ba\\B', 'a-', '', ascii(None)),
  1661. ('\\Ba\\B', '-a', '', ascii(None)),
  1662. ('\\Ba\\B', '-a-', '', ascii(None)),
  1663. ('\\By\\B', 'xy', '', ascii(None)),
  1664. ('\\By\\B', 'yz', '', ascii(None)),
  1665. ('\\By\\b', 'xy', '0', ascii('y')),
  1666. ('\\by\\B', 'yz', '0', ascii('y')),
  1667. ('\\By\\B', 'xyz', '0', ascii('y')),
  1668. ('ab|cd', 'abc', '0', ascii('ab')),
  1669. ('ab|cd', 'abcd', '0', ascii('ab')),
  1670. ('()ef', 'def', '0,1', ascii(('ef', ''))),
  1671. ('$b', 'b', '', ascii(None)),
  1672. ('a\\(b', 'a(b', '', ascii(('a(b',))),
  1673. ('a\\(*b', 'ab', '0', ascii('ab')),
  1674. ('a\\(*b', 'a((b', '0', ascii('a((b')),
  1675. ('a\\\\b', 'a\\b', '0', ascii('a\\b')),
  1676. ('((a))', 'abc', '0,1,2', ascii(('a', 'a', 'a'))),
  1677. ('(a)b(c)', 'abc', '0,1,2', ascii(('abc', 'a', 'c'))),
  1678. ('a+b+c', 'aabbabc', '0', ascii('abc')),
  1679. ('(a+|b)*', 'ab', '0,1', ascii(('ab', 'b'))),
  1680. ('(a+|b)+', 'ab', '0,1', ascii(('ab', 'b'))),
  1681. ('(a+|b)?', 'ab', '0,1', ascii(('a', 'a'))),
  1682. (')(', '-', '', regex.error, self.TRAILING_CHARS),
  1683. ('[^ab]*', 'cde', '0', ascii('cde')),
  1684. ('abc', '', '', ascii(None)),
  1685. ('a*', '', '0', ascii('')),
  1686. ('a|b|c|d|e', 'e', '0', ascii('e')),
  1687. ('(a|b|c|d|e)f', 'ef', '0,1', ascii(('ef', 'e'))),
  1688. ('abcd*efg', 'abcdefg', '0', ascii('abcdefg')),
  1689. ('ab*', 'xabyabbbz', '0', ascii('ab')),
  1690. ('ab*', 'xayabbbz', '0', ascii('a')),
  1691. ('(ab|cd)e', 'abcde', '0,1', ascii(('cde', 'cd'))),
  1692. ('[abhgefdc]ij', 'hij', '0', ascii('hij')),
  1693. ('^(ab|cd)e', 'abcde', '', ascii(None)),
  1694. ('(abc|)ef', 'abcdef', '0,1', ascii(('ef', ''))),
  1695. ('(a|b)c*d', 'abcd', '0,1', ascii(('bcd', 'b'))),
  1696. ('(ab|ab*)bc', 'abc', '0,1', ascii(('abc', 'a'))),
  1697. ('a([bc]*)c*', 'abc', '0,1', ascii(('abc', 'bc'))),
  1698. ('a([bc]*)(c*d)', 'abcd', '0,1,2', ascii(('abcd', 'bc', 'd'))),
  1699. ('a([bc]+)(c*d)', 'abcd', '0,1,2', ascii(('abcd', 'bc', 'd'))),
  1700. ('a([bc]*)(c+d)', 'abcd', '0,1,2', ascii(('abcd', 'b', 'cd'))),
  1701. ('a[bcd]*dcdcde', 'adcdcde', '0', ascii('adcdcde')),
  1702. ('a[bcd]+dcdcde', 'adcdcde', '', ascii(None)),
  1703. ('(ab|a)b*c', 'abc', '0,1', ascii(('abc', 'ab'))),
  1704. ('((a)(b)c)(d)', 'abcd', '1,2,3,4', ascii(('abc', 'a', 'b', 'd'))),
  1705. ('[a-zA-Z_][a-zA-Z0-9_]*', 'alpha', '0', ascii('alpha')),
  1706. ('^a(bc+|b[eh])g|.h$', 'abh', '0,1', ascii(('bh', None))),
  1707. ('(bc+d$|ef*g.|h?i(j|k))', 'effgz', '0,1,2', ascii(('effgz',
  1708. 'effgz', None))),
  1709. ('(bc+d$|ef*g.|h?i(j|k))', 'ij', '0,1,2', ascii(('ij', 'ij',
  1710. 'j'))),
  1711. ('(bc+d$|ef*g.|h?i(j|k))', 'effg', '', ascii(None)),
  1712. ('(bc+d$|ef*g.|h?i(j|k))', 'bcdd', '', ascii(None)),
  1713. ('(bc+d$|ef*g.|h?i(j|k))', 'reffgz', '0,1,2', ascii(('effgz',
  1714. 'effgz', None))),
  1715. ('(((((((((a)))))))))', 'a', '0', ascii('a')),
  1716. ('multiple words of text', 'uh-uh', '', ascii(None)),
  1717. ('multiple words', 'multiple words, yeah', '0',
  1718. ascii('multiple words')),
  1719. ('(.*)c(.*)', 'abcde', '0,1,2', ascii(('abcde', 'ab', 'de'))),
  1720. ('\\((.*), (.*)\\)', '(a, b)', '2,1', ascii(('b', 'a'))),
  1721. ('[k]', 'ab', '', ascii(None)),
  1722. ('a[-]?c', 'ac', '0', ascii('ac')),
  1723. ('(abc)\\1', 'abcabc', '1', ascii('abc')),
  1724. ('([a-c]*)\\1', 'abcabc', '1', ascii('abc')),
  1725. ('^(.+)?B', 'AB', '1', ascii('A')),
  1726. ('(a+).\\1$', 'aaaaa', '0,1', ascii(('aaaaa', 'aa'))),
  1727. ('^(a+).\\1$', 'aaaa', '', ascii(None)),
  1728. ('(abc)\\1', 'abcabc', '0,1', ascii(('abcabc', 'abc'))),
  1729. ('([a-c]+)\\1', 'abcabc', '0,1', ascii(('abcabc', 'abc'))),
  1730. ('(a)\\1', 'aa', '0,1', ascii(('aa', 'a'))),
  1731. ('(a+)\\1', 'aa', '0,1', ascii(('aa', 'a'))),
  1732. ('(a+)+\\1', 'aa', '0,1', ascii(('aa', 'a'))),
  1733. ('(a).+\\1', 'aba', '0,1', ascii(('aba', 'a'))),
  1734. ('(a)ba*\\1', 'aba', '0,1', ascii(('aba', 'a'))),
  1735. ('(aa|a)a\\1$', 'aaa', '0,1', ascii(('aaa', 'a'))),
  1736. ('(a|aa)a\\1$', 'aaa', '0,1', ascii(('aaa', 'a'))),
  1737. ('(a+)a\\1$', 'aaa', '0,1', ascii(('aaa', 'a'))),
  1738. ('([abc]*)\\1', 'abcabc', '0,1', ascii(('abcabc', 'abc'))),
  1739. ('(a)(b)c|ab', 'ab', '0,1,2', ascii(('ab', None, None))),
  1740. ('(a)+x', 'aaax', '0,1', ascii(('aaax', 'a'))),
  1741. ('([ac])+x', 'aacx', '0,1', ascii(('aacx', 'c'))),
  1742. ('([^/]*/)*sub1/', 'd:msgs/tdir/sub1/trial/away.cpp', '0,1',
  1743. ascii(('d:msgs/tdir/sub1/', 'tdir/'))),
  1744. ('([^.]*)\\.([^:]*):[T ]+(.*)', 'track1.title:TBlah blah blah',
  1745. '0,1,2,3', ascii(('track1.title:TBlah blah blah', 'track1',
  1746. 'title', 'Blah blah blah'))),
  1747. ('([^N]*N)+', 'abNNxyzN', '0,1', ascii(('abNNxyzN', 'xyzN'))),
  1748. ('([^N]*N)+', 'abNNxyz', '0,1', ascii(('abNN', 'N'))),
  1749. ('([abc]*)x', 'abcx', '0,1', ascii(('abcx', 'abc'))),
  1750. ('([abc]*)x', 'abc', '', ascii(None)),
  1751. ('([xyz]*)x', 'abcx', '0,1', ascii(('x', ''))),
  1752. ('(a)+b|aac', 'aac', '0,1', ascii(('aac', None))),
  1753. # Test symbolic groups.
  1754. ('(?P<i d>aaa)a', 'aaaa', '', regex.error, self.BAD_GROUP_NAME),
  1755. ('(?P<id>aaa)a', 'aaaa', '0,id', ascii(('aaaa', 'aaa'))),
  1756. ('(?P<id>aa)(?P=id)', 'aaaa', '0,id', ascii(('aaaa', 'aa'))),
  1757. ('(?P<id>aa)(?P=xd)', 'aaaa', '', regex.error, self.UNKNOWN_GROUP),
  1758. # Character properties.
  1759. (r"\g", "g", '0', ascii('g')),
  1760. (r"\g<1>", "g", '', regex.error, self.INVALID_GROUP_REF),
  1761. (r"(.)\g<1>", "gg", '0', ascii('gg')),
  1762. (r"(.)\g<1>", "gg", '', ascii(('gg', 'g'))),
  1763. (r"\N", "N", '0', ascii('N')),
  1764. (r"\N{LATIN SMALL LETTER A}", "a", '0', ascii('a')),
  1765. (r"\p", "p", '0', ascii('p')),
  1766. (r"\p{Ll}", "a", '0', ascii('a')),
  1767. (r"\P", "P", '0', ascii('P')),
  1768. (r"\P{Lu}", "p", '0', ascii('p')),
  1769. # All tests from Perl.
  1770. ('abc', 'abc', '0', ascii('abc')),
  1771. ('abc', 'xbc', '', ascii(None)),
  1772. ('abc', 'axc', '', ascii(None)),
  1773. ('abc', 'abx', '', ascii(None)),
  1774. ('abc', 'xabcy', '0', ascii('abc')),
  1775. ('abc', 'ababc', '0', ascii('abc')),
  1776. ('ab*c', 'abc', '0', ascii('abc')),
  1777. ('ab*bc', 'abc', '0', ascii('abc')),
  1778. ('ab*bc', 'abbc', '0', ascii('abbc')),
  1779. ('ab*bc', 'abbbbc', '0', ascii('abbbbc')),
  1780. ('ab{0,}bc', 'abbbbc', '0', ascii('abbbbc')),
  1781. ('ab+bc', 'abbc', '0', ascii('abbc')),
  1782. ('ab+bc', 'abc', '', ascii(None)),
  1783. ('ab+bc', 'abq', '', ascii(None)),
  1784. ('ab{1,}bc', 'abq', '', ascii(None)),
  1785. ('ab+bc', 'abbbbc', '0', ascii('abbbbc')),
  1786. ('ab{1,}bc', 'abbbbc', '0', ascii('abbbbc')),
  1787. ('ab{1,3}bc', 'abbbbc', '0', ascii('abbbbc')),
  1788. ('ab{3,4}bc', 'abbbbc', '0', ascii('abbbbc')),
  1789. ('ab{4,5}bc', 'abbbbc', '', ascii(None)),
  1790. ('ab?bc', 'abbc', '0', ascii('abbc')),
  1791. ('ab?bc', 'abc', '0', ascii('abc')),
  1792. ('ab{0,1}bc', 'abc', '0', ascii('abc')),
  1793. ('ab?bc', 'abbbbc', '', ascii(None)),
  1794. ('ab?c', 'abc', '0', ascii('abc')),
  1795. ('ab{0,1}c', 'abc', '0', ascii('abc')),
  1796. ('^abc$', 'abc', '0', ascii('abc')),
  1797. ('^abc$', 'abcc', '', ascii(None)),
  1798. ('^abc', 'abcc', '0', ascii('abc')),
  1799. ('^abc$', 'aabc', '', ascii(None)),
  1800. ('abc$', 'aabc', '0', ascii('abc')),
  1801. ('^', 'abc', '0', ascii('')),
  1802. ('$', 'abc', '0', ascii('')),
  1803. ('a.c', 'abc', '0', ascii('abc')),
  1804. ('a.c', 'axc', '0', ascii('axc')),
  1805. ('a.*c', 'axyzc', '0', ascii('axyzc')),
  1806. ('a.*c', 'axyzd', '', ascii(None)),
  1807. ('a[bc]d', 'abc', '', ascii(None)),
  1808. ('a[bc]d', 'abd', '0', ascii('abd')),
  1809. ('a[b-d]e', 'abd', '', ascii(None)),
  1810. ('a[b-d]e', 'ace', '0', ascii('ace')),
  1811. ('a[b-d]', 'aac', '0', ascii('ac')),
  1812. ('a[-b]', 'a-', '0', ascii('a-')),
  1813. ('a[b-]', 'a-', '0', ascii('a-')),
  1814. ('a[b-a]', '-', '', regex.error, self.BAD_CHAR_RANGE),
  1815. ('a[]b', '-', '', regex.error, self.BAD_SET),
  1816. ('a[', '-', '', regex.error, self.BAD_SET),
  1817. ('a]', 'a]', '0', ascii('a]')),
  1818. ('a[]]b', 'a]b', '0', ascii('a]b')),
  1819. ('a[^bc]d', 'aed', '0', ascii('aed')),
  1820. ('a[^bc]d', 'abd', '', ascii(None)),
  1821. ('a[^-b]c', 'adc', '0', ascii('adc')),
  1822. ('a[^-b]c', 'a-c', '', ascii(None)),
  1823. ('a[^]b]c', 'a]c', '', ascii(None)),
  1824. ('a[^]b]c', 'adc', '0', ascii('adc')),
  1825. ('ab|cd', 'abc', '0', ascii('ab')),
  1826. ('ab|cd', 'abcd', '0', ascii('ab')),
  1827. ('()ef', 'def', '0,1', ascii(('ef', ''))),
  1828. ('*a', '-', '', regex.error, self.NOTHING_TO_REPEAT),
  1829. ('(*)b', '-', '', regex.error, self.NOTHING_TO_REPEAT),
  1830. ('$b', 'b', '', ascii(None)),
  1831. ('a\\', '-', '', regex.error, self.BAD_ESCAPE),
  1832. ('a\\(b', 'a(b', '', ascii(('a(b',))),
  1833. ('a\\(*b', 'ab', '0', ascii('ab')),
  1834. ('a\\(*b', 'a((b', '0', ascii('a((b')),
  1835. ('a\\\\b', 'a\\b', '0', ascii('a\\b')),
  1836. ('abc)', '-', '', regex.error, self.TRAILING_CHARS),
  1837. ('(abc', '-', '', regex.error, self.MISSING_RPAREN),
  1838. ('((a))', 'abc', '0,1,2', ascii(('a', 'a', 'a'))),
  1839. ('(a)b(c)', 'abc', '0,1,2', ascii(('abc', 'a', 'c'))),
  1840. ('a+b+c', 'aabbabc', '0', ascii('abc')),
  1841. ('a{1,}b{1,}c', 'aabbabc', '0', ascii('abc')),
  1842. ('a**', '-', '', regex.error, self.MULTIPLE_REPEAT),
  1843. ('a.+?c', 'abcabc', '0', ascii('abc')),
  1844. ('(a+|b)*', 'ab', '0,1', ascii(('ab', 'b'))),
  1845. ('(a+|b){0,}', 'ab', '0,1', ascii(('ab', 'b'))),
  1846. ('(a+|b)+', 'ab', '0,1', ascii(('ab', 'b'))),
  1847. ('(a+|b){1,}', 'ab', '0,1', ascii(('ab', 'b'))),
  1848. ('(a+|b)?', 'ab', '0,1', ascii(('a', 'a'))),
  1849. ('(a+|b){0,1}', 'ab', '0,1', ascii(('a', 'a'))),
  1850. (')(', '-', '', regex.error, self.TRAILING_CHARS),
  1851. ('[^ab]*', 'cde', '0', ascii('cde')),
  1852. ('abc', '', '', ascii(None)),
  1853. ('a*', '', '0', ascii('')),
  1854. ('([abc])*d', 'abbbcd', '0,1', ascii(('abbbcd', 'c'))),
  1855. ('([abc])*bcd', 'abcd', '0,1', ascii(('abcd', 'a'))),
  1856. ('a|b|c|d|e', 'e', '0', ascii('e')),
  1857. ('(a|b|c|d|e)f', 'ef', '0,1', ascii(('ef', 'e'))),
  1858. ('abcd*efg', 'abcdefg', '0', ascii('abcdefg')),
  1859. ('ab*', 'xabyabbbz', '0', ascii('ab')),
  1860. ('ab*', 'xayabbbz', '0', ascii('a')),
  1861. ('(ab|cd)e', 'abcde', '0,1', ascii(('cde', 'cd'))),
  1862. ('[abhgefdc]ij', 'hij', '0', ascii('hij')),
  1863. ('^(ab|cd)e', 'abcde', '', ascii(None)),
  1864. ('(abc|)ef', 'abcdef', '0,1', ascii(('ef', ''))),
  1865. ('(a|b)c*d', 'abcd', '0,1', ascii(('bcd', 'b'))),
  1866. ('(ab|ab*)bc', 'abc', '0,1', ascii(('abc', 'a'))),
  1867. ('a([bc]*)c*', 'abc', '0,1', ascii(('abc', 'bc'))),
  1868. ('a([bc]*)(c*d)', 'abcd', '0,1,2', ascii(('abcd', 'bc', 'd'))),
  1869. ('a([bc]+)(c*d)', 'abcd', '0,1,2', ascii(('abcd', 'bc', 'd'))),
  1870. ('a([bc]*)(c+d)', 'abcd', '0,1,2', ascii(('abcd', 'b', 'cd'))),
  1871. ('a[bcd]*dcdcde', 'adcdcde', '0', ascii('adcdcde')),
  1872. ('a[bcd]+dcdcde', 'adcdcde', '', ascii(None)),
  1873. ('(ab|a)b*c', 'abc', '0,1', ascii(('abc', 'ab'))),
  1874. ('((a)(b)c)(d)', 'abcd', '1,2,3,4', ascii(('abc', 'a', 'b', 'd'))),
  1875. ('[a-zA-Z_][a-zA-Z0-9_]*', 'alpha', '0', ascii('alpha')),
  1876. ('^a(bc+|b[eh])g|.h$', 'abh', '0,1', ascii(('bh', None))),
  1877. ('(bc+d$|ef*g.|h?i(j|k))', 'effgz', '0,1,2', ascii(('effgz',
  1878. 'effgz', None))),
  1879. ('(bc+d$|ef*g.|h?i(j|k))', 'ij', '0,1,2', ascii(('ij', 'ij',
  1880. 'j'))),
  1881. ('(bc+d$|ef*g.|h?i(j|k))', 'effg', '', ascii(None)),
  1882. ('(bc+d$|ef*g.|h?i(j|k))', 'bcdd', '', ascii(None)),
  1883. ('(bc+d$|ef*g.|h?i(j|k))', 'reffgz', '0,1,2', ascii(('effgz',
  1884. 'effgz', None))),
  1885. ('((((((((((a))))))))))', 'a', '10', ascii('a')),
  1886. ('((((((((((a))))))))))\\10', 'aa', '0', ascii('aa')),
  1887. # Python does not have the same rules for \\41 so this is a syntax error
  1888. # ('((((((((((a))))))))))\\41', 'aa', '', ascii(None)),
  1889. # ('((((((((((a))))))))))\\41', 'a!', '0', ascii('a!')),
  1890. ('((((((((((a))))))))))\\41', '', '', regex.error,
  1891. self.INVALID_GROUP_REF),
  1892. ('(?i)((((((((((a))))))))))\\41', '', '', regex.error,
  1893. self.INVALID_GROUP_REF),
  1894. ('(((((((((a)))))))))', 'a', '0', ascii('a')),
  1895. ('multiple words of text', 'uh-uh', '', ascii(None)),
  1896. ('multiple words', 'multiple words, yeah', '0',
  1897. ascii('multiple words')),
  1898. ('(.*)c(.*)', 'abcde', '0,1,2', ascii(('abcde', 'ab', 'de'))),
  1899. ('\\((.*), (.*)\\)', '(a, b)', '2,1', ascii(('b', 'a'))),
  1900. ('[k]', 'ab', '', ascii(None)),
  1901. ('a[-]?c', 'ac', '0', ascii('ac')),
  1902. ('(abc)\\1', 'abcabc', '1', ascii('abc')),
  1903. ('([a-c]*)\\1', 'abcabc', '1', ascii('abc')),
  1904. ('(?i)abc', 'ABC', '0', ascii('ABC')),
  1905. ('(?i)abc', 'XBC', '', ascii(None)),
  1906. ('(?i)abc', 'AXC', '', ascii(None)),
  1907. ('(?i)abc', 'ABX', '', ascii(None)),
  1908. ('(?i)abc', 'XABCY', '0', ascii('ABC')),
  1909. ('(?i)abc', 'ABABC', '0', ascii('ABC')),
  1910. ('(?i)ab*c', 'ABC', '0', ascii('ABC')),
  1911. ('(?i)ab*bc', 'ABC', '0', ascii('ABC')),
  1912. ('(?i)ab*bc', 'ABBC', '0', ascii('ABBC')),
  1913. ('(?i)ab*?bc', 'ABBBBC', '0', ascii('ABBBBC')),
  1914. ('(?i)ab{0,}?bc', 'ABBBBC', '0', ascii('ABBBBC')),
  1915. ('(?i)ab+?bc', 'ABBC', '0', ascii('ABBC')),
  1916. ('(?i)ab+bc', 'ABC', '', ascii(None)),
  1917. ('(?i)ab+bc', 'ABQ', '', ascii(None)),
  1918. ('(?i)ab{1,}bc', 'ABQ', '', ascii(None)),
  1919. ('(?i)ab+bc', 'ABBBBC', '0', ascii('ABBBBC')),
  1920. ('(?i)ab{1,}?bc', 'ABBBBC', '0', ascii('ABBBBC')),
  1921. ('(?i)ab{1,3}?bc', 'ABBBBC', '0', ascii('ABBBBC')),
  1922. ('(?i)ab{3,4}?bc', 'ABBBBC', '0', ascii('ABBBBC')),
  1923. ('(?i)ab{4,5}?bc', 'ABBBBC', '', ascii(None)),
  1924. ('(?i)ab??bc', 'ABBC', '0', ascii('ABBC')),
  1925. ('(?i)ab??bc', 'ABC', '0', ascii('ABC')),
  1926. ('(?i)ab{0,1}?bc', 'ABC', '0', ascii('ABC')),
  1927. ('(?i)ab??bc', 'ABBBBC', '', ascii(None)),
  1928. ('(?i)ab??c', 'ABC', '0', ascii('ABC')),
  1929. ('(?i)ab{0,1}?c', 'ABC', '0', ascii('ABC')),
  1930. ('(?i)^abc$', 'ABC', '0', ascii('ABC')),
  1931. ('(?i)^abc$', 'ABCC', '', ascii(None)),
  1932. ('(?i)^abc', 'ABCC', '0', ascii('ABC')),
  1933. ('(?i)^abc$', 'AABC', '', ascii(None)),
  1934. ('(?i)abc$', 'AABC', '0', ascii('ABC')),
  1935. ('(?i)^', 'ABC', '0', ascii('')),
  1936. ('(?i)$', 'ABC', '0', ascii('')),
  1937. ('(?i)a.c', 'ABC', '0', ascii('ABC')),
  1938. ('(?i)a.c', 'AXC', '0', ascii('AXC')),
  1939. ('(?i)a.*?c', 'AXYZC', '0', ascii('AXYZC')),
  1940. ('(?i)a.*c', 'AXYZD', '', ascii(None)),
  1941. ('(?i)a[bc]d', 'ABC', '', ascii(None)),
  1942. ('(?i)a[bc]d', 'ABD', '0', ascii('ABD')),
  1943. ('(?i)a[b-d]e', 'ABD', '', ascii(None)),
  1944. ('(?i)a[b-d]e', 'ACE', '0', ascii('ACE')),
  1945. ('(?i)a[b-d]', 'AAC', '0', ascii('AC')),
  1946. ('(?i)a[-b]', 'A-', '0', ascii('A-')),
  1947. ('(?i)a[b-]', 'A-', '0', ascii('A-')),
  1948. ('(?i)a[b-a]', '-', '', regex.error, self.BAD_CHAR_RANGE),
  1949. ('(?i)a[]b', '-', '', regex.error, self.BAD_SET),
  1950. ('(?i)a[', '-', '', regex.error, self.BAD_SET),
  1951. ('(?i)a]', 'A]', '0', ascii('A]')),
  1952. ('(?i)a[]]b', 'A]B', '0', ascii('A]B')),
  1953. ('(?i)a[^bc]d', 'AED', '0', ascii('AED')),
  1954. ('(?i)a[^bc]d', 'ABD', '', ascii(None)),
  1955. ('(?i)a[^-b]c', 'ADC', '0', ascii('ADC')),
  1956. ('(?i)a[^-b]c', 'A-C', '', ascii(None)),
  1957. ('(?i)a[^]b]c', 'A]C', '', ascii(None)),
  1958. ('(?i)a[^]b]c', 'ADC', '0', ascii('ADC')),
  1959. ('(?i)ab|cd', 'ABC', '0', ascii('AB')),
  1960. ('(?i)ab|cd', 'ABCD', '0', ascii('AB')),
  1961. ('(?i)()ef', 'DEF', '0,1', ascii(('EF', ''))),
  1962. ('(?i)*a', '-', '', regex.error, self.NOTHING_TO_REPEAT),
  1963. ('(?i)(*)b', '-', '', regex.error, self.NOTHING_TO_REPEAT),
  1964. ('(?i)$b', 'B', '', ascii(None)),
  1965. ('(?i)a\\', '-', '', regex.error, self.BAD_ESCAPE),
  1966. ('(?i)a\\(b', 'A(B', '', ascii(('A(B',))),
  1967. ('(?i)a\\(*b', 'AB', '0', ascii('AB')),
  1968. ('(?i)a\\(*b', 'A((B', '0', ascii('A((B')),
  1969. ('(?i)a\\\\b', 'A\\B', '0', ascii('A\\B')),
  1970. ('(?i)abc)', '-', '', regex.error, self.TRAILING_CHARS),
  1971. ('(?i)(abc', '-', '', regex.error, self.MISSING_RPAREN),
  1972. ('(?i)((a))', 'ABC', '0,1,2', ascii(('A', 'A', 'A'))),
  1973. ('(?i)(a)b(c)', 'ABC', '0,1,2', ascii(('ABC', 'A', 'C'))),
  1974. ('(?i)a+b+c', 'AABBABC', '0', ascii('ABC')),
  1975. ('(?i)a{1,}b{1,}c', 'AABBABC', '0', ascii('ABC')),
  1976. ('(?i)a**', '-', '', regex.error, self.MULTIPLE_REPEAT),
  1977. ('(?i)a.+?c', 'ABCABC', '0', ascii('ABC')),
  1978. ('(?i)a.*?c', 'ABCABC', '0', ascii('ABC')),
  1979. ('(?i)a.{0,5}?c', 'ABCABC', '0', ascii('ABC')),
  1980. ('(?i)(a+|b)*', 'AB', '0,1', ascii(('AB', 'B'))),
  1981. ('(?i)(a+|b){0,}', 'AB', '0,1', ascii(('AB', 'B'))),
  1982. ('(?i)(a+|b)+', 'AB', '0,1', ascii(('AB', 'B'))),
  1983. ('(?i)(a+|b){1,}', 'AB', '0,1', ascii(('AB', 'B'))),
  1984. ('(?i)(a+|b)?', 'AB', '0,1', ascii(('A', 'A'))),
  1985. ('(?i)(a+|b){0,1}', 'AB', '0,1', ascii(('A', 'A'))),
  1986. ('(?i)(a+|b){0,1}?', 'AB', '0,1', ascii(('', None))),
  1987. ('(?i))(', '-', '', regex.error, self.TRAILING_CHARS),
  1988. ('(?i)[^ab]*', 'CDE', '0', ascii('CDE')),
  1989. ('(?i)abc', '', '', ascii(None)),
  1990. ('(?i)a*', '', '0', ascii('')),
  1991. ('(?i)([abc])*d', 'ABBBCD', '0,1', ascii(('ABBBCD', 'C'))),
  1992. ('(?i)([abc])*bcd', 'ABCD', '0,1', ascii(('ABCD', 'A'))),
  1993. ('(?i)a|b|c|d|e', 'E', '0', ascii('E')),
  1994. ('(?i)(a|b|c|d|e)f', 'EF', '0,1', ascii(('EF', 'E'))),
  1995. ('(?i)abcd*efg', 'ABCDEFG', '0', ascii('ABCDEFG')),
  1996. ('(?i)ab*', 'XABYABBBZ', '0', ascii('AB')),
  1997. ('(?i)ab*', 'XAYABBBZ', '0', ascii('A')),
  1998. ('(?i)(ab|cd)e', 'ABCDE', '0,1', ascii(('CDE', 'CD'))),
  1999. ('(?i)[abhgefdc]ij', 'HIJ', '0', ascii('HIJ')),
  2000. ('(?i)^(ab|cd)e', 'ABCDE', '', ascii(None)),
  2001. ('(?i)(abc|)ef', 'ABCDEF', '0,1', ascii(('EF', ''))),
  2002. ('(?i)(a|b)c*d', 'ABCD', '0,1', ascii(('BCD', 'B'))),
  2003. ('(?i)(ab|ab*)bc', 'ABC', '0,1', ascii(('ABC', 'A'))),
  2004. ('(?i)a([bc]*)c*', 'ABC', '0,1', ascii(('ABC', 'BC'))),
  2005. ('(?i)a([bc]*)(c*d)', 'ABCD', '0,1,2', ascii(('ABCD', 'BC', 'D'))),
  2006. ('(?i)a([bc]+)(c*d)', 'ABCD', '0,1,2', ascii(('ABCD', 'BC', 'D'))),
  2007. ('(?i)a([bc]*)(c+d)', 'ABCD', '0,1,2', ascii(('ABCD', 'B', 'CD'))),
  2008. ('(?i)a[bcd]*dcdcde', 'ADCDCDE', '0', ascii('ADCDCDE')),
  2009. ('(?i)a[bcd]+dcdcde', 'ADCDCDE', '', ascii(None)),
  2010. ('(?i)(ab|a)b*c', 'ABC', '0,1', ascii(('ABC', 'AB'))),
  2011. ('(?i)((a)(b)c)(d)', 'ABCD', '1,2,3,4', ascii(('ABC', 'A', 'B',
  2012. 'D'))),
  2013. ('(?i)[a-zA-Z_][a-zA-Z0-9_]*', 'ALPHA', '0', ascii('ALPHA')),
  2014. ('(?i)^a(bc+|b[eh])g|.h$', 'ABH', '0,1', ascii(('BH', None))),
  2015. ('(?i)(bc+d$|ef*g.|h?i(j|k))', 'EFFGZ', '0,1,2', ascii(('EFFGZ',
  2016. 'EFFGZ', None))),
  2017. ('(?i)(bc+d$|ef*g.|h?i(j|k))', 'IJ', '0,1,2', ascii(('IJ', 'IJ',
  2018. 'J'))),
  2019. ('(?i)(bc+d$|ef*g.|h?i(j|k))', 'EFFG', '', ascii(None)),
  2020. ('(?i)(bc+d$|ef*g.|h?i(j|k))', 'BCDD', '', ascii(None)),
  2021. ('(?i)(bc+d$|ef*g.|h?i(j|k))', 'REFFGZ', '0,1,2', ascii(('EFFGZ',
  2022. 'EFFGZ', None))),
  2023. ('(?i)((((((((((a))))))))))', 'A', '10', ascii('A')),
  2024. ('(?i)((((((((((a))))))))))\\10', 'AA', '0', ascii('AA')),
  2025. #('(?i)((((((((((a))))))))))\\41', 'AA', '', ascii(None)),
  2026. #('(?i)((((((((((a))))))))))\\41', 'A!', '0', ascii('A!')),
  2027. ('(?i)(((((((((a)))))))))', 'A', '0', ascii('A')),
  2028. ('(?i)(?:(?:(?:(?:(?:(?:(?:(?:(?:(a))))))))))', 'A', '1',
  2029. ascii('A')),
  2030. ('(?i)(?:(?:(?:(?:(?:(?:(?:(?:(?:(a|b|c))))))))))', 'C', '1',
  2031. ascii('C')),
  2032. ('(?i)multiple words of text', 'UH-UH', '', ascii(None)),
  2033. ('(?i)multiple words', 'MULTIPLE WORDS, YEAH', '0',
  2034. ascii('MULTIPLE WORDS')),
  2035. ('(?i)(.*)c(.*)', 'ABCDE', '0,1,2', ascii(('ABCDE', 'AB', 'DE'))),
  2036. ('(?i)\\((.*), (.*)\\)', '(A, B)', '2,1', ascii(('B', 'A'))),
  2037. ('(?i)[k]', 'AB', '', ascii(None)),
  2038. # ('(?i)abcd', 'ABCD', SUCCEED, 'found+"-"+\\found+"-"+\\\\found', ascii(ABCD-$&-\\ABCD)),
  2039. # ('(?i)a(bc)d', 'ABCD', SUCCEED, 'g1+"-"+\\g1+"-"+\\\\g1', ascii(BC-$1-\\BC)),
  2040. ('(?i)a[-]?c', 'AC', '0', ascii('AC')),
  2041. ('(?i)(abc)\\1', 'ABCABC', '1', ascii('ABC')),
  2042. ('(?i)([a-c]*)\\1', 'ABCABC', '1', ascii('ABC')),
  2043. ('a(?!b).', 'abad', '0', ascii('ad')),
  2044. ('a(?=d).', 'abad', '0', ascii('ad')),
  2045. ('a(?=c|d).', 'abad', '0', ascii('ad')),
  2046. ('a(?:b|c|d)(.)', 'ace', '1', ascii('e')),
  2047. ('a(?:b|c|d)*(.)', 'ace', '1', ascii('e')),
  2048. ('a(?:b|c|d)+?(.)', 'ace', '1', ascii('e')),
  2049. ('a(?:b|(c|e){1,2}?|d)+?(.)', 'ace', '1,2', ascii(('c', 'e'))),
  2050. # Lookbehind: split by : but not if it is escaped by -.
  2051. ('(?<!-):(.*?)(?<!-):', 'a:bc-:de:f', '1', ascii('bc-:de')),
  2052. # Escaping with \ as we know it.
  2053. ('(?<!\\\\):(.*?)(?<!\\\\):', 'a:bc\\:de:f', '1', ascii('bc\\:de')),
  2054. # Terminating with ' and escaping with ? as in edifact.
  2055. ("(?<!\\?)'(.*?)(?<!\\?)'", "a'bc?'de'f", '1', ascii("bc?'de")),
  2056. # Comments using the (?#...) syntax.
  2057. ('w(?# comment', 'w', '', regex.error, self.MISSING_RPAREN),
  2058. ('w(?# comment 1)xy(?# comment 2)z', 'wxyz', '0', ascii('wxyz')),
  2059. # Check odd placement of embedded pattern modifiers.
  2060. # Not an error under PCRE/PRE:
  2061. # When the new behaviour is turned on positional inline flags affect
  2062. # only what follows.
  2063. ('w(?i)', 'W', '0', ascii('W')),
  2064. ('w(?iV1)', 'W', '0', ascii(None)),
  2065. ('w(?i)', 'w', '0', ascii('w')),
  2066. ('w(?iV1)', 'w', '0', ascii('w')),
  2067. ('(?i)w', 'W', '0', ascii('W')),
  2068. ('(?iV1)w', 'W', '0', ascii('W')),
  2069. # Comments using the x embedded pattern modifier.
  2070. ("""(?x)w# comment 1
  2071. x y
  2072. # comment 2
  2073. z""", 'wxyz', '0', ascii('wxyz')),
  2074. # Using the m embedded pattern modifier.
  2075. ('^abc', """jkl
  2076. abc
  2077. xyz""", '', ascii(None)),
  2078. ('(?m)^abc', """jkl
  2079. abc
  2080. xyz""", '0', ascii('abc')),
  2081. ('(?m)abc$', """jkl
  2082. xyzabc
  2083. 123""", '0', ascii('abc')),
  2084. # Using the s embedded pattern modifier.
  2085. ('a.b', 'a\nb', '', ascii(None)),
  2086. ('(?s)a.b', 'a\nb', '0', ascii('a\nb')),
  2087. # Test \w, etc. both inside and outside character classes.
  2088. ('\\w+', '--ab_cd0123--', '0', ascii('ab_cd0123')),
  2089. ('[\\w]+', '--ab_cd0123--', '0', ascii('ab_cd0123')),
  2090. ('\\D+', '1234abc5678', '0', ascii('abc')),
  2091. ('[\\D]+', '1234abc5678', '0', ascii('abc')),
  2092. ('[\\da-fA-F]+', '123abc', '0', ascii('123abc')),
  2093. # Not an error under PCRE/PRE:
  2094. # ('[\\d-x]', '-', '', regex.error, self.BAD_CHAR_RANGE),
  2095. (r'([\s]*)([\S]*)([\s]*)', ' testing!1972', '3,2,1', ascii(('',
  2096. 'testing!1972', ' '))),
  2097. (r'(\s*)(\S*)(\s*)', ' testing!1972', '3,2,1', ascii(('',
  2098. 'testing!1972', ' '))),
  2099. #
  2100. # Post-1.5.2 additions.
  2101. # xmllib problem.
  2102. (r'(([a-z]+):)?([a-z]+)$', 'smil', '1,2,3', ascii((None, None,
  2103. 'smil'))),
  2104. # Bug 110866: reference to undefined group.
  2105. (r'((.)\1+)', '', '', regex.error, self.OPEN_GROUP),
  2106. # Bug 111869: search (PRE/PCRE fails on this one, SRE doesn't).
  2107. (r'.*d', 'abc\nabd', '0', ascii('abd')),
  2108. # Bug 112468: various expected syntax errors.
  2109. (r'(', '', '', regex.error, self.MISSING_RPAREN),
  2110. (r'[\41]', '!', '0', ascii('!')),
  2111. # Bug 114033: nothing to repeat.
  2112. (r'(x?)?', 'x', '0', ascii('x')),
  2113. # Bug 115040: rescan if flags are modified inside pattern.
  2114. # If the new behaviour is turned on then positional inline flags
  2115. # affect only what follows.
  2116. (r' (?x)foo ', 'foo', '0', ascii('foo')),
  2117. (r' (?V1x)foo ', 'foo', '0', ascii(None)),
  2118. (r'(?x) foo ', 'foo', '0', ascii('foo')),
  2119. (r'(?V1x) foo ', 'foo', '0', ascii('foo')),
  2120. (r'(?x)foo ', 'foo', '0', ascii('foo')),
  2121. (r'(?V1x)foo ', 'foo', '0', ascii('foo')),
  2122. # Bug 115618: negative lookahead.
  2123. (r'(?<!abc)(d.f)', 'abcdefdof', '0', ascii('dof')),
  2124. # Bug 116251: character class bug.
  2125. (r'[\w-]+', 'laser_beam', '0', ascii('laser_beam')),
  2126. # Bug 123769+127259: non-greedy backtracking bug.
  2127. (r'.*?\S *:', 'xx:', '0', ascii('xx:')),
  2128. (r'a[ ]*?\ (\d+).*', 'a 10', '0', ascii('a 10')),
  2129. (r'a[ ]*?\ (\d+).*', 'a 10', '0', ascii('a 10')),
  2130. # Bug 127259: \Z shouldn't depend on multiline mode.
  2131. (r'(?ms).*?x\s*\Z(.*)','xx\nx\n', '1', ascii('')),
  2132. # Bug 128899: uppercase literals under the ignorecase flag.
  2133. (r'(?i)M+', 'MMM', '0', ascii('MMM')),
  2134. (r'(?i)m+', 'MMM', '0', ascii('MMM')),
  2135. (r'(?i)[M]+', 'MMM', '0', ascii('MMM')),
  2136. (r'(?i)[m]+', 'MMM', '0', ascii('MMM')),
  2137. # Bug 130748: ^* should be an error (nothing to repeat).
  2138. # In 'regex' we won't bother to complain about this.
  2139. # (r'^*', '', '', regex.error, self.NOTHING_TO_REPEAT),
  2140. # Bug 133283: minimizing repeat problem.
  2141. (r'"(?:\\"|[^"])*?"', r'"\""', '0', ascii(r'"\""')),
  2142. # Bug 477728: minimizing repeat problem.
  2143. (r'^.*?$', 'one\ntwo\nthree\n', '', ascii(None)),
  2144. # Bug 483789: minimizing repeat problem.
  2145. (r'a[^>]*?b', 'a>b', '', ascii(None)),
  2146. # Bug 490573: minimizing repeat problem.
  2147. (r'^a*?$', 'foo', '', ascii(None)),
  2148. # Bug 470582: nested groups problem.
  2149. (r'^((a)c)?(ab)$', 'ab', '1,2,3', ascii((None, None, 'ab'))),
  2150. # Another minimizing repeat problem (capturing groups in assertions).
  2151. ('^([ab]*?)(?=(b)?)c', 'abc', '1,2', ascii(('ab', None))),
  2152. ('^([ab]*?)(?!(b))c', 'abc', '1,2', ascii(('ab', None))),
  2153. ('^([ab]*?)(?<!(a))c', 'abc', '1,2', ascii(('ab', None))),
  2154. # Bug 410271: \b broken under locales.
  2155. (r'\b.\b', 'a', '0', ascii('a')),
  2156. (r'\b.\b', '\N{LATIN CAPITAL LETTER A WITH DIAERESIS}', '0',
  2157. ascii('\xc4')),
  2158. (r'\w', '\N{LATIN CAPITAL LETTER A WITH DIAERESIS}', '0',
  2159. ascii('\xc4')),
  2160. ]
  2161. for t in tests:
  2162. excval = None
  2163. try:
  2164. if len(t) == 4:
  2165. pattern, string, groups, expected = t
  2166. else:
  2167. pattern, string, groups, expected, excval = t
  2168. except ValueError:
  2169. fields = ", ".join([ascii(f) for f in t[ : 3]] + ["..."])
  2170. self.fail("Incorrect number of test fields: ({})".format(fields))
  2171. else:
  2172. group_list = []
  2173. if groups:
  2174. for group in groups.split(","):
  2175. try:
  2176. group_list.append(int(group))
  2177. except ValueError:
  2178. group_list.append(group)
  2179. if excval is not None:
  2180. if (sys.version_info.major, sys.version_info.minor) >= (3, 4):
  2181. with self.subTest(pattern=pattern, string=string):
  2182. self.assertRaisesRegex(expected, excval, regex.search,
  2183. pattern, string)
  2184. else:
  2185. m = regex.search(pattern, string)
  2186. if m:
  2187. if group_list:
  2188. actual = ascii(m.group(*group_list))
  2189. else:
  2190. actual = ascii(m[:])
  2191. else:
  2192. actual = ascii(m)
  2193. self.assertEqual(actual, expected)
  2194. def test_replacement(self):
  2195. self.assertEqual(regex.sub(r"test\?", "result\\?\\.\a\\q\\m\n", "test?"),
  2196. "result\\?\\.\a\\q\\m\n")
  2197. self.assertEqual(regex.sub('(.)', r"\1\1", 'x'), 'xx')
  2198. self.assertEqual(regex.sub('(.)', regex.escape(r"\1\1"), 'x'), r"\1\1")
  2199. self.assertEqual(regex.sub('(.)', r"\\1\\1", 'x'), r"\1\1")
  2200. self.assertEqual(regex.sub('(.)', lambda m: r"\1\1", 'x'), r"\1\1")
  2201. def test_common_prefix(self):
  2202. # Very long common prefix
  2203. all = string.ascii_lowercase + string.digits + string.ascii_uppercase
  2204. side = all * 4
  2205. regexp = '(' + side + '|' + side + ')'
  2206. self.assertEqual(repr(type(regex.compile(regexp))), self.PATTERN_CLASS)
  2207. def test_captures(self):
  2208. self.assertEqual(regex.search(r"(\w)+", "abc").captures(1), ['a', 'b',
  2209. 'c'])
  2210. self.assertEqual(regex.search(r"(\w{3})+", "abcdef").captures(0, 1),
  2211. (['abcdef'], ['abc', 'def']))
  2212. self.assertEqual(regex.search(r"^(\d{1,3})(?:\.(\d{1,3})){3}$",
  2213. "192.168.0.1").captures(1, 2), (['192', ], ['168', '0', '1']))
  2214. self.assertEqual(regex.match(r"^([0-9A-F]{2}){4} ([a-z]\d){5}$",
  2215. "3FB52A0C a2c4g3k9d3").captures(1, 2), (['3F', 'B5', '2A', '0C'],
  2216. ['a2', 'c4', 'g3', 'k9', 'd3']))
  2217. self.assertEqual(regex.match("([a-z]W)([a-z]X)+([a-z]Y)",
  2218. "aWbXcXdXeXfY").captures(1, 2, 3), (['aW'], ['bX', 'cX', 'dX', 'eX'],
  2219. ['fY']))
  2220. self.assertEqual(regex.search(r".*?(?=(.)+)b", "ab").captures(1),
  2221. ['b'])
  2222. self.assertEqual(regex.search(r".*?(?>(.){0,2})d", "abcd").captures(1),
  2223. ['b', 'c'])
  2224. self.assertEqual(regex.search(r"(.)+", "a").captures(1), ['a'])
  2225. def test_guards(self):
  2226. m = regex.search(r"(X.*?Y\s*){3}(X\s*)+AB:",
  2227. "XY\nX Y\nX Y\nXY\nXX AB:")
  2228. self.assertEqual(m.span(0, 1, 2), ((3, 21), (12, 15), (16, 18)))
  2229. m = regex.search(r"(X.*?Y\s*){3,}(X\s*)+AB:",
  2230. "XY\nX Y\nX Y\nXY\nXX AB:")
  2231. self.assertEqual(m.span(0, 1, 2), ((0, 21), (12, 15), (16, 18)))
  2232. m = regex.search(r'\d{4}(\s*\w)?\W*((?!\d)\w){2}', "9999XX")
  2233. self.assertEqual(m.span(0, 1, 2), ((0, 6), (-1, -1), (5, 6)))
  2234. m = regex.search(r'A\s*?.*?(\n+.*?\s*?){0,2}\(X', 'A\n1\nS\n1 (X')
  2235. self.assertEqual(m.span(0, 1), ((0, 10), (5, 8)))
  2236. m = regex.search(r'Derde\s*:', 'aaaaaa:\nDerde:')
  2237. self.assertEqual(m.span(), (8, 14))
  2238. m = regex.search(r'Derde\s*:', 'aaaaa:\nDerde:')
  2239. self.assertEqual(m.span(), (7, 13))
  2240. def test_turkic(self):
  2241. # Turkish has dotted and dotless I/i.
  2242. pairs = "I=i;I=\u0131;i=\u0130"
  2243. all_chars = set()
  2244. matching = set()
  2245. for pair in pairs.split(";"):
  2246. ch1, ch2 = pair.split("=")
  2247. all_chars.update((ch1, ch2))
  2248. matching.add((ch1, ch1))
  2249. matching.add((ch1, ch2))
  2250. matching.add((ch2, ch1))
  2251. matching.add((ch2, ch2))
  2252. for ch1 in all_chars:
  2253. for ch2 in all_chars:
  2254. m = regex.match(r"(?i)\A" + ch1 + r"\Z", ch2)
  2255. if m:
  2256. if (ch1, ch2) not in matching:
  2257. self.fail("{} matching {}".format(ascii(ch1),
  2258. ascii(ch2)))
  2259. else:
  2260. if (ch1, ch2) in matching:
  2261. self.fail("{} not matching {}".format(ascii(ch1),
  2262. ascii(ch2)))
  2263. def test_named_lists(self):
  2264. options = ["one", "two", "three"]
  2265. self.assertEqual(regex.match(r"333\L<bar>444", "333one444",
  2266. bar=options).group(), "333one444")
  2267. self.assertEqual(regex.match(r"(?i)333\L<bar>444", "333TWO444",
  2268. bar=options).group(), "333TWO444")
  2269. self.assertEqual(regex.match(r"333\L<bar>444", "333four444",
  2270. bar=options), None)
  2271. options = [b"one", b"two", b"three"]
  2272. self.assertEqual(regex.match(br"333\L<bar>444", b"333one444",
  2273. bar=options).group(), b"333one444")
  2274. self.assertEqual(regex.match(br"(?i)333\L<bar>444", b"333TWO444",
  2275. bar=options).group(), b"333TWO444")
  2276. self.assertEqual(regex.match(br"333\L<bar>444", b"333four444",
  2277. bar=options), None)
  2278. self.assertEqual(repr(type(regex.compile(r"3\L<bar>4\L<bar>+5",
  2279. bar=["one", "two", "three"]))), self.PATTERN_CLASS)
  2280. self.assertEqual(regex.findall(r"^\L<options>", "solid QWERT",
  2281. options=set(['good', 'brilliant', '+s\\ol[i}d'])), [])
  2282. self.assertEqual(regex.findall(r"^\L<options>", "+solid QWERT",
  2283. options=set(['good', 'brilliant', '+solid'])), ['+solid'])
  2284. options = ["STRASSE"]
  2285. self.assertEqual(regex.match(r"(?fi)\L<words>",
  2286. "stra\N{LATIN SMALL LETTER SHARP S}e", words=options).span(), (0,
  2287. 6))
  2288. options = ["STRASSE", "stress"]
  2289. self.assertEqual(regex.match(r"(?fi)\L<words>",
  2290. "stra\N{LATIN SMALL LETTER SHARP S}e", words=options).span(), (0,
  2291. 6))
  2292. options = ["stra\N{LATIN SMALL LETTER SHARP S}e"]
  2293. self.assertEqual(regex.match(r"(?fi)\L<words>", "STRASSE",
  2294. words=options).span(), (0, 7))
  2295. options = ["kit"]
  2296. self.assertEqual(regex.search(r"(?i)\L<words>", "SKITS",
  2297. words=options).span(), (1, 4))
  2298. self.assertEqual(regex.search(r"(?i)\L<words>",
  2299. "SK\N{LATIN CAPITAL LETTER I WITH DOT ABOVE}TS",
  2300. words=options).span(), (1, 4))
  2301. self.assertEqual(regex.search(r"(?fi)\b(\w+) +\1\b",
  2302. " stra\N{LATIN SMALL LETTER SHARP S}e STRASSE ").span(), (1, 15))
  2303. self.assertEqual(regex.search(r"(?fi)\b(\w+) +\1\b",
  2304. " STRASSE stra\N{LATIN SMALL LETTER SHARP S}e ").span(), (1, 15))
  2305. self.assertEqual(regex.search(r"^\L<options>$", "", options=[]).span(),
  2306. (0, 0))
  2307. def test_fuzzy(self):
  2308. # Some tests borrowed from TRE library tests.
  2309. self.assertEqual(repr(type(regex.compile('(fou){s,e<=1}'))),
  2310. self.PATTERN_CLASS)
  2311. self.assertEqual(repr(type(regex.compile('(fuu){s}'))),
  2312. self.PATTERN_CLASS)
  2313. self.assertEqual(repr(type(regex.compile('(fuu){s,e}'))),
  2314. self.PATTERN_CLASS)
  2315. self.assertEqual(repr(type(regex.compile('(anaconda){1i+1d<1,s<=1}'))),
  2316. self.PATTERN_CLASS)
  2317. self.assertEqual(repr(type(regex.compile('(anaconda){1i+1d<1,s<=1,e<=10}'))),
  2318. self.PATTERN_CLASS)
  2319. self.assertEqual(repr(type(regex.compile('(anaconda){s<=1,e<=1,1i+1d<1}'))),
  2320. self.PATTERN_CLASS)
  2321. text = 'molasses anaconda foo bar baz smith anderson '
  2322. self.assertEqual(regex.search('(znacnda){s<=1,e<=3,1i+1d<1}', text),
  2323. None)
  2324. self.assertEqual(regex.search('(znacnda){s<=1,e<=3,1i+1d<2}',
  2325. text).span(0, 1), ((9, 17), (9, 17)))
  2326. self.assertEqual(regex.search('(ananda){1i+1d<2}', text), None)
  2327. self.assertEqual(regex.search(r"(?:\bznacnda){e<=2}", text)[0],
  2328. "anaconda")
  2329. self.assertEqual(regex.search(r"(?:\bnacnda){e<=2}", text)[0],
  2330. "anaconda")
  2331. text = 'anaconda foo bar baz smith anderson'
  2332. self.assertEqual(regex.search('(fuu){i<=3,d<=3,e<=5}', text).span(0,
  2333. 1), ((0, 0), (0, 0)))
  2334. self.assertEqual(regex.search('(?b)(fuu){i<=3,d<=3,e<=5}',
  2335. text).span(0, 1), ((9, 10), (9, 10)))
  2336. self.assertEqual(regex.search('(fuu){i<=2,d<=2,e<=5}', text).span(0,
  2337. 1), ((7, 10), (7, 10)))
  2338. self.assertEqual(regex.search('(?e)(fuu){i<=2,d<=2,e<=5}',
  2339. text).span(0, 1), ((9, 10), (9, 10)))
  2340. self.assertEqual(regex.search('(fuu){i<=3,d<=3,e}', text).span(0, 1),
  2341. ((0, 0), (0, 0)))
  2342. self.assertEqual(regex.search('(?b)(fuu){i<=3,d<=3,e}', text).span(0,
  2343. 1), ((9, 10), (9, 10)))
  2344. self.assertEqual(repr(type(regex.compile('(approximate){s<=3,1i+1d<3}'))),
  2345. self.PATTERN_CLASS)
  2346. # No cost limit.
  2347. self.assertEqual(regex.search('(foobar){e}',
  2348. 'xirefoabralfobarxie').span(0, 1), ((0, 6), (0, 6)))
  2349. self.assertEqual(regex.search('(?e)(foobar){e}',
  2350. 'xirefoabralfobarxie').span(0, 1), ((0, 3), (0, 3)))
  2351. self.assertEqual(regex.search('(?b)(foobar){e}',
  2352. 'xirefoabralfobarxie').span(0, 1), ((11, 16), (11, 16)))
  2353. # At most two errors.
  2354. self.assertEqual(regex.search('(foobar){e<=2}',
  2355. 'xirefoabrzlfd').span(0, 1), ((4, 9), (4, 9)))
  2356. self.assertEqual(regex.search('(foobar){e<=2}', 'xirefoabzlfd'), None)
  2357. # At most two inserts or substitutions and max two errors total.
  2358. self.assertEqual(regex.search('(foobar){i<=2,s<=2,e<=2}',
  2359. 'oobargoobaploowap').span(0, 1), ((5, 11), (5, 11)))
  2360. # Find best whole word match for "foobar".
  2361. self.assertEqual(regex.search('\\b(foobar){e}\\b', 'zfoobarz').span(0,
  2362. 1), ((0, 8), (0, 8)))
  2363. self.assertEqual(regex.search('\\b(foobar){e}\\b',
  2364. 'boing zfoobarz goobar woop').span(0, 1), ((0, 6), (0, 6)))
  2365. self.assertEqual(regex.search('(?b)\\b(foobar){e}\\b',
  2366. 'boing zfoobarz goobar woop').span(0, 1), ((15, 21), (15, 21)))
  2367. # Match whole string, allow only 1 error.
  2368. self.assertEqual(regex.search('^(foobar){e<=1}$', 'foobar').span(0, 1),
  2369. ((0, 6), (0, 6)))
  2370. self.assertEqual(regex.search('^(foobar){e<=1}$', 'xfoobar').span(0,
  2371. 1), ((0, 7), (0, 7)))
  2372. self.assertEqual(regex.search('^(foobar){e<=1}$', 'foobarx').span(0,
  2373. 1), ((0, 7), (0, 7)))
  2374. self.assertEqual(regex.search('^(foobar){e<=1}$', 'fooxbar').span(0,
  2375. 1), ((0, 7), (0, 7)))
  2376. self.assertEqual(regex.search('^(foobar){e<=1}$', 'foxbar').span(0, 1),
  2377. ((0, 6), (0, 6)))
  2378. self.assertEqual(regex.search('^(foobar){e<=1}$', 'xoobar').span(0, 1),
  2379. ((0, 6), (0, 6)))
  2380. self.assertEqual(regex.search('^(foobar){e<=1}$', 'foobax').span(0, 1),
  2381. ((0, 6), (0, 6)))
  2382. self.assertEqual(regex.search('^(foobar){e<=1}$', 'oobar').span(0, 1),
  2383. ((0, 5), (0, 5)))
  2384. self.assertEqual(regex.search('^(foobar){e<=1}$', 'fobar').span(0, 1),
  2385. ((0, 5), (0, 5)))
  2386. self.assertEqual(regex.search('^(foobar){e<=1}$', 'fooba').span(0, 1),
  2387. ((0, 5), (0, 5)))
  2388. self.assertEqual(regex.search('^(foobar){e<=1}$', 'xfoobarx'), None)
  2389. self.assertEqual(regex.search('^(foobar){e<=1}$', 'foobarxx'), None)
  2390. self.assertEqual(regex.search('^(foobar){e<=1}$', 'xxfoobar'), None)
  2391. self.assertEqual(regex.search('^(foobar){e<=1}$', 'xfoxbar'), None)
  2392. self.assertEqual(regex.search('^(foobar){e<=1}$', 'foxbarx'), None)
  2393. # At most one insert, two deletes, and three substitutions.
  2394. # Additionally, deletes cost two and substitutes one, and total
  2395. # cost must be less than 4.
  2396. self.assertEqual(regex.search('(foobar){i<=1,d<=2,s<=3,2d+1s<4}',
  2397. '3oifaowefbaoraofuiebofasebfaobfaorfeoaro').span(0, 1), ((6, 13), (6,
  2398. 13)))
  2399. self.assertEqual(regex.search('(?b)(foobar){i<=1,d<=2,s<=3,2d+1s<4}',
  2400. '3oifaowefbaoraofuiebofasebfaobfaorfeoaro').span(0, 1), ((34, 39),
  2401. (34, 39)))
  2402. # Partially fuzzy matches.
  2403. self.assertEqual(regex.search('foo(bar){e<=1}zap', 'foobarzap').span(0,
  2404. 1), ((0, 9), (3, 6)))
  2405. self.assertEqual(regex.search('foo(bar){e<=1}zap', 'fobarzap'), None)
  2406. self.assertEqual(regex.search('foo(bar){e<=1}zap', 'foobrzap').span(0,
  2407. 1), ((0, 8), (3, 5)))
  2408. text = ('www.cnn.com 64.236.16.20\nwww.slashdot.org 66.35.250.150\n'
  2409. 'For useful information, use www.slashdot.org\nthis is demo data!\n')
  2410. self.assertEqual(regex.search(r'(?s)^.*(dot.org){e}.*$', text).span(0,
  2411. 1), ((0, 120), (120, 120)))
  2412. self.assertEqual(regex.search(r'(?es)^.*(dot.org){e}.*$', text).span(0,
  2413. 1), ((0, 120), (93, 100)))
  2414. self.assertEqual(regex.search(r'^.*(dot.org){e}.*$', text).span(0, 1),
  2415. ((0, 119), (24, 101)))
  2416. # Behaviour is unexpected, but arguably not wrong. It first finds the
  2417. # best match, then the best in what follows, etc.
  2418. self.assertEqual(regex.findall(r"\b\L<words>{e<=1}\b",
  2419. " book cot dog desk ", words="cat dog".split()), ["cot", "dog"])
  2420. self.assertEqual(regex.findall(r"\b\L<words>{e<=1}\b",
  2421. " book dog cot desk ", words="cat dog".split()), [" dog", "cot"])
  2422. self.assertEqual(regex.findall(r"(?e)\b\L<words>{e<=1}\b",
  2423. " book dog cot desk ", words="cat dog".split()), ["dog", "cot"])
  2424. self.assertEqual(regex.findall(r"(?r)\b\L<words>{e<=1}\b",
  2425. " book cot dog desk ", words="cat dog".split()), ["dog ", "cot"])
  2426. self.assertEqual(regex.findall(r"(?er)\b\L<words>{e<=1}\b",
  2427. " book cot dog desk ", words="cat dog".split()), ["dog", "cot"])
  2428. self.assertEqual(regex.findall(r"(?r)\b\L<words>{e<=1}\b",
  2429. " book dog cot desk ", words="cat dog".split()), ["cot", "dog"])
  2430. self.assertEqual(regex.findall(br"\b\L<words>{e<=1}\b",
  2431. b" book cot dog desk ", words=b"cat dog".split()), [b"cot", b"dog"])
  2432. self.assertEqual(regex.findall(br"\b\L<words>{e<=1}\b",
  2433. b" book dog cot desk ", words=b"cat dog".split()), [b" dog", b"cot"])
  2434. self.assertEqual(regex.findall(br"(?e)\b\L<words>{e<=1}\b",
  2435. b" book dog cot desk ", words=b"cat dog".split()), [b"dog", b"cot"])
  2436. self.assertEqual(regex.findall(br"(?r)\b\L<words>{e<=1}\b",
  2437. b" book cot dog desk ", words=b"cat dog".split()), [b"dog ", b"cot"])
  2438. self.assertEqual(regex.findall(br"(?er)\b\L<words>{e<=1}\b",
  2439. b" book cot dog desk ", words=b"cat dog".split()), [b"dog", b"cot"])
  2440. self.assertEqual(regex.findall(br"(?r)\b\L<words>{e<=1}\b",
  2441. b" book dog cot desk ", words=b"cat dog".split()), [b"cot", b"dog"])
  2442. self.assertEqual(regex.search(r"(\w+) (\1{e<=1})", "foo fou").groups(),
  2443. ("foo", "fou"))
  2444. self.assertEqual(regex.search(r"(?r)(\2{e<=1}) (\w+)",
  2445. "foo fou").groups(), ("foo", "fou"))
  2446. self.assertEqual(regex.search(br"(\w+) (\1{e<=1})",
  2447. b"foo fou").groups(), (b"foo", b"fou"))
  2448. self.assertEqual(regex.findall(r"(?:(?:QR)+){e}", "abcde"), ["abcde",
  2449. ""])
  2450. self.assertEqual(regex.findall(r"(?:Q+){e}", "abc"), ["abc", ""])
  2451. # Hg issue 41: = for fuzzy matches
  2452. self.assertEqual(regex.match(r"(?:service detection){0<e<5}",
  2453. "servic detection").span(), (0, 16))
  2454. self.assertEqual(regex.match(r"(?:service detection){0<e<5}",
  2455. "service detect").span(), (0, 14))
  2456. self.assertEqual(regex.match(r"(?:service detection){0<e<5}",
  2457. "service detecti").span(), (0, 15))
  2458. self.assertEqual(regex.match(r"(?:service detection){0<e<5}",
  2459. "service detection"), None)
  2460. self.assertEqual(regex.match(r"(?:service detection){0<e<5}",
  2461. "in service detection").span(), (0, 20))
  2462. # Hg issue 109: Edit distance of fuzzy match
  2463. self.assertEqual(regex.fullmatch(r"(?:cats|cat){e<=1}",
  2464. "cat").fuzzy_counts, (0, 0, 1))
  2465. self.assertEqual(regex.fullmatch(r"(?e)(?:cats|cat){e<=1}",
  2466. "cat").fuzzy_counts, (0, 0, 0))
  2467. self.assertEqual(regex.fullmatch(r"(?:cat|cats){e<=1}",
  2468. "cats").fuzzy_counts, (0, 1, 0))
  2469. self.assertEqual(regex.fullmatch(r"(?e)(?:cat|cats){e<=1}",
  2470. "cats").fuzzy_counts, (0, 0, 0))
  2471. self.assertEqual(regex.fullmatch(r"(?:cat){e<=1} (?:cat){e<=1}",
  2472. "cat cot").fuzzy_counts, (1, 0, 0))
  2473. def test_recursive(self):
  2474. self.assertEqual(regex.search(r"(\w)(?:(?R)|(\w?))\1", "xx")[ : ],
  2475. ("xx", "x", ""))
  2476. self.assertEqual(regex.search(r"(\w)(?:(?R)|(\w?))\1", "aba")[ : ],
  2477. ("aba", "a", "b"))
  2478. self.assertEqual(regex.search(r"(\w)(?:(?R)|(\w?))\1", "abba")[ : ],
  2479. ("abba", "a", None))
  2480. self.assertEqual(regex.search(r"(\w)(?:(?R)|(\w?))\1", "kayak")[ : ],
  2481. ("kayak", "k", None))
  2482. self.assertEqual(regex.search(r"(\w)(?:(?R)|(\w?))\1", "paper")[ : ],
  2483. ("pap", "p", "a"))
  2484. self.assertEqual(regex.search(r"(\w)(?:(?R)|(\w?))\1", "dontmatchme"),
  2485. None)
  2486. self.assertEqual(regex.search(r"(?r)\2(?:(\w?)|(?R))(\w)", "xx")[ : ],
  2487. ("xx", "", "x"))
  2488. self.assertEqual(regex.search(r"(?r)\2(?:(\w?)|(?R))(\w)", "aba")[ : ],
  2489. ("aba", "b", "a"))
  2490. self.assertEqual(regex.search(r"(?r)\2(?:(\w?)|(?R))(\w)", "abba")[ :
  2491. ], ("abba", None, "a"))
  2492. self.assertEqual(regex.search(r"(?r)\2(?:(\w?)|(?R))(\w)", "kayak")[ :
  2493. ], ("kayak", None, "k"))
  2494. self.assertEqual(regex.search(r"(?r)\2(?:(\w?)|(?R))(\w)", "paper")[ :
  2495. ], ("pap", "a", "p"))
  2496. self.assertEqual(regex.search(r"(?r)\2(?:(\w?)|(?R))(\w)",
  2497. "dontmatchme"), None)
  2498. self.assertEqual(regex.search(r"\(((?>[^()]+)|(?R))*\)", "(ab(cd)ef)")[
  2499. : ], ("(ab(cd)ef)", "ef"))
  2500. self.assertEqual(regex.search(r"\(((?>[^()]+)|(?R))*\)",
  2501. "(ab(cd)ef)").captures(1), ["ab", "cd", "(cd)", "ef"])
  2502. self.assertEqual(regex.search(r"(?r)\(((?R)|(?>[^()]+))*\)",
  2503. "(ab(cd)ef)")[ : ], ("(ab(cd)ef)", "ab"))
  2504. self.assertEqual(regex.search(r"(?r)\(((?R)|(?>[^()]+))*\)",
  2505. "(ab(cd)ef)").captures(1), ["ef", "cd", "(cd)", "ab"])
  2506. self.assertEqual(regex.search(r"\(([^()]+|(?R))*\)",
  2507. "some text (a(b(c)d)e) more text")[ : ], ("(a(b(c)d)e)", "e"))
  2508. self.assertEqual(regex.search(r"(?r)\(((?R)|[^()]+)*\)",
  2509. "some text (a(b(c)d)e) more text")[ : ], ("(a(b(c)d)e)", "a"))
  2510. self.assertEqual(regex.search(r"(foo(\(((?:(?>[^()]+)|(?2))*)\)))",
  2511. "foo(bar(baz)+baz(bop))")[ : ], ("foo(bar(baz)+baz(bop))",
  2512. "foo(bar(baz)+baz(bop))", "(bar(baz)+baz(bop))",
  2513. "bar(baz)+baz(bop)"))
  2514. self.assertEqual(regex.search(r"(?r)(foo(\(((?:(?2)|(?>[^()]+))*)\)))",
  2515. "foo(bar(baz)+baz(bop))")[ : ], ("foo(bar(baz)+baz(bop))",
  2516. "foo(bar(baz)+baz(bop))", "(bar(baz)+baz(bop))",
  2517. "bar(baz)+baz(bop)"))
  2518. rgx = regex.compile(r"""^\s*(<\s*([a-zA-Z:]+)(?:\s*[a-zA-Z:]*\s*=\s*(?:'[^']*'|"[^"]*"))*\s*(/\s*)?>(?:[^<>]*|(?1))*(?(3)|<\s*/\s*\2\s*>))\s*$""")
  2519. self.assertEqual(bool(rgx.search('<foo><bar></bar></foo>')), True)
  2520. self.assertEqual(bool(rgx.search('<foo><bar></foo></bar>')), False)
  2521. self.assertEqual(bool(rgx.search('<foo><bar/></foo>')), True)
  2522. self.assertEqual(bool(rgx.search('<foo><bar></foo>')), False)
  2523. self.assertEqual(bool(rgx.search('<foo bar=baz/>')), False)
  2524. self.assertEqual(bool(rgx.search('<foo bar="baz">')), False)
  2525. self.assertEqual(bool(rgx.search('<foo bar="baz"/>')), True)
  2526. self.assertEqual(bool(rgx.search('< fooo / >')), True)
  2527. # The next regex should and does match. Perl 5.14 agrees.
  2528. #self.assertEqual(bool(rgx.search('<foo/>foo')), False)
  2529. self.assertEqual(bool(rgx.search('foo<foo/>')), False)
  2530. self.assertEqual(bool(rgx.search('<foo>foo</foo>')), True)
  2531. self.assertEqual(bool(rgx.search('<foo><bar/>foo</foo>')), True)
  2532. self.assertEqual(bool(rgx.search('<a><b><c></c></b></a>')), True)
  2533. def test_copy(self):
  2534. # PatternObjects are immutable, therefore there's no need to clone them.
  2535. r = regex.compile("a")
  2536. self.assertTrue(copy.copy(r) is r)
  2537. self.assertTrue(copy.deepcopy(r) is r)
  2538. # MatchObjects are normally mutable because the target string can be
  2539. # detached. However, after the target string has been detached, a
  2540. # MatchObject becomes immutable, so there's no need to clone it.
  2541. m = r.match("a")
  2542. self.assertTrue(copy.copy(m) is not m)
  2543. self.assertTrue(copy.deepcopy(m) is not m)
  2544. self.assertTrue(m.string is not None)
  2545. m2 = copy.copy(m)
  2546. m2.detach_string()
  2547. self.assertTrue(m.string is not None)
  2548. self.assertTrue(m2.string is None)
  2549. # The following behaviour matches that of the re module.
  2550. it = regex.finditer(".", "ab")
  2551. it2 = copy.copy(it)
  2552. self.assertEqual(next(it).group(), "a")
  2553. self.assertEqual(next(it2).group(), "b")
  2554. # The following behaviour matches that of the re module.
  2555. it = regex.finditer(".", "ab")
  2556. it2 = copy.deepcopy(it)
  2557. self.assertEqual(next(it).group(), "a")
  2558. self.assertEqual(next(it2).group(), "b")
  2559. # The following behaviour is designed to match that of copying 'finditer'.
  2560. it = regex.splititer(" ", "a b")
  2561. it2 = copy.copy(it)
  2562. self.assertEqual(next(it), "a")
  2563. self.assertEqual(next(it2), "b")
  2564. # The following behaviour is designed to match that of copying 'finditer'.
  2565. it = regex.splititer(" ", "a b")
  2566. it2 = copy.deepcopy(it)
  2567. self.assertEqual(next(it), "a")
  2568. self.assertEqual(next(it2), "b")
  2569. def test_format(self):
  2570. self.assertEqual(regex.subf(r"(\w+) (\w+)", "{0} => {2} {1}",
  2571. "foo bar"), "foo bar => bar foo")
  2572. self.assertEqual(regex.subf(r"(?<word1>\w+) (?<word2>\w+)",
  2573. "{word2} {word1}", "foo bar"), "bar foo")
  2574. self.assertEqual(regex.subfn(r"(\w+) (\w+)", "{0} => {2} {1}",
  2575. "foo bar"), ("foo bar => bar foo", 1))
  2576. self.assertEqual(regex.subfn(r"(?<word1>\w+) (?<word2>\w+)",
  2577. "{word2} {word1}", "foo bar"), ("bar foo", 1))
  2578. self.assertEqual(regex.match(r"(\w+) (\w+)",
  2579. "foo bar").expandf("{0} => {2} {1}"), "foo bar => bar foo")
  2580. def test_fullmatch(self):
  2581. self.assertEqual(bool(regex.fullmatch(r"abc", "abc")), True)
  2582. self.assertEqual(bool(regex.fullmatch(r"abc", "abcx")), False)
  2583. self.assertEqual(bool(regex.fullmatch(r"abc", "abcx", endpos=3)), True)
  2584. self.assertEqual(bool(regex.fullmatch(r"abc", "xabc", pos=1)), True)
  2585. self.assertEqual(bool(regex.fullmatch(r"abc", "xabcy", pos=1)), False)
  2586. self.assertEqual(bool(regex.fullmatch(r"abc", "xabcy", pos=1,
  2587. endpos=4)), True)
  2588. self.assertEqual(bool(regex.fullmatch(r"(?r)abc", "abc")), True)
  2589. self.assertEqual(bool(regex.fullmatch(r"(?r)abc", "abcx")), False)
  2590. self.assertEqual(bool(regex.fullmatch(r"(?r)abc", "abcx", endpos=3)),
  2591. True)
  2592. self.assertEqual(bool(regex.fullmatch(r"(?r)abc", "xabc", pos=1)),
  2593. True)
  2594. self.assertEqual(bool(regex.fullmatch(r"(?r)abc", "xabcy", pos=1)),
  2595. False)
  2596. self.assertEqual(bool(regex.fullmatch(r"(?r)abc", "xabcy", pos=1,
  2597. endpos=4)), True)
  2598. def test_issue_18468(self):
  2599. # Applies only after Python 3.4 for compatibility with re.
  2600. if (sys.version_info.major, sys.version_info.minor) < (3, 4):
  2601. return
  2602. self.assertTypedEqual(regex.sub('y', 'a', 'xyz'), 'xaz')
  2603. self.assertTypedEqual(regex.sub('y', StrSubclass('a'),
  2604. StrSubclass('xyz')), 'xaz')
  2605. self.assertTypedEqual(regex.sub(b'y', b'a', b'xyz'), b'xaz')
  2606. self.assertTypedEqual(regex.sub(b'y', BytesSubclass(b'a'),
  2607. BytesSubclass(b'xyz')), b'xaz')
  2608. self.assertTypedEqual(regex.sub(b'y', bytearray(b'a'),
  2609. bytearray(b'xyz')), b'xaz')
  2610. self.assertTypedEqual(regex.sub(b'y', memoryview(b'a'),
  2611. memoryview(b'xyz')), b'xaz')
  2612. for string in ":a:b::c", StrSubclass(":a:b::c"):
  2613. self.assertTypedEqual(regex.split(":", string), ['', 'a', 'b', '',
  2614. 'c'])
  2615. if sys.version_info >= (3, 7, 0):
  2616. self.assertTypedEqual(regex.split(":*", string), ['', '', 'a',
  2617. '', 'b', '', 'c', ''])
  2618. self.assertTypedEqual(regex.split("(:*)", string), ['', ':',
  2619. '', '', 'a', ':', '', '', 'b', '::', '', '', 'c', '', ''])
  2620. else:
  2621. self.assertTypedEqual(regex.split(":*", string), ['', 'a', 'b',
  2622. 'c'])
  2623. self.assertTypedEqual(regex.split("(:*)", string), ['', ':',
  2624. 'a', ':', 'b', '::', 'c'])
  2625. for string in (b":a:b::c", BytesSubclass(b":a:b::c"),
  2626. bytearray(b":a:b::c"), memoryview(b":a:b::c")):
  2627. self.assertTypedEqual(regex.split(b":", string), [b'', b'a', b'b',
  2628. b'', b'c'])
  2629. if sys.version_info >= (3, 7, 0):
  2630. self.assertTypedEqual(regex.split(b":*", string), [b'', b'',
  2631. b'a', b'', b'b', b'', b'c', b''])
  2632. self.assertTypedEqual(regex.split(b"(:*)", string), [b'', b':',
  2633. b'', b'', b'a', b':', b'', b'', b'b', b'::', b'', b'', b'c',
  2634. b'', b''])
  2635. else:
  2636. self.assertTypedEqual(regex.split(b":*", string), [b'', b'a',
  2637. b'b', b'c'])
  2638. self.assertTypedEqual(regex.split(b"(:*)", string), [b'', b':',
  2639. b'a', b':', b'b', b'::', b'c'])
  2640. for string in "a:b::c:::d", StrSubclass("a:b::c:::d"):
  2641. self.assertTypedEqual(regex.findall(":+", string), [":", "::",
  2642. ":::"])
  2643. self.assertTypedEqual(regex.findall("(:+)", string), [":", "::",
  2644. ":::"])
  2645. self.assertTypedEqual(regex.findall("(:)(:*)", string), [(":", ""),
  2646. (":", ":"), (":", "::")])
  2647. for string in (b"a:b::c:::d", BytesSubclass(b"a:b::c:::d"),
  2648. bytearray(b"a:b::c:::d"), memoryview(b"a:b::c:::d")):
  2649. self.assertTypedEqual(regex.findall(b":+", string), [b":", b"::",
  2650. b":::"])
  2651. self.assertTypedEqual(regex.findall(b"(:+)", string), [b":", b"::",
  2652. b":::"])
  2653. self.assertTypedEqual(regex.findall(b"(:)(:*)", string), [(b":",
  2654. b""), (b":", b":"), (b":", b"::")])
  2655. for string in 'a', StrSubclass('a'):
  2656. self.assertEqual(regex.match('a', string).groups(), ())
  2657. self.assertEqual(regex.match('(a)', string).groups(), ('a',))
  2658. self.assertEqual(regex.match('(a)', string).group(0), 'a')
  2659. self.assertEqual(regex.match('(a)', string).group(1), 'a')
  2660. self.assertEqual(regex.match('(a)', string).group(1, 1), ('a',
  2661. 'a'))
  2662. for string in (b'a', BytesSubclass(b'a'), bytearray(b'a'),
  2663. memoryview(b'a')):
  2664. self.assertEqual(regex.match(b'a', string).groups(), ())
  2665. self.assertEqual(regex.match(b'(a)', string).groups(), (b'a',))
  2666. self.assertEqual(regex.match(b'(a)', string).group(0), b'a')
  2667. self.assertEqual(regex.match(b'(a)', string).group(1), b'a')
  2668. self.assertEqual(regex.match(b'(a)', string).group(1, 1), (b'a',
  2669. b'a'))
  2670. def test_partial(self):
  2671. self.assertEqual(regex.match('ab', 'a', partial=True).partial, True)
  2672. self.assertEqual(regex.match('ab', 'a', partial=True).span(), (0, 1))
  2673. self.assertEqual(regex.match(r'cats', 'cat', partial=True).partial,
  2674. True)
  2675. self.assertEqual(regex.match(r'cats', 'cat', partial=True).span(), (0,
  2676. 3))
  2677. self.assertEqual(regex.match(r'cats', 'catch', partial=True), None)
  2678. self.assertEqual(regex.match(r'abc\w{3}', 'abcdef',
  2679. partial=True).partial, False)
  2680. self.assertEqual(regex.match(r'abc\w{3}', 'abcdef',
  2681. partial=True).span(), (0, 6))
  2682. self.assertEqual(regex.match(r'abc\w{3}', 'abcde',
  2683. partial=True).partial, True)
  2684. self.assertEqual(regex.match(r'abc\w{3}', 'abcde',
  2685. partial=True).span(), (0, 5))
  2686. self.assertEqual(regex.match(r'\d{4}$', '1234', partial=True).partial,
  2687. False)
  2688. self.assertEqual(regex.match(r'\L<words>', 'post', partial=True,
  2689. words=['post']).partial, False)
  2690. self.assertEqual(regex.match(r'\L<words>', 'post', partial=True,
  2691. words=['post']).span(), (0, 4))
  2692. self.assertEqual(regex.match(r'\L<words>', 'pos', partial=True,
  2693. words=['post']).partial, True)
  2694. self.assertEqual(regex.match(r'\L<words>', 'pos', partial=True,
  2695. words=['post']).span(), (0, 3))
  2696. self.assertEqual(regex.match(r'(?fi)\L<words>', 'POST', partial=True,
  2697. words=['po\uFB06']).partial, False)
  2698. self.assertEqual(regex.match(r'(?fi)\L<words>', 'POST', partial=True,
  2699. words=['po\uFB06']).span(), (0, 4))
  2700. self.assertEqual(regex.match(r'(?fi)\L<words>', 'POS', partial=True,
  2701. words=['po\uFB06']).partial, True)
  2702. self.assertEqual(regex.match(r'(?fi)\L<words>', 'POS', partial=True,
  2703. words=['po\uFB06']).span(), (0, 3))
  2704. self.assertEqual(regex.match(r'(?fi)\L<words>', 'po\uFB06',
  2705. partial=True, words=['POS']), None)
  2706. self.assertEqual(regex.match(r'[a-z]*4R$', 'a', partial=True).span(),
  2707. (0, 1))
  2708. self.assertEqual(regex.match(r'[a-z]*4R$', 'ab', partial=True).span(),
  2709. (0, 2))
  2710. self.assertEqual(regex.match(r'[a-z]*4R$', 'ab4', partial=True).span(),
  2711. (0, 3))
  2712. self.assertEqual(regex.match(r'[a-z]*4R$', 'a4', partial=True).span(),
  2713. (0, 2))
  2714. self.assertEqual(regex.match(r'[a-z]*4R$', 'a4R', partial=True).span(),
  2715. (0, 3))
  2716. self.assertEqual(regex.match(r'[a-z]*4R$', '4a', partial=True), None)
  2717. self.assertEqual(regex.match(r'[a-z]*4R$', 'a44', partial=True), None)
  2718. def test_hg_bugs(self):
  2719. # Hg issue 28: regex.compile("(?>b)") causes "TypeError: 'Character'
  2720. # object is not subscriptable"
  2721. self.assertEqual(bool(regex.compile("(?>b)", flags=regex.V1)), True)
  2722. # Hg issue 29: regex.compile("^((?>\w+)|(?>\s+))*$") causes
  2723. # "TypeError: 'GreedyRepeat' object is not iterable"
  2724. self.assertEqual(bool(regex.compile(r"^((?>\w+)|(?>\s+))*$",
  2725. flags=regex.V1)), True)
  2726. # Hg issue 31: atomic and normal groups in recursive patterns
  2727. self.assertEqual(regex.findall(r"\((?:(?>[^()]+)|(?R))*\)",
  2728. "a(bcd(e)f)g(h)"), ['(bcd(e)f)', '(h)'])
  2729. self.assertEqual(regex.findall(r"\((?:(?:[^()]+)|(?R))*\)",
  2730. "a(bcd(e)f)g(h)"), ['(bcd(e)f)', '(h)'])
  2731. self.assertEqual(regex.findall(r"\((?:(?>[^()]+)|(?R))*\)",
  2732. "a(b(cd)e)f)g)h"), ['(b(cd)e)'])
  2733. self.assertEqual(regex.findall(r"\((?:(?>[^()]+)|(?R))*\)",
  2734. "a(bc(d(e)f)gh"), ['(d(e)f)'])
  2735. self.assertEqual(regex.findall(r"(?r)\((?:(?>[^()]+)|(?R))*\)",
  2736. "a(bc(d(e)f)gh"), ['(d(e)f)'])
  2737. self.assertEqual([m.group() for m in
  2738. regex.finditer(r"\((?:[^()]*+|(?0))*\)", "a(b(c(de)fg)h")],
  2739. ['(c(de)fg)'])
  2740. # Hg issue 32: regex.search("a(bc)d", "abcd", regex.I|regex.V1) returns
  2741. # None
  2742. self.assertEqual(regex.search("a(bc)d", "abcd", regex.I |
  2743. regex.V1).group(0), "abcd")
  2744. # Hg issue 33: regex.search("([\da-f:]+)$", "E", regex.I|regex.V1)
  2745. # returns None
  2746. self.assertEqual(regex.search(r"([\da-f:]+)$", "E", regex.I |
  2747. regex.V1).group(0), "E")
  2748. self.assertEqual(regex.search(r"([\da-f:]+)$", "e", regex.I |
  2749. regex.V1).group(0), "e")
  2750. # Hg issue 34: regex.search("^(?=ab(de))(abd)(e)", "abde").groups()
  2751. # returns (None, 'abd', 'e') instead of ('de', 'abd', 'e')
  2752. self.assertEqual(regex.search("^(?=ab(de))(abd)(e)", "abde").groups(),
  2753. ('de', 'abd', 'e'))
  2754. # Hg issue 35: regex.compile("\ ", regex.X) causes "_regex_core.error:
  2755. # bad escape"
  2756. self.assertEqual(bool(regex.match(r"\ ", " ", flags=regex.X)), True)
  2757. # Hg issue 36: regex.search("^(a|)\1{2}b", "b") returns None
  2758. self.assertEqual(regex.search(r"^(a|)\1{2}b", "b").group(0, 1), ('b',
  2759. ''))
  2760. # Hg issue 37: regex.search("^(a){0,0}", "abc").group(0,1) returns
  2761. # ('a', 'a') instead of ('', None)
  2762. self.assertEqual(regex.search("^(a){0,0}", "abc").group(0, 1), ('',
  2763. None))
  2764. # Hg issue 38: regex.search("(?>.*/)b", "a/b") returns None
  2765. self.assertEqual(regex.search("(?>.*/)b", "a/b").group(0), "a/b")
  2766. # Hg issue 39: regex.search("((?i)blah)\\s+\\1", "blah BLAH") doesn't
  2767. # return None
  2768. self.assertEqual(regex.search(r"(?V0)((?i)blah)\s+\1",
  2769. "blah BLAH").group(0, 1), ("blah BLAH", "blah"))
  2770. self.assertEqual(regex.search(r"(?V1)((?i)blah)\s+\1", "blah BLAH"),
  2771. None)
  2772. # Hg issue 40: regex.search("(\()?[^()]+(?(1)\)|)", "(abcd").group(0)
  2773. # returns "bcd" instead of "abcd"
  2774. self.assertEqual(regex.search(r"(\()?[^()]+(?(1)\)|)",
  2775. "(abcd").group(0), "abcd")
  2776. # Hg issue 42: regex.search("(a*)*", "a", flags=regex.V1).span(1)
  2777. # returns (0, 1) instead of (1, 1)
  2778. self.assertEqual(regex.search("(a*)*", "a").span(1), (1, 1))
  2779. self.assertEqual(regex.search("(a*)*", "aa").span(1), (2, 2))
  2780. self.assertEqual(regex.search("(a*)*", "aaa").span(1), (3, 3))
  2781. # Hg issue 43: regex.compile("a(?#xxx)*") causes "_regex_core.error:
  2782. # nothing to repeat"
  2783. self.assertEqual(regex.search("a(?#xxx)*", "aaa").group(), "aaa")
  2784. # Hg issue 44: regex.compile("(?=abc){3}abc") causes
  2785. # "_regex_core.error: nothing to repeat"
  2786. self.assertEqual(regex.search("(?=abc){3}abc", "abcabcabc").span(), (0,
  2787. 3))
  2788. # Hg issue 45: regex.compile("^(?:a(?:(?:))+)+") causes
  2789. # "_regex_core.error: nothing to repeat"
  2790. self.assertEqual(regex.search("^(?:a(?:(?:))+)+", "a").span(), (0, 1))
  2791. self.assertEqual(regex.search("^(?:a(?:(?:))+)+", "aa").span(), (0, 2))
  2792. # Hg issue 46: regex.compile("a(?x: b c )d") causes
  2793. # "_regex_core.error: missing )"
  2794. self.assertEqual(regex.search("a(?x: b c )d", "abcd").group(0), "abcd")
  2795. # Hg issue 47: regex.compile("a#comment\n*", flags=regex.X) causes
  2796. # "_regex_core.error: nothing to repeat"
  2797. self.assertEqual(regex.search("a#comment\n*", "aaa",
  2798. flags=regex.X).group(0), "aaa")
  2799. # Hg issue 48: regex.search("(a(?(1)\\1)){4}", "a"*10,
  2800. # flags=regex.V1).group(0,1) returns ('aaaaa', 'a') instead of ('aaaaaaaaaa', 'aaaa')
  2801. self.assertEqual(regex.search(r"(?V1)(a(?(1)\1)){1}",
  2802. "aaaaaaaaaa").span(0, 1), ((0, 1), (0, 1)))
  2803. self.assertEqual(regex.search(r"(?V1)(a(?(1)\1)){2}",
  2804. "aaaaaaaaaa").span(0, 1), ((0, 3), (1, 3)))
  2805. self.assertEqual(regex.search(r"(?V1)(a(?(1)\1)){3}",
  2806. "aaaaaaaaaa").span(0, 1), ((0, 6), (3, 6)))
  2807. self.assertEqual(regex.search(r"(?V1)(a(?(1)\1)){4}",
  2808. "aaaaaaaaaa").span(0, 1), ((0, 10), (6, 10)))
  2809. # Hg issue 49: regex.search("(a)(?<=b(?1))", "baz", regex.V1) returns
  2810. # None incorrectly
  2811. self.assertEqual(regex.search("(?V1)(a)(?<=b(?1))", "baz").group(0),
  2812. "a")
  2813. # Hg issue 50: not all keywords are found by named list with
  2814. # overlapping keywords when full Unicode casefolding is required
  2815. self.assertEqual(regex.findall(r'(?fi)\L<keywords>',
  2816. 'POST, Post, post, po\u017Ft, po\uFB06, and po\uFB05',
  2817. keywords=['post','pos']), ['POST', 'Post', 'post', 'po\u017Ft',
  2818. 'po\uFB06', 'po\uFB05'])
  2819. self.assertEqual(regex.findall(r'(?fi)pos|post',
  2820. 'POST, Post, post, po\u017Ft, po\uFB06, and po\uFB05'), ['POS',
  2821. 'Pos', 'pos', 'po\u017F', 'po\uFB06', 'po\uFB05'])
  2822. self.assertEqual(regex.findall(r'(?fi)post|pos',
  2823. 'POST, Post, post, po\u017Ft, po\uFB06, and po\uFB05'), ['POST',
  2824. 'Post', 'post', 'po\u017Ft', 'po\uFB06', 'po\uFB05'])
  2825. self.assertEqual(regex.findall(r'(?fi)post|another',
  2826. 'POST, Post, post, po\u017Ft, po\uFB06, and po\uFB05'), ['POST',
  2827. 'Post', 'post', 'po\u017Ft', 'po\uFB06', 'po\uFB05'])
  2828. # Hg issue 51: regex.search("((a)(?1)|(?2))", "a", flags=regex.V1)
  2829. # returns None incorrectly
  2830. self.assertEqual(regex.search("(?V1)((a)(?1)|(?2))", "a").group(0, 1,
  2831. 2), ('a', 'a', None))
  2832. # Hg issue 52: regex.search("(\\1xx|){6}", "xx",
  2833. # flags=regex.V1).span(0,1) returns incorrect value
  2834. self.assertEqual(regex.search(r"(?V1)(\1xx|){6}", "xx").span(0, 1),
  2835. ((0, 2), (2, 2)))
  2836. # Hg issue 53: regex.search("(a|)+", "a") causes MemoryError
  2837. self.assertEqual(regex.search("(a|)+", "a").group(0, 1), ("a", ""))
  2838. # Hg issue 54: regex.search("(a|)*\\d", "a"*80) causes MemoryError
  2839. self.assertEqual(regex.search(r"(a|)*\d", "a" * 80), None)
  2840. # Hg issue 55: regex.search("^(?:a?b?)*$", "ac") take a very long time.
  2841. self.assertEqual(regex.search("^(?:a?b?)*$", "ac"), None)
  2842. # Hg issue 58: bad named character escape sequences like "\\N{1}"
  2843. # treats as "N"
  2844. self.assertRaisesRegex(regex.error, self.UNDEF_CHAR_NAME, lambda:
  2845. regex.compile("\\N{1}"))
  2846. # Hg issue 59: regex.search("\\Z", "a\na\n") returns None incorrectly
  2847. self.assertEqual(regex.search("\\Z", "a\na\n").span(0), (4, 4))
  2848. # Hg issue 60: regex.search("(q1|.)*(q2|.)*(x(a|bc)*y){2,}", "xayxay")
  2849. # returns None incorrectly
  2850. self.assertEqual(regex.search("(q1|.)*(q2|.)*(x(a|bc)*y){2,}",
  2851. "xayxay").group(0), "xayxay")
  2852. # Hg issue 61: regex.search("[^a]", "A", regex.I).group(0) returns ''
  2853. # incorrectly
  2854. self.assertEqual(regex.search("(?i)[^a]", "A"), None)
  2855. # Hg issue 63: regex.search("[[:ascii:]]", "\N{KELVIN SIGN}",
  2856. # flags=regex.I|regex.V1) doesn't return None
  2857. self.assertEqual(regex.search("(?i)[[:ascii:]]", "\N{KELVIN SIGN}"),
  2858. None)
  2859. # Hg issue 66: regex.search("((a|b(?1)c){3,5})", "baaaaca",
  2860. # flags=regex.V1).groups() returns ('baaaac', 'baaaac') instead of ('aaaa', 'a')
  2861. self.assertEqual(regex.search("((a|b(?1)c){3,5})", "baaaaca").group(0,
  2862. 1, 2), ('aaaa', 'aaaa', 'a'))
  2863. # Hg issue 71: non-greedy quantifier in lookbehind
  2864. self.assertEqual(regex.findall(r"(?<=:\S+ )\w+", ":9 abc :10 def"),
  2865. ['abc', 'def'])
  2866. self.assertEqual(regex.findall(r"(?<=:\S* )\w+", ":9 abc :10 def"),
  2867. ['abc', 'def'])
  2868. self.assertEqual(regex.findall(r"(?<=:\S+? )\w+", ":9 abc :10 def"),
  2869. ['abc', 'def'])
  2870. self.assertEqual(regex.findall(r"(?<=:\S*? )\w+", ":9 abc :10 def"),
  2871. ['abc', 'def'])
  2872. # Hg issue 73: conditional patterns
  2873. self.assertEqual(regex.search(r"(?:fe)?male", "female").group(),
  2874. "female")
  2875. self.assertEqual([m.group() for m in
  2876. regex.finditer(r"(fe)?male: h(?(1)(er)|(is)) (\w+)",
  2877. "female: her dog; male: his cat. asdsasda")], ['female: her dog',
  2878. 'male: his cat'])
  2879. # Hg issue 78: "Captures"doesn't work for recursive calls
  2880. self.assertEqual(regex.search(r'(?<rec>\((?:[^()]++|(?&rec))*\))',
  2881. 'aaa(((1+0)+1)+1)bbb').captures('rec'), ['(1+0)', '((1+0)+1)',
  2882. '(((1+0)+1)+1)'])
  2883. # Hg issue 80: Escape characters throws an exception
  2884. self.assertRaisesRegex(regex.error, self.BAD_ESCAPE, lambda:
  2885. regex.sub('x', '\\', 'x'), )
  2886. # Hg issue 82: error range does not work
  2887. fz = "(CAGCCTCCCATTTCAGAATATACATCC){1<e<=2}"
  2888. seq = "tcagacgagtgcgttgtaaaacgacggccagtCAGCCTCCCATTCAGAATATACATCCcgacggccagttaaaaacaatgccaaggaggtcatagctgtttcctgccagttaaaaacaatgccaaggaggtcatagctgtttcctgacgcactcgtctgagcgggctggcaagg"
  2889. self.assertEqual(regex.search(fz, seq, regex.BESTMATCH)[0],
  2890. "tCAGCCTCCCATTCAGAATATACATCC")
  2891. # Hg issue 83: slash handling in presence of a quantifier
  2892. self.assertEqual(regex.findall(r"c..+/c", "cA/c\ncAb/c"), ['cAb/c'])
  2893. # Hg issue 85: Non-conformance to Unicode UAX#29 re: ZWJ / ZWNJ
  2894. self.assertEqual(ascii(regex.sub(r"(\w+)", r"[\1]",
  2895. '\u0905\u0928\u094d\u200d\u0928 \u0d28\u0d4d\u200d \u0915\u093f\u0928',
  2896. regex.WORD)),
  2897. ascii('[\u0905\u0928\u094d\u200d\u0928] [\u0d28\u0d4d\u200d] [\u0915\u093f\u0928]'))
  2898. # Hg issue 88: regex.match() hangs
  2899. self.assertEqual(regex.match(r".*a.*ba.*aa", "ababba"), None)
  2900. # Hg issue 87: Allow duplicate names of groups
  2901. self.assertEqual(regex.match(r'(?<x>a(?<x>b))', "ab").spans("x"), [(1,
  2902. 2), (0, 2)])
  2903. # Hg issue 91: match.expand is extremely slow
  2904. # Check that the replacement cache works.
  2905. self.assertEqual(regex.sub(r'(-)', lambda m: m.expand(r'x'), 'a-b-c'),
  2906. 'axbxc')
  2907. # Hg issue 94: Python crashes when executing regex updates
  2908. # pattern.findall
  2909. rx = regex.compile(r'\bt(est){i<2}', flags=regex.V1)
  2910. self.assertEqual(rx.search("Some text"), None)
  2911. self.assertEqual(rx.findall("Some text"), [])
  2912. # Hg issue 95: 'pos' for regex.error
  2913. self.assertRaisesRegex(regex.error, self.MULTIPLE_REPEAT, lambda:
  2914. regex.compile(r'.???'))
  2915. # Hg issue 97: behaviour of regex.escape's special_only is wrong
  2916. #
  2917. # Hg issue 244: Make `special_only=True` the default in
  2918. # `regex.escape()`
  2919. self.assertEqual(regex.escape('foo!?', special_only=False), 'foo\\!\\?')
  2920. self.assertEqual(regex.escape('foo!?', special_only=True), 'foo!\\?')
  2921. self.assertEqual(regex.escape('foo!?'), 'foo!\\?')
  2922. self.assertEqual(regex.escape(b'foo!?', special_only=False), b'foo\\!\\?')
  2923. self.assertEqual(regex.escape(b'foo!?', special_only=True),
  2924. b'foo!\\?')
  2925. self.assertEqual(regex.escape(b'foo!?'), b'foo!\\?')
  2926. # Hg issue 100: strange results from regex.search
  2927. self.assertEqual(regex.search('^([^z]*(?:WWWi|W))?$',
  2928. 'WWWi').groups(), ('WWWi', ))
  2929. self.assertEqual(regex.search('^([^z]*(?:WWWi|w))?$',
  2930. 'WWWi').groups(), ('WWWi', ))
  2931. self.assertEqual(regex.search('^([^z]*?(?:WWWi|W))?$',
  2932. 'WWWi').groups(), ('WWWi', ))
  2933. # Hg issue 101: findall() broken (seems like memory corruption)
  2934. pat = regex.compile(r'xxx', flags=regex.FULLCASE | regex.UNICODE)
  2935. self.assertEqual([x.group() for x in pat.finditer('yxxx')], ['xxx'])
  2936. self.assertEqual(pat.findall('yxxx'), ['xxx'])
  2937. raw = 'yxxx'
  2938. self.assertEqual([x.group() for x in pat.finditer(raw)], ['xxx'])
  2939. self.assertEqual(pat.findall(raw), ['xxx'])
  2940. pat = regex.compile(r'xxx', flags=regex.FULLCASE | regex.IGNORECASE |
  2941. regex.UNICODE)
  2942. self.assertEqual([x.group() for x in pat.finditer('yxxx')], ['xxx'])
  2943. self.assertEqual(pat.findall('yxxx'), ['xxx'])
  2944. raw = 'yxxx'
  2945. self.assertEqual([x.group() for x in pat.finditer(raw)], ['xxx'])
  2946. self.assertEqual(pat.findall(raw), ['xxx'])
  2947. # Hg issue 106: * operator not working correctly with sub()
  2948. if sys.version_info >= (3, 7, 0):
  2949. self.assertEqual(regex.sub('(?V0).*', 'x', 'test'), 'xx')
  2950. else:
  2951. self.assertEqual(regex.sub('(?V0).*', 'x', 'test'), 'x')
  2952. self.assertEqual(regex.sub('(?V1).*', 'x', 'test'), 'xx')
  2953. if sys.version_info >= (3, 7, 0):
  2954. self.assertEqual(regex.sub('(?V0).*?', '|', 'test'), '|||||||||')
  2955. else:
  2956. self.assertEqual(regex.sub('(?V0).*?', '|', 'test'), '|t|e|s|t|')
  2957. self.assertEqual(regex.sub('(?V1).*?', '|', 'test'), '|||||||||')
  2958. # Hg issue 112: re: OK, but regex: SystemError
  2959. self.assertEqual(regex.sub(r'^(@)\n(?!.*?@)(.*)',
  2960. r'\1\n==========\n\2', '@\n', flags=regex.DOTALL), '@\n==========\n')
  2961. # Hg issue 109: Edit distance of fuzzy match
  2962. self.assertEqual(regex.match(r'(?:cats|cat){e<=1}',
  2963. 'caz').fuzzy_counts, (1, 0, 0))
  2964. self.assertEqual(regex.match(r'(?e)(?:cats|cat){e<=1}',
  2965. 'caz').fuzzy_counts, (1, 0, 0))
  2966. self.assertEqual(regex.match(r'(?b)(?:cats|cat){e<=1}',
  2967. 'caz').fuzzy_counts, (1, 0, 0))
  2968. self.assertEqual(regex.match(r'(?:cat){e<=1}', 'caz').fuzzy_counts,
  2969. (1, 0, 0))
  2970. self.assertEqual(regex.match(r'(?e)(?:cat){e<=1}',
  2971. 'caz').fuzzy_counts, (1, 0, 0))
  2972. self.assertEqual(regex.match(r'(?b)(?:cat){e<=1}',
  2973. 'caz').fuzzy_counts, (1, 0, 0))
  2974. self.assertEqual(regex.match(r'(?:cats){e<=2}', 'c ats').fuzzy_counts,
  2975. (1, 1, 0))
  2976. self.assertEqual(regex.match(r'(?e)(?:cats){e<=2}',
  2977. 'c ats').fuzzy_counts, (0, 1, 0))
  2978. self.assertEqual(regex.match(r'(?b)(?:cats){e<=2}',
  2979. 'c ats').fuzzy_counts, (0, 1, 0))
  2980. self.assertEqual(regex.match(r'(?:cats){e<=2}',
  2981. 'c a ts').fuzzy_counts, (0, 2, 0))
  2982. self.assertEqual(regex.match(r'(?e)(?:cats){e<=2}',
  2983. 'c a ts').fuzzy_counts, (0, 2, 0))
  2984. self.assertEqual(regex.match(r'(?b)(?:cats){e<=2}',
  2985. 'c a ts').fuzzy_counts, (0, 2, 0))
  2986. self.assertEqual(regex.match(r'(?:cats){e<=1}', 'c ats').fuzzy_counts,
  2987. (0, 1, 0))
  2988. self.assertEqual(regex.match(r'(?e)(?:cats){e<=1}',
  2989. 'c ats').fuzzy_counts, (0, 1, 0))
  2990. self.assertEqual(regex.match(r'(?b)(?:cats){e<=1}',
  2991. 'c ats').fuzzy_counts, (0, 1, 0))
  2992. # Hg issue 115: Infinite loop when processing backreferences
  2993. self.assertEqual(regex.findall(r'\bof ([a-z]+) of \1\b',
  2994. 'To make use of one of these modules'), [])
  2995. # Hg issue 125: Reference to entire match (\g&lt;0&gt;) in
  2996. # Pattern.sub() doesn't work as of 2014.09.22 release.
  2997. self.assertEqual(regex.sub(r'x', r'\g<0>', 'x'), 'x')
  2998. # Unreported issue: no such builtin as 'ascii' in Python 2.
  2999. self.assertEqual(bool(regex.match(r'a', 'a', regex.DEBUG)), True)
  3000. # Hg issue 131: nested sets behaviour
  3001. self.assertEqual(regex.findall(r'(?V1)[[b-e]--cd]', 'abcdef'), ['b',
  3002. 'e'])
  3003. self.assertEqual(regex.findall(r'(?V1)[b-e--cd]', 'abcdef'), ['b',
  3004. 'e'])
  3005. self.assertEqual(regex.findall(r'(?V1)[[bcde]--cd]', 'abcdef'), ['b',
  3006. 'e'])
  3007. self.assertEqual(regex.findall(r'(?V1)[bcde--cd]', 'abcdef'), ['b',
  3008. 'e'])
  3009. # Hg issue 132: index out of range on null property \p{}
  3010. self.assertRaisesRegex(regex.error, '^unknown property at position 4$',
  3011. lambda: regex.compile(r'\p{}'))
  3012. # Issue 23692.
  3013. self.assertEqual(regex.match('(?:()|(?(1)()|z)){2}(?(2)a|z)',
  3014. 'a').group(0, 1, 2), ('a', '', ''))
  3015. self.assertEqual(regex.match('(?:()|(?(1)()|z)){0,2}(?(2)a|z)',
  3016. 'a').group(0, 1, 2), ('a', '', ''))
  3017. # Hg issue 137: Posix character class :punct: does not seem to be
  3018. # supported.
  3019. # Posix compatibility as recommended here:
  3020. # http://www.unicode.org/reports/tr18/#Compatibility_Properties
  3021. # Posix in Unicode.
  3022. chars = ''.join(chr(c) for c in range(0x10000))
  3023. self.assertEqual(ascii(''.join(regex.findall(r'''[[:alnum:]]+''',
  3024. chars))), ascii(''.join(regex.findall(r'''[\p{Alpha}\p{PosixDigit}]+''',
  3025. chars))))
  3026. self.assertEqual(ascii(''.join(regex.findall(r'''[[:alpha:]]+''',
  3027. chars))), ascii(''.join(regex.findall(r'''\p{Alpha}+''',
  3028. chars))))
  3029. self.assertEqual(ascii(''.join(regex.findall(r'''[[:ascii:]]+''',
  3030. chars))), ascii(''.join(regex.findall(r'''[\p{InBasicLatin}]+''',
  3031. chars))))
  3032. self.assertEqual(ascii(''.join(regex.findall(r'''[[:blank:]]+''',
  3033. chars))), ascii(''.join(regex.findall(r'''[\p{gc=Space_Separator}\t]+''',
  3034. chars))))
  3035. self.assertEqual(ascii(''.join(regex.findall(r'''[[:cntrl:]]+''',
  3036. chars))), ascii(''.join(regex.findall(r'''\p{gc=Control}+''', chars))))
  3037. self.assertEqual(ascii(''.join(regex.findall(r'''[[:digit:]]+''',
  3038. chars))), ascii(''.join(regex.findall(r'''[0-9]+''', chars))))
  3039. self.assertEqual(ascii(''.join(regex.findall(r'''[[:graph:]]+''',
  3040. chars))), ascii(''.join(regex.findall(r'''[^\p{Space}\p{gc=Control}\p{gc=Surrogate}\p{gc=Unassigned}]+''',
  3041. chars))))
  3042. self.assertEqual(ascii(''.join(regex.findall(r'''[[:lower:]]+''',
  3043. chars))), ascii(''.join(regex.findall(r'''\p{Lower}+''',
  3044. chars))))
  3045. self.assertEqual(ascii(''.join(regex.findall(r'''[[:print:]]+''',
  3046. chars))), ascii(''.join(regex.findall(r'''(?V1)[\p{Graph}\p{Blank}--\p{Cntrl}]+''', chars))))
  3047. self.assertEqual(ascii(''.join(regex.findall(r'''[[:punct:]]+''',
  3048. chars))),
  3049. ascii(''.join(regex.findall(r'''(?V1)[\p{gc=Punctuation}\p{gc=Symbol}--\p{Alpha}]+''',
  3050. chars))))
  3051. self.assertEqual(ascii(''.join(regex.findall(r'''[[:space:]]+''',
  3052. chars))), ascii(''.join(regex.findall(r'''\p{Whitespace}+''',
  3053. chars))))
  3054. self.assertEqual(ascii(''.join(regex.findall(r'''[[:upper:]]+''',
  3055. chars))), ascii(''.join(regex.findall(r'''\p{Upper}+''',
  3056. chars))))
  3057. self.assertEqual(ascii(''.join(regex.findall(r'''[[:word:]]+''',
  3058. chars))), ascii(''.join(regex.findall(r'''[\p{Alpha}\p{gc=Mark}\p{Digit}\p{gc=Connector_Punctuation}\p{Join_Control}]+''',
  3059. chars))))
  3060. self.assertEqual(ascii(''.join(regex.findall(r'''[[:xdigit:]]+''',
  3061. chars))), ascii(''.join(regex.findall(r'''[0-9A-Fa-f]+''',
  3062. chars))))
  3063. # Posix in ASCII.
  3064. chars = bytes(range(0x100))
  3065. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:alnum:]]+''',
  3066. chars))), ascii(b''.join(regex.findall(br'''(?a)[\p{Alpha}\p{PosixDigit}]+''',
  3067. chars))))
  3068. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:alpha:]]+''',
  3069. chars))), ascii(b''.join(regex.findall(br'''(?a)\p{Alpha}+''', chars))))
  3070. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:ascii:]]+''',
  3071. chars))), ascii(b''.join(regex.findall(br'''(?a)[\x00-\x7F]+''', chars))))
  3072. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:blank:]]+''',
  3073. chars))), ascii(b''.join(regex.findall(br'''(?a)[\p{gc=Space_Separator}\t]+''',
  3074. chars))))
  3075. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:cntrl:]]+''',
  3076. chars))), ascii(b''.join(regex.findall(br'''(?a)\p{gc=Control}+''',
  3077. chars))))
  3078. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:digit:]]+''',
  3079. chars))), ascii(b''.join(regex.findall(br'''(?a)[0-9]+''', chars))))
  3080. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:graph:]]+''',
  3081. chars))), ascii(b''.join(regex.findall(br'''(?a)[^\p{Space}\p{gc=Control}\p{gc=Surrogate}\p{gc=Unassigned}]+''', chars))))
  3082. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:lower:]]+''',
  3083. chars))), ascii(b''.join(regex.findall(br'''(?a)\p{Lower}+''', chars))))
  3084. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:print:]]+''',
  3085. chars))), ascii(b''.join(regex.findall(br'''(?aV1)[\p{Graph}\p{Blank}--\p{Cntrl}]+''', chars))))
  3086. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:punct:]]+''',
  3087. chars))), ascii(b''.join(regex.findall(br'''(?aV1)[\p{gc=Punctuation}\p{gc=Symbol}--\p{Alpha}]+''',
  3088. chars))))
  3089. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:space:]]+''',
  3090. chars))), ascii(b''.join(regex.findall(br'''(?a)\p{Whitespace}+''', chars))))
  3091. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:upper:]]+''',
  3092. chars))), ascii(b''.join(regex.findall(br'''(?a)\p{Upper}+''', chars))))
  3093. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:word:]]+''',
  3094. chars))), ascii(b''.join(regex.findall(br'''(?a)[\p{Alpha}\p{gc=Mark}\p{Digit}\p{gc=Connector_Punctuation}\p{Join_Control}]+''', chars))))
  3095. self.assertEqual(ascii(b''.join(regex.findall(br'''(?a)[[:xdigit:]]+''',
  3096. chars))), ascii(b''.join(regex.findall(br'''(?a)[0-9A-Fa-f]+''', chars))))
  3097. # Hg issue 138: grapheme anchored search not working properly.
  3098. self.assertEqual(ascii(regex.search(r'\X$', 'ab\u2103').group()),
  3099. ascii('\u2103'))
  3100. # Hg issue 139: Regular expression with multiple wildcards where first
  3101. # should match empty string does not always work.
  3102. self.assertEqual(regex.search("([^L]*)([^R]*R)", "LtR").groups(), ('',
  3103. 'LtR'))
  3104. # Hg issue 140: Replace with REVERSE and groups has unexpected
  3105. # behavior.
  3106. self.assertEqual(regex.sub(r'(.)', r'x\1y', 'ab'), 'xayxby')
  3107. self.assertEqual(regex.sub(r'(?r)(.)', r'x\1y', 'ab'), 'xayxby')
  3108. self.assertEqual(regex.subf(r'(.)', 'x{1}y', 'ab'), 'xayxby')
  3109. self.assertEqual(regex.subf(r'(?r)(.)', 'x{1}y', 'ab'), 'xayxby')
  3110. # Hg issue 141: Crash on a certain partial match.
  3111. self.assertEqual(regex.fullmatch('(a)*abc', 'ab',
  3112. partial=True).span(), (0, 2))
  3113. self.assertEqual(regex.fullmatch('(a)*abc', 'ab',
  3114. partial=True).partial, True)
  3115. # Hg issue 143: Partial matches have incorrect span if prefix is '.'
  3116. # wildcard.
  3117. self.assertEqual(regex.search('OXRG', 'OOGOX', partial=True).span(),
  3118. (3, 5))
  3119. self.assertEqual(regex.search('.XRG', 'OOGOX', partial=True).span(),
  3120. (3, 5))
  3121. self.assertEqual(regex.search('.{1,3}XRG', 'OOGOX',
  3122. partial=True).span(), (1, 5))
  3123. # Hg issue 144: Latest version problem with matching 'R|R'.
  3124. self.assertEqual(regex.match('R|R', 'R').span(), (0, 1))
  3125. # Hg issue 146: Forced-fail (?!) works improperly in conditional.
  3126. self.assertEqual(regex.match(r'(.)(?(1)(?!))', 'xy'), None)
  3127. # Groups cleared after failure.
  3128. self.assertEqual(regex.findall(r'(y)?(\d)(?(1)\b\B)', 'ax1y2z3b'),
  3129. [('', '1'), ('', '2'), ('', '3')])
  3130. self.assertEqual(regex.findall(r'(y)?+(\d)(?(1)\b\B)', 'ax1y2z3b'),
  3131. [('', '1'), ('', '2'), ('', '3')])
  3132. # Hg issue 147: Fuzzy match can return match points beyond buffer end.
  3133. self.assertEqual([m.span() for m in regex.finditer(r'(?i)(?:error){e}',
  3134. 'regex failure')], [(0, 5), (5, 10), (10, 13), (13, 13)])
  3135. self.assertEqual([m.span() for m in
  3136. regex.finditer(r'(?fi)(?:error){e}', 'regex failure')], [(0, 5), (5,
  3137. 10), (10, 13), (13, 13)])
  3138. # Hg issue 150: Have an option for POSIX-compatible longest match of
  3139. # alternates.
  3140. self.assertEqual(regex.search(r'(?p)\d+(\w(\d*)?|[eE]([+-]\d+))',
  3141. '10b12')[0], '10b12')
  3142. self.assertEqual(regex.search(r'(?p)\d+(\w(\d*)?|[eE]([+-]\d+))',
  3143. '10E+12')[0], '10E+12')
  3144. self.assertEqual(regex.search(r'(?p)(\w|ae|oe|ue|ss)', 'ae')[0], 'ae')
  3145. self.assertEqual(regex.search(r'(?p)one(self)?(selfsufficient)?',
  3146. 'oneselfsufficient')[0], 'oneselfsufficient')
  3147. # Hg issue 151: Request: \K.
  3148. self.assertEqual(regex.search(r'(ab\Kcd)', 'abcd').group(0, 1), ('cd',
  3149. 'abcd'))
  3150. self.assertEqual(regex.findall(r'\w\w\K\w\w', 'abcdefgh'), ['cd',
  3151. 'gh'])
  3152. self.assertEqual(regex.findall(r'(\w\w\K\w\w)', 'abcdefgh'), ['abcd',
  3153. 'efgh'])
  3154. self.assertEqual(regex.search(r'(?r)(ab\Kcd)', 'abcd').group(0, 1),
  3155. ('ab', 'abcd'))
  3156. self.assertEqual(regex.findall(r'(?r)\w\w\K\w\w', 'abcdefgh'), ['ef',
  3157. 'ab'])
  3158. self.assertEqual(regex.findall(r'(?r)(\w\w\K\w\w)', 'abcdefgh'),
  3159. ['efgh', 'abcd'])
  3160. # Hg issue 152: Request: Request: (?(DEFINE)...).
  3161. self.assertEqual(regex.search(r'(?(DEFINE)(?<quant>\d+)(?<item>\w+))(?&quant) (?&item)',
  3162. '5 elephants')[0], '5 elephants')
  3163. # Hg issue 153: Request: (*SKIP).
  3164. self.assertEqual(regex.search(r'12(*FAIL)|3', '123')[0], '3')
  3165. self.assertEqual(regex.search(r'(?r)12(*FAIL)|3', '123')[0], '3')
  3166. self.assertEqual(regex.search(r'\d+(*PRUNE)\d', '123'), None)
  3167. self.assertEqual(regex.search(r'\d+(?=(*PRUNE))\d', '123')[0], '123')
  3168. self.assertEqual(regex.search(r'\d+(*PRUNE)bcd|[3d]', '123bcd')[0],
  3169. '123bcd')
  3170. self.assertEqual(regex.search(r'\d+(*PRUNE)bcd|[3d]', '123zzd')[0],
  3171. 'd')
  3172. self.assertEqual(regex.search(r'\d+?(*PRUNE)bcd|[3d]', '123bcd')[0],
  3173. '3bcd')
  3174. self.assertEqual(regex.search(r'\d+?(*PRUNE)bcd|[3d]', '123zzd')[0],
  3175. 'd')
  3176. self.assertEqual(regex.search(r'\d++(?<=3(*PRUNE))zzd|[4d]$',
  3177. '123zzd')[0], '123zzd')
  3178. self.assertEqual(regex.search(r'\d++(?<=3(*PRUNE))zzd|[4d]$',
  3179. '124zzd')[0], 'd')
  3180. self.assertEqual(regex.search(r'\d++(?<=(*PRUNE)3)zzd|[4d]$',
  3181. '124zzd')[0], 'd')
  3182. self.assertEqual(regex.search(r'\d++(?<=2(*PRUNE)3)zzd|[3d]$',
  3183. '124zzd')[0], 'd')
  3184. self.assertEqual(regex.search(r'(?r)\d(*PRUNE)\d+', '123'), None)
  3185. self.assertEqual(regex.search(r'(?r)\d(?<=(*PRUNE))\d+', '123')[0],
  3186. '123')
  3187. self.assertEqual(regex.search(r'(?r)\d+(*PRUNE)bcd|[3d]',
  3188. '123bcd')[0], '123bcd')
  3189. self.assertEqual(regex.search(r'(?r)\d+(*PRUNE)bcd|[3d]',
  3190. '123zzd')[0], 'd')
  3191. self.assertEqual(regex.search(r'(?r)\d++(?<=3(*PRUNE))zzd|[4d]$',
  3192. '123zzd')[0], '123zzd')
  3193. self.assertEqual(regex.search(r'(?r)\d++(?<=3(*PRUNE))zzd|[4d]$',
  3194. '124zzd')[0], 'd')
  3195. self.assertEqual(regex.search(r'(?r)\d++(?<=(*PRUNE)3)zzd|[4d]$',
  3196. '124zzd')[0], 'd')
  3197. self.assertEqual(regex.search(r'(?r)\d++(?<=2(*PRUNE)3)zzd|[3d]$',
  3198. '124zzd')[0], 'd')
  3199. self.assertEqual(regex.search(r'\d+(*SKIP)bcd|[3d]', '123bcd')[0],
  3200. '123bcd')
  3201. self.assertEqual(regex.search(r'\d+(*SKIP)bcd|[3d]', '123zzd')[0],
  3202. 'd')
  3203. self.assertEqual(regex.search(r'\d+?(*SKIP)bcd|[3d]', '123bcd')[0],
  3204. '3bcd')
  3205. self.assertEqual(regex.search(r'\d+?(*SKIP)bcd|[3d]', '123zzd')[0],
  3206. 'd')
  3207. self.assertEqual(regex.search(r'\d++(?<=3(*SKIP))zzd|[4d]$',
  3208. '123zzd')[0], '123zzd')
  3209. self.assertEqual(regex.search(r'\d++(?<=3(*SKIP))zzd|[4d]$',
  3210. '124zzd')[0], 'd')
  3211. self.assertEqual(regex.search(r'\d++(?<=(*SKIP)3)zzd|[4d]$',
  3212. '124zzd')[0], 'd')
  3213. self.assertEqual(regex.search(r'\d++(?<=2(*SKIP)3)zzd|[3d]$',
  3214. '124zzd')[0], 'd')
  3215. self.assertEqual(regex.search(r'(?r)\d+(*SKIP)bcd|[3d]', '123bcd')[0],
  3216. '123bcd')
  3217. self.assertEqual(regex.search(r'(?r)\d+(*SKIP)bcd|[3d]', '123zzd')[0],
  3218. 'd')
  3219. self.assertEqual(regex.search(r'(?r)\d++(?<=3(*SKIP))zzd|[4d]$',
  3220. '123zzd')[0], '123zzd')
  3221. self.assertEqual(regex.search(r'(?r)\d++(?<=3(*SKIP))zzd|[4d]$',
  3222. '124zzd')[0], 'd')
  3223. self.assertEqual(regex.search(r'(?r)\d++(?<=(*SKIP)3)zzd|[4d]$',
  3224. '124zzd')[0], 'd')
  3225. self.assertEqual(regex.search(r'(?r)\d++(?<=2(*SKIP)3)zzd|[3d]$',
  3226. '124zzd')[0], 'd')
  3227. # Hg issue 154: Segmentation fault 11 when working with an atomic group
  3228. text = """June 30, December 31, 2013 2012
  3229. some words follow:
  3230. more words and numbers 1,234,567 9,876,542
  3231. more words and numbers 1,234,567 9,876,542"""
  3232. self.assertEqual(len(regex.findall(r'(?<!\d)(?>2014|2013 ?2012)', text)), 1)
  3233. # Hg issue 156: regression on atomic grouping
  3234. self.assertEqual(regex.match('1(?>2)', '12').span(), (0, 2))
  3235. # Hg issue 157: regression: segfault on complex lookaround
  3236. self.assertEqual(regex.match(r'(?V1w)(?=(?=[^A-Z]*+[A-Z])(?=[^a-z]*+[a-z]))(?=\D*+\d)(?=\p{Alphanumeric}*+\P{Alphanumeric})\A(?s:.){8,255}+\Z',
  3237. 'AAaa11!!')[0], 'AAaa11!!')
  3238. # Hg issue 158: Group issue with (?(DEFINE)...)
  3239. TEST_REGEX = regex.compile(r'''(?smx)
  3240. (?(DEFINE)
  3241. (?<subcat>
  3242. ^,[^,]+,
  3243. )
  3244. )
  3245. # Group 2 is defined on this line
  3246. ^,([^,]+),
  3247. (?:(?!(?&subcat)[\r\n]+(?&subcat)).)+
  3248. ''')
  3249. TEST_DATA = '''
  3250. ,Cat 1,
  3251. ,Brand 1,
  3252. some
  3253. thing
  3254. ,Brand 2,
  3255. other
  3256. things
  3257. ,Cat 2,
  3258. ,Brand,
  3259. Some
  3260. thing
  3261. '''
  3262. self.assertEqual([m.span(1, 2) for m in
  3263. TEST_REGEX.finditer(TEST_DATA)], [((-1, -1), (2, 7)), ((-1, -1), (54,
  3264. 59))])
  3265. # Hg issue 161: Unexpected fuzzy match results
  3266. self.assertEqual(regex.search('(abcdefgh){e}',
  3267. '******abcdefghijklmnopqrtuvwxyz', regex.BESTMATCH).span(), (6, 14))
  3268. self.assertEqual(regex.search('(abcdefghi){e}',
  3269. '******abcdefghijklmnopqrtuvwxyz', regex.BESTMATCH).span(), (6, 15))
  3270. # Hg issue 163: allow lookarounds in conditionals.
  3271. self.assertEqual(regex.match(r'(?:(?=\d)\d+\b|\w+)', '123abc').span(),
  3272. (0, 6))
  3273. self.assertEqual(regex.match(r'(?(?=\d)\d+\b|\w+)', '123abc'), None)
  3274. self.assertEqual(regex.search(r'(?(?<=love\s)you|(?<=hate\s)her)',
  3275. "I love you").span(), (7, 10))
  3276. self.assertEqual(regex.findall(r'(?(?<=love\s)you|(?<=hate\s)her)',
  3277. "I love you but I don't hate her either"), ['you', 'her'])
  3278. # Hg issue 180: bug of POSIX matching.
  3279. self.assertEqual(regex.search(r'(?p)a*(.*?)', 'aaabbb').group(0, 1),
  3280. ('aaabbb', 'bbb'))
  3281. self.assertEqual(regex.search(r'(?p)a*(.*)', 'aaabbb').group(0, 1),
  3282. ('aaabbb', 'bbb'))
  3283. self.assertEqual(regex.sub(r'(?p)a*(.*?)', r'\1', 'aaabbb'), 'bbb')
  3284. self.assertEqual(regex.sub(r'(?p)a*(.*)', r'\1', 'aaabbb'), 'bbb')
  3285. # Hg issue 192: Named lists reverse matching doesn't work with
  3286. # IGNORECASE and V1
  3287. self.assertEqual(regex.match(r'(?irV0)\L<kw>', '21', kw=['1']).span(),
  3288. (1, 2))
  3289. self.assertEqual(regex.match(r'(?irV1)\L<kw>', '21', kw=['1']).span(),
  3290. (1, 2))
  3291. # Hg issue 193: Alternation and .REVERSE flag.
  3292. self.assertEqual(regex.search('a|b', '111a222').span(), (3, 4))
  3293. self.assertEqual(regex.search('(?r)a|b', '111a222').span(), (3, 4))
  3294. # Hg issue 194: .FULLCASE and Backreference
  3295. self.assertEqual(regex.search(r'(?if)<(CLI)><\1>',
  3296. '<cli><cli>').span(), (0, 10))
  3297. self.assertEqual(regex.search(r'(?if)<(CLI)><\1>',
  3298. '<cli><clI>').span(), (0, 10))
  3299. self.assertEqual(regex.search(r'(?ifr)<\1><(CLI)>',
  3300. '<cli><clI>').span(), (0, 10))
  3301. # Hg issue 195: Pickle (or otherwise serial) the compiled regex
  3302. r = regex.compile(r'\L<options>', options=['foo', 'bar'])
  3303. p = pickle.dumps(r)
  3304. r = pickle.loads(p)
  3305. self.assertEqual(r.match('foo').span(), (0, 3))
  3306. # Hg issue 196: Fuzzy matching on repeated regex not working as
  3307. # expected
  3308. self.assertEqual(regex.match('(x{6}){e<=1}', 'xxxxxx',
  3309. flags=regex.BESTMATCH).span(), (0, 6))
  3310. self.assertEqual(regex.match('(x{6}){e<=1}', 'xxxxx',
  3311. flags=regex.BESTMATCH).span(), (0, 5))
  3312. self.assertEqual(regex.match('(x{6}){e<=1}', 'x',
  3313. flags=regex.BESTMATCH), None)
  3314. self.assertEqual(regex.match('(?r)(x{6}){e<=1}', 'xxxxxx',
  3315. flags=regex.BESTMATCH).span(), (0, 6))
  3316. self.assertEqual(regex.match('(?r)(x{6}){e<=1}', 'xxxxx',
  3317. flags=regex.BESTMATCH).span(), (0, 5))
  3318. self.assertEqual(regex.match('(?r)(x{6}){e<=1}', 'x',
  3319. flags=regex.BESTMATCH), None)
  3320. # Hg issue 197: ValueError in regex.compile
  3321. self.assertRaises(regex.error, lambda:
  3322. regex.compile(b'00000\\0\\00\\^\50\\00\\U05000000'))
  3323. # Hg issue 198: ValueError in regex.compile
  3324. self.assertRaises(regex.error, lambda: regex.compile(b"{e<l"))
  3325. # Hg issue 199: Segfault in re.compile
  3326. self.assertEqual(bool(regex.compile('((?0)){e}')), True)
  3327. # Hg issue 200: AttributeError in regex.compile with latest regex
  3328. self.assertEqual(bool(regex.compile('\x00?(?0){e}')), True)
  3329. # Hg issue 201: ENHANCEMATCH crashes interpreter
  3330. self.assertEqual(regex.findall(r'((brown)|(lazy)){1<=e<=3} ((dog)|(fox)){1<=e<=3}',
  3331. 'The quick borwn fax jumped over the lzy hog', regex.ENHANCEMATCH),
  3332. [('borwn', 'borwn', '', 'fax', '', 'fax'), ('lzy', '', 'lzy', 'hog',
  3333. 'hog', '')])
  3334. # Hg issue 203: partial matching bug
  3335. self.assertEqual(regex.search(r'\d\d\d-\d\d-\d\d\d\d',
  3336. "My SSN is 999-89-76, but don't tell.", partial=True).span(), (36,
  3337. 36))
  3338. # Hg issue 204: confusion of (?aif) flags
  3339. upper_i = '\N{CYRILLIC CAPITAL LETTER SHORT I}'
  3340. lower_i = '\N{CYRILLIC SMALL LETTER SHORT I}'
  3341. self.assertEqual(bool(regex.match(r'(?ui)' + upper_i,
  3342. lower_i)), True)
  3343. self.assertEqual(bool(regex.match(r'(?ui)' + lower_i,
  3344. upper_i)), True)
  3345. self.assertEqual(bool(regex.match(r'(?ai)' + upper_i,
  3346. lower_i)), False)
  3347. self.assertEqual(bool(regex.match(r'(?ai)' + lower_i,
  3348. upper_i)), False)
  3349. self.assertEqual(bool(regex.match(r'(?afi)' + upper_i,
  3350. lower_i)), False)
  3351. self.assertEqual(bool(regex.match(r'(?afi)' + lower_i,
  3352. upper_i)), False)
  3353. # Hg issue 205: Named list and (?ri) flags
  3354. self.assertEqual(bool(regex.search(r'(?i)\L<aa>', '22', aa=['121',
  3355. '22'])), True)
  3356. self.assertEqual(bool(regex.search(r'(?ri)\L<aa>', '22', aa=['121',
  3357. '22'])), True)
  3358. self.assertEqual(bool(regex.search(r'(?fi)\L<aa>', '22', aa=['121',
  3359. '22'])), True)
  3360. self.assertEqual(bool(regex.search(r'(?fri)\L<aa>', '22', aa=['121',
  3361. '22'])), True)
  3362. # Hg issue 208: Named list, (?ri) flags, Backreference
  3363. self.assertEqual(regex.search(r'(?r)\1dog..(?<=(\L<aa>))$', 'ccdogcc',
  3364. aa=['bcb', 'cc']). span(), (0, 7))
  3365. self.assertEqual(regex.search(r'(?ir)\1dog..(?<=(\L<aa>))$',
  3366. 'ccdogcc', aa=['bcb', 'cc']). span(), (0, 7))
  3367. # Hg issue 210: Fuzzy matching and Backreference
  3368. self.assertEqual(regex.search(r'(2)(?:\1{5}){e<=1}',
  3369. '3222212').span(), (1, 7))
  3370. self.assertEqual(regex.search(r'(\d)(?:\1{5}){e<=1}',
  3371. '3222212').span(), (1, 7))
  3372. # Hg issue 211: Segmentation fault with recursive matches and atomic
  3373. # groups
  3374. self.assertEqual(regex.match(r'''\A(?P<whole>(?>\((?&whole)\)|[+\-]))\Z''',
  3375. '((-))').span(), (0, 5))
  3376. self.assertEqual(regex.match(r'''\A(?P<whole>(?>\((?&whole)\)|[+\-]))\Z''',
  3377. '((-)+)'), None)
  3378. # Hg issue 212: Unexpected matching difference with .*? between re and
  3379. # regex
  3380. self.assertEqual(regex.match(r"x.*? (.).*\1(.*)\1",
  3381. 'x |y| z|').span(), (0, 9))
  3382. self.assertEqual(regex.match(r"\.sr (.*?) (.)(.*)\2(.*)\2(.*)",
  3383. r'.sr h |<nw>|<span class="locked">|').span(), (0, 35))
  3384. # Hg issue 213: Segmentation Fault
  3385. a = '"\\xF9\\x80\\xAEqdz\\x95L\\xA7\\x89[\\xFE \\x91)\\xF9]\\xDB\'\\x99\\x09=\\x00\\xFD\\x98\\x22\\xDD\\xF1\\xB6\\xC3 Z\\xB6gv\\xA5x\\x93P\\xE1r\\x14\\x8Cv\\x0C\\xC0w\\x15r\\xFFc%" '
  3386. py_regex_pattern = r'''(?P<http_referer>((?>(?<!\\)(?>"(?>\\.|[^\\"]+)+"|""|(?>'(?>\\.|[^\\']+)+')|''|(?>`(?>\\.|[^\\`]+)+`)|``)))) (?P<useragent>((?>(?<!\\)(?>"(?>\\.|[^\\"]+)+"|""|(?>'(?>\\.|[^\\']+)+')|''|(?>`(?>\\.|[^\\`]+)+`)|``))))'''
  3387. self.assertEqual(bool(regex.search(py_regex_pattern, a)), False)
  3388. # Hg Issue 216: Invalid match when using negative lookbehind and pipe
  3389. self.assertEqual(bool(regex.match('foo(?<=foo)', 'foo')), True)
  3390. self.assertEqual(bool(regex.match('foo(?<!foo)', 'foo')), False)
  3391. self.assertEqual(bool(regex.match('foo(?<=foo|x)', 'foo')), True)
  3392. self.assertEqual(bool(regex.match('foo(?<!foo|x)', 'foo')), False)
  3393. # Hg issue 217: Core dump in conditional ahead match and matching \!
  3394. # character
  3395. self.assertEqual(bool(regex.match(r'(?(?=.*\!.*)(?P<true>.*\!\w*\:.*)|(?P<false>.*))',
  3396. '!')), False)
  3397. # Hg issue 220: Misbehavior of group capture with OR operand
  3398. self.assertEqual(regex.match(r'\w*(ea)\w*|\w*e(?!a)\w*',
  3399. 'easier').groups(), ('ea', ))
  3400. # Hg issue 225: BESTMATCH in fuzzy match not working
  3401. self.assertEqual(regex.search('(^1234$){i,d}', '12234',
  3402. regex.BESTMATCH).span(), (0, 5))
  3403. self.assertEqual(regex.search('(^1234$){i,d}', '12234',
  3404. regex.BESTMATCH).fuzzy_counts, (0, 1, 0))
  3405. self.assertEqual(regex.search('(^1234$){s,i,d}', '12234',
  3406. regex.BESTMATCH).span(), (0, 5))
  3407. self.assertEqual(regex.search('(^1234$){s,i,d}', '12234',
  3408. regex.BESTMATCH).fuzzy_counts, (0, 1, 0))
  3409. # Hg issue 226: Error matching at start of string
  3410. self.assertEqual(regex.search('(^123$){s,i,d}', 'xxxxxxxx123',
  3411. regex.BESTMATCH).span(), (0, 11))
  3412. self.assertEqual(regex.search('(^123$){s,i,d}', 'xxxxxxxx123',
  3413. regex.BESTMATCH).fuzzy_counts, (0, 8, 0))
  3414. # Hg issue 227: Incorrect behavior for ? operator with UNICODE +
  3415. # IGNORECASE
  3416. self.assertEqual(regex.search(r'a?yz', 'xxxxyz', flags=regex.FULLCASE |
  3417. regex.IGNORECASE).span(), (4, 6))
  3418. # Hg issue 230: Is it a bug of (?(DEFINE)...)
  3419. self.assertEqual(regex.findall(r'(?:(?![a-d]).)+', 'abcdefgh'),
  3420. ['efgh'])
  3421. self.assertEqual(regex.findall(r'''(?(DEFINE)(?P<mydef>(?:(?![a-d]).)))(?&mydef)+''',
  3422. 'abcdefgh'), ['efgh'])
  3423. # Hg issue 238: Not fully re backward compatible
  3424. self.assertEqual(regex.findall(r'((\w{1,3})(\.{2,10})){1,3}',
  3425. '"Erm....yes. T..T...Thank you for that."'), [('Erm....', 'Erm',
  3426. '....'), ('T...', 'T', '...')])
  3427. self.assertEqual(regex.findall(r'((\w{1,3})(\.{2,10})){3}',
  3428. '"Erm....yes. T..T...Thank you for that."'), [])
  3429. self.assertEqual(regex.findall(r'((\w{1,3})(\.{2,10})){2}',
  3430. '"Erm....yes. T..T...Thank you for that."'), [('T...', 'T', '...')])
  3431. self.assertEqual(regex.findall(r'((\w{1,3})(\.{2,10})){1}',
  3432. '"Erm....yes. T..T...Thank you for that."'), [('Erm....', 'Erm',
  3433. '....'), ('T..', 'T', '..'), ('T...', 'T', '...')])
  3434. # Hg issue 247: Unexpected result with fuzzy matching and lookahead
  3435. # expression
  3436. self.assertEqual(regex.search(r'(?:ESTONIA(?!\w)){e<=1}',
  3437. 'ESTONIAN WORKERS').group(), 'ESTONIAN')
  3438. self.assertEqual(regex.search(r'(?:ESTONIA(?=\W)){e<=1}',
  3439. 'ESTONIAN WORKERS').group(), 'ESTONIAN')
  3440. self.assertEqual(regex.search(r'(?:(?<!\w)ESTONIA){e<=1}',
  3441. 'BLUB NESTONIA').group(), 'NESTONIA')
  3442. self.assertEqual(regex.search(r'(?:(?<=\W)ESTONIA){e<=1}',
  3443. 'BLUB NESTONIA').group(), 'NESTONIA')
  3444. self.assertEqual(regex.search(r'(?r)(?:ESTONIA(?!\w)){e<=1}',
  3445. 'ESTONIAN WORKERS').group(), 'ESTONIAN')
  3446. self.assertEqual(regex.search(r'(?r)(?:ESTONIA(?=\W)){e<=1}',
  3447. 'ESTONIAN WORKERS').group(), 'ESTONIAN')
  3448. self.assertEqual(regex.search(r'(?r)(?:(?<!\w)ESTONIA){e<=1}',
  3449. 'BLUB NESTONIA').group(), 'NESTONIA')
  3450. self.assertEqual(regex.search(r'(?r)(?:(?<=\W)ESTONIA){e<=1}',
  3451. 'BLUB NESTONIA').group(), 'NESTONIA')
  3452. # Hg issue 248: Unexpected result with fuzzy matching and more than one
  3453. # non-greedy quantifier
  3454. self.assertEqual(regex.search(r'(?:A.*B.*CDE){e<=2}',
  3455. 'A B CYZ').group(), 'A B CYZ')
  3456. self.assertEqual(regex.search(r'(?:A.*B.*?CDE){e<=2}',
  3457. 'A B CYZ').group(), 'A B CYZ')
  3458. self.assertEqual(regex.search(r'(?:A.*?B.*CDE){e<=2}',
  3459. 'A B CYZ').group(), 'A B CYZ')
  3460. self.assertEqual(regex.search(r'(?:A.*?B.*?CDE){e<=2}',
  3461. 'A B CYZ').group(), 'A B CYZ')
  3462. # Hg issue 249: Add an option to regex.escape() to not escape spaces
  3463. self.assertEqual(regex.escape(' ,0A[', special_only=False, literal_spaces=False), '\\ \\,0A\\[')
  3464. self.assertEqual(regex.escape(' ,0A[', special_only=False, literal_spaces=True), ' \\,0A\\[')
  3465. self.assertEqual(regex.escape(' ,0A[', special_only=True, literal_spaces=False), '\\ ,0A\\[')
  3466. self.assertEqual(regex.escape(' ,0A[', special_only=True, literal_spaces=True), ' ,0A\\[')
  3467. self.assertEqual(regex.escape(' ,0A['), '\\ ,0A\\[')
  3468. # Hg issue 251: Segfault with a particular expression
  3469. self.assertEqual(regex.search(r'(?(?=A)A|B)', 'A').span(), (0, 1))
  3470. self.assertEqual(regex.search(r'(?(?=A)A|B)', 'B').span(), (0, 1))
  3471. self.assertEqual(regex.search(r'(?(?=A)A|)', 'B').span(), (0, 0))
  3472. self.assertEqual(regex.search(r'(?(?=X)X|)', '').span(), (0, 0))
  3473. self.assertEqual(regex.search(r'(?(?=X))', '').span(), (0, 0))
  3474. # Hg issue 252: Empty capture strings when using DEFINE group reference
  3475. # within look-behind expression
  3476. self.assertEqual(regex.search(r'(?(DEFINE)(?<func>.))(?&func)',
  3477. 'abc').groups(), (None, ))
  3478. self.assertEqual(regex.search(r'(?(DEFINE)(?<func>.))(?&func)',
  3479. 'abc').groupdict(), {'func': None})
  3480. self.assertEqual(regex.search(r'(?(DEFINE)(?<func>.))(?&func)',
  3481. 'abc').capturesdict(), {'func': ['a']})
  3482. self.assertEqual(regex.search(r'(?(DEFINE)(?<func>.))(?=(?&func))',
  3483. 'abc').groups(), (None, ))
  3484. self.assertEqual(regex.search(r'(?(DEFINE)(?<func>.))(?=(?&func))',
  3485. 'abc').groupdict(), {'func': None})
  3486. self.assertEqual(regex.search(r'(?(DEFINE)(?<func>.))(?=(?&func))',
  3487. 'abc').capturesdict(), {'func': ['a']})
  3488. self.assertEqual(regex.search(r'(?(DEFINE)(?<func>.)).(?<=(?&func))',
  3489. 'abc').groups(), (None, ))
  3490. self.assertEqual(regex.search(r'(?(DEFINE)(?<func>.)).(?<=(?&func))',
  3491. 'abc').groupdict(), {'func': None})
  3492. self.assertEqual(regex.search(r'(?(DEFINE)(?<func>.)).(?<=(?&func))',
  3493. 'abc').capturesdict(), {'func': ['a']})
  3494. def test_subscripted_captures(self):
  3495. self.assertEqual(regex.match(r'(?P<x>.)+',
  3496. 'abc').expandf('{0} {0[0]} {0[-1]}'), 'abc abc abc')
  3497. self.assertEqual(regex.match(r'(?P<x>.)+',
  3498. 'abc').expandf('{1} {1[0]} {1[1]} {1[2]} {1[-1]} {1[-2]} {1[-3]}'),
  3499. 'c a b c c b a')
  3500. self.assertEqual(regex.match(r'(?P<x>.)+',
  3501. 'abc').expandf('{x} {x[0]} {x[1]} {x[2]} {x[-1]} {x[-2]} {x[-3]}'),
  3502. 'c a b c c b a')
  3503. self.assertEqual(regex.subf(r'(?P<x>.)+', r'{0} {0[0]} {0[-1]}',
  3504. 'abc'), 'abc abc abc')
  3505. self.assertEqual(regex.subf(r'(?P<x>.)+',
  3506. '{1} {1[0]} {1[1]} {1[2]} {1[-1]} {1[-2]} {1[-3]}', 'abc'),
  3507. 'c a b c c b a')
  3508. self.assertEqual(regex.subf(r'(?P<x>.)+',
  3509. '{x} {x[0]} {x[1]} {x[2]} {x[-1]} {x[-2]} {x[-3]}', 'abc'),
  3510. 'c a b c c b a')
  3511. def test_more_zerowidth(self):
  3512. if sys.version_info >= (3, 7, 0):
  3513. self.assertEqual(regex.split(r'\b|:+', 'a::bc'), ['', 'a', '', '',
  3514. 'bc', ''])
  3515. self.assertEqual(regex.sub(r'\b|:+', '-', 'a::bc'), '-a---bc-')
  3516. self.assertEqual(regex.findall(r'\b|:+', 'a::bc'), ['', '', '::',
  3517. '', ''])
  3518. self.assertEqual([m.span() for m in regex.finditer(r'\b|:+',
  3519. 'a::bc')], [(0, 0), (1, 1), (1, 3), (3, 3), (5, 5)])
  3520. self.assertEqual([m.span() for m in regex.finditer(r'(?m)^\s*?$',
  3521. 'foo\n\n\nbar')], [(4, 4), (4, 5), (5, 5)])
  3522. if sys.version_info < (3, 2, 0):
  3523. # In Python 3.1 it's called assertRaisesRegexp.
  3524. RegexTests.assertRaisesRegex = RegexTests.assertRaisesRegexp
  3525. def test_main():
  3526. run_unittest(RegexTests)
  3527. if __name__ == "__main__":
  3528. test_main()