blob: 1cd553c859051df7cefa96ffa8091c9b0c3b842f [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
Guido van Rossumebe3e162007-05-17 18:20:34 +0000110 compiler - Allow test_tokenize to verify round-trip lexing on
111 every file in the test library.
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000112
Tim Peterseba28be2005-03-28 01:08:02 +0000113 subprocess Run all tests for the subprocess module.
Peter Astrandf7f1bb72005-03-03 20:47:37 +0000114
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000115 urlfetch - It is okay to download files required on testing.
116
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000117To enable all resources except one, use '-uall,-<resource>'. For
118example, to run all the tests except for the bsddb tests, give the
119option '-uall,-bsddb'.
Guido van Rossum152494a1996-12-20 03:12:20 +0000120"""
121
Guido van Rossum152494a1996-12-20 03:12:20 +0000122import os
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000123import sys
Guido van Rossum152494a1996-12-20 03:12:20 +0000124import getopt
Skip Montanaroab1c7912000-06-30 16:39:27 +0000125import random
Guido van Rossumdc15c272002-08-12 21:55:51 +0000126import warnings
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000127import re
Guido van Rossum62b88712007-05-09 23:45:09 +0000128import StringIO
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000129import traceback
Guido van Rossumdc15c272002-08-12 21:55:51 +0000130
131# I see no other way to suppress these warnings;
132# putting them in test_grammar.py has no effect:
Guido van Rossum88b1def2002-08-14 17:54:48 +0000133warnings.filterwarnings("ignore", "hex/oct constants", FutureWarning,
Guido van Rossumdc15c272002-08-12 21:55:51 +0000134 ".*test.test_grammar$")
Guido van Rossumc34c4fc2002-09-19 00:42:16 +0000135if sys.maxint > 0x7fffffff:
136 # Also suppress them in <string>, because for 64-bit platforms,
137 # that's where test_grammar.py hides them.
138 warnings.filterwarnings("ignore", "hex/oct constants", FutureWarning,
139 "<string>")
Guido van Rossum152494a1996-12-20 03:12:20 +0000140
Thomas Wouters477c8d52006-05-27 19:21:47 +0000141# Ignore ImportWarnings that only occur in the source tree,
142# (because of modules with the same name as source-directories in Modules/)
143for mod in ("ctypes", "gzip", "zipfile", "tarfile", "encodings.zlib_codec",
144 "test.test_zipimport", "test.test_zlib", "test.test_zipfile",
145 "test.test_codecs", "test.string_tests"):
146 warnings.filterwarnings(module=".*%s$" % (mod,),
147 action="ignore", category=ImportWarning)
148
Guido van Rossumbb484652002-12-02 09:56:21 +0000149# MacOSX (a.k.a. Darwin) has a default stack size that is too small
150# for deeply recursive regular expressions. We see this as crashes in
151# the Python test suite when running test_re.py and test_sre.py. The
152# fix is to set the stack limit to 2048.
153# This approach may also be useful for other Unixy platforms that
154# suffer from small default stack limits.
155if sys.platform == 'darwin':
156 try:
157 import resource
158 except ImportError:
159 pass
160 else:
161 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
162 newsoft = min(hard, max(soft, 1024*2048))
163 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
164
Barry Warsaw04f357c2002-07-23 19:04:11 +0000165from test import test_support
Fred Drake3a15dac2002-04-11 16:39:16 +0000166
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000167RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network', 'bsddb',
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000168 'decimal', 'compiler', 'subprocess', 'urlfetch')
Fred Drake3a15dac2002-04-11 16:39:16 +0000169
170
Barry Warsaw08fca522001-08-20 22:33:46 +0000171def usage(code, msg=''):
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000172 print(__doc__)
173 if msg: print(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000174 sys.exit(code)
175
176
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000177def main(tests=None, testdir=None, verbose=0, quiet=False, generate=False,
178 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000179 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Guido van Rossumce4a4752007-02-26 22:01:28 +0000180 runleaks=False, huntrleaks=False, verbose2=False, debug=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000181 """Execute a test suite.
182
Thomas Wouters7e474022000-07-16 12:04:32 +0000183 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000184 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000185
186 tests -- a list of strings containing test names (optional)
187 testdir -- the directory in which to look for tests (optional)
188
189 Users other than the Python test suite will certainly want to
190 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000191 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000192
193 If the tests argument is omitted, the tests listed on the
194 command-line will be used. If that's empty, too, then all *.py
195 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000196
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000197 The other default arguments (verbose, quiet, generate, exclude, single,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000198 randomize, findleaks, use_resources, trace and coverdir) allow programmers
199 calling main() directly to set the values that would normally be set by
200 flags on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000201 """
Fred Drake004d5e62000-10-23 17:22:08 +0000202
Tim Peters8dee8092001-09-25 20:05:11 +0000203 test_support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000204 try:
Guido van Rossumce4a4752007-02-26 22:01:28 +0000205 opts, args = getopt.getopt(sys.argv[1:], 'dhvgqxsrf:lu:t:TD:NLR:wM:',
Barry Warsaw08fca522001-08-20 22:33:46 +0000206 ['help', 'verbose', 'quiet', 'generate',
Tim Petersc5000df2002-06-02 21:42:01 +0000207 'exclude', 'single', 'random', 'fromfile',
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000208 'findleaks', 'use=', 'threshold=', 'trace',
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000209 'coverdir=', 'nocoverdir', 'runleaks',
Thomas Wouters477c8d52006-05-27 19:21:47 +0000210 'huntrleaks=', 'verbose2', 'memlimit=',
Guido van Rossumce4a4752007-02-26 22:01:28 +0000211 'debug',
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000212 ])
Guido van Rossumb940e112007-01-10 16:19:56 +0000213 except getopt.error as msg:
Barry Warsaw08fca522001-08-20 22:33:46 +0000214 usage(2, msg)
215
216 # Defaults
217 if use_resources is None:
218 use_resources = []
Guido van Rossum152494a1996-12-20 03:12:20 +0000219 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000220 if o in ('-h', '--help'):
221 usage(0)
222 elif o in ('-v', '--verbose'):
223 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000224 elif o in ('-w', '--verbose2'):
225 verbose2 = True
Guido van Rossumce4a4752007-02-26 22:01:28 +0000226 elif o in ('-d', '--debug'):
227 debug = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000228 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000229 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000230 verbose = 0
231 elif o in ('-g', '--generate'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000232 generate = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000233 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000234 exclude = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000235 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000236 single = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000237 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000238 randomize = True
Tim Petersc5000df2002-06-02 21:42:01 +0000239 elif o in ('-f', '--fromfile'):
240 fromfile = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000241 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000242 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000243 elif o in ('-L', '--runleaks'):
244 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000245 elif o in ('-t', '--threshold'):
246 import gc
247 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000248 elif o in ('-T', '--coverage'):
249 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000250 elif o in ('-D', '--coverdir'):
251 coverdir = os.path.join(os.getcwd(), a)
252 elif o in ('-N', '--nocoverdir'):
253 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000254 elif o in ('-R', '--huntrleaks'):
255 huntrleaks = a.split(':')
256 if len(huntrleaks) != 3:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000257 print(a, huntrleaks)
Michael W. Hudson61147f62004-08-03 11:33:28 +0000258 usage(2, '-R takes three colon-separated arguments')
259 if len(huntrleaks[0]) == 0:
260 huntrleaks[0] = 5
261 else:
262 huntrleaks[0] = int(huntrleaks[0])
263 if len(huntrleaks[1]) == 0:
264 huntrleaks[1] = 4
265 else:
266 huntrleaks[1] = int(huntrleaks[1])
267 if len(huntrleaks[2]) == 0:
268 huntrleaks[2] = "reflog.txt"
Thomas Wouters477c8d52006-05-27 19:21:47 +0000269 elif o in ('-M', '--memlimit'):
270 test_support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000271 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000272 u = [x.lower() for x in a.split(',')]
273 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000274 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000275 use_resources[:] = RESOURCE_NAMES
276 continue
277 remove = False
278 if r[0] == '-':
279 remove = True
280 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000281 if r not in RESOURCE_NAMES:
282 usage(1, 'Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000283 if remove:
284 if r in use_resources:
285 use_resources.remove(r)
286 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000287 use_resources.append(r)
Guido van Rossuma4122201997-08-18 20:08:24 +0000288 if generate and verbose:
Barry Warsaw08fca522001-08-20 22:33:46 +0000289 usage(2, "-g and -v don't go together!")
Tim Petersc5000df2002-06-02 21:42:01 +0000290 if single and fromfile:
291 usage(2, "-s and -f don't go together!")
Barry Warsaw08fca522001-08-20 22:33:46 +0000292
Guido van Rossum152494a1996-12-20 03:12:20 +0000293 good = []
294 bad = []
295 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000296 resource_denieds = []
Barry Warsawe11e3de1999-01-28 19:51:51 +0000297
Neil Schemenauerd569f232000-09-22 15:29:28 +0000298 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000299 try:
300 import gc
301 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000302 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000303 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000304 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000305 # Uncomment the line below to report garbage that is not
306 # freeable by reference counting alone. By default only
307 # garbage that is not collectable by the GC is reported.
308 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000309 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000310
Barry Warsawe11e3de1999-01-28 19:51:51 +0000311 if single:
312 from tempfile import gettempdir
313 filename = os.path.join(gettempdir(), 'pynexttest')
314 try:
315 fp = open(filename, 'r')
Eric S. Raymondfc170b12001-02-09 11:51:27 +0000316 next = fp.read().strip()
Barry Warsawe11e3de1999-01-28 19:51:51 +0000317 tests = [next]
318 fp.close()
319 except IOError:
320 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000321
322 if fromfile:
323 tests = []
324 fp = open(fromfile)
325 for line in fp:
326 guts = line.split() # assuming no test has whitespace in its name
327 if guts and not guts[0].startswith('#'):
328 tests.extend(guts)
329 fp.close()
330
331 # Strip .py extensions.
332 if args:
333 args = map(removepy, args)
334 if tests:
335 tests = map(removepy, tests)
336
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000337 stdtests = STDTESTS[:]
338 nottests = NOTTESTS[:]
Guido van Rossum152494a1996-12-20 03:12:20 +0000339 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000340 for arg in args:
341 if arg in stdtests:
342 stdtests.remove(arg)
343 nottests[:0] = args
Guido van Rossum41360a41998-03-26 19:42:58 +0000344 args = []
Guido van Rossum747e1ca1998-08-24 13:48:36 +0000345 tests = tests or args or findtests(testdir, stdtests, nottests)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000346 if single:
347 tests = tests[:1]
Skip Montanaroab1c7912000-06-30 16:39:27 +0000348 if randomize:
349 random.shuffle(tests)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000350 if trace:
351 import trace
352 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix],
353 trace=False, count=True)
Guido van Rossum41360a41998-03-26 19:42:58 +0000354 test_support.verbose = verbose # Tell tests to be moderately quiet
Barry Warsaw08fca522001-08-20 22:33:46 +0000355 test_support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000356 save_modules = sys.modules.keys()
Guido van Rossum152494a1996-12-20 03:12:20 +0000357 for test in tests:
Guido van Rossum41360a41998-03-26 19:42:58 +0000358 if not quiet:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000359 print(test)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000360 sys.stdout.flush()
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000361 if trace:
362 # If we're tracing code coverage, then we don't exit with status
363 # if on a false return value from main.
364 tracer.runctx('runtest(test, generate, verbose, quiet, testdir)',
365 globals=globals(), locals=vars())
Guido van Rossum41360a41998-03-26 19:42:58 +0000366 else:
Neal Norwitz14ca3272006-02-28 18:05:43 +0000367 try:
368 ok = runtest(test, generate, verbose, quiet, testdir,
369 huntrleaks)
370 except KeyboardInterrupt:
371 # print a newline separate from the ^C
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000372 print()
Neal Norwitz14ca3272006-02-28 18:05:43 +0000373 break
374 except:
375 raise
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000376 if ok > 0:
377 good.append(test)
378 elif ok == 0:
379 bad.append(test)
380 else:
381 skipped.append(test)
382 if ok == -2:
383 resource_denieds.append(test)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000384 if findleaks:
385 gc.collect()
386 if gc.garbage:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000387 print("Warning: test created", len(gc.garbage), end=' ')
388 print("uncollectable object(s).")
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000389 # move the uncollectable objects somewhere so we don't see
390 # them again
Neil Schemenauerd569f232000-09-22 15:29:28 +0000391 found_garbage.extend(gc.garbage)
392 del gc.garbage[:]
Guido van Rossum5796d262000-04-21 21:35:06 +0000393 # Unload the newly imported modules (best effort finalization)
394 for module in sys.modules.keys():
Guido van Rossum51931142000-05-05 14:27:39 +0000395 if module not in save_modules and module.startswith("test."):
Guido van Rossum5796d262000-04-21 21:35:06 +0000396 test_support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000397
398 # The lists won't be sorted if running with -r
399 good.sort()
400 bad.sort()
401 skipped.sort()
Tim Peterse0c446b2001-10-18 21:57:37 +0000402
Guido van Rossum152494a1996-12-20 03:12:20 +0000403 if good and not quiet:
Guido van Rossum41360a41998-03-26 19:42:58 +0000404 if not bad and not skipped and len(good) > 1:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000405 print("All", end=' ')
406 print(count(len(good), "test"), "OK.")
Tim Peters1a4d77b2000-12-30 22:21:22 +0000407 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000408 print("CAUTION: stdout isn't compared in verbose mode:")
409 print("a test that passes in verbose mode may fail without it.")
Guido van Rossum152494a1996-12-20 03:12:20 +0000410 if bad:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000411 print(count(len(bad), "test"), "failed:")
Tim Petersa45da922001-08-12 03:45:50 +0000412 printlist(bad)
Guido van Rossum152494a1996-12-20 03:12:20 +0000413 if skipped and not quiet:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000414 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000415 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000416
Tim Petersb5b7b782001-08-12 01:20:39 +0000417 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000418 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000419 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000420 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000421 if surprise:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000422 print(count(len(surprise), "skip"), \
423 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000424 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000425 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000426 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000427 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000428 print("Ask someone to teach regrtest.py about which tests are")
429 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000430
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000431 if verbose2 and bad:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000432 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000433 for test in bad:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000434 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000435 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000436 try:
437 test_support.verbose = 1
438 ok = runtest(test, generate, 1, quiet, testdir,
Guido van Rossumce4a4752007-02-26 22:01:28 +0000439 huntrleaks, debug)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000440 except KeyboardInterrupt:
441 # print a newline separate from the ^C
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000442 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000443 break
444 except:
445 raise
446
Barry Warsawe11e3de1999-01-28 19:51:51 +0000447 if single:
448 alltests = findtests(testdir, stdtests, nottests)
449 for i in range(len(alltests)):
450 if tests[0] == alltests[i]:
451 if i == len(alltests) - 1:
452 os.unlink(filename)
453 else:
454 fp = open(filename, 'w')
455 fp.write(alltests[i+1] + '\n')
456 fp.close()
457 break
458 else:
459 os.unlink(filename)
460
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000461 if trace:
462 r = tracer.results()
463 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
464
Skip Montanaro0179a182004-06-06 15:53:18 +0000465 if runleaks:
466 os.system("leaks %d" % os.getpid())
467
Tim Peters5943b4a2003-07-23 00:30:39 +0000468 sys.exit(len(bad) > 0)
Barry Warsaw08fca522001-08-20 22:33:46 +0000469
Guido van Rossum152494a1996-12-20 03:12:20 +0000470
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000471STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000472 'test_grammar',
473 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000474 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000475 'test_builtin',
476 'test_exceptions',
477 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000478 'test_unittest',
479 'test_doctest',
480 'test_doctest2',
Guido van Rossum152494a1996-12-20 03:12:20 +0000481 ]
482
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000483NOTTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000484 'test_support',
Jeremy Hylton62e2c7e2001-02-28 17:48:06 +0000485 'test_future1',
486 'test_future2',
Jeremy Hylton8471a352001-08-20 20:33:42 +0000487 'test_future3',
Guido van Rossum152494a1996-12-20 03:12:20 +0000488 ]
489
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000490def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000491 """Return a list of all applicable test modules."""
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000492 if not testdir: testdir = findtestdir()
Guido van Rossum152494a1996-12-20 03:12:20 +0000493 names = os.listdir(testdir)
494 tests = []
495 for name in names:
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000496 if name[:5] == "test_" and name[-3:] == os.extsep+"py":
Guido van Rossum41360a41998-03-26 19:42:58 +0000497 modname = name[:-3]
498 if modname not in stdtests and modname not in nottests:
499 tests.append(modname)
Guido van Rossum152494a1996-12-20 03:12:20 +0000500 tests.sort()
501 return stdtests + tests
502
Guido van Rossumce4a4752007-02-26 22:01:28 +0000503def runtest(test, generate, verbose, quiet, testdir=None,
504 huntrleaks=False, debug=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000505 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000506
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000507 test -- the name of the test
508 generate -- if true, generate output, instead of running the test
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000509 and comparing it to a previously created output file
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000510 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000511 quiet -- if true, don't print 'skipped' messages (probably redundant)
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000512 testdir -- test directory
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000513 huntrleaks -- run multiple times to test for leaks; requires a debug
514 build; a triple corresponding to -R's three arguments
Guido van Rossumce4a4752007-02-26 22:01:28 +0000515 debug -- if true, print tracebacks for failed tests regardless of
516 verbose setting
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000517 Return:
518 -2 test skipped because resource denied
519 -1 test skipped for some other reason
520 0 test failed
521 1 test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000522 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000523
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000524 try:
525 return runtest_inner(test, generate, verbose, quiet, testdir,
Guido van Rossumce4a4752007-02-26 22:01:28 +0000526 huntrleaks, debug)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000527 finally:
528 cleanup_test_droppings(test, verbose)
529
530def runtest_inner(test, generate, verbose, quiet,
Guido van Rossumce4a4752007-02-26 22:01:28 +0000531 testdir=None, huntrleaks=False, debug=False):
Guido van Rossum152494a1996-12-20 03:12:20 +0000532 test_support.unload(test)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000533 if not testdir:
534 testdir = findtestdir()
Guido van Rossum152494a1996-12-20 03:12:20 +0000535 outputdir = os.path.join(testdir, "output")
536 outputfile = os.path.join(outputdir, test)
Tim Peters9390cc12001-09-28 20:14:46 +0000537 if verbose:
Guido van Rossum41360a41998-03-26 19:42:58 +0000538 cfp = None
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000539 else:
Guido van Rossum62b88712007-05-09 23:45:09 +0000540 cfp = StringIO.StringIO() # XXX Should use io.StringIO()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000541
Guido van Rossum152494a1996-12-20 03:12:20 +0000542 try:
Tim Peters342ca752001-09-25 19:13:20 +0000543 save_stdout = sys.stdout
Guido van Rossum41360a41998-03-26 19:42:58 +0000544 try:
545 if cfp:
546 sys.stdout = cfp
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000547 print(test) # Output file starts with test name
Barry Warsaw408b6d32002-07-30 23:27:12 +0000548 if test.startswith('test.'):
549 abstest = test
550 else:
551 # Always import it from the test package
552 abstest = 'test.' + test
553 the_package = __import__(abstest, globals(), locals(), [])
554 the_module = getattr(the_package, test)
Tim Petersd9742212001-05-22 18:28:25 +0000555 # Most tests run to completion simply as a side-effect of
556 # being imported. For the benefit of tests that can't run
557 # that way (like test_threaded_import), explicitly invoke
558 # their test_main() function (if it exists).
559 indirect_test = getattr(the_module, "test_main", None)
560 if indirect_test is not None:
561 indirect_test()
Michael W. Hudson61147f62004-08-03 11:33:28 +0000562 if huntrleaks:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000563 dash_R(the_module, test, indirect_test, huntrleaks)
Guido van Rossum41360a41998-03-26 19:42:58 +0000564 finally:
Tim Peters342ca752001-09-25 19:13:20 +0000565 sys.stdout = save_stdout
Guido van Rossumb940e112007-01-10 16:19:56 +0000566 except test_support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +0000567 if not quiet:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000568 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +0000569 sys.stdout.flush()
570 return -2
Guido van Rossumb940e112007-01-10 16:19:56 +0000571 except (ImportError, test_support.TestSkipped) as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +0000572 if not quiet:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000573 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000574 sys.stdout.flush()
Guido van Rossum41360a41998-03-26 19:42:58 +0000575 return -1
Fred Drakefe5c22a2000-08-18 16:04:05 +0000576 except KeyboardInterrupt:
577 raise
Guido van Rossumb940e112007-01-10 16:19:56 +0000578 except test_support.TestFailed as msg:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000579 print("test", test, "failed --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000580 sys.stdout.flush()
Guido van Rossum41360a41998-03-26 19:42:58 +0000581 return 0
Guido van Rossum9e48b271997-07-16 01:56:13 +0000582 except:
Guido van Rossum41360a41998-03-26 19:42:58 +0000583 type, value = sys.exc_info()[:2]
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000584 print("test", test, "crashed --", str(type) + ":", value)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000585 sys.stdout.flush()
Guido van Rossumce4a4752007-02-26 22:01:28 +0000586 if verbose or debug:
Guido van Rossum41360a41998-03-26 19:42:58 +0000587 traceback.print_exc(file=sys.stdout)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000588 sys.stdout.flush()
Guido van Rossum41360a41998-03-26 19:42:58 +0000589 return 0
Guido van Rossum152494a1996-12-20 03:12:20 +0000590 else:
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000591 if not cfp:
592 return 1
593 output = cfp.getvalue()
Fred Drakee51fe8d2001-05-29 17:10:51 +0000594 if generate:
Fred Drakee51fe8d2001-05-29 17:10:51 +0000595 if output == test + "\n":
596 if os.path.exists(outputfile):
597 # Write it since it already exists (and the contents
598 # may have changed), but let the user know it isn't
599 # needed:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000600 print("output file", outputfile, \
601 "is no longer needed; consider removing it")
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000602 else:
603 # We don't need it, so don't create it.
604 return 1
605 fp = open(outputfile, "w")
606 fp.write(output)
607 fp.close()
608 return 1
609 if os.path.exists(outputfile):
610 fp = open(outputfile, "r")
611 expected = fp.read()
612 fp.close()
613 else:
614 expected = test + "\n"
Michael W. Hudson61147f62004-08-03 11:33:28 +0000615 if output == expected or huntrleaks:
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000616 return 1
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000617 print("test", test, "produced unexpected output:")
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000618 sys.stdout.flush()
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000619 reportdiff(expected, output)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000620 sys.stdout.flush()
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000621 return 0
622
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000623def cleanup_test_droppings(testname, verbose):
624 import shutil
625
626 # Try to clean up junk commonly left behind. While tests shouldn't leave
627 # any files or directories behind, when a test fails that can be tedious
628 # for it to arrange. The consequences can be especially nasty on Windows,
629 # since if a test leaves a file open, it cannot be deleted by name (while
630 # there's nothing we can do about that here either, we can display the
631 # name of the offending test, which is a real help).
632 for name in (test_support.TESTFN,
633 "db_home",
634 ):
635 if not os.path.exists(name):
636 continue
637
638 if os.path.isdir(name):
639 kind, nuker = "directory", shutil.rmtree
640 elif os.path.isfile(name):
641 kind, nuker = "file", os.unlink
642 else:
643 raise SystemError("os.path says %r exists but is neither "
644 "directory nor file" % name)
645
646 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000647 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000648 try:
649 nuker(name)
Guido van Rossumb940e112007-01-10 16:19:56 +0000650 except Exception as msg:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000651 print(("%r left behind %s %r and it couldn't be "
652 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000653
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000654def dash_R(the_module, test, indirect_test, huntrleaks):
655 # This code is hackish and inelegant, but it seems to do the job.
656 import copy_reg
657
658 if not hasattr(sys, 'gettotalrefcount'):
659 raise Exception("Tracking reference leaks requires a debug build "
660 "of Python")
661
662 # Save current values for dash_R_cleanup() to restore.
663 fs = warnings.filters[:]
664 ps = copy_reg.dispatch_table.copy()
665 pic = sys.path_importer_cache.copy()
666
667 if indirect_test:
668 def run_the_test():
669 indirect_test()
670 else:
671 def run_the_test():
672 reload(the_module)
673
674 deltas = []
675 nwarmup, ntracked, fname = huntrleaks
676 repcount = nwarmup + ntracked
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000677 print("beginning", repcount, "repetitions", file=sys.stderr)
678 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000679 dash_R_cleanup(fs, ps, pic)
680 for i in range(repcount):
681 rc = sys.gettotalrefcount()
682 run_the_test()
683 sys.stderr.write('.')
684 dash_R_cleanup(fs, ps, pic)
685 if i >= nwarmup:
686 deltas.append(sys.gettotalrefcount() - rc - 2)
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000687 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000688 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000689 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
690 print(msg, file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000691 refrep = open(fname, "a")
Guido van Rossum360e4b82007-05-14 22:51:27 +0000692 print(msg, file=refrep)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000693 refrep.close()
694
695def dash_R_cleanup(fs, ps, pic):
696 import gc, copy_reg
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000697 import _strptime, linecache, dircache
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000698 import urlparse, urllib, urllib2, mimetypes, doctest
699 import struct, filecmp
700 from distutils.dir_util import _path_created
701
702 # Restore some original values.
703 warnings.filters[:] = fs
704 copy_reg.dispatch_table.clear()
705 copy_reg.dispatch_table.update(ps)
706 sys.path_importer_cache.clear()
707 sys.path_importer_cache.update(pic)
708
709 # Clear assorted module caches.
710 _path_created.clear()
711 re.purge()
712 _strptime._regex_cache.clear()
713 urlparse.clear_cache()
714 urllib.urlcleanup()
715 urllib2.install_opener(None)
716 dircache.reset()
717 linecache.clearcache()
718 mimetypes._default_mime_types()
719 struct._cache.clear()
720 filecmp._cache.clear()
721 doctest.master = None
722
723 # Collect cyclic trash.
724 gc.collect()
725
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000726def reportdiff(expected, output):
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000727 import difflib
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000728 print("*" * 70)
Tim Petersc377b162001-09-22 05:31:03 +0000729 a = expected.splitlines(1)
730 b = output.splitlines(1)
Guido van Rossumcf691932001-09-21 21:06:22 +0000731 sm = difflib.SequenceMatcher(a=a, b=b)
732 tuples = sm.get_opcodes()
Tim Petersc377b162001-09-22 05:31:03 +0000733
Guido van Rossumcf691932001-09-21 21:06:22 +0000734 def pair(x0, x1):
Tim Petersc377b162001-09-22 05:31:03 +0000735 # x0:x1 are 0-based slice indices; convert to 1-based line indices.
Guido van Rossumcf691932001-09-21 21:06:22 +0000736 x0 += 1
737 if x0 >= x1:
Tim Petersc377b162001-09-22 05:31:03 +0000738 return "line " + str(x0)
Guido van Rossumcf691932001-09-21 21:06:22 +0000739 else:
Tim Petersc377b162001-09-22 05:31:03 +0000740 return "lines %d-%d" % (x0, x1)
741
Guido van Rossumcf691932001-09-21 21:06:22 +0000742 for op, a0, a1, b0, b1 in tuples:
743 if op == 'equal':
744 pass
Tim Petersc377b162001-09-22 05:31:03 +0000745
Guido van Rossumcf691932001-09-21 21:06:22 +0000746 elif op == 'delete':
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000747 print("***", pair(a0, a1), "of expected output missing:")
Guido van Rossumcf691932001-09-21 21:06:22 +0000748 for line in a[a0:a1]:
Guido van Rossum08c47ba2007-02-09 20:50:08 +0000749 print("-", line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000750
Guido van Rossumcf691932001-09-21 21:06:22 +0000751 elif op == 'replace':
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000752 print("*** mismatch between", pair(a0, a1), "of expected", \
753 "output and", pair(b0, b1), "of actual output:")
Tim Petersc377b162001-09-22 05:31:03 +0000754 for line in difflib.ndiff(a[a0:a1], b[b0:b1]):
Guido van Rossum08c47ba2007-02-09 20:50:08 +0000755 print(line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000756
Guido van Rossumcf691932001-09-21 21:06:22 +0000757 elif op == 'insert':
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000758 print("***", pair(b0, b1), "of actual output doesn't appear", \
759 "in expected output after line", str(a1)+":")
Guido van Rossumcf691932001-09-21 21:06:22 +0000760 for line in b[b0:b1]:
Guido van Rossum08c47ba2007-02-09 20:50:08 +0000761 print("+", line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000762
Guido van Rossumcf691932001-09-21 21:06:22 +0000763 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000764 print("get_opcodes() returned bad tuple?!?!", (op, a0, a1, b0, b1))
Tim Petersc377b162001-09-22 05:31:03 +0000765
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000766 print("*" * 70)
Guido van Rossum152494a1996-12-20 03:12:20 +0000767
768def findtestdir():
769 if __name__ == '__main__':
Guido van Rossum41360a41998-03-26 19:42:58 +0000770 file = sys.argv[0]
Guido van Rossum152494a1996-12-20 03:12:20 +0000771 else:
Guido van Rossum41360a41998-03-26 19:42:58 +0000772 file = __file__
Guido van Rossum152494a1996-12-20 03:12:20 +0000773 testdir = os.path.dirname(file) or os.curdir
774 return testdir
775
Tim Petersc5000df2002-06-02 21:42:01 +0000776def removepy(name):
777 if name.endswith(os.extsep + "py"):
778 name = name[:-3]
779 return name
780
Guido van Rossum152494a1996-12-20 03:12:20 +0000781def count(n, word):
782 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +0000783 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +0000784 else:
Guido van Rossum41360a41998-03-26 19:42:58 +0000785 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +0000786
Tim Petersa45da922001-08-12 03:45:50 +0000787def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +0000788 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +0000789
790 Optional arg width (default 70) is the maximum line length.
791 Optional arg indent (default 4) is the number of blanks with which to
792 begin each line.
793 """
794
Tim Petersba78bc42002-07-04 19:45:06 +0000795 from textwrap import fill
796 blanks = ' ' * indent
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000797 print(fill(' '.join(map(str, x)), width,
798 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +0000799
Tim Petersde14a302002-04-01 05:04:46 +0000800# Map sys.platform to a string containing the basenames of tests
801# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +0000802#
803# Special cases:
804# test_pep277
805# The _ExpectedSkips constructor adds this to the set of expected
806# skips if not os.path.supports_unicode_filenames.
Tim Petersb4ee4eb2002-12-04 03:26:57 +0000807# test_socket_ssl
808# Controlled by test_socket_ssl.skip_expected. Requires the network
809# resource, and a socket module with ssl support.
Neal Norwitz55b61d22003-02-28 19:57:03 +0000810# test_timeout
811# Controlled by test_timeout.skip_expected. Requires the network
812# resource and a socket module.
Tim Petersde14a302002-04-01 05:04:46 +0000813
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000814_expectations = {
815 'win32':
816 """
Tim Petersc7c516a2003-09-20 22:06:13 +0000817 test__locale
Raymond Hettinger901dc982003-11-20 19:02:02 +0000818 test_applesingle
Tim Peters78e35f92002-11-22 20:00:34 +0000819 test_bsddb3
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000820 test_commands
821 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +0000822 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000823 test_dbm
824 test_dl
825 test_fcntl
826 test_fork1
827 test_gdbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000828 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +0000829 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000830 test_largefile
831 test_linuxaudiodev
832 test_mhlib
833 test_nis
834 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +0000835 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000836 test_poll
Tim Peters003eb302003-02-17 21:48:48 +0000837 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000838 test_pty
839 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +0000840 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000841 test_signal
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000842 test_sunaudiodev
Tim Peterscea2cc42004-08-04 02:32:03 +0000843 test_threadsignals
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000844 test_timing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000845 test_wait3
846 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000847 """,
848 'linux2':
849 """
Guido van Rossum944a6c32003-11-20 22:11:29 +0000850 test_applesingle
Guido van Rossumf66dacd2001-10-23 15:10:55 +0000851 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000852 test_dl
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000853 test_largefile
Guido van Rossum4507ec72003-02-14 19:29:22 +0000854 test_linuxaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000855 test_nis
856 test_ntpath
Guido van Rossum4507ec72003-02-14 19:29:22 +0000857 test_ossaudiodev
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000858 test_sqlite
859 test_startfile
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000860 test_sunaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000861 """,
Jack Jansen49a806e2001-08-28 14:49:00 +0000862 'mac':
Guido van Rossumaa782362001-09-02 03:58:41 +0000863 """
Jack Jansen67975142003-01-08 16:31:11 +0000864 test_atexit
Guido van Rossumaa782362001-09-02 03:58:41 +0000865 test_bsddb
Jack Jansen67975142003-01-08 16:31:11 +0000866 test_bsddb3
867 test_bz2
Guido van Rossumaa782362001-09-02 03:58:41 +0000868 test_commands
869 test_crypt
Jack Jansenb3be2162001-11-30 14:16:36 +0000870 test_curses
Guido van Rossumaa782362001-09-02 03:58:41 +0000871 test_dbm
872 test_dl
873 test_fcntl
874 test_fork1
Guido van Rossumaa782362001-09-02 03:58:41 +0000875 test_grp
Jack Jansenc4d6bdd2003-03-07 15:38:11 +0000876 test_ioctl
Guido van Rossumaa782362001-09-02 03:58:41 +0000877 test_largefile
878 test_linuxaudiodev
879 test_locale
880 test_mmap
881 test_nis
882 test_ntpath
883 test_openpty
Jack Jansen67975142003-01-08 16:31:11 +0000884 test_ossaudiodev
Guido van Rossumaa782362001-09-02 03:58:41 +0000885 test_poll
Jack Jansen67975142003-01-08 16:31:11 +0000886 test_popen
Jack Jansen5bb97e62003-02-21 22:33:55 +0000887 test_posix
Guido van Rossumaa782362001-09-02 03:58:41 +0000888 test_pty
889 test_pwd
Jack Jansen67975142003-01-08 16:31:11 +0000890 test_resource
Guido van Rossumaa782362001-09-02 03:58:41 +0000891 test_signal
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000892 test_sqlite
893 test_startfile
Guido van Rossumaa782362001-09-02 03:58:41 +0000894 test_sunaudiodev
895 test_sundry
Jack Jansenc4d6bdd2003-03-07 15:38:11 +0000896 test_tarfile
Guido van Rossumaa782362001-09-02 03:58:41 +0000897 test_timing
Guido van Rossumaa782362001-09-02 03:58:41 +0000898 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000899 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000900 """
Guido van Rossum944a6c32003-11-20 22:11:29 +0000901 test_applesingle
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000902 test_bsddb
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000903 test_dl
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000904 test_largefile
905 test_linuxaudiodev
906 test_minidom
907 test_nis
908 test_ntpath
909 test_openpty
910 test_pyexpat
911 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000912 test_startfile
913 test_sqlite
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000914 test_sunaudiodev
915 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000916 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000917 'openunix8':
918 """
Guido van Rossum944a6c32003-11-20 22:11:29 +0000919 test_applesingle
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000920 test_bsddb
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000921 test_dl
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000922 test_largefile
923 test_linuxaudiodev
924 test_minidom
925 test_nis
926 test_ntpath
927 test_openpty
928 test_pyexpat
929 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000930 test_sqlite
931 test_startfile
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000932 test_sunaudiodev
933 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000934 """,
935 'sco_sv3':
936 """
Guido van Rossum944a6c32003-11-20 22:11:29 +0000937 test_applesingle
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000938 test_asynchat
939 test_bsddb
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000940 test_dl
941 test_fork1
942 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000943 test_largefile
944 test_linuxaudiodev
945 test_locale
946 test_minidom
947 test_nis
948 test_ntpath
949 test_openpty
950 test_pyexpat
951 test_queue
952 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000953 test_sqlite
954 test_startfile
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000955 test_sunaudiodev
956 test_sundry
957 test_thread
958 test_threaded_import
959 test_threadedtempfile
960 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000961 """,
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000962 'riscos':
963 """
Guido van Rossum944a6c32003-11-20 22:11:29 +0000964 test_applesingle
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000965 test_asynchat
Martin v. Löwisa94568a2003-05-10 07:36:56 +0000966 test_atexit
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000967 test_bsddb
Martin v. Löwisa94568a2003-05-10 07:36:56 +0000968 test_bsddb3
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000969 test_commands
970 test_crypt
971 test_dbm
972 test_dl
973 test_fcntl
974 test_fork1
975 test_gdbm
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000976 test_grp
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000977 test_largefile
978 test_linuxaudiodev
979 test_locale
980 test_mmap
981 test_nis
982 test_ntpath
983 test_openpty
984 test_poll
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000985 test_pty
986 test_pwd
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000987 test_sqlite
988 test_startfile
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000989 test_sunaudiodev
990 test_sundry
991 test_thread
992 test_threaded_import
993 test_threadedtempfile
994 test_threading
995 test_timing
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000996 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +0000997 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +0000998 """
Jack Jansen398c2362001-12-02 21:41:36 +0000999 test_gdbm
Jack Jansen398c2362001-12-02 21:41:36 +00001000 test_largefile
1001 test_linuxaudiodev
Jack Jansenacda3392002-12-30 23:03:13 +00001002 test_locale
Jack Jansen398c2362001-12-02 21:41:36 +00001003 test_nis
Jack Jansenacda3392002-12-30 23:03:13 +00001004 test_ossaudiodev
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001005 test_startfile
Jack Jansen398c2362001-12-02 21:41:36 +00001006 test_sunaudiodev
Jack Jansen398c2362001-12-02 21:41:36 +00001007 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001008 'sunos5':
1009 """
Guido van Rossum944a6c32003-11-20 22:11:29 +00001010 test_applesingle
Guido van Rossum11c3f092002-07-17 15:08:24 +00001011 test_bsddb
Guido van Rossum11c3f092002-07-17 15:08:24 +00001012 test_curses
1013 test_dbm
Guido van Rossum11c3f092002-07-17 15:08:24 +00001014 test_gdbm
Guido van Rossum11c3f092002-07-17 15:08:24 +00001015 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001016 test_linuxaudiodev
Guido van Rossum11c3f092002-07-17 15:08:24 +00001017 test_openpty
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001018 test_sqlite
1019 test_startfile
Guido van Rossum11c3f092002-07-17 15:08:24 +00001020 test_zipfile
1021 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001022 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001023 'hp-ux11':
1024 """
Guido van Rossum944a6c32003-11-20 22:11:29 +00001025 test_applesingle
Skip Montanarob3230212002-03-15 02:54:03 +00001026 test_bsddb
Skip Montanarob3230212002-03-15 02:54:03 +00001027 test_curses
1028 test_dl
1029 test_gdbm
Skip Montanarob3230212002-03-15 02:54:03 +00001030 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001031 test_largefile
1032 test_linuxaudiodev
1033 test_locale
1034 test_minidom
1035 test_nis
1036 test_ntpath
1037 test_openpty
1038 test_pyexpat
1039 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001040 test_sqlite
1041 test_startfile
Skip Montanarob3230212002-03-15 02:54:03 +00001042 test_sunaudiodev
Skip Montanarob3230212002-03-15 02:54:03 +00001043 test_zipfile
1044 test_zlib
1045 """,
Martin v. Löwisf90ae202002-06-11 06:22:31 +00001046 'atheos':
Tim Petersc411dba2002-07-16 21:35:23 +00001047 """
Guido van Rossum944a6c32003-11-20 22:11:29 +00001048 test_applesingle
Tim Petersc411dba2002-07-16 21:35:23 +00001049 test_curses
1050 test_dl
Tim Petersc411dba2002-07-16 21:35:23 +00001051 test_gdbm
Tim Petersc411dba2002-07-16 21:35:23 +00001052 test_largefile
1053 test_linuxaudiodev
1054 test_locale
1055 test_mhlib
1056 test_mmap
Tim Petersc411dba2002-07-16 21:35:23 +00001057 test_nis
1058 test_poll
Tim Petersc411dba2002-07-16 21:35:23 +00001059 test_resource
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001060 test_sqlite
1061 test_startfile
Tim Petersc411dba2002-07-16 21:35:23 +00001062 test_sunaudiodev
Tim Petersc411dba2002-07-16 21:35:23 +00001063 """,
Jason Tishler25115942002-12-05 15:18:15 +00001064 'cygwin':
1065 """
Guido van Rossum944a6c32003-11-20 22:11:29 +00001066 test_applesingle
Tim Petersb0f89e02002-12-05 17:20:25 +00001067 test_bsddb3
Jason Tishler25115942002-12-05 15:18:15 +00001068 test_curses
1069 test_dbm
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001070 test_ioctl
Jason Tishler25115942002-12-05 15:18:15 +00001071 test_largefile
1072 test_linuxaudiodev
1073 test_locale
Jason Tishler25115942002-12-05 15:18:15 +00001074 test_nis
Jason Tishler5c4ded22003-02-05 16:46:01 +00001075 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001076 test_socketserver
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001077 test_sqlite
Jason Tishler25115942002-12-05 15:18:15 +00001078 test_sunaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001079 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001080 'os2emx':
1081 """
Guido van Rossum944a6c32003-11-20 22:11:29 +00001082 test_applesingle
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001083 test_audioop
1084 test_bsddb3
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001085 test_commands
1086 test_curses
1087 test_dl
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001088 test_largefile
1089 test_linuxaudiodev
1090 test_mhlib
1091 test_mmap
1092 test_nis
1093 test_openpty
1094 test_ossaudiodev
1095 test_pty
1096 test_resource
1097 test_signal
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001098 test_sqlite
1099 test_startfile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001100 test_sunaudiodev
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001101 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001102 'freebsd4':
1103 """
1104 test_aepack
Guido van Rossum944a6c32003-11-20 22:11:29 +00001105 test_applesingle
1106 test_bsddb
1107 test_bsddb3
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001108 test_gdbm
Guido van Rossum944a6c32003-11-20 22:11:29 +00001109 test_linuxaudiodev
1110 test_locale
1111 test_macfs
1112 test_macostools
1113 test_nis
Guido van Rossum944a6c32003-11-20 22:11:29 +00001114 test_ossaudiodev
1115 test_pep277
1116 test_plistlib
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001117 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001118 test_scriptpackages
1119 test_socket_ssl
1120 test_socketserver
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001121 test_sqlite
1122 test_startfile
Guido van Rossum944a6c32003-11-20 22:11:29 +00001123 test_sunaudiodev
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001124 test_tcl
Guido van Rossum944a6c32003-11-20 22:11:29 +00001125 test_timeout
1126 test_unicode_file
1127 test_urllibnet
1128 test_winreg
1129 test_winsound
Martin v. Löwis56f88112003-06-07 20:01:37 +00001130 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001131 'aix5':
1132 """
1133 test_aepack
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001134 test_applesingle
1135 test_bsddb
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001136 test_bsddb3
1137 test_bz2
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001138 test_dl
1139 test_gdbm
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001140 test_gzip
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001141 test_linuxaudiodev
1142 test_macfs
1143 test_macostools
1144 test_nis
1145 test_ossaudiodev
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001146 test_sqlite
1147 test_startfile
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001148 test_sunaudiodev
1149 test_tcl
1150 test_winreg
1151 test_winsound
1152 test_zipimport
1153 test_zlib
1154 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001155 'openbsd3':
1156 """
1157 test_aepack
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001158 test_applesingle
1159 test_bsddb
1160 test_bsddb3
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001161 test_ctypes
1162 test_dl
1163 test_gdbm
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001164 test_linuxaudiodev
1165 test_locale
1166 test_macfs
1167 test_macostools
1168 test_nis
1169 test_normalization
1170 test_ossaudiodev
1171 test_pep277
1172 test_plistlib
1173 test_scriptpackages
1174 test_tcl
1175 test_sqlite
1176 test_startfile
1177 test_sunaudiodev
1178 test_unicode_file
1179 test_winreg
1180 test_winsound
1181 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001182 'netbsd3':
1183 """
1184 test_aepack
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001185 test_applesingle
1186 test_bsddb
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001187 test_bsddb3
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001188 test_ctypes
1189 test_curses
1190 test_dl
1191 test_gdbm
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001192 test_linuxaudiodev
1193 test_locale
1194 test_macfs
1195 test_macostools
1196 test_nis
1197 test_ossaudiodev
1198 test_pep277
1199 test_sqlite
1200 test_startfile
1201 test_sunaudiodev
1202 test_tcl
1203 test_unicode_file
1204 test_winreg
1205 test_winsound
1206 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001207}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001208_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001209_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001210_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001211
Tim Petersb5b7b782001-08-12 01:20:39 +00001212class _ExpectedSkips:
1213 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001214 import os.path
Tim Petersb4ee4eb2002-12-04 03:26:57 +00001215 from test import test_socket_ssl
Neal Norwitz55b61d22003-02-28 19:57:03 +00001216 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001217
Tim Peters7c7efe92002-08-23 17:55:54 +00001218 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001219 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001220 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001221 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001222
Tim Peters2a182db2002-10-09 01:07:11 +00001223 if not os.path.supports_unicode_filenames:
1224 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001225
Tim Petersb4ee4eb2002-12-04 03:26:57 +00001226 if test_socket_ssl.skip_expected:
1227 self.expected.add('test_socket_ssl')
1228
Neal Norwitz55b61d22003-02-28 19:57:03 +00001229 if test_timeout.skip_expected:
1230 self.expected.add('test_timeout')
1231
Guido van Rossume2a383d2007-01-15 16:59:06 +00001232 if sys.maxint == 9223372036854775807:
Martin v. Löwisfba73692004-11-13 11:13:35 +00001233 self.expected.add('test_rgbimg')
1234 self.expected.add('test_imageop')
1235
Jack Jansen6afc5e02003-01-29 16:24:16 +00001236 if not sys.platform in ("mac", "darwin"):
Neal Norwitz7035c982003-03-29 22:01:17 +00001237 MAC_ONLY = ["test_macostools", "test_macfs", "test_aepack",
1238 "test_plistlib", "test_scriptpackages"]
1239 for skip in MAC_ONLY:
1240 self.expected.add(skip)
Tim Petersecd79eb2003-01-29 00:35:32 +00001241
1242 if sys.platform != "win32":
Neal Norwitz7035c982003-03-29 22:01:17 +00001243 WIN_ONLY = ["test_unicode_file", "test_winreg",
1244 "test_winsound"]
1245 for skip in WIN_ONLY:
1246 self.expected.add(skip)
Tim Petersf2715e02003-02-19 02:35:07 +00001247
Tim Peters7c7efe92002-08-23 17:55:54 +00001248 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001249
1250 def isvalid(self):
1251 "Return true iff _ExpectedSkips knows about the current platform."
1252 return self.valid
1253
1254 def getexpected(self):
1255 """Return set of test names we expect to skip on current platform.
1256
1257 self.isvalid() must be true.
1258 """
1259
1260 assert self.isvalid()
1261 return self.expected
1262
Guido van Rossum152494a1996-12-20 03:12:20 +00001263if __name__ == '__main__':
Barry Warsaw408b6d32002-07-30 23:27:12 +00001264 # Remove regrtest.py's own directory from the module search path. This
1265 # prevents relative imports from working, and relative imports will screw
1266 # up the testing framework. E.g. if both test.test_support and
1267 # test_support are imported, they will not contain the same globals, and
1268 # much of the testing framework relies on the globals in the
1269 # test.test_support module.
1270 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1271 i = pathlen = len(sys.path)
1272 while i >= 0:
1273 i -= 1
1274 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1275 del sys.path[i]
1276 if len(sys.path) == pathlen:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001277 print('Could not find %r in sys.path to remove it' % mydir)
Barry Warsaw08fca522001-08-20 22:33:46 +00001278 main()