blob: 426a92cc41db1e6388ea22df148706722a9b1f05 [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',
477 'test_operations',
478 'test_builtin',
479 'test_exceptions',
480 'test_types',
481 ]
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:
Raymond Hettinger74e67662003-05-17 20:44:12 +0000540 cfp = cStringIO.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 Rossumbe19ed72007-02-09 05:37:30 +0000689 print(test, 'leaked', deltas, 'references', file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000690 refrep = open(fname, "a")
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000691 print(test, 'leaked', deltas, 'references', file=refrep)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000692 refrep.close()
693
694def dash_R_cleanup(fs, ps, pic):
695 import gc, copy_reg
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000696 import _strptime, linecache, dircache
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000697 import urlparse, urllib, urllib2, mimetypes, doctest
698 import struct, filecmp
699 from distutils.dir_util import _path_created
700
701 # Restore some original values.
702 warnings.filters[:] = fs
703 copy_reg.dispatch_table.clear()
704 copy_reg.dispatch_table.update(ps)
705 sys.path_importer_cache.clear()
706 sys.path_importer_cache.update(pic)
707
708 # Clear assorted module caches.
709 _path_created.clear()
710 re.purge()
711 _strptime._regex_cache.clear()
712 urlparse.clear_cache()
713 urllib.urlcleanup()
714 urllib2.install_opener(None)
715 dircache.reset()
716 linecache.clearcache()
717 mimetypes._default_mime_types()
718 struct._cache.clear()
719 filecmp._cache.clear()
720 doctest.master = None
721
722 # Collect cyclic trash.
723 gc.collect()
724
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000725def reportdiff(expected, output):
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000726 import difflib
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000727 print("*" * 70)
Tim Petersc377b162001-09-22 05:31:03 +0000728 a = expected.splitlines(1)
729 b = output.splitlines(1)
Guido van Rossumcf691932001-09-21 21:06:22 +0000730 sm = difflib.SequenceMatcher(a=a, b=b)
731 tuples = sm.get_opcodes()
Tim Petersc377b162001-09-22 05:31:03 +0000732
Guido van Rossumcf691932001-09-21 21:06:22 +0000733 def pair(x0, x1):
Tim Petersc377b162001-09-22 05:31:03 +0000734 # x0:x1 are 0-based slice indices; convert to 1-based line indices.
Guido van Rossumcf691932001-09-21 21:06:22 +0000735 x0 += 1
736 if x0 >= x1:
Tim Petersc377b162001-09-22 05:31:03 +0000737 return "line " + str(x0)
Guido van Rossumcf691932001-09-21 21:06:22 +0000738 else:
Tim Petersc377b162001-09-22 05:31:03 +0000739 return "lines %d-%d" % (x0, x1)
740
Guido van Rossumcf691932001-09-21 21:06:22 +0000741 for op, a0, a1, b0, b1 in tuples:
742 if op == 'equal':
743 pass
Tim Petersc377b162001-09-22 05:31:03 +0000744
Guido van Rossumcf691932001-09-21 21:06:22 +0000745 elif op == 'delete':
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000746 print("***", pair(a0, a1), "of expected output missing:")
Guido van Rossumcf691932001-09-21 21:06:22 +0000747 for line in a[a0:a1]:
Guido van Rossum08c47ba2007-02-09 20:50:08 +0000748 print("-", line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000749
Guido van Rossumcf691932001-09-21 21:06:22 +0000750 elif op == 'replace':
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000751 print("*** mismatch between", pair(a0, a1), "of expected", \
752 "output and", pair(b0, b1), "of actual output:")
Tim Petersc377b162001-09-22 05:31:03 +0000753 for line in difflib.ndiff(a[a0:a1], b[b0:b1]):
Guido van Rossum08c47ba2007-02-09 20:50:08 +0000754 print(line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000755
Guido van Rossumcf691932001-09-21 21:06:22 +0000756 elif op == 'insert':
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000757 print("***", pair(b0, b1), "of actual output doesn't appear", \
758 "in expected output after line", str(a1)+":")
Guido van Rossumcf691932001-09-21 21:06:22 +0000759 for line in b[b0:b1]:
Guido van Rossum08c47ba2007-02-09 20:50:08 +0000760 print("+", line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000761
Guido van Rossumcf691932001-09-21 21:06:22 +0000762 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000763 print("get_opcodes() returned bad tuple?!?!", (op, a0, a1, b0, b1))
Tim Petersc377b162001-09-22 05:31:03 +0000764
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000765 print("*" * 70)
Guido van Rossum152494a1996-12-20 03:12:20 +0000766
767def findtestdir():
768 if __name__ == '__main__':
Guido van Rossum41360a41998-03-26 19:42:58 +0000769 file = sys.argv[0]
Guido van Rossum152494a1996-12-20 03:12:20 +0000770 else:
Guido van Rossum41360a41998-03-26 19:42:58 +0000771 file = __file__
Guido van Rossum152494a1996-12-20 03:12:20 +0000772 testdir = os.path.dirname(file) or os.curdir
773 return testdir
774
Tim Petersc5000df2002-06-02 21:42:01 +0000775def removepy(name):
776 if name.endswith(os.extsep + "py"):
777 name = name[:-3]
778 return name
779
Guido van Rossum152494a1996-12-20 03:12:20 +0000780def count(n, word):
781 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +0000782 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +0000783 else:
Guido van Rossum41360a41998-03-26 19:42:58 +0000784 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +0000785
Tim Petersa45da922001-08-12 03:45:50 +0000786def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +0000787 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +0000788
789 Optional arg width (default 70) is the maximum line length.
790 Optional arg indent (default 4) is the number of blanks with which to
791 begin each line.
792 """
793
Tim Petersba78bc42002-07-04 19:45:06 +0000794 from textwrap import fill
795 blanks = ' ' * indent
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000796 print(fill(' '.join(map(str, x)), width,
797 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +0000798
Tim Petersde14a302002-04-01 05:04:46 +0000799# Map sys.platform to a string containing the basenames of tests
800# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +0000801#
802# Special cases:
803# test_pep277
804# The _ExpectedSkips constructor adds this to the set of expected
805# skips if not os.path.supports_unicode_filenames.
Tim Petersb4ee4eb2002-12-04 03:26:57 +0000806# test_socket_ssl
807# Controlled by test_socket_ssl.skip_expected. Requires the network
808# resource, and a socket module with ssl support.
Neal Norwitz55b61d22003-02-28 19:57:03 +0000809# test_timeout
810# Controlled by test_timeout.skip_expected. Requires the network
811# resource and a socket module.
Tim Petersde14a302002-04-01 05:04:46 +0000812
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000813_expectations = {
814 'win32':
815 """
Tim Petersc7c516a2003-09-20 22:06:13 +0000816 test__locale
Raymond Hettinger901dc982003-11-20 19:02:02 +0000817 test_applesingle
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000818 test_al
Skip Montanaro823ba282003-05-06 20:36:24 +0000819 test_bsddb185
Tim Peters78e35f92002-11-22 20:00:34 +0000820 test_bsddb3
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000821 test_cd
822 test_cl
823 test_commands
824 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +0000825 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000826 test_dbm
827 test_dl
828 test_fcntl
829 test_fork1
830 test_gdbm
831 test_gl
832 test_grp
833 test_imgfile
Tim Petersfd8e6e52003-03-04 00:26:38 +0000834 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000835 test_largefile
836 test_linuxaudiodev
837 test_mhlib
838 test_nis
839 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +0000840 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000841 test_poll
Tim Peters003eb302003-02-17 21:48:48 +0000842 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000843 test_pty
844 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +0000845 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000846 test_signal
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000847 test_sunaudiodev
Tim Peterscea2cc42004-08-04 02:32:03 +0000848 test_threadsignals
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000849 test_timing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000850 test_wait3
851 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000852 """,
853 'linux2':
854 """
855 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +0000856 test_applesingle
Skip Montanaro823ba282003-05-06 20:36:24 +0000857 test_bsddb185
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000858 test_cd
859 test_cl
Guido van Rossumf66dacd2001-10-23 15:10:55 +0000860 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000861 test_dl
862 test_gl
863 test_imgfile
864 test_largefile
Guido van Rossum4507ec72003-02-14 19:29:22 +0000865 test_linuxaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000866 test_nis
867 test_ntpath
Guido van Rossum4507ec72003-02-14 19:29:22 +0000868 test_ossaudiodev
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000869 test_sqlite
870 test_startfile
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000871 test_sunaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000872 """,
Jack Jansen49a806e2001-08-28 14:49:00 +0000873 'mac':
Guido van Rossumaa782362001-09-02 03:58:41 +0000874 """
875 test_al
Jack Jansen67975142003-01-08 16:31:11 +0000876 test_atexit
Guido van Rossumaa782362001-09-02 03:58:41 +0000877 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +0000878 test_bsddb185
Jack Jansen67975142003-01-08 16:31:11 +0000879 test_bsddb3
880 test_bz2
Guido van Rossumaa782362001-09-02 03:58:41 +0000881 test_cd
882 test_cl
883 test_commands
884 test_crypt
Jack Jansenb3be2162001-11-30 14:16:36 +0000885 test_curses
Guido van Rossumaa782362001-09-02 03:58:41 +0000886 test_dbm
887 test_dl
888 test_fcntl
889 test_fork1
890 test_gl
891 test_grp
Jack Jansenc4d6bdd2003-03-07 15:38:11 +0000892 test_ioctl
Guido van Rossumaa782362001-09-02 03:58:41 +0000893 test_imgfile
894 test_largefile
895 test_linuxaudiodev
896 test_locale
897 test_mmap
898 test_nis
899 test_ntpath
900 test_openpty
Jack Jansen67975142003-01-08 16:31:11 +0000901 test_ossaudiodev
Guido van Rossumaa782362001-09-02 03:58:41 +0000902 test_poll
Jack Jansen67975142003-01-08 16:31:11 +0000903 test_popen
Guido van Rossumaa782362001-09-02 03:58:41 +0000904 test_popen2
Jack Jansen5bb97e62003-02-21 22:33:55 +0000905 test_posix
Guido van Rossumaa782362001-09-02 03:58:41 +0000906 test_pty
907 test_pwd
Jack Jansen67975142003-01-08 16:31:11 +0000908 test_resource
Guido van Rossumaa782362001-09-02 03:58:41 +0000909 test_signal
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000910 test_sqlite
911 test_startfile
Guido van Rossumaa782362001-09-02 03:58:41 +0000912 test_sunaudiodev
913 test_sundry
Jack Jansenc4d6bdd2003-03-07 15:38:11 +0000914 test_tarfile
Guido van Rossumaa782362001-09-02 03:58:41 +0000915 test_timing
Guido van Rossumaa782362001-09-02 03:58:41 +0000916 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000917 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000918 """
919 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +0000920 test_applesingle
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000921 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +0000922 test_bsddb185
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000923 test_cd
924 test_cl
925 test_dl
926 test_gl
927 test_imgfile
928 test_largefile
929 test_linuxaudiodev
930 test_minidom
931 test_nis
932 test_ntpath
933 test_openpty
934 test_pyexpat
935 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000936 test_startfile
937 test_sqlite
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000938 test_sunaudiodev
939 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000940 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000941 'openunix8':
942 """
943 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +0000944 test_applesingle
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000945 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +0000946 test_bsddb185
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000947 test_cd
948 test_cl
949 test_dl
950 test_gl
951 test_imgfile
952 test_largefile
953 test_linuxaudiodev
954 test_minidom
955 test_nis
956 test_ntpath
957 test_openpty
958 test_pyexpat
959 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000960 test_sqlite
961 test_startfile
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000962 test_sunaudiodev
963 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000964 """,
965 'sco_sv3':
966 """
967 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +0000968 test_applesingle
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000969 test_asynchat
970 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +0000971 test_bsddb185
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000972 test_cd
973 test_cl
974 test_dl
975 test_fork1
976 test_gettext
977 test_gl
978 test_imgfile
979 test_largefile
980 test_linuxaudiodev
981 test_locale
982 test_minidom
983 test_nis
984 test_ntpath
985 test_openpty
986 test_pyexpat
987 test_queue
988 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000989 test_sqlite
990 test_startfile
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000991 test_sunaudiodev
992 test_sundry
993 test_thread
994 test_threaded_import
995 test_threadedtempfile
996 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000997 """,
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000998 'riscos':
999 """
1000 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +00001001 test_applesingle
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001002 test_asynchat
Martin v. Löwisa94568a2003-05-10 07:36:56 +00001003 test_atexit
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001004 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +00001005 test_bsddb185
Martin v. Löwisa94568a2003-05-10 07:36:56 +00001006 test_bsddb3
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001007 test_cd
1008 test_cl
1009 test_commands
1010 test_crypt
1011 test_dbm
1012 test_dl
1013 test_fcntl
1014 test_fork1
1015 test_gdbm
1016 test_gl
1017 test_grp
1018 test_imgfile
1019 test_largefile
1020 test_linuxaudiodev
1021 test_locale
1022 test_mmap
1023 test_nis
1024 test_ntpath
1025 test_openpty
1026 test_poll
1027 test_popen2
1028 test_pty
1029 test_pwd
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001030 test_strop
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001031 test_sqlite
1032 test_startfile
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001033 test_sunaudiodev
1034 test_sundry
1035 test_thread
1036 test_threaded_import
1037 test_threadedtempfile
1038 test_threading
1039 test_timing
Guido van Rossume2ae77b2001-10-24 20:42:55 +00001040 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001041 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001042 """
1043 test_al
1044 test_cd
1045 test_cl
Jack Jansen398c2362001-12-02 21:41:36 +00001046 test_gdbm
1047 test_gl
1048 test_imgfile
1049 test_largefile
1050 test_linuxaudiodev
Jack Jansenacda3392002-12-30 23:03:13 +00001051 test_locale
Jack Jansen398c2362001-12-02 21:41:36 +00001052 test_nis
Jack Jansenacda3392002-12-30 23:03:13 +00001053 test_ossaudiodev
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001054 test_startfile
Jack Jansen398c2362001-12-02 21:41:36 +00001055 test_sunaudiodev
Jack Jansen398c2362001-12-02 21:41:36 +00001056 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001057 'sunos5':
1058 """
1059 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +00001060 test_applesingle
Guido van Rossum11c3f092002-07-17 15:08:24 +00001061 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +00001062 test_bsddb185
Guido van Rossum11c3f092002-07-17 15:08:24 +00001063 test_cd
1064 test_cl
1065 test_curses
1066 test_dbm
Guido van Rossum11c3f092002-07-17 15:08:24 +00001067 test_gdbm
1068 test_gl
1069 test_gzip
1070 test_imgfile
1071 test_linuxaudiodev
Guido van Rossum11c3f092002-07-17 15:08:24 +00001072 test_openpty
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001073 test_sqlite
1074 test_startfile
Guido van Rossum11c3f092002-07-17 15:08:24 +00001075 test_zipfile
1076 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001077 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001078 'hp-ux11':
1079 """
1080 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +00001081 test_applesingle
Skip Montanarob3230212002-03-15 02:54:03 +00001082 test_bsddb
Skip Montanaro823ba282003-05-06 20:36:24 +00001083 test_bsddb185
Skip Montanarob3230212002-03-15 02:54:03 +00001084 test_cd
1085 test_cl
1086 test_curses
1087 test_dl
1088 test_gdbm
1089 test_gl
1090 test_gzip
1091 test_imgfile
1092 test_largefile
1093 test_linuxaudiodev
1094 test_locale
1095 test_minidom
1096 test_nis
1097 test_ntpath
1098 test_openpty
1099 test_pyexpat
1100 test_sax
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001101 test_sqlite
1102 test_startfile
Skip Montanarob3230212002-03-15 02:54:03 +00001103 test_sunaudiodev
Skip Montanarob3230212002-03-15 02:54:03 +00001104 test_zipfile
1105 test_zlib
1106 """,
Martin v. Löwisf90ae202002-06-11 06:22:31 +00001107 'atheos':
Tim Petersc411dba2002-07-16 21:35:23 +00001108 """
1109 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +00001110 test_applesingle
Skip Montanaro823ba282003-05-06 20:36:24 +00001111 test_bsddb185
Tim Petersc411dba2002-07-16 21:35:23 +00001112 test_cd
1113 test_cl
1114 test_curses
1115 test_dl
Tim Petersc411dba2002-07-16 21:35:23 +00001116 test_gdbm
1117 test_gl
1118 test_imgfile
1119 test_largefile
1120 test_linuxaudiodev
1121 test_locale
1122 test_mhlib
1123 test_mmap
Tim Petersc411dba2002-07-16 21:35:23 +00001124 test_nis
1125 test_poll
1126 test_popen2
1127 test_resource
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001128 test_sqlite
1129 test_startfile
Tim Petersc411dba2002-07-16 21:35:23 +00001130 test_sunaudiodev
Tim Petersc411dba2002-07-16 21:35:23 +00001131 """,
Jason Tishler25115942002-12-05 15:18:15 +00001132 'cygwin':
1133 """
1134 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +00001135 test_applesingle
Skip Montanaro823ba282003-05-06 20:36:24 +00001136 test_bsddb185
Tim Petersb0f89e02002-12-05 17:20:25 +00001137 test_bsddb3
Jason Tishler25115942002-12-05 15:18:15 +00001138 test_cd
1139 test_cl
1140 test_curses
1141 test_dbm
Jason Tishler25115942002-12-05 15:18:15 +00001142 test_gl
1143 test_imgfile
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001144 test_ioctl
Jason Tishler25115942002-12-05 15:18:15 +00001145 test_largefile
1146 test_linuxaudiodev
1147 test_locale
Jason Tishler25115942002-12-05 15:18:15 +00001148 test_nis
Jason Tishler5c4ded22003-02-05 16:46:01 +00001149 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001150 test_socketserver
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001151 test_sqlite
Jason Tishler25115942002-12-05 15:18:15 +00001152 test_sunaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001153 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001154 'os2emx':
1155 """
1156 test_al
Guido van Rossum944a6c32003-11-20 22:11:29 +00001157 test_applesingle
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001158 test_audioop
Skip Montanaro823ba282003-05-06 20:36:24 +00001159 test_bsddb185
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001160 test_bsddb3
1161 test_cd
1162 test_cl
1163 test_commands
1164 test_curses
1165 test_dl
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001166 test_gl
1167 test_imgfile
1168 test_largefile
1169 test_linuxaudiodev
1170 test_mhlib
1171 test_mmap
1172 test_nis
1173 test_openpty
1174 test_ossaudiodev
1175 test_pty
1176 test_resource
1177 test_signal
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001178 test_sqlite
1179 test_startfile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001180 test_sunaudiodev
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001181 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001182 'freebsd4':
1183 """
1184 test_aepack
1185 test_al
1186 test_applesingle
1187 test_bsddb
1188 test_bsddb3
1189 test_cd
1190 test_cl
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001191 test_gdbm
Guido van Rossum944a6c32003-11-20 22:11:29 +00001192 test_gl
1193 test_imgfile
1194 test_linuxaudiodev
1195 test_locale
1196 test_macfs
1197 test_macostools
1198 test_nis
Guido van Rossum944a6c32003-11-20 22:11:29 +00001199 test_ossaudiodev
1200 test_pep277
1201 test_plistlib
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001202 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001203 test_scriptpackages
1204 test_socket_ssl
1205 test_socketserver
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001206 test_sqlite
1207 test_startfile
Guido van Rossum944a6c32003-11-20 22:11:29 +00001208 test_sunaudiodev
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001209 test_tcl
Guido van Rossum944a6c32003-11-20 22:11:29 +00001210 test_timeout
1211 test_unicode_file
1212 test_urllibnet
1213 test_winreg
1214 test_winsound
Martin v. Löwis56f88112003-06-07 20:01:37 +00001215 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001216 'aix5':
1217 """
1218 test_aepack
1219 test_al
1220 test_applesingle
1221 test_bsddb
1222 test_bsddb185
1223 test_bsddb3
1224 test_bz2
1225 test_cd
1226 test_cl
1227 test_dl
1228 test_gdbm
1229 test_gl
1230 test_gzip
1231 test_imgfile
1232 test_linuxaudiodev
1233 test_macfs
1234 test_macostools
1235 test_nis
1236 test_ossaudiodev
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001237 test_sqlite
1238 test_startfile
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001239 test_sunaudiodev
1240 test_tcl
1241 test_winreg
1242 test_winsound
1243 test_zipimport
1244 test_zlib
1245 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001246 'openbsd3':
1247 """
1248 test_aepack
1249 test_al
1250 test_applesingle
1251 test_bsddb
1252 test_bsddb3
1253 test_cd
1254 test_cl
1255 test_ctypes
1256 test_dl
1257 test_gdbm
1258 test_gl
1259 test_imgfile
1260 test_linuxaudiodev
1261 test_locale
1262 test_macfs
1263 test_macostools
1264 test_nis
1265 test_normalization
1266 test_ossaudiodev
1267 test_pep277
1268 test_plistlib
1269 test_scriptpackages
1270 test_tcl
1271 test_sqlite
1272 test_startfile
1273 test_sunaudiodev
1274 test_unicode_file
1275 test_winreg
1276 test_winsound
1277 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001278 'netbsd3':
1279 """
1280 test_aepack
1281 test_al
1282 test_applesingle
1283 test_bsddb
1284 test_bsddb185
1285 test_bsddb3
1286 test_cd
1287 test_cl
1288 test_ctypes
1289 test_curses
1290 test_dl
1291 test_gdbm
1292 test_gl
1293 test_imgfile
1294 test_linuxaudiodev
1295 test_locale
1296 test_macfs
1297 test_macostools
1298 test_nis
1299 test_ossaudiodev
1300 test_pep277
1301 test_sqlite
1302 test_startfile
1303 test_sunaudiodev
1304 test_tcl
1305 test_unicode_file
1306 test_winreg
1307 test_winsound
1308 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001309}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001310_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001311_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001312_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001313
Tim Petersb5b7b782001-08-12 01:20:39 +00001314class _ExpectedSkips:
1315 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001316 import os.path
Tim Petersb4ee4eb2002-12-04 03:26:57 +00001317 from test import test_socket_ssl
Neal Norwitz55b61d22003-02-28 19:57:03 +00001318 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001319
Tim Peters7c7efe92002-08-23 17:55:54 +00001320 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001321 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001322 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001323 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001324
Tim Peters2a182db2002-10-09 01:07:11 +00001325 if not os.path.supports_unicode_filenames:
1326 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001327
Tim Petersb4ee4eb2002-12-04 03:26:57 +00001328 if test_socket_ssl.skip_expected:
1329 self.expected.add('test_socket_ssl')
1330
Neal Norwitz55b61d22003-02-28 19:57:03 +00001331 if test_timeout.skip_expected:
1332 self.expected.add('test_timeout')
1333
Guido van Rossume2a383d2007-01-15 16:59:06 +00001334 if sys.maxint == 9223372036854775807:
Martin v. Löwisfba73692004-11-13 11:13:35 +00001335 self.expected.add('test_rgbimg')
1336 self.expected.add('test_imageop')
1337
Jack Jansen6afc5e02003-01-29 16:24:16 +00001338 if not sys.platform in ("mac", "darwin"):
Neal Norwitz7035c982003-03-29 22:01:17 +00001339 MAC_ONLY = ["test_macostools", "test_macfs", "test_aepack",
1340 "test_plistlib", "test_scriptpackages"]
1341 for skip in MAC_ONLY:
1342 self.expected.add(skip)
Tim Petersecd79eb2003-01-29 00:35:32 +00001343
1344 if sys.platform != "win32":
Neal Norwitz7035c982003-03-29 22:01:17 +00001345 WIN_ONLY = ["test_unicode_file", "test_winreg",
1346 "test_winsound"]
1347 for skip in WIN_ONLY:
1348 self.expected.add(skip)
Tim Petersf2715e02003-02-19 02:35:07 +00001349
Tim Peters7c7efe92002-08-23 17:55:54 +00001350 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001351
1352 def isvalid(self):
1353 "Return true iff _ExpectedSkips knows about the current platform."
1354 return self.valid
1355
1356 def getexpected(self):
1357 """Return set of test names we expect to skip on current platform.
1358
1359 self.isvalid() must be true.
1360 """
1361
1362 assert self.isvalid()
1363 return self.expected
1364
Guido van Rossum152494a1996-12-20 03:12:20 +00001365if __name__ == '__main__':
Barry Warsaw408b6d32002-07-30 23:27:12 +00001366 # Remove regrtest.py's own directory from the module search path. This
1367 # prevents relative imports from working, and relative imports will screw
1368 # up the testing framework. E.g. if both test.test_support and
1369 # test_support are imported, they will not contain the same globals, and
1370 # much of the testing framework relies on the globals in the
1371 # test.test_support module.
1372 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1373 i = pathlen = len(sys.path)
1374 while i >= 0:
1375 i -= 1
1376 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1377 del sys.path[i]
1378 if len(sys.path) == pathlen:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001379 print('Could not find %r in sys.path to remove it' % mydir)
Barry Warsaw08fca522001-08-20 22:33:46 +00001380 main()