blob: 87d9dc9c37fc4b565b2d0e4180b800749daf00f1 [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
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000131import cStringIO
132import 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:
Raymond Hettinger74e67662003-05-17 20:44:12 +0000543 cfp = cStringIO.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 Rossumbe19ed72007-02-09 05:37:30 +0000692 print(test, 'leaked', deltas, 'references', file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000693 refrep = open(fname, "a")
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000694 print(test, 'leaked', deltas, 'references', file=refrep)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000695 refrep.close()
696
697def dash_R_cleanup(fs, ps, pic):
698 import gc, copy_reg
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000699 import _strptime, linecache, dircache
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000700 import urlparse, urllib, urllib2, mimetypes, doctest
701 import struct, filecmp
702 from distutils.dir_util import _path_created
703
704 # Restore some original values.
705 warnings.filters[:] = fs
706 copy_reg.dispatch_table.clear()
707 copy_reg.dispatch_table.update(ps)
708 sys.path_importer_cache.clear()
709 sys.path_importer_cache.update(pic)
710
711 # Clear assorted module caches.
712 _path_created.clear()
713 re.purge()
714 _strptime._regex_cache.clear()
715 urlparse.clear_cache()
716 urllib.urlcleanup()
717 urllib2.install_opener(None)
718 dircache.reset()
719 linecache.clearcache()
720 mimetypes._default_mime_types()
721 struct._cache.clear()
722 filecmp._cache.clear()
723 doctest.master = None
724
725 # Collect cyclic trash.
726 gc.collect()
727
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000728def reportdiff(expected, output):
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000729 import difflib
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000730 print("*" * 70)
Tim Petersc377b162001-09-22 05:31:03 +0000731 a = expected.splitlines(1)
732 b = output.splitlines(1)
Guido van Rossumcf691932001-09-21 21:06:22 +0000733 sm = difflib.SequenceMatcher(a=a, b=b)
734 tuples = sm.get_opcodes()
Tim Petersc377b162001-09-22 05:31:03 +0000735
Guido van Rossumcf691932001-09-21 21:06:22 +0000736 def pair(x0, x1):
Tim Petersc377b162001-09-22 05:31:03 +0000737 # x0:x1 are 0-based slice indices; convert to 1-based line indices.
Guido van Rossumcf691932001-09-21 21:06:22 +0000738 x0 += 1
739 if x0 >= x1:
Tim Petersc377b162001-09-22 05:31:03 +0000740 return "line " + str(x0)
Guido van Rossumcf691932001-09-21 21:06:22 +0000741 else:
Tim Petersc377b162001-09-22 05:31:03 +0000742 return "lines %d-%d" % (x0, x1)
743
Guido van Rossumcf691932001-09-21 21:06:22 +0000744 for op, a0, a1, b0, b1 in tuples:
745 if op == 'equal':
746 pass
Tim Petersc377b162001-09-22 05:31:03 +0000747
Guido van Rossumcf691932001-09-21 21:06:22 +0000748 elif op == 'delete':
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000749 print("***", pair(a0, a1), "of expected output missing:")
Guido van Rossumcf691932001-09-21 21:06:22 +0000750 for line in a[a0:a1]:
Guido van Rossum08c47ba2007-02-09 20:50:08 +0000751 print("-", line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000752
Guido van Rossumcf691932001-09-21 21:06:22 +0000753 elif op == 'replace':
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000754 print("*** mismatch between", pair(a0, a1), "of expected", \
755 "output and", pair(b0, b1), "of actual output:")
Tim Petersc377b162001-09-22 05:31:03 +0000756 for line in difflib.ndiff(a[a0:a1], b[b0:b1]):
Guido van Rossum08c47ba2007-02-09 20:50:08 +0000757 print(line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000758
Guido van Rossumcf691932001-09-21 21:06:22 +0000759 elif op == 'insert':
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000760 print("***", pair(b0, b1), "of actual output doesn't appear", \
761 "in expected output after line", str(a1)+":")
Guido van Rossumcf691932001-09-21 21:06:22 +0000762 for line in b[b0:b1]:
Guido van Rossum08c47ba2007-02-09 20:50:08 +0000763 print("+", line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000764
Guido van Rossumcf691932001-09-21 21:06:22 +0000765 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000766 print("get_opcodes() returned bad tuple?!?!", (op, a0, a1, b0, b1))
Tim Petersc377b162001-09-22 05:31:03 +0000767
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000768 print("*" * 70)
Guido van Rossum152494a1996-12-20 03:12:20 +0000769
770def findtestdir():
771 if __name__ == '__main__':
Guido van Rossum41360a41998-03-26 19:42:58 +0000772 file = sys.argv[0]
Guido van Rossum152494a1996-12-20 03:12:20 +0000773 else:
Guido van Rossum41360a41998-03-26 19:42:58 +0000774 file = __file__
Guido van Rossum152494a1996-12-20 03:12:20 +0000775 testdir = os.path.dirname(file) or os.curdir
776 return testdir
777
Tim Petersc5000df2002-06-02 21:42:01 +0000778def removepy(name):
779 if name.endswith(os.extsep + "py"):
780 name = name[:-3]
781 return name
782
Guido van Rossum152494a1996-12-20 03:12:20 +0000783def count(n, word):
784 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +0000785 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +0000786 else:
Guido van Rossum41360a41998-03-26 19:42:58 +0000787 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +0000788
Tim Petersa45da922001-08-12 03:45:50 +0000789def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +0000790 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +0000791
792 Optional arg width (default 70) is the maximum line length.
793 Optional arg indent (default 4) is the number of blanks with which to
794 begin each line.
795 """
796
Tim Petersba78bc42002-07-04 19:45:06 +0000797 from textwrap import fill
798 blanks = ' ' * indent
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000799 print(fill(' '.join(map(str, x)), width,
800 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +0000801
Tim Petersde14a302002-04-01 05:04:46 +0000802# Map sys.platform to a string containing the basenames of tests
803# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +0000804#
805# Special cases:
806# test_pep277
807# The _ExpectedSkips constructor adds this to the set of expected
808# skips if not os.path.supports_unicode_filenames.
Tim Petersb4ee4eb2002-12-04 03:26:57 +0000809# test_socket_ssl
810# Controlled by test_socket_ssl.skip_expected. Requires the network
811# resource, and a socket module with ssl support.
Neal Norwitz55b61d22003-02-28 19:57:03 +0000812# test_timeout
813# Controlled by test_timeout.skip_expected. Requires the network
814# resource and a socket module.
Tim Petersde14a302002-04-01 05:04:46 +0000815
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000816_expectations = {
817 'win32':
818 """
Tim Petersc7c516a2003-09-20 22:06:13 +0000819 test__locale
Raymond Hettinger901dc982003-11-20 19:02:02 +0000820 test_applesingle
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000821 test_al
Skip Montanaro823ba282003-05-06 20:36:24 +0000822 test_bsddb185
Tim Peters78e35f92002-11-22 20:00:34 +0000823 test_bsddb3
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000824 test_cd
825 test_cl
826 test_commands
827 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +0000828 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000829 test_dbm
830 test_dl
831 test_fcntl
832 test_fork1
833 test_gdbm
834 test_gl
835 test_grp
836 test_imgfile
Tim Petersfd8e6e52003-03-04 00:26:38 +0000837 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000838 test_largefile
839 test_linuxaudiodev
840 test_mhlib
841 test_nis
842 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +0000843 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000844 test_poll
Tim Peters003eb302003-02-17 21:48:48 +0000845 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000846 test_pty
847 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +0000848 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000849 test_signal
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000850 test_sunaudiodev
Tim Peterscea2cc42004-08-04 02:32:03 +0000851 test_threadsignals
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000852 test_timing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000853 test_wait3
854 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000855 """,
856 'linux2':
857 """
858 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +0000859 test_applesingle
Skip Montanaro823ba282003-05-06 20:36:24 +0000860 test_bsddb185
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000861 test_cd
862 test_cl
Guido van Rossumf66dacd2001-10-23 15:10:55 +0000863 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000864 test_dl
865 test_gl
866 test_imgfile
867 test_largefile
Guido van Rossum4507ec72003-02-14 19:29:22 +0000868 test_linuxaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000869 test_nis
870 test_ntpath
Guido van Rossum4507ec72003-02-14 19:29:22 +0000871 test_ossaudiodev
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000872 test_sqlite
873 test_startfile
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000874 test_sunaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000875 """,
Jack Jansen49a806e2001-08-28 14:49:00 +0000876 'mac':
Guido van Rossumaa782362001-09-02 03:58:41 +0000877 """
878 test_al
Jack Jansen67975142003-01-08 16:31:11 +0000879 test_atexit
Guido van Rossumaa782362001-09-02 03:58:41 +0000880 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +0000881 test_bsddb185
Jack Jansen67975142003-01-08 16:31:11 +0000882 test_bsddb3
883 test_bz2
Guido van Rossumaa782362001-09-02 03:58:41 +0000884 test_cd
885 test_cl
886 test_commands
887 test_crypt
Jack Jansenb3be2162001-11-30 14:16:36 +0000888 test_curses
Guido van Rossumaa782362001-09-02 03:58:41 +0000889 test_dbm
890 test_dl
891 test_fcntl
892 test_fork1
893 test_gl
894 test_grp
Jack Jansenc4d6bdd2003-03-07 15:38:11 +0000895 test_ioctl
Guido van Rossumaa782362001-09-02 03:58:41 +0000896 test_imgfile
897 test_largefile
898 test_linuxaudiodev
899 test_locale
900 test_mmap
901 test_nis
902 test_ntpath
903 test_openpty
Jack Jansen67975142003-01-08 16:31:11 +0000904 test_ossaudiodev
Guido van Rossumaa782362001-09-02 03:58:41 +0000905 test_poll
Jack Jansen67975142003-01-08 16:31:11 +0000906 test_popen
Guido van Rossumaa782362001-09-02 03:58:41 +0000907 test_popen2
Jack Jansen5bb97e62003-02-21 22:33:55 +0000908 test_posix
Guido van Rossumaa782362001-09-02 03:58:41 +0000909 test_pty
910 test_pwd
Jack Jansen67975142003-01-08 16:31:11 +0000911 test_resource
Guido van Rossumaa782362001-09-02 03:58:41 +0000912 test_signal
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000913 test_sqlite
914 test_startfile
Guido van Rossumaa782362001-09-02 03:58:41 +0000915 test_sunaudiodev
916 test_sundry
Jack Jansenc4d6bdd2003-03-07 15:38:11 +0000917 test_tarfile
Guido van Rossumaa782362001-09-02 03:58:41 +0000918 test_timing
Guido van Rossumaa782362001-09-02 03:58:41 +0000919 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000920 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000921 """
922 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +0000923 test_applesingle
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000924 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +0000925 test_bsddb185
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000926 test_cd
927 test_cl
928 test_dl
929 test_gl
930 test_imgfile
931 test_largefile
932 test_linuxaudiodev
933 test_minidom
934 test_nis
935 test_ntpath
936 test_openpty
937 test_pyexpat
938 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000939 test_startfile
940 test_sqlite
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000941 test_sunaudiodev
942 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000943 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000944 'openunix8':
945 """
946 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +0000947 test_applesingle
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000948 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +0000949 test_bsddb185
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000950 test_cd
951 test_cl
952 test_dl
953 test_gl
954 test_imgfile
955 test_largefile
956 test_linuxaudiodev
957 test_minidom
958 test_nis
959 test_ntpath
960 test_openpty
961 test_pyexpat
962 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000963 test_sqlite
964 test_startfile
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000965 test_sunaudiodev
966 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000967 """,
968 'sco_sv3':
969 """
970 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +0000971 test_applesingle
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000972 test_asynchat
973 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +0000974 test_bsddb185
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000975 test_cd
976 test_cl
977 test_dl
978 test_fork1
979 test_gettext
980 test_gl
981 test_imgfile
982 test_largefile
983 test_linuxaudiodev
984 test_locale
985 test_minidom
986 test_nis
987 test_ntpath
988 test_openpty
989 test_pyexpat
990 test_queue
991 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000992 test_sqlite
993 test_startfile
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000994 test_sunaudiodev
995 test_sundry
996 test_thread
997 test_threaded_import
998 test_threadedtempfile
999 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001000 """,
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001001 'riscos':
1002 """
1003 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +00001004 test_applesingle
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001005 test_asynchat
Martin v. Löwisa94568a2003-05-10 07:36:56 +00001006 test_atexit
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001007 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +00001008 test_bsddb185
Martin v. Löwisa94568a2003-05-10 07:36:56 +00001009 test_bsddb3
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001010 test_cd
1011 test_cl
1012 test_commands
1013 test_crypt
1014 test_dbm
1015 test_dl
1016 test_fcntl
1017 test_fork1
1018 test_gdbm
1019 test_gl
1020 test_grp
1021 test_imgfile
1022 test_largefile
1023 test_linuxaudiodev
1024 test_locale
1025 test_mmap
1026 test_nis
1027 test_ntpath
1028 test_openpty
1029 test_poll
1030 test_popen2
1031 test_pty
1032 test_pwd
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001033 test_strop
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001034 test_sqlite
1035 test_startfile
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001036 test_sunaudiodev
1037 test_sundry
1038 test_thread
1039 test_threaded_import
1040 test_threadedtempfile
1041 test_threading
1042 test_timing
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001043 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001044 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001045 """
1046 test_al
1047 test_cd
1048 test_cl
Jack Jansen398c2362001-12-02 21:41:36 +00001049 test_gdbm
1050 test_gl
1051 test_imgfile
1052 test_largefile
1053 test_linuxaudiodev
Jack Jansenacda3392002-12-30 23:03:13 +00001054 test_locale
Jack Jansen398c2362001-12-02 21:41:36 +00001055 test_nis
Jack Jansenacda3392002-12-30 23:03:13 +00001056 test_ossaudiodev
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001057 test_startfile
Jack Jansen398c2362001-12-02 21:41:36 +00001058 test_sunaudiodev
Jack Jansen398c2362001-12-02 21:41:36 +00001059 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001060 'sunos5':
1061 """
1062 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +00001063 test_applesingle
Guido van Rossum11c3f092002-07-17 15:08:24 +00001064 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +00001065 test_bsddb185
Guido van Rossum11c3f092002-07-17 15:08:24 +00001066 test_cd
1067 test_cl
1068 test_curses
1069 test_dbm
Guido van Rossum11c3f092002-07-17 15:08:24 +00001070 test_gdbm
1071 test_gl
1072 test_gzip
1073 test_imgfile
1074 test_linuxaudiodev
Guido van Rossum11c3f092002-07-17 15:08:24 +00001075 test_openpty
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001076 test_sqlite
1077 test_startfile
Guido van Rossum11c3f092002-07-17 15:08:24 +00001078 test_zipfile
1079 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001080 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001081 'hp-ux11':
1082 """
1083 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +00001084 test_applesingle
Skip Montanarob3230212002-03-15 02:54:03 +00001085 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +00001086 test_bsddb185
Skip Montanarob3230212002-03-15 02:54:03 +00001087 test_cd
1088 test_cl
1089 test_curses
1090 test_dl
1091 test_gdbm
1092 test_gl
1093 test_gzip
1094 test_imgfile
1095 test_largefile
1096 test_linuxaudiodev
1097 test_locale
1098 test_minidom
1099 test_nis
1100 test_ntpath
1101 test_openpty
1102 test_pyexpat
1103 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001104 test_sqlite
1105 test_startfile
Skip Montanarob3230212002-03-15 02:54:03 +00001106 test_sunaudiodev
Skip Montanarob3230212002-03-15 02:54:03 +00001107 test_zipfile
1108 test_zlib
1109 """,
Martin v. Löwisf90ae202002-06-11 06:22:31 +00001110 'atheos':
Tim Petersc411dba2002-07-16 21:35:23 +00001111 """
1112 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +00001113 test_applesingle
Skip Montanaro823ba282003-05-06 20:36:24 +00001114 test_bsddb185
Tim Petersc411dba2002-07-16 21:35:23 +00001115 test_cd
1116 test_cl
1117 test_curses
1118 test_dl
Tim Petersc411dba2002-07-16 21:35:23 +00001119 test_gdbm
1120 test_gl
1121 test_imgfile
1122 test_largefile
1123 test_linuxaudiodev
1124 test_locale
1125 test_mhlib
1126 test_mmap
Tim Petersc411dba2002-07-16 21:35:23 +00001127 test_nis
1128 test_poll
1129 test_popen2
1130 test_resource
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001131 test_sqlite
1132 test_startfile
Tim Petersc411dba2002-07-16 21:35:23 +00001133 test_sunaudiodev
Tim Petersc411dba2002-07-16 21:35:23 +00001134 """,
Jason Tishler25115942002-12-05 15:18:15 +00001135 'cygwin':
1136 """
1137 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +00001138 test_applesingle
Skip Montanaro823ba282003-05-06 20:36:24 +00001139 test_bsddb185
Tim Petersb0f89e02002-12-05 17:20:25 +00001140 test_bsddb3
Jason Tishler25115942002-12-05 15:18:15 +00001141 test_cd
1142 test_cl
1143 test_curses
1144 test_dbm
Jason Tishler25115942002-12-05 15:18:15 +00001145 test_gl
1146 test_imgfile
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001147 test_ioctl
Jason Tishler25115942002-12-05 15:18:15 +00001148 test_largefile
1149 test_linuxaudiodev
1150 test_locale
Jason Tishler25115942002-12-05 15:18:15 +00001151 test_nis
Jason Tishler5c4ded22003-02-05 16:46:01 +00001152 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001153 test_socketserver
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001154 test_sqlite
Jason Tishler25115942002-12-05 15:18:15 +00001155 test_sunaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001156 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001157 'os2emx':
1158 """
1159 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +00001160 test_applesingle
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001161 test_audioop
Skip Montanaro823ba282003-05-06 20:36:24 +00001162 test_bsddb185
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001163 test_bsddb3
1164 test_cd
1165 test_cl
1166 test_commands
1167 test_curses
1168 test_dl
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001169 test_gl
1170 test_imgfile
1171 test_largefile
1172 test_linuxaudiodev
1173 test_mhlib
1174 test_mmap
1175 test_nis
1176 test_openpty
1177 test_ossaudiodev
1178 test_pty
1179 test_resource
1180 test_signal
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001181 test_sqlite
1182 test_startfile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001183 test_sunaudiodev
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001184 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001185 'freebsd4':
1186 """
1187 test_aepack
1188 test_al
1189 test_applesingle
1190 test_bsddb
1191 test_bsddb3
1192 test_cd
1193 test_cl
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001194 test_gdbm
Guido van Rossum944a6c32003-11-20 22:11:29 +00001195 test_gl
1196 test_imgfile
1197 test_linuxaudiodev
1198 test_locale
1199 test_macfs
1200 test_macostools
1201 test_nis
Guido van Rossum944a6c32003-11-20 22:11:29 +00001202 test_ossaudiodev
1203 test_pep277
1204 test_plistlib
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001205 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001206 test_scriptpackages
1207 test_socket_ssl
1208 test_socketserver
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001209 test_sqlite
1210 test_startfile
Guido van Rossum944a6c32003-11-20 22:11:29 +00001211 test_sunaudiodev
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001212 test_tcl
Guido van Rossum944a6c32003-11-20 22:11:29 +00001213 test_timeout
1214 test_unicode_file
1215 test_urllibnet
1216 test_winreg
1217 test_winsound
Martin v. Löwis56f88112003-06-07 20:01:37 +00001218 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001219 'aix5':
1220 """
1221 test_aepack
1222 test_al
1223 test_applesingle
1224 test_bsddb
1225 test_bsddb185
1226 test_bsddb3
1227 test_bz2
1228 test_cd
1229 test_cl
1230 test_dl
1231 test_gdbm
1232 test_gl
1233 test_gzip
1234 test_imgfile
1235 test_linuxaudiodev
1236 test_macfs
1237 test_macostools
1238 test_nis
1239 test_ossaudiodev
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001240 test_sqlite
1241 test_startfile
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001242 test_sunaudiodev
1243 test_tcl
1244 test_winreg
1245 test_winsound
1246 test_zipimport
1247 test_zlib
1248 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001249 'openbsd3':
1250 """
1251 test_aepack
1252 test_al
1253 test_applesingle
1254 test_bsddb
1255 test_bsddb3
1256 test_cd
1257 test_cl
1258 test_ctypes
1259 test_dl
1260 test_gdbm
1261 test_gl
1262 test_imgfile
1263 test_linuxaudiodev
1264 test_locale
1265 test_macfs
1266 test_macostools
1267 test_nis
1268 test_normalization
1269 test_ossaudiodev
1270 test_pep277
1271 test_plistlib
1272 test_scriptpackages
1273 test_tcl
1274 test_sqlite
1275 test_startfile
1276 test_sunaudiodev
1277 test_unicode_file
1278 test_winreg
1279 test_winsound
1280 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001281 'netbsd3':
1282 """
1283 test_aepack
1284 test_al
1285 test_applesingle
1286 test_bsddb
1287 test_bsddb185
1288 test_bsddb3
1289 test_cd
1290 test_cl
1291 test_ctypes
1292 test_curses
1293 test_dl
1294 test_gdbm
1295 test_gl
1296 test_imgfile
1297 test_linuxaudiodev
1298 test_locale
1299 test_macfs
1300 test_macostools
1301 test_nis
1302 test_ossaudiodev
1303 test_pep277
1304 test_sqlite
1305 test_startfile
1306 test_sunaudiodev
1307 test_tcl
1308 test_unicode_file
1309 test_winreg
1310 test_winsound
1311 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001312}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001313_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001314_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001315_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001316
Tim Petersb5b7b782001-08-12 01:20:39 +00001317class _ExpectedSkips:
1318 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001319 import os.path
Tim Petersb4ee4eb2002-12-04 03:26:57 +00001320 from test import test_socket_ssl
Neal Norwitz55b61d22003-02-28 19:57:03 +00001321 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001322
Tim Peters7c7efe92002-08-23 17:55:54 +00001323 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001324 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001325 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001326 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001327
Tim Peters2a182db2002-10-09 01:07:11 +00001328 if not os.path.supports_unicode_filenames:
1329 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001330
Tim Petersb4ee4eb2002-12-04 03:26:57 +00001331 if test_socket_ssl.skip_expected:
1332 self.expected.add('test_socket_ssl')
1333
Neal Norwitz55b61d22003-02-28 19:57:03 +00001334 if test_timeout.skip_expected:
1335 self.expected.add('test_timeout')
1336
Guido van Rossume2a383d2007-01-15 16:59:06 +00001337 if sys.maxint == 9223372036854775807:
Martin v. Löwisfba73692004-11-13 11:13:35 +00001338 self.expected.add('test_rgbimg')
1339 self.expected.add('test_imageop')
1340
Jack Jansen6afc5e02003-01-29 16:24:16 +00001341 if not sys.platform in ("mac", "darwin"):
Neal Norwitz7035c982003-03-29 22:01:17 +00001342 MAC_ONLY = ["test_macostools", "test_macfs", "test_aepack",
1343 "test_plistlib", "test_scriptpackages"]
1344 for skip in MAC_ONLY:
1345 self.expected.add(skip)
Tim Petersecd79eb2003-01-29 00:35:32 +00001346
1347 if sys.platform != "win32":
Neal Norwitz7035c982003-03-29 22:01:17 +00001348 WIN_ONLY = ["test_unicode_file", "test_winreg",
1349 "test_winsound"]
1350 for skip in WIN_ONLY:
1351 self.expected.add(skip)
Tim Petersf2715e02003-02-19 02:35:07 +00001352
Tim Peters7c7efe92002-08-23 17:55:54 +00001353 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001354
1355 def isvalid(self):
1356 "Return true iff _ExpectedSkips knows about the current platform."
1357 return self.valid
1358
1359 def getexpected(self):
1360 """Return set of test names we expect to skip on current platform.
1361
1362 self.isvalid() must be true.
1363 """
1364
1365 assert self.isvalid()
1366 return self.expected
1367
Guido van Rossum152494a1996-12-20 03:12:20 +00001368if __name__ == '__main__':
Barry Warsaw408b6d32002-07-30 23:27:12 +00001369 # Remove regrtest.py's own directory from the module search path. This
1370 # prevents relative imports from working, and relative imports will screw
1371 # up the testing framework. E.g. if both test.test_support and
1372 # test_support are imported, they will not contain the same globals, and
1373 # much of the testing framework relies on the globals in the
1374 # test.test_support module.
1375 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1376 i = pathlen = len(sys.path)
1377 while i >= 0:
1378 i -= 1
1379 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1380 del sys.path[i]
1381 if len(sys.path) == pathlen:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001382 print('Could not find %r in sys.path to remove it' % mydir)
Barry Warsaw08fca522001-08-20 22:33:46 +00001383 main()