blob: 1581856131cf89c5e0e7c3f5055f8d5f968502bb [file] [log] [blame]
Guido van Rossuma0e4c1b1997-07-17 14:52:48 +00001#!/usr/local/bin/python
2# -*- mode: python -*-
3# $Id$
4
Guido van Rossum23b22571997-07-17 22:36:14 +00005from test_support import verbose, TestFailed
Guido van Rossum8e0ce301997-07-11 19:34:44 +00006import re
Guido van Rossuma0e4c1b1997-07-17 14:52:48 +00007import reop
Guido van Rossume8b81311997-07-15 18:47:48 +00008import sys, os, string, traceback
Guido van Rossum8e0ce301997-07-11 19:34:44 +00009
Guido van Rossum23b22571997-07-17 22:36:14 +000010# Misc tests from Tim Peters' re.doc
11
Guido van Rossum49946571997-07-18 04:26:25 +000012if verbose:
13 print 'Running tests on re.sub'
14
Guido van Rossum23b22571997-07-17 22:36:14 +000015try:
Guido van Rossum23b22571997-07-17 22:36:14 +000016 assert re.sub("(?i)b+", "x", "bbbb BBBB") == 'x x'
Guido van Rossum49946571997-07-18 04:26:25 +000017
Guido van Rossum23b22571997-07-17 22:36:14 +000018 def bump_num(matchobj):
19 int_value = int(matchobj.group(0))
20 return str(int_value + 1)
21
Guido van Rossum49946571997-07-18 04:26:25 +000022 assert re.sub(r'\d+', bump_num, '08.2 -2 23x99y') == '9.3 -3 24x100y'
23
Guido van Rossum23b22571997-07-17 22:36:14 +000024 assert re.sub('.', lambda m: r"\n", 'x') == '\\n'
25 assert re.sub('.', r"\n", 'x') == '\n'
Guido van Rossum49946571997-07-18 04:26:25 +000026
Guido van Rossum23b22571997-07-17 22:36:14 +000027 s = r"\1\1"
28 assert re.sub('(.)', s, 'x') == 'xx'
29 assert re.sub('(.)', re.escape(s), 'x') == s
30 assert re.sub('(.)', lambda m: s, 'x') == s
31
Guido van Rossum49946571997-07-18 04:26:25 +000032 assert re.sub('(?P<a>x)', '\g<a>\g<a>', 'xx') == 'xxxx'
33
Guido van Rossum23b22571997-07-17 22:36:14 +000034except AssertionError:
35 raise TestFailed, "re.sub"
36
Guido van Rossum49946571997-07-18 04:26:25 +000037if verbose:
38 print 'Running tests on symbolic references'
39
40try:
41 re.sub('(?P<a>x)', '\g<a', 'xx')
42except re.error, reason:
43 pass
44else:
45 raise TestFailed, "symbolic reference"
46
47try:
48 re.sub('(?P<a>x)', '\g<', 'xx')
49except re.error, reason:
50 pass
51else:
52 raise TestFailed, "symbolic reference"
53
54try:
55 re.sub('(?P<a>x)', '\g', 'xx')
56except re.error, reason:
57 pass
58else:
59 raise TestFailed, "symbolic reference"
60
61try:
62 re.sub('(?P<a>x)', '\g<a a>', 'xx')
63except re.error, reason:
64 pass
65else:
66 raise TestFailed, "symbolic reference"
67
68try:
69 re.sub('(?P<a>x)', '\g<ab>', 'xx')
70except IndexError, reason:
71 pass
72else:
73 raise TestFailed, "symbolic reference"
74
75try:
76 re.sub('(?P<a>x)|(?P<b>y)', '\g<b>', 'xx')
77except re.error, reason:
78 pass
79else:
80 raise TestFailed, "symbolic reference"
81
82try:
83 re.sub('(?P<a>x)|(?P<b>y)', '\\2', 'xx')
84except re.error, reason:
85 pass
86else:
87 raise TestFailed, "symbolic reference"
88
89if verbose:
90 print 'Running tests on re.subn'
91
Guido van Rossum23b22571997-07-17 22:36:14 +000092try:
93 assert re.subn("(?i)b+", "x", "bbbb BBBB") == ('x x', 2)
94 assert re.subn("b+", "x", "bbbb BBBB") == ('x BBBB', 1)
95 assert re.subn("b+", "x", "xyz") == ('xyz', 0)
96 assert re.subn("b*", "x", "xyz") == ('xxxyxzx', 4)
Guido van Rossum49946571997-07-18 04:26:25 +000097
Guido van Rossum23b22571997-07-17 22:36:14 +000098except AssertionError:
99 raise TestFailed, "re.subn"
100
Guido van Rossum49946571997-07-18 04:26:25 +0000101if verbose:
102 print 'Running tests on re.split'
103
Guido van Rossum23b22571997-07-17 22:36:14 +0000104try:
105 assert re.split(":", ":a:b::c") == ['', 'a', 'b', '', 'c']
106 assert re.split(":*", ":a:b::c") == ['', 'a', 'b', 'c']
107 assert re.split("(:*)", ":a:b::c") == ['', ':', 'a', ':', 'b', '::', 'c']
108 assert re.split("(?::*)", ":a:b::c") == ['', 'a', 'b', 'c']
109 assert re.split("(:)*", ":a:b::c") == ['', ':', 'a', ':', 'b', ':', 'c']
110 assert re.split("([b:]+)", ":a:b::c") == ['', ':', 'a', ':b::', 'c']
111 assert re.split("(b)|(:+)", ":a:b::c") == \
Guido van Rossum49946571997-07-18 04:26:25 +0000112 ['', None, ':', 'a', None, ':', '', 'b', None, '', None, '::', 'c']
Guido van Rossum23b22571997-07-17 22:36:14 +0000113 assert re.split("(?:b)|(?::+)", ":a:b::c") == ['', 'a', '', '', 'c']
Guido van Rossum49946571997-07-18 04:26:25 +0000114
Guido van Rossum23b22571997-07-17 22:36:14 +0000115except AssertionError:
116 raise TestFailed, "re.split"
117
Guido van Rossum8e0ce301997-07-11 19:34:44 +0000118from re_tests import *
Guido van Rossum49946571997-07-18 04:26:25 +0000119if verbose:
120 print 'Running re_tests test suite'
Guido van Rossum8e0ce301997-07-11 19:34:44 +0000121
122for t in tests:
123 print t
Guido van Rossuma0e4c1b1997-07-17 14:52:48 +0000124 sys.stdout.flush()
Guido van Rossum8e0ce301997-07-11 19:34:44 +0000125 pattern=s=outcome=repl=expected=None
126 if len(t)==5:
127 pattern, s, outcome, repl, expected = t
128 elif len(t)==3:
129 pattern, s, outcome = t
130 else:
131 raise ValueError, ('Test tuples should have 3 or 5 fields',t)
132
133 try:
134 obj=re.compile(pattern)
135 except re.error:
136 if outcome==SYNTAX_ERROR: pass # Expected a syntax error
137 else:
138 print '=== Syntax error:', t
139 except:
140 print '*** Unexpected error ***'
Guido van Rossum1d8b7581997-07-11 21:14:53 +0000141 if verbose:
142 traceback.print_exc(file=sys.stdout)
Guido van Rossum8e0ce301997-07-11 19:34:44 +0000143 else:
144 try:
145 result=obj.search(s)
Guido van Rossuma0e4c1b1997-07-17 14:52:48 +0000146 except (re.error, reop.error), msg:
Guido van Rossum8e0ce301997-07-11 19:34:44 +0000147 print '=== Unexpected exception', t, repr(msg)
148 if outcome==SYNTAX_ERROR:
149 # This should have been a syntax error; forget it.
150 pass
151 elif outcome==FAIL:
152 if result is None: pass # No match, as expected
153 else: print '=== Succeeded incorrectly', t
154 elif outcome==SUCCEED:
155 if result is not None:
156 # Matched, as expected, so now we compute the
157 # result string and compare it to our expected result.
158 start, end = result.span(0)
Guido van Rossuma0e4c1b1997-07-17 14:52:48 +0000159 vardict={'found': result.group(0),
160 'groups': result.group(),
161 'flags': result.re.flags}
Guido van Rossum8e0ce301997-07-11 19:34:44 +0000162 for i in range(1, 100):
163 try:
164 gi = result.group(i)
Guido van Rossum035aae01997-07-11 20:47:58 +0000165 # Special hack because else the string concat fails:
Guido van Rossuma0e4c1b1997-07-17 14:52:48 +0000166 if gi is None:
167 gi = "None"
Guido van Rossum8e0ce301997-07-11 19:34:44 +0000168 except IndexError:
Guido van Rossum23b8d4c1997-07-15 15:49:52 +0000169 gi = "Error"
Guido van Rossum847ed4a1997-07-15 15:40:57 +0000170 vardict['g%d' % i] = gi
Guido van Rossume8b81311997-07-15 18:47:48 +0000171 for i in result.re.groupindex.keys():
172 try:
173 gi = result.group(i)
Guido van Rossuma0e4c1b1997-07-17 14:52:48 +0000174 if gi is None:
175 gi = "None"
Guido van Rossume8b81311997-07-15 18:47:48 +0000176 except IndexError:
Guido van Rossuma0e4c1b1997-07-17 14:52:48 +0000177 gi = "Error"
178 vardict[i] = gi
Guido van Rossum8e0ce301997-07-11 19:34:44 +0000179 repl=eval(repl, vardict)
180 if repl!=expected:
181 print '=== grouping error', t,
182 print repr(repl)+' should be '+repr(expected)
183 else:
184 print '=== Failed incorrectly', t