Guido van Rossum | a0e4c1b | 1997-07-17 14:52:48 +0000 | [diff] [blame] | 1 | #!/usr/local/bin/python |
| 2 | # -*- mode: python -*- |
| 3 | # $Id$ |
| 4 | |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 5 | from test_support import verbose, TestFailed |
Guido van Rossum | 8e0ce30 | 1997-07-11 19:34:44 +0000 | [diff] [blame] | 6 | import re |
Guido van Rossum | a0e4c1b | 1997-07-17 14:52:48 +0000 | [diff] [blame] | 7 | import reop |
Guido van Rossum | e8b8131 | 1997-07-15 18:47:48 +0000 | [diff] [blame] | 8 | import sys, os, string, traceback |
Guido van Rossum | 8e0ce30 | 1997-07-11 19:34:44 +0000 | [diff] [blame] | 9 | |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 10 | # Misc tests from Tim Peters' re.doc |
| 11 | |
Guido van Rossum | 4994657 | 1997-07-18 04:26:25 +0000 | [diff] [blame] | 12 | if verbose: |
| 13 | print 'Running tests on re.sub' |
| 14 | |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 15 | try: |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 16 | assert re.sub("(?i)b+", "x", "bbbb BBBB") == 'x x' |
Guido van Rossum | 4994657 | 1997-07-18 04:26:25 +0000 | [diff] [blame] | 17 | |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 18 | def bump_num(matchobj): |
| 19 | int_value = int(matchobj.group(0)) |
| 20 | return str(int_value + 1) |
| 21 | |
Guido van Rossum | 4994657 | 1997-07-18 04:26:25 +0000 | [diff] [blame] | 22 | assert re.sub(r'\d+', bump_num, '08.2 -2 23x99y') == '9.3 -3 24x100y' |
| 23 | |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 24 | assert re.sub('.', lambda m: r"\n", 'x') == '\\n' |
| 25 | assert re.sub('.', r"\n", 'x') == '\n' |
Guido van Rossum | 4994657 | 1997-07-18 04:26:25 +0000 | [diff] [blame] | 26 | |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 27 | 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 Rossum | 4994657 | 1997-07-18 04:26:25 +0000 | [diff] [blame] | 32 | assert re.sub('(?P<a>x)', '\g<a>\g<a>', 'xx') == 'xxxx' |
| 33 | |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 34 | except AssertionError: |
| 35 | raise TestFailed, "re.sub" |
| 36 | |
Guido van Rossum | 4994657 | 1997-07-18 04:26:25 +0000 | [diff] [blame] | 37 | if verbose: |
| 38 | print 'Running tests on symbolic references' |
| 39 | |
| 40 | try: |
| 41 | re.sub('(?P<a>x)', '\g<a', 'xx') |
| 42 | except re.error, reason: |
| 43 | pass |
| 44 | else: |
| 45 | raise TestFailed, "symbolic reference" |
| 46 | |
| 47 | try: |
| 48 | re.sub('(?P<a>x)', '\g<', 'xx') |
| 49 | except re.error, reason: |
| 50 | pass |
| 51 | else: |
| 52 | raise TestFailed, "symbolic reference" |
| 53 | |
| 54 | try: |
| 55 | re.sub('(?P<a>x)', '\g', 'xx') |
| 56 | except re.error, reason: |
| 57 | pass |
| 58 | else: |
| 59 | raise TestFailed, "symbolic reference" |
| 60 | |
| 61 | try: |
| 62 | re.sub('(?P<a>x)', '\g<a a>', 'xx') |
| 63 | except re.error, reason: |
| 64 | pass |
| 65 | else: |
| 66 | raise TestFailed, "symbolic reference" |
| 67 | |
| 68 | try: |
| 69 | re.sub('(?P<a>x)', '\g<ab>', 'xx') |
| 70 | except IndexError, reason: |
| 71 | pass |
| 72 | else: |
| 73 | raise TestFailed, "symbolic reference" |
| 74 | |
| 75 | try: |
| 76 | re.sub('(?P<a>x)|(?P<b>y)', '\g<b>', 'xx') |
| 77 | except re.error, reason: |
| 78 | pass |
| 79 | else: |
| 80 | raise TestFailed, "symbolic reference" |
| 81 | |
| 82 | try: |
| 83 | re.sub('(?P<a>x)|(?P<b>y)', '\\2', 'xx') |
| 84 | except re.error, reason: |
| 85 | pass |
| 86 | else: |
| 87 | raise TestFailed, "symbolic reference" |
| 88 | |
| 89 | if verbose: |
| 90 | print 'Running tests on re.subn' |
| 91 | |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 92 | try: |
| 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 Rossum | 4994657 | 1997-07-18 04:26:25 +0000 | [diff] [blame] | 97 | |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 98 | except AssertionError: |
| 99 | raise TestFailed, "re.subn" |
| 100 | |
Guido van Rossum | 4994657 | 1997-07-18 04:26:25 +0000 | [diff] [blame] | 101 | if verbose: |
| 102 | print 'Running tests on re.split' |
| 103 | |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 104 | try: |
| 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 Rossum | 4994657 | 1997-07-18 04:26:25 +0000 | [diff] [blame] | 112 | ['', None, ':', 'a', None, ':', '', 'b', None, '', None, '::', 'c'] |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 113 | assert re.split("(?:b)|(?::+)", ":a:b::c") == ['', 'a', '', '', 'c'] |
Guido van Rossum | 4994657 | 1997-07-18 04:26:25 +0000 | [diff] [blame] | 114 | |
Guido van Rossum | 23b2257 | 1997-07-17 22:36:14 +0000 | [diff] [blame] | 115 | except AssertionError: |
| 116 | raise TestFailed, "re.split" |
| 117 | |
Guido van Rossum | 8e0ce30 | 1997-07-11 19:34:44 +0000 | [diff] [blame] | 118 | from re_tests import * |
Guido van Rossum | 4994657 | 1997-07-18 04:26:25 +0000 | [diff] [blame] | 119 | if verbose: |
| 120 | print 'Running re_tests test suite' |
Guido van Rossum | 8e0ce30 | 1997-07-11 19:34:44 +0000 | [diff] [blame] | 121 | |
| 122 | for t in tests: |
| 123 | print t |
Guido van Rossum | a0e4c1b | 1997-07-17 14:52:48 +0000 | [diff] [blame] | 124 | sys.stdout.flush() |
Guido van Rossum | 8e0ce30 | 1997-07-11 19:34:44 +0000 | [diff] [blame] | 125 | 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 Rossum | 1d8b758 | 1997-07-11 21:14:53 +0000 | [diff] [blame] | 141 | if verbose: |
| 142 | traceback.print_exc(file=sys.stdout) |
Guido van Rossum | 8e0ce30 | 1997-07-11 19:34:44 +0000 | [diff] [blame] | 143 | else: |
| 144 | try: |
| 145 | result=obj.search(s) |
Guido van Rossum | a0e4c1b | 1997-07-17 14:52:48 +0000 | [diff] [blame] | 146 | except (re.error, reop.error), msg: |
Guido van Rossum | 8e0ce30 | 1997-07-11 19:34:44 +0000 | [diff] [blame] | 147 | 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 Rossum | a0e4c1b | 1997-07-17 14:52:48 +0000 | [diff] [blame] | 159 | vardict={'found': result.group(0), |
| 160 | 'groups': result.group(), |
| 161 | 'flags': result.re.flags} |
Guido van Rossum | 8e0ce30 | 1997-07-11 19:34:44 +0000 | [diff] [blame] | 162 | for i in range(1, 100): |
| 163 | try: |
| 164 | gi = result.group(i) |
Guido van Rossum | 035aae0 | 1997-07-11 20:47:58 +0000 | [diff] [blame] | 165 | # Special hack because else the string concat fails: |
Guido van Rossum | a0e4c1b | 1997-07-17 14:52:48 +0000 | [diff] [blame] | 166 | if gi is None: |
| 167 | gi = "None" |
Guido van Rossum | 8e0ce30 | 1997-07-11 19:34:44 +0000 | [diff] [blame] | 168 | except IndexError: |
Guido van Rossum | 23b8d4c | 1997-07-15 15:49:52 +0000 | [diff] [blame] | 169 | gi = "Error" |
Guido van Rossum | 847ed4a | 1997-07-15 15:40:57 +0000 | [diff] [blame] | 170 | vardict['g%d' % i] = gi |
Guido van Rossum | e8b8131 | 1997-07-15 18:47:48 +0000 | [diff] [blame] | 171 | for i in result.re.groupindex.keys(): |
| 172 | try: |
| 173 | gi = result.group(i) |
Guido van Rossum | a0e4c1b | 1997-07-17 14:52:48 +0000 | [diff] [blame] | 174 | if gi is None: |
| 175 | gi = "None" |
Guido van Rossum | e8b8131 | 1997-07-15 18:47:48 +0000 | [diff] [blame] | 176 | except IndexError: |
Guido van Rossum | a0e4c1b | 1997-07-17 14:52:48 +0000 | [diff] [blame] | 177 | gi = "Error" |
| 178 | vardict[i] = gi |
Guido van Rossum | 8e0ce30 | 1997-07-11 19:34:44 +0000 | [diff] [blame] | 179 | 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 |