blob: edfaa9a01cfe23c3e2a44e3f9e94387a4fb778d4 [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
Neal Norwitz94fa2ee2008-03-31 02:55:15 +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
Michael W. Hudson61147f62004-08-03 11:33:28 +000015-x: exclude -- arguments are tests to *exclude*
16-s: single -- run only a single test (see below)
Christian Heimesb186d002008-03-18 15:15:01 +000017-S: slow -- print the slowest 10 tests
Michael W. Hudson61147f62004-08-03 11:33:28 +000018-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
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000030-n: nowindows -- suppress error message boxes on Windows
Guido van Rossum152494a1996-12-20 03:12:20 +000031
32If non-option arguments are present, they are names for tests to run,
33unless -x is given, in which case they are names for tests not to run.
34If no test names are given, all tests are run.
Guido van Rossumf58ed251997-03-07 21:04:33 +000035
Guido van Rossuma4122201997-08-18 20:08:24 +000036-v is incompatible with -g and does not compare test output files.
Barry Warsawe11e3de1999-01-28 19:51:51 +000037
Collin Winterfd12f492009-03-29 04:05:05 +000038-r randomizes test execution order. You can use --randseed=int to provide a
39int seed value for the randomizer; this is useful for reproducing troublesome
40test orders.
41
Barry Warsaw3b6d0252004-02-07 22:43:03 +000042-T turns on code coverage tracing with the trace module.
43
Walter Dörwaldaee4da62004-11-12 18:51:27 +000044-D specifies the directory where coverage files are put.
45
46-N Put coverage files alongside modules.
47
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000048-s means to run only a single test and exit. This is useful when
49doing memory analysis on the Python interpreter (which tend to consume
50too many resources to run the full regression test non-stop). The
51file /tmp/pynexttest is read to find the next test to run. If this
52file is missing, the first test_*.py file in testdir or on the command
53line is used. (actually tempfile.gettempdir() is used instead of
54/tmp).
Barry Warsawe11e3de1999-01-28 19:51:51 +000055
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000056-S is used to continue running tests after an aborted run. It will
57maintain the order a standard run (ie, this assumes -r is not used).
58This is useful after the tests have prematurely stopped for some external
59reason and you want to start running from where you left off rather
60than starting from the beginning.
61
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000062-f reads the names of tests from the file given as f's argument, one
63or more test names per line. Whitespace is ignored. Blank lines and
64lines beginning with '#' are ignored. This is especially useful for
65whittling down failures involving interactions among tests.
Tim Petersc5000df2002-06-02 21:42:01 +000066
Skip Montanaro0179a182004-06-06 15:53:18 +000067-L causes the leaks(1) command to be run just before exit if it exists.
68leaks(1) is available on Mac OS X and presumably on some other
69FreeBSD-derived systems.
70
Michael W. Hudson61147f62004-08-03 11:33:28 +000071-R runs each test several times and examines sys.gettotalrefcount() to
72see if the test appears to be leaking references. The argument should
73be of the form stab:run:fname where 'stab' is the number of times the
74test is run to let gettotalrefcount settle down, 'run' is the number
75of times further it is run and 'fname' is the name of the file the
76reports are written to. These parameters all have defaults (5, 4 and
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000077"reflog.txt" respectively), and the minimal invocation is '-R :'.
Michael W. Hudson61147f62004-08-03 11:33:28 +000078
Thomas Wouters477c8d52006-05-27 19:21:47 +000079-M runs tests that require an exorbitant amount of memory. These tests
80typically try to ascertain containers keep working when containing more than
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000812 billion objects, which only works on 64-bit systems. There are also some
82tests that try to exhaust the address space of the process, which only makes
83sense on 32-bit systems with at least 2Gb of memory. The passed-in memlimit,
Thomas Wouters477c8d52006-05-27 19:21:47 +000084which is a string in the form of '2.5Gb', determines howmuch memory the
85tests will limit themselves to (but they may go slightly over.) The number
86shouldn't be more memory than the machine has (including swap memory). You
87should also keep in mind that swap memory is generally much, much slower
88than RAM, and setting memlimit to all available RAM or higher will heavily
89tax the machine. On the other hand, it is no use running these tests with a
90limit of less than 2.5Gb, and many require more than 20Gb. Tests that expect
91to use more than memlimit memory will be skipped. The big-memory tests
92generally run very, very long.
93
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000094-u is used to specify which special resource intensive tests to run,
95such as those requiring large file support or network connectivity.
96The argument is a comma-separated list of words indicating the
97resources to test. Currently only the following are defined:
Barry Warsaw08fca522001-08-20 22:33:46 +000098
Fred Drake3a15dac2002-04-11 16:39:16 +000099 all - Enable all special resources.
100
Guido van Rossum315aa362003-03-11 14:46:48 +0000101 audio - Tests that use the audio device. (There are known
102 cases of broken audio drivers that can crash Python or
103 even the Linux kernel.)
104
Andrew M. Kuchling2158df02001-10-22 15:26:09 +0000105 curses - Tests that use curses and will modify the terminal's
106 state and output modes.
Tim Peters1633a2e2001-10-30 05:56:40 +0000107
Georg Brandl86b2fb92008-07-16 03:43:04 +0000108 lib2to3 - Run the tests for 2to3 (They take a while.)
109
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000110 largefile - It is okay to run some test that may create huge
111 files. These tests can take a long time and may
112 consume >2GB of disk space temporarily.
Barry Warsaw08fca522001-08-20 22:33:46 +0000113
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000114 network - It is okay to run tests that use external network
115 resource, e.g. testing SSL support for sockets.
Martin v. Löwis1c6b1a22002-11-19 17:47:07 +0000116
117 bsddb - It is okay to run the bsddb testsuite, which takes
118 a long time to complete.
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000119
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000120 decimal - Test the decimal module against a large suite that
121 verifies compliance with standards.
122
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000123 compiler - Allow test_tokenize to verify round-trip lexing on
124 every file in the test library.
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000125
Tim Peterseba28be2005-03-28 01:08:02 +0000126 subprocess Run all tests for the subprocess module.
Peter Astrandf7f1bb72005-03-03 20:47:37 +0000127
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000128 urlfetch - It is okay to download files required on testing.
129
Guilherme Polo9de29af2009-01-28 20:40:48 +0000130 gui - Run tests that require a running GUI.
131
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000132To enable all resources except one, use '-uall,-<resource>'. For
133example, to run all the tests except for the bsddb tests, give the
134option '-uall,-bsddb'.
Guido van Rossum152494a1996-12-20 03:12:20 +0000135"""
136
Guido van Rossum152494a1996-12-20 03:12:20 +0000137import getopt
Christian Heimesb186d002008-03-18 15:15:01 +0000138import os
Skip Montanaroab1c7912000-06-30 16:39:27 +0000139import random
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000140import re
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000141import io
Christian Heimesb186d002008-03-18 15:15:01 +0000142import sys
143import time
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000144import traceback
Christian Heimesb186d002008-03-18 15:15:01 +0000145import warnings
Benjamin Petersone549ead2009-03-28 21:42:05 +0000146import unittest
Christian Heimesbe5b30b2008-03-03 19:18:51 +0000147from inspect import isabstract
Guido van Rossumdc15c272002-08-12 21:55:51 +0000148
149# I see no other way to suppress these warnings;
150# putting them in test_grammar.py has no effect:
Guido van Rossum88b1def2002-08-14 17:54:48 +0000151warnings.filterwarnings("ignore", "hex/oct constants", FutureWarning,
Guido van Rossumdc15c272002-08-12 21:55:51 +0000152 ".*test.test_grammar$")
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000153if sys.maxsize > 0x7fffffff:
Guido van Rossumc34c4fc2002-09-19 00:42:16 +0000154 # Also suppress them in <string>, because for 64-bit platforms,
155 # that's where test_grammar.py hides them.
156 warnings.filterwarnings("ignore", "hex/oct constants", FutureWarning,
157 "<string>")
Guido van Rossum152494a1996-12-20 03:12:20 +0000158
Thomas Wouters477c8d52006-05-27 19:21:47 +0000159# Ignore ImportWarnings that only occur in the source tree,
160# (because of modules with the same name as source-directories in Modules/)
161for mod in ("ctypes", "gzip", "zipfile", "tarfile", "encodings.zlib_codec",
162 "test.test_zipimport", "test.test_zlib", "test.test_zipfile",
163 "test.test_codecs", "test.string_tests"):
164 warnings.filterwarnings(module=".*%s$" % (mod,),
165 action="ignore", category=ImportWarning)
166
Guido van Rossumbb484652002-12-02 09:56:21 +0000167# MacOSX (a.k.a. Darwin) has a default stack size that is too small
168# for deeply recursive regular expressions. We see this as crashes in
169# the Python test suite when running test_re.py and test_sre.py. The
170# fix is to set the stack limit to 2048.
171# This approach may also be useful for other Unixy platforms that
172# suffer from small default stack limits.
173if sys.platform == 'darwin':
174 try:
175 import resource
176 except ImportError:
177 pass
178 else:
179 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
180 newsoft = min(hard, max(soft, 1024*2048))
181 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
182
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000183from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000184
Benjamin Peterson5a722402008-07-18 14:35:44 +0000185RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network', 'bsddb',
Guilherme Polo9de29af2009-01-28 20:40:48 +0000186 'decimal', 'compiler', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000187
188
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000189def usage(msg):
190 print(msg, file=sys.stderr)
191 print("Use --help for usage", file=sys.stderr)
192 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000193
194
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000195def main(tests=None, testdir=None, verbose=0, quiet=False, generate=False,
196 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000197 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000198 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
199 random_seed=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000200 """Execute a test suite.
201
Thomas Wouters7e474022000-07-16 12:04:32 +0000202 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000203 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000204
205 tests -- a list of strings containing test names (optional)
206 testdir -- the directory in which to look for tests (optional)
207
208 Users other than the Python test suite will certainly want to
209 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000210 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000211
212 If the tests argument is omitted, the tests listed on the
213 command-line will be used. If that's empty, too, then all *.py
214 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000215
Christian Heimesb186d002008-03-18 15:15:01 +0000216 The other default arguments (verbose, quiet, generate, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000217 single, randomize, findleaks, use_resources, trace, coverdir,
218 print_slow, and random_seed) allow programmers calling main()
219 directly to set the values that would normally be set by flags
220 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000221 """
Fred Drake004d5e62000-10-23 17:22:08 +0000222
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000223 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000224 try:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000225 opts, args = getopt.getopt(sys.argv[1:], 'hvgqxsSrf:lu:t:TD:NLR:wM:n',
Christian Heimesb186d002008-03-18 15:15:01 +0000226 ['help', 'verbose', 'quiet', 'exclude',
227 'single', 'slow', 'random', 'fromfile',
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000228 'findleaks', 'use=', 'threshold=', 'trace',
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000229 'coverdir=', 'nocoverdir', 'runleaks',
Thomas Wouters477c8d52006-05-27 19:21:47 +0000230 'huntrleaks=', 'verbose2', 'memlimit=',
Collin Winterfd12f492009-03-29 04:05:05 +0000231 'debug', 'start=', 'nowindows',
232 'randseed=',
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000233 ])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000234 except getopt.error as msg:
235 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000236
237 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000238 if random_seed is None:
239 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000240 if use_resources is None:
241 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000242 debug = False
243 start = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000244 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000245 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000246 print(__doc__)
247 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000248 elif o in ('-v', '--verbose'):
249 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000250 elif o in ('-w', '--verbose2'):
251 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000252 elif o in ('-d', '--debug'):
253 debug = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000254 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000255 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000256 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000257 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000258 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000259 elif o in ('-S', '--start'):
260 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000261 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000262 single = True
Christian Heimesb186d002008-03-18 15:15:01 +0000263 elif o in ('-S', '--slow'):
264 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000265 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000266 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000267 elif o == '--randseed':
268 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000269 elif o in ('-f', '--fromfile'):
270 fromfile = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000271 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000272 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000273 elif o in ('-L', '--runleaks'):
274 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000275 elif o in ('-t', '--threshold'):
276 import gc
277 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000278 elif o in ('-T', '--coverage'):
279 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000280 elif o in ('-D', '--coverdir'):
281 coverdir = os.path.join(os.getcwd(), a)
282 elif o in ('-N', '--nocoverdir'):
283 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000284 elif o in ('-R', '--huntrleaks'):
285 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000286 if len(huntrleaks) not in (2, 3):
287 print(a, huntrleaks)
288 usage('-R takes 2 or 3 colon-separated arguments')
289 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000290 huntrleaks[0] = 5
291 else:
292 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000293 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000294 huntrleaks[1] = 4
295 else:
296 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000297 if len(huntrleaks) == 2 or not huntrleaks[2]:
298 huntrleaks[2:] = ["reflog.txt"]
299 # Avoid false positives due to the character cache in
300 # stringobject.c filling slowly with random data
301 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000302 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000303 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000304 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000305 u = [x.lower() for x in a.split(',')]
306 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000307 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000308 use_resources[:] = RESOURCE_NAMES
309 continue
310 remove = False
311 if r[0] == '-':
312 remove = True
313 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000314 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000315 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000316 if remove:
317 if r in use_resources:
318 use_resources.remove(r)
319 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000320 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000321 elif o in ('-n', '--nowindows'):
322 import msvcrt
323 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
324 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
325 msvcrt.SEM_NOGPFAULTERRORBOX|
326 msvcrt.SEM_NOOPENFILEERRORBOX)
327 try:
328 msvcrt.CrtSetReportMode
329 except AttributeError:
330 # release build
331 pass
332 else:
333 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
334 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
335 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
336 if generate and verbose:
337 usage("-g and -v don't go together!")
Tim Petersc5000df2002-06-02 21:42:01 +0000338 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000339 usage("-s and -f don't go together!")
Barry Warsaw08fca522001-08-20 22:33:46 +0000340
Guido van Rossum152494a1996-12-20 03:12:20 +0000341 good = []
342 bad = []
343 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000344 resource_denieds = []
Barry Warsawe11e3de1999-01-28 19:51:51 +0000345
Neil Schemenauerd569f232000-09-22 15:29:28 +0000346 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000347 try:
348 import gc
349 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000350 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000351 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000352 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000353 # Uncomment the line below to report garbage that is not
354 # freeable by reference counting alone. By default only
355 # garbage that is not collectable by the GC is reported.
356 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000357 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000358
Barry Warsawe11e3de1999-01-28 19:51:51 +0000359 if single:
360 from tempfile import gettempdir
361 filename = os.path.join(gettempdir(), 'pynexttest')
362 try:
363 fp = open(filename, 'r')
Eric S. Raymondfc170b12001-02-09 11:51:27 +0000364 next = fp.read().strip()
Barry Warsawe11e3de1999-01-28 19:51:51 +0000365 tests = [next]
366 fp.close()
367 except IOError:
368 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000369
370 if fromfile:
371 tests = []
372 fp = open(fromfile)
373 for line in fp:
374 guts = line.split() # assuming no test has whitespace in its name
375 if guts and not guts[0].startswith('#'):
376 tests.extend(guts)
377 fp.close()
378
379 # Strip .py extensions.
380 if args:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000381 args = list(map(removepy, args))
Tim Petersc5000df2002-06-02 21:42:01 +0000382 if tests:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000383 tests = list(map(removepy, tests))
Tim Petersc5000df2002-06-02 21:42:01 +0000384
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000385 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000386 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000387 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000388 for arg in args:
389 if arg in stdtests:
390 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000391 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000392 args = []
Guido van Rossum747e1ca1998-08-24 13:48:36 +0000393 tests = tests or args or findtests(testdir, stdtests, nottests)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000394 if single:
395 tests = tests[:1]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000396 # Remove all the tests that precede start if it's set.
397 if start:
398 try:
399 del tests[:tests.index(start)]
400 except ValueError:
401 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000402 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000403 random.seed(random_seed)
404 print("Using random seed", random_seed)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000405 random.shuffle(tests)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000406 if trace:
407 import trace
408 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix],
409 trace=False, count=True)
Christian Heimesb186d002008-03-18 15:15:01 +0000410 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000411 support.verbose = verbose # Tell tests to be moderately quiet
412 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000413 save_modules = sys.modules.keys()
Guido van Rossum152494a1996-12-20 03:12:20 +0000414 for test in tests:
Guido van Rossum41360a41998-03-26 19:42:58 +0000415 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000416 print(test)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000417 sys.stdout.flush()
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000418 if trace:
419 # If we're tracing code coverage, then we don't exit with status
420 # if on a false return value from main.
Christian Heimesb186d002008-03-18 15:15:01 +0000421 tracer.runctx('runtest(test, generate, verbose, quiet,'
422 ' test_times, testdir)',
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000423 globals=globals(), locals=vars())
Guido van Rossum41360a41998-03-26 19:42:58 +0000424 else:
Neal Norwitz14ca3272006-02-28 18:05:43 +0000425 try:
Christian Heimesb186d002008-03-18 15:15:01 +0000426 ok = runtest(test, generate, verbose, quiet, test_times,
427 testdir, huntrleaks)
Neal Norwitz14ca3272006-02-28 18:05:43 +0000428 except KeyboardInterrupt:
429 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000430 print()
Neal Norwitz14ca3272006-02-28 18:05:43 +0000431 break
432 except:
433 raise
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000434 if ok > 0:
435 good.append(test)
436 elif ok == 0:
437 bad.append(test)
438 else:
439 skipped.append(test)
440 if ok == -2:
441 resource_denieds.append(test)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000442 if findleaks:
443 gc.collect()
444 if gc.garbage:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000445 print("Warning: test created", len(gc.garbage), end=' ')
446 print("uncollectable object(s).")
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000447 # move the uncollectable objects somewhere so we don't see
448 # them again
Neil Schemenauerd569f232000-09-22 15:29:28 +0000449 found_garbage.extend(gc.garbage)
450 del gc.garbage[:]
Guido van Rossum5796d262000-04-21 21:35:06 +0000451 # Unload the newly imported modules (best effort finalization)
452 for module in sys.modules.keys():
Guido van Rossum51931142000-05-05 14:27:39 +0000453 if module not in save_modules and module.startswith("test."):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000454 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000455
456 # The lists won't be sorted if running with -r
457 good.sort()
458 bad.sort()
459 skipped.sort()
Tim Peterse0c446b2001-10-18 21:57:37 +0000460
Guido van Rossum152494a1996-12-20 03:12:20 +0000461 if good and not quiet:
Guido van Rossum41360a41998-03-26 19:42:58 +0000462 if not bad and not skipped and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000463 print("All", end=' ')
464 print(count(len(good), "test"), "OK.")
465 if verbose:
466 print("CAUTION: stdout isn't compared in verbose mode:")
467 print("a test that passes in verbose mode may fail without it.")
Christian Heimesb186d002008-03-18 15:15:01 +0000468 if print_slow:
469 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000470 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000471 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000472 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000473 if bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000474 print(count(len(bad), "test"), "failed:")
Tim Petersa45da922001-08-12 03:45:50 +0000475 printlist(bad)
Guido van Rossum152494a1996-12-20 03:12:20 +0000476 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000477 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000478 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000479
Tim Petersb5b7b782001-08-12 01:20:39 +0000480 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000481 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000482 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000483 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000484 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000485 print(count(len(surprise), "skip"), \
486 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000487 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000488 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000489 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000490 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000491 print("Ask someone to teach regrtest.py about which tests are")
492 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000493
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000494 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000495 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000496 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000497 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000498 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000499 try:
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000500 support.verbose = True
Neal Norwitz931936e2008-03-18 17:44:29 +0000501 ok = runtest(test, generate, True, quiet, test_times, testdir,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000502 huntrleaks, debug)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000503 except KeyboardInterrupt:
504 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000505 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000506 break
507 except:
508 raise
509
Barry Warsawe11e3de1999-01-28 19:51:51 +0000510 if single:
511 alltests = findtests(testdir, stdtests, nottests)
512 for i in range(len(alltests)):
513 if tests[0] == alltests[i]:
514 if i == len(alltests) - 1:
515 os.unlink(filename)
516 else:
517 fp = open(filename, 'w')
518 fp.write(alltests[i+1] + '\n')
519 fp.close()
520 break
521 else:
522 os.unlink(filename)
523
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000524 if trace:
525 r = tracer.results()
526 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
527
Skip Montanaro0179a182004-06-06 15:53:18 +0000528 if runleaks:
529 os.system("leaks %d" % os.getpid())
530
Tim Peters5943b4a2003-07-23 00:30:39 +0000531 sys.exit(len(bad) > 0)
Barry Warsaw08fca522001-08-20 22:33:46 +0000532
Guido van Rossum152494a1996-12-20 03:12:20 +0000533
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000534STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000535 'test_grammar',
536 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000537 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000538 'test_builtin',
539 'test_exceptions',
540 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000541 'test_unittest',
542 'test_doctest',
543 'test_doctest2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000544]
Guido van Rossum152494a1996-12-20 03:12:20 +0000545
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000546NOTTESTS = {
Jeremy Hylton62e2c7e2001-02-28 17:48:06 +0000547 'test_future1',
548 'test_future2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000549}
Guido van Rossum152494a1996-12-20 03:12:20 +0000550
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000551def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000552 """Return a list of all applicable test modules."""
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000553 if not testdir: testdir = findtestdir()
Guido van Rossum152494a1996-12-20 03:12:20 +0000554 names = os.listdir(testdir)
555 tests = []
556 for name in names:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000557 if name[:5] == "test_" and name[-3:] == ".py":
Guido van Rossum41360a41998-03-26 19:42:58 +0000558 modname = name[:-3]
559 if modname not in stdtests and modname not in nottests:
560 tests.append(modname)
Guido van Rossum152494a1996-12-20 03:12:20 +0000561 tests.sort()
562 return stdtests + tests
563
Christian Heimesb186d002008-03-18 15:15:01 +0000564def runtest(test, generate, verbose, quiet, test_times,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000565 testdir=None, huntrleaks=False, debug=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000566 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000567
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000568 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000569 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000570 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000571 test_times -- a list of (time, test_name) pairs
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000572 testdir -- test directory
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000573 huntrleaks -- run multiple times to test for leaks; requires a debug
574 build; a triple corresponding to -R's three arguments
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000575 debug -- if true, print tracebacks for failed tests regardless of
576 verbose setting
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000577 Return:
578 -2 test skipped because resource denied
579 -1 test skipped for some other reason
580 0 test failed
581 1 test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000582 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000583
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000584 try:
Christian Heimesb186d002008-03-18 15:15:01 +0000585 return runtest_inner(test, generate, verbose, quiet, test_times,
586 testdir, huntrleaks)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000587 finally:
588 cleanup_test_droppings(test, verbose)
589
Christian Heimesb186d002008-03-18 15:15:01 +0000590def runtest_inner(test, generate, verbose, quiet, test_times,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000591 testdir=None, huntrleaks=False, debug=False):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000592 support.unload(test)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000593 if not testdir:
594 testdir = findtestdir()
Tim Peters9390cc12001-09-28 20:14:46 +0000595 if verbose:
Guido van Rossum41360a41998-03-26 19:42:58 +0000596 cfp = None
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000597 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000598 cfp = io.StringIO() # XXX Should use io.StringIO()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000599
Guido van Rossum152494a1996-12-20 03:12:20 +0000600 try:
Tim Peters342ca752001-09-25 19:13:20 +0000601 save_stdout = sys.stdout
Guido van Rossum41360a41998-03-26 19:42:58 +0000602 try:
603 if cfp:
604 sys.stdout = cfp
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000605 print(test) # Output file starts with test name
Barry Warsaw408b6d32002-07-30 23:27:12 +0000606 if test.startswith('test.'):
607 abstest = test
608 else:
609 # Always import it from the test package
610 abstest = 'test.' + test
Christian Heimesb186d002008-03-18 15:15:01 +0000611 start_time = time.time()
Barry Warsaw408b6d32002-07-30 23:27:12 +0000612 the_package = __import__(abstest, globals(), locals(), [])
613 the_module = getattr(the_package, test)
Guido van Rossum58da9312007-11-10 23:39:45 +0000614 # Old tests run to completion simply as a side-effect of
615 # being imported. For tests based on unittest or doctest,
616 # explicitly invoke their test_main() function (if it exists).
Tim Petersd9742212001-05-22 18:28:25 +0000617 indirect_test = getattr(the_module, "test_main", None)
618 if indirect_test is not None:
619 indirect_test()
Michael W. Hudson61147f62004-08-03 11:33:28 +0000620 if huntrleaks:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000621 dash_R(the_module, test, indirect_test, huntrleaks)
Christian Heimesb186d002008-03-18 15:15:01 +0000622 test_time = time.time() - start_time
623 test_times.append((test_time, test))
Guido van Rossum41360a41998-03-26 19:42:58 +0000624 finally:
Tim Peters342ca752001-09-25 19:13:20 +0000625 sys.stdout = save_stdout
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000626 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +0000627 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000628 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +0000629 sys.stdout.flush()
630 return -2
R. David Murraya21e4ca2009-03-31 23:16:50 +0000631 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +0000632 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000633 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000634 sys.stdout.flush()
Guido van Rossum41360a41998-03-26 19:42:58 +0000635 return -1
Fred Drakefe5c22a2000-08-18 16:04:05 +0000636 except KeyboardInterrupt:
637 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000638 except support.TestFailed as msg:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000639 print("test", test, "failed --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000640 sys.stdout.flush()
Guido van Rossum41360a41998-03-26 19:42:58 +0000641 return 0
Guido van Rossum9e48b271997-07-16 01:56:13 +0000642 except:
Guido van Rossum41360a41998-03-26 19:42:58 +0000643 type, value = sys.exc_info()[:2]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000644 print("test", test, "crashed --", str(type) + ":", value)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000645 sys.stdout.flush()
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000646 if verbose or debug:
Guido van Rossum41360a41998-03-26 19:42:58 +0000647 traceback.print_exc(file=sys.stdout)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000648 sys.stdout.flush()
Guido van Rossum41360a41998-03-26 19:42:58 +0000649 return 0
Guido van Rossum152494a1996-12-20 03:12:20 +0000650 else:
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000651 if not cfp:
652 return 1
653 output = cfp.getvalue()
Christian Heimesb186d002008-03-18 15:15:01 +0000654 expected = test + "\n"
Michael W. Hudson61147f62004-08-03 11:33:28 +0000655 if output == expected or huntrleaks:
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000656 return 1
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000657 print("test", test, "produced unexpected output:")
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000658 sys.stdout.flush()
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000659 reportdiff(expected, output)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000660 sys.stdout.flush()
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000661 return 0
662
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000663def cleanup_test_droppings(testname, verbose):
664 import shutil
665
666 # Try to clean up junk commonly left behind. While tests shouldn't leave
667 # any files or directories behind, when a test fails that can be tedious
668 # for it to arrange. The consequences can be especially nasty on Windows,
669 # since if a test leaves a file open, it cannot be deleted by name (while
670 # there's nothing we can do about that here either, we can display the
671 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000672 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000673 "db_home",
674 ):
675 if not os.path.exists(name):
676 continue
677
678 if os.path.isdir(name):
679 kind, nuker = "directory", shutil.rmtree
680 elif os.path.isfile(name):
681 kind, nuker = "file", os.unlink
682 else:
683 raise SystemError("os.path says %r exists but is neither "
684 "directory nor file" % name)
685
686 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000687 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000688 try:
689 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000690 except Exception as msg:
691 print(("%r left behind %s %r and it couldn't be "
692 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000693
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000694def dash_R(the_module, test, indirect_test, huntrleaks):
695 # This code is hackish and inelegant, but it seems to do the job.
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +0000696 import copyreg, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000697
698 if not hasattr(sys, 'gettotalrefcount'):
699 raise Exception("Tracking reference leaks requires a debug build "
700 "of Python")
701
702 # Save current values for dash_R_cleanup() to restore.
703 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +0000704 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000705 pic = sys.path_importer_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +0000706 abcs = {}
707 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +0000708 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +0000709 continue
710 for obj in abc.__subclasses__() + [abc]:
711 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000712
713 if indirect_test:
714 def run_the_test():
715 indirect_test()
716 else:
717 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000718 del sys.modules[the_module.__name__]
719 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000720
721 deltas = []
722 nwarmup, ntracked, fname = huntrleaks
723 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000724 print("beginning", repcount, "repetitions", file=sys.stderr)
725 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Guido van Rossum3de862d2007-08-18 00:10:33 +0000726 dash_R_cleanup(fs, ps, pic, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000727 for i in range(repcount):
728 rc = sys.gettotalrefcount()
729 run_the_test()
730 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000731 sys.stderr.flush()
Guido van Rossum3de862d2007-08-18 00:10:33 +0000732 dash_R_cleanup(fs, ps, pic, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000733 if i >= nwarmup:
734 deltas.append(sys.gettotalrefcount() - rc - 2)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000735 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000736 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000737 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000738 print(msg, file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000739 refrep = open(fname, "a")
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000740 print(msg, file=refrep)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000741 refrep.close()
742
Guido van Rossum3de862d2007-08-18 00:10:33 +0000743def dash_R_cleanup(fs, ps, pic, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +0000744 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +0000745 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000746 import urllib.parse, urllib.request, mimetypes, doctest
Guido van Rossum7eaf8222007-06-18 17:58:50 +0000747 import struct, filecmp, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000748 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000749 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000750
Christian Heimesdae2a892008-04-19 00:55:37 +0000751 # Clear the warnings registry, so they can be displayed again
752 for mod in sys.modules.values():
753 if hasattr(mod, '__warningregistry__'):
754 del mod.__warningregistry__
755
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000756 # Restore some original values.
757 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +0000758 copyreg.dispatch_table.clear()
759 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000760 sys.path_importer_cache.clear()
761 sys.path_importer_cache.update(pic)
762
Christian Heimes26855632008-01-27 23:50:43 +0000763 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +0000764 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +0000765
Guido van Rossum3de862d2007-08-18 00:10:33 +0000766 # Clear ABC registries, restoring previously saved ABC registries.
Guido van Rossum7eaf8222007-06-18 17:58:50 +0000767 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +0000768 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +0000769 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +0000770 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000771 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +0000772 obj._abc_cache.clear()
773 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000774
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000775 # Clear assorted module caches.
776 _path_created.clear()
777 re.purge()
778 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000779 urllib.parse.clear_cache()
780 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000781 linecache.clearcache()
782 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000783 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +0000784 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000785 doctest.master = None
786
787 # Collect cyclic trash.
788 gc.collect()
789
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000790def warm_char_cache():
791 s = bytes(range(256))
792 for i in range(256):
793 s[i:i+1]
794
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000795def reportdiff(expected, output):
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000796 import difflib
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000797 print("*" * 70)
Tim Petersc377b162001-09-22 05:31:03 +0000798 a = expected.splitlines(1)
799 b = output.splitlines(1)
Guido van Rossumcf691932001-09-21 21:06:22 +0000800 sm = difflib.SequenceMatcher(a=a, b=b)
801 tuples = sm.get_opcodes()
Tim Petersc377b162001-09-22 05:31:03 +0000802
Guido van Rossumcf691932001-09-21 21:06:22 +0000803 def pair(x0, x1):
Tim Petersc377b162001-09-22 05:31:03 +0000804 # x0:x1 are 0-based slice indices; convert to 1-based line indices.
Guido van Rossumcf691932001-09-21 21:06:22 +0000805 x0 += 1
806 if x0 >= x1:
Tim Petersc377b162001-09-22 05:31:03 +0000807 return "line " + str(x0)
Guido van Rossumcf691932001-09-21 21:06:22 +0000808 else:
Tim Petersc377b162001-09-22 05:31:03 +0000809 return "lines %d-%d" % (x0, x1)
810
Guido van Rossumcf691932001-09-21 21:06:22 +0000811 for op, a0, a1, b0, b1 in tuples:
812 if op == 'equal':
813 pass
Tim Petersc377b162001-09-22 05:31:03 +0000814
Guido van Rossumcf691932001-09-21 21:06:22 +0000815 elif op == 'delete':
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000816 print("***", pair(a0, a1), "of expected output missing:")
Guido van Rossumcf691932001-09-21 21:06:22 +0000817 for line in a[a0:a1]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000818 print("-", line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000819
Guido van Rossumcf691932001-09-21 21:06:22 +0000820 elif op == 'replace':
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000821 print("*** mismatch between", pair(a0, a1), "of expected", \
822 "output and", pair(b0, b1), "of actual output:")
Tim Petersc377b162001-09-22 05:31:03 +0000823 for line in difflib.ndiff(a[a0:a1], b[b0:b1]):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000824 print(line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000825
Guido van Rossumcf691932001-09-21 21:06:22 +0000826 elif op == 'insert':
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000827 print("***", pair(b0, b1), "of actual output doesn't appear", \
828 "in expected output after line", str(a1)+":")
Guido van Rossumcf691932001-09-21 21:06:22 +0000829 for line in b[b0:b1]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000830 print("+", line, end='')
Tim Petersc377b162001-09-22 05:31:03 +0000831
Guido van Rossumcf691932001-09-21 21:06:22 +0000832 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000833 print("get_opcodes() returned bad tuple?!?!", (op, a0, a1, b0, b1))
Tim Petersc377b162001-09-22 05:31:03 +0000834
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000835 print("*" * 70)
Guido van Rossum152494a1996-12-20 03:12:20 +0000836
837def findtestdir():
838 if __name__ == '__main__':
Guido van Rossum41360a41998-03-26 19:42:58 +0000839 file = sys.argv[0]
Guido van Rossum152494a1996-12-20 03:12:20 +0000840 else:
Guido van Rossum41360a41998-03-26 19:42:58 +0000841 file = __file__
Guido van Rossum152494a1996-12-20 03:12:20 +0000842 testdir = os.path.dirname(file) or os.curdir
843 return testdir
844
Tim Petersc5000df2002-06-02 21:42:01 +0000845def removepy(name):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000846 if name.endswith(".py"):
Tim Petersc5000df2002-06-02 21:42:01 +0000847 name = name[:-3]
848 return name
849
Guido van Rossum152494a1996-12-20 03:12:20 +0000850def count(n, word):
851 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +0000852 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +0000853 else:
Guido van Rossum41360a41998-03-26 19:42:58 +0000854 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +0000855
Tim Petersa45da922001-08-12 03:45:50 +0000856def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +0000857 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +0000858
859 Optional arg width (default 70) is the maximum line length.
860 Optional arg indent (default 4) is the number of blanks with which to
861 begin each line.
862 """
863
Tim Petersba78bc42002-07-04 19:45:06 +0000864 from textwrap import fill
865 blanks = ' ' * indent
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000866 print(fill(' '.join(map(str, x)), width,
867 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +0000868
Tim Petersde14a302002-04-01 05:04:46 +0000869# Map sys.platform to a string containing the basenames of tests
870# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +0000871#
872# Special cases:
873# test_pep277
874# The _ExpectedSkips constructor adds this to the set of expected
875# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +0000876# test_timeout
877# Controlled by test_timeout.skip_expected. Requires the network
878# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000879#
880# Tests that are expected to be skipped everywhere except on one platform
881# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +0000882
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000883_expectations = {
884 'win32':
885 """
Tim Petersc7c516a2003-09-20 22:06:13 +0000886 test__locale
Tim Peters78e35f92002-11-22 20:00:34 +0000887 test_bsddb3
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000888 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +0000889 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000890 test_dbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000891 test_fcntl
892 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000893 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +0000894 test_dbm_gnu
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000895 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +0000896 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000897 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000898 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000899 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +0000900 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000901 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000902 test_poll
Tim Peters003eb302003-02-17 21:48:48 +0000903 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000904 test_pty
905 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +0000906 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000907 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000908 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +0000909 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000910 test_wait3
911 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000912 """,
913 'linux2':
914 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +0000915 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000916 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000917 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +0000918 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +0000919 """,
Jack Jansen49a806e2001-08-28 14:49:00 +0000920 'mac':
Guido van Rossumaa782362001-09-02 03:58:41 +0000921 """
Jack Jansen67975142003-01-08 16:31:11 +0000922 test_atexit
Guido van Rossumaa782362001-09-02 03:58:41 +0000923 test_bsddb
Jack Jansen67975142003-01-08 16:31:11 +0000924 test_bsddb3
925 test_bz2
Guido van Rossumaa782362001-09-02 03:58:41 +0000926 test_crypt
Jack Jansenb3be2162001-11-30 14:16:36 +0000927 test_curses
Guido van Rossumaa782362001-09-02 03:58:41 +0000928 test_dbm
Guido van Rossumaa782362001-09-02 03:58:41 +0000929 test_fcntl
930 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000931 test_epoll
Guido van Rossumaa782362001-09-02 03:58:41 +0000932 test_grp
Jack Jansenc4d6bdd2003-03-07 15:38:11 +0000933 test_ioctl
Guido van Rossumaa782362001-09-02 03:58:41 +0000934 test_largefile
Guido van Rossumaa782362001-09-02 03:58:41 +0000935 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000936 test_kqueue
Guido van Rossumaa782362001-09-02 03:58:41 +0000937 test_mmap
Guido van Rossumaa782362001-09-02 03:58:41 +0000938 test_openpty
Jack Jansen67975142003-01-08 16:31:11 +0000939 test_ossaudiodev
Guido van Rossumaa782362001-09-02 03:58:41 +0000940 test_poll
Jack Jansen67975142003-01-08 16:31:11 +0000941 test_popen
Jack Jansen5bb97e62003-02-21 22:33:55 +0000942 test_posix
Guido van Rossumaa782362001-09-02 03:58:41 +0000943 test_pty
944 test_pwd
Jack Jansen67975142003-01-08 16:31:11 +0000945 test_resource
Guido van Rossumaa782362001-09-02 03:58:41 +0000946 test_signal
Guido van Rossumaa782362001-09-02 03:58:41 +0000947 test_sundry
Jack Jansenc4d6bdd2003-03-07 15:38:11 +0000948 test_tarfile
Guido van Rossumaa782362001-09-02 03:58:41 +0000949 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000950 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000951 """
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000952 test_bsddb
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000953 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000954 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000955 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000956 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000957 test_openpty
958 test_pyexpat
959 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000960 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +0000961 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000962 'openunix8':
963 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000964 test_bsddb
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000965 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000966 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000967 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000968 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000969 test_openpty
970 test_pyexpat
971 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000972 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000973 """,
974 'sco_sv3':
975 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000976 test_asynchat
977 test_bsddb
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000978 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000979 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000980 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000981 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000982 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +0000983 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000984 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000985 test_openpty
986 test_pyexpat
987 test_queue
988 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000989 test_sundry
990 test_thread
991 test_threaded_import
992 test_threadedtempfile
993 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +0000994 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +0000995 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +0000996 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000997 test__locale
998 test_bsddb
999 test_bsddb3
1000 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001001 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001002 test_dbm_gnu
Jack Jansen398c2362001-12-02 21:41:36 +00001003 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001004 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001005 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001006 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001007 test_poll
Jack Jansen398c2362001-12-02 21:41:36 +00001008 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001009 'sunos5':
1010 """
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
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001014 test_epoll
1015 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001016 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001017 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001018 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001019 test_zipfile
1020 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001021 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001022 'hp-ux11':
1023 """
Skip Montanarob3230212002-03-15 02:54:03 +00001024 test_bsddb
Skip Montanarob3230212002-03-15 02:54:03 +00001025 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001026 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001027 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001028 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001029 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001030 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001031 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001032 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001033 test_openpty
1034 test_pyexpat
1035 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001036 test_zipfile
1037 test_zlib
1038 """,
Martin v. Löwisf90ae202002-06-11 06:22:31 +00001039 'atheos':
Tim Petersc411dba2002-07-16 21:35:23 +00001040 """
Tim Petersc411dba2002-07-16 21:35:23 +00001041 test_curses
Benjamin Peterson4d480532008-05-26 19:08:31 +00001042 test_dbm_gnu
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001043 test_epoll
Tim Petersc411dba2002-07-16 21:35:23 +00001044 test_largefile
Tim Petersc411dba2002-07-16 21:35:23 +00001045 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001046 test_kqueue
Tim Petersc411dba2002-07-16 21:35:23 +00001047 test_mhlib
1048 test_mmap
Tim Petersc411dba2002-07-16 21:35:23 +00001049 test_poll
Tim Petersc411dba2002-07-16 21:35:23 +00001050 test_resource
Tim Petersc411dba2002-07-16 21:35:23 +00001051 """,
Jason Tishler25115942002-12-05 15:18:15 +00001052 'cygwin':
1053 """
Tim Petersb0f89e02002-12-05 17:20:25 +00001054 test_bsddb3
Jason Tishler25115942002-12-05 15:18:15 +00001055 test_curses
1056 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001057 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001058 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001059 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001060 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001061 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001062 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001063 test_socketserver
Jason Tishler25115942002-12-05 15:18:15 +00001064 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001065 'os2emx':
1066 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001067 test_audioop
1068 test_bsddb3
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001069 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001070 test_epoll
1071 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001072 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001073 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001074 test_openpty
1075 test_ossaudiodev
1076 test_pty
1077 test_resource
1078 test_signal
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001079 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001080 'freebsd4':
1081 """
Guido van Rossum944a6c32003-11-20 22:11:29 +00001082 test_bsddb
1083 test_bsddb3
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001084 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001085 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001086 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001087 test_ossaudiodev
1088 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001089 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001090 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001091 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001092 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001093 test_ttk_guionly
1094 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001095 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001096 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001097 test_multiprocessing
Martin v. Löwis56f88112003-06-07 20:01:37 +00001098 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001099 'aix5':
1100 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001101 test_bsddb
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001102 test_bsddb3
1103 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001104 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001105 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001106 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001107 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001108 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001109 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001110 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001111 test_ttk_guionly
1112 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001113 test_zipimport
1114 test_zlib
1115 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001116 'openbsd3':
1117 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001118 test_bsddb
1119 test_bsddb3
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001120 test_ctypes
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001121 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001122 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001123 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001124 test_normalization
1125 test_ossaudiodev
1126 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001127 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001128 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001129 test_ttk_guionly
1130 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001131 test_multiprocessing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001132 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001133 'netbsd3':
1134 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001135 test_bsddb
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001136 test_bsddb3
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001137 test_ctypes
1138 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001139 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001140 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001141 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001142 test_ossaudiodev
1143 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001144 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001145 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001146 test_ttk_guionly
1147 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001148 test_multiprocessing
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001149 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001150}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001151_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001152_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001153_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossum8ce8a782007-11-01 19:42:39 +00001154_expectations['freebsd8'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001155
Tim Petersb5b7b782001-08-12 01:20:39 +00001156class _ExpectedSkips:
1157 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001158 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001159 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001160
Tim Peters7c7efe92002-08-23 17:55:54 +00001161 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001162 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001163 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001164 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001165
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001166 # These are broken tests, for now skipped on every platform.
1167 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001168 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001169
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001170 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001171 if not os.path.supports_unicode_filenames:
1172 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001173
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001174 # doctest, profile and cProfile tests fail when the codec for the
1175 # fs encoding isn't built in because PyUnicode_Decode() adds two
1176 # calls into Python.
1177 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1178 if sys.getfilesystemencoding().lower() not in encs:
1179 self.expected.add('test_profile')
1180 self.expected.add('test_cProfile')
1181 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001182
Neal Norwitz55b61d22003-02-28 19:57:03 +00001183 if test_timeout.skip_expected:
1184 self.expected.add('test_timeout')
1185
Tim Petersecd79eb2003-01-29 00:35:32 +00001186 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001187 # test_sqlite is only reliable on Windows where the library
1188 # is distributed with Python
Neal Norwitz7035c982003-03-29 22:01:17 +00001189 WIN_ONLY = ["test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001190 "test_winsound", "test_startfile",
1191 "test_sqlite"]
Neal Norwitz7035c982003-03-29 22:01:17 +00001192 for skip in WIN_ONLY:
1193 self.expected.add(skip)
Tim Petersf2715e02003-02-19 02:35:07 +00001194
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001195 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001196 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001197
Tim Peters7c7efe92002-08-23 17:55:54 +00001198 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001199
1200 def isvalid(self):
1201 "Return true iff _ExpectedSkips knows about the current platform."
1202 return self.valid
1203
1204 def getexpected(self):
1205 """Return set of test names we expect to skip on current platform.
1206
1207 self.isvalid() must be true.
1208 """
1209
1210 assert self.isvalid()
1211 return self.expected
1212
Guido van Rossum152494a1996-12-20 03:12:20 +00001213if __name__ == '__main__':
Barry Warsaw408b6d32002-07-30 23:27:12 +00001214 # Remove regrtest.py's own directory from the module search path. This
1215 # prevents relative imports from working, and relative imports will screw
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001216 # up the testing framework. E.g. if both test.support and
1217 # support are imported, they will not contain the same globals, and
Barry Warsaw408b6d32002-07-30 23:27:12 +00001218 # much of the testing framework relies on the globals in the
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001219 # test.support module.
Barry Warsaw408b6d32002-07-30 23:27:12 +00001220 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1221 i = pathlen = len(sys.path)
1222 while i >= 0:
1223 i -= 1
1224 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1225 del sys.path[i]
1226 if len(sys.path) == pathlen:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001227 print('Could not find %r in sys.path to remove it' % mydir)
Barry Warsaw08fca522001-08-20 22:33:46 +00001228 main()