test_re.py 130 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834
  1. from test.support import (gc_collect, bigmemtest, _2G,
  2. cpython_only, captured_stdout,
  3. check_disallow_instantiation, is_emscripten, is_wasi,
  4. SHORT_TIMEOUT)
  5. import locale
  6. import re
  7. import string
  8. import sys
  9. import time
  10. import unittest
  11. import warnings
  12. from re import Scanner
  13. from weakref import proxy
  14. # some platforms lack working multiprocessing
  15. try:
  16. import _multiprocessing
  17. except ImportError:
  18. multiprocessing = None
  19. else:
  20. import multiprocessing
  21. # Misc tests from Tim Peters' re.doc
  22. # WARNING: Don't change details in these tests if you don't know
  23. # what you're doing. Some of these tests were carefully modeled to
  24. # cover most of the code.
  25. class S(str):
  26. def __getitem__(self, index):
  27. return S(super().__getitem__(index))
  28. class B(bytes):
  29. def __getitem__(self, index):
  30. return B(super().__getitem__(index))
  31. class ReTests(unittest.TestCase):
  32. def assertTypedEqual(self, actual, expect, msg=None):
  33. self.assertEqual(actual, expect, msg)
  34. def recurse(actual, expect):
  35. if isinstance(expect, (tuple, list)):
  36. for x, y in zip(actual, expect):
  37. recurse(x, y)
  38. else:
  39. self.assertIs(type(actual), type(expect), msg)
  40. recurse(actual, expect)
  41. def checkPatternError(self, pattern, errmsg, pos=None):
  42. with self.assertRaises(re.error) as cm:
  43. re.compile(pattern)
  44. with self.subTest(pattern=pattern):
  45. err = cm.exception
  46. self.assertEqual(err.msg, errmsg)
  47. if pos is not None:
  48. self.assertEqual(err.pos, pos)
  49. def checkTemplateError(self, pattern, repl, string, errmsg, pos=None):
  50. with self.assertRaises(re.error) as cm:
  51. re.sub(pattern, repl, string)
  52. with self.subTest(pattern=pattern, repl=repl):
  53. err = cm.exception
  54. self.assertEqual(err.msg, errmsg)
  55. if pos is not None:
  56. self.assertEqual(err.pos, pos)
  57. def test_keep_buffer(self):
  58. # See bug 14212
  59. b = bytearray(b'x')
  60. it = re.finditer(b'a', b)
  61. with self.assertRaises(BufferError):
  62. b.extend(b'x'*400)
  63. list(it)
  64. del it
  65. gc_collect()
  66. b.extend(b'x'*400)
  67. def test_weakref(self):
  68. s = 'QabbbcR'
  69. x = re.compile('ab+c')
  70. y = proxy(x)
  71. self.assertEqual(x.findall('QabbbcR'), y.findall('QabbbcR'))
  72. def test_search_star_plus(self):
  73. self.assertEqual(re.search('x*', 'axx').span(0), (0, 0))
  74. self.assertEqual(re.search('x*', 'axx').span(), (0, 0))
  75. self.assertEqual(re.search('x+', 'axx').span(0), (1, 3))
  76. self.assertEqual(re.search('x+', 'axx').span(), (1, 3))
  77. self.assertIsNone(re.search('x', 'aaa'))
  78. self.assertEqual(re.match('a*', 'xxx').span(0), (0, 0))
  79. self.assertEqual(re.match('a*', 'xxx').span(), (0, 0))
  80. self.assertEqual(re.match('x*', 'xxxa').span(0), (0, 3))
  81. self.assertEqual(re.match('x*', 'xxxa').span(), (0, 3))
  82. self.assertIsNone(re.match('a+', 'xxx'))
  83. def test_branching(self):
  84. """Test Branching
  85. Test expressions using the OR ('|') operator."""
  86. self.assertEqual(re.match('(ab|ba)', 'ab').span(), (0, 2))
  87. self.assertEqual(re.match('(ab|ba)', 'ba').span(), (0, 2))
  88. self.assertEqual(re.match('(abc|bac|ca|cb)', 'abc').span(),
  89. (0, 3))
  90. self.assertEqual(re.match('(abc|bac|ca|cb)', 'bac').span(),
  91. (0, 3))
  92. self.assertEqual(re.match('(abc|bac|ca|cb)', 'ca').span(),
  93. (0, 2))
  94. self.assertEqual(re.match('(abc|bac|ca|cb)', 'cb').span(),
  95. (0, 2))
  96. self.assertEqual(re.match('((a)|(b)|(c))', 'a').span(), (0, 1))
  97. self.assertEqual(re.match('((a)|(b)|(c))', 'b').span(), (0, 1))
  98. self.assertEqual(re.match('((a)|(b)|(c))', 'c').span(), (0, 1))
  99. def bump_num(self, matchobj):
  100. int_value = int(matchobj.group(0))
  101. return str(int_value + 1)
  102. def test_basic_re_sub(self):
  103. self.assertTypedEqual(re.sub('y', 'a', 'xyz'), 'xaz')
  104. self.assertTypedEqual(re.sub('y', S('a'), S('xyz')), 'xaz')
  105. self.assertTypedEqual(re.sub(b'y', b'a', b'xyz'), b'xaz')
  106. self.assertTypedEqual(re.sub(b'y', B(b'a'), B(b'xyz')), b'xaz')
  107. self.assertTypedEqual(re.sub(b'y', bytearray(b'a'), bytearray(b'xyz')), b'xaz')
  108. self.assertTypedEqual(re.sub(b'y', memoryview(b'a'), memoryview(b'xyz')), b'xaz')
  109. for y in ("\xe0", "\u0430", "\U0001d49c"):
  110. self.assertEqual(re.sub(y, 'a', 'x%sz' % y), 'xaz')
  111. self.assertEqual(re.sub("(?i)b+", "x", "bbbb BBBB"), 'x x')
  112. self.assertEqual(re.sub(r'\d+', self.bump_num, '08.2 -2 23x99y'),
  113. '9.3 -3 24x100y')
  114. self.assertEqual(re.sub(r'\d+', self.bump_num, '08.2 -2 23x99y', 3),
  115. '9.3 -3 23x99y')
  116. self.assertEqual(re.sub(r'\d+', self.bump_num, '08.2 -2 23x99y', count=3),
  117. '9.3 -3 23x99y')
  118. self.assertEqual(re.sub('.', lambda m: r"\n", 'x'), '\\n')
  119. self.assertEqual(re.sub('.', r"\n", 'x'), '\n')
  120. s = r"\1\1"
  121. self.assertEqual(re.sub('(.)', s, 'x'), 'xx')
  122. self.assertEqual(re.sub('(.)', s.replace('\\', r'\\'), 'x'), s)
  123. self.assertEqual(re.sub('(.)', lambda m: s, 'x'), s)
  124. self.assertEqual(re.sub('(?P<a>x)', r'\g<a>\g<a>', 'xx'), 'xxxx')
  125. self.assertEqual(re.sub('(?P<a>x)', r'\g<a>\g<1>', 'xx'), 'xxxx')
  126. self.assertEqual(re.sub('(?P<unk>x)', r'\g<unk>\g<unk>', 'xx'), 'xxxx')
  127. self.assertEqual(re.sub('(?P<unk>x)', r'\g<1>\g<1>', 'xx'), 'xxxx')
  128. self.assertEqual(re.sub('()x', r'\g<0>\g<0>', 'xx'), 'xxxx')
  129. self.assertEqual(re.sub('a', r'\t\n\v\r\f\a\b', 'a'), '\t\n\v\r\f\a\b')
  130. self.assertEqual(re.sub('a', '\t\n\v\r\f\a\b', 'a'), '\t\n\v\r\f\a\b')
  131. self.assertEqual(re.sub('a', '\t\n\v\r\f\a\b', 'a'),
  132. (chr(9)+chr(10)+chr(11)+chr(13)+chr(12)+chr(7)+chr(8)))
  133. for c in 'cdehijklmopqsuwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ':
  134. with self.subTest(c):
  135. with self.assertRaises(re.error):
  136. self.assertEqual(re.sub('a', '\\' + c, 'a'), '\\' + c)
  137. self.assertEqual(re.sub(r'^\s*', 'X', 'test'), 'Xtest')
  138. def test_bug_449964(self):
  139. # fails for group followed by other escape
  140. self.assertEqual(re.sub(r'(?P<unk>x)', r'\g<1>\g<1>\b', 'xx'),
  141. 'xx\bxx\b')
  142. def test_bug_449000(self):
  143. # Test for sub() on escaped characters
  144. self.assertEqual(re.sub(r'\r\n', r'\n', 'abc\r\ndef\r\n'),
  145. 'abc\ndef\n')
  146. self.assertEqual(re.sub('\r\n', r'\n', 'abc\r\ndef\r\n'),
  147. 'abc\ndef\n')
  148. self.assertEqual(re.sub(r'\r\n', '\n', 'abc\r\ndef\r\n'),
  149. 'abc\ndef\n')
  150. self.assertEqual(re.sub('\r\n', '\n', 'abc\r\ndef\r\n'),
  151. 'abc\ndef\n')
  152. def test_bug_1661(self):
  153. # Verify that flags do not get silently ignored with compiled patterns
  154. pattern = re.compile('.')
  155. self.assertRaises(ValueError, re.match, pattern, 'A', re.I)
  156. self.assertRaises(ValueError, re.search, pattern, 'A', re.I)
  157. self.assertRaises(ValueError, re.findall, pattern, 'A', re.I)
  158. self.assertRaises(ValueError, re.compile, pattern, re.I)
  159. def test_bug_3629(self):
  160. # A regex that triggered a bug in the sre-code validator
  161. re.compile("(?P<quote>)(?(quote))")
  162. def test_sub_template_numeric_escape(self):
  163. # bug 776311 and friends
  164. self.assertEqual(re.sub('x', r'\0', 'x'), '\0')
  165. self.assertEqual(re.sub('x', r'\000', 'x'), '\000')
  166. self.assertEqual(re.sub('x', r'\001', 'x'), '\001')
  167. self.assertEqual(re.sub('x', r'\008', 'x'), '\0' + '8')
  168. self.assertEqual(re.sub('x', r'\009', 'x'), '\0' + '9')
  169. self.assertEqual(re.sub('x', r'\111', 'x'), '\111')
  170. self.assertEqual(re.sub('x', r'\117', 'x'), '\117')
  171. self.assertEqual(re.sub('x', r'\377', 'x'), '\377')
  172. self.assertEqual(re.sub('x', r'\1111', 'x'), '\1111')
  173. self.assertEqual(re.sub('x', r'\1111', 'x'), '\111' + '1')
  174. self.assertEqual(re.sub('x', r'\00', 'x'), '\x00')
  175. self.assertEqual(re.sub('x', r'\07', 'x'), '\x07')
  176. self.assertEqual(re.sub('x', r'\08', 'x'), '\0' + '8')
  177. self.assertEqual(re.sub('x', r'\09', 'x'), '\0' + '9')
  178. self.assertEqual(re.sub('x', r'\0a', 'x'), '\0' + 'a')
  179. self.checkTemplateError('x', r'\400', 'x',
  180. r'octal escape value \400 outside of '
  181. r'range 0-0o377', 0)
  182. self.checkTemplateError('x', r'\777', 'x',
  183. r'octal escape value \777 outside of '
  184. r'range 0-0o377', 0)
  185. self.checkTemplateError('x', r'\1', 'x', 'invalid group reference 1', 1)
  186. self.checkTemplateError('x', r'\8', 'x', 'invalid group reference 8', 1)
  187. self.checkTemplateError('x', r'\9', 'x', 'invalid group reference 9', 1)
  188. self.checkTemplateError('x', r'\11', 'x', 'invalid group reference 11', 1)
  189. self.checkTemplateError('x', r'\18', 'x', 'invalid group reference 18', 1)
  190. self.checkTemplateError('x', r'\1a', 'x', 'invalid group reference 1', 1)
  191. self.checkTemplateError('x', r'\90', 'x', 'invalid group reference 90', 1)
  192. self.checkTemplateError('x', r'\99', 'x', 'invalid group reference 99', 1)
  193. self.checkTemplateError('x', r'\118', 'x', 'invalid group reference 11', 1)
  194. self.checkTemplateError('x', r'\11a', 'x', 'invalid group reference 11', 1)
  195. self.checkTemplateError('x', r'\181', 'x', 'invalid group reference 18', 1)
  196. self.checkTemplateError('x', r'\800', 'x', 'invalid group reference 80', 1)
  197. self.checkTemplateError('x', r'\8', '', 'invalid group reference 8', 1)
  198. # in python2.3 (etc), these loop endlessly in sre_parser.py
  199. self.assertEqual(re.sub('(((((((((((x)))))))))))', r'\11', 'x'), 'x')
  200. self.assertEqual(re.sub('((((((((((y))))))))))(.)', r'\118', 'xyz'),
  201. 'xz8')
  202. self.assertEqual(re.sub('((((((((((y))))))))))(.)', r'\11a', 'xyz'),
  203. 'xza')
  204. def test_qualified_re_sub(self):
  205. self.assertEqual(re.sub('a', 'b', 'aaaaa'), 'bbbbb')
  206. self.assertEqual(re.sub('a', 'b', 'aaaaa', 1), 'baaaa')
  207. self.assertEqual(re.sub('a', 'b', 'aaaaa', count=1), 'baaaa')
  208. def test_bug_114660(self):
  209. self.assertEqual(re.sub(r'(\S)\s+(\S)', r'\1 \2', 'hello there'),
  210. 'hello there')
  211. def test_symbolic_groups(self):
  212. re.compile(r'(?P<a>x)(?P=a)(?(a)y)')
  213. re.compile(r'(?P<a1>x)(?P=a1)(?(a1)y)')
  214. re.compile(r'(?P<a1>x)\1(?(1)y)')
  215. re.compile(b'(?P<a1>x)(?P=a1)(?(a1)y)')
  216. # New valid identifiers in Python 3
  217. re.compile('(?P<µ>x)(?P=µ)(?(µ)y)')
  218. re.compile('(?P<𝔘𝔫𝔦𝔠𝔬𝔡𝔢>x)(?P=𝔘𝔫𝔦𝔠𝔬𝔡𝔢)(?(𝔘𝔫𝔦𝔠𝔬𝔡𝔢)y)')
  219. # Support > 100 groups.
  220. pat = '|'.join('x(?P<a%d>%x)y' % (i, i) for i in range(1, 200 + 1))
  221. pat = '(?:%s)(?(200)z|t)' % pat
  222. self.assertEqual(re.match(pat, 'xc8yz').span(), (0, 5))
  223. def test_symbolic_groups_errors(self):
  224. self.checkPatternError(r'(?P<a>)(?P<a>)',
  225. "redefinition of group name 'a' as group 2; "
  226. "was group 1")
  227. self.checkPatternError(r'(?P<a>(?P=a))',
  228. "cannot refer to an open group", 10)
  229. self.checkPatternError(r'(?Pxy)', 'unknown extension ?Px')
  230. self.checkPatternError(r'(?P<a>)(?P=a', 'missing ), unterminated name', 11)
  231. self.checkPatternError(r'(?P=', 'missing group name', 4)
  232. self.checkPatternError(r'(?P=)', 'missing group name', 4)
  233. self.checkPatternError(r'(?P=1)', "bad character in group name '1'", 4)
  234. self.checkPatternError(r'(?P=a)', "unknown group name 'a'")
  235. self.checkPatternError(r'(?P=a1)', "unknown group name 'a1'")
  236. self.checkPatternError(r'(?P=a.)', "bad character in group name 'a.'", 4)
  237. self.checkPatternError(r'(?P<)', 'missing >, unterminated name', 4)
  238. self.checkPatternError(r'(?P<a', 'missing >, unterminated name', 4)
  239. self.checkPatternError(r'(?P<', 'missing group name', 4)
  240. self.checkPatternError(r'(?P<>)', 'missing group name', 4)
  241. self.checkPatternError(r'(?P<1>)', "bad character in group name '1'", 4)
  242. self.checkPatternError(r'(?P<a.>)', "bad character in group name 'a.'", 4)
  243. self.checkPatternError(r'(?(', 'missing group name', 3)
  244. self.checkPatternError(r'(?())', 'missing group name', 3)
  245. self.checkPatternError(r'(?(a))', "unknown group name 'a'", 3)
  246. self.checkPatternError(r'(?(-1))', "bad character in group name '-1'", 3)
  247. self.checkPatternError(r'(?(1a))', "bad character in group name '1a'", 3)
  248. self.checkPatternError(r'(?(a.))', "bad character in group name 'a.'", 3)
  249. self.checkPatternError('(?P<©>x)', "bad character in group name '©'", 4)
  250. self.checkPatternError('(?P=©)', "bad character in group name '©'", 4)
  251. self.checkPatternError('(?(©)y)', "bad character in group name '©'", 3)
  252. with self.assertWarnsRegex(DeprecationWarning,
  253. r"bad character in group name '\\xc2\\xb5' "
  254. r"at position 4") as w:
  255. re.compile(b'(?P<\xc2\xb5>x)')
  256. self.assertEqual(w.filename, __file__)
  257. with self.assertWarnsRegex(DeprecationWarning,
  258. r"bad character in group name '\\xc2\\xb5' "
  259. r"at position 4"):
  260. self.checkPatternError(b'(?P=\xc2\xb5)',
  261. r"unknown group name '\xc2\xb5'", 4)
  262. with self.assertWarnsRegex(DeprecationWarning,
  263. r"bad character in group name '\\xc2\\xb5' "
  264. r"at position 3"):
  265. self.checkPatternError(b'(?(\xc2\xb5)y)',
  266. r"unknown group name '\xc2\xb5'", 3)
  267. def test_symbolic_refs(self):
  268. self.assertEqual(re.sub('(?P<a>x)|(?P<b>y)', r'\g<b>', 'xx'), '')
  269. self.assertEqual(re.sub('(?P<a>x)|(?P<b>y)', r'\2', 'xx'), '')
  270. self.assertEqual(re.sub(b'(?P<a1>x)', br'\g<a1>', b'xx'), b'xx')
  271. # New valid identifiers in Python 3
  272. self.assertEqual(re.sub('(?P<µ>x)', r'\g<µ>', 'xx'), 'xx')
  273. self.assertEqual(re.sub('(?P<𝔘𝔫𝔦𝔠𝔬𝔡𝔢>x)', r'\g<𝔘𝔫𝔦𝔠𝔬𝔡𝔢>', 'xx'), 'xx')
  274. # Support > 100 groups.
  275. pat = '|'.join('x(?P<a%d>%x)y' % (i, i) for i in range(1, 200 + 1))
  276. self.assertEqual(re.sub(pat, r'\g<200>', 'xc8yzxc8y'), 'c8zc8')
  277. def test_symbolic_refs_errors(self):
  278. self.checkTemplateError('(?P<a>x)', r'\g<a', 'xx',
  279. 'missing >, unterminated name', 3)
  280. self.checkTemplateError('(?P<a>x)', r'\g<', 'xx',
  281. 'missing group name', 3)
  282. self.checkTemplateError('(?P<a>x)', r'\g', 'xx', 'missing <', 2)
  283. self.checkTemplateError('(?P<a>x)', r'\g<a a>', 'xx',
  284. "bad character in group name 'a a'", 3)
  285. self.checkTemplateError('(?P<a>x)', r'\g<>', 'xx',
  286. 'missing group name', 3)
  287. self.checkTemplateError('(?P<a>x)', r'\g<1a1>', 'xx',
  288. "bad character in group name '1a1'", 3)
  289. self.checkTemplateError('(?P<a>x)', r'\g<2>', 'xx',
  290. 'invalid group reference 2', 3)
  291. self.checkTemplateError('(?P<a>x)', r'\2', 'xx',
  292. 'invalid group reference 2', 1)
  293. with self.assertRaisesRegex(IndexError, "unknown group name 'ab'"):
  294. re.sub('(?P<a>x)', r'\g<ab>', 'xx')
  295. self.checkTemplateError('(?P<a>x)', r'\g<-1>', 'xx',
  296. "bad character in group name '-1'", 3)
  297. with self.assertWarnsRegex(DeprecationWarning,
  298. r"bad character in group name '\+1' "
  299. r"at position 3") as w:
  300. re.sub('(?P<a>x)', r'\g<+1>', 'xx')
  301. self.assertEqual(w.filename, __file__)
  302. with self.assertWarnsRegex(DeprecationWarning,
  303. r"bad character in group name '1_0' "
  304. r"at position 3"):
  305. re.sub('()'*10, r'\g<1_0>', 'xx')
  306. with self.assertWarnsRegex(DeprecationWarning,
  307. r"bad character in group name ' 1 ' "
  308. r"at position 3"):
  309. re.sub('(?P<a>x)', r'\g< 1 >', 'xx')
  310. self.checkTemplateError('(?P<a>x)', r'\g<©>', 'xx',
  311. "bad character in group name '©'", 3)
  312. with self.assertWarnsRegex(DeprecationWarning,
  313. r"bad character in group name '\\xc2\\xb5' "
  314. r"at position 3") as w:
  315. with self.assertRaisesRegex(IndexError, "unknown group name '\xc2\xb5'"):
  316. re.sub(b'(?P<a>x)', b'\\g<\xc2\xb5>', b'xx')
  317. self.assertEqual(w.filename, __file__)
  318. self.checkTemplateError('(?P<a>x)', r'\g<㊀>', 'xx',
  319. "bad character in group name '㊀'", 3)
  320. self.checkTemplateError('(?P<a>x)', r'\g<¹>', 'xx',
  321. "bad character in group name '¹'", 3)
  322. with self.assertWarnsRegex(DeprecationWarning,
  323. r"bad character in group name '१' "
  324. r"at position 3"):
  325. re.sub('(?P<a>x)', r'\g<१>', 'xx')
  326. def test_re_subn(self):
  327. self.assertEqual(re.subn("(?i)b+", "x", "bbbb BBBB"), ('x x', 2))
  328. self.assertEqual(re.subn("b+", "x", "bbbb BBBB"), ('x BBBB', 1))
  329. self.assertEqual(re.subn("b+", "x", "xyz"), ('xyz', 0))
  330. self.assertEqual(re.subn("b*", "x", "xyz"), ('xxxyxzx', 4))
  331. self.assertEqual(re.subn("b*", "x", "xyz", 2), ('xxxyz', 2))
  332. self.assertEqual(re.subn("b*", "x", "xyz", count=2), ('xxxyz', 2))
  333. def test_re_split(self):
  334. for string in ":a:b::c", S(":a:b::c"):
  335. self.assertTypedEqual(re.split(":", string),
  336. ['', 'a', 'b', '', 'c'])
  337. self.assertTypedEqual(re.split(":+", string),
  338. ['', 'a', 'b', 'c'])
  339. self.assertTypedEqual(re.split("(:+)", string),
  340. ['', ':', 'a', ':', 'b', '::', 'c'])
  341. for string in (b":a:b::c", B(b":a:b::c"), bytearray(b":a:b::c"),
  342. memoryview(b":a:b::c")):
  343. self.assertTypedEqual(re.split(b":", string),
  344. [b'', b'a', b'b', b'', b'c'])
  345. self.assertTypedEqual(re.split(b":+", string),
  346. [b'', b'a', b'b', b'c'])
  347. self.assertTypedEqual(re.split(b"(:+)", string),
  348. [b'', b':', b'a', b':', b'b', b'::', b'c'])
  349. for a, b, c in ("\xe0\xdf\xe7", "\u0430\u0431\u0432",
  350. "\U0001d49c\U0001d49e\U0001d4b5"):
  351. string = ":%s:%s::%s" % (a, b, c)
  352. self.assertEqual(re.split(":", string), ['', a, b, '', c])
  353. self.assertEqual(re.split(":+", string), ['', a, b, c])
  354. self.assertEqual(re.split("(:+)", string),
  355. ['', ':', a, ':', b, '::', c])
  356. self.assertEqual(re.split("(?::+)", ":a:b::c"), ['', 'a', 'b', 'c'])
  357. self.assertEqual(re.split("(:)+", ":a:b::c"),
  358. ['', ':', 'a', ':', 'b', ':', 'c'])
  359. self.assertEqual(re.split("([b:]+)", ":a:b::c"),
  360. ['', ':', 'a', ':b::', 'c'])
  361. self.assertEqual(re.split("(b)|(:+)", ":a:b::c"),
  362. ['', None, ':', 'a', None, ':', '', 'b', None, '',
  363. None, '::', 'c'])
  364. self.assertEqual(re.split("(?:b)|(?::+)", ":a:b::c"),
  365. ['', 'a', '', '', 'c'])
  366. for sep, expected in [
  367. (':*', ['', '', 'a', '', 'b', '', 'c', '']),
  368. ('(?::*)', ['', '', 'a', '', 'b', '', 'c', '']),
  369. ('(:*)', ['', ':', '', '', 'a', ':', '', '', 'b', '::', '', '', 'c', '', '']),
  370. ('(:)*', ['', ':', '', None, 'a', ':', '', None, 'b', ':', '', None, 'c', None, '']),
  371. ]:
  372. with self.subTest(sep=sep):
  373. self.assertTypedEqual(re.split(sep, ':a:b::c'), expected)
  374. for sep, expected in [
  375. ('', ['', ':', 'a', ':', 'b', ':', ':', 'c', '']),
  376. (r'\b', [':', 'a', ':', 'b', '::', 'c', '']),
  377. (r'(?=:)', ['', ':a', ':b', ':', ':c']),
  378. (r'(?<=:)', [':', 'a:', 'b:', ':', 'c']),
  379. ]:
  380. with self.subTest(sep=sep):
  381. self.assertTypedEqual(re.split(sep, ':a:b::c'), expected)
  382. def test_qualified_re_split(self):
  383. self.assertEqual(re.split(":", ":a:b::c", 2), ['', 'a', 'b::c'])
  384. self.assertEqual(re.split(":", ":a:b::c", maxsplit=2), ['', 'a', 'b::c'])
  385. self.assertEqual(re.split(':', 'a:b:c:d', maxsplit=2), ['a', 'b', 'c:d'])
  386. self.assertEqual(re.split("(:)", ":a:b::c", maxsplit=2),
  387. ['', ':', 'a', ':', 'b::c'])
  388. self.assertEqual(re.split("(:+)", ":a:b::c", maxsplit=2),
  389. ['', ':', 'a', ':', 'b::c'])
  390. self.assertEqual(re.split("(:*)", ":a:b::c", maxsplit=2),
  391. ['', ':', '', '', 'a:b::c'])
  392. def test_re_findall(self):
  393. self.assertEqual(re.findall(":+", "abc"), [])
  394. for string in "a:b::c:::d", S("a:b::c:::d"):
  395. self.assertTypedEqual(re.findall(":+", string),
  396. [":", "::", ":::"])
  397. self.assertTypedEqual(re.findall("(:+)", string),
  398. [":", "::", ":::"])
  399. self.assertTypedEqual(re.findall("(:)(:*)", string),
  400. [(":", ""), (":", ":"), (":", "::")])
  401. for string in (b"a:b::c:::d", B(b"a:b::c:::d"), bytearray(b"a:b::c:::d"),
  402. memoryview(b"a:b::c:::d")):
  403. self.assertTypedEqual(re.findall(b":+", string),
  404. [b":", b"::", b":::"])
  405. self.assertTypedEqual(re.findall(b"(:+)", string),
  406. [b":", b"::", b":::"])
  407. self.assertTypedEqual(re.findall(b"(:)(:*)", string),
  408. [(b":", b""), (b":", b":"), (b":", b"::")])
  409. for x in ("\xe0", "\u0430", "\U0001d49c"):
  410. xx = x * 2
  411. xxx = x * 3
  412. string = "a%sb%sc%sd" % (x, xx, xxx)
  413. self.assertEqual(re.findall("%s+" % x, string), [x, xx, xxx])
  414. self.assertEqual(re.findall("(%s+)" % x, string), [x, xx, xxx])
  415. self.assertEqual(re.findall("(%s)(%s*)" % (x, x), string),
  416. [(x, ""), (x, x), (x, xx)])
  417. def test_bug_117612(self):
  418. self.assertEqual(re.findall(r"(a|(b))", "aba"),
  419. [("a", ""),("b", "b"),("a", "")])
  420. def test_re_match(self):
  421. for string in 'a', S('a'):
  422. self.assertEqual(re.match('a', string).groups(), ())
  423. self.assertEqual(re.match('(a)', string).groups(), ('a',))
  424. self.assertEqual(re.match('(a)', string).group(0), 'a')
  425. self.assertEqual(re.match('(a)', string).group(1), 'a')
  426. self.assertEqual(re.match('(a)', string).group(1, 1), ('a', 'a'))
  427. for string in b'a', B(b'a'), bytearray(b'a'), memoryview(b'a'):
  428. self.assertEqual(re.match(b'a', string).groups(), ())
  429. self.assertEqual(re.match(b'(a)', string).groups(), (b'a',))
  430. self.assertEqual(re.match(b'(a)', string).group(0), b'a')
  431. self.assertEqual(re.match(b'(a)', string).group(1), b'a')
  432. self.assertEqual(re.match(b'(a)', string).group(1, 1), (b'a', b'a'))
  433. for a in ("\xe0", "\u0430", "\U0001d49c"):
  434. self.assertEqual(re.match(a, a).groups(), ())
  435. self.assertEqual(re.match('(%s)' % a, a).groups(), (a,))
  436. self.assertEqual(re.match('(%s)' % a, a).group(0), a)
  437. self.assertEqual(re.match('(%s)' % a, a).group(1), a)
  438. self.assertEqual(re.match('(%s)' % a, a).group(1, 1), (a, a))
  439. pat = re.compile('((a)|(b))(c)?')
  440. self.assertEqual(pat.match('a').groups(), ('a', 'a', None, None))
  441. self.assertEqual(pat.match('b').groups(), ('b', None, 'b', None))
  442. self.assertEqual(pat.match('ac').groups(), ('a', 'a', None, 'c'))
  443. self.assertEqual(pat.match('bc').groups(), ('b', None, 'b', 'c'))
  444. self.assertEqual(pat.match('bc').groups(""), ('b', "", 'b', 'c'))
  445. pat = re.compile('(?:(?P<a1>a)|(?P<b2>b))(?P<c3>c)?')
  446. self.assertEqual(pat.match('a').group(1, 2, 3), ('a', None, None))
  447. self.assertEqual(pat.match('b').group('a1', 'b2', 'c3'),
  448. (None, 'b', None))
  449. self.assertEqual(pat.match('ac').group(1, 'b2', 3), ('a', None, 'c'))
  450. def test_group(self):
  451. class Index:
  452. def __init__(self, value):
  453. self.value = value
  454. def __index__(self):
  455. return self.value
  456. # A single group
  457. m = re.match('(a)(b)', 'ab')
  458. self.assertEqual(m.group(), 'ab')
  459. self.assertEqual(m.group(0), 'ab')
  460. self.assertEqual(m.group(1), 'a')
  461. self.assertEqual(m.group(Index(1)), 'a')
  462. self.assertRaises(IndexError, m.group, -1)
  463. self.assertRaises(IndexError, m.group, 3)
  464. self.assertRaises(IndexError, m.group, 1<<1000)
  465. self.assertRaises(IndexError, m.group, Index(1<<1000))
  466. self.assertRaises(IndexError, m.group, 'x')
  467. # Multiple groups
  468. self.assertEqual(m.group(2, 1), ('b', 'a'))
  469. self.assertEqual(m.group(Index(2), Index(1)), ('b', 'a'))
  470. def test_match_getitem(self):
  471. pat = re.compile('(?:(?P<a1>a)|(?P<b2>b))(?P<c3>c)?')
  472. m = pat.match('a')
  473. self.assertEqual(m['a1'], 'a')
  474. self.assertEqual(m['b2'], None)
  475. self.assertEqual(m['c3'], None)
  476. self.assertEqual('a1={a1} b2={b2} c3={c3}'.format_map(m), 'a1=a b2=None c3=None')
  477. self.assertEqual(m[0], 'a')
  478. self.assertEqual(m[1], 'a')
  479. self.assertEqual(m[2], None)
  480. self.assertEqual(m[3], None)
  481. with self.assertRaisesRegex(IndexError, 'no such group'):
  482. m['X']
  483. with self.assertRaisesRegex(IndexError, 'no such group'):
  484. m[-1]
  485. with self.assertRaisesRegex(IndexError, 'no such group'):
  486. m[4]
  487. with self.assertRaisesRegex(IndexError, 'no such group'):
  488. m[0, 1]
  489. with self.assertRaisesRegex(IndexError, 'no such group'):
  490. m[(0,)]
  491. with self.assertRaisesRegex(IndexError, 'no such group'):
  492. m[(0, 1)]
  493. with self.assertRaisesRegex(IndexError, 'no such group'):
  494. 'a1={a2}'.format_map(m)
  495. m = pat.match('ac')
  496. self.assertEqual(m['a1'], 'a')
  497. self.assertEqual(m['b2'], None)
  498. self.assertEqual(m['c3'], 'c')
  499. self.assertEqual('a1={a1} b2={b2} c3={c3}'.format_map(m), 'a1=a b2=None c3=c')
  500. self.assertEqual(m[0], 'ac')
  501. self.assertEqual(m[1], 'a')
  502. self.assertEqual(m[2], None)
  503. self.assertEqual(m[3], 'c')
  504. # Cannot assign.
  505. with self.assertRaises(TypeError):
  506. m[0] = 1
  507. # No len().
  508. self.assertRaises(TypeError, len, m)
  509. def test_re_fullmatch(self):
  510. # Issue 16203: Proposal: add re.fullmatch() method.
  511. self.assertEqual(re.fullmatch(r"a", "a").span(), (0, 1))
  512. for string in "ab", S("ab"):
  513. self.assertEqual(re.fullmatch(r"a|ab", string).span(), (0, 2))
  514. for string in b"ab", B(b"ab"), bytearray(b"ab"), memoryview(b"ab"):
  515. self.assertEqual(re.fullmatch(br"a|ab", string).span(), (0, 2))
  516. for a, b in "\xe0\xdf", "\u0430\u0431", "\U0001d49c\U0001d49e":
  517. r = r"%s|%s" % (a, a + b)
  518. self.assertEqual(re.fullmatch(r, a + b).span(), (0, 2))
  519. self.assertEqual(re.fullmatch(r".*?$", "abc").span(), (0, 3))
  520. self.assertEqual(re.fullmatch(r".*?", "abc").span(), (0, 3))
  521. self.assertEqual(re.fullmatch(r"a.*?b", "ab").span(), (0, 2))
  522. self.assertEqual(re.fullmatch(r"a.*?b", "abb").span(), (0, 3))
  523. self.assertEqual(re.fullmatch(r"a.*?b", "axxb").span(), (0, 4))
  524. self.assertIsNone(re.fullmatch(r"a+", "ab"))
  525. self.assertIsNone(re.fullmatch(r"abc$", "abc\n"))
  526. self.assertIsNone(re.fullmatch(r"abc\Z", "abc\n"))
  527. self.assertIsNone(re.fullmatch(r"(?m)abc$", "abc\n"))
  528. self.assertEqual(re.fullmatch(r"ab(?=c)cd", "abcd").span(), (0, 4))
  529. self.assertEqual(re.fullmatch(r"ab(?<=b)cd", "abcd").span(), (0, 4))
  530. self.assertEqual(re.fullmatch(r"(?=a|ab)ab", "ab").span(), (0, 2))
  531. self.assertEqual(
  532. re.compile(r"bc").fullmatch("abcd", pos=1, endpos=3).span(), (1, 3))
  533. self.assertEqual(
  534. re.compile(r".*?$").fullmatch("abcd", pos=1, endpos=3).span(), (1, 3))
  535. self.assertEqual(
  536. re.compile(r".*?").fullmatch("abcd", pos=1, endpos=3).span(), (1, 3))
  537. def test_re_groupref_exists(self):
  538. self.assertEqual(re.match(r'^(\()?([^()]+)(?(1)\))$', '(a)').groups(),
  539. ('(', 'a'))
  540. self.assertEqual(re.match(r'^(\()?([^()]+)(?(1)\))$', 'a').groups(),
  541. (None, 'a'))
  542. self.assertIsNone(re.match(r'^(\()?([^()]+)(?(1)\))$', 'a)'))
  543. self.assertIsNone(re.match(r'^(\()?([^()]+)(?(1)\))$', '(a'))
  544. self.assertEqual(re.match('^(?:(a)|c)((?(1)b|d))$', 'ab').groups(),
  545. ('a', 'b'))
  546. self.assertEqual(re.match(r'^(?:(a)|c)((?(1)b|d))$', 'cd').groups(),
  547. (None, 'd'))
  548. self.assertEqual(re.match(r'^(?:(a)|c)((?(1)|d))$', 'cd').groups(),
  549. (None, 'd'))
  550. self.assertEqual(re.match(r'^(?:(a)|c)((?(1)|d))$', 'a').groups(),
  551. ('a', ''))
  552. # Tests for bug #1177831: exercise groups other than the first group
  553. p = re.compile('(?P<g1>a)(?P<g2>b)?((?(g2)c|d))')
  554. self.assertEqual(p.match('abc').groups(),
  555. ('a', 'b', 'c'))
  556. self.assertEqual(p.match('ad').groups(),
  557. ('a', None, 'd'))
  558. self.assertIsNone(p.match('abd'))
  559. self.assertIsNone(p.match('ac'))
  560. # Support > 100 groups.
  561. pat = '|'.join('x(?P<a%d>%x)y' % (i, i) for i in range(1, 200 + 1))
  562. pat = '(?:%s)(?(200)z)' % pat
  563. self.assertEqual(re.match(pat, 'xc8yz').span(), (0, 5))
  564. def test_re_groupref_exists_errors(self):
  565. self.checkPatternError(r'(?P<a>)(?(0)a|b)', 'bad group number', 10)
  566. self.checkPatternError(r'()(?(-1)a|b)',
  567. "bad character in group name '-1'", 5)
  568. with self.assertWarnsRegex(DeprecationWarning,
  569. r"bad character in group name '\+1' "
  570. r"at position 5") as w:
  571. re.compile(r'()(?(+1)a|b)')
  572. self.assertEqual(w.filename, __file__)
  573. with self.assertWarnsRegex(DeprecationWarning,
  574. r"bad character in group name '1_0' "
  575. r"at position 23"):
  576. re.compile(r'()'*10 + r'(?(1_0)a|b)')
  577. with self.assertWarnsRegex(DeprecationWarning,
  578. r"bad character in group name ' 1 ' "
  579. r"at position 5"):
  580. re.compile(r'()(?( 1 )a|b)')
  581. self.checkPatternError(r'()(?(㊀)a|b)',
  582. "bad character in group name '㊀'", 5)
  583. self.checkPatternError(r'()(?(¹)a|b)',
  584. "bad character in group name '¹'", 5)
  585. with self.assertWarnsRegex(DeprecationWarning,
  586. r"bad character in group name '१' "
  587. r"at position 5"):
  588. re.compile(r'()(?(१)a|b)')
  589. self.checkPatternError(r'()(?(1',
  590. "missing ), unterminated name", 5)
  591. self.checkPatternError(r'()(?(1)a',
  592. "missing ), unterminated subpattern", 2)
  593. self.checkPatternError(r'()(?(1)a|b',
  594. 'missing ), unterminated subpattern', 2)
  595. self.checkPatternError(r'()(?(1)a|b|c',
  596. 'conditional backref with more than '
  597. 'two branches', 10)
  598. self.checkPatternError(r'()(?(1)a|b|c)',
  599. 'conditional backref with more than '
  600. 'two branches', 10)
  601. self.checkPatternError(r'()(?(2)a)',
  602. "invalid group reference 2", 5)
  603. def test_re_groupref_exists_validation_bug(self):
  604. for i in range(256):
  605. with self.subTest(code=i):
  606. re.compile(r'()(?(1)\x%02x?)' % i)
  607. def test_re_groupref_overflow(self):
  608. from re._constants import MAXGROUPS
  609. self.checkTemplateError('()', r'\g<%s>' % MAXGROUPS, 'xx',
  610. 'invalid group reference %d' % MAXGROUPS, 3)
  611. self.checkPatternError(r'(?P<a>)(?(%d))' % MAXGROUPS,
  612. 'invalid group reference %d' % MAXGROUPS, 10)
  613. def test_re_groupref(self):
  614. self.assertEqual(re.match(r'^(\|)?([^()]+)\1$', '|a|').groups(),
  615. ('|', 'a'))
  616. self.assertEqual(re.match(r'^(\|)?([^()]+)\1?$', 'a').groups(),
  617. (None, 'a'))
  618. self.assertIsNone(re.match(r'^(\|)?([^()]+)\1$', 'a|'))
  619. self.assertIsNone(re.match(r'^(\|)?([^()]+)\1$', '|a'))
  620. self.assertEqual(re.match(r'^(?:(a)|c)(\1)$', 'aa').groups(),
  621. ('a', 'a'))
  622. self.assertEqual(re.match(r'^(?:(a)|c)(\1)?$', 'c').groups(),
  623. (None, None))
  624. self.checkPatternError(r'(abc\1)', 'cannot refer to an open group', 4)
  625. def test_groupdict(self):
  626. self.assertEqual(re.match('(?P<first>first) (?P<second>second)',
  627. 'first second').groupdict(),
  628. {'first':'first', 'second':'second'})
  629. def test_expand(self):
  630. self.assertEqual(re.match("(?P<first>first) (?P<second>second)",
  631. "first second")
  632. .expand(r"\2 \1 \g<second> \g<first>"),
  633. "second first second first")
  634. self.assertEqual(re.match("(?P<first>first)|(?P<second>second)",
  635. "first")
  636. .expand(r"\2 \g<second>"),
  637. " ")
  638. def test_repeat_minmax(self):
  639. self.assertIsNone(re.match(r"^(\w){1}$", "abc"))
  640. self.assertIsNone(re.match(r"^(\w){1}?$", "abc"))
  641. self.assertIsNone(re.match(r"^(\w){1,2}$", "abc"))
  642. self.assertIsNone(re.match(r"^(\w){1,2}?$", "abc"))
  643. self.assertEqual(re.match(r"^(\w){3}$", "abc").group(1), "c")
  644. self.assertEqual(re.match(r"^(\w){1,3}$", "abc").group(1), "c")
  645. self.assertEqual(re.match(r"^(\w){1,4}$", "abc").group(1), "c")
  646. self.assertEqual(re.match(r"^(\w){3,4}?$", "abc").group(1), "c")
  647. self.assertEqual(re.match(r"^(\w){3}?$", "abc").group(1), "c")
  648. self.assertEqual(re.match(r"^(\w){1,3}?$", "abc").group(1), "c")
  649. self.assertEqual(re.match(r"^(\w){1,4}?$", "abc").group(1), "c")
  650. self.assertEqual(re.match(r"^(\w){3,4}?$", "abc").group(1), "c")
  651. self.assertIsNone(re.match(r"^x{1}$", "xxx"))
  652. self.assertIsNone(re.match(r"^x{1}?$", "xxx"))
  653. self.assertIsNone(re.match(r"^x{1,2}$", "xxx"))
  654. self.assertIsNone(re.match(r"^x{1,2}?$", "xxx"))
  655. self.assertTrue(re.match(r"^x{3}$", "xxx"))
  656. self.assertTrue(re.match(r"^x{1,3}$", "xxx"))
  657. self.assertTrue(re.match(r"^x{3,3}$", "xxx"))
  658. self.assertTrue(re.match(r"^x{1,4}$", "xxx"))
  659. self.assertTrue(re.match(r"^x{3,4}?$", "xxx"))
  660. self.assertTrue(re.match(r"^x{3}?$", "xxx"))
  661. self.assertTrue(re.match(r"^x{1,3}?$", "xxx"))
  662. self.assertTrue(re.match(r"^x{1,4}?$", "xxx"))
  663. self.assertTrue(re.match(r"^x{3,4}?$", "xxx"))
  664. self.assertIsNone(re.match(r"^x{}$", "xxx"))
  665. self.assertTrue(re.match(r"^x{}$", "x{}"))
  666. self.checkPatternError(r'x{2,1}',
  667. 'min repeat greater than max repeat', 2)
  668. def test_getattr(self):
  669. self.assertEqual(re.compile("(?i)(a)(b)").pattern, "(?i)(a)(b)")
  670. self.assertEqual(re.compile("(?i)(a)(b)").flags, re.I | re.U)
  671. self.assertEqual(re.compile("(?i)(a)(b)").groups, 2)
  672. self.assertEqual(re.compile("(?i)(a)(b)").groupindex, {})
  673. self.assertEqual(re.compile("(?i)(?P<first>a)(?P<other>b)").groupindex,
  674. {'first': 1, 'other': 2})
  675. self.assertEqual(re.match("(a)", "a").pos, 0)
  676. self.assertEqual(re.match("(a)", "a").endpos, 1)
  677. self.assertEqual(re.match("(a)", "a").string, "a")
  678. self.assertEqual(re.match("(a)", "a").regs, ((0, 1), (0, 1)))
  679. self.assertTrue(re.match("(a)", "a").re)
  680. # Issue 14260. groupindex should be non-modifiable mapping.
  681. p = re.compile(r'(?i)(?P<first>a)(?P<other>b)')
  682. self.assertEqual(sorted(p.groupindex), ['first', 'other'])
  683. self.assertEqual(p.groupindex['other'], 2)
  684. with self.assertRaises(TypeError):
  685. p.groupindex['other'] = 0
  686. self.assertEqual(p.groupindex['other'], 2)
  687. def test_special_escapes(self):
  688. self.assertEqual(re.search(r"\b(b.)\b",
  689. "abcd abc bcd bx").group(1), "bx")
  690. self.assertEqual(re.search(r"\B(b.)\B",
  691. "abc bcd bc abxd").group(1), "bx")
  692. self.assertEqual(re.search(r"\b(b.)\b",
  693. "abcd abc bcd bx", re.ASCII).group(1), "bx")
  694. self.assertEqual(re.search(r"\B(b.)\B",
  695. "abc bcd bc abxd", re.ASCII).group(1), "bx")
  696. self.assertEqual(re.search(r"^abc$", "\nabc\n", re.M).group(0), "abc")
  697. self.assertEqual(re.search(r"^\Aabc\Z$", "abc", re.M).group(0), "abc")
  698. self.assertIsNone(re.search(r"^\Aabc\Z$", "\nabc\n", re.M))
  699. self.assertEqual(re.search(br"\b(b.)\b",
  700. b"abcd abc bcd bx").group(1), b"bx")
  701. self.assertEqual(re.search(br"\B(b.)\B",
  702. b"abc bcd bc abxd").group(1), b"bx")
  703. self.assertEqual(re.search(br"\b(b.)\b",
  704. b"abcd abc bcd bx", re.LOCALE).group(1), b"bx")
  705. self.assertEqual(re.search(br"\B(b.)\B",
  706. b"abc bcd bc abxd", re.LOCALE).group(1), b"bx")
  707. self.assertEqual(re.search(br"^abc$", b"\nabc\n", re.M).group(0), b"abc")
  708. self.assertEqual(re.search(br"^\Aabc\Z$", b"abc", re.M).group(0), b"abc")
  709. self.assertIsNone(re.search(br"^\Aabc\Z$", b"\nabc\n", re.M))
  710. self.assertEqual(re.search(r"\d\D\w\W\s\S",
  711. "1aa! a").group(0), "1aa! a")
  712. self.assertEqual(re.search(br"\d\D\w\W\s\S",
  713. b"1aa! a").group(0), b"1aa! a")
  714. self.assertEqual(re.search(r"\d\D\w\W\s\S",
  715. "1aa! a", re.ASCII).group(0), "1aa! a")
  716. self.assertEqual(re.search(br"\d\D\w\W\s\S",
  717. b"1aa! a", re.LOCALE).group(0), b"1aa! a")
  718. def test_other_escapes(self):
  719. self.checkPatternError("\\", 'bad escape (end of pattern)', 0)
  720. self.assertEqual(re.match(r"\(", '(').group(), '(')
  721. self.assertIsNone(re.match(r"\(", ')'))
  722. self.assertEqual(re.match(r"\\", '\\').group(), '\\')
  723. self.assertEqual(re.match(r"[\]]", ']').group(), ']')
  724. self.assertIsNone(re.match(r"[\]]", '['))
  725. self.assertEqual(re.match(r"[a\-c]", '-').group(), '-')
  726. self.assertIsNone(re.match(r"[a\-c]", 'b'))
  727. self.assertEqual(re.match(r"[\^a]+", 'a^').group(), 'a^')
  728. self.assertIsNone(re.match(r"[\^a]+", 'b'))
  729. re.purge() # for warnings
  730. for c in 'ceghijklmopqyzCEFGHIJKLMNOPQRTVXY':
  731. with self.subTest(c):
  732. self.assertRaises(re.error, re.compile, '\\%c' % c)
  733. for c in 'ceghijklmopqyzABCEFGHIJKLMNOPQRTVXYZ':
  734. with self.subTest(c):
  735. self.assertRaises(re.error, re.compile, '[\\%c]' % c)
  736. def test_named_unicode_escapes(self):
  737. # test individual Unicode named escapes
  738. self.assertTrue(re.match(r'\N{LESS-THAN SIGN}', '<'))
  739. self.assertTrue(re.match(r'\N{less-than sign}', '<'))
  740. self.assertIsNone(re.match(r'\N{LESS-THAN SIGN}', '>'))
  741. self.assertTrue(re.match(r'\N{SNAKE}', '\U0001f40d'))
  742. self.assertTrue(re.match(r'\N{ARABIC LIGATURE UIGHUR KIRGHIZ YEH WITH '
  743. r'HAMZA ABOVE WITH ALEF MAKSURA ISOLATED FORM}',
  744. '\ufbf9'))
  745. self.assertTrue(re.match(r'[\N{LESS-THAN SIGN}-\N{GREATER-THAN SIGN}]',
  746. '='))
  747. self.assertIsNone(re.match(r'[\N{LESS-THAN SIGN}-\N{GREATER-THAN SIGN}]',
  748. ';'))
  749. # test errors in \N{name} handling - only valid names should pass
  750. self.checkPatternError(r'\N', 'missing {', 2)
  751. self.checkPatternError(r'[\N]', 'missing {', 3)
  752. self.checkPatternError(r'\N{', 'missing character name', 3)
  753. self.checkPatternError(r'[\N{', 'missing character name', 4)
  754. self.checkPatternError(r'\N{}', 'missing character name', 3)
  755. self.checkPatternError(r'[\N{}]', 'missing character name', 4)
  756. self.checkPatternError(r'\NSNAKE}', 'missing {', 2)
  757. self.checkPatternError(r'[\NSNAKE}]', 'missing {', 3)
  758. self.checkPatternError(r'\N{SNAKE',
  759. 'missing }, unterminated name', 3)
  760. self.checkPatternError(r'[\N{SNAKE]',
  761. 'missing }, unterminated name', 4)
  762. self.checkPatternError(r'[\N{SNAKE]}',
  763. "undefined character name 'SNAKE]'", 1)
  764. self.checkPatternError(r'\N{SPAM}',
  765. "undefined character name 'SPAM'", 0)
  766. self.checkPatternError(r'[\N{SPAM}]',
  767. "undefined character name 'SPAM'", 1)
  768. self.checkPatternError(r'\N{KEYCAP NUMBER SIGN}',
  769. "undefined character name 'KEYCAP NUMBER SIGN'", 0)
  770. self.checkPatternError(r'[\N{KEYCAP NUMBER SIGN}]',
  771. "undefined character name 'KEYCAP NUMBER SIGN'", 1)
  772. self.checkPatternError(br'\N{LESS-THAN SIGN}', r'bad escape \N', 0)
  773. self.checkPatternError(br'[\N{LESS-THAN SIGN}]', r'bad escape \N', 1)
  774. def test_string_boundaries(self):
  775. # See http://bugs.python.org/issue10713
  776. self.assertEqual(re.search(r"\b(abc)\b", "abc").group(1),
  777. "abc")
  778. # There's a word boundary at the start of a string.
  779. self.assertTrue(re.match(r"\b", "abc"))
  780. # A non-empty string includes a non-boundary zero-length match.
  781. self.assertTrue(re.search(r"\B", "abc"))
  782. # There is no non-boundary match at the start of a string.
  783. self.assertFalse(re.match(r"\B", "abc"))
  784. # However, an empty string contains no word boundaries, and also no
  785. # non-boundaries.
  786. self.assertIsNone(re.search(r"\B", ""))
  787. # This one is questionable and different from the perlre behaviour,
  788. # but describes current behavior.
  789. self.assertIsNone(re.search(r"\b", ""))
  790. # A single word-character string has two boundaries, but no
  791. # non-boundary gaps.
  792. self.assertEqual(len(re.findall(r"\b", "a")), 2)
  793. self.assertEqual(len(re.findall(r"\B", "a")), 0)
  794. # If there are no words, there are no boundaries
  795. self.assertEqual(len(re.findall(r"\b", " ")), 0)
  796. self.assertEqual(len(re.findall(r"\b", " ")), 0)
  797. # Can match around the whitespace.
  798. self.assertEqual(len(re.findall(r"\B", " ")), 2)
  799. def test_bigcharset(self):
  800. self.assertEqual(re.match("([\u2222\u2223])",
  801. "\u2222").group(1), "\u2222")
  802. r = '[%s]' % ''.join(map(chr, range(256, 2**16, 255)))
  803. self.assertEqual(re.match(r, "\uff01").group(), "\uff01")
  804. def test_big_codesize(self):
  805. # Issue #1160
  806. r = re.compile('|'.join(('%d'%x for x in range(10000))))
  807. self.assertTrue(r.match('1000'))
  808. self.assertTrue(r.match('9999'))
  809. def test_anyall(self):
  810. self.assertEqual(re.match("a.b", "a\nb", re.DOTALL).group(0),
  811. "a\nb")
  812. self.assertEqual(re.match("a.*b", "a\n\nb", re.DOTALL).group(0),
  813. "a\n\nb")
  814. def test_lookahead(self):
  815. self.assertEqual(re.match(r"(a(?=\s[^a]))", "a b").group(1), "a")
  816. self.assertEqual(re.match(r"(a(?=\s[^a]*))", "a b").group(1), "a")
  817. self.assertEqual(re.match(r"(a(?=\s[abc]))", "a b").group(1), "a")
  818. self.assertEqual(re.match(r"(a(?=\s[abc]*))", "a bc").group(1), "a")
  819. self.assertEqual(re.match(r"(a)(?=\s\1)", "a a").group(1), "a")
  820. self.assertEqual(re.match(r"(a)(?=\s\1*)", "a aa").group(1), "a")
  821. self.assertEqual(re.match(r"(a)(?=\s(abc|a))", "a a").group(1), "a")
  822. self.assertEqual(re.match(r"(a(?!\s[^a]))", "a a").group(1), "a")
  823. self.assertEqual(re.match(r"(a(?!\s[abc]))", "a d").group(1), "a")
  824. self.assertEqual(re.match(r"(a)(?!\s\1)", "a b").group(1), "a")
  825. self.assertEqual(re.match(r"(a)(?!\s(abc|a))", "a b").group(1), "a")
  826. # Group reference.
  827. self.assertTrue(re.match(r'(a)b(?=\1)a', 'aba'))
  828. self.assertIsNone(re.match(r'(a)b(?=\1)c', 'abac'))
  829. # Conditional group reference.
  830. self.assertTrue(re.match(r'(?:(a)|(x))b(?=(?(2)x|c))c', 'abc'))
  831. self.assertIsNone(re.match(r'(?:(a)|(x))b(?=(?(2)c|x))c', 'abc'))
  832. self.assertTrue(re.match(r'(?:(a)|(x))b(?=(?(2)x|c))c', 'abc'))
  833. self.assertIsNone(re.match(r'(?:(a)|(x))b(?=(?(1)b|x))c', 'abc'))
  834. self.assertTrue(re.match(r'(?:(a)|(x))b(?=(?(1)c|x))c', 'abc'))
  835. # Group used before defined.
  836. self.assertTrue(re.match(r'(a)b(?=(?(2)x|c))(c)', 'abc'))
  837. self.assertIsNone(re.match(r'(a)b(?=(?(2)b|x))(c)', 'abc'))
  838. self.assertTrue(re.match(r'(a)b(?=(?(1)c|x))(c)', 'abc'))
  839. def test_lookbehind(self):
  840. self.assertTrue(re.match(r'ab(?<=b)c', 'abc'))
  841. self.assertIsNone(re.match(r'ab(?<=c)c', 'abc'))
  842. self.assertIsNone(re.match(r'ab(?<!b)c', 'abc'))
  843. self.assertTrue(re.match(r'ab(?<!c)c', 'abc'))
  844. # Group reference.
  845. self.assertTrue(re.match(r'(a)a(?<=\1)c', 'aac'))
  846. self.assertIsNone(re.match(r'(a)b(?<=\1)a', 'abaa'))
  847. self.assertIsNone(re.match(r'(a)a(?<!\1)c', 'aac'))
  848. self.assertTrue(re.match(r'(a)b(?<!\1)a', 'abaa'))
  849. # Conditional group reference.
  850. self.assertIsNone(re.match(r'(?:(a)|(x))b(?<=(?(2)x|c))c', 'abc'))
  851. self.assertIsNone(re.match(r'(?:(a)|(x))b(?<=(?(2)b|x))c', 'abc'))
  852. self.assertTrue(re.match(r'(?:(a)|(x))b(?<=(?(2)x|b))c', 'abc'))
  853. self.assertIsNone(re.match(r'(?:(a)|(x))b(?<=(?(1)c|x))c', 'abc'))
  854. self.assertTrue(re.match(r'(?:(a)|(x))b(?<=(?(1)b|x))c', 'abc'))
  855. # Group used before defined.
  856. self.assertRaises(re.error, re.compile, r'(a)b(?<=(?(2)b|x))(c)')
  857. self.assertIsNone(re.match(r'(a)b(?<=(?(1)c|x))(c)', 'abc'))
  858. self.assertTrue(re.match(r'(a)b(?<=(?(1)b|x))(c)', 'abc'))
  859. # Group defined in the same lookbehind pattern
  860. self.assertRaises(re.error, re.compile, r'(a)b(?<=(.)\2)(c)')
  861. self.assertRaises(re.error, re.compile, r'(a)b(?<=(?P<a>.)(?P=a))(c)')
  862. self.assertRaises(re.error, re.compile, r'(a)b(?<=(a)(?(2)b|x))(c)')
  863. self.assertRaises(re.error, re.compile, r'(a)b(?<=(.)(?<=\2))(c)')
  864. def test_ignore_case(self):
  865. self.assertEqual(re.match("abc", "ABC", re.I).group(0), "ABC")
  866. self.assertEqual(re.match(b"abc", b"ABC", re.I).group(0), b"ABC")
  867. self.assertEqual(re.match(r"(a\s[^a])", "a b", re.I).group(1), "a b")
  868. self.assertEqual(re.match(r"(a\s[^a]*)", "a bb", re.I).group(1), "a bb")
  869. self.assertEqual(re.match(r"(a\s[abc])", "a b", re.I).group(1), "a b")
  870. self.assertEqual(re.match(r"(a\s[abc]*)", "a bb", re.I).group(1), "a bb")
  871. self.assertEqual(re.match(r"((a)\s\2)", "a a", re.I).group(1), "a a")
  872. self.assertEqual(re.match(r"((a)\s\2*)", "a aa", re.I).group(1), "a aa")
  873. self.assertEqual(re.match(r"((a)\s(abc|a))", "a a", re.I).group(1), "a a")
  874. self.assertEqual(re.match(r"((a)\s(abc|a)*)", "a aa", re.I).group(1), "a aa")
  875. # Two different characters have the same lowercase.
  876. assert 'K'.lower() == '\u212a'.lower() == 'k' # 'K'
  877. self.assertTrue(re.match(r'K', '\u212a', re.I))
  878. self.assertTrue(re.match(r'k', '\u212a', re.I))
  879. self.assertTrue(re.match(r'\u212a', 'K', re.I))
  880. self.assertTrue(re.match(r'\u212a', 'k', re.I))
  881. # Two different characters have the same uppercase.
  882. assert 's'.upper() == '\u017f'.upper() == 'S' # 'ſ'
  883. self.assertTrue(re.match(r'S', '\u017f', re.I))
  884. self.assertTrue(re.match(r's', '\u017f', re.I))
  885. self.assertTrue(re.match(r'\u017f', 'S', re.I))
  886. self.assertTrue(re.match(r'\u017f', 's', re.I))
  887. # Two different characters have the same uppercase. Unicode 9.0+.
  888. assert '\u0432'.upper() == '\u1c80'.upper() == '\u0412' # 'в', 'ᲀ', 'В'
  889. self.assertTrue(re.match(r'\u0412', '\u0432', re.I))
  890. self.assertTrue(re.match(r'\u0412', '\u1c80', re.I))
  891. self.assertTrue(re.match(r'\u0432', '\u0412', re.I))
  892. self.assertTrue(re.match(r'\u0432', '\u1c80', re.I))
  893. self.assertTrue(re.match(r'\u1c80', '\u0412', re.I))
  894. self.assertTrue(re.match(r'\u1c80', '\u0432', re.I))
  895. # Two different characters have the same multicharacter uppercase.
  896. assert '\ufb05'.upper() == '\ufb06'.upper() == 'ST' # 'ſt', 'st'
  897. self.assertTrue(re.match(r'\ufb05', '\ufb06', re.I))
  898. self.assertTrue(re.match(r'\ufb06', '\ufb05', re.I))
  899. def test_ignore_case_set(self):
  900. self.assertTrue(re.match(r'[19A]', 'A', re.I))
  901. self.assertTrue(re.match(r'[19a]', 'a', re.I))
  902. self.assertTrue(re.match(r'[19a]', 'A', re.I))
  903. self.assertTrue(re.match(r'[19A]', 'a', re.I))
  904. self.assertTrue(re.match(br'[19A]', b'A', re.I))
  905. self.assertTrue(re.match(br'[19a]', b'a', re.I))
  906. self.assertTrue(re.match(br'[19a]', b'A', re.I))
  907. self.assertTrue(re.match(br'[19A]', b'a', re.I))
  908. # Two different characters have the same lowercase.
  909. assert 'K'.lower() == '\u212a'.lower() == 'k' # 'K'
  910. self.assertTrue(re.match(r'[19K]', '\u212a', re.I))
  911. self.assertTrue(re.match(r'[19k]', '\u212a', re.I))
  912. self.assertTrue(re.match(r'[19\u212a]', 'K', re.I))
  913. self.assertTrue(re.match(r'[19\u212a]', 'k', re.I))
  914. # Two different characters have the same uppercase.
  915. assert 's'.upper() == '\u017f'.upper() == 'S' # 'ſ'
  916. self.assertTrue(re.match(r'[19S]', '\u017f', re.I))
  917. self.assertTrue(re.match(r'[19s]', '\u017f', re.I))
  918. self.assertTrue(re.match(r'[19\u017f]', 'S', re.I))
  919. self.assertTrue(re.match(r'[19\u017f]', 's', re.I))
  920. # Two different characters have the same uppercase. Unicode 9.0+.
  921. assert '\u0432'.upper() == '\u1c80'.upper() == '\u0412' # 'в', 'ᲀ', 'В'
  922. self.assertTrue(re.match(r'[19\u0412]', '\u0432', re.I))
  923. self.assertTrue(re.match(r'[19\u0412]', '\u1c80', re.I))
  924. self.assertTrue(re.match(r'[19\u0432]', '\u0412', re.I))
  925. self.assertTrue(re.match(r'[19\u0432]', '\u1c80', re.I))
  926. self.assertTrue(re.match(r'[19\u1c80]', '\u0412', re.I))
  927. self.assertTrue(re.match(r'[19\u1c80]', '\u0432', re.I))
  928. # Two different characters have the same multicharacter uppercase.
  929. assert '\ufb05'.upper() == '\ufb06'.upper() == 'ST' # 'ſt', 'st'
  930. self.assertTrue(re.match(r'[19\ufb05]', '\ufb06', re.I))
  931. self.assertTrue(re.match(r'[19\ufb06]', '\ufb05', re.I))
  932. def test_ignore_case_range(self):
  933. # Issues #3511, #17381.
  934. self.assertTrue(re.match(r'[9-a]', '_', re.I))
  935. self.assertIsNone(re.match(r'[9-A]', '_', re.I))
  936. self.assertTrue(re.match(br'[9-a]', b'_', re.I))
  937. self.assertIsNone(re.match(br'[9-A]', b'_', re.I))
  938. self.assertTrue(re.match(r'[\xc0-\xde]', '\xd7', re.I))
  939. self.assertIsNone(re.match(r'[\xc0-\xde]', '\xf7', re.I))
  940. self.assertTrue(re.match(r'[\xe0-\xfe]', '\xf7', re.I))
  941. self.assertIsNone(re.match(r'[\xe0-\xfe]', '\xd7', re.I))
  942. self.assertTrue(re.match(r'[\u0430-\u045f]', '\u0450', re.I))
  943. self.assertTrue(re.match(r'[\u0430-\u045f]', '\u0400', re.I))
  944. self.assertTrue(re.match(r'[\u0400-\u042f]', '\u0450', re.I))
  945. self.assertTrue(re.match(r'[\u0400-\u042f]', '\u0400', re.I))
  946. self.assertTrue(re.match(r'[\U00010428-\U0001044f]', '\U00010428', re.I))
  947. self.assertTrue(re.match(r'[\U00010428-\U0001044f]', '\U00010400', re.I))
  948. self.assertTrue(re.match(r'[\U00010400-\U00010427]', '\U00010428', re.I))
  949. self.assertTrue(re.match(r'[\U00010400-\U00010427]', '\U00010400', re.I))
  950. # Two different characters have the same lowercase.
  951. assert 'K'.lower() == '\u212a'.lower() == 'k' # 'K'
  952. self.assertTrue(re.match(r'[J-M]', '\u212a', re.I))
  953. self.assertTrue(re.match(r'[j-m]', '\u212a', re.I))
  954. self.assertTrue(re.match(r'[\u2129-\u212b]', 'K', re.I))
  955. self.assertTrue(re.match(r'[\u2129-\u212b]', 'k', re.I))
  956. # Two different characters have the same uppercase.
  957. assert 's'.upper() == '\u017f'.upper() == 'S' # 'ſ'
  958. self.assertTrue(re.match(r'[R-T]', '\u017f', re.I))
  959. self.assertTrue(re.match(r'[r-t]', '\u017f', re.I))
  960. self.assertTrue(re.match(r'[\u017e-\u0180]', 'S', re.I))
  961. self.assertTrue(re.match(r'[\u017e-\u0180]', 's', re.I))
  962. # Two different characters have the same uppercase. Unicode 9.0+.
  963. assert '\u0432'.upper() == '\u1c80'.upper() == '\u0412' # 'в', 'ᲀ', 'В'
  964. self.assertTrue(re.match(r'[\u0411-\u0413]', '\u0432', re.I))
  965. self.assertTrue(re.match(r'[\u0411-\u0413]', '\u1c80', re.I))
  966. self.assertTrue(re.match(r'[\u0431-\u0433]', '\u0412', re.I))
  967. self.assertTrue(re.match(r'[\u0431-\u0433]', '\u1c80', re.I))
  968. self.assertTrue(re.match(r'[\u1c80-\u1c82]', '\u0412', re.I))
  969. self.assertTrue(re.match(r'[\u1c80-\u1c82]', '\u0432', re.I))
  970. # Two different characters have the same multicharacter uppercase.
  971. assert '\ufb05'.upper() == '\ufb06'.upper() == 'ST' # 'ſt', 'st'
  972. self.assertTrue(re.match(r'[\ufb04-\ufb05]', '\ufb06', re.I))
  973. self.assertTrue(re.match(r'[\ufb06-\ufb07]', '\ufb05', re.I))
  974. def test_category(self):
  975. self.assertEqual(re.match(r"(\s)", " ").group(1), " ")
  976. @cpython_only
  977. def test_case_helpers(self):
  978. import _sre
  979. for i in range(128):
  980. c = chr(i)
  981. lo = ord(c.lower())
  982. self.assertEqual(_sre.ascii_tolower(i), lo)
  983. self.assertEqual(_sre.unicode_tolower(i), lo)
  984. iscased = c in string.ascii_letters
  985. self.assertEqual(_sre.ascii_iscased(i), iscased)
  986. self.assertEqual(_sre.unicode_iscased(i), iscased)
  987. for i in list(range(128, 0x1000)) + [0x10400, 0x10428]:
  988. c = chr(i)
  989. self.assertEqual(_sre.ascii_tolower(i), i)
  990. if i != 0x0130:
  991. self.assertEqual(_sre.unicode_tolower(i), ord(c.lower()))
  992. iscased = c != c.lower() or c != c.upper()
  993. self.assertFalse(_sre.ascii_iscased(i))
  994. self.assertEqual(_sre.unicode_iscased(i),
  995. c != c.lower() or c != c.upper())
  996. self.assertEqual(_sre.ascii_tolower(0x0130), 0x0130)
  997. self.assertEqual(_sre.unicode_tolower(0x0130), ord('i'))
  998. self.assertFalse(_sre.ascii_iscased(0x0130))
  999. self.assertTrue(_sre.unicode_iscased(0x0130))
  1000. def test_not_literal(self):
  1001. self.assertEqual(re.search(r"\s([^a])", " b").group(1), "b")
  1002. self.assertEqual(re.search(r"\s([^a]*)", " bb").group(1), "bb")
  1003. def test_possible_set_operations(self):
  1004. s = bytes(range(128)).decode()
  1005. with self.assertWarns(FutureWarning):
  1006. p = re.compile(r'[0-9--1]')
  1007. self.assertEqual(p.findall(s), list('-./0123456789'))
  1008. self.assertEqual(re.findall(r'[--1]', s), list('-./01'))
  1009. with self.assertWarns(FutureWarning):
  1010. p = re.compile(r'[%--1]')
  1011. self.assertEqual(p.findall(s), list("%&'()*+,-1"))
  1012. with self.assertWarns(FutureWarning):
  1013. p = re.compile(r'[%--]')
  1014. self.assertEqual(p.findall(s), list("%&'()*+,-"))
  1015. with self.assertWarns(FutureWarning):
  1016. p = re.compile(r'[0-9&&1]')
  1017. self.assertEqual(p.findall(s), list('&0123456789'))
  1018. with self.assertWarns(FutureWarning):
  1019. p = re.compile(r'[\d&&1]')
  1020. self.assertEqual(p.findall(s), list('&0123456789'))
  1021. self.assertEqual(re.findall(r'[&&1]', s), list('&1'))
  1022. with self.assertWarns(FutureWarning):
  1023. p = re.compile(r'[0-9||a]')
  1024. self.assertEqual(p.findall(s), list('0123456789a|'))
  1025. with self.assertWarns(FutureWarning):
  1026. p = re.compile(r'[\d||a]')
  1027. self.assertEqual(p.findall(s), list('0123456789a|'))
  1028. self.assertEqual(re.findall(r'[||1]', s), list('1|'))
  1029. with self.assertWarns(FutureWarning):
  1030. p = re.compile(r'[0-9~~1]')
  1031. self.assertEqual(p.findall(s), list('0123456789~'))
  1032. with self.assertWarns(FutureWarning):
  1033. p = re.compile(r'[\d~~1]')
  1034. self.assertEqual(p.findall(s), list('0123456789~'))
  1035. self.assertEqual(re.findall(r'[~~1]', s), list('1~'))
  1036. with self.assertWarns(FutureWarning):
  1037. p = re.compile(r'[[0-9]|]')
  1038. self.assertEqual(p.findall(s), list('0123456789[]'))
  1039. with self.assertWarns(FutureWarning):
  1040. p = re.compile(r'[[:digit:]|]')
  1041. self.assertEqual(p.findall(s), list(':[]dgit'))
  1042. def test_search_coverage(self):
  1043. self.assertEqual(re.search(r"\s(b)", " b").group(1), "b")
  1044. self.assertEqual(re.search(r"a\s", "a ").group(0), "a ")
  1045. def assertMatch(self, pattern, text, match=None, span=None,
  1046. matcher=re.fullmatch):
  1047. if match is None and span is None:
  1048. # the pattern matches the whole text
  1049. match = text
  1050. span = (0, len(text))
  1051. elif match is None or span is None:
  1052. raise ValueError('If match is not None, span should be specified '
  1053. '(and vice versa).')
  1054. m = matcher(pattern, text)
  1055. self.assertTrue(m)
  1056. self.assertEqual(m.group(), match)
  1057. self.assertEqual(m.span(), span)
  1058. LITERAL_CHARS = string.ascii_letters + string.digits + '!"%\',/:;<=>@_`'
  1059. def test_re_escape(self):
  1060. p = ''.join(chr(i) for i in range(256))
  1061. for c in p:
  1062. self.assertMatch(re.escape(c), c)
  1063. self.assertMatch('[' + re.escape(c) + ']', c)
  1064. self.assertMatch('(?x)' + re.escape(c), c)
  1065. self.assertMatch(re.escape(p), p)
  1066. for c in '-.]{}':
  1067. self.assertEqual(re.escape(c)[:1], '\\')
  1068. literal_chars = self.LITERAL_CHARS
  1069. self.assertEqual(re.escape(literal_chars), literal_chars)
  1070. def test_re_escape_bytes(self):
  1071. p = bytes(range(256))
  1072. for i in p:
  1073. b = bytes([i])
  1074. self.assertMatch(re.escape(b), b)
  1075. self.assertMatch(b'[' + re.escape(b) + b']', b)
  1076. self.assertMatch(b'(?x)' + re.escape(b), b)
  1077. self.assertMatch(re.escape(p), p)
  1078. for i in b'-.]{}':
  1079. b = bytes([i])
  1080. self.assertEqual(re.escape(b)[:1], b'\\')
  1081. literal_chars = self.LITERAL_CHARS.encode('ascii')
  1082. self.assertEqual(re.escape(literal_chars), literal_chars)
  1083. def test_re_escape_non_ascii(self):
  1084. s = 'xxx\u2620\u2620\u2620xxx'
  1085. s_escaped = re.escape(s)
  1086. self.assertEqual(s_escaped, s)
  1087. self.assertMatch(s_escaped, s)
  1088. self.assertMatch('.%s+.' % re.escape('\u2620'), s,
  1089. 'x\u2620\u2620\u2620x', (2, 7), re.search)
  1090. def test_re_escape_non_ascii_bytes(self):
  1091. b = 'y\u2620y\u2620y'.encode('utf-8')
  1092. b_escaped = re.escape(b)
  1093. self.assertEqual(b_escaped, b)
  1094. self.assertMatch(b_escaped, b)
  1095. res = re.findall(re.escape('\u2620'.encode('utf-8')), b)
  1096. self.assertEqual(len(res), 2)
  1097. def test_pickling(self):
  1098. import pickle
  1099. oldpat = re.compile('a(?:b|(c|e){1,2}?|d)+?(.)', re.UNICODE)
  1100. for proto in range(pickle.HIGHEST_PROTOCOL + 1):
  1101. pickled = pickle.dumps(oldpat, proto)
  1102. newpat = pickle.loads(pickled)
  1103. self.assertEqual(newpat, oldpat)
  1104. # current pickle expects the _compile() reconstructor in re module
  1105. from re import _compile
  1106. def test_copying(self):
  1107. import copy
  1108. p = re.compile(r'(?P<int>\d+)(?:\.(?P<frac>\d*))?')
  1109. self.assertIs(copy.copy(p), p)
  1110. self.assertIs(copy.deepcopy(p), p)
  1111. m = p.match('12.34')
  1112. self.assertIs(copy.copy(m), m)
  1113. self.assertIs(copy.deepcopy(m), m)
  1114. def test_constants(self):
  1115. self.assertEqual(re.I, re.IGNORECASE)
  1116. self.assertEqual(re.L, re.LOCALE)
  1117. self.assertEqual(re.M, re.MULTILINE)
  1118. self.assertEqual(re.S, re.DOTALL)
  1119. self.assertEqual(re.X, re.VERBOSE)
  1120. def test_flags(self):
  1121. for flag in [re.I, re.M, re.X, re.S, re.A, re.U]:
  1122. self.assertTrue(re.compile('^pattern$', flag))
  1123. for flag in [re.I, re.M, re.X, re.S, re.A, re.L]:
  1124. self.assertTrue(re.compile(b'^pattern$', flag))
  1125. def test_sre_character_literals(self):
  1126. for i in [0, 8, 16, 32, 64, 127, 128, 255, 256, 0xFFFF, 0x10000, 0x10FFFF]:
  1127. if i < 256:
  1128. self.assertTrue(re.match(r"\%03o" % i, chr(i)))
  1129. self.assertTrue(re.match(r"\%03o0" % i, chr(i)+"0"))
  1130. self.assertTrue(re.match(r"\%03o8" % i, chr(i)+"8"))
  1131. self.assertTrue(re.match(r"\x%02x" % i, chr(i)))
  1132. self.assertTrue(re.match(r"\x%02x0" % i, chr(i)+"0"))
  1133. self.assertTrue(re.match(r"\x%02xz" % i, chr(i)+"z"))
  1134. if i < 0x10000:
  1135. self.assertTrue(re.match(r"\u%04x" % i, chr(i)))
  1136. self.assertTrue(re.match(r"\u%04x0" % i, chr(i)+"0"))
  1137. self.assertTrue(re.match(r"\u%04xz" % i, chr(i)+"z"))
  1138. self.assertTrue(re.match(r"\U%08x" % i, chr(i)))
  1139. self.assertTrue(re.match(r"\U%08x0" % i, chr(i)+"0"))
  1140. self.assertTrue(re.match(r"\U%08xz" % i, chr(i)+"z"))
  1141. self.assertTrue(re.match(r"\0", "\000"))
  1142. self.assertTrue(re.match(r"\08", "\0008"))
  1143. self.assertTrue(re.match(r"\01", "\001"))
  1144. self.assertTrue(re.match(r"\018", "\0018"))
  1145. self.checkPatternError(r"\567",
  1146. r'octal escape value \567 outside of '
  1147. r'range 0-0o377', 0)
  1148. self.checkPatternError(r"\911", 'invalid group reference 91', 1)
  1149. self.checkPatternError(r"\x1", r'incomplete escape \x1', 0)
  1150. self.checkPatternError(r"\x1z", r'incomplete escape \x1', 0)
  1151. self.checkPatternError(r"\u123", r'incomplete escape \u123', 0)
  1152. self.checkPatternError(r"\u123z", r'incomplete escape \u123', 0)
  1153. self.checkPatternError(r"\U0001234", r'incomplete escape \U0001234', 0)
  1154. self.checkPatternError(r"\U0001234z", r'incomplete escape \U0001234', 0)
  1155. self.checkPatternError(r"\U00110000", r'bad escape \U00110000', 0)
  1156. def test_sre_character_class_literals(self):
  1157. for i in [0, 8, 16, 32, 64, 127, 128, 255, 256, 0xFFFF, 0x10000, 0x10FFFF]:
  1158. if i < 256:
  1159. self.assertTrue(re.match(r"[\%o]" % i, chr(i)))
  1160. self.assertTrue(re.match(r"[\%o8]" % i, chr(i)))
  1161. self.assertTrue(re.match(r"[\%03o]" % i, chr(i)))
  1162. self.assertTrue(re.match(r"[\%03o0]" % i, chr(i)))
  1163. self.assertTrue(re.match(r"[\%03o8]" % i, chr(i)))
  1164. self.assertTrue(re.match(r"[\x%02x]" % i, chr(i)))
  1165. self.assertTrue(re.match(r"[\x%02x0]" % i, chr(i)))
  1166. self.assertTrue(re.match(r"[\x%02xz]" % i, chr(i)))
  1167. if i < 0x10000:
  1168. self.assertTrue(re.match(r"[\u%04x]" % i, chr(i)))
  1169. self.assertTrue(re.match(r"[\u%04x0]" % i, chr(i)))
  1170. self.assertTrue(re.match(r"[\u%04xz]" % i, chr(i)))
  1171. self.assertTrue(re.match(r"[\U%08x]" % i, chr(i)))
  1172. self.assertTrue(re.match(r"[\U%08x0]" % i, chr(i)+"0"))
  1173. self.assertTrue(re.match(r"[\U%08xz]" % i, chr(i)+"z"))
  1174. self.checkPatternError(r"[\567]",
  1175. r'octal escape value \567 outside of '
  1176. r'range 0-0o377', 1)
  1177. self.checkPatternError(r"[\911]", r'bad escape \9', 1)
  1178. self.checkPatternError(r"[\x1z]", r'incomplete escape \x1', 1)
  1179. self.checkPatternError(r"[\u123z]", r'incomplete escape \u123', 1)
  1180. self.checkPatternError(r"[\U0001234z]", r'incomplete escape \U0001234', 1)
  1181. self.checkPatternError(r"[\U00110000]", r'bad escape \U00110000', 1)
  1182. self.assertTrue(re.match(r"[\U0001d49c-\U0001d4b5]", "\U0001d49e"))
  1183. def test_sre_byte_literals(self):
  1184. for i in [0, 8, 16, 32, 64, 127, 128, 255]:
  1185. self.assertTrue(re.match((r"\%03o" % i).encode(), bytes([i])))
  1186. self.assertTrue(re.match((r"\%03o0" % i).encode(), bytes([i])+b"0"))
  1187. self.assertTrue(re.match((r"\%03o8" % i).encode(), bytes([i])+b"8"))
  1188. self.assertTrue(re.match((r"\x%02x" % i).encode(), bytes([i])))
  1189. self.assertTrue(re.match((r"\x%02x0" % i).encode(), bytes([i])+b"0"))
  1190. self.assertTrue(re.match((r"\x%02xz" % i).encode(), bytes([i])+b"z"))
  1191. self.assertRaises(re.error, re.compile, br"\u1234")
  1192. self.assertRaises(re.error, re.compile, br"\U00012345")
  1193. self.assertTrue(re.match(br"\0", b"\000"))
  1194. self.assertTrue(re.match(br"\08", b"\0008"))
  1195. self.assertTrue(re.match(br"\01", b"\001"))
  1196. self.assertTrue(re.match(br"\018", b"\0018"))
  1197. self.checkPatternError(br"\567",
  1198. r'octal escape value \567 outside of '
  1199. r'range 0-0o377', 0)
  1200. self.checkPatternError(br"\911", 'invalid group reference 91', 1)
  1201. self.checkPatternError(br"\x1", r'incomplete escape \x1', 0)
  1202. self.checkPatternError(br"\x1z", r'incomplete escape \x1', 0)
  1203. def test_sre_byte_class_literals(self):
  1204. for i in [0, 8, 16, 32, 64, 127, 128, 255]:
  1205. self.assertTrue(re.match((r"[\%o]" % i).encode(), bytes([i])))
  1206. self.assertTrue(re.match((r"[\%o8]" % i).encode(), bytes([i])))
  1207. self.assertTrue(re.match((r"[\%03o]" % i).encode(), bytes([i])))
  1208. self.assertTrue(re.match((r"[\%03o0]" % i).encode(), bytes([i])))
  1209. self.assertTrue(re.match((r"[\%03o8]" % i).encode(), bytes([i])))
  1210. self.assertTrue(re.match((r"[\x%02x]" % i).encode(), bytes([i])))
  1211. self.assertTrue(re.match((r"[\x%02x0]" % i).encode(), bytes([i])))
  1212. self.assertTrue(re.match((r"[\x%02xz]" % i).encode(), bytes([i])))
  1213. self.assertRaises(re.error, re.compile, br"[\u1234]")
  1214. self.assertRaises(re.error, re.compile, br"[\U00012345]")
  1215. self.checkPatternError(br"[\567]",
  1216. r'octal escape value \567 outside of '
  1217. r'range 0-0o377', 1)
  1218. self.checkPatternError(br"[\911]", r'bad escape \9', 1)
  1219. self.checkPatternError(br"[\x1z]", r'incomplete escape \x1', 1)
  1220. def test_character_set_errors(self):
  1221. self.checkPatternError(r'[', 'unterminated character set', 0)
  1222. self.checkPatternError(r'[^', 'unterminated character set', 0)
  1223. self.checkPatternError(r'[a', 'unterminated character set', 0)
  1224. # bug 545855 -- This pattern failed to cause a compile error as it
  1225. # should, instead provoking a TypeError.
  1226. self.checkPatternError(r"[a-", 'unterminated character set', 0)
  1227. self.checkPatternError(r"[\w-b]", r'bad character range \w-b', 1)
  1228. self.checkPatternError(r"[a-\w]", r'bad character range a-\w', 1)
  1229. self.checkPatternError(r"[b-a]", 'bad character range b-a', 1)
  1230. def test_bug_113254(self):
  1231. self.assertEqual(re.match(r'(a)|(b)', 'b').start(1), -1)
  1232. self.assertEqual(re.match(r'(a)|(b)', 'b').end(1), -1)
  1233. self.assertEqual(re.match(r'(a)|(b)', 'b').span(1), (-1, -1))
  1234. def test_bug_527371(self):
  1235. # bug described in patches 527371/672491
  1236. self.assertIsNone(re.match(r'(a)?a','a').lastindex)
  1237. self.assertEqual(re.match(r'(a)(b)?b','ab').lastindex, 1)
  1238. self.assertEqual(re.match(r'(?P<a>a)(?P<b>b)?b','ab').lastgroup, 'a')
  1239. self.assertEqual(re.match(r"(?P<a>a(b))", "ab").lastgroup, 'a')
  1240. self.assertEqual(re.match(r"((a))", "a").lastindex, 1)
  1241. def test_bug_418626(self):
  1242. # bugs 418626 at al. -- Testing Greg Chapman's addition of op code
  1243. # SRE_OP_MIN_REPEAT_ONE for eliminating recursion on simple uses of
  1244. # pattern '*?' on a long string.
  1245. self.assertEqual(re.match('.*?c', 10000*'ab'+'cd').end(0), 20001)
  1246. self.assertEqual(re.match('.*?cd', 5000*'ab'+'c'+5000*'ab'+'cde').end(0),
  1247. 20003)
  1248. self.assertEqual(re.match('.*?cd', 20000*'abc'+'de').end(0), 60001)
  1249. # non-simple '*?' still used to hit the recursion limit, before the
  1250. # non-recursive scheme was implemented.
  1251. self.assertEqual(re.search('(a|b)*?c', 10000*'ab'+'cd').end(0), 20001)
  1252. def test_bug_612074(self):
  1253. pat="["+re.escape("\u2039")+"]"
  1254. self.assertEqual(re.compile(pat) and 1, 1)
  1255. def test_stack_overflow(self):
  1256. # nasty cases that used to overflow the straightforward recursive
  1257. # implementation of repeated groups.
  1258. self.assertEqual(re.match('(x)*', 50000*'x').group(1), 'x')
  1259. self.assertEqual(re.match('(x)*y', 50000*'x'+'y').group(1), 'x')
  1260. self.assertEqual(re.match('(x)*?y', 50000*'x'+'y').group(1), 'x')
  1261. def test_nothing_to_repeat(self):
  1262. for reps in '*', '+', '?', '{1,2}':
  1263. for mod in '', '?':
  1264. self.checkPatternError('%s%s' % (reps, mod),
  1265. 'nothing to repeat', 0)
  1266. self.checkPatternError('(?:%s%s)' % (reps, mod),
  1267. 'nothing to repeat', 3)
  1268. def test_multiple_repeat(self):
  1269. for outer_reps in '*', '+', '?', '{1,2}':
  1270. for outer_mod in '', '?', '+':
  1271. outer_op = outer_reps + outer_mod
  1272. for inner_reps in '*', '+', '?', '{1,2}':
  1273. for inner_mod in '', '?', '+':
  1274. if inner_mod + outer_reps in ('?', '+'):
  1275. continue
  1276. inner_op = inner_reps + inner_mod
  1277. self.checkPatternError(r'x%s%s' % (inner_op, outer_op),
  1278. 'multiple repeat', 1 + len(inner_op))
  1279. def test_unlimited_zero_width_repeat(self):
  1280. # Issue #9669
  1281. self.assertIsNone(re.match(r'(?:a?)*y', 'z'))
  1282. self.assertIsNone(re.match(r'(?:a?)+y', 'z'))
  1283. self.assertIsNone(re.match(r'(?:a?){2,}y', 'z'))
  1284. self.assertIsNone(re.match(r'(?:a?)*?y', 'z'))
  1285. self.assertIsNone(re.match(r'(?:a?)+?y', 'z'))
  1286. self.assertIsNone(re.match(r'(?:a?){2,}?y', 'z'))
  1287. def test_scanner(self):
  1288. def s_ident(scanner, token): return token
  1289. def s_operator(scanner, token): return "op%s" % token
  1290. def s_float(scanner, token): return float(token)
  1291. def s_int(scanner, token): return int(token)
  1292. scanner = Scanner([
  1293. (r"[a-zA-Z_]\w*", s_ident),
  1294. (r"\d+\.\d*", s_float),
  1295. (r"\d+", s_int),
  1296. (r"=|\+|-|\*|/", s_operator),
  1297. (r"\s+", None),
  1298. ])
  1299. self.assertTrue(scanner.scanner.scanner("").pattern)
  1300. self.assertEqual(scanner.scan("sum = 3*foo + 312.50 + bar"),
  1301. (['sum', 'op=', 3, 'op*', 'foo', 'op+', 312.5,
  1302. 'op+', 'bar'], ''))
  1303. def test_bug_448951(self):
  1304. # bug 448951 (similar to 429357, but with single char match)
  1305. # (Also test greedy matches.)
  1306. for op in '','?','*':
  1307. self.assertEqual(re.match(r'((.%s):)?z'%op, 'z').groups(),
  1308. (None, None))
  1309. self.assertEqual(re.match(r'((.%s):)?z'%op, 'a:z').groups(),
  1310. ('a:', 'a'))
  1311. def test_bug_725106(self):
  1312. # capturing groups in alternatives in repeats
  1313. self.assertEqual(re.match('^((a)|b)*', 'abc').groups(),
  1314. ('b', 'a'))
  1315. self.assertEqual(re.match('^(([ab])|c)*', 'abc').groups(),
  1316. ('c', 'b'))
  1317. self.assertEqual(re.match('^((d)|[ab])*', 'abc').groups(),
  1318. ('b', None))
  1319. self.assertEqual(re.match('^((a)c|[ab])*', 'abc').groups(),
  1320. ('b', None))
  1321. self.assertEqual(re.match('^((a)|b)*?c', 'abc').groups(),
  1322. ('b', 'a'))
  1323. self.assertEqual(re.match('^(([ab])|c)*?d', 'abcd').groups(),
  1324. ('c', 'b'))
  1325. self.assertEqual(re.match('^((d)|[ab])*?c', 'abc').groups(),
  1326. ('b', None))
  1327. self.assertEqual(re.match('^((a)c|[ab])*?c', 'abc').groups(),
  1328. ('b', None))
  1329. def test_bug_725149(self):
  1330. # mark_stack_base restoring before restoring marks
  1331. self.assertEqual(re.match('(a)(?:(?=(b)*)c)*', 'abb').groups(),
  1332. ('a', None))
  1333. self.assertEqual(re.match('(a)((?!(b)*))*', 'abb').groups(),
  1334. ('a', None, None))
  1335. def test_bug_764548(self):
  1336. # bug 764548, re.compile() barfs on str/unicode subclasses
  1337. class my_unicode(str): pass
  1338. pat = re.compile(my_unicode("abc"))
  1339. self.assertIsNone(pat.match("xyz"))
  1340. def test_finditer(self):
  1341. iter = re.finditer(r":+", "a:b::c:::d")
  1342. self.assertEqual([item.group(0) for item in iter],
  1343. [":", "::", ":::"])
  1344. pat = re.compile(r":+")
  1345. iter = pat.finditer("a:b::c:::d", 1, 10)
  1346. self.assertEqual([item.group(0) for item in iter],
  1347. [":", "::", ":::"])
  1348. pat = re.compile(r":+")
  1349. iter = pat.finditer("a:b::c:::d", pos=1, endpos=10)
  1350. self.assertEqual([item.group(0) for item in iter],
  1351. [":", "::", ":::"])
  1352. pat = re.compile(r":+")
  1353. iter = pat.finditer("a:b::c:::d", endpos=10, pos=1)
  1354. self.assertEqual([item.group(0) for item in iter],
  1355. [":", "::", ":::"])
  1356. pat = re.compile(r":+")
  1357. iter = pat.finditer("a:b::c:::d", pos=3, endpos=8)
  1358. self.assertEqual([item.group(0) for item in iter],
  1359. ["::", "::"])
  1360. def test_bug_926075(self):
  1361. self.assertIsNot(re.compile('bug_926075'),
  1362. re.compile(b'bug_926075'))
  1363. def test_bug_931848(self):
  1364. pattern = "[\u002E\u3002\uFF0E\uFF61]"
  1365. self.assertEqual(re.compile(pattern).split("a.b.c"),
  1366. ['a','b','c'])
  1367. def test_bug_581080(self):
  1368. iter = re.finditer(r"\s", "a b")
  1369. self.assertEqual(next(iter).span(), (1,2))
  1370. self.assertRaises(StopIteration, next, iter)
  1371. scanner = re.compile(r"\s").scanner("a b")
  1372. self.assertEqual(scanner.search().span(), (1, 2))
  1373. self.assertIsNone(scanner.search())
  1374. def test_bug_817234(self):
  1375. iter = re.finditer(r".*", "asdf")
  1376. self.assertEqual(next(iter).span(), (0, 4))
  1377. self.assertEqual(next(iter).span(), (4, 4))
  1378. self.assertRaises(StopIteration, next, iter)
  1379. def test_bug_6561(self):
  1380. # '\d' should match characters in Unicode category 'Nd'
  1381. # (Number, Decimal Digit), but not those in 'Nl' (Number,
  1382. # Letter) or 'No' (Number, Other).
  1383. decimal_digits = [
  1384. '\u0037', # '\N{DIGIT SEVEN}', category 'Nd'
  1385. '\u0e58', # '\N{THAI DIGIT SIX}', category 'Nd'
  1386. '\uff10', # '\N{FULLWIDTH DIGIT ZERO}', category 'Nd'
  1387. ]
  1388. for x in decimal_digits:
  1389. self.assertEqual(re.match(r'^\d$', x).group(0), x)
  1390. not_decimal_digits = [
  1391. '\u2165', # '\N{ROMAN NUMERAL SIX}', category 'Nl'
  1392. '\u3039', # '\N{HANGZHOU NUMERAL TWENTY}', category 'Nl'
  1393. '\u2082', # '\N{SUBSCRIPT TWO}', category 'No'
  1394. '\u32b4', # '\N{CIRCLED NUMBER THIRTY NINE}', category 'No'
  1395. ]
  1396. for x in not_decimal_digits:
  1397. self.assertIsNone(re.match(r'^\d$', x))
  1398. def test_empty_array(self):
  1399. # SF buf 1647541
  1400. import array
  1401. for typecode in 'bBuhHiIlLfd':
  1402. a = array.array(typecode)
  1403. self.assertIsNone(re.compile(b"bla").match(a))
  1404. self.assertEqual(re.compile(b"").match(a).groups(), ())
  1405. def test_inline_flags(self):
  1406. # Bug #1700
  1407. upper_char = '\u1ea0' # Latin Capital Letter A with Dot Below
  1408. lower_char = '\u1ea1' # Latin Small Letter A with Dot Below
  1409. p = re.compile('.' + upper_char, re.I | re.S)
  1410. q = p.match('\n' + lower_char)
  1411. self.assertTrue(q)
  1412. p = re.compile('.' + lower_char, re.I | re.S)
  1413. q = p.match('\n' + upper_char)
  1414. self.assertTrue(q)
  1415. p = re.compile('(?i).' + upper_char, re.S)
  1416. q = p.match('\n' + lower_char)
  1417. self.assertTrue(q)
  1418. p = re.compile('(?i).' + lower_char, re.S)
  1419. q = p.match('\n' + upper_char)
  1420. self.assertTrue(q)
  1421. p = re.compile('(?is).' + upper_char)
  1422. q = p.match('\n' + lower_char)
  1423. self.assertTrue(q)
  1424. p = re.compile('(?is).' + lower_char)
  1425. q = p.match('\n' + upper_char)
  1426. self.assertTrue(q)
  1427. p = re.compile('(?s)(?i).' + upper_char)
  1428. q = p.match('\n' + lower_char)
  1429. self.assertTrue(q)
  1430. p = re.compile('(?s)(?i).' + lower_char)
  1431. q = p.match('\n' + upper_char)
  1432. self.assertTrue(q)
  1433. self.assertTrue(re.match('(?ix) ' + upper_char, lower_char))
  1434. self.assertTrue(re.match('(?ix) ' + lower_char, upper_char))
  1435. self.assertTrue(re.match(' (?i) ' + upper_char, lower_char, re.X))
  1436. self.assertTrue(re.match('(?x) (?i) ' + upper_char, lower_char))
  1437. self.assertTrue(re.match(' (?x) (?i) ' + upper_char, lower_char, re.X))
  1438. msg = "global flags not at the start of the expression"
  1439. self.checkPatternError(upper_char + '(?i)', msg, 1)
  1440. # bpo-30605: Compiling a bytes instance regex was throwing a BytesWarning
  1441. with warnings.catch_warnings():
  1442. warnings.simplefilter('error', BytesWarning)
  1443. self.checkPatternError(b'A(?i)', msg, 1)
  1444. self.checkPatternError('(?s).(?i)' + upper_char, msg, 5)
  1445. self.checkPatternError('(?i) ' + upper_char + ' (?x)', msg, 7)
  1446. self.checkPatternError(' (?x) (?i) ' + upper_char, msg, 1)
  1447. self.checkPatternError('^(?i)' + upper_char, msg, 1)
  1448. self.checkPatternError('$|(?i)' + upper_char, msg, 2)
  1449. self.checkPatternError('(?:(?i)' + upper_char + ')', msg, 3)
  1450. self.checkPatternError('(^)?(?(1)(?i)' + upper_char + ')', msg, 9)
  1451. self.checkPatternError('($)?(?(1)|(?i)' + upper_char + ')', msg, 10)
  1452. def test_dollar_matches_twice(self):
  1453. r"""Test that $ does not include \n
  1454. $ matches the end of string, and just before the terminating \n"""
  1455. pattern = re.compile('$')
  1456. self.assertEqual(pattern.sub('#', 'a\nb\n'), 'a\nb#\n#')
  1457. self.assertEqual(pattern.sub('#', 'a\nb\nc'), 'a\nb\nc#')
  1458. self.assertEqual(pattern.sub('#', '\n'), '#\n#')
  1459. pattern = re.compile('$', re.MULTILINE)
  1460. self.assertEqual(pattern.sub('#', 'a\nb\n' ), 'a#\nb#\n#' )
  1461. self.assertEqual(pattern.sub('#', 'a\nb\nc'), 'a#\nb#\nc#')
  1462. self.assertEqual(pattern.sub('#', '\n'), '#\n#')
  1463. def test_bytes_str_mixing(self):
  1464. # Mixing str and bytes is disallowed
  1465. pat = re.compile('.')
  1466. bpat = re.compile(b'.')
  1467. self.assertRaises(TypeError, pat.match, b'b')
  1468. self.assertRaises(TypeError, bpat.match, 'b')
  1469. self.assertRaises(TypeError, pat.sub, b'b', 'c')
  1470. self.assertRaises(TypeError, pat.sub, 'b', b'c')
  1471. self.assertRaises(TypeError, pat.sub, b'b', b'c')
  1472. self.assertRaises(TypeError, bpat.sub, b'b', 'c')
  1473. self.assertRaises(TypeError, bpat.sub, 'b', b'c')
  1474. self.assertRaises(TypeError, bpat.sub, 'b', 'c')
  1475. def test_ascii_and_unicode_flag(self):
  1476. # String patterns
  1477. for flags in (0, re.UNICODE):
  1478. pat = re.compile('\xc0', flags | re.IGNORECASE)
  1479. self.assertTrue(pat.match('\xe0'))
  1480. pat = re.compile(r'\w', flags)
  1481. self.assertTrue(pat.match('\xe0'))
  1482. pat = re.compile('\xc0', re.ASCII | re.IGNORECASE)
  1483. self.assertIsNone(pat.match('\xe0'))
  1484. pat = re.compile('(?a)\xc0', re.IGNORECASE)
  1485. self.assertIsNone(pat.match('\xe0'))
  1486. pat = re.compile(r'\w', re.ASCII)
  1487. self.assertIsNone(pat.match('\xe0'))
  1488. pat = re.compile(r'(?a)\w')
  1489. self.assertIsNone(pat.match('\xe0'))
  1490. # Bytes patterns
  1491. for flags in (0, re.ASCII):
  1492. pat = re.compile(b'\xc0', flags | re.IGNORECASE)
  1493. self.assertIsNone(pat.match(b'\xe0'))
  1494. pat = re.compile(br'\w', flags)
  1495. self.assertIsNone(pat.match(b'\xe0'))
  1496. # Incompatibilities
  1497. self.assertRaises(ValueError, re.compile, br'\w', re.UNICODE)
  1498. self.assertRaises(re.error, re.compile, br'(?u)\w')
  1499. self.assertRaises(ValueError, re.compile, r'\w', re.UNICODE | re.ASCII)
  1500. self.assertRaises(ValueError, re.compile, r'(?u)\w', re.ASCII)
  1501. self.assertRaises(ValueError, re.compile, r'(?a)\w', re.UNICODE)
  1502. self.assertRaises(re.error, re.compile, r'(?au)\w')
  1503. def test_locale_flag(self):
  1504. enc = locale.getpreferredencoding()
  1505. # Search non-ASCII letter
  1506. for i in range(128, 256):
  1507. try:
  1508. c = bytes([i]).decode(enc)
  1509. sletter = c.lower()
  1510. if sletter == c: continue
  1511. bletter = sletter.encode(enc)
  1512. if len(bletter) != 1: continue
  1513. if bletter.decode(enc) != sletter: continue
  1514. bpat = re.escape(bytes([i]))
  1515. break
  1516. except (UnicodeError, TypeError):
  1517. pass
  1518. else:
  1519. bletter = None
  1520. bpat = b'A'
  1521. # Bytes patterns
  1522. pat = re.compile(bpat, re.LOCALE | re.IGNORECASE)
  1523. if bletter:
  1524. self.assertTrue(pat.match(bletter))
  1525. pat = re.compile(b'(?L)' + bpat, re.IGNORECASE)
  1526. if bletter:
  1527. self.assertTrue(pat.match(bletter))
  1528. pat = re.compile(bpat, re.IGNORECASE)
  1529. if bletter:
  1530. self.assertIsNone(pat.match(bletter))
  1531. pat = re.compile(br'\w', re.LOCALE)
  1532. if bletter:
  1533. self.assertTrue(pat.match(bletter))
  1534. pat = re.compile(br'(?L)\w')
  1535. if bletter:
  1536. self.assertTrue(pat.match(bletter))
  1537. pat = re.compile(br'\w')
  1538. if bletter:
  1539. self.assertIsNone(pat.match(bletter))
  1540. # Incompatibilities
  1541. self.assertRaises(ValueError, re.compile, '', re.LOCALE)
  1542. self.assertRaises(re.error, re.compile, '(?L)')
  1543. self.assertRaises(ValueError, re.compile, b'', re.LOCALE | re.ASCII)
  1544. self.assertRaises(ValueError, re.compile, b'(?L)', re.ASCII)
  1545. self.assertRaises(ValueError, re.compile, b'(?a)', re.LOCALE)
  1546. self.assertRaises(re.error, re.compile, b'(?aL)')
  1547. def test_scoped_flags(self):
  1548. self.assertTrue(re.match(r'(?i:a)b', 'Ab'))
  1549. self.assertIsNone(re.match(r'(?i:a)b', 'aB'))
  1550. self.assertIsNone(re.match(r'(?-i:a)b', 'Ab', re.IGNORECASE))
  1551. self.assertTrue(re.match(r'(?-i:a)b', 'aB', re.IGNORECASE))
  1552. self.assertIsNone(re.match(r'(?i:(?-i:a)b)', 'Ab'))
  1553. self.assertTrue(re.match(r'(?i:(?-i:a)b)', 'aB'))
  1554. self.assertTrue(re.match(r'\w(?a:\W)\w', '\xe0\xe0\xe0'))
  1555. self.assertTrue(re.match(r'(?a:\W(?u:\w)\W)', '\xe0\xe0\xe0'))
  1556. self.assertTrue(re.match(r'\W(?u:\w)\W', '\xe0\xe0\xe0', re.ASCII))
  1557. self.checkPatternError(r'(?a)(?-a:\w)',
  1558. "bad inline flags: cannot turn off flags 'a', 'u' and 'L'", 8)
  1559. self.checkPatternError(r'(?i-i:a)',
  1560. 'bad inline flags: flag turned on and off', 5)
  1561. self.checkPatternError(r'(?au:a)',
  1562. "bad inline flags: flags 'a', 'u' and 'L' are incompatible", 4)
  1563. self.checkPatternError(br'(?aL:a)',
  1564. "bad inline flags: flags 'a', 'u' and 'L' are incompatible", 4)
  1565. self.checkPatternError(r'(?-', 'missing flag', 3)
  1566. self.checkPatternError(r'(?-+', 'missing flag', 3)
  1567. self.checkPatternError(r'(?-z', 'unknown flag', 3)
  1568. self.checkPatternError(r'(?-i', 'missing :', 4)
  1569. self.checkPatternError(r'(?-i)', 'missing :', 4)
  1570. self.checkPatternError(r'(?-i+', 'missing :', 4)
  1571. self.checkPatternError(r'(?-iz', 'unknown flag', 4)
  1572. self.checkPatternError(r'(?i:', 'missing ), unterminated subpattern', 0)
  1573. self.checkPatternError(r'(?i', 'missing -, : or )', 3)
  1574. self.checkPatternError(r'(?i+', 'missing -, : or )', 3)
  1575. self.checkPatternError(r'(?iz', 'unknown flag', 3)
  1576. def test_ignore_spaces(self):
  1577. for space in " \t\n\r\v\f":
  1578. self.assertTrue(re.fullmatch(space + 'a', 'a', re.VERBOSE))
  1579. for space in b" ", b"\t", b"\n", b"\r", b"\v", b"\f":
  1580. self.assertTrue(re.fullmatch(space + b'a', b'a', re.VERBOSE))
  1581. self.assertTrue(re.fullmatch('(?x) a', 'a'))
  1582. self.assertTrue(re.fullmatch(' (?x) a', 'a', re.VERBOSE))
  1583. self.assertTrue(re.fullmatch('(?x) (?x) a', 'a'))
  1584. self.assertTrue(re.fullmatch(' a(?x: b) c', ' ab c'))
  1585. self.assertTrue(re.fullmatch(' a(?-x: b) c', 'a bc', re.VERBOSE))
  1586. self.assertTrue(re.fullmatch('(?x) a(?-x: b) c', 'a bc'))
  1587. self.assertTrue(re.fullmatch('(?x) a| b', 'a'))
  1588. self.assertTrue(re.fullmatch('(?x) a| b', 'b'))
  1589. def test_comments(self):
  1590. self.assertTrue(re.fullmatch('#x\na', 'a', re.VERBOSE))
  1591. self.assertTrue(re.fullmatch(b'#x\na', b'a', re.VERBOSE))
  1592. self.assertTrue(re.fullmatch('(?x)#x\na', 'a'))
  1593. self.assertTrue(re.fullmatch('#x\n(?x)#y\na', 'a', re.VERBOSE))
  1594. self.assertTrue(re.fullmatch('(?x)#x\n(?x)#y\na', 'a'))
  1595. self.assertTrue(re.fullmatch('#x\na(?x:#y\nb)#z\nc', '#x\nab#z\nc'))
  1596. self.assertTrue(re.fullmatch('#x\na(?-x:#y\nb)#z\nc', 'a#y\nbc',
  1597. re.VERBOSE))
  1598. self.assertTrue(re.fullmatch('(?x)#x\na(?-x:#y\nb)#z\nc', 'a#y\nbc'))
  1599. self.assertTrue(re.fullmatch('(?x)#x\na|#y\nb', 'a'))
  1600. self.assertTrue(re.fullmatch('(?x)#x\na|#y\nb', 'b'))
  1601. def test_bug_6509(self):
  1602. # Replacement strings of both types must parse properly.
  1603. # all strings
  1604. pat = re.compile(r'a(\w)')
  1605. self.assertEqual(pat.sub('b\\1', 'ac'), 'bc')
  1606. pat = re.compile('a(.)')
  1607. self.assertEqual(pat.sub('b\\1', 'a\u1234'), 'b\u1234')
  1608. pat = re.compile('..')
  1609. self.assertEqual(pat.sub(lambda m: 'str', 'a5'), 'str')
  1610. # all bytes
  1611. pat = re.compile(br'a(\w)')
  1612. self.assertEqual(pat.sub(b'b\\1', b'ac'), b'bc')
  1613. pat = re.compile(b'a(.)')
  1614. self.assertEqual(pat.sub(b'b\\1', b'a\xCD'), b'b\xCD')
  1615. pat = re.compile(b'..')
  1616. self.assertEqual(pat.sub(lambda m: b'bytes', b'a5'), b'bytes')
  1617. def test_dealloc(self):
  1618. # issue 3299: check for segfault in debug build
  1619. import _sre
  1620. # the overflow limit is different on wide and narrow builds and it
  1621. # depends on the definition of SRE_CODE (see sre.h).
  1622. # 2**128 should be big enough to overflow on both. For smaller values
  1623. # a RuntimeError is raised instead of OverflowError.
  1624. long_overflow = 2**128
  1625. self.assertRaises(TypeError, re.finditer, "a", {})
  1626. with self.assertRaises(OverflowError):
  1627. _sre.compile("abc", 0, [long_overflow], 0, {}, ())
  1628. with self.assertRaises(TypeError):
  1629. _sre.compile({}, 0, [], 0, [], [])
  1630. def test_search_dot_unicode(self):
  1631. self.assertTrue(re.search("123.*-", '123abc-'))
  1632. self.assertTrue(re.search("123.*-", '123\xe9-'))
  1633. self.assertTrue(re.search("123.*-", '123\u20ac-'))
  1634. self.assertTrue(re.search("123.*-", '123\U0010ffff-'))
  1635. self.assertTrue(re.search("123.*-", '123\xe9\u20ac\U0010ffff-'))
  1636. def test_compile(self):
  1637. # Test return value when given string and pattern as parameter
  1638. pattern = re.compile('random pattern')
  1639. self.assertIsInstance(pattern, re.Pattern)
  1640. same_pattern = re.compile(pattern)
  1641. self.assertIsInstance(same_pattern, re.Pattern)
  1642. self.assertIs(same_pattern, pattern)
  1643. # Test behaviour when not given a string or pattern as parameter
  1644. self.assertRaises(TypeError, re.compile, 0)
  1645. @bigmemtest(size=_2G, memuse=1)
  1646. def test_large_search(self, size):
  1647. # Issue #10182: indices were 32-bit-truncated.
  1648. s = 'a' * size
  1649. m = re.search('$', s)
  1650. self.assertIsNotNone(m)
  1651. self.assertEqual(m.start(), size)
  1652. self.assertEqual(m.end(), size)
  1653. # The huge memuse is because of re.sub() using a list and a join()
  1654. # to create the replacement result.
  1655. @bigmemtest(size=_2G, memuse=16 + 2)
  1656. def test_large_subn(self, size):
  1657. # Issue #10182: indices were 32-bit-truncated.
  1658. s = 'a' * size
  1659. r, n = re.subn('', '', s)
  1660. self.assertEqual(r, s)
  1661. self.assertEqual(n, size + 1)
  1662. def test_bug_16688(self):
  1663. # Issue 16688: Backreferences make case-insensitive regex fail on
  1664. # non-ASCII strings.
  1665. self.assertEqual(re.findall(r"(?i)(a)\1", "aa \u0100"), ['a'])
  1666. self.assertEqual(re.match(r"(?s).{1,3}", "\u0100\u0100").span(), (0, 2))
  1667. def test_repeat_minmax_overflow(self):
  1668. # Issue #13169
  1669. string = "x" * 100000
  1670. self.assertEqual(re.match(r".{65535}", string).span(), (0, 65535))
  1671. self.assertEqual(re.match(r".{,65535}", string).span(), (0, 65535))
  1672. self.assertEqual(re.match(r".{65535,}?", string).span(), (0, 65535))
  1673. self.assertEqual(re.match(r".{65536}", string).span(), (0, 65536))
  1674. self.assertEqual(re.match(r".{,65536}", string).span(), (0, 65536))
  1675. self.assertEqual(re.match(r".{65536,}?", string).span(), (0, 65536))
  1676. # 2**128 should be big enough to overflow both SRE_CODE and Py_ssize_t.
  1677. self.assertRaises(OverflowError, re.compile, r".{%d}" % 2**128)
  1678. self.assertRaises(OverflowError, re.compile, r".{,%d}" % 2**128)
  1679. self.assertRaises(OverflowError, re.compile, r".{%d,}?" % 2**128)
  1680. self.assertRaises(OverflowError, re.compile, r".{%d,%d}" % (2**129, 2**128))
  1681. @cpython_only
  1682. def test_repeat_minmax_overflow_maxrepeat(self):
  1683. try:
  1684. from _sre import MAXREPEAT
  1685. except ImportError:
  1686. self.skipTest('requires _sre.MAXREPEAT constant')
  1687. string = "x" * 100000
  1688. self.assertIsNone(re.match(r".{%d}" % (MAXREPEAT - 1), string))
  1689. self.assertEqual(re.match(r".{,%d}" % (MAXREPEAT - 1), string).span(),
  1690. (0, 100000))
  1691. self.assertIsNone(re.match(r".{%d,}?" % (MAXREPEAT - 1), string))
  1692. self.assertRaises(OverflowError, re.compile, r".{%d}" % MAXREPEAT)
  1693. self.assertRaises(OverflowError, re.compile, r".{,%d}" % MAXREPEAT)
  1694. self.assertRaises(OverflowError, re.compile, r".{%d,}?" % MAXREPEAT)
  1695. def test_backref_group_name_in_exception(self):
  1696. # Issue 17341: Poor error message when compiling invalid regex
  1697. self.checkPatternError('(?P=<foo>)',
  1698. "bad character in group name '<foo>'", 4)
  1699. def test_group_name_in_exception(self):
  1700. # Issue 17341: Poor error message when compiling invalid regex
  1701. self.checkPatternError('(?P<?foo>)',
  1702. "bad character in group name '?foo'", 4)
  1703. def test_issue17998(self):
  1704. for reps in '*', '+', '?', '{1}':
  1705. for mod in '', '?':
  1706. pattern = '.' + reps + mod + 'yz'
  1707. self.assertEqual(re.compile(pattern, re.S).findall('xyz'),
  1708. ['xyz'], msg=pattern)
  1709. pattern = pattern.encode()
  1710. self.assertEqual(re.compile(pattern, re.S).findall(b'xyz'),
  1711. [b'xyz'], msg=pattern)
  1712. def test_match_repr(self):
  1713. for string in '[abracadabra]', S('[abracadabra]'):
  1714. m = re.search(r'(.+)(.*?)\1', string)
  1715. pattern = r"<(%s\.)?%s object; span=\(1, 12\), match='abracadabra'>" % (
  1716. type(m).__module__, type(m).__qualname__
  1717. )
  1718. self.assertRegex(repr(m), pattern)
  1719. for string in (b'[abracadabra]', B(b'[abracadabra]'),
  1720. bytearray(b'[abracadabra]'),
  1721. memoryview(b'[abracadabra]')):
  1722. m = re.search(br'(.+)(.*?)\1', string)
  1723. pattern = r"<(%s\.)?%s object; span=\(1, 12\), match=b'abracadabra'>" % (
  1724. type(m).__module__, type(m).__qualname__
  1725. )
  1726. self.assertRegex(repr(m), pattern)
  1727. first, second = list(re.finditer("(aa)|(bb)", "aa bb"))
  1728. pattern = r"<(%s\.)?%s object; span=\(0, 2\), match='aa'>" % (
  1729. type(second).__module__, type(second).__qualname__
  1730. )
  1731. self.assertRegex(repr(first), pattern)
  1732. pattern = r"<(%s\.)?%s object; span=\(3, 5\), match='bb'>" % (
  1733. type(second).__module__, type(second).__qualname__
  1734. )
  1735. self.assertRegex(repr(second), pattern)
  1736. def test_zerowidth(self):
  1737. # Issues 852532, 1647489, 3262, 25054.
  1738. self.assertEqual(re.split(r"\b", "a::bc"), ['', 'a', '::', 'bc', ''])
  1739. self.assertEqual(re.split(r"\b|:+", "a::bc"), ['', 'a', '', '', 'bc', ''])
  1740. self.assertEqual(re.split(r"(?<!\w)(?=\w)|:+", "a::bc"), ['', 'a', '', 'bc'])
  1741. self.assertEqual(re.split(r"(?<=\w)(?!\w)|:+", "a::bc"), ['a', '', 'bc', ''])
  1742. self.assertEqual(re.sub(r"\b", "-", "a::bc"), '-a-::-bc-')
  1743. self.assertEqual(re.sub(r"\b|:+", "-", "a::bc"), '-a---bc-')
  1744. self.assertEqual(re.sub(r"(\b|:+)", r"[\1]", "a::bc"), '[]a[][::][]bc[]')
  1745. self.assertEqual(re.findall(r"\b|:+", "a::bc"), ['', '', '::', '', ''])
  1746. self.assertEqual(re.findall(r"\b|\w+", "a::bc"),
  1747. ['', 'a', '', '', 'bc', ''])
  1748. self.assertEqual([m.span() for m in re.finditer(r"\b|:+", "a::bc")],
  1749. [(0, 0), (1, 1), (1, 3), (3, 3), (5, 5)])
  1750. self.assertEqual([m.span() for m in re.finditer(r"\b|\w+", "a::bc")],
  1751. [(0, 0), (0, 1), (1, 1), (3, 3), (3, 5), (5, 5)])
  1752. def test_bug_2537(self):
  1753. # issue 2537: empty submatches
  1754. for outer_op in ('{0,}', '*', '+', '{1,187}'):
  1755. for inner_op in ('{0,}', '*', '?'):
  1756. r = re.compile("^((x|y)%s)%s" % (inner_op, outer_op))
  1757. m = r.match("xyyzy")
  1758. self.assertEqual(m.group(0), "xyy")
  1759. self.assertEqual(m.group(1), "")
  1760. self.assertEqual(m.group(2), "y")
  1761. def test_keyword_parameters(self):
  1762. # Issue #20283: Accepting the string keyword parameter.
  1763. pat = re.compile(r'(ab)')
  1764. self.assertEqual(
  1765. pat.match(string='abracadabra', pos=7, endpos=10).span(), (7, 9))
  1766. self.assertEqual(
  1767. pat.fullmatch(string='abracadabra', pos=7, endpos=9).span(), (7, 9))
  1768. self.assertEqual(
  1769. pat.search(string='abracadabra', pos=3, endpos=10).span(), (7, 9))
  1770. self.assertEqual(
  1771. pat.findall(string='abracadabra', pos=3, endpos=10), ['ab'])
  1772. self.assertEqual(
  1773. pat.split(string='abracadabra', maxsplit=1),
  1774. ['', 'ab', 'racadabra'])
  1775. self.assertEqual(
  1776. pat.scanner(string='abracadabra', pos=3, endpos=10).search().span(),
  1777. (7, 9))
  1778. def test_bug_20998(self):
  1779. # Issue #20998: Fullmatch of repeated single character pattern
  1780. # with ignore case.
  1781. self.assertEqual(re.fullmatch('[a-c]+', 'ABC', re.I).span(), (0, 3))
  1782. @unittest.skipIf(
  1783. is_emscripten or is_wasi,
  1784. "musl libc issue on Emscripten/WASI, bpo-46390"
  1785. )
  1786. def test_locale_caching(self):
  1787. # Issue #22410
  1788. oldlocale = locale.setlocale(locale.LC_CTYPE)
  1789. self.addCleanup(locale.setlocale, locale.LC_CTYPE, oldlocale)
  1790. for loc in 'en_US.iso88591', 'en_US.utf8':
  1791. try:
  1792. locale.setlocale(locale.LC_CTYPE, loc)
  1793. except locale.Error:
  1794. # Unsupported locale on this system
  1795. self.skipTest('test needs %s locale' % loc)
  1796. re.purge()
  1797. self.check_en_US_iso88591()
  1798. self.check_en_US_utf8()
  1799. re.purge()
  1800. self.check_en_US_utf8()
  1801. self.check_en_US_iso88591()
  1802. def check_en_US_iso88591(self):
  1803. locale.setlocale(locale.LC_CTYPE, 'en_US.iso88591')
  1804. self.assertTrue(re.match(b'\xc5\xe5', b'\xc5\xe5', re.L|re.I))
  1805. self.assertTrue(re.match(b'\xc5', b'\xe5', re.L|re.I))
  1806. self.assertTrue(re.match(b'\xe5', b'\xc5', re.L|re.I))
  1807. self.assertTrue(re.match(b'(?Li)\xc5\xe5', b'\xc5\xe5'))
  1808. self.assertTrue(re.match(b'(?Li)\xc5', b'\xe5'))
  1809. self.assertTrue(re.match(b'(?Li)\xe5', b'\xc5'))
  1810. def check_en_US_utf8(self):
  1811. locale.setlocale(locale.LC_CTYPE, 'en_US.utf8')
  1812. self.assertTrue(re.match(b'\xc5\xe5', b'\xc5\xe5', re.L|re.I))
  1813. self.assertIsNone(re.match(b'\xc5', b'\xe5', re.L|re.I))
  1814. self.assertIsNone(re.match(b'\xe5', b'\xc5', re.L|re.I))
  1815. self.assertTrue(re.match(b'(?Li)\xc5\xe5', b'\xc5\xe5'))
  1816. self.assertIsNone(re.match(b'(?Li)\xc5', b'\xe5'))
  1817. self.assertIsNone(re.match(b'(?Li)\xe5', b'\xc5'))
  1818. @unittest.skipIf(
  1819. is_emscripten or is_wasi,
  1820. "musl libc issue on Emscripten/WASI, bpo-46390"
  1821. )
  1822. def test_locale_compiled(self):
  1823. oldlocale = locale.setlocale(locale.LC_CTYPE)
  1824. self.addCleanup(locale.setlocale, locale.LC_CTYPE, oldlocale)
  1825. for loc in 'en_US.iso88591', 'en_US.utf8':
  1826. try:
  1827. locale.setlocale(locale.LC_CTYPE, loc)
  1828. except locale.Error:
  1829. # Unsupported locale on this system
  1830. self.skipTest('test needs %s locale' % loc)
  1831. locale.setlocale(locale.LC_CTYPE, 'en_US.iso88591')
  1832. p1 = re.compile(b'\xc5\xe5', re.L|re.I)
  1833. p2 = re.compile(b'[a\xc5][a\xe5]', re.L|re.I)
  1834. p3 = re.compile(b'[az\xc5][az\xe5]', re.L|re.I)
  1835. p4 = re.compile(b'[^\xc5][^\xe5]', re.L|re.I)
  1836. for p in p1, p2, p3:
  1837. self.assertTrue(p.match(b'\xc5\xe5'))
  1838. self.assertTrue(p.match(b'\xe5\xe5'))
  1839. self.assertTrue(p.match(b'\xc5\xc5'))
  1840. self.assertIsNone(p4.match(b'\xe5\xc5'))
  1841. self.assertIsNone(p4.match(b'\xe5\xe5'))
  1842. self.assertIsNone(p4.match(b'\xc5\xc5'))
  1843. locale.setlocale(locale.LC_CTYPE, 'en_US.utf8')
  1844. for p in p1, p2, p3:
  1845. self.assertTrue(p.match(b'\xc5\xe5'))
  1846. self.assertIsNone(p.match(b'\xe5\xe5'))
  1847. self.assertIsNone(p.match(b'\xc5\xc5'))
  1848. self.assertTrue(p4.match(b'\xe5\xc5'))
  1849. self.assertIsNone(p4.match(b'\xe5\xe5'))
  1850. self.assertIsNone(p4.match(b'\xc5\xc5'))
  1851. def test_error(self):
  1852. with self.assertRaises(re.error) as cm:
  1853. re.compile('(\u20ac))')
  1854. err = cm.exception
  1855. self.assertIsInstance(err.pattern, str)
  1856. self.assertEqual(err.pattern, '(\u20ac))')
  1857. self.assertEqual(err.pos, 3)
  1858. self.assertEqual(err.lineno, 1)
  1859. self.assertEqual(err.colno, 4)
  1860. self.assertIn(err.msg, str(err))
  1861. self.assertIn(' at position 3', str(err))
  1862. self.assertNotIn(' at position 3', err.msg)
  1863. # Bytes pattern
  1864. with self.assertRaises(re.error) as cm:
  1865. re.compile(b'(\xa4))')
  1866. err = cm.exception
  1867. self.assertIsInstance(err.pattern, bytes)
  1868. self.assertEqual(err.pattern, b'(\xa4))')
  1869. self.assertEqual(err.pos, 3)
  1870. # Multiline pattern
  1871. with self.assertRaises(re.error) as cm:
  1872. re.compile("""
  1873. (
  1874. abc
  1875. )
  1876. )
  1877. (
  1878. """, re.VERBOSE)
  1879. err = cm.exception
  1880. self.assertEqual(err.pos, 77)
  1881. self.assertEqual(err.lineno, 5)
  1882. self.assertEqual(err.colno, 17)
  1883. self.assertIn(err.msg, str(err))
  1884. self.assertIn(' at position 77', str(err))
  1885. self.assertIn('(line 5, column 17)', str(err))
  1886. def test_misc_errors(self):
  1887. self.checkPatternError(r'(', 'missing ), unterminated subpattern', 0)
  1888. self.checkPatternError(r'((a|b)', 'missing ), unterminated subpattern', 0)
  1889. self.checkPatternError(r'(a|b))', 'unbalanced parenthesis', 5)
  1890. self.checkPatternError(r'(?P', 'unexpected end of pattern', 3)
  1891. self.checkPatternError(r'(?z)', 'unknown extension ?z', 1)
  1892. self.checkPatternError(r'(?iz)', 'unknown flag', 3)
  1893. self.checkPatternError(r'(?i', 'missing -, : or )', 3)
  1894. self.checkPatternError(r'(?#abc', 'missing ), unterminated comment', 0)
  1895. self.checkPatternError(r'(?<', 'unexpected end of pattern', 3)
  1896. self.checkPatternError(r'(?<>)', 'unknown extension ?<>', 1)
  1897. self.checkPatternError(r'(?', 'unexpected end of pattern', 2)
  1898. def test_enum(self):
  1899. # Issue #28082: Check that str(flag) returns a human readable string
  1900. # instead of an integer
  1901. self.assertIn('ASCII', str(re.A))
  1902. self.assertIn('DOTALL', str(re.S))
  1903. def test_pattern_compare(self):
  1904. pattern1 = re.compile('abc', re.IGNORECASE)
  1905. # equal to itself
  1906. self.assertEqual(pattern1, pattern1)
  1907. self.assertFalse(pattern1 != pattern1)
  1908. # equal
  1909. re.purge()
  1910. pattern2 = re.compile('abc', re.IGNORECASE)
  1911. self.assertEqual(hash(pattern2), hash(pattern1))
  1912. self.assertEqual(pattern2, pattern1)
  1913. # not equal: different pattern
  1914. re.purge()
  1915. pattern3 = re.compile('XYZ', re.IGNORECASE)
  1916. # Don't test hash(pattern3) != hash(pattern1) because there is no
  1917. # warranty that hash values are different
  1918. self.assertNotEqual(pattern3, pattern1)
  1919. # not equal: different flag (flags=0)
  1920. re.purge()
  1921. pattern4 = re.compile('abc')
  1922. self.assertNotEqual(pattern4, pattern1)
  1923. # only == and != comparison operators are supported
  1924. with self.assertRaises(TypeError):
  1925. pattern1 < pattern2
  1926. def test_pattern_compare_bytes(self):
  1927. pattern1 = re.compile(b'abc')
  1928. # equal: test bytes patterns
  1929. re.purge()
  1930. pattern2 = re.compile(b'abc')
  1931. self.assertEqual(hash(pattern2), hash(pattern1))
  1932. self.assertEqual(pattern2, pattern1)
  1933. # not equal: pattern of a different types (str vs bytes),
  1934. # comparison must not raise a BytesWarning
  1935. re.purge()
  1936. pattern3 = re.compile('abc')
  1937. with warnings.catch_warnings():
  1938. warnings.simplefilter('error', BytesWarning)
  1939. self.assertNotEqual(pattern3, pattern1)
  1940. def test_bug_29444(self):
  1941. s = bytearray(b'abcdefgh')
  1942. m = re.search(b'[a-h]+', s)
  1943. m2 = re.search(b'[e-h]+', s)
  1944. self.assertEqual(m.group(), b'abcdefgh')
  1945. self.assertEqual(m2.group(), b'efgh')
  1946. s[:] = b'xyz'
  1947. self.assertEqual(m.group(), b'xyz')
  1948. self.assertEqual(m2.group(), b'')
  1949. def test_bug_34294(self):
  1950. # Issue 34294: wrong capturing groups
  1951. # exists since Python 2
  1952. s = "a\tx"
  1953. p = r"\b(?=(\t)|(x))x"
  1954. self.assertEqual(re.search(p, s).groups(), (None, 'x'))
  1955. # introduced in Python 3.7.0
  1956. s = "ab"
  1957. p = r"(?=(.)(.)?)"
  1958. self.assertEqual(re.findall(p, s),
  1959. [('a', 'b'), ('b', '')])
  1960. self.assertEqual([m.groups() for m in re.finditer(p, s)],
  1961. [('a', 'b'), ('b', None)])
  1962. # test-cases provided by issue34294, introduced in Python 3.7.0
  1963. p = r"(?=<(?P<tag>\w+)/?>(?:(?P<text>.+?)</(?P=tag)>)?)"
  1964. s = "<test><foo2/></test>"
  1965. self.assertEqual(re.findall(p, s),
  1966. [('test', '<foo2/>'), ('foo2', '')])
  1967. self.assertEqual([m.groupdict() for m in re.finditer(p, s)],
  1968. [{'tag': 'test', 'text': '<foo2/>'},
  1969. {'tag': 'foo2', 'text': None}])
  1970. s = "<test>Hello</test><foo/>"
  1971. self.assertEqual([m.groupdict() for m in re.finditer(p, s)],
  1972. [{'tag': 'test', 'text': 'Hello'},
  1973. {'tag': 'foo', 'text': None}])
  1974. s = "<test>Hello</test><foo/><foo/>"
  1975. self.assertEqual([m.groupdict() for m in re.finditer(p, s)],
  1976. [{'tag': 'test', 'text': 'Hello'},
  1977. {'tag': 'foo', 'text': None},
  1978. {'tag': 'foo', 'text': None}])
  1979. def test_MARK_PUSH_macro_bug(self):
  1980. # issue35859, MARK_PUSH() macro didn't protect MARK-0 if it
  1981. # was the only available mark.
  1982. self.assertEqual(re.match(r'(ab|a)*?b', 'ab').groups(), ('a',))
  1983. self.assertEqual(re.match(r'(ab|a)+?b', 'ab').groups(), ('a',))
  1984. self.assertEqual(re.match(r'(ab|a){0,2}?b', 'ab').groups(), ('a',))
  1985. self.assertEqual(re.match(r'(.b|a)*?b', 'ab').groups(), ('a',))
  1986. def test_MIN_UNTIL_mark_bug(self):
  1987. # Fixed in issue35859, reported in issue9134.
  1988. # JUMP_MIN_UNTIL_2 should MARK_PUSH() if in a repeat
  1989. s = 'axxzbcz'
  1990. p = r'(?:(?:a|bc)*?(xx)??z)*'
  1991. self.assertEqual(re.match(p, s).groups(), ('xx',))
  1992. # test-case provided by issue9134
  1993. s = 'xtcxyzxc'
  1994. p = r'((x|yz)+?(t)??c)*'
  1995. m = re.match(p, s)
  1996. self.assertEqual(m.span(), (0, 8))
  1997. self.assertEqual(m.span(2), (6, 7))
  1998. self.assertEqual(m.groups(), ('xyzxc', 'x', 't'))
  1999. def test_REPEAT_ONE_mark_bug(self):
  2000. # issue35859
  2001. # JUMP_REPEAT_ONE_1 should MARK_PUSH() if in a repeat
  2002. s = 'aabaab'
  2003. p = r'(?:[^b]*a(?=(b)|(a))ab)*'
  2004. m = re.match(p, s)
  2005. self.assertEqual(m.span(), (0, 6))
  2006. self.assertEqual(m.span(2), (4, 5))
  2007. self.assertEqual(m.groups(), (None, 'a'))
  2008. # JUMP_REPEAT_ONE_2 should MARK_PUSH() if in a repeat
  2009. s = 'abab'
  2010. p = r'(?:[^b]*(?=(b)|(a))ab)*'
  2011. m = re.match(p, s)
  2012. self.assertEqual(m.span(), (0, 4))
  2013. self.assertEqual(m.span(2), (2, 3))
  2014. self.assertEqual(m.groups(), (None, 'a'))
  2015. self.assertEqual(re.match(r'(ab?)*?b', 'ab').groups(), ('a',))
  2016. def test_MIN_REPEAT_ONE_mark_bug(self):
  2017. # issue35859
  2018. # JUMP_MIN_REPEAT_ONE should MARK_PUSH() if in a repeat
  2019. s = 'abab'
  2020. p = r'(?:.*?(?=(a)|(b))b)*'
  2021. m = re.match(p, s)
  2022. self.assertEqual(m.span(), (0, 4))
  2023. self.assertEqual(m.span(2), (3, 4))
  2024. self.assertEqual(m.groups(), (None, 'b'))
  2025. s = 'axxzaz'
  2026. p = r'(?:a*?(xx)??z)*'
  2027. self.assertEqual(re.match(p, s).groups(), ('xx',))
  2028. def test_ASSERT_NOT_mark_bug(self):
  2029. # Fixed in issue35859, reported in issue725149.
  2030. # JUMP_ASSERT_NOT should LASTMARK_SAVE()
  2031. self.assertEqual(re.match(r'(?!(..)c)', 'ab').groups(), (None,))
  2032. # JUMP_ASSERT_NOT should MARK_PUSH() if in a repeat
  2033. m = re.match(r'((?!(ab)c)(.))*', 'abab')
  2034. self.assertEqual(m.span(), (0, 4))
  2035. self.assertEqual(m.span(1), (3, 4))
  2036. self.assertEqual(m.span(3), (3, 4))
  2037. self.assertEqual(m.groups(), ('b', None, 'b'))
  2038. def test_bug_40736(self):
  2039. with self.assertRaisesRegex(TypeError, "got 'int'"):
  2040. re.search("x*", 5)
  2041. with self.assertRaisesRegex(TypeError, "got 'type'"):
  2042. re.search("x*", type)
  2043. def test_search_anchor_at_beginning(self):
  2044. s = 'x'*10**7
  2045. start = time.perf_counter()
  2046. for p in r'\Ay', r'^y':
  2047. self.assertIsNone(re.search(p, s))
  2048. self.assertEqual(re.split(p, s), [s])
  2049. self.assertEqual(re.findall(p, s), [])
  2050. self.assertEqual(list(re.finditer(p, s)), [])
  2051. self.assertEqual(re.sub(p, '', s), s)
  2052. t = time.perf_counter() - start
  2053. # Without optimization it takes 1 second on my computer.
  2054. # With optimization -- 0.0003 seconds.
  2055. self.assertLess(t, 0.1)
  2056. def test_possessive_quantifiers(self):
  2057. """Test Possessive Quantifiers
  2058. Test quantifiers of the form @+ for some repetition operator @,
  2059. e.g. x{3,5}+ meaning match from 3 to 5 greadily and proceed
  2060. without creating a stack frame for rolling the stack back and
  2061. trying 1 or more fewer matches."""
  2062. self.assertIsNone(re.match('e*+e', 'eeee'))
  2063. self.assertEqual(re.match('e++a', 'eeea').group(0), 'eeea')
  2064. self.assertEqual(re.match('e?+a', 'ea').group(0), 'ea')
  2065. self.assertEqual(re.match('e{2,4}+a', 'eeea').group(0), 'eeea')
  2066. self.assertIsNone(re.match('(.)++.', 'ee'))
  2067. self.assertEqual(re.match('(ae)*+a', 'aea').groups(), ('ae',))
  2068. self.assertEqual(re.match('([ae][ae])?+a', 'aea').groups(),
  2069. ('ae',))
  2070. self.assertEqual(re.match('(e?){2,4}+a', 'eeea').groups(),
  2071. ('',))
  2072. self.assertEqual(re.match('()*+a', 'a').groups(), ('',))
  2073. self.assertEqual(re.search('x*+', 'axx').span(), (0, 0))
  2074. self.assertEqual(re.search('x++', 'axx').span(), (1, 3))
  2075. self.assertEqual(re.match('a*+', 'xxx').span(), (0, 0))
  2076. self.assertEqual(re.match('x*+', 'xxxa').span(), (0, 3))
  2077. self.assertIsNone(re.match('a++', 'xxx'))
  2078. self.assertIsNone(re.match(r"^(\w){1}+$", "abc"))
  2079. self.assertIsNone(re.match(r"^(\w){1,2}+$", "abc"))
  2080. self.assertEqual(re.match(r"^(\w){3}+$", "abc").group(1), "c")
  2081. self.assertEqual(re.match(r"^(\w){1,3}+$", "abc").group(1), "c")
  2082. self.assertEqual(re.match(r"^(\w){1,4}+$", "abc").group(1), "c")
  2083. self.assertIsNone(re.match("^x{1}+$", "xxx"))
  2084. self.assertIsNone(re.match("^x{1,2}+$", "xxx"))
  2085. self.assertTrue(re.match("^x{3}+$", "xxx"))
  2086. self.assertTrue(re.match("^x{1,3}+$", "xxx"))
  2087. self.assertTrue(re.match("^x{1,4}+$", "xxx"))
  2088. self.assertIsNone(re.match("^x{}+$", "xxx"))
  2089. self.assertTrue(re.match("^x{}+$", "x{}"))
  2090. def test_fullmatch_possessive_quantifiers(self):
  2091. self.assertTrue(re.fullmatch(r'a++', 'a'))
  2092. self.assertTrue(re.fullmatch(r'a*+', 'a'))
  2093. self.assertTrue(re.fullmatch(r'a?+', 'a'))
  2094. self.assertTrue(re.fullmatch(r'a{1,3}+', 'a'))
  2095. self.assertIsNone(re.fullmatch(r'a++', 'ab'))
  2096. self.assertIsNone(re.fullmatch(r'a*+', 'ab'))
  2097. self.assertIsNone(re.fullmatch(r'a?+', 'ab'))
  2098. self.assertIsNone(re.fullmatch(r'a{1,3}+', 'ab'))
  2099. self.assertTrue(re.fullmatch(r'a++b', 'ab'))
  2100. self.assertTrue(re.fullmatch(r'a*+b', 'ab'))
  2101. self.assertTrue(re.fullmatch(r'a?+b', 'ab'))
  2102. self.assertTrue(re.fullmatch(r'a{1,3}+b', 'ab'))
  2103. self.assertTrue(re.fullmatch(r'(?:ab)++', 'ab'))
  2104. self.assertTrue(re.fullmatch(r'(?:ab)*+', 'ab'))
  2105. self.assertTrue(re.fullmatch(r'(?:ab)?+', 'ab'))
  2106. self.assertTrue(re.fullmatch(r'(?:ab){1,3}+', 'ab'))
  2107. self.assertIsNone(re.fullmatch(r'(?:ab)++', 'abc'))
  2108. self.assertIsNone(re.fullmatch(r'(?:ab)*+', 'abc'))
  2109. self.assertIsNone(re.fullmatch(r'(?:ab)?+', 'abc'))
  2110. self.assertIsNone(re.fullmatch(r'(?:ab){1,3}+', 'abc'))
  2111. self.assertTrue(re.fullmatch(r'(?:ab)++c', 'abc'))
  2112. self.assertTrue(re.fullmatch(r'(?:ab)*+c', 'abc'))
  2113. self.assertTrue(re.fullmatch(r'(?:ab)?+c', 'abc'))
  2114. self.assertTrue(re.fullmatch(r'(?:ab){1,3}+c', 'abc'))
  2115. def test_findall_possessive_quantifiers(self):
  2116. self.assertEqual(re.findall(r'a++', 'aab'), ['aa'])
  2117. self.assertEqual(re.findall(r'a*+', 'aab'), ['aa', '', ''])
  2118. self.assertEqual(re.findall(r'a?+', 'aab'), ['a', 'a', '', ''])
  2119. self.assertEqual(re.findall(r'a{1,3}+', 'aab'), ['aa'])
  2120. self.assertEqual(re.findall(r'(?:ab)++', 'ababc'), ['abab'])
  2121. self.assertEqual(re.findall(r'(?:ab)*+', 'ababc'), ['abab', '', ''])
  2122. self.assertEqual(re.findall(r'(?:ab)?+', 'ababc'), ['ab', 'ab', '', ''])
  2123. self.assertEqual(re.findall(r'(?:ab){1,3}+', 'ababc'), ['abab'])
  2124. def test_atomic_grouping(self):
  2125. """Test Atomic Grouping
  2126. Test non-capturing groups of the form (?>...), which does
  2127. not maintain any stack point created within the group once the
  2128. group is finished being evaluated."""
  2129. pattern1 = re.compile(r'a(?>bc|b)c')
  2130. self.assertIsNone(pattern1.match('abc'))
  2131. self.assertTrue(pattern1.match('abcc'))
  2132. self.assertIsNone(re.match(r'(?>.*).', 'abc'))
  2133. self.assertTrue(re.match(r'(?>x)++', 'xxx'))
  2134. self.assertTrue(re.match(r'(?>x++)', 'xxx'))
  2135. self.assertIsNone(re.match(r'(?>x)++x', 'xxx'))
  2136. self.assertIsNone(re.match(r'(?>x++)x', 'xxx'))
  2137. def test_fullmatch_atomic_grouping(self):
  2138. self.assertTrue(re.fullmatch(r'(?>a+)', 'a'))
  2139. self.assertTrue(re.fullmatch(r'(?>a*)', 'a'))
  2140. self.assertTrue(re.fullmatch(r'(?>a?)', 'a'))
  2141. self.assertTrue(re.fullmatch(r'(?>a{1,3})', 'a'))
  2142. self.assertIsNone(re.fullmatch(r'(?>a+)', 'ab'))
  2143. self.assertIsNone(re.fullmatch(r'(?>a*)', 'ab'))
  2144. self.assertIsNone(re.fullmatch(r'(?>a?)', 'ab'))
  2145. self.assertIsNone(re.fullmatch(r'(?>a{1,3})', 'ab'))
  2146. self.assertTrue(re.fullmatch(r'(?>a+)b', 'ab'))
  2147. self.assertTrue(re.fullmatch(r'(?>a*)b', 'ab'))
  2148. self.assertTrue(re.fullmatch(r'(?>a?)b', 'ab'))
  2149. self.assertTrue(re.fullmatch(r'(?>a{1,3})b', 'ab'))
  2150. self.assertTrue(re.fullmatch(r'(?>(?:ab)+)', 'ab'))
  2151. self.assertTrue(re.fullmatch(r'(?>(?:ab)*)', 'ab'))
  2152. self.assertTrue(re.fullmatch(r'(?>(?:ab)?)', 'ab'))
  2153. self.assertTrue(re.fullmatch(r'(?>(?:ab){1,3})', 'ab'))
  2154. self.assertIsNone(re.fullmatch(r'(?>(?:ab)+)', 'abc'))
  2155. self.assertIsNone(re.fullmatch(r'(?>(?:ab)*)', 'abc'))
  2156. self.assertIsNone(re.fullmatch(r'(?>(?:ab)?)', 'abc'))
  2157. self.assertIsNone(re.fullmatch(r'(?>(?:ab){1,3})', 'abc'))
  2158. self.assertTrue(re.fullmatch(r'(?>(?:ab)+)c', 'abc'))
  2159. self.assertTrue(re.fullmatch(r'(?>(?:ab)*)c', 'abc'))
  2160. self.assertTrue(re.fullmatch(r'(?>(?:ab)?)c', 'abc'))
  2161. self.assertTrue(re.fullmatch(r'(?>(?:ab){1,3})c', 'abc'))
  2162. def test_findall_atomic_grouping(self):
  2163. self.assertEqual(re.findall(r'(?>a+)', 'aab'), ['aa'])
  2164. self.assertEqual(re.findall(r'(?>a*)', 'aab'), ['aa', '', ''])
  2165. self.assertEqual(re.findall(r'(?>a?)', 'aab'), ['a', 'a', '', ''])
  2166. self.assertEqual(re.findall(r'(?>a{1,3})', 'aab'), ['aa'])
  2167. self.assertEqual(re.findall(r'(?>(?:ab)+)', 'ababc'), ['abab'])
  2168. self.assertEqual(re.findall(r'(?>(?:ab)*)', 'ababc'), ['abab', '', ''])
  2169. self.assertEqual(re.findall(r'(?>(?:ab)?)', 'ababc'), ['ab', 'ab', '', ''])
  2170. self.assertEqual(re.findall(r'(?>(?:ab){1,3})', 'ababc'), ['abab'])
  2171. def test_bug_gh91616(self):
  2172. self.assertTrue(re.fullmatch(r'(?s:(?>.*?\.).*)\Z', "a.txt")) # reproducer
  2173. self.assertTrue(re.fullmatch(r'(?s:(?=(?P<g0>.*?\.))(?P=g0).*)\Z', "a.txt"))
  2174. def test_template_function_and_flag_is_deprecated(self):
  2175. with self.assertWarns(DeprecationWarning) as cm:
  2176. template_re1 = re.template(r'a')
  2177. self.assertIn('re.template()', str(cm.warning))
  2178. self.assertIn('is deprecated', str(cm.warning))
  2179. self.assertIn('function', str(cm.warning))
  2180. self.assertNotIn('flag', str(cm.warning))
  2181. with self.assertWarns(DeprecationWarning) as cm:
  2182. # we deliberately use more flags here to test that that still
  2183. # triggers the warning
  2184. # if paranoid, we could test multiple different combinations,
  2185. # but it's probably not worth it
  2186. template_re2 = re.compile(r'a', flags=re.TEMPLATE|re.UNICODE)
  2187. self.assertIn('re.TEMPLATE', str(cm.warning))
  2188. self.assertIn('is deprecated', str(cm.warning))
  2189. self.assertIn('flag', str(cm.warning))
  2190. self.assertNotIn('function', str(cm.warning))
  2191. # while deprecated, is should still function
  2192. self.assertEqual(template_re1, template_re2)
  2193. self.assertTrue(template_re1.match('ahoy'))
  2194. self.assertFalse(template_re1.match('nope'))
  2195. @unittest.skipIf(multiprocessing is None, 'test requires multiprocessing')
  2196. def test_regression_gh94675(self):
  2197. pattern = re.compile(r'(?<=[({}])(((//[^\n]*)?[\n])([\000-\040])*)*'
  2198. r'((/[^/\[\n]*(([^\n]|(\[\n]*(]*)*\]))'
  2199. r'[^/\[]*)*/))((((//[^\n]*)?[\n])'
  2200. r'([\000-\040]|(/\*[^*]*\*+'
  2201. r'([^/*]\*+)*/))*)+(?=[^\000-\040);\]}]))')
  2202. input_js = '''a(function() {
  2203. ///////////////////////////////////////////////////////////////////
  2204. });'''
  2205. p = multiprocessing.Process(target=pattern.sub, args=('', input_js))
  2206. p.start()
  2207. p.join(SHORT_TIMEOUT)
  2208. try:
  2209. self.assertFalse(p.is_alive(), 'pattern.sub() timed out')
  2210. finally:
  2211. if p.is_alive():
  2212. p.terminate()
  2213. p.join()
  2214. def get_debug_out(pat):
  2215. with captured_stdout() as out:
  2216. re.compile(pat, re.DEBUG)
  2217. return out.getvalue()
  2218. @cpython_only
  2219. class DebugTests(unittest.TestCase):
  2220. maxDiff = None
  2221. def test_debug_flag(self):
  2222. pat = r'(\.)(?:[ch]|py)(?(1)$|: )'
  2223. dump = '''\
  2224. SUBPATTERN 1 0 0
  2225. LITERAL 46
  2226. BRANCH
  2227. IN
  2228. LITERAL 99
  2229. LITERAL 104
  2230. OR
  2231. LITERAL 112
  2232. LITERAL 121
  2233. GROUPREF_EXISTS 1
  2234. AT AT_END
  2235. ELSE
  2236. LITERAL 58
  2237. LITERAL 32
  2238. 0. INFO 8 0b1 2 5 (to 9)
  2239. prefix_skip 0
  2240. prefix [0x2e] ('.')
  2241. overlap [0]
  2242. 9: MARK 0
  2243. 11. LITERAL 0x2e ('.')
  2244. 13. MARK 1
  2245. 15. BRANCH 10 (to 26)
  2246. 17. IN 6 (to 24)
  2247. 19. LITERAL 0x63 ('c')
  2248. 21. LITERAL 0x68 ('h')
  2249. 23. FAILURE
  2250. 24: JUMP 9 (to 34)
  2251. 26: branch 7 (to 33)
  2252. 27. LITERAL 0x70 ('p')
  2253. 29. LITERAL 0x79 ('y')
  2254. 31. JUMP 2 (to 34)
  2255. 33: FAILURE
  2256. 34: GROUPREF_EXISTS 0 6 (to 41)
  2257. 37. AT END
  2258. 39. JUMP 5 (to 45)
  2259. 41: LITERAL 0x3a (':')
  2260. 43. LITERAL 0x20 (' ')
  2261. 45: SUCCESS
  2262. '''
  2263. self.assertEqual(get_debug_out(pat), dump)
  2264. # Debug output is output again even a second time (bypassing
  2265. # the cache -- issue #20426).
  2266. self.assertEqual(get_debug_out(pat), dump)
  2267. def test_atomic_group(self):
  2268. self.assertEqual(get_debug_out(r'(?>ab?)'), '''\
  2269. ATOMIC_GROUP [(LITERAL, 97), (MAX_REPEAT, (0, 1, [(LITERAL, 98)]))]
  2270. 0. INFO 4 0b0 1 2 (to 5)
  2271. 5: ATOMIC_GROUP 11 (to 17)
  2272. 7. LITERAL 0x61 ('a')
  2273. 9. REPEAT_ONE 6 0 1 (to 16)
  2274. 13. LITERAL 0x62 ('b')
  2275. 15. SUCCESS
  2276. 16: SUCCESS
  2277. 17: SUCCESS
  2278. ''')
  2279. def test_possesive_repeat_one(self):
  2280. self.assertEqual(get_debug_out(r'a?+'), '''\
  2281. POSSESSIVE_REPEAT 0 1
  2282. LITERAL 97
  2283. 0. INFO 4 0b0 0 1 (to 5)
  2284. 5: POSSESSIVE_REPEAT_ONE 6 0 1 (to 12)
  2285. 9. LITERAL 0x61 ('a')
  2286. 11. SUCCESS
  2287. 12: SUCCESS
  2288. ''')
  2289. def test_possesive_repeat(self):
  2290. self.assertEqual(get_debug_out(r'(?:ab)?+'), '''\
  2291. POSSESSIVE_REPEAT 0 1
  2292. LITERAL 97
  2293. LITERAL 98
  2294. 0. INFO 4 0b0 0 2 (to 5)
  2295. 5: POSSESSIVE_REPEAT 7 0 1 (to 13)
  2296. 9. LITERAL 0x61 ('a')
  2297. 11. LITERAL 0x62 ('b')
  2298. 13: SUCCESS
  2299. 14. SUCCESS
  2300. ''')
  2301. class PatternReprTests(unittest.TestCase):
  2302. def check(self, pattern, expected):
  2303. self.assertEqual(repr(re.compile(pattern)), expected)
  2304. def check_flags(self, pattern, flags, expected):
  2305. self.assertEqual(repr(re.compile(pattern, flags)), expected)
  2306. def test_without_flags(self):
  2307. self.check('random pattern',
  2308. "re.compile('random pattern')")
  2309. def test_single_flag(self):
  2310. self.check_flags('random pattern', re.IGNORECASE,
  2311. "re.compile('random pattern', re.IGNORECASE)")
  2312. def test_multiple_flags(self):
  2313. self.check_flags('random pattern', re.I|re.S|re.X,
  2314. "re.compile('random pattern', "
  2315. "re.IGNORECASE|re.DOTALL|re.VERBOSE)")
  2316. def test_unicode_flag(self):
  2317. self.check_flags('random pattern', re.U,
  2318. "re.compile('random pattern')")
  2319. self.check_flags('random pattern', re.I|re.S|re.U,
  2320. "re.compile('random pattern', "
  2321. "re.IGNORECASE|re.DOTALL)")
  2322. def test_inline_flags(self):
  2323. self.check('(?i)pattern',
  2324. "re.compile('(?i)pattern', re.IGNORECASE)")
  2325. def test_unknown_flags(self):
  2326. self.check_flags('random pattern', 0x123000,
  2327. "re.compile('random pattern', 0x123000)")
  2328. self.check_flags('random pattern', 0x123000|re.I,
  2329. "re.compile('random pattern', re.IGNORECASE|0x123000)")
  2330. def test_bytes(self):
  2331. self.check(b'bytes pattern',
  2332. "re.compile(b'bytes pattern')")
  2333. self.check_flags(b'bytes pattern', re.A,
  2334. "re.compile(b'bytes pattern', re.ASCII)")
  2335. def test_locale(self):
  2336. self.check_flags(b'bytes pattern', re.L,
  2337. "re.compile(b'bytes pattern', re.LOCALE)")
  2338. def test_quotes(self):
  2339. self.check('random "double quoted" pattern',
  2340. '''re.compile('random "double quoted" pattern')''')
  2341. self.check("random 'single quoted' pattern",
  2342. '''re.compile("random 'single quoted' pattern")''')
  2343. self.check('''both 'single' and "double" quotes''',
  2344. '''re.compile('both \\'single\\' and "double" quotes')''')
  2345. def test_long_pattern(self):
  2346. pattern = 'Very %spattern' % ('long ' * 1000)
  2347. r = repr(re.compile(pattern))
  2348. self.assertLess(len(r), 300)
  2349. self.assertEqual(r[:30], "re.compile('Very long long lon")
  2350. r = repr(re.compile(pattern, re.I))
  2351. self.assertLess(len(r), 300)
  2352. self.assertEqual(r[:30], "re.compile('Very long long lon")
  2353. self.assertEqual(r[-16:], ", re.IGNORECASE)")
  2354. def test_flags_repr(self):
  2355. self.assertEqual(repr(re.I), "re.IGNORECASE")
  2356. self.assertEqual(repr(re.I|re.S|re.X),
  2357. "re.IGNORECASE|re.DOTALL|re.VERBOSE")
  2358. self.assertEqual(repr(re.I|re.S|re.X|(1<<20)),
  2359. "re.IGNORECASE|re.DOTALL|re.VERBOSE|0x100000")
  2360. self.assertEqual(
  2361. repr(~re.I),
  2362. "re.ASCII|re.LOCALE|re.UNICODE|re.MULTILINE|re.DOTALL|re.VERBOSE|re.TEMPLATE|re.DEBUG")
  2363. self.assertEqual(repr(~(re.I|re.S|re.X)),
  2364. "re.ASCII|re.LOCALE|re.UNICODE|re.MULTILINE|re.TEMPLATE|re.DEBUG")
  2365. self.assertEqual(repr(~(re.I|re.S|re.X|(1<<20))),
  2366. "re.ASCII|re.LOCALE|re.UNICODE|re.MULTILINE|re.TEMPLATE|re.DEBUG|0xffe00")
  2367. class ImplementationTest(unittest.TestCase):
  2368. """
  2369. Test implementation details of the re module.
  2370. """
  2371. @cpython_only
  2372. def test_immutable(self):
  2373. # bpo-43908: check that re types are immutable
  2374. with self.assertRaises(TypeError):
  2375. re.Match.foo = 1
  2376. with self.assertRaises(TypeError):
  2377. re.Pattern.foo = 1
  2378. with self.assertRaises(TypeError):
  2379. pat = re.compile("")
  2380. tp = type(pat.scanner(""))
  2381. tp.foo = 1
  2382. def test_overlap_table(self):
  2383. f = re._compiler._generate_overlap_table
  2384. self.assertEqual(f(""), [])
  2385. self.assertEqual(f("a"), [0])
  2386. self.assertEqual(f("abcd"), [0, 0, 0, 0])
  2387. self.assertEqual(f("aaaa"), [0, 1, 2, 3])
  2388. self.assertEqual(f("ababba"), [0, 0, 1, 2, 0, 1])
  2389. self.assertEqual(f("abcabdac"), [0, 0, 0, 1, 2, 0, 1, 0])
  2390. def test_signedness(self):
  2391. self.assertGreaterEqual(re._compiler.MAXREPEAT, 0)
  2392. self.assertGreaterEqual(re._compiler.MAXGROUPS, 0)
  2393. @cpython_only
  2394. def test_disallow_instantiation(self):
  2395. # Ensure that the type disallows instantiation (bpo-43916)
  2396. check_disallow_instantiation(self, re.Match)
  2397. check_disallow_instantiation(self, re.Pattern)
  2398. pat = re.compile("")
  2399. check_disallow_instantiation(self, type(pat.scanner("")))
  2400. def test_deprecated_modules(self):
  2401. deprecated = {
  2402. 'sre_compile': ['compile', 'error',
  2403. 'SRE_FLAG_IGNORECASE', 'SUBPATTERN',
  2404. '_compile_info'],
  2405. 'sre_constants': ['error', 'SRE_FLAG_IGNORECASE', 'SUBPATTERN',
  2406. '_NamedIntConstant'],
  2407. 'sre_parse': ['SubPattern', 'parse',
  2408. 'SRE_FLAG_IGNORECASE', 'SUBPATTERN',
  2409. '_parse_sub'],
  2410. }
  2411. for name in deprecated:
  2412. with self.subTest(module=name):
  2413. sys.modules.pop(name, None)
  2414. with self.assertWarns(DeprecationWarning) as w:
  2415. __import__(name)
  2416. self.assertEqual(str(w.warning),
  2417. f"module {name!r} is deprecated")
  2418. self.assertEqual(w.filename, __file__)
  2419. self.assertIn(name, sys.modules)
  2420. mod = sys.modules[name]
  2421. self.assertEqual(mod.__name__, name)
  2422. self.assertEqual(mod.__package__, '')
  2423. for attr in deprecated[name]:
  2424. self.assertTrue(hasattr(mod, attr))
  2425. del sys.modules[name]
  2426. class ExternalTests(unittest.TestCase):
  2427. def test_re_benchmarks(self):
  2428. 're_tests benchmarks'
  2429. from test.re_tests import benchmarks
  2430. for pattern, s in benchmarks:
  2431. with self.subTest(pattern=pattern, string=s):
  2432. p = re.compile(pattern)
  2433. self.assertTrue(p.search(s))
  2434. self.assertTrue(p.match(s))
  2435. self.assertTrue(p.fullmatch(s))
  2436. s2 = ' '*10000 + s + ' '*10000
  2437. self.assertTrue(p.search(s2))
  2438. self.assertTrue(p.match(s2, 10000))
  2439. self.assertTrue(p.match(s2, 10000, 10000 + len(s)))
  2440. self.assertTrue(p.fullmatch(s2, 10000, 10000 + len(s)))
  2441. def test_re_tests(self):
  2442. 're_tests test suite'
  2443. from test.re_tests import tests, FAIL, SYNTAX_ERROR
  2444. for t in tests:
  2445. pattern = s = outcome = repl = expected = None
  2446. if len(t) == 5:
  2447. pattern, s, outcome, repl, expected = t
  2448. elif len(t) == 3:
  2449. pattern, s, outcome = t
  2450. else:
  2451. raise ValueError('Test tuples should have 3 or 5 fields', t)
  2452. with self.subTest(pattern=pattern, string=s):
  2453. if outcome == SYNTAX_ERROR: # Expected a syntax error
  2454. with self.assertRaises(re.error):
  2455. re.compile(pattern)
  2456. continue
  2457. obj = re.compile(pattern)
  2458. result = obj.search(s)
  2459. if outcome == FAIL:
  2460. self.assertIsNone(result, 'Succeeded incorrectly')
  2461. continue
  2462. with self.subTest():
  2463. self.assertTrue(result, 'Failed incorrectly')
  2464. # Matched, as expected, so now we compute the
  2465. # result string and compare it to our expected result.
  2466. start, end = result.span(0)
  2467. vardict = {'found': result.group(0),
  2468. 'groups': result.group(),
  2469. 'flags': result.re.flags}
  2470. for i in range(1, 100):
  2471. try:
  2472. gi = result.group(i)
  2473. # Special hack because else the string concat fails:
  2474. if gi is None:
  2475. gi = "None"
  2476. except IndexError:
  2477. gi = "Error"
  2478. vardict['g%d' % i] = gi
  2479. for i in result.re.groupindex.keys():
  2480. try:
  2481. gi = result.group(i)
  2482. if gi is None:
  2483. gi = "None"
  2484. except IndexError:
  2485. gi = "Error"
  2486. vardict[i] = gi
  2487. self.assertEqual(eval(repl, vardict), expected,
  2488. 'grouping error')
  2489. # Try the match with both pattern and string converted to
  2490. # bytes, and check that it still succeeds.
  2491. try:
  2492. bpat = bytes(pattern, "ascii")
  2493. bs = bytes(s, "ascii")
  2494. except UnicodeEncodeError:
  2495. # skip non-ascii tests
  2496. pass
  2497. else:
  2498. with self.subTest('bytes pattern match'):
  2499. obj = re.compile(bpat)
  2500. self.assertTrue(obj.search(bs))
  2501. # Try the match with LOCALE enabled, and check that it
  2502. # still succeeds.
  2503. with self.subTest('locale-sensitive match'):
  2504. obj = re.compile(bpat, re.LOCALE)
  2505. result = obj.search(bs)
  2506. if result is None:
  2507. print('=== Fails on locale-sensitive match', t)
  2508. # Try the match with the search area limited to the extent
  2509. # of the match and see if it still succeeds. \B will
  2510. # break (because it won't match at the end or start of a
  2511. # string), so we'll ignore patterns that feature it.
  2512. if (pattern[:2] != r'\B' and pattern[-2:] != r'\B'
  2513. and result is not None):
  2514. with self.subTest('range-limited match'):
  2515. obj = re.compile(pattern)
  2516. self.assertTrue(obj.search(s, start, end + 1))
  2517. # Try the match with IGNORECASE enabled, and check that it
  2518. # still succeeds.
  2519. with self.subTest('case-insensitive match'):
  2520. obj = re.compile(pattern, re.IGNORECASE)
  2521. self.assertTrue(obj.search(s))
  2522. # Try the match with UNICODE locale enabled, and check
  2523. # that it still succeeds.
  2524. with self.subTest('unicode-sensitive match'):
  2525. obj = re.compile(pattern, re.UNICODE)
  2526. self.assertTrue(obj.search(s))
  2527. if __name__ == "__main__":
  2528. unittest.main()