blob: 3681a9d9caef874976dedd9e7dc9265ee54d8182 [file] [log] [blame]
Guido van Rossum152494a1996-12-20 03:12:20 +00001#! /usr/bin/env python
2
3"""Regression test.
4
5This will find all modules whose name is "test_*" in the test
6directory, and run them. Various command line options provide
7additional facilities.
8
9Command line options:
10
Michael W. Hudson61147f62004-08-03 11:33:28 +000011-v: verbose -- run tests in verbose mode with output to stdout
Martin v. Löwis04824ce2006-03-10 21:26:16 +000012-w: verbose2 -- re-run failed tests in verbose mode
Guido van Rossumce4a4752007-02-26 22:01:28 +000013-d: debug -- print traceback for failed tests
Michael W. Hudson61147f62004-08-03 11:33:28 +000014-q: quiet -- don't print anything except if a test fails
15-g: generate -- write the output file for a test instead of comparing it
16-x: exclude -- arguments are tests to *exclude*
17-s: single -- run only a single test (see below)
18-r: random -- randomize test execution order
19-f: fromfile -- read names of tests to run from a file (see below)
20-l: findleaks -- if GC is available detect tests that leak memory
21-u: use -- specify which special resource intensive tests to run
22-h: help -- print this text and exit
23-t: threshold -- call gc.set_threshold(N)
24-T: coverage -- turn on code coverage using the trace module
Walter Dörwaldaee4da62004-11-12 18:51:27 +000025-D: coverdir -- Directory where coverage files are put
26-N: nocoverdir -- Put coverage files alongside modules
Michael W. Hudson61147f62004-08-03 11:33:28 +000027-L: runleaks -- run the leaks(1) command just before exit
28-R: huntrleaks -- search for reference leaks (needs debug build, v. slow)
Thomas Wouters477c8d52006-05-27 19:21:47 +000029-M: memlimit -- run very large memory-consuming tests
Guido van Rossum152494a1996-12-20 03:12:20 +000030
31If non-option arguments are present, they are names for tests to run,
32unless -x is given, in which case they are names for tests not to run.
33If no test names are given, all tests are run.
Guido van Rossumf58ed251997-03-07 21:04:33 +000034
Guido van Rossuma4122201997-08-18 20:08:24 +000035-v is incompatible with -g and does not compare test output files.
Barry Warsawe11e3de1999-01-28 19:51:51 +000036
Barry Warsaw3b6d0252004-02-07 22:43:03 +000037-T turns on code coverage tracing with the trace module.
38
Walter Dörwaldaee4da62004-11-12 18:51:27 +000039-D specifies the directory where coverage files are put.
40
41-N Put coverage files alongside modules.
42
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000043-s means to run only a single test and exit. This is useful when
44doing memory analysis on the Python interpreter (which tend to consume
45too many resources to run the full regression test non-stop). The
46file /tmp/pynexttest is read to find the next test to run. If this
47file is missing, the first test_*.py file in testdir or on the command
48line is used. (actually tempfile.gettempdir() is used instead of
49/tmp).
Barry Warsawe11e3de1999-01-28 19:51:51 +000050
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000051-f reads the names of tests from the file given as f's argument, one
52or more test names per line. Whitespace is ignored. Blank lines and
53lines beginning with '#' are ignored. This is especially useful for
54whittling down failures involving interactions among tests.
Tim Petersc5000df2002-06-02 21:42:01 +000055
Skip Montanaro0179a182004-06-06 15:53:18 +000056-L causes the leaks(1) command to be run just before exit if it exists.
57leaks(1) is available on Mac OS X and presumably on some other
58FreeBSD-derived systems.
59
Michael W. Hudson61147f62004-08-03 11:33:28 +000060-R runs each test several times and examines sys.gettotalrefcount() to
61see if the test appears to be leaking references. The argument should
62be of the form stab:run:fname where 'stab' is the number of times the
63test is run to let gettotalrefcount settle down, 'run' is the number
64of times further it is run and 'fname' is the name of the file the
65reports are written to. These parameters all have defaults (5, 4 and
66"reflog.txt" respectively), so the minimal invocation is '-R ::'.
67
Thomas Wouters477c8d52006-05-27 19:21:47 +000068-M runs tests that require an exorbitant amount of memory. These tests
69typically try to ascertain containers keep working when containing more than
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000702 billion objects, which only works on 64-bit systems. There are also some
71tests that try to exhaust the address space of the process, which only makes
72sense on 32-bit systems with at least 2Gb of memory. The passed-in memlimit,
Thomas Wouters477c8d52006-05-27 19:21:47 +000073which is a string in the form of '2.5Gb', determines howmuch memory the
74tests will limit themselves to (but they may go slightly over.) The number
75shouldn't be more memory than the machine has (including swap memory). You
76should also keep in mind that swap memory is generally much, much slower
77than RAM, and setting memlimit to all available RAM or higher will heavily
78tax the machine. On the other hand, it is no use running these tests with a
79limit of less than 2.5Gb, and many require more than 20Gb. Tests that expect
80to use more than memlimit memory will be skipped. The big-memory tests
81generally run very, very long.
82
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000083-u is used to specify which special resource intensive tests to run,
84such as those requiring large file support or network connectivity.
85The argument is a comma-separated list of words indicating the
86resources to test. Currently only the following are defined:
Barry Warsaw08fca522001-08-20 22:33:46 +000087
Fred Drake3a15dac2002-04-11 16:39:16 +000088 all - Enable all special resources.
89
Guido van Rossum315aa362003-03-11 14:46:48 +000090 audio - Tests that use the audio device. (There are known
91 cases of broken audio drivers that can crash Python or
92 even the Linux kernel.)
93
Andrew M. Kuchling2158df02001-10-22 15:26:09 +000094 curses - Tests that use curses and will modify the terminal's
95 state and output modes.
Tim Peters1633a2e2001-10-30 05:56:40 +000096
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000097 largefile - It is okay to run some test that may create huge
98 files. These tests can take a long time and may
99 consume >2GB of disk space temporarily.
Barry Warsaw08fca522001-08-20 22:33:46 +0000100
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000101 network - It is okay to run tests that use external network
102 resource, e.g. testing SSL support for sockets.
Martin v. Löwis1c6b1a22002-11-19 17:47:07 +0000103
104 bsddb - It is okay to run the bsddb testsuite, which takes
105 a long time to complete.
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000106
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000107 decimal - Test the decimal module against a large suite that
108 verifies compliance with standards.
109
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000110 compiler - Test the compiler package by compiling all the source
111 in the standard library and test suite. This takes
Raymond Hettinger68c04532005-06-10 11:05:19 +0000112 a long time. Enabling this resource also allows
113 test_tokenize to verify round-trip lexing on every
114 file in the test library.
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000115
Tim Peterseba28be2005-03-28 01:08:02 +0000116 subprocess Run all tests for the subprocess module.
Peter Astrandf7f1bb72005-03-03 20:47:37 +0000117
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000118 urlfetch - It is okay to download files required on testing.
119
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000120To enable all resources except one, use '-uall,-<resource>'. For
121example, to run all the tests except for the bsddb tests, give the
122option '-uall,-bsddb'.
Guido van Rossum152494a1996-12-20 03:12:20 +0000123"""
124
Guido van Rossum152494a1996-12-20 03:12:20 +0000125import os
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000126import sys
Guido van Rossum152494a1996-12-20 03:12:20 +0000127import getopt
Skip Montanaroab1c7912000-06-30 16:39:27 +0000128import random
Guido van Rossumdc15c272002-08-12 21:55:51 +0000129import warnings
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000130import re
Guido van Rossum62b88712007-05-09 23:45:09 +0000131import StringIO
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000132import traceback
Guido van Rossumdc15c272002-08-12 21:55:51 +0000133
134# I see no other way to suppress these warnings;
135# putting them in test_grammar.py has no effect:
Guido van Rossum88b1def2002-08-14 17:54:48 +0000136warnings.filterwarnings("ignore", "hex/oct constants", FutureWarning,
Guido van Rossumdc15c272002-08-12 21:55:51 +0000137 ".*test.test_grammar$")
Guido van Rossumc34c4fc2002-09-19 00:42:16 +0000138if sys.maxint > 0x7fffffff:
139 # Also suppress them in <string>, because for 64-bit platforms,
140 # that's where test_grammar.py hides them.
141 warnings.filterwarnings("ignore", "hex/oct constants", FutureWarning,
142 "<string>")
Guido van Rossum152494a1996-12-20 03:12:20 +0000143
Thomas Wouters477c8d52006-05-27 19:21:47 +0000144# Ignore ImportWarnings that only occur in the source tree,
145# (because of modules with the same name as source-directories in Modules/)
146for mod in ("ctypes", "gzip", "zipfile", "tarfile", "encodings.zlib_codec",
147 "test.test_zipimport", "test.test_zlib", "test.test_zipfile",
148 "test.test_codecs", "test.string_tests"):
149 warnings.filterwarnings(module=".*%s$" % (mod,),
150 action="ignore", category=ImportWarning)
151
Guido van Rossumbb484652002-12-02 09:56:21 +0000152# MacOSX (a.k.a. Darwin) has a default stack size that is too small
153# for deeply recursive regular expressions. We see this as crashes in
154# the Python test suite when running test_re.py and test_sre.py. The
155# fix is to set the stack limit to 2048.
156# This approach may also be useful for other Unixy platforms that
157# suffer from small default stack limits.
158if sys.platform == 'darwin':
159 try:
160 import resource
161 except ImportError:
162 pass
163 else:
164 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
165 newsoft = min(hard, max(soft, 1024*2048))
166 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
167
Barry Warsaw04f357c2002-07-23 19:04:11 +0000168from test import test_support
Fred Drake3a15dac2002-04-11 16:39:16 +0000169
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000170RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network', 'bsddb',
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000171 'decimal', 'compiler', 'subprocess', 'urlfetch')
Fred Drake3a15dac2002-04-11 16:39:16 +0000172
173
Barry Warsaw08fca522001-08-20 22:33:46 +0000174def usage(code, msg=''):
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000175 print(__doc__)
176 if msg: print(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000177 sys.exit(code)
178
179
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000180def main(tests=None, testdir=None, verbose=0, quiet=False, generate=False,
181 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000182 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Guido van Rossumce4a4752007-02-26 22:01:28 +0000183 runleaks=False, huntrleaks=False, verbose2=False, debug=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000184 """Execute a test suite.
185
Thomas Wouters7e474022000-07-16 12:04:32 +0000186 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000187 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000188
189 tests -- a list of strings containing test names (optional)
190 testdir -- the directory in which to look for tests (optional)
191
192 Users other than the Python test suite will certainly want to
193 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000194 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000195
196 If the tests argument is omitted, the tests listed on the
197 command-line will be used. If that's empty, too, then all *.py
198 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000199
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000200 The other default arguments (verbose, quiet, generate, exclude, single,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000201 randomize, findleaks, use_resources, trace and coverdir) allow programmers
202 calling main() directly to set the values that would normally be set by
203 flags on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000204 """
Fred Drake004d5e62000-10-23 17:22:08 +0000205
Tim Peters8dee8092001-09-25 20:05:11 +0000206 test_support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000207 try:
Guido van Rossumce4a4752007-02-26 22:01:28 +0000208 opts, args = getopt.getopt(sys.argv[1:], 'dhvgqxsrf:lu:t:TD:NLR:wM:',
Barry Warsaw08fca522001-08-20 22:33:46 +0000209 ['help', 'verbose', 'quiet', 'generate',
Tim Petersc5000df2002-06-02 21:42:01 +0000210 'exclude', 'single', 'random', 'fromfile',
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000211 'findleaks', 'use=', 'threshold=', 'trace',
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000212 'coverdir=', 'nocoverdir', 'runleaks',
Thomas Wouters477c8d52006-05-27 19:21:47 +0000213 'huntrleaks=', 'verbose2', 'memlimit=',
Guido van Rossumce4a4752007-02-26 22:01:28 +0000214 'debug',
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000215 ])
Guido van Rossumb940e112007-01-10 16:19:56 +0000216 except getopt.error as msg:
Barry Warsaw08fca522001-08-20 22:33:46 +0000217 usage(2, msg)
218
219 # Defaults
220 if use_resources is None:
221 use_resources = []
Guido van Rossum152494a1996-12-20 03:12:20 +0000222 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000223 if o in ('-h', '--help'):
224 usage(0)
225 elif o in ('-v', '--verbose'):
226 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000227 elif o in ('-w', '--verbose2'):
228 verbose2 = True
Guido van Rossumce4a4752007-02-26 22:01:28 +0000229 elif o in ('-d', '--debug'):
230 debug = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000231 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000232 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000233 verbose = 0
234 elif o in ('-g', '--generate'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000235 generate = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000236 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000237 exclude = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000238 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000239 single = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000240 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000241 randomize = True
Tim Petersc5000df2002-06-02 21:42:01 +0000242 elif o in ('-f', '--fromfile'):
243 fromfile = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000244 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000245 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000246 elif o in ('-L', '--runleaks'):
247 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000248 elif o in ('-t', '--threshold'):
249 import gc
250 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000251 elif o in ('-T', '--coverage'):
252 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000253 elif o in ('-D', '--coverdir'):
254 coverdir = os.path.join(os.getcwd(), a)
255 elif o in ('-N', '--nocoverdir'):
256 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000257 elif o in ('-R', '--huntrleaks'):
258 huntrleaks = a.split(':')
259 if len(huntrleaks) != 3:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000260 print(a, huntrleaks)
Michael W. Hudson61147f62004-08-03 11:33:28 +0000261 usage(2, '-R takes three colon-separated arguments')
262 if len(huntrleaks[0]) == 0:
263 huntrleaks[0] = 5
264 else:
265 huntrleaks[0] = int(huntrleaks[0])
266 if len(huntrleaks[1]) == 0:
267 huntrleaks[1] = 4
268 else:
269 huntrleaks[1] = int(huntrleaks[1])
270 if len(huntrleaks[2]) == 0:
271 huntrleaks[2] = "reflog.txt"
Thomas Wouters477c8d52006-05-27 19:21:47 +0000272 elif o in ('-M', '--memlimit'):
273 test_support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000274 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000275 u = [x.lower() for x in a.split(',')]
276 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000277 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000278 use_resources[:] = RESOURCE_NAMES
279 continue
280 remove = False
281 if r[0] == '-':
282 remove = True
283 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000284 if r not in RESOURCE_NAMES:
285 usage(1, 'Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000286 if remove:
287 if r in use_resources:
288 use_resources.remove(r)
289 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000290 use_resources.append(r)
Guido van Rossuma4122201997-08-18 20:08:24 +0000291 if generate and verbose:
Barry Warsaw08fca522001-08-20 22:33:46 +0000292 usage(2, "-g and -v don't go together!")
Tim Petersc5000df2002-06-02 21:42:01 +0000293 if single and fromfile:
294 usage(2, "-s and -f don't go together!")
Barry Warsaw08fca522001-08-20 22:33:46 +0000295
Guido van Rossum152494a1996-12-20 03:12:20 +0000296 good = []
297 bad = []
298 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000299 resource_denieds = []
Barry Warsawe11e3de1999-01-28 19:51:51 +0000300
Neil Schemenauerd569f232000-09-22 15:29:28 +0000301 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000302 try:
303 import gc
304 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000305 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000306 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000307 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000308 # Uncomment the line below to report garbage that is not
309 # freeable by reference counting alone. By default only
310 # garbage that is not collectable by the GC is reported.
311 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000312 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000313
Barry Warsawe11e3de1999-01-28 19:51:51 +0000314 if single:
315 from tempfile import gettempdir
316 filename = os.path.join(gettempdir(), 'pynexttest')
317 try:
318 fp = open(filename, 'r')
Eric S. Raymondfc170b12001-02-09 11:51:27 +0000319 next = fp.read().strip()
Barry Warsawe11e3de1999-01-28 19:51:51 +0000320 tests = [next]
321 fp.close()
322 except IOError:
323 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000324
325 if fromfile:
326 tests = []
327 fp = open(fromfile)
328 for line in fp:
329 guts = line.split() # assuming no test has whitespace in its name
330 if guts and not guts[0].startswith('#'):
331 tests.extend(guts)
332 fp.close()
333
334 # Strip .py extensions.
335 if args:
336 args = map(removepy, args)
337 if tests:
338 tests = map(removepy, tests)
339
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000340 stdtests = STDTESTS[:]
341 nottests = NOTTESTS[:]
Guido van Rossum152494a1996-12-20 03:12:20 +0000342 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000343 for arg in args:
344 if arg in stdtests:
345 stdtests.remove(arg)
346 nottests[:0] = args
Guido van Rossum41360a41998-03-26 19:42:58 +0000347 args = []
Guido van Rossum747e1ca1998-08-24 13:48:36 +0000348 tests = tests or args or findtests(testdir, stdtests, nottests)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000349 if single:
350 tests = tests[:1]
Skip Montanaroab1c7912000-06-30 16:39:27 +0000351 if randomize:
352 random.shuffle(tests)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000353 if trace:
354 import trace
355 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix],
356 trace=False, count=True)
Guido van Rossum41360a41998-03-26 19:42:58 +0000357 test_support.verbose = verbose # Tell tests to be moderately quiet
Barry Warsaw08fca522001-08-20 22:33:46 +0000358 test_support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000359 save_modules = sys.modules.keys()
Guido van Rossum152494a1996-12-20 03:12:20 +0000360 for test in tests:
Guido van Rossum41360a41998-03-26 19:42:58 +0000361 if not quiet:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000362 print(test)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000363 sys.stdout.flush()
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000364 if trace:
365 # If we're tracing code coverage, then we don't exit with status
366 # if on a false return value from main.
367 tracer.runctx('runtest(test, generate, verbose, quiet, testdir)',
368 globals=globals(), locals=vars())
Guido van Rossum41360a41998-03-26 19:42:58 +0000369 else:
Neal Norwitz14ca3272006-02-28 18:05:43 +0000370 try:
371 ok = runtest(test, generate, verbose, quiet, testdir,
372 huntrleaks)
373 except KeyboardInterrupt:
374 # print a newline separate from the ^C
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000375 print()
Neal Norwitz14ca3272006-02-28 18:05:43 +0000376 break
377 except:
378 raise
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000379 if ok > 0:
380 good.append(test)
381 elif ok == 0:
382 bad.append(test)
383 else:
384 skipped.append(test)
385 if ok == -2:
386 resource_denieds.append(test)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000387 if findleaks:
388 gc.collect()
389 if gc.garbage:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000390 print("Warning: test created", len(gc.garbage), end=' ')
391 print("uncollectable object(s).")
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000392 # move the uncollectable objects somewhere so we don't see
393 # them again
Neil Schemenauerd569f232000-09-22 15:29:28 +0000394 found_garbage.extend(gc.garbage)
395 del gc.garbage[:]
Guido van Rossum5796d262000-04-21 21:35:06 +0000396 # Unload the newly imported modules (best effort finalization)
397 for module in sys.modules.keys():
Guido van Rossum51931142000-05-05 14:27:39 +0000398 if module not in save_modules and module.startswith("test."):
Guido van Rossum5796d262000-04-21 21:35:06 +0000399 test_support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000400
401 # The lists won't be sorted if running with -r
402 good.sort()
403 bad.sort()
404 skipped.sort()
Tim Peterse0c446b2001-10-18 21:57:37 +0000405
Guido van Rossum152494a1996-12-20 03:12:20 +0000406 if good and not quiet:
Guido van Rossum41360a41998-03-26 19:42:58 +0000407 if not bad and not skipped and len(good) > 1:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000408 print("All", end=' ')
409 print(count(len(good), "test"), "OK.")
Tim Peters1a4d77b2000-12-30 22:21:22 +0000410 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000411 print("CAUTION: stdout isn't compared in verbose mode:")
412 print("a test that passes in verbose mode may fail without it.")
Guido van Rossum152494a1996-12-20 03:12:20 +0000413 if bad:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000414 print(count(len(bad), "test"), "failed:")
Tim Petersa45da922001-08-12 03:45:50 +0000415 printlist(bad)
Guido van Rossum152494a1996-12-20 03:12:20 +0000416 if skipped and not quiet:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000417 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000418 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000419
Tim Petersb5b7b782001-08-12 01:20:39 +0000420 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000421 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000422 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000423 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000424 if surprise:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000425 print(count(len(surprise), "skip"), \
426 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000427 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000428 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000429 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000430 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000431 print("Ask someone to teach regrtest.py about which tests are")
432 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000433
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000434 if verbose2 and bad:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000435 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000436 for test in bad:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000437 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000438 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000439 try:
440 test_support.verbose = 1
441 ok = runtest(test, generate, 1, quiet, testdir,
Guido van Rossumce4a4752007-02-26 22:01:28 +0000442 huntrleaks, debug)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000443 except KeyboardInterrupt:
444 # print a newline separate from the ^C
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000445 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000446 break
447 except:
448 raise
449
Barry Warsawe11e3de1999-01-28 19:51:51 +0000450 if single:
451 alltests = findtests(testdir, stdtests, nottests)
452 for i in range(len(alltests)):
453 if tests[0] == alltests[i]:
454 if i == len(alltests) - 1:
455 os.unlink(filename)
456 else:
457 fp = open(filename, 'w')
458 fp.write(alltests[i+1] + '\n')
459 fp.close()
460 break
461 else:
462 os.unlink(filename)
463
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000464 if trace:
465 r = tracer.results()
466 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
467
Skip Montanaro0179a182004-06-06 15:53:18 +0000468 if runleaks:
469 os.system("leaks %d" % os.getpid())
470
Tim Peters5943b4a2003-07-23 00:30:39 +0000471 sys.exit(len(bad) > 0)
Barry Warsaw08fca522001-08-20 22:33:46 +0000472
Guido van Rossum152494a1996-12-20 03:12:20 +0000473
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000474STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000475 'test_grammar',
476 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000477 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000478 'test_builtin',
479 'test_exceptions',
480 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000481 'test_unittest',
482 'test_doctest',
483 'test_doctest2',
Guido van Rossum152494a1996-12-20 03:12:20 +0000484 ]
485
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000486NOTTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000487 'test_support',
Jeremy Hylton62e2c7e2001-02-28 17:48:06 +0000488 'test_future1',
489 'test_future2',
Jeremy Hylton8471a352001-08-20 20:33:42 +0000490 'test_future3',
Guido van Rossum152494a1996-12-20 03:12:20 +0000491 ]
492
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000493def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000494 """Return a list of all applicable test modules."""
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000495 if not testdir: testdir = findtestdir()
Guido van Rossum152494a1996-12-20 03:12:20 +0000496 names = os.listdir(testdir)
497 tests = []
498 for name in names:
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000499 if name[:5] == "test_" and name[-3:] == os.extsep+"py":
Guido van Rossum41360a41998-03-26 19:42:58 +0000500 modname = name[:-3]
501 if modname not in stdtests and modname not in nottests:
502 tests.append(modname)
Guido van Rossum152494a1996-12-20 03:12:20 +0000503 tests.sort()
504 return stdtests + tests
505
Guido van Rossumce4a4752007-02-26 22:01:28 +0000506def runtest(test, generate, verbose, quiet, testdir=None,
507 huntrleaks=False, debug=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000508 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000509
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000510 test -- the name of the test
511 generate -- if true, generate output, instead of running the test
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000512 and comparing it to a previously created output file
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000513 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000514 quiet -- if true, don't print 'skipped' messages (probably redundant)
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000515 testdir -- test directory
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000516 huntrleaks -- run multiple times to test for leaks; requires a debug
517 build; a triple corresponding to -R's three arguments
Guido van Rossumce4a4752007-02-26 22:01:28 +0000518 debug -- if true, print tracebacks for failed tests regardless of
519 verbose setting
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000520 Return:
521 -2 test skipped because resource denied
522 -1 test skipped for some other reason
523 0 test failed
524 1 test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000525 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000526
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000527 try:
528 return runtest_inner(test, generate, verbose, quiet, testdir,
Guido van Rossumce4a4752007-02-26 22:01:28 +0000529 huntrleaks, debug)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000530 finally:
531 cleanup_test_droppings(test, verbose)
532
533def runtest_inner(test, generate, verbose, quiet,
Guido van Rossumce4a4752007-02-26 22:01:28 +0000534 testdir=None, huntrleaks=False, debug=False):
Guido van Rossum152494a1996-12-20 03:12:20 +0000535 test_support.unload(test)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000536 if not testdir:
537 testdir = findtestdir()
Guido van Rossum152494a1996-12-20 03:12:20 +0000538 outputdir = os.path.join(testdir, "output")
539 outputfile = os.path.join(outputdir, test)
Tim Peters9390cc12001-09-28 20:14:46 +0000540 if verbose:
Guido van Rossum41360a41998-03-26 19:42:58 +0000541 cfp = None
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000542 else:
Guido van Rossum62b88712007-05-09 23:45:09 +0000543 cfp = StringIO.StringIO() # XXX Should use io.StringIO()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000544
Guido van Rossum152494a1996-12-20 03:12:20 +0000545 try:
Tim Peters342ca752001-09-25 19:13:20 +0000546 save_stdout = sys.stdout
Guido van Rossum41360a41998-03-26 19:42:58 +0000547 try:
548 if cfp:
549 sys.stdout = cfp
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000550 print(test) # Output file starts with test name
Barry Warsaw408b6d32002-07-30 23:27:12 +0000551 if test.startswith('test.'):
552 abstest = test
553 else:
554 # Always import it from the test package
555 abstest = 'test.' + test
556 the_package = __import__(abstest, globals(), locals(), [])
557 the_module = getattr(the_package, test)
Tim Petersd9742212001-05-22 18:28:25 +0000558 # Most tests run to completion simply as a side-effect of
559 # being imported. For the benefit of tests that can't run
560 # that way (like test_threaded_import), explicitly invoke
561 # their test_main() function (if it exists).
562 indirect_test = getattr(the_module, "test_main", None)
563 if indirect_test is not None:
564 indirect_test()
Michael W. Hudson61147f62004-08-03 11:33:28 +0000565 if huntrleaks:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000566 dash_R(the_module, test, indirect_test, huntrleaks)
Guido van Rossum41360a41998-03-26 19:42:58 +0000567 finally:
Tim Peters342ca752001-09-25 19:13:20 +0000568 sys.stdout = save_stdout
Guido van Rossumb940e112007-01-10 16:19:56 +0000569 except test_support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +0000570 if not quiet:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000571 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +0000572 sys.stdout.flush()
573 return -2
Guido van Rossumb940e112007-01-10 16:19:56 +0000574 except (ImportError, test_support.TestSkipped) as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +0000575 if not quiet:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000576 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000577 sys.stdout.flush()
Guido van Rossum41360a41998-03-26 19:42:58 +0000578 return -1
Fred Drakefe5c22a2000-08-18 16:04:05 +0000579 except KeyboardInterrupt:
580 raise
Guido van Rossumb940e112007-01-10 16:19:56 +0000581 except test_support.TestFailed as msg:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000582 print("test", test, "failed --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000583 sys.stdout.flush()
Guido van Rossum41360a41998-03-26 19:42:58 +0000584 return 0
Guido van Rossum9e48b271997-07-16 01:56:13 +0000585 except:
Guido van Rossum41360a41998-03-26 19:42:58 +0000586 type, value = sys.exc_info()[:2]
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000587 print("test", test, "crashed --", str(type) + ":", value)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000588 sys.stdout.flush()
Guido van Rossumce4a4752007-02-26 22:01:28 +0000589 if verbose or debug:
Guido van Rossum41360a41998-03-26 19:42:58 +0000590 traceback.print_exc(file=sys.stdout)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000591 sys.stdout.flush()
Guido van Rossum41360a41998-03-26 19:42:58 +0000592 return 0
Guido van Rossum152494a1996-12-20 03:12:20 +0000593 else:
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000594 if not cfp:
595 return 1
596 output = cfp.getvalue()
Fred Drakee51fe8d2001-05-29 17:10:51 +0000597 if generate:
Fred Drakee51fe8d2001-05-29 17:10:51 +0000598 if output == test + "\n":
599 if os.path.exists(outputfile):
600 # Write it since it already exists (and the contents
601 # may have changed), but let the user know it isn't
602 # needed:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000603 print("output file", outputfile, \
604 "is no longer needed; consider removing it")
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000605 else:
606 # We don't need it, so don't create it.
607 return 1
608 fp = open(outputfile, "w")
609 fp.write(output)
610 fp.close()
611 return 1
612 if os.path.exists(outputfile):
613 fp = open(outputfile, "r")
614 expected = fp.read()
615 fp.close()
616 else:
617 expected = test + "\n"
Michael W. Hudson61147f62004-08-03 11:33:28 +0000618 if output == expected or huntrleaks:
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000619 return 1
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000620 print("test", test, "produced unexpected output:")
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000621 sys.stdout.flush()
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000622 reportdiff(expected, output)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000623 sys.stdout.flush()
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000624 return 0
625
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000626def cleanup_test_droppings(testname, verbose):
627 import shutil
628
629 # Try to clean up junk commonly left behind. While tests shouldn't leave
630 # any files or directories behind, when a test fails that can be tedious
631 # for it to arrange. The consequences can be especially nasty on Windows,
632 # since if a test leaves a file open, it cannot be deleted by name (while
633 # there's nothing we can do about that here either, we can display the
634 # name of the offending test, which is a real help).
635 for name in (test_support.TESTFN,
636 "db_home",
637 ):
638 if not os.path.exists(name):
639 continue
640
641 if os.path.isdir(name):
642 kind, nuker = "directory", shutil.rmtree
643 elif os.path.isfile(name):
644 kind, nuker = "file", os.unlink
645 else:
646 raise SystemError("os.path says %r exists but is neither "
647 "directory nor file" % name)
648
649 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000650 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000651 try:
652 nuker(name)
Guido van Rossumb940e112007-01-10 16:19:56 +0000653 except Exception as msg:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000654 print(("%r left behind %s %r and it couldn't be "
655 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000656
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000657def dash_R(the_module, test, indirect_test, huntrleaks):
658 # This code is hackish and inelegant, but it seems to do the job.
659 import copy_reg
660
661 if not hasattr(sys, 'gettotalrefcount'):
662 raise Exception("Tracking reference leaks requires a debug build "
663 "of Python")
664
665 # Save current values for dash_R_cleanup() to restore.
666 fs = warnings.filters[:]
667 ps = copy_reg.dispatch_table.copy()
668 pic = sys.path_importer_cache.copy()
669
670 if indirect_test:
671 def run_the_test():
672 indirect_test()
673 else:
674 def run_the_test():
675 reload(the_module)
676
677 deltas = []
678 nwarmup, ntracked, fname = huntrleaks
679 repcount = nwarmup + ntracked
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000680 print("beginning", repcount, "repetitions", file=sys.stderr)
681 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000682 dash_R_cleanup(fs, ps, pic)
683 for i in range(repcount):
684 rc = sys.gettotalrefcount()
685 run_the_test()
686 sys.stderr.write('.')
687 dash_R_cleanup(fs, ps, pic)
688 if i >= nwarmup:
689 deltas.append(sys.gettotalrefcount() - rc - 2)
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000690 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000691 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000692 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
693 print(msg, file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000694 refrep = open(fname, "a")
Guido van Rossum360e4b82007-05-14 22:51:27 +0000695 print(msg, file=refrep)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000696 refrep.close()
697
698def dash_R_cleanup(fs, ps, pic):
699 import gc, copy_reg
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000700 import _strptime, linecache, dircache
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000701 import urlparse, urllib, urllib2, mimetypes, doctest
702 import struct, filecmp
703 from distutils.dir_util import _path_created
704
705 # Restore some original values.
706 warnings.filters[:] = fs
707 copy_reg.dispatch_table.clear()
708 copy_reg.dispatch_table.update(ps)
709 sys.path_importer_cache.clear()
710 sys.path_importer_cache.update(pic)
711
712 # Clear assorted module caches.
713 _path_created.clear()
714 re.purge()
715 _strptime._regex_cache.clear()
716 urlparse.clear_cache()
717 urllib.urlcleanup()
718 urllib2.install_opener(None)
719 dircache.reset()
720 linecache.clearcache()
721 mimetypes._default_mime_types()
722 struct._cache.clear()
723 filecmp._cache.clear()
724 doctest.master = None
725
726 # Collect cyclic trash.
727 gc.collect()
728
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000729def reportdiff(expected, output):
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000730 import difflib
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000731 print("*" * 70)
Tim Petersc377b162001-09-22 05:31:03 +0000732 a = expected.splitlines(1)
733 b = output.splitlines(1)
Guido van Rossumcf691932001-09-21 21:06:22 +0000734 sm = difflib.SequenceMatcher(a=a, b=b)
735 tuples = sm.get_opcodes()
Tim Petersc377b162001-09-22 05:31:03 +0000736
Guido van Rossumcf691932001-09-21 21:06:22 +0000737 def pair(x0, x1):
Tim Petersc377b162001-09-22 05:31:03 +0000738 # x0:x1 are 0-based slice indices; convert to 1-based line indices.
Guido van Rossumcf691932001-09-21 21:06:22 +0000739 x0 += 1
740 if x0 >= x1:
Tim Petersc377b162001-09-22 05:31:03 +0000741 return "line " + str(x0)
Guido van Rossumcf691932001-09-21 21:06:22 +0000742 else:
Tim Petersc377b162001-09-22 05:31:03 +0000743 return "lines %d-%d" % (x0, x1)
744
Guido van Rossumcf691932001-09-21 21:06:22 +0000745 for op, a0, a1, b0, b1 in tuples:
746 if op == 'equal':
747 pass
Tim Petersc377b162001-09-22 05:31:03 +0000748
Guido van Rossumcf691932001-09-21 21:06:22 +0000749 elif op == 'delete':
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000750 print("***", pair(a0, a1), "of expected output missing:")
Guido van Rossumcf691932001-09-21 21:06:22 +0000751 for line in a[a0:a1]:
Guido van Rossum08c47ba2007-02-09 20:50:08 +0000752 print("-", line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000753
Guido van Rossumcf691932001-09-21 21:06:22 +0000754 elif op == 'replace':
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000755 print("*** mismatch between", pair(a0, a1), "of expected", \
756 "output and", pair(b0, b1), "of actual output:")
Tim Petersc377b162001-09-22 05:31:03 +0000757 for line in difflib.ndiff(a[a0:a1], b[b0:b1]):
Guido van Rossum08c47ba2007-02-09 20:50:08 +0000758 print(line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000759
Guido van Rossumcf691932001-09-21 21:06:22 +0000760 elif op == 'insert':
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000761 print("***", pair(b0, b1), "of actual output doesn't appear", \
762 "in expected output after line", str(a1)+":")
Guido van Rossumcf691932001-09-21 21:06:22 +0000763 for line in b[b0:b1]:
Guido van Rossum08c47ba2007-02-09 20:50:08 +0000764 print("+", line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000765
Guido van Rossumcf691932001-09-21 21:06:22 +0000766 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000767 print("get_opcodes() returned bad tuple?!?!", (op, a0, a1, b0, b1))
Tim Petersc377b162001-09-22 05:31:03 +0000768
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000769 print("*" * 70)
Guido van Rossum152494a1996-12-20 03:12:20 +0000770
771def findtestdir():
772 if __name__ == '__main__':
Guido van Rossum41360a41998-03-26 19:42:58 +0000773 file = sys.argv[0]
Guido van Rossum152494a1996-12-20 03:12:20 +0000774 else:
Guido van Rossum41360a41998-03-26 19:42:58 +0000775 file = __file__
Guido van Rossum152494a1996-12-20 03:12:20 +0000776 testdir = os.path.dirname(file) or os.curdir
777 return testdir
778
Tim Petersc5000df2002-06-02 21:42:01 +0000779def removepy(name):
780 if name.endswith(os.extsep + "py"):
781 name = name[:-3]
782 return name
783
Guido van Rossum152494a1996-12-20 03:12:20 +0000784def count(n, word):
785 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +0000786 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +0000787 else:
Guido van Rossum41360a41998-03-26 19:42:58 +0000788 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +0000789
Tim Petersa45da922001-08-12 03:45:50 +0000790def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +0000791 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +0000792
793 Optional arg width (default 70) is the maximum line length.
794 Optional arg indent (default 4) is the number of blanks with which to
795 begin each line.
796 """
797
Tim Petersba78bc42002-07-04 19:45:06 +0000798 from textwrap import fill
799 blanks = ' ' * indent
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000800 print(fill(' '.join(map(str, x)), width,
801 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +0000802
Tim Petersde14a302002-04-01 05:04:46 +0000803# Map sys.platform to a string containing the basenames of tests
804# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +0000805#
806# Special cases:
807# test_pep277
808# The _ExpectedSkips constructor adds this to the set of expected
809# skips if not os.path.supports_unicode_filenames.
Tim Petersb4ee4eb2002-12-04 03:26:57 +0000810# test_socket_ssl
811# Controlled by test_socket_ssl.skip_expected. Requires the network
812# resource, and a socket module with ssl support.
Neal Norwitz55b61d22003-02-28 19:57:03 +0000813# test_timeout
814# Controlled by test_timeout.skip_expected. Requires the network
815# resource and a socket module.
Tim Petersde14a302002-04-01 05:04:46 +0000816
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000817_expectations = {
818 'win32':
819 """
Tim Petersc7c516a2003-09-20 22:06:13 +0000820 test__locale
Raymond Hettinger901dc982003-11-20 19:02:02 +0000821 test_applesingle
Tim Peters78e35f92002-11-22 20:00:34 +0000822 test_bsddb3
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000823 test_commands
824 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +0000825 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000826 test_dbm
827 test_dl
828 test_fcntl
829 test_fork1
830 test_gdbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000831 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +0000832 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000833 test_largefile
834 test_linuxaudiodev
835 test_mhlib
836 test_nis
837 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +0000838 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000839 test_poll
Tim Peters003eb302003-02-17 21:48:48 +0000840 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000841 test_pty
842 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +0000843 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000844 test_signal
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000845 test_sunaudiodev
Tim Peterscea2cc42004-08-04 02:32:03 +0000846 test_threadsignals
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000847 test_timing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000848 test_wait3
849 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000850 """,
851 'linux2':
852 """
Guido van Rossum944a6c32003-11-20 22:11:29 +0000853 test_applesingle
Guido van Rossumf66dacd2001-10-23 15:10:55 +0000854 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000855 test_dl
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000856 test_largefile
Guido van Rossum4507ec72003-02-14 19:29:22 +0000857 test_linuxaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000858 test_nis
859 test_ntpath
Guido van Rossum4507ec72003-02-14 19:29:22 +0000860 test_ossaudiodev
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000861 test_sqlite
862 test_startfile
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000863 test_sunaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000864 """,
Jack Jansen49a806e2001-08-28 14:49:00 +0000865 'mac':
Guido van Rossumaa782362001-09-02 03:58:41 +0000866 """
Jack Jansen67975142003-01-08 16:31:11 +0000867 test_atexit
Guido van Rossumaa782362001-09-02 03:58:41 +0000868 test_bsddb
Jack Jansen67975142003-01-08 16:31:11 +0000869 test_bsddb3
870 test_bz2
Guido van Rossumaa782362001-09-02 03:58:41 +0000871 test_commands
872 test_crypt
Jack Jansenb3be2162001-11-30 14:16:36 +0000873 test_curses
Guido van Rossumaa782362001-09-02 03:58:41 +0000874 test_dbm
875 test_dl
876 test_fcntl
877 test_fork1
Guido van Rossumaa782362001-09-02 03:58:41 +0000878 test_grp
Jack Jansenc4d6bdd2003-03-07 15:38:11 +0000879 test_ioctl
Guido van Rossumaa782362001-09-02 03:58:41 +0000880 test_largefile
881 test_linuxaudiodev
882 test_locale
883 test_mmap
884 test_nis
885 test_ntpath
886 test_openpty
Jack Jansen67975142003-01-08 16:31:11 +0000887 test_ossaudiodev
Guido van Rossumaa782362001-09-02 03:58:41 +0000888 test_poll
Jack Jansen67975142003-01-08 16:31:11 +0000889 test_popen
Guido van Rossumaa782362001-09-02 03:58:41 +0000890 test_popen2
Jack Jansen5bb97e62003-02-21 22:33:55 +0000891 test_posix
Guido van Rossumaa782362001-09-02 03:58:41 +0000892 test_pty
893 test_pwd
Jack Jansen67975142003-01-08 16:31:11 +0000894 test_resource
Guido van Rossumaa782362001-09-02 03:58:41 +0000895 test_signal
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000896 test_sqlite
897 test_startfile
Guido van Rossumaa782362001-09-02 03:58:41 +0000898 test_sunaudiodev
899 test_sundry
Jack Jansenc4d6bdd2003-03-07 15:38:11 +0000900 test_tarfile
Guido van Rossumaa782362001-09-02 03:58:41 +0000901 test_timing
Guido van Rossumaa782362001-09-02 03:58:41 +0000902 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000903 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000904 """
Guido van Rossum944a6c32003-11-20 22:11:29 +0000905 test_applesingle
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000906 test_bsddb
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000907 test_dl
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000908 test_largefile
909 test_linuxaudiodev
910 test_minidom
911 test_nis
912 test_ntpath
913 test_openpty
914 test_pyexpat
915 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000916 test_startfile
917 test_sqlite
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000918 test_sunaudiodev
919 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000920 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000921 'openunix8':
922 """
Guido van Rossum944a6c32003-11-20 22:11:29 +0000923 test_applesingle
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000924 test_bsddb
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000925 test_dl
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000926 test_largefile
927 test_linuxaudiodev
928 test_minidom
929 test_nis
930 test_ntpath
931 test_openpty
932 test_pyexpat
933 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000934 test_sqlite
935 test_startfile
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000936 test_sunaudiodev
937 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000938 """,
939 'sco_sv3':
940 """
Guido van Rossum944a6c32003-11-20 22:11:29 +0000941 test_applesingle
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000942 test_asynchat
943 test_bsddb
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000944 test_dl
945 test_fork1
946 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000947 test_largefile
948 test_linuxaudiodev
949 test_locale
950 test_minidom
951 test_nis
952 test_ntpath
953 test_openpty
954 test_pyexpat
955 test_queue
956 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000957 test_sqlite
958 test_startfile
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000959 test_sunaudiodev
960 test_sundry
961 test_thread
962 test_threaded_import
963 test_threadedtempfile
964 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000965 """,
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000966 'riscos':
967 """
Guido van Rossum944a6c32003-11-20 22:11:29 +0000968 test_applesingle
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000969 test_asynchat
Martin v. Löwisa94568a2003-05-10 07:36:56 +0000970 test_atexit
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000971 test_bsddb
Martin v. Löwisa94568a2003-05-10 07:36:56 +0000972 test_bsddb3
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000973 test_commands
974 test_crypt
975 test_dbm
976 test_dl
977 test_fcntl
978 test_fork1
979 test_gdbm
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000980 test_grp
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000981 test_largefile
982 test_linuxaudiodev
983 test_locale
984 test_mmap
985 test_nis
986 test_ntpath
987 test_openpty
988 test_poll
989 test_popen2
990 test_pty
991 test_pwd
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000992 test_strop
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000993 test_sqlite
994 test_startfile
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000995 test_sunaudiodev
996 test_sundry
997 test_thread
998 test_threaded_import
999 test_threadedtempfile
1000 test_threading
1001 test_timing
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001002 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001003 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001004 """
Jack Jansen398c2362001-12-02 21:41:36 +00001005 test_gdbm
Jack Jansen398c2362001-12-02 21:41:36 +00001006 test_largefile
1007 test_linuxaudiodev
Jack Jansenacda3392002-12-30 23:03:13 +00001008 test_locale
Jack Jansen398c2362001-12-02 21:41:36 +00001009 test_nis
Jack Jansenacda3392002-12-30 23:03:13 +00001010 test_ossaudiodev
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001011 test_startfile
Jack Jansen398c2362001-12-02 21:41:36 +00001012 test_sunaudiodev
Jack Jansen398c2362001-12-02 21:41:36 +00001013 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001014 'sunos5':
1015 """
Guido van Rossum944a6c32003-11-20 22:11:29 +00001016 test_applesingle
Guido van Rossum11c3f092002-07-17 15:08:24 +00001017 test_bsddb
Guido van Rossum11c3f092002-07-17 15:08:24 +00001018 test_curses
1019 test_dbm
Guido van Rossum11c3f092002-07-17 15:08:24 +00001020 test_gdbm
Guido van Rossum11c3f092002-07-17 15:08:24 +00001021 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001022 test_linuxaudiodev
Guido van Rossum11c3f092002-07-17 15:08:24 +00001023 test_openpty
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001024 test_sqlite
1025 test_startfile
Guido van Rossum11c3f092002-07-17 15:08:24 +00001026 test_zipfile
1027 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001028 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001029 'hp-ux11':
1030 """
Guido van Rossum944a6c32003-11-20 22:11:29 +00001031 test_applesingle
Skip Montanarob3230212002-03-15 02:54:03 +00001032 test_bsddb
Skip Montanarob3230212002-03-15 02:54:03 +00001033 test_curses
1034 test_dl
1035 test_gdbm
Skip Montanarob3230212002-03-15 02:54:03 +00001036 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001037 test_largefile
1038 test_linuxaudiodev
1039 test_locale
1040 test_minidom
1041 test_nis
1042 test_ntpath
1043 test_openpty
1044 test_pyexpat
1045 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001046 test_sqlite
1047 test_startfile
Skip Montanarob3230212002-03-15 02:54:03 +00001048 test_sunaudiodev
Skip Montanarob3230212002-03-15 02:54:03 +00001049 test_zipfile
1050 test_zlib
1051 """,
Martin v. Löwisf90ae202002-06-11 06:22:31 +00001052 'atheos':
Tim Petersc411dba2002-07-16 21:35:23 +00001053 """
Guido van Rossum944a6c32003-11-20 22:11:29 +00001054 test_applesingle
Tim Petersc411dba2002-07-16 21:35:23 +00001055 test_curses
1056 test_dl
Tim Petersc411dba2002-07-16 21:35:23 +00001057 test_gdbm
Tim Petersc411dba2002-07-16 21:35:23 +00001058 test_largefile
1059 test_linuxaudiodev
1060 test_locale
1061 test_mhlib
1062 test_mmap
Tim Petersc411dba2002-07-16 21:35:23 +00001063 test_nis
1064 test_poll
1065 test_popen2
1066 test_resource
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001067 test_sqlite
1068 test_startfile
Tim Petersc411dba2002-07-16 21:35:23 +00001069 test_sunaudiodev
Tim Petersc411dba2002-07-16 21:35:23 +00001070 """,
Jason Tishler25115942002-12-05 15:18:15 +00001071 'cygwin':
1072 """
Guido van Rossum944a6c32003-11-20 22:11:29 +00001073 test_applesingle
Tim Petersb0f89e02002-12-05 17:20:25 +00001074 test_bsddb3
Jason Tishler25115942002-12-05 15:18:15 +00001075 test_curses
1076 test_dbm
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001077 test_ioctl
Jason Tishler25115942002-12-05 15:18:15 +00001078 test_largefile
1079 test_linuxaudiodev
1080 test_locale
Jason Tishler25115942002-12-05 15:18:15 +00001081 test_nis
Jason Tishler5c4ded22003-02-05 16:46:01 +00001082 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001083 test_socketserver
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001084 test_sqlite
Jason Tishler25115942002-12-05 15:18:15 +00001085 test_sunaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001086 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001087 'os2emx':
1088 """
Guido van Rossum944a6c32003-11-20 22:11:29 +00001089 test_applesingle
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001090 test_audioop
1091 test_bsddb3
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001092 test_commands
1093 test_curses
1094 test_dl
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001095 test_largefile
1096 test_linuxaudiodev
1097 test_mhlib
1098 test_mmap
1099 test_nis
1100 test_openpty
1101 test_ossaudiodev
1102 test_pty
1103 test_resource
1104 test_signal
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001105 test_sqlite
1106 test_startfile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001107 test_sunaudiodev
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001108 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001109 'freebsd4':
1110 """
1111 test_aepack
Guido van Rossum944a6c32003-11-20 22:11:29 +00001112 test_applesingle
1113 test_bsddb
1114 test_bsddb3
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001115 test_gdbm
Guido van Rossum944a6c32003-11-20 22:11:29 +00001116 test_linuxaudiodev
1117 test_locale
1118 test_macfs
1119 test_macostools
1120 test_nis
Guido van Rossum944a6c32003-11-20 22:11:29 +00001121 test_ossaudiodev
1122 test_pep277
1123 test_plistlib
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001124 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001125 test_scriptpackages
1126 test_socket_ssl
1127 test_socketserver
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001128 test_sqlite
1129 test_startfile
Guido van Rossum944a6c32003-11-20 22:11:29 +00001130 test_sunaudiodev
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001131 test_tcl
Guido van Rossum944a6c32003-11-20 22:11:29 +00001132 test_timeout
1133 test_unicode_file
1134 test_urllibnet
1135 test_winreg
1136 test_winsound
Martin v. Löwis56f88112003-06-07 20:01:37 +00001137 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001138 'aix5':
1139 """
1140 test_aepack
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001141 test_applesingle
1142 test_bsddb
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001143 test_bsddb3
1144 test_bz2
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001145 test_dl
1146 test_gdbm
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001147 test_gzip
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001148 test_linuxaudiodev
1149 test_macfs
1150 test_macostools
1151 test_nis
1152 test_ossaudiodev
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001153 test_sqlite
1154 test_startfile
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001155 test_sunaudiodev
1156 test_tcl
1157 test_winreg
1158 test_winsound
1159 test_zipimport
1160 test_zlib
1161 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001162 'openbsd3':
1163 """
1164 test_aepack
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001165 test_applesingle
1166 test_bsddb
1167 test_bsddb3
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001168 test_ctypes
1169 test_dl
1170 test_gdbm
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001171 test_linuxaudiodev
1172 test_locale
1173 test_macfs
1174 test_macostools
1175 test_nis
1176 test_normalization
1177 test_ossaudiodev
1178 test_pep277
1179 test_plistlib
1180 test_scriptpackages
1181 test_tcl
1182 test_sqlite
1183 test_startfile
1184 test_sunaudiodev
1185 test_unicode_file
1186 test_winreg
1187 test_winsound
1188 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001189 'netbsd3':
1190 """
1191 test_aepack
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001192 test_applesingle
1193 test_bsddb
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001194 test_bsddb3
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001195 test_ctypes
1196 test_curses
1197 test_dl
1198 test_gdbm
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001199 test_linuxaudiodev
1200 test_locale
1201 test_macfs
1202 test_macostools
1203 test_nis
1204 test_ossaudiodev
1205 test_pep277
1206 test_sqlite
1207 test_startfile
1208 test_sunaudiodev
1209 test_tcl
1210 test_unicode_file
1211 test_winreg
1212 test_winsound
1213 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001214}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001215_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001216_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001217_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001218
Tim Petersb5b7b782001-08-12 01:20:39 +00001219class _ExpectedSkips:
1220 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001221 import os.path
Tim Petersb4ee4eb2002-12-04 03:26:57 +00001222 from test import test_socket_ssl
Neal Norwitz55b61d22003-02-28 19:57:03 +00001223 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001224
Tim Peters7c7efe92002-08-23 17:55:54 +00001225 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001226 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001227 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001228 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001229
Tim Peters2a182db2002-10-09 01:07:11 +00001230 if not os.path.supports_unicode_filenames:
1231 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001232
Tim Petersb4ee4eb2002-12-04 03:26:57 +00001233 if test_socket_ssl.skip_expected:
1234 self.expected.add('test_socket_ssl')
1235
Neal Norwitz55b61d22003-02-28 19:57:03 +00001236 if test_timeout.skip_expected:
1237 self.expected.add('test_timeout')
1238
Guido van Rossume2a383d2007-01-15 16:59:06 +00001239 if sys.maxint == 9223372036854775807:
Martin v. Löwisfba73692004-11-13 11:13:35 +00001240 self.expected.add('test_rgbimg')
1241 self.expected.add('test_imageop')
1242
Jack Jansen6afc5e02003-01-29 16:24:16 +00001243 if not sys.platform in ("mac", "darwin"):
Neal Norwitz7035c982003-03-29 22:01:17 +00001244 MAC_ONLY = ["test_macostools", "test_macfs", "test_aepack",
1245 "test_plistlib", "test_scriptpackages"]
1246 for skip in MAC_ONLY:
1247 self.expected.add(skip)
Tim Petersecd79eb2003-01-29 00:35:32 +00001248
1249 if sys.platform != "win32":
Neal Norwitz7035c982003-03-29 22:01:17 +00001250 WIN_ONLY = ["test_unicode_file", "test_winreg",
1251 "test_winsound"]
1252 for skip in WIN_ONLY:
1253 self.expected.add(skip)
Tim Petersf2715e02003-02-19 02:35:07 +00001254
Tim Peters7c7efe92002-08-23 17:55:54 +00001255 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001256
1257 def isvalid(self):
1258 "Return true iff _ExpectedSkips knows about the current platform."
1259 return self.valid
1260
1261 def getexpected(self):
1262 """Return set of test names we expect to skip on current platform.
1263
1264 self.isvalid() must be true.
1265 """
1266
1267 assert self.isvalid()
1268 return self.expected
1269
Guido van Rossum152494a1996-12-20 03:12:20 +00001270if __name__ == '__main__':
Barry Warsaw408b6d32002-07-30 23:27:12 +00001271 # Remove regrtest.py's own directory from the module search path. This
1272 # prevents relative imports from working, and relative imports will screw
1273 # up the testing framework. E.g. if both test.test_support and
1274 # test_support are imported, they will not contain the same globals, and
1275 # much of the testing framework relies on the globals in the
1276 # test.test_support module.
1277 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1278 i = pathlen = len(sys.path)
1279 while i >= 0:
1280 i -= 1
1281 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1282 del sys.path[i]
1283 if len(sys.path) == pathlen:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001284 print('Could not find %r in sys.path to remove it' % mydir)
Barry Warsaw08fca522001-08-20 22:33:46 +00001285 main()