| Benjamin Peterson | 90f5ba5 | 2010-03-11 22:53:45 +0000 | [diff] [blame] | 1 | #! /usr/bin/env python3 | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 2 |  | 
| R. David Murray | 0ba81e0 | 2010-04-26 17:02:32 +0000 | [diff] [blame] | 3 | """ | 
|  | 4 | Usage: | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 5 |  | 
| Georg Brandl | be41a48 | 2011-01-05 21:47:47 +0000 | [diff] [blame] | 6 | python -m test [options] [test_name1 [test_name2 ...]] | 
| R. David Murray | 0ba81e0 | 2010-04-26 17:02:32 +0000 | [diff] [blame] | 7 | python path/to/Lib/test/regrtest.py [options] [test_name1 [test_name2 ...]] | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 8 |  | 
| R. David Murray | 0ba81e0 | 2010-04-26 17:02:32 +0000 | [diff] [blame] | 9 |  | 
|  | 10 | If no arguments or options are provided, finds all files matching | 
|  | 11 | the pattern "test_*" in the Lib/test subdirectory and runs | 
|  | 12 | them in alphabetical order (but see -M and -u, below, for exceptions). | 
|  | 13 |  | 
|  | 14 | For more rigorous testing, it is useful to use the following | 
|  | 15 | command line: | 
|  | 16 |  | 
| Georg Brandl | be41a48 | 2011-01-05 21:47:47 +0000 | [diff] [blame] | 17 | python -E -Wd -m test [options] [test_name1 ...] | 
| R. David Murray | 0ba81e0 | 2010-04-26 17:02:32 +0000 | [diff] [blame] | 18 |  | 
|  | 19 |  | 
|  | 20 | Options: | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 21 |  | 
| Georg Brandl | 96f7897 | 2009-10-27 21:02:11 +0000 | [diff] [blame] | 22 | -h/--help       -- print this text and exit | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 23 | --timeout TIMEOUT | 
|  | 24 | -- dump the traceback and exit if a test takes more | 
| Victor Stinner | d91a5ca | 2011-04-01 18:16:36 +0200 | [diff] [blame] | 25 | than TIMEOUT seconds (default: 30 minutes); disable | 
| Victor Stinner | 305bff1 | 2011-04-01 15:59:59 +0200 | [diff] [blame] | 26 | the timeout if TIMEOUT is zero | 
| Georg Brandl | 96f7897 | 2009-10-27 21:02:11 +0000 | [diff] [blame] | 27 |  | 
|  | 28 | Verbosity | 
|  | 29 |  | 
|  | 30 | -v/--verbose    -- run tests in verbose mode with output to stdout | 
|  | 31 | -w/--verbose2   -- re-run failed tests in verbose mode | 
|  | 32 | -W/--verbose3   -- re-run failed tests in verbose mode immediately | 
|  | 33 | -d/--debug      -- print traceback for failed tests | 
| R. David Murray | 0ba81e0 | 2010-04-26 17:02:32 +0000 | [diff] [blame] | 34 | -q/--quiet      -- no output unless one or more tests fail | 
| Georg Brandl | 96f7897 | 2009-10-27 21:02:11 +0000 | [diff] [blame] | 35 | -S/--slow       -- print the slowest 10 tests | 
| Antoine Pitrou | 3c4402f | 2011-01-03 20:38:52 +0000 | [diff] [blame] | 36 | --header     -- print header with interpreter info | 
| Georg Brandl | 96f7897 | 2009-10-27 21:02:11 +0000 | [diff] [blame] | 37 |  | 
|  | 38 | Selecting tests | 
|  | 39 |  | 
| R. David Murray | 0ba81e0 | 2010-04-26 17:02:32 +0000 | [diff] [blame] | 40 | -r/--random     -- randomize test execution order (see below) | 
| Antoine Pitrou | 68530ac | 2011-01-03 20:40:07 +0000 | [diff] [blame] | 41 | --randseed   -- pass a random seed to reproduce a previous random run | 
| Georg Brandl | 96f7897 | 2009-10-27 21:02:11 +0000 | [diff] [blame] | 42 | -f/--fromfile   -- read names of tests to run from a file (see below) | 
|  | 43 | -x/--exclude    -- arguments are tests to *exclude* | 
| R. David Murray | 0ba81e0 | 2010-04-26 17:02:32 +0000 | [diff] [blame] | 44 | -s/--single     -- single step through a set of tests (see below) | 
| Georg Brandl | 96f7897 | 2009-10-27 21:02:11 +0000 | [diff] [blame] | 45 | -u/--use RES1,RES2,... | 
|  | 46 | -- specify which special resource intensive tests to run | 
|  | 47 | -M/--memlimit LIMIT | 
|  | 48 | -- run very large memory-consuming tests | 
| R David Murray | b588f8d | 2011-03-24 14:42:58 -0400 | [diff] [blame] | 49 | --testdir DIR | 
|  | 50 | -- execute test files in the specified directory (instead | 
|  | 51 | of the Python stdlib test suite) | 
| Georg Brandl | 96f7897 | 2009-10-27 21:02:11 +0000 | [diff] [blame] | 52 |  | 
|  | 53 | Special runs | 
|  | 54 |  | 
|  | 55 | -l/--findleaks  -- if GC is available detect tests that leak memory | 
|  | 56 | -L/--runleaks   -- run the leaks(1) command just before exit | 
|  | 57 | -R/--huntrleaks RUNCOUNTS | 
|  | 58 | -- search for reference leaks (needs debug build, v. slow) | 
|  | 59 | -j/--multiprocess PROCESSES | 
|  | 60 | -- run PROCESSES processes at once | 
| R. David Murray | 0ba81e0 | 2010-04-26 17:02:32 +0000 | [diff] [blame] | 61 | -T/--coverage   -- turn on code coverage tracing using the trace module | 
| Georg Brandl | 96f7897 | 2009-10-27 21:02:11 +0000 | [diff] [blame] | 62 | -D/--coverdir DIRECTORY | 
|  | 63 | -- Directory where coverage files are put | 
|  | 64 | -N/--nocoverdir -- Put coverage files alongside modules | 
|  | 65 | -t/--threshold THRESHOLD | 
|  | 66 | -- call gc.set_threshold(THRESHOLD) | 
|  | 67 | -n/--nowindows  -- suppress error message boxes on Windows | 
| R. David Murray | 0ba81e0 | 2010-04-26 17:02:32 +0000 | [diff] [blame] | 68 | -F/--forever    -- run the specified tests in a loop, until an error happens | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 69 |  | 
| R. David Murray | 0ba81e0 | 2010-04-26 17:02:32 +0000 | [diff] [blame] | 70 |  | 
|  | 71 | Additional Option Details: | 
| Guido van Rossum | f58ed25 | 1997-03-07 21:04:33 +0000 | [diff] [blame] | 72 |  | 
| Collin Winter | fd12f49 | 2009-03-29 04:05:05 +0000 | [diff] [blame] | 73 | -r randomizes test execution order. You can use --randseed=int to provide a | 
|  | 74 | int seed value for the randomizer; this is useful for reproducing troublesome | 
|  | 75 | test orders. | 
|  | 76 |  | 
| R. David Murray | 0ba81e0 | 2010-04-26 17:02:32 +0000 | [diff] [blame] | 77 | -s On the first invocation of regrtest using -s, the first test file found | 
|  | 78 | or the first test file given on the command line is run, and the name of | 
|  | 79 | the next test is recorded in a file named pynexttest.  If run from the | 
|  | 80 | Python build directory, pynexttest is located in the 'build' subdirectory, | 
|  | 81 | otherwise it is located in tempfile.gettempdir().  On subsequent runs, | 
|  | 82 | the test in pynexttest is run, and the next test is written to pynexttest. | 
|  | 83 | When the last test has been run, pynexttest is deleted.  In this way it | 
|  | 84 | is possible to single step through the test files.  This is useful when | 
|  | 85 | doing memory analysis on the Python interpreter, which process tends to | 
|  | 86 | consume too many resources to run the full regression test non-stop. | 
| Barry Warsaw | e11e3de | 1999-01-28 19:51:51 +0000 | [diff] [blame] | 87 |  | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 88 | -S is used to continue running tests after an aborted run.  It will | 
|  | 89 | maintain the order a standard run (ie, this assumes -r is not used). | 
|  | 90 | This is useful after the tests have prematurely stopped for some external | 
|  | 91 | reason and you want to start running from where you left off rather | 
|  | 92 | than starting from the beginning. | 
|  | 93 |  | 
| Guido van Rossum | 9e9d4f8 | 2002-06-07 15:17:03 +0000 | [diff] [blame] | 94 | -f reads the names of tests from the file given as f's argument, one | 
|  | 95 | or more test names per line.  Whitespace is ignored.  Blank lines and | 
|  | 96 | lines beginning with '#' are ignored.  This is especially useful for | 
|  | 97 | whittling down failures involving interactions among tests. | 
| Tim Peters | c5000df | 2002-06-02 21:42:01 +0000 | [diff] [blame] | 98 |  | 
| Skip Montanaro | 0179a18 | 2004-06-06 15:53:18 +0000 | [diff] [blame] | 99 | -L causes the leaks(1) command to be run just before exit if it exists. | 
|  | 100 | leaks(1) is available on Mac OS X and presumably on some other | 
|  | 101 | FreeBSD-derived systems. | 
|  | 102 |  | 
| Michael W. Hudson | 61147f6 | 2004-08-03 11:33:28 +0000 | [diff] [blame] | 103 | -R runs each test several times and examines sys.gettotalrefcount() to | 
|  | 104 | see if the test appears to be leaking references.  The argument should | 
|  | 105 | be of the form stab:run:fname where 'stab' is the number of times the | 
|  | 106 | test is run to let gettotalrefcount settle down, 'run' is the number | 
|  | 107 | of times further it is run and 'fname' is the name of the file the | 
|  | 108 | reports are written to.  These parameters all have defaults (5, 4 and | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 109 | "reflog.txt" respectively), and the minimal invocation is '-R :'. | 
| Michael W. Hudson | 61147f6 | 2004-08-03 11:33:28 +0000 | [diff] [blame] | 110 |  | 
| Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 111 | -M runs tests that require an exorbitant amount of memory. These tests | 
|  | 112 | typically try to ascertain containers keep working when containing more than | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 113 | 2 billion objects, which only works on 64-bit systems. There are also some | 
|  | 114 | tests that try to exhaust the address space of the process, which only makes | 
|  | 115 | sense on 32-bit systems with at least 2Gb of memory. The passed-in memlimit, | 
| Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 116 | which is a string in the form of '2.5Gb', determines howmuch memory the | 
|  | 117 | tests will limit themselves to (but they may go slightly over.) The number | 
|  | 118 | shouldn't be more memory than the machine has (including swap memory). You | 
|  | 119 | should also keep in mind that swap memory is generally much, much slower | 
|  | 120 | than RAM, and setting memlimit to all available RAM or higher will heavily | 
|  | 121 | tax the machine. On the other hand, it is no use running these tests with a | 
|  | 122 | limit of less than 2.5Gb, and many require more than 20Gb. Tests that expect | 
|  | 123 | to use more than memlimit memory will be skipped. The big-memory tests | 
|  | 124 | generally run very, very long. | 
|  | 125 |  | 
| Guido van Rossum | 9e9d4f8 | 2002-06-07 15:17:03 +0000 | [diff] [blame] | 126 | -u is used to specify which special resource intensive tests to run, | 
|  | 127 | such as those requiring large file support or network connectivity. | 
|  | 128 | The argument is a comma-separated list of words indicating the | 
|  | 129 | resources to test.  Currently only the following are defined: | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 130 |  | 
| Fred Drake | 3a15dac | 2002-04-11 16:39:16 +0000 | [diff] [blame] | 131 | all -       Enable all special resources. | 
|  | 132 |  | 
| Guido van Rossum | 315aa36 | 2003-03-11 14:46:48 +0000 | [diff] [blame] | 133 | audio -     Tests that use the audio device.  (There are known | 
|  | 134 | cases of broken audio drivers that can crash Python or | 
|  | 135 | even the Linux kernel.) | 
|  | 136 |  | 
| Andrew M. Kuchling | 2158df0 | 2001-10-22 15:26:09 +0000 | [diff] [blame] | 137 | curses -    Tests that use curses and will modify the terminal's | 
|  | 138 | state and output modes. | 
| Tim Peters | 1633a2e | 2001-10-30 05:56:40 +0000 | [diff] [blame] | 139 |  | 
| Guido van Rossum | 9e9d4f8 | 2002-06-07 15:17:03 +0000 | [diff] [blame] | 140 | largefile - It is okay to run some test that may create huge | 
|  | 141 | files.  These tests can take a long time and may | 
|  | 142 | consume >2GB of disk space temporarily. | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 143 |  | 
| Guido van Rossum | 9e9d4f8 | 2002-06-07 15:17:03 +0000 | [diff] [blame] | 144 | network -   It is okay to run tests that use external network | 
|  | 145 | resource, e.g. testing SSL support for sockets. | 
| Martin v. Löwis | 1c6b1a2 | 2002-11-19 17:47:07 +0000 | [diff] [blame] | 146 |  | 
| Raymond Hettinger | 7c85fa4 | 2004-07-01 11:01:35 +0000 | [diff] [blame] | 147 | decimal -   Test the decimal module against a large suite that | 
|  | 148 | verifies compliance with standards. | 
|  | 149 |  | 
| Antoine Pitrou | 5bc4fa7 | 2010-10-14 15:34:31 +0000 | [diff] [blame] | 150 | cpu -       Used for certain CPU-heavy tests. | 
| Jeremy Hylton | 4336eda | 2004-08-07 19:25:33 +0000 | [diff] [blame] | 151 |  | 
| Tim Peters | eba28be | 2005-03-28 01:08:02 +0000 | [diff] [blame] | 152 | subprocess  Run all tests for the subprocess module. | 
| Peter Astrand | f7f1bb7 | 2005-03-03 20:47:37 +0000 | [diff] [blame] | 153 |  | 
| Hye-Shik Chang | aaa2f1d | 2005-12-10 17:44:27 +0000 | [diff] [blame] | 154 | urlfetch -  It is okay to download files required on testing. | 
|  | 155 |  | 
| Guilherme Polo | 9de29af | 2009-01-28 20:40:48 +0000 | [diff] [blame] | 156 | gui -       Run tests that require a running GUI. | 
|  | 157 |  | 
| Fred Drake | 4dd0f7e | 2002-11-26 21:44:56 +0000 | [diff] [blame] | 158 | To enable all resources except one, use '-uall,-<resource>'.  For | 
| Georg Brandl | 1158a33 | 2009-06-04 09:30:30 +0000 | [diff] [blame] | 159 | example, to run all the tests except for the gui tests, give the | 
|  | 160 | option '-uall,-gui'. | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 161 | """ | 
|  | 162 |  | 
| Brett Cannon | 4507190 | 2010-06-14 22:22:54 +0000 | [diff] [blame] | 163 | import builtins | 
| Victor Stinner | 024e37a | 2011-03-31 01:31:06 +0200 | [diff] [blame] | 164 | import faulthandler | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 165 | import getopt | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 166 | import json | 
| Christian Heimes | b186d00 | 2008-03-18 15:15:01 +0000 | [diff] [blame] | 167 | import os | 
| Skip Montanaro | ab1c791 | 2000-06-30 16:39:27 +0000 | [diff] [blame] | 168 | import random | 
| Thomas Wouters | 9ada3d6 | 2006-04-21 09:47:09 +0000 | [diff] [blame] | 169 | import re | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 170 | import io | 
| Christian Heimes | b186d00 | 2008-03-18 15:15:01 +0000 | [diff] [blame] | 171 | import sys | 
|  | 172 | import time | 
| Antoine Pitrou | ee42934 | 2011-04-16 18:53:59 +0200 | [diff] [blame] | 173 | import errno | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 174 | import traceback | 
| Christian Heimes | b186d00 | 2008-03-18 15:15:01 +0000 | [diff] [blame] | 175 | import warnings | 
| Benjamin Peterson | e549ead | 2009-03-28 21:42:05 +0000 | [diff] [blame] | 176 | import unittest | 
| Christian Heimes | be5b30b | 2008-03-03 19:18:51 +0000 | [diff] [blame] | 177 | from inspect import isabstract | 
| Ezio Melotti | 184bdfb | 2010-02-18 09:37:05 +0000 | [diff] [blame] | 178 | import tempfile | 
| Florent Xicluna | da7bfd5 | 2010-03-06 11:43:55 +0000 | [diff] [blame] | 179 | import platform | 
|  | 180 | import sysconfig | 
| Nick Coghlan | 7d81975 | 2010-12-05 06:45:03 +0000 | [diff] [blame] | 181 | import logging | 
| Ezio Melotti | 184bdfb | 2010-02-18 09:37:05 +0000 | [diff] [blame] | 182 |  | 
| Florent Xicluna | 64fb18e | 2010-03-06 14:43:34 +0000 | [diff] [blame] | 183 |  | 
| Ezio Melotti | 184bdfb | 2010-02-18 09:37:05 +0000 | [diff] [blame] | 184 | # Some times __path__ and __file__ are not absolute (e.g. while running from | 
|  | 185 | # Lib/) and, if we change the CWD to run the tests in a temporary dir, some | 
|  | 186 | # imports might fail.  This affects only the modules imported before os.chdir(). | 
|  | 187 | # These modules are searched first in sys.path[0] (so '' -- the CWD) and if | 
|  | 188 | # they are found in the CWD their __file__ and __path__ will be relative (this | 
|  | 189 | # happens before the chdir).  All the modules imported after the chdir, are | 
|  | 190 | # not found in the CWD, and since the other paths in sys.path[1:] are absolute | 
|  | 191 | # (site.py absolutize them), the __file__ and __path__ will be absolute too. | 
|  | 192 | # Therefore it is necessary to absolutize manually the __file__ and __path__ of | 
|  | 193 | # the packages to prevent later imports to fail when the CWD is different. | 
|  | 194 | for module in sys.modules.values(): | 
|  | 195 | if hasattr(module, '__path__'): | 
|  | 196 | module.__path__ = [os.path.abspath(path) for path in module.__path__] | 
|  | 197 | if hasattr(module, '__file__'): | 
|  | 198 | module.__file__ = os.path.abspath(module.__file__) | 
|  | 199 |  | 
| Guido van Rossum | dc15c27 | 2002-08-12 21:55:51 +0000 | [diff] [blame] | 200 |  | 
| Guido van Rossum | bb48465 | 2002-12-02 09:56:21 +0000 | [diff] [blame] | 201 | # MacOSX (a.k.a. Darwin) has a default stack size that is too small | 
|  | 202 | # for deeply recursive regular expressions.  We see this as crashes in | 
|  | 203 | # the Python test suite when running test_re.py and test_sre.py.  The | 
|  | 204 | # fix is to set the stack limit to 2048. | 
|  | 205 | # This approach may also be useful for other Unixy platforms that | 
|  | 206 | # suffer from small default stack limits. | 
|  | 207 | if sys.platform == 'darwin': | 
|  | 208 | try: | 
|  | 209 | import resource | 
|  | 210 | except ImportError: | 
|  | 211 | pass | 
|  | 212 | else: | 
|  | 213 | soft, hard = resource.getrlimit(resource.RLIMIT_STACK) | 
|  | 214 | newsoft = min(hard, max(soft, 1024*2048)) | 
|  | 215 | resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard)) | 
|  | 216 |  | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 217 | # Test result constants. | 
|  | 218 | PASSED = 1 | 
|  | 219 | FAILED = 0 | 
|  | 220 | ENV_CHANGED = -1 | 
|  | 221 | SKIPPED = -2 | 
|  | 222 | RESOURCE_DENIED = -3 | 
|  | 223 | INTERRUPTED = -4 | 
| Victor Stinner | 4b73988 | 2011-03-31 18:02:36 +0200 | [diff] [blame] | 224 | CHILD_ERROR = -5   # error in a child process | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 225 |  | 
| Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 226 | from test import support | 
| Fred Drake | 3a15dac | 2002-04-11 16:39:16 +0000 | [diff] [blame] | 227 |  | 
| Georg Brandl | 1158a33 | 2009-06-04 09:30:30 +0000 | [diff] [blame] | 228 | RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network', | 
| Antoine Pitrou | 5bc4fa7 | 2010-10-14 15:34:31 +0000 | [diff] [blame] | 229 | 'decimal', 'cpu', 'subprocess', 'urlfetch', 'gui') | 
| Fred Drake | 3a15dac | 2002-04-11 16:39:16 +0000 | [diff] [blame] | 230 |  | 
| Florent Xicluna | ec88221 | 2010-08-09 16:56:43 +0000 | [diff] [blame] | 231 | TEMPDIR = os.path.abspath(tempfile.gettempdir()) | 
| Fred Drake | 3a15dac | 2002-04-11 16:39:16 +0000 | [diff] [blame] | 232 |  | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 233 | def usage(msg): | 
|  | 234 | print(msg, file=sys.stderr) | 
|  | 235 | print("Use --help for usage", file=sys.stderr) | 
|  | 236 | sys.exit(2) | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 237 |  | 
|  | 238 |  | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 239 | def main(tests=None, testdir=None, verbose=0, quiet=False, | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 240 | exclude=False, single=False, randomize=False, fromfile=None, | 
| Walter Dörwald | aee4da6 | 2004-11-12 18:51:27 +0000 | [diff] [blame] | 241 | findleaks=False, use_resources=None, trace=False, coverdir='coverage', | 
| Collin Winter | fd12f49 | 2009-03-29 04:05:05 +0000 | [diff] [blame] | 242 | runleaks=False, huntrleaks=False, verbose2=False, print_slow=False, | 
| Antoine Pitrou | 3c4402f | 2011-01-03 20:38:52 +0000 | [diff] [blame] | 243 | random_seed=None, use_mp=None, verbose3=False, forever=False, | 
| Victor Stinner | 9bf133c | 2011-04-13 01:06:27 +0200 | [diff] [blame] | 244 | header=False): | 
| Guido van Rossum | 6fd83b7 | 1998-08-01 17:04:08 +0000 | [diff] [blame] | 245 | """Execute a test suite. | 
|  | 246 |  | 
| Thomas Wouters | 7e47402 | 2000-07-16 12:04:32 +0000 | [diff] [blame] | 247 | This also parses command-line options and modifies its behavior | 
| Fred Drake | 004d5e6 | 2000-10-23 17:22:08 +0000 | [diff] [blame] | 248 | accordingly. | 
| Guido van Rossum | 6fd83b7 | 1998-08-01 17:04:08 +0000 | [diff] [blame] | 249 |  | 
|  | 250 | tests -- a list of strings containing test names (optional) | 
|  | 251 | testdir -- the directory in which to look for tests (optional) | 
|  | 252 |  | 
|  | 253 | Users other than the Python test suite will certainly want to | 
|  | 254 | specify testdir; if it's omitted, the directory containing the | 
| Fred Drake | 004d5e6 | 2000-10-23 17:22:08 +0000 | [diff] [blame] | 255 | Python test suite is searched for. | 
| Guido van Rossum | 6fd83b7 | 1998-08-01 17:04:08 +0000 | [diff] [blame] | 256 |  | 
|  | 257 | If the tests argument is omitted, the tests listed on the | 
|  | 258 | command-line will be used.  If that's empty, too, then all *.py | 
|  | 259 | files beginning with test_ will be used. | 
| Skip Montanaro | ab1c791 | 2000-06-30 16:39:27 +0000 | [diff] [blame] | 260 |  | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 261 | The other default arguments (verbose, quiet, exclude, | 
| Collin Winter | fd12f49 | 2009-03-29 04:05:05 +0000 | [diff] [blame] | 262 | single, randomize, findleaks, use_resources, trace, coverdir, | 
|  | 263 | print_slow, and random_seed) allow programmers calling main() | 
|  | 264 | directly to set the values that would normally be set by flags | 
|  | 265 | on the command line. | 
| Guido van Rossum | 6fd83b7 | 1998-08-01 17:04:08 +0000 | [diff] [blame] | 266 | """ | 
| Victor Stinner | 9bf133c | 2011-04-13 01:06:27 +0200 | [diff] [blame] | 267 | if hasattr(faulthandler, 'dump_tracebacks_later'): | 
|  | 268 | timeout = 60*60 | 
|  | 269 | else: | 
|  | 270 | timeout = None | 
| Fred Drake | 004d5e6 | 2000-10-23 17:22:08 +0000 | [diff] [blame] | 271 |  | 
| Victor Stinner | 1802d3f | 2010-05-19 17:11:19 +0000 | [diff] [blame] | 272 | replace_stdout() | 
|  | 273 |  | 
| Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 274 | support.record_original_stdout(sys.stdout) | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 275 | try: | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 276 | opts, args = getopt.getopt(sys.argv[1:], 'hvqxsSrf:lu:t:TD:NLR:FwWM:nj:', | 
| Georg Brandl | 96f7897 | 2009-10-27 21:02:11 +0000 | [diff] [blame] | 277 | ['help', 'verbose', 'verbose2', 'verbose3', 'quiet', | 
|  | 278 | 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks', | 
|  | 279 | 'use=', 'threshold=', 'trace', 'coverdir=', 'nocoverdir', | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 280 | 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=', | 
| Brett Cannon | 63eef1e | 2011-01-06 22:32:41 +0000 | [diff] [blame] | 281 | 'multiprocess=', 'coverage', 'slaveargs=', 'forever', 'debug', | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 282 | 'start=', 'nowindows', 'header', 'testdir=', 'timeout=']) | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 283 | except getopt.error as msg: | 
|  | 284 | usage(msg) | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 285 |  | 
|  | 286 | # Defaults | 
| Collin Winter | fd12f49 | 2009-03-29 04:05:05 +0000 | [diff] [blame] | 287 | if random_seed is None: | 
|  | 288 | random_seed = random.randrange(10000000) | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 289 | if use_resources is None: | 
|  | 290 | use_resources = [] | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 291 | debug = False | 
|  | 292 | start = None | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 293 | for o, a in opts: | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 294 | if o in ('-h', '--help'): | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 295 | print(__doc__) | 
|  | 296 | return | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 297 | elif o in ('-v', '--verbose'): | 
|  | 298 | verbose += 1 | 
| Martin v. Löwis | 04824ce | 2006-03-10 21:26:16 +0000 | [diff] [blame] | 299 | elif o in ('-w', '--verbose2'): | 
|  | 300 | verbose2 = True | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 301 | elif o in ('-d', '--debug'): | 
|  | 302 | debug = True | 
| Georg Brandl | 96f7897 | 2009-10-27 21:02:11 +0000 | [diff] [blame] | 303 | elif o in ('-W', '--verbose3'): | 
|  | 304 | verbose3 = True | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 305 | elif o in ('-q', '--quiet'): | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 306 | quiet = True; | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 307 | verbose = 0 | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 308 | elif o in ('-x', '--exclude'): | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 309 | exclude = True | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 310 | elif o in ('-S', '--start'): | 
|  | 311 | start = a | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 312 | elif o in ('-s', '--single'): | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 313 | single = True | 
| Christian Heimes | b186d00 | 2008-03-18 15:15:01 +0000 | [diff] [blame] | 314 | elif o in ('-S', '--slow'): | 
|  | 315 | print_slow = True | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 316 | elif o in ('-r', '--randomize'): | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 317 | randomize = True | 
| Collin Winter | fd12f49 | 2009-03-29 04:05:05 +0000 | [diff] [blame] | 318 | elif o == '--randseed': | 
|  | 319 | random_seed = int(a) | 
| Tim Peters | c5000df | 2002-06-02 21:42:01 +0000 | [diff] [blame] | 320 | elif o in ('-f', '--fromfile'): | 
|  | 321 | fromfile = a | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 322 | elif o in ('-l', '--findleaks'): | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 323 | findleaks = True | 
| Skip Montanaro | 0179a18 | 2004-06-06 15:53:18 +0000 | [diff] [blame] | 324 | elif o in ('-L', '--runleaks'): | 
|  | 325 | runleaks = True | 
| Guido van Rossum | 9e9d4f8 | 2002-06-07 15:17:03 +0000 | [diff] [blame] | 326 | elif o in ('-t', '--threshold'): | 
|  | 327 | import gc | 
|  | 328 | gc.set_threshold(int(a)) | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 329 | elif o in ('-T', '--coverage'): | 
|  | 330 | trace = True | 
| Walter Dörwald | aee4da6 | 2004-11-12 18:51:27 +0000 | [diff] [blame] | 331 | elif o in ('-D', '--coverdir'): | 
| R David Murray | 03504fc | 2011-03-24 14:35:30 -0400 | [diff] [blame] | 332 | # CWD is replaced with a temporary dir before calling main(), so we | 
|  | 333 | # need  join it with the saved CWD so it goes where the user expects. | 
|  | 334 | coverdir = os.path.join(support.SAVEDCWD, a) | 
| Walter Dörwald | aee4da6 | 2004-11-12 18:51:27 +0000 | [diff] [blame] | 335 | elif o in ('-N', '--nocoverdir'): | 
|  | 336 | coverdir = None | 
| Michael W. Hudson | 61147f6 | 2004-08-03 11:33:28 +0000 | [diff] [blame] | 337 | elif o in ('-R', '--huntrleaks'): | 
|  | 338 | huntrleaks = a.split(':') | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 339 | if len(huntrleaks) not in (2, 3): | 
|  | 340 | print(a, huntrleaks) | 
|  | 341 | usage('-R takes 2 or 3 colon-separated arguments') | 
|  | 342 | if not huntrleaks[0]: | 
| Michael W. Hudson | 61147f6 | 2004-08-03 11:33:28 +0000 | [diff] [blame] | 343 | huntrleaks[0] = 5 | 
|  | 344 | else: | 
|  | 345 | huntrleaks[0] = int(huntrleaks[0]) | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 346 | if not huntrleaks[1]: | 
| Michael W. Hudson | 61147f6 | 2004-08-03 11:33:28 +0000 | [diff] [blame] | 347 | huntrleaks[1] = 4 | 
|  | 348 | else: | 
|  | 349 | huntrleaks[1] = int(huntrleaks[1]) | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 350 | if len(huntrleaks) == 2 or not huntrleaks[2]: | 
|  | 351 | huntrleaks[2:] = ["reflog.txt"] | 
|  | 352 | # Avoid false positives due to the character cache in | 
|  | 353 | # stringobject.c filling slowly with random data | 
|  | 354 | warm_char_cache() | 
| Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 355 | elif o in ('-M', '--memlimit'): | 
| Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 356 | support.set_memlimit(a) | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 357 | elif o in ('-u', '--use'): | 
| Guido van Rossum | fe3f696 | 2001-09-06 16:09:41 +0000 | [diff] [blame] | 358 | u = [x.lower() for x in a.split(',')] | 
|  | 359 | for r in u: | 
| Fred Drake | 3a15dac | 2002-04-11 16:39:16 +0000 | [diff] [blame] | 360 | if r == 'all': | 
| Fred Drake | 4dd0f7e | 2002-11-26 21:44:56 +0000 | [diff] [blame] | 361 | use_resources[:] = RESOURCE_NAMES | 
|  | 362 | continue | 
|  | 363 | remove = False | 
|  | 364 | if r[0] == '-': | 
|  | 365 | remove = True | 
|  | 366 | r = r[1:] | 
| Fred Drake | 3a15dac | 2002-04-11 16:39:16 +0000 | [diff] [blame] | 367 | if r not in RESOURCE_NAMES: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 368 | usage('Invalid -u/--use option: ' + a) | 
| Fred Drake | 4dd0f7e | 2002-11-26 21:44:56 +0000 | [diff] [blame] | 369 | if remove: | 
|  | 370 | if r in use_resources: | 
|  | 371 | use_resources.remove(r) | 
|  | 372 | elif r not in use_resources: | 
| Andrew MacIntyre | e41abab | 2002-04-30 12:11:04 +0000 | [diff] [blame] | 373 | use_resources.append(r) | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 374 | elif o in ('-n', '--nowindows'): | 
|  | 375 | import msvcrt | 
|  | 376 | msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS| | 
|  | 377 | msvcrt.SEM_NOALIGNMENTFAULTEXCEPT| | 
|  | 378 | msvcrt.SEM_NOGPFAULTERRORBOX| | 
|  | 379 | msvcrt.SEM_NOOPENFILEERRORBOX) | 
|  | 380 | try: | 
|  | 381 | msvcrt.CrtSetReportMode | 
|  | 382 | except AttributeError: | 
|  | 383 | # release build | 
|  | 384 | pass | 
|  | 385 | else: | 
|  | 386 | for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]: | 
|  | 387 | msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE) | 
|  | 388 | msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR) | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 389 | elif o in ('-F', '--forever'): | 
|  | 390 | forever = True | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 391 | elif o in ('-j', '--multiprocess'): | 
|  | 392 | use_mp = int(a) | 
| Antoine Pitrou | 6ab79d9 | 2011-03-23 20:17:45 +0100 | [diff] [blame] | 393 | if use_mp <= 0: | 
|  | 394 | try: | 
|  | 395 | import multiprocessing | 
|  | 396 | # Use all cores + extras for tests that like to sleep | 
|  | 397 | use_mp = 2 + multiprocessing.cpu_count() | 
|  | 398 | except (ImportError, NotImplementedError): | 
|  | 399 | use_mp = 3 | 
| Antoine Pitrou | 3c4402f | 2011-01-03 20:38:52 +0000 | [diff] [blame] | 400 | elif o == '--header': | 
|  | 401 | header = True | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 402 | elif o == '--slaveargs': | 
|  | 403 | args, kwargs = json.loads(a) | 
|  | 404 | try: | 
|  | 405 | result = runtest(*args, **kwargs) | 
|  | 406 | except BaseException as e: | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 407 | result = INTERRUPTED, e.__class__.__name__ | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 408 | sys.stdout.flush() | 
|  | 409 | print()   # Force a newline (just in case) | 
|  | 410 | print(json.dumps(result)) | 
|  | 411 | sys.exit(0) | 
| R David Murray | b588f8d | 2011-03-24 14:42:58 -0400 | [diff] [blame] | 412 | elif o == '--testdir': | 
|  | 413 | # CWD is replaced with a temporary dir before calling main(), so we | 
|  | 414 | # join it with the saved CWD so it ends up where the user expects. | 
|  | 415 | testdir = os.path.join(support.SAVEDCWD, a) | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 416 | elif o == '--timeout': | 
| Victor Stinner | 9bf133c | 2011-04-13 01:06:27 +0200 | [diff] [blame] | 417 | if not hasattr(faulthandler, 'dump_tracebacks_later'): | 
|  | 418 | print("--timeout option requires " | 
|  | 419 | "faulthandler.dump_tracebacks_later", file=sys.stderr) | 
|  | 420 | sys.exit(1) | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 421 | timeout = float(a) | 
| R. David Murray | 35768ad | 2009-11-15 00:23:21 +0000 | [diff] [blame] | 422 | else: | 
|  | 423 | print(("No handler for option {}.  Please report this as a bug " | 
| Georg Brandl | dee7b85 | 2010-08-02 18:59:52 +0000 | [diff] [blame] | 424 | "at http://bugs.python.org.").format(o), file=sys.stderr) | 
| R. David Murray | 35768ad | 2009-11-15 00:23:21 +0000 | [diff] [blame] | 425 | sys.exit(1) | 
| Tim Peters | c5000df | 2002-06-02 21:42:01 +0000 | [diff] [blame] | 426 | if single and fromfile: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 427 | usage("-s and -f don't go together!") | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 428 | if use_mp and trace: | 
| Florent Xicluna | 64fb18e | 2010-03-06 14:43:34 +0000 | [diff] [blame] | 429 | usage("-T and -j don't go together!") | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 430 | if use_mp and findleaks: | 
| Florent Xicluna | 64fb18e | 2010-03-06 14:43:34 +0000 | [diff] [blame] | 431 | usage("-l and -j don't go together!") | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 432 |  | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 433 | good = [] | 
|  | 434 | bad = [] | 
|  | 435 | skipped = [] | 
| Fred Drake | 9a0db07 | 2003-02-03 15:19:30 +0000 | [diff] [blame] | 436 | resource_denieds = [] | 
| Nick Coghlan | 6ead552 | 2009-10-18 13:19:33 +0000 | [diff] [blame] | 437 | environment_changed = [] | 
| Florent Xicluna | d6995eb | 2010-03-30 19:43:09 +0000 | [diff] [blame] | 438 | interrupted = False | 
| Barry Warsaw | e11e3de | 1999-01-28 19:51:51 +0000 | [diff] [blame] | 439 |  | 
| Neil Schemenauer | d569f23 | 2000-09-22 15:29:28 +0000 | [diff] [blame] | 440 | if findleaks: | 
| Barry Warsaw | a873b03 | 2000-08-03 15:50:37 +0000 | [diff] [blame] | 441 | try: | 
|  | 442 | import gc | 
|  | 443 | except ImportError: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 444 | print('No GC available, disabling findleaks.') | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 445 | findleaks = False | 
| Barry Warsaw | a873b03 | 2000-08-03 15:50:37 +0000 | [diff] [blame] | 446 | else: | 
| Neil Schemenauer | 8a00abc | 2000-10-13 01:32:42 +0000 | [diff] [blame] | 447 | # Uncomment the line below to report garbage that is not | 
|  | 448 | # freeable by reference counting alone.  By default only | 
|  | 449 | # garbage that is not collectable by the GC is reported. | 
|  | 450 | #gc.set_debug(gc.DEBUG_SAVEALL) | 
| Neil Schemenauer | d569f23 | 2000-09-22 15:29:28 +0000 | [diff] [blame] | 451 | found_garbage = [] | 
| Barry Warsaw | a873b03 | 2000-08-03 15:50:37 +0000 | [diff] [blame] | 452 |  | 
| Barry Warsaw | e11e3de | 1999-01-28 19:51:51 +0000 | [diff] [blame] | 453 | if single: | 
| Florent Xicluna | ec88221 | 2010-08-09 16:56:43 +0000 | [diff] [blame] | 454 | filename = os.path.join(TEMPDIR, 'pynexttest') | 
| Barry Warsaw | e11e3de | 1999-01-28 19:51:51 +0000 | [diff] [blame] | 455 | try: | 
|  | 456 | fp = open(filename, 'r') | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 457 | next_test = fp.read().strip() | 
|  | 458 | tests = [next_test] | 
| Barry Warsaw | e11e3de | 1999-01-28 19:51:51 +0000 | [diff] [blame] | 459 | fp.close() | 
|  | 460 | except IOError: | 
|  | 461 | pass | 
| Tim Peters | c5000df | 2002-06-02 21:42:01 +0000 | [diff] [blame] | 462 |  | 
|  | 463 | if fromfile: | 
|  | 464 | tests = [] | 
| Ezio Melotti | 184bdfb | 2010-02-18 09:37:05 +0000 | [diff] [blame] | 465 | fp = open(os.path.join(support.SAVEDCWD, fromfile)) | 
| Antoine Pitrou | d95c7b5 | 2010-11-07 20:50:51 +0000 | [diff] [blame] | 466 | count_pat = re.compile(r'\[\s*\d+/\s*\d+\]') | 
| Tim Peters | c5000df | 2002-06-02 21:42:01 +0000 | [diff] [blame] | 467 | for line in fp: | 
| Antoine Pitrou | d95c7b5 | 2010-11-07 20:50:51 +0000 | [diff] [blame] | 468 | line = count_pat.sub('', line) | 
| Tim Peters | c5000df | 2002-06-02 21:42:01 +0000 | [diff] [blame] | 469 | guts = line.split() # assuming no test has whitespace in its name | 
|  | 470 | if guts and not guts[0].startswith('#'): | 
|  | 471 | tests.extend(guts) | 
|  | 472 | fp.close() | 
|  | 473 |  | 
|  | 474 | # Strip .py extensions. | 
| Florent Xicluna | da7bfd5 | 2010-03-06 11:43:55 +0000 | [diff] [blame] | 475 | removepy(args) | 
|  | 476 | removepy(tests) | 
| Tim Peters | c5000df | 2002-06-02 21:42:01 +0000 | [diff] [blame] | 477 |  | 
| Guido van Rossum | 6c74fea | 1998-08-25 12:29:08 +0000 | [diff] [blame] | 478 | stdtests = STDTESTS[:] | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 479 | nottests = NOTTESTS.copy() | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 480 | if exclude: | 
| Guido van Rossum | 6c74fea | 1998-08-25 12:29:08 +0000 | [diff] [blame] | 481 | for arg in args: | 
|  | 482 | if arg in stdtests: | 
|  | 483 | stdtests.remove(arg) | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 484 | nottests.add(arg) | 
| Guido van Rossum | 41360a4 | 1998-03-26 19:42:58 +0000 | [diff] [blame] | 485 | args = [] | 
| Florent Xicluna | 0e62a14 | 2010-03-06 17:34:48 +0000 | [diff] [blame] | 486 |  | 
|  | 487 | # For a partial run, we do not need to clutter the output. | 
| Antoine Pitrou | 3c4402f | 2011-01-03 20:38:52 +0000 | [diff] [blame] | 488 | if verbose or header or not (quiet or single or tests or args): | 
| Florent Xicluna | 0e62a14 | 2010-03-06 17:34:48 +0000 | [diff] [blame] | 489 | # Print basic platform information | 
|  | 490 | print("==", platform.python_implementation(), *sys.version.split()) | 
| Florent Xicluna | ec88221 | 2010-08-09 16:56:43 +0000 | [diff] [blame] | 491 | print("==  ", platform.platform(aliased=True), | 
|  | 492 | "%s-endian" % sys.byteorder) | 
| Florent Xicluna | 0e62a14 | 2010-03-06 17:34:48 +0000 | [diff] [blame] | 493 | print("==  ", os.getcwd()) | 
| Antoine Pitrou | 3c4402f | 2011-01-03 20:38:52 +0000 | [diff] [blame] | 494 | print("Testing with flags:", sys.flags) | 
| Florent Xicluna | 0e62a14 | 2010-03-06 17:34:48 +0000 | [diff] [blame] | 495 |  | 
| R David Murray | b588f8d | 2011-03-24 14:42:58 -0400 | [diff] [blame] | 496 | # if testdir is set, then we are not running the python tests suite, so | 
|  | 497 | # don't add default tests to be executed or skipped (pass empty values) | 
|  | 498 | if testdir: | 
|  | 499 | alltests = findtests(testdir, list(), set()) | 
|  | 500 | else: | 
|  | 501 | alltests = findtests(testdir, stdtests, nottests) | 
|  | 502 |  | 
| Florent Xicluna | d6995eb | 2010-03-30 19:43:09 +0000 | [diff] [blame] | 503 | selected = tests or args or alltests | 
| Barry Warsaw | e11e3de | 1999-01-28 19:51:51 +0000 | [diff] [blame] | 504 | if single: | 
| Florent Xicluna | d6995eb | 2010-03-30 19:43:09 +0000 | [diff] [blame] | 505 | selected = selected[:1] | 
| R. David Murray | ef1992b | 2009-12-16 15:19:27 +0000 | [diff] [blame] | 506 | try: | 
| Florent Xicluna | d6995eb | 2010-03-30 19:43:09 +0000 | [diff] [blame] | 507 | next_single_test = alltests[alltests.index(selected[0])+1] | 
| R. David Murray | ef1992b | 2009-12-16 15:19:27 +0000 | [diff] [blame] | 508 | except IndexError: | 
|  | 509 | next_single_test = None | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 510 | # Remove all the tests that precede start if it's set. | 
|  | 511 | if start: | 
|  | 512 | try: | 
|  | 513 | del tests[:tests.index(start)] | 
|  | 514 | except ValueError: | 
|  | 515 | print("Couldn't find starting test (%s), using all tests" % start) | 
| Skip Montanaro | ab1c791 | 2000-06-30 16:39:27 +0000 | [diff] [blame] | 516 | if randomize: | 
| Collin Winter | fd12f49 | 2009-03-29 04:05:05 +0000 | [diff] [blame] | 517 | random.seed(random_seed) | 
|  | 518 | print("Using random seed", random_seed) | 
| Florent Xicluna | d6995eb | 2010-03-30 19:43:09 +0000 | [diff] [blame] | 519 | random.shuffle(selected) | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 520 | if trace: | 
| Georg Brandl | 33c2881 | 2009-04-01 23:07:29 +0000 | [diff] [blame] | 521 | import trace, tempfile | 
|  | 522 | tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix, | 
|  | 523 | tempfile.gettempdir()], | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 524 | trace=False, count=True) | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 525 |  | 
| Christian Heimes | b186d00 | 2008-03-18 15:15:01 +0000 | [diff] [blame] | 526 | test_times = [] | 
| Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 527 | support.verbose = verbose      # Tell tests to be moderately quiet | 
|  | 528 | support.use_resources = use_resources | 
| Guido van Rossum | 5796d26 | 2000-04-21 21:35:06 +0000 | [diff] [blame] | 529 | save_modules = sys.modules.keys() | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 530 |  | 
|  | 531 | def accumulate_result(test, result): | 
|  | 532 | ok, test_time = result | 
|  | 533 | test_times.append((test_time, test)) | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 534 | if ok == PASSED: | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 535 | good.append(test) | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 536 | elif ok == FAILED: | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 537 | bad.append(test) | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 538 | elif ok == ENV_CHANGED: | 
|  | 539 | bad.append(test) | 
|  | 540 | environment_changed.append(test) | 
|  | 541 | elif ok == SKIPPED: | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 542 | skipped.append(test) | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 543 | elif ok == RESOURCE_DENIED: | 
|  | 544 | skipped.append(test) | 
|  | 545 | resource_denieds.append(test) | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 546 |  | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 547 | if forever: | 
| Florent Xicluna | d6995eb | 2010-03-30 19:43:09 +0000 | [diff] [blame] | 548 | def test_forever(tests=list(selected)): | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 549 | while True: | 
|  | 550 | for test in tests: | 
|  | 551 | yield test | 
|  | 552 | if bad: | 
|  | 553 | return | 
|  | 554 | tests = test_forever() | 
| Georg Brandl | e8e02e3 | 2010-08-03 07:56:50 +0000 | [diff] [blame] | 555 | test_count = '' | 
|  | 556 | test_count_width = 3 | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 557 | else: | 
| Florent Xicluna | d6995eb | 2010-03-30 19:43:09 +0000 | [diff] [blame] | 558 | tests = iter(selected) | 
| Georg Brandl | e8e02e3 | 2010-08-03 07:56:50 +0000 | [diff] [blame] | 559 | test_count = '/{}'.format(len(selected)) | 
|  | 560 | test_count_width = len(test_count) - 1 | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 561 |  | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 562 | if use_mp: | 
| Victor Stinner | 45df820 | 2010-04-28 22:31:17 +0000 | [diff] [blame] | 563 | try: | 
|  | 564 | from threading import Thread | 
|  | 565 | except ImportError: | 
|  | 566 | print("Multiprocess option requires thread support") | 
|  | 567 | sys.exit(2) | 
| Georg Brandl | 1b37e87 | 2010-03-14 10:45:50 +0000 | [diff] [blame] | 568 | from queue import Queue | 
|  | 569 | from subprocess import Popen, PIPE | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 570 | debug_output_pat = re.compile(r"\[\d+ refs\]$") | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 571 | output = Queue() | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 572 | def tests_and_args(): | 
|  | 573 | for test in tests: | 
|  | 574 | args_tuple = ( | 
| R. David Murray | 8e286c4 | 2010-12-27 20:09:32 +0000 | [diff] [blame] | 575 | (test, verbose, quiet), | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 576 | dict(huntrleaks=huntrleaks, use_resources=use_resources, | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 577 | debug=debug, rerun_failed=verbose3, timeout=timeout) | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 578 | ) | 
|  | 579 | yield (test, args_tuple) | 
|  | 580 | pending = tests_and_args() | 
| Antoine Pitrou | 1b03f2c | 2010-10-14 11:12:00 +0000 | [diff] [blame] | 581 | opt_args = support.args_from_interpreter_flags() | 
|  | 582 | base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest'] | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 583 | def work(): | 
|  | 584 | # A worker thread. | 
| Neal Norwitz | 14ca327 | 2006-02-28 18:05:43 +0000 | [diff] [blame] | 585 | try: | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 586 | while True: | 
|  | 587 | try: | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 588 | test, args_tuple = next(pending) | 
|  | 589 | except StopIteration: | 
| R. David Murray | 2714460 | 2009-10-19 15:26:16 +0000 | [diff] [blame] | 590 | output.put((None, None, None, None)) | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 591 | return | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 592 | # -E is needed by some tests, e.g. test_import | 
| Antoine Pitrou | 1b03f2c | 2010-10-14 11:12:00 +0000 | [diff] [blame] | 593 | popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)], | 
| R. David Murray | 2714460 | 2009-10-19 15:26:16 +0000 | [diff] [blame] | 594 | stdout=PIPE, stderr=PIPE, | 
| Antoine Pitrou | 36e778e | 2010-08-18 20:44:14 +0000 | [diff] [blame] | 595 | universal_newlines=True, | 
|  | 596 | close_fds=(os.name != 'nt')) | 
| R. David Murray | 2714460 | 2009-10-19 15:26:16 +0000 | [diff] [blame] | 597 | stdout, stderr = popen.communicate() | 
| Victor Stinner | 4b73988 | 2011-03-31 18:02:36 +0200 | [diff] [blame] | 598 | retcode = popen.wait() | 
| R. David Murray | 2714460 | 2009-10-19 15:26:16 +0000 | [diff] [blame] | 599 | # Strip last refcount output line if it exists, since it | 
|  | 600 | # comes from the shutdown of the interpreter in the subcommand. | 
|  | 601 | stderr = debug_output_pat.sub("", stderr) | 
|  | 602 | stdout, _, result = stdout.strip().rpartition("\n") | 
| Victor Stinner | 4b73988 | 2011-03-31 18:02:36 +0200 | [diff] [blame] | 603 | if retcode != 0: | 
|  | 604 | result = (CHILD_ERROR, "Exit code %s" % retcode) | 
|  | 605 | output.put((test, stdout.rstrip(), stderr.rstrip(), result)) | 
|  | 606 | return | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 607 | if not result: | 
|  | 608 | output.put((None, None, None, None)) | 
|  | 609 | return | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 610 | result = json.loads(result) | 
| R. David Murray | 2714460 | 2009-10-19 15:26:16 +0000 | [diff] [blame] | 611 | output.put((test, stdout.rstrip(), stderr.rstrip(), result)) | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 612 | except BaseException: | 
| R. David Murray | 2714460 | 2009-10-19 15:26:16 +0000 | [diff] [blame] | 613 | output.put((None, None, None, None)) | 
| Neal Norwitz | 14ca327 | 2006-02-28 18:05:43 +0000 | [diff] [blame] | 614 | raise | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 615 | workers = [Thread(target=work) for i in range(use_mp)] | 
|  | 616 | for worker in workers: | 
|  | 617 | worker.start() | 
|  | 618 | finished = 0 | 
| Georg Brandl | dee7b85 | 2010-08-02 18:59:52 +0000 | [diff] [blame] | 619 | test_index = 1 | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 620 | try: | 
|  | 621 | while finished < use_mp: | 
|  | 622 | test, stdout, stderr, result = output.get() | 
|  | 623 | if test is None: | 
|  | 624 | finished += 1 | 
|  | 625 | continue | 
| Georg Brandl | dee7b85 | 2010-08-02 18:59:52 +0000 | [diff] [blame] | 626 | if not quiet: | 
| Georg Brandl | e8e02e3 | 2010-08-03 07:56:50 +0000 | [diff] [blame] | 627 | print("[{1:{0}}{2}] {3}".format( | 
| Georg Brandl | dee7b85 | 2010-08-02 18:59:52 +0000 | [diff] [blame] | 628 | test_count_width, test_index, test_count, test)) | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 629 | if stdout: | 
|  | 630 | print(stdout) | 
|  | 631 | if stderr: | 
|  | 632 | print(stderr, file=sys.stderr) | 
|  | 633 | if result[0] == INTERRUPTED: | 
|  | 634 | assert result[1] == 'KeyboardInterrupt' | 
|  | 635 | raise KeyboardInterrupt   # What else? | 
| Victor Stinner | 4b73988 | 2011-03-31 18:02:36 +0200 | [diff] [blame] | 636 | if result[0] == CHILD_ERROR: | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 637 | raise Exception("Child error: {}".format(result[1])) | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 638 | accumulate_result(test, result) | 
| Georg Brandl | dee7b85 | 2010-08-02 18:59:52 +0000 | [diff] [blame] | 639 | test_index += 1 | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 640 | except KeyboardInterrupt: | 
| Florent Xicluna | d6995eb | 2010-03-30 19:43:09 +0000 | [diff] [blame] | 641 | interrupted = True | 
| R. David Murray | 7dc72cc | 2009-11-14 16:13:02 +0000 | [diff] [blame] | 642 | pending.close() | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 643 | for worker in workers: | 
|  | 644 | worker.join() | 
|  | 645 | else: | 
| Georg Brandl | dee7b85 | 2010-08-02 18:59:52 +0000 | [diff] [blame] | 646 | for test_index, test in enumerate(tests, 1): | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 647 | if not quiet: | 
| Georg Brandl | e8e02e3 | 2010-08-03 07:56:50 +0000 | [diff] [blame] | 648 | print("[{1:{0}}{2}] {3}".format( | 
| Georg Brandl | dee7b85 | 2010-08-02 18:59:52 +0000 | [diff] [blame] | 649 | test_count_width, test_index, test_count, test)) | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 650 | sys.stdout.flush() | 
|  | 651 | if trace: | 
|  | 652 | # If we're tracing code coverage, then we don't exit with status | 
|  | 653 | # if on a false return value from main. | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 654 | tracer.runctx('runtest(test, verbose, quiet, timeout=timeout)', | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 655 | globals=globals(), locals=vars()) | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 656 | else: | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 657 | try: | 
| Antoine Pitrou | 293954d | 2011-03-23 23:01:49 +0100 | [diff] [blame] | 658 | result = runtest(test, verbose, quiet, huntrleaks, debug, | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 659 | rerun_failed=verbose3, timeout=timeout) | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 660 | accumulate_result(test, result) | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 661 | except KeyboardInterrupt: | 
| Florent Xicluna | d6995eb | 2010-03-30 19:43:09 +0000 | [diff] [blame] | 662 | interrupted = True | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 663 | break | 
|  | 664 | except: | 
|  | 665 | raise | 
|  | 666 | if findleaks: | 
|  | 667 | gc.collect() | 
|  | 668 | if gc.garbage: | 
|  | 669 | print("Warning: test created", len(gc.garbage), end=' ') | 
|  | 670 | print("uncollectable object(s).") | 
|  | 671 | # move the uncollectable objects somewhere so we don't see | 
|  | 672 | # them again | 
|  | 673 | found_garbage.extend(gc.garbage) | 
|  | 674 | del gc.garbage[:] | 
|  | 675 | # Unload the newly imported modules (best effort finalization) | 
|  | 676 | for module in sys.modules.keys(): | 
|  | 677 | if module not in save_modules and module.startswith("test."): | 
|  | 678 | support.unload(module) | 
| Jeremy Hylton | 7a1ea0e | 2001-10-17 13:45:28 +0000 | [diff] [blame] | 679 |  | 
| Florent Xicluna | d6995eb | 2010-03-30 19:43:09 +0000 | [diff] [blame] | 680 | if interrupted: | 
|  | 681 | # print a newline after ^C | 
|  | 682 | print() | 
|  | 683 | print("Test suite interrupted by signal SIGINT.") | 
|  | 684 | omitted = set(selected) - set(good) - set(bad) - set(skipped) | 
|  | 685 | print(count(len(omitted), "test"), "omitted:") | 
|  | 686 | printlist(omitted) | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 687 | if good and not quiet: | 
| Florent Xicluna | d6995eb | 2010-03-30 19:43:09 +0000 | [diff] [blame] | 688 | if not bad and not skipped and not interrupted and len(good) > 1: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 689 | print("All", end=' ') | 
|  | 690 | print(count(len(good), "test"), "OK.") | 
| Christian Heimes | b186d00 | 2008-03-18 15:15:01 +0000 | [diff] [blame] | 691 | if print_slow: | 
|  | 692 | test_times.sort(reverse=True) | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 693 | print("10 slowest tests:") | 
| Christian Heimes | b186d00 | 2008-03-18 15:15:01 +0000 | [diff] [blame] | 694 | for time, test in test_times[:10]: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 695 | print("%s: %.1fs" % (test, time)) | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 696 | if bad: | 
| Nick Coghlan | 6ead552 | 2009-10-18 13:19:33 +0000 | [diff] [blame] | 697 | bad = sorted(set(bad) - set(environment_changed)) | 
|  | 698 | if bad: | 
|  | 699 | print(count(len(bad), "test"), "failed:") | 
|  | 700 | printlist(bad) | 
|  | 701 | if environment_changed: | 
|  | 702 | print("{} altered the execution environment:".format( | 
|  | 703 | count(len(environment_changed), "test"))) | 
|  | 704 | printlist(environment_changed) | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 705 | if skipped and not quiet: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 706 | print(count(len(skipped), "test"), "skipped:") | 
| Tim Peters | a45da92 | 2001-08-12 03:45:50 +0000 | [diff] [blame] | 707 | printlist(skipped) | 
| Barry Warsaw | e11e3de | 1999-01-28 19:51:51 +0000 | [diff] [blame] | 708 |  | 
| Tim Peters | b5b7b78 | 2001-08-12 01:20:39 +0000 | [diff] [blame] | 709 | e = _ExpectedSkips() | 
| Tim Peters | a2be2d6 | 2001-08-12 02:01:09 +0000 | [diff] [blame] | 710 | plat = sys.platform | 
| Tim Peters | b5b7b78 | 2001-08-12 01:20:39 +0000 | [diff] [blame] | 711 | if e.isvalid(): | 
| Raymond Hettinger | a690a99 | 2003-11-16 16:17:49 +0000 | [diff] [blame] | 712 | surprise = set(skipped) - e.getexpected() - set(resource_denieds) | 
| Tim Peters | b5b7b78 | 2001-08-12 01:20:39 +0000 | [diff] [blame] | 713 | if surprise: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 714 | print(count(len(surprise), "skip"), \ | 
|  | 715 | "unexpected on", plat + ":") | 
| Tim Peters | a45da92 | 2001-08-12 03:45:50 +0000 | [diff] [blame] | 716 | printlist(surprise) | 
| Tim Peters | b5b7b78 | 2001-08-12 01:20:39 +0000 | [diff] [blame] | 717 | else: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 718 | print("Those skips are all expected on", plat + ".") | 
| Tim Peters | b5b7b78 | 2001-08-12 01:20:39 +0000 | [diff] [blame] | 719 | else: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 720 | print("Ask someone to teach regrtest.py about which tests are") | 
|  | 721 | print("expected to get skipped on", plat + ".") | 
| Tim Peters | b5b7b78 | 2001-08-12 01:20:39 +0000 | [diff] [blame] | 722 |  | 
| Martin v. Löwis | 04824ce | 2006-03-10 21:26:16 +0000 | [diff] [blame] | 723 | if verbose2 and bad: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 724 | print("Re-running failed tests in verbose mode") | 
| Martin v. Löwis | 04824ce | 2006-03-10 21:26:16 +0000 | [diff] [blame] | 725 | for test in bad: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 726 | print("Re-running test %r in verbose mode" % test) | 
| Tim Peters | 922dd7d | 2006-03-10 23:37:10 +0000 | [diff] [blame] | 727 | sys.stdout.flush() | 
| Martin v. Löwis | 04824ce | 2006-03-10 21:26:16 +0000 | [diff] [blame] | 728 | try: | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 729 | verbose = True | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 730 | ok = runtest(test, True, quiet, huntrleaks, debug, timeout=timeout) | 
| Martin v. Löwis | 04824ce | 2006-03-10 21:26:16 +0000 | [diff] [blame] | 731 | except KeyboardInterrupt: | 
|  | 732 | # print a newline separate from the ^C | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 733 | print() | 
| Martin v. Löwis | 04824ce | 2006-03-10 21:26:16 +0000 | [diff] [blame] | 734 | break | 
|  | 735 | except: | 
|  | 736 | raise | 
|  | 737 |  | 
| Barry Warsaw | e11e3de | 1999-01-28 19:51:51 +0000 | [diff] [blame] | 738 | if single: | 
| R. David Murray | ef1992b | 2009-12-16 15:19:27 +0000 | [diff] [blame] | 739 | if next_single_test: | 
|  | 740 | with open(filename, 'w') as fp: | 
|  | 741 | fp.write(next_single_test + '\n') | 
| Barry Warsaw | e11e3de | 1999-01-28 19:51:51 +0000 | [diff] [blame] | 742 | else: | 
|  | 743 | os.unlink(filename) | 
|  | 744 |  | 
| Barry Warsaw | 3b6d025 | 2004-02-07 22:43:03 +0000 | [diff] [blame] | 745 | if trace: | 
|  | 746 | r = tracer.results() | 
|  | 747 | r.write_results(show_missing=True, summary=True, coverdir=coverdir) | 
|  | 748 |  | 
| Skip Montanaro | 0179a18 | 2004-06-06 15:53:18 +0000 | [diff] [blame] | 749 | if runleaks: | 
|  | 750 | os.system("leaks %d" % os.getpid()) | 
|  | 751 |  | 
| Florent Xicluna | d6995eb | 2010-03-30 19:43:09 +0000 | [diff] [blame] | 752 | sys.exit(len(bad) > 0 or interrupted) | 
| Barry Warsaw | 08fca52 | 2001-08-20 22:33:46 +0000 | [diff] [blame] | 753 |  | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 754 |  | 
| R David Murray | b588f8d | 2011-03-24 14:42:58 -0400 | [diff] [blame] | 755 | # small set of tests to determine if we have a basically functioning interpreter | 
|  | 756 | # (i.e. if any of these fail, then anything else is likely to follow) | 
| Guido van Rossum | 6fd83b7 | 1998-08-01 17:04:08 +0000 | [diff] [blame] | 757 | STDTESTS = [ | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 758 | 'test_grammar', | 
|  | 759 | 'test_opcodes', | 
| Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 760 | 'test_dict', | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 761 | 'test_builtin', | 
|  | 762 | 'test_exceptions', | 
|  | 763 | 'test_types', | 
| Collin Winter | 7afaa88 | 2007-03-08 19:54:43 +0000 | [diff] [blame] | 764 | 'test_unittest', | 
|  | 765 | 'test_doctest', | 
|  | 766 | 'test_doctest2', | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 767 | ] | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 768 |  | 
| R David Murray | b588f8d | 2011-03-24 14:42:58 -0400 | [diff] [blame] | 769 | # set of tests that we don't want to be executed when using regrtest | 
| R David Murray | 5764830 | 2011-03-24 14:57:05 -0400 | [diff] [blame] | 770 | NOTTESTS = set() | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 771 |  | 
| Guido van Rossum | 6fd83b7 | 1998-08-01 17:04:08 +0000 | [diff] [blame] | 772 | def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS): | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 773 | """Return a list of all applicable test modules.""" | 
| Florent Xicluna | da7bfd5 | 2010-03-06 11:43:55 +0000 | [diff] [blame] | 774 | testdir = findtestdir(testdir) | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 775 | names = os.listdir(testdir) | 
|  | 776 | tests = [] | 
| Florent Xicluna | da7bfd5 | 2010-03-06 11:43:55 +0000 | [diff] [blame] | 777 | others = set(stdtests) | nottests | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 778 | for name in names: | 
| R David Murray | 661720e | 2011-03-21 15:14:34 -0400 | [diff] [blame] | 779 | mod, ext = os.path.splitext(name) | 
|  | 780 | if mod[:5] == "test_" and ext in (".py", "") and mod not in others: | 
|  | 781 | tests.append(mod) | 
| Florent Xicluna | da7bfd5 | 2010-03-06 11:43:55 +0000 | [diff] [blame] | 782 | return stdtests + sorted(tests) | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 783 |  | 
| Victor Stinner | f58087b | 2010-05-02 17:24:51 +0000 | [diff] [blame] | 784 | def replace_stdout(): | 
|  | 785 | """Set stdout encoder error handler to backslashreplace (as stderr error | 
|  | 786 | handler) to avoid UnicodeEncodeError when printing a traceback""" | 
| Victor Stinner | a303280 | 2010-05-07 00:57:12 +0000 | [diff] [blame] | 787 | if os.name == "nt": | 
|  | 788 | # Replace sys.stdout breaks the stdout newlines on Windows: issue #8533 | 
|  | 789 | return | 
| Victor Stinner | 4b2b43d | 2011-01-05 03:54:26 +0000 | [diff] [blame] | 790 |  | 
|  | 791 | import atexit | 
|  | 792 |  | 
| Victor Stinner | f58087b | 2010-05-02 17:24:51 +0000 | [diff] [blame] | 793 | stdout = sys.stdout | 
|  | 794 | sys.stdout = open(stdout.fileno(), 'w', | 
|  | 795 | encoding=stdout.encoding, | 
| Victor Stinner | 4b2b43d | 2011-01-05 03:54:26 +0000 | [diff] [blame] | 796 | errors="backslashreplace", | 
|  | 797 | closefd=False) | 
|  | 798 |  | 
|  | 799 | def restore_stdout(): | 
|  | 800 | sys.stdout.close() | 
|  | 801 | sys.stdout = stdout | 
|  | 802 | atexit.register(restore_stdout) | 
| Victor Stinner | f58087b | 2010-05-02 17:24:51 +0000 | [diff] [blame] | 803 |  | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 804 | def runtest(test, verbose, quiet, | 
| Antoine Pitrou | 293954d | 2011-03-23 23:01:49 +0100 | [diff] [blame] | 805 | huntrleaks=False, debug=False, use_resources=None, | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 806 | rerun_failed=False, timeout=None): | 
| Guido van Rossum | 6fd83b7 | 1998-08-01 17:04:08 +0000 | [diff] [blame] | 807 | """Run a single test. | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 808 |  | 
| Guido van Rossum | 6fd83b7 | 1998-08-01 17:04:08 +0000 | [diff] [blame] | 809 | test -- the name of the test | 
| Guido van Rossum | 6fd83b7 | 1998-08-01 17:04:08 +0000 | [diff] [blame] | 810 | verbose -- if true, print more messages | 
| Trent Mick | f29f47b | 2000-08-11 19:02:59 +0000 | [diff] [blame] | 811 | quiet -- if true, don't print 'skipped' messages (probably redundant) | 
| Christian Heimes | b186d00 | 2008-03-18 15:15:01 +0000 | [diff] [blame] | 812 | test_times -- a list of (time, test_name) pairs | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 813 | huntrleaks -- run multiple times to test for leaks; requires a debug | 
|  | 814 | build; a triple corresponding to -R's three arguments | 
| Antoine Pitrou | 293954d | 2011-03-23 23:01:49 +0100 | [diff] [blame] | 815 | rerun_failed -- if true, re-run in verbose mode when failed | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 816 | timeout -- dump the traceback and exit if a test takes more than | 
|  | 817 | timeout seconds | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 818 |  | 
|  | 819 | Returns one of the test result constants: | 
|  | 820 | INTERRUPTED      KeyboardInterrupt when run under -j | 
|  | 821 | RESOURCE_DENIED  test skipped because resource denied | 
|  | 822 | SKIPPED          test skipped for some other reason | 
|  | 823 | ENV_CHANGED      test failed because it changed the execution environment | 
|  | 824 | FAILED           test failed | 
|  | 825 | PASSED           test passed | 
| Guido van Rossum | 6fd83b7 | 1998-08-01 17:04:08 +0000 | [diff] [blame] | 826 | """ | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 827 |  | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 828 | support.verbose = verbose  # Tell tests to be moderately quiet | 
|  | 829 | if use_resources is not None: | 
|  | 830 | support.use_resources = use_resources | 
| Victor Stinner | 7d648a0 | 2011-03-31 18:27:50 +0200 | [diff] [blame] | 831 | use_timeout = (timeout is not None and timeout > 0) | 
|  | 832 | if use_timeout: | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 833 | faulthandler.dump_tracebacks_later(timeout, exit=True) | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 834 | try: | 
| Antoine Pitrou | 293954d | 2011-03-23 23:01:49 +0100 | [diff] [blame] | 835 | result = runtest_inner(test, verbose, quiet, huntrleaks, debug) | 
|  | 836 | if result[0] == FAILED and rerun_failed: | 
|  | 837 | cleanup_test_droppings(test, verbose) | 
|  | 838 | sys.stdout.flush() | 
|  | 839 | sys.stderr.flush() | 
|  | 840 | print("Re-running test {} in verbose mode".format(test)) | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 841 | runtest(test, True, quiet, huntrleaks, debug, timeout=timeout) | 
| Antoine Pitrou | 293954d | 2011-03-23 23:01:49 +0100 | [diff] [blame] | 842 | return result | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 843 | finally: | 
| Victor Stinner | 7d648a0 | 2011-03-31 18:27:50 +0200 | [diff] [blame] | 844 | if use_timeout: | 
| Victor Stinner | 0cc8d59 | 2011-03-31 18:10:13 +0200 | [diff] [blame] | 845 | faulthandler.cancel_dump_tracebacks_later() | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 846 | cleanup_test_droppings(test, verbose) | 
|  | 847 |  | 
| Nick Coghlan | 6ead552 | 2009-10-18 13:19:33 +0000 | [diff] [blame] | 848 | # Unit tests are supposed to leave the execution environment unchanged | 
|  | 849 | # once they complete.  But sometimes tests have bugs, especially when | 
|  | 850 | # tests fail, and the changes to environment go on to mess up other | 
|  | 851 | # tests.  This can cause issues with buildbot stability, since tests | 
|  | 852 | # are run in random order and so problems may appear to come and go. | 
|  | 853 | # There are a few things we can save and restore to mitigate this, and | 
|  | 854 | # the following context manager handles this task. | 
|  | 855 |  | 
|  | 856 | class saved_test_environment: | 
|  | 857 | """Save bits of the test environment and restore them at block exit. | 
|  | 858 |  | 
|  | 859 | with saved_test_environment(testname, verbose, quiet): | 
|  | 860 | #stuff | 
|  | 861 |  | 
|  | 862 | Unless quiet is True, a warning is printed to stderr if any of | 
|  | 863 | the saved items was changed by the test.  The attribute 'changed' | 
|  | 864 | is initially False, but is set to True if a change is detected. | 
|  | 865 |  | 
|  | 866 | If verbose is more than 1, the before and after state of changed | 
|  | 867 | items is also printed. | 
|  | 868 | """ | 
|  | 869 |  | 
|  | 870 | changed = False | 
|  | 871 |  | 
|  | 872 | def __init__(self, testname, verbose=0, quiet=False): | 
|  | 873 | self.testname = testname | 
|  | 874 | self.verbose = verbose | 
|  | 875 | self.quiet = quiet | 
|  | 876 |  | 
|  | 877 | # To add things to save and restore, add a name XXX to the resources list | 
|  | 878 | # and add corresponding get_XXX/restore_XXX functions.  get_XXX should | 
|  | 879 | # return the value to be saved and compared against a second call to the | 
|  | 880 | # get function when test execution completes.  restore_XXX should accept | 
|  | 881 | # the saved value and restore the resource using it.  It will be called if | 
|  | 882 | # and only if a change in the value is detected. | 
|  | 883 | # | 
|  | 884 | # Note: XXX will have any '.' replaced with '_' characters when determining | 
|  | 885 | # the corresponding method names. | 
|  | 886 |  | 
|  | 887 | resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr', | 
| Brett Cannon | 29c0e4f | 2010-03-20 22:22:57 +0000 | [diff] [blame] | 888 | 'os.environ', 'sys.path', 'sys.path_hooks', '__import__', | 
| Nick Coghlan | 7bd5dbe | 2010-12-05 07:17:25 +0000 | [diff] [blame] | 889 | 'warnings.filters', 'asyncore.socket_map', | 
| Ezio Melotti | 45763d0 | 2011-03-20 15:34:28 +0200 | [diff] [blame] | 890 | 'logging._handlers', 'logging._handlerList', 'sys.gettrace', | 
| Ezio Melotti | 0123e05 | 2011-03-20 15:09:26 +0200 | [diff] [blame] | 891 | 'sys.warnoptions') | 
| Nick Coghlan | 6ead552 | 2009-10-18 13:19:33 +0000 | [diff] [blame] | 892 |  | 
|  | 893 | def get_sys_argv(self): | 
|  | 894 | return id(sys.argv), sys.argv, sys.argv[:] | 
|  | 895 | def restore_sys_argv(self, saved_argv): | 
|  | 896 | sys.argv = saved_argv[1] | 
|  | 897 | sys.argv[:] = saved_argv[2] | 
|  | 898 |  | 
|  | 899 | def get_cwd(self): | 
|  | 900 | return os.getcwd() | 
|  | 901 | def restore_cwd(self, saved_cwd): | 
|  | 902 | os.chdir(saved_cwd) | 
|  | 903 |  | 
|  | 904 | def get_sys_stdout(self): | 
|  | 905 | return sys.stdout | 
|  | 906 | def restore_sys_stdout(self, saved_stdout): | 
|  | 907 | sys.stdout = saved_stdout | 
|  | 908 |  | 
|  | 909 | def get_sys_stderr(self): | 
|  | 910 | return sys.stderr | 
|  | 911 | def restore_sys_stderr(self, saved_stderr): | 
|  | 912 | sys.stderr = saved_stderr | 
|  | 913 |  | 
|  | 914 | def get_sys_stdin(self): | 
|  | 915 | return sys.stdin | 
|  | 916 | def restore_sys_stdin(self, saved_stdin): | 
|  | 917 | sys.stdin = saved_stdin | 
|  | 918 |  | 
|  | 919 | def get_os_environ(self): | 
|  | 920 | return id(os.environ), os.environ, dict(os.environ) | 
|  | 921 | def restore_os_environ(self, saved_environ): | 
|  | 922 | os.environ = saved_environ[1] | 
|  | 923 | os.environ.clear() | 
|  | 924 | os.environ.update(saved_environ[2]) | 
|  | 925 |  | 
|  | 926 | def get_sys_path(self): | 
|  | 927 | return id(sys.path), sys.path, sys.path[:] | 
|  | 928 | def restore_sys_path(self, saved_path): | 
|  | 929 | sys.path = saved_path[1] | 
|  | 930 | sys.path[:] = saved_path[2] | 
|  | 931 |  | 
| Brett Cannon | 055470a | 2010-02-19 15:57:10 +0000 | [diff] [blame] | 932 | def get_sys_path_hooks(self): | 
|  | 933 | return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:] | 
|  | 934 | def restore_sys_path_hooks(self, saved_hooks): | 
|  | 935 | sys.path_hooks = saved_hooks[1] | 
|  | 936 | sys.path_hooks[:] = saved_hooks[2] | 
|  | 937 |  | 
| Brett Cannon | 31f5929 | 2011-02-21 19:29:56 +0000 | [diff] [blame] | 938 | def get_sys_gettrace(self): | 
|  | 939 | return sys.gettrace() | 
|  | 940 | def restore_sys_gettrace(self, trace_fxn): | 
|  | 941 | sys.settrace(trace_fxn) | 
|  | 942 |  | 
| Brett Cannon | 055470a | 2010-02-19 15:57:10 +0000 | [diff] [blame] | 943 | def get___import__(self): | 
| Brett Cannon | 4507190 | 2010-06-14 22:22:54 +0000 | [diff] [blame] | 944 | return builtins.__import__ | 
| Brett Cannon | 055470a | 2010-02-19 15:57:10 +0000 | [diff] [blame] | 945 | def restore___import__(self, import_): | 
| Brett Cannon | 4507190 | 2010-06-14 22:22:54 +0000 | [diff] [blame] | 946 | builtins.__import__ = import_ | 
| Brett Cannon | 055470a | 2010-02-19 15:57:10 +0000 | [diff] [blame] | 947 |  | 
| Brett Cannon | 29c0e4f | 2010-03-20 22:22:57 +0000 | [diff] [blame] | 948 | def get_warnings_filters(self): | 
|  | 949 | return id(warnings.filters), warnings.filters, warnings.filters[:] | 
|  | 950 | def restore_warnings_filters(self, saved_filters): | 
|  | 951 | warnings.filters = saved_filters[1] | 
|  | 952 | warnings.filters[:] = saved_filters[2] | 
|  | 953 |  | 
| Antoine Pitrou | b14ac8c | 2010-08-16 00:28:05 +0000 | [diff] [blame] | 954 | def get_asyncore_socket_map(self): | 
|  | 955 | asyncore = sys.modules.get('asyncore') | 
| Antoine Pitrou | aa87965 | 2010-10-29 11:54:38 +0000 | [diff] [blame] | 956 | # XXX Making a copy keeps objects alive until __exit__ gets called. | 
|  | 957 | return asyncore and asyncore.socket_map.copy() or {} | 
| Antoine Pitrou | b14ac8c | 2010-08-16 00:28:05 +0000 | [diff] [blame] | 958 | def restore_asyncore_socket_map(self, saved_map): | 
|  | 959 | asyncore = sys.modules.get('asyncore') | 
|  | 960 | if asyncore is not None: | 
| Antoine Pitrou | aa87965 | 2010-10-29 11:54:38 +0000 | [diff] [blame] | 961 | asyncore.close_all(ignore_all=True) | 
| Antoine Pitrou | b14ac8c | 2010-08-16 00:28:05 +0000 | [diff] [blame] | 962 | asyncore.socket_map.update(saved_map) | 
|  | 963 |  | 
| Nick Coghlan | 7d81975 | 2010-12-05 06:45:03 +0000 | [diff] [blame] | 964 | def get_logging__handlers(self): | 
|  | 965 | # _handlers is a WeakValueDictionary | 
| Nick Coghlan | 7bd5dbe | 2010-12-05 07:17:25 +0000 | [diff] [blame] | 966 | return id(logging._handlers), logging._handlers, logging._handlers.copy() | 
| Nick Coghlan | 7d81975 | 2010-12-05 06:45:03 +0000 | [diff] [blame] | 967 | def restore_logging__handlers(self, saved_handlers): | 
|  | 968 | # Can't easily revert the logging state | 
|  | 969 | pass | 
|  | 970 |  | 
| Nick Coghlan | 7bd5dbe | 2010-12-05 07:17:25 +0000 | [diff] [blame] | 971 | def get_logging__handlerList(self): | 
|  | 972 | # _handlerList is a list of weakrefs to handlers | 
|  | 973 | return id(logging._handlerList), logging._handlerList, logging._handlerList[:] | 
|  | 974 | def restore_logging__handlerList(self, saved_handlerList): | 
|  | 975 | # Can't easily revert the logging state | 
|  | 976 | pass | 
|  | 977 |  | 
| Ezio Melotti | 0123e05 | 2011-03-20 15:09:26 +0200 | [diff] [blame] | 978 | def get_sys_warnoptions(self): | 
|  | 979 | return id(sys.warnoptions), sys.warnoptions, sys.warnoptions[:] | 
|  | 980 | def restore_sys_warnoptions(self, saved_options): | 
|  | 981 | sys.warnoptions = saved_options[1] | 
|  | 982 | sys.warnoptions[:] = saved_options[2] | 
|  | 983 |  | 
| Nick Coghlan | 6ead552 | 2009-10-18 13:19:33 +0000 | [diff] [blame] | 984 | def resource_info(self): | 
|  | 985 | for name in self.resources: | 
|  | 986 | method_suffix = name.replace('.', '_') | 
|  | 987 | get_name = 'get_' + method_suffix | 
|  | 988 | restore_name = 'restore_' + method_suffix | 
|  | 989 | yield name, getattr(self, get_name), getattr(self, restore_name) | 
|  | 990 |  | 
|  | 991 | def __enter__(self): | 
|  | 992 | self.saved_values = dict((name, get()) for name, get, restore | 
|  | 993 | in self.resource_info()) | 
|  | 994 | return self | 
|  | 995 |  | 
|  | 996 | def __exit__(self, exc_type, exc_val, exc_tb): | 
| Antoine Pitrou | aa87965 | 2010-10-29 11:54:38 +0000 | [diff] [blame] | 997 | saved_values = self.saved_values | 
|  | 998 | del self.saved_values | 
| Nick Coghlan | 6ead552 | 2009-10-18 13:19:33 +0000 | [diff] [blame] | 999 | for name, get, restore in self.resource_info(): | 
|  | 1000 | current = get() | 
| Antoine Pitrou | aa87965 | 2010-10-29 11:54:38 +0000 | [diff] [blame] | 1001 | original = saved_values.pop(name) | 
| Nick Coghlan | 6ead552 | 2009-10-18 13:19:33 +0000 | [diff] [blame] | 1002 | # Check for changes to the resource's value | 
|  | 1003 | if current != original: | 
|  | 1004 | self.changed = True | 
|  | 1005 | restore(original) | 
|  | 1006 | if not self.quiet: | 
|  | 1007 | print("Warning -- {} was modified by {}".format( | 
|  | 1008 | name, self.testname), | 
|  | 1009 | file=sys.stderr) | 
|  | 1010 | if self.verbose > 1: | 
|  | 1011 | print("  Before: {}\n  After:  {} ".format( | 
|  | 1012 | original, current), | 
|  | 1013 | file=sys.stderr) | 
|  | 1014 | return False | 
|  | 1015 |  | 
|  | 1016 |  | 
| R. David Murray | 8e286c4 | 2010-12-27 20:09:32 +0000 | [diff] [blame] | 1017 | def runtest_inner(test, verbose, quiet, huntrleaks=False, debug=False): | 
| Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 1018 | support.unload(test) | 
| Benjamin Peterson | b48af54 | 2010-04-11 20:43:16 +0000 | [diff] [blame] | 1019 | if verbose: | 
|  | 1020 | capture_stdout = None | 
|  | 1021 | else: | 
|  | 1022 | capture_stdout = io.StringIO() | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1023 |  | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 1024 | test_time = 0.0 | 
| Collin Winter | a5503d5 | 2009-05-15 01:20:21 +0000 | [diff] [blame] | 1025 | refleak = False  # True if the test leaked references. | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 1026 | try: | 
| R. David Murray | 0bc11ae | 2009-10-18 22:18:17 +0000 | [diff] [blame] | 1027 | if test.startswith('test.'): | 
|  | 1028 | abstest = test | 
|  | 1029 | else: | 
|  | 1030 | # Always import it from the test package | 
|  | 1031 | abstest = 'test.' + test | 
|  | 1032 | with saved_test_environment(test, verbose, quiet) as environment: | 
|  | 1033 | start_time = time.time() | 
|  | 1034 | the_package = __import__(abstest, globals(), locals(), []) | 
|  | 1035 | the_module = getattr(the_package, test) | 
|  | 1036 | # Old tests run to completion simply as a side-effect of | 
|  | 1037 | # being imported.  For tests based on unittest or doctest, | 
|  | 1038 | # explicitly invoke their test_main() function (if it exists). | 
|  | 1039 | indirect_test = getattr(the_module, "test_main", None) | 
|  | 1040 | if indirect_test is not None: | 
|  | 1041 | indirect_test() | 
|  | 1042 | if huntrleaks: | 
|  | 1043 | refleak = dash_R(the_module, test, indirect_test, | 
|  | 1044 | huntrleaks) | 
|  | 1045 | test_time = time.time() - start_time | 
| Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 1046 | except support.ResourceDenied as msg: | 
| Fred Drake | 9a0db07 | 2003-02-03 15:19:30 +0000 | [diff] [blame] | 1047 | if not quiet: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1048 | print(test, "skipped --", msg) | 
| Fred Drake | 9a0db07 | 2003-02-03 15:19:30 +0000 | [diff] [blame] | 1049 | sys.stdout.flush() | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 1050 | return RESOURCE_DENIED, test_time | 
| R. David Murray | a21e4ca | 2009-03-31 23:16:50 +0000 | [diff] [blame] | 1051 | except unittest.SkipTest as msg: | 
| Trent Mick | f29f47b | 2000-08-11 19:02:59 +0000 | [diff] [blame] | 1052 | if not quiet: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1053 | print(test, "skipped --", msg) | 
| Guido van Rossum | 3cda93e | 2002-09-13 21:28:03 +0000 | [diff] [blame] | 1054 | sys.stdout.flush() | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 1055 | return SKIPPED, test_time | 
| Fred Drake | fe5c22a | 2000-08-18 16:04:05 +0000 | [diff] [blame] | 1056 | except KeyboardInterrupt: | 
|  | 1057 | raise | 
| Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 1058 | except support.TestFailed as msg: | 
| R. David Murray | 11cabcf | 2010-09-29 01:08:05 +0000 | [diff] [blame] | 1059 | print("test", test, "failed --", msg, file=sys.stderr) | 
|  | 1060 | sys.stderr.flush() | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 1061 | return FAILED, test_time | 
| Guido van Rossum | 9e48b27 | 1997-07-16 01:56:13 +0000 | [diff] [blame] | 1062 | except: | 
| Antoine Pitrou | 779a5b0 | 2011-03-21 19:55:16 +0100 | [diff] [blame] | 1063 | msg = traceback.format_exc() | 
|  | 1064 | print("test", test, "crashed --", msg, file=sys.stderr) | 
| R. David Murray | 11cabcf | 2010-09-29 01:08:05 +0000 | [diff] [blame] | 1065 | sys.stderr.flush() | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 1066 | return FAILED, test_time | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 1067 | else: | 
| Collin Winter | a5503d5 | 2009-05-15 01:20:21 +0000 | [diff] [blame] | 1068 | if refleak: | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 1069 | return FAILED, test_time | 
| Nick Coghlan | 6ead552 | 2009-10-18 13:19:33 +0000 | [diff] [blame] | 1070 | if environment.changed: | 
| Benjamin Peterson | a0dfa82 | 2009-11-13 02:25:08 +0000 | [diff] [blame] | 1071 | return ENV_CHANGED, test_time | 
|  | 1072 | return PASSED, test_time | 
| Guido van Rossum | 0fcca4e | 2001-09-21 20:31:52 +0000 | [diff] [blame] | 1073 |  | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1074 | def cleanup_test_droppings(testname, verbose): | 
|  | 1075 | import shutil | 
| Benjamin Peterson | 25c95f1 | 2009-05-08 20:42:26 +0000 | [diff] [blame] | 1076 | import stat | 
| Antoine Pitrou | c14efc4 | 2010-10-29 19:34:45 +0000 | [diff] [blame] | 1077 | import gc | 
|  | 1078 |  | 
|  | 1079 | # First kill any dangling references to open files etc. | 
|  | 1080 | # This can also issue some ResourceWarnings which would otherwise get | 
| Antoine Pitrou | 2b40efd | 2010-10-29 19:36:37 +0000 | [diff] [blame] | 1081 | # triggered during the following test run, and possibly produce failures. | 
| Antoine Pitrou | c14efc4 | 2010-10-29 19:34:45 +0000 | [diff] [blame] | 1082 | gc.collect() | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1083 |  | 
|  | 1084 | # Try to clean up junk commonly left behind.  While tests shouldn't leave | 
|  | 1085 | # any files or directories behind, when a test fails that can be tedious | 
|  | 1086 | # for it to arrange.  The consequences can be especially nasty on Windows, | 
|  | 1087 | # since if a test leaves a file open, it cannot be deleted by name (while | 
|  | 1088 | # there's nothing we can do about that here either, we can display the | 
|  | 1089 | # name of the offending test, which is a real help). | 
| Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 1090 | for name in (support.TESTFN, | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1091 | "db_home", | 
|  | 1092 | ): | 
|  | 1093 | if not os.path.exists(name): | 
|  | 1094 | continue | 
|  | 1095 |  | 
|  | 1096 | if os.path.isdir(name): | 
|  | 1097 | kind, nuker = "directory", shutil.rmtree | 
|  | 1098 | elif os.path.isfile(name): | 
|  | 1099 | kind, nuker = "file", os.unlink | 
|  | 1100 | else: | 
|  | 1101 | raise SystemError("os.path says %r exists but is neither " | 
|  | 1102 | "directory nor file" % name) | 
|  | 1103 |  | 
|  | 1104 | if verbose: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1105 | print("%r left behind %s %r" % (testname, kind, name)) | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1106 | try: | 
| Benjamin Peterson | 25c95f1 | 2009-05-08 20:42:26 +0000 | [diff] [blame] | 1107 | # if we have chmod, fix possible permissions problems | 
|  | 1108 | # that might prevent cleanup | 
|  | 1109 | if (hasattr(os, 'chmod')): | 
|  | 1110 | os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO) | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1111 | nuker(name) | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1112 | except Exception as msg: | 
|  | 1113 | print(("%r left behind %s %r and it couldn't be " | 
|  | 1114 | "removed: %s" % (testname, kind, name, msg)), file=sys.stderr) | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1115 |  | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1116 | def dash_R(the_module, test, indirect_test, huntrleaks): | 
| Collin Winter | a5503d5 | 2009-05-15 01:20:21 +0000 | [diff] [blame] | 1117 | """Run a test multiple times, looking for reference leaks. | 
|  | 1118 |  | 
|  | 1119 | Returns: | 
|  | 1120 | False if the test didn't leak references; True if we detected refleaks. | 
|  | 1121 | """ | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1122 | # This code is hackish and inelegant, but it seems to do the job. | 
| Raymond Hettinger | 158c9c2 | 2011-02-22 00:41:50 +0000 | [diff] [blame] | 1123 | import copyreg | 
|  | 1124 | import collections.abc | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1125 |  | 
|  | 1126 | if not hasattr(sys, 'gettotalrefcount'): | 
|  | 1127 | raise Exception("Tracking reference leaks requires a debug build " | 
|  | 1128 | "of Python") | 
|  | 1129 |  | 
|  | 1130 | # Save current values for dash_R_cleanup() to restore. | 
|  | 1131 | fs = warnings.filters[:] | 
| Alexandre Vassalotti | f7fa63d | 2008-05-11 08:55:36 +0000 | [diff] [blame] | 1132 | ps = copyreg.dispatch_table.copy() | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1133 | pic = sys.path_importer_cache.copy() | 
| Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 1134 | try: | 
|  | 1135 | import zipimport | 
|  | 1136 | except ImportError: | 
|  | 1137 | zdc = None # Run unmodified on platforms without zipimport support | 
|  | 1138 | else: | 
|  | 1139 | zdc = zipimport._zip_directory_cache.copy() | 
| Christian Heimes | 9385266 | 2007-12-01 12:22:32 +0000 | [diff] [blame] | 1140 | abcs = {} | 
| Raymond Hettinger | 158c9c2 | 2011-02-22 00:41:50 +0000 | [diff] [blame] | 1141 | for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]: | 
| Christian Heimes | be5b30b | 2008-03-03 19:18:51 +0000 | [diff] [blame] | 1142 | if not isabstract(abc): | 
| Christian Heimes | 9385266 | 2007-12-01 12:22:32 +0000 | [diff] [blame] | 1143 | continue | 
|  | 1144 | for obj in abc.__subclasses__() + [abc]: | 
|  | 1145 | abcs[obj] = obj._abc_registry.copy() | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1146 |  | 
|  | 1147 | if indirect_test: | 
|  | 1148 | def run_the_test(): | 
|  | 1149 | indirect_test() | 
|  | 1150 | else: | 
|  | 1151 | def run_the_test(): | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1152 | del sys.modules[the_module.__name__] | 
|  | 1153 | exec('import ' + the_module.__name__) | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1154 |  | 
|  | 1155 | deltas = [] | 
|  | 1156 | nwarmup, ntracked, fname = huntrleaks | 
| Ezio Melotti | 184bdfb | 2010-02-18 09:37:05 +0000 | [diff] [blame] | 1157 | fname = os.path.join(support.SAVEDCWD, fname) | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1158 | repcount = nwarmup + ntracked | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1159 | print("beginning", repcount, "repetitions", file=sys.stderr) | 
|  | 1160 | print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr) | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 1161 | sys.stderr.flush() | 
| Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 1162 | dash_R_cleanup(fs, ps, pic, zdc, abcs) | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1163 | for i in range(repcount): | 
| Collin Winter | 58bee59 | 2010-03-17 02:08:57 +0000 | [diff] [blame] | 1164 | rc_before = sys.gettotalrefcount() | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1165 | run_the_test() | 
|  | 1166 | sys.stderr.write('.') | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1167 | sys.stderr.flush() | 
| Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 1168 | dash_R_cleanup(fs, ps, pic, zdc, abcs) | 
| Collin Winter | 58bee59 | 2010-03-17 02:08:57 +0000 | [diff] [blame] | 1169 | rc_after = sys.gettotalrefcount() | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1170 | if i >= nwarmup: | 
| Collin Winter | 58bee59 | 2010-03-17 02:08:57 +0000 | [diff] [blame] | 1171 | deltas.append(rc_after - rc_before) | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1172 | print(file=sys.stderr) | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1173 | if any(deltas): | 
| Guido van Rossum | 360e4b8 | 2007-05-14 22:51:27 +0000 | [diff] [blame] | 1174 | msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas)) | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1175 | print(msg, file=sys.stderr) | 
| Antoine Pitrou | 8890954 | 2009-06-29 13:54:42 +0000 | [diff] [blame] | 1176 | sys.stderr.flush() | 
|  | 1177 | with open(fname, "a") as refrep: | 
|  | 1178 | print(msg, file=refrep) | 
|  | 1179 | refrep.flush() | 
| Collin Winter | a5503d5 | 2009-05-15 01:20:21 +0000 | [diff] [blame] | 1180 | return True | 
|  | 1181 | return False | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1182 |  | 
| Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 1183 | def dash_R_cleanup(fs, ps, pic, zdc, abcs): | 
| Alexandre Vassalotti | f7fa63d | 2008-05-11 08:55:36 +0000 | [diff] [blame] | 1184 | import gc, copyreg | 
| Brett Cannon | f4fd993 | 2008-05-10 21:11:46 +0000 | [diff] [blame] | 1185 | import _strptime, linecache | 
| Jeremy Hylton | 1afc169 | 2008-06-18 20:49:58 +0000 | [diff] [blame] | 1186 | import urllib.parse, urllib.request, mimetypes, doctest | 
| Raymond Hettinger | 158c9c2 | 2011-02-22 00:41:50 +0000 | [diff] [blame] | 1187 | import struct, filecmp, collections.abc | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1188 | from distutils.dir_util import _path_created | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1189 | from weakref import WeakSet | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1190 |  | 
| Christian Heimes | dae2a89 | 2008-04-19 00:55:37 +0000 | [diff] [blame] | 1191 | # Clear the warnings registry, so they can be displayed again | 
|  | 1192 | for mod in sys.modules.values(): | 
|  | 1193 | if hasattr(mod, '__warningregistry__'): | 
|  | 1194 | del mod.__warningregistry__ | 
|  | 1195 |  | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1196 | # Restore some original values. | 
|  | 1197 | warnings.filters[:] = fs | 
| Alexandre Vassalotti | f7fa63d | 2008-05-11 08:55:36 +0000 | [diff] [blame] | 1198 | copyreg.dispatch_table.clear() | 
|  | 1199 | copyreg.dispatch_table.update(ps) | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1200 | sys.path_importer_cache.clear() | 
|  | 1201 | sys.path_importer_cache.update(pic) | 
| Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 1202 | try: | 
|  | 1203 | import zipimport | 
|  | 1204 | except ImportError: | 
|  | 1205 | pass # Run unmodified on platforms without zipimport support | 
|  | 1206 | else: | 
|  | 1207 | zipimport._zip_directory_cache.clear() | 
|  | 1208 | zipimport._zip_directory_cache.update(zdc) | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1209 |  | 
| Christian Heimes | 2685563 | 2008-01-27 23:50:43 +0000 | [diff] [blame] | 1210 | # clear type cache | 
| Christian Heimes | 15ebc88 | 2008-02-04 18:48:49 +0000 | [diff] [blame] | 1211 | sys._clear_type_cache() | 
| Christian Heimes | 2685563 | 2008-01-27 23:50:43 +0000 | [diff] [blame] | 1212 |  | 
| Guido van Rossum | 3de862d | 2007-08-18 00:10:33 +0000 | [diff] [blame] | 1213 | # Clear ABC registries, restoring previously saved ABC registries. | 
| Raymond Hettinger | 158c9c2 | 2011-02-22 00:41:50 +0000 | [diff] [blame] | 1214 | for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]: | 
| Christian Heimes | be5b30b | 2008-03-03 19:18:51 +0000 | [diff] [blame] | 1215 | if not isabstract(abc): | 
| Christian Heimes | 941973a | 2007-11-30 21:53:03 +0000 | [diff] [blame] | 1216 | continue | 
| Guido van Rossum | 7eaf822 | 2007-06-18 17:58:50 +0000 | [diff] [blame] | 1217 | for obj in abc.__subclasses__() + [abc]: | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1218 | obj._abc_registry = abcs.get(obj, WeakSet()).copy() | 
| Guido van Rossum | c1e315d | 2007-08-20 19:29:24 +0000 | [diff] [blame] | 1219 | obj._abc_cache.clear() | 
|  | 1220 | obj._abc_negative_cache.clear() | 
| Guido van Rossum | cd16bf6 | 2007-06-13 18:07:49 +0000 | [diff] [blame] | 1221 |  | 
| Antoine Pitrou | 046467c | 2009-10-30 18:30:35 +0000 | [diff] [blame] | 1222 | # Flush standard output, so that buffered data is sent to the OS and | 
|  | 1223 | # associated Python objects are reclaimed. | 
|  | 1224 | for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__): | 
|  | 1225 | if stream is not None: | 
|  | 1226 | stream.flush() | 
|  | 1227 |  | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1228 | # Clear assorted module caches. | 
|  | 1229 | _path_created.clear() | 
|  | 1230 | re.purge() | 
|  | 1231 | _strptime._regex_cache.clear() | 
| Jeremy Hylton | 1afc169 | 2008-06-18 20:49:58 +0000 | [diff] [blame] | 1232 | urllib.parse.clear_cache() | 
|  | 1233 | urllib.request.urlcleanup() | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1234 | linecache.clearcache() | 
|  | 1235 | mimetypes._default_mime_types() | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1236 | filecmp._cache.clear() | 
| Christian Heimes | a34706f | 2008-01-04 03:06:10 +0000 | [diff] [blame] | 1237 | struct._clearcache() | 
| Thomas Wouters | 73e5a5b | 2006-06-08 15:35:45 +0000 | [diff] [blame] | 1238 | doctest.master = None | 
|  | 1239 |  | 
|  | 1240 | # Collect cyclic trash. | 
|  | 1241 | gc.collect() | 
|  | 1242 |  | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1243 | def warm_char_cache(): | 
|  | 1244 | s = bytes(range(256)) | 
|  | 1245 | for i in range(256): | 
|  | 1246 | s[i:i+1] | 
|  | 1247 |  | 
| Florent Xicluna | da7bfd5 | 2010-03-06 11:43:55 +0000 | [diff] [blame] | 1248 | def findtestdir(path=None): | 
|  | 1249 | return path or os.path.dirname(__file__) or os.curdir | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 1250 |  | 
| Florent Xicluna | da7bfd5 | 2010-03-06 11:43:55 +0000 | [diff] [blame] | 1251 | def removepy(names): | 
|  | 1252 | if not names: | 
|  | 1253 | return | 
|  | 1254 | for idx, name in enumerate(names): | 
|  | 1255 | basename, ext = os.path.splitext(name) | 
|  | 1256 | if ext == '.py': | 
|  | 1257 | names[idx] = basename | 
| Tim Peters | c5000df | 2002-06-02 21:42:01 +0000 | [diff] [blame] | 1258 |  | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 1259 | def count(n, word): | 
|  | 1260 | if n == 1: | 
| Guido van Rossum | 41360a4 | 1998-03-26 19:42:58 +0000 | [diff] [blame] | 1261 | return "%d %s" % (n, word) | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 1262 | else: | 
| Guido van Rossum | 41360a4 | 1998-03-26 19:42:58 +0000 | [diff] [blame] | 1263 | return "%d %ss" % (n, word) | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 1264 |  | 
| Tim Peters | a45da92 | 2001-08-12 03:45:50 +0000 | [diff] [blame] | 1265 | def printlist(x, width=70, indent=4): | 
| Tim Peters | 7c7efe9 | 2002-08-23 17:55:54 +0000 | [diff] [blame] | 1266 | """Print the elements of iterable x to stdout. | 
| Tim Peters | a45da92 | 2001-08-12 03:45:50 +0000 | [diff] [blame] | 1267 |  | 
|  | 1268 | Optional arg width (default 70) is the maximum line length. | 
|  | 1269 | Optional arg indent (default 4) is the number of blanks with which to | 
|  | 1270 | begin each line. | 
|  | 1271 | """ | 
|  | 1272 |  | 
| Tim Peters | ba78bc4 | 2002-07-04 19:45:06 +0000 | [diff] [blame] | 1273 | from textwrap import fill | 
|  | 1274 | blanks = ' ' * indent | 
| Florent Xicluna | fd1b093 | 2010-03-28 00:25:02 +0000 | [diff] [blame] | 1275 | # Print the sorted list: 'x' may be a '--random' list or a set() | 
|  | 1276 | print(fill(' '.join(str(elt) for elt in sorted(x)), width, | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1277 | initial_indent=blanks, subsequent_indent=blanks)) | 
| Tim Peters | a45da92 | 2001-08-12 03:45:50 +0000 | [diff] [blame] | 1278 |  | 
| Tim Peters | de14a30 | 2002-04-01 05:04:46 +0000 | [diff] [blame] | 1279 | # Map sys.platform to a string containing the basenames of tests | 
|  | 1280 | # expected to be skipped on that platform. | 
| Tim Peters | 2a182db | 2002-10-09 01:07:11 +0000 | [diff] [blame] | 1281 | # | 
|  | 1282 | # Special cases: | 
|  | 1283 | #     test_pep277 | 
|  | 1284 | #         The _ExpectedSkips constructor adds this to the set of expected | 
|  | 1285 | #         skips if not os.path.supports_unicode_filenames. | 
| Neal Norwitz | 55b61d2 | 2003-02-28 19:57:03 +0000 | [diff] [blame] | 1286 | #     test_timeout | 
|  | 1287 | #         Controlled by test_timeout.skip_expected.  Requires the network | 
|  | 1288 | #         resource and a socket module. | 
| Guido van Rossum | b5a755e | 2007-07-18 18:15:48 +0000 | [diff] [blame] | 1289 | # | 
|  | 1290 | # Tests that are expected to be skipped everywhere except on one platform | 
|  | 1291 | # are also handled separately. | 
| Tim Peters | de14a30 | 2002-04-01 05:04:46 +0000 | [diff] [blame] | 1292 |  | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1293 | _expectations = { | 
|  | 1294 | 'win32': | 
|  | 1295 | """ | 
| Tim Peters | c7c516a | 2003-09-20 22:06:13 +0000 | [diff] [blame] | 1296 | test__locale | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1297 | test_crypt | 
| Tim Peters | d703057 | 2001-10-22 22:06:08 +0000 | [diff] [blame] | 1298 | test_curses | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1299 | test_dbm | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1300 | test_fcntl | 
|  | 1301 | test_fork1 | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1302 | test_epoll | 
| Benjamin Peterson | 4d48053 | 2008-05-26 19:08:31 +0000 | [diff] [blame] | 1303 | test_dbm_gnu | 
| Kristján Valur Jónsson | 42a40c5 | 2009-04-01 11:28:47 +0000 | [diff] [blame] | 1304 | test_dbm_ndbm | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1305 | test_grp | 
| Tim Peters | fd8e6e5 | 2003-03-04 00:26:38 +0000 | [diff] [blame] | 1306 | test_ioctl | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1307 | test_largefile | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1308 | test_kqueue | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1309 | test_openpty | 
| Tim Peters | efc4b12 | 2002-12-10 18:47:56 +0000 | [diff] [blame] | 1310 | test_ossaudiodev | 
| Thomas Wouters | 47b49bf | 2007-08-30 22:15:33 +0000 | [diff] [blame] | 1311 | test_pipes | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1312 | test_poll | 
| Tim Peters | 003eb30 | 2003-02-17 21:48:48 +0000 | [diff] [blame] | 1313 | test_posix | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1314 | test_pty | 
|  | 1315 | test_pwd | 
| Tim Peters | 1e33ffa | 2002-04-23 23:09:02 +0000 | [diff] [blame] | 1316 | test_resource | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1317 | test_signal | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1318 | test_syslog | 
| Tim Peters | cea2cc4 | 2004-08-04 02:32:03 +0000 | [diff] [blame] | 1319 | test_threadsignals | 
| Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 1320 | test_wait3 | 
|  | 1321 | test_wait4 | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1322 | """, | 
|  | 1323 | 'linux2': | 
|  | 1324 | """ | 
| Guido van Rossum | f66dacd | 2001-10-23 15:10:55 +0000 | [diff] [blame] | 1325 | test_curses | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1326 | test_largefile | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1327 | test_kqueue | 
| Guido van Rossum | 4507ec7 | 2003-02-14 19:29:22 +0000 | [diff] [blame] | 1328 | test_ossaudiodev | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1329 | """, | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1330 | 'unixware7': | 
| Martin v. Löwis | 0ace326 | 2001-09-05 14:38:48 +0000 | [diff] [blame] | 1331 | """ | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1332 | test_epoll | 
| Martin v. Löwis | 0ace326 | 2001-09-05 14:38:48 +0000 | [diff] [blame] | 1333 | test_largefile | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1334 | test_kqueue | 
| Martin v. Löwis | 0ace326 | 2001-09-05 14:38:48 +0000 | [diff] [blame] | 1335 | test_minidom | 
| Martin v. Löwis | 0ace326 | 2001-09-05 14:38:48 +0000 | [diff] [blame] | 1336 | test_openpty | 
|  | 1337 | test_pyexpat | 
|  | 1338 | test_sax | 
| Martin v. Löwis | 0ace326 | 2001-09-05 14:38:48 +0000 | [diff] [blame] | 1339 | test_sundry | 
| Martin v. Löwis | 0ace326 | 2001-09-05 14:38:48 +0000 | [diff] [blame] | 1340 | """, | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1341 | 'openunix8': | 
|  | 1342 | """ | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1343 | test_epoll | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1344 | test_largefile | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1345 | test_kqueue | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1346 | test_minidom | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1347 | test_openpty | 
|  | 1348 | test_pyexpat | 
|  | 1349 | test_sax | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1350 | test_sundry | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1351 | """, | 
|  | 1352 | 'sco_sv3': | 
|  | 1353 | """ | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1354 | test_asynchat | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1355 | test_fork1 | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1356 | test_epoll | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1357 | test_gettext | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1358 | test_largefile | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1359 | test_locale | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1360 | test_kqueue | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1361 | test_minidom | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1362 | test_openpty | 
|  | 1363 | test_pyexpat | 
|  | 1364 | test_queue | 
|  | 1365 | test_sax | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1366 | test_sundry | 
|  | 1367 | test_thread | 
|  | 1368 | test_threaded_import | 
|  | 1369 | test_threadedtempfile | 
|  | 1370 | test_threading | 
| Martin v. Löwis | 21ee409 | 2002-09-30 16:19:48 +0000 | [diff] [blame] | 1371 | """, | 
| Jack Jansen | 8a97f4a | 2001-12-05 23:27:32 +0000 | [diff] [blame] | 1372 | 'darwin': | 
| Jack Jansen | 398c236 | 2001-12-02 21:41:36 +0000 | [diff] [blame] | 1373 | """ | 
| Guido van Rossum | b5a755e | 2007-07-18 18:15:48 +0000 | [diff] [blame] | 1374 | test__locale | 
| Guido van Rossum | b5a755e | 2007-07-18 18:15:48 +0000 | [diff] [blame] | 1375 | test_curses | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1376 | test_epoll | 
| Benjamin Peterson | 4d48053 | 2008-05-26 19:08:31 +0000 | [diff] [blame] | 1377 | test_dbm_gnu | 
| Skip Montanaro | 97019ff | 2010-10-25 01:35:48 +0000 | [diff] [blame] | 1378 | test_gdb | 
| Jack Jansen | 398c236 | 2001-12-02 21:41:36 +0000 | [diff] [blame] | 1379 | test_largefile | 
| Jack Jansen | acda339 | 2002-12-30 23:03:13 +0000 | [diff] [blame] | 1380 | test_locale | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1381 | test_minidom | 
| Jack Jansen | acda339 | 2002-12-30 23:03:13 +0000 | [diff] [blame] | 1382 | test_ossaudiodev | 
| Guido van Rossum | b5a755e | 2007-07-18 18:15:48 +0000 | [diff] [blame] | 1383 | test_poll | 
| Jack Jansen | 398c236 | 2001-12-02 21:41:36 +0000 | [diff] [blame] | 1384 | """, | 
| Guido van Rossum | 11c3f09 | 2002-07-17 15:08:24 +0000 | [diff] [blame] | 1385 | 'sunos5': | 
|  | 1386 | """ | 
| Guido van Rossum | 11c3f09 | 2002-07-17 15:08:24 +0000 | [diff] [blame] | 1387 | test_curses | 
|  | 1388 | test_dbm | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1389 | test_epoll | 
|  | 1390 | test_kqueue | 
| Benjamin Peterson | 4d48053 | 2008-05-26 19:08:31 +0000 | [diff] [blame] | 1391 | test_dbm_gnu | 
| Guido van Rossum | 11c3f09 | 2002-07-17 15:08:24 +0000 | [diff] [blame] | 1392 | test_gzip | 
| Guido van Rossum | 11c3f09 | 2002-07-17 15:08:24 +0000 | [diff] [blame] | 1393 | test_openpty | 
| Guido van Rossum | 11c3f09 | 2002-07-17 15:08:24 +0000 | [diff] [blame] | 1394 | test_zipfile | 
|  | 1395 | test_zlib | 
| Jeremy Hylton | ed375e1 | 2002-07-17 15:56:55 +0000 | [diff] [blame] | 1396 | """, | 
| Skip Montanaro | b323021 | 2002-03-15 02:54:03 +0000 | [diff] [blame] | 1397 | 'hp-ux11': | 
|  | 1398 | """ | 
| Skip Montanaro | b323021 | 2002-03-15 02:54:03 +0000 | [diff] [blame] | 1399 | test_curses | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1400 | test_epoll | 
| Benjamin Peterson | 4d48053 | 2008-05-26 19:08:31 +0000 | [diff] [blame] | 1401 | test_dbm_gnu | 
| Skip Montanaro | b323021 | 2002-03-15 02:54:03 +0000 | [diff] [blame] | 1402 | test_gzip | 
| Skip Montanaro | b323021 | 2002-03-15 02:54:03 +0000 | [diff] [blame] | 1403 | test_largefile | 
| Skip Montanaro | b323021 | 2002-03-15 02:54:03 +0000 | [diff] [blame] | 1404 | test_locale | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1405 | test_kqueue | 
| Skip Montanaro | b323021 | 2002-03-15 02:54:03 +0000 | [diff] [blame] | 1406 | test_minidom | 
| Skip Montanaro | b323021 | 2002-03-15 02:54:03 +0000 | [diff] [blame] | 1407 | test_openpty | 
|  | 1408 | test_pyexpat | 
|  | 1409 | test_sax | 
| Skip Montanaro | b323021 | 2002-03-15 02:54:03 +0000 | [diff] [blame] | 1410 | test_zipfile | 
|  | 1411 | test_zlib | 
|  | 1412 | """, | 
| Jason Tishler | 2511594 | 2002-12-05 15:18:15 +0000 | [diff] [blame] | 1413 | 'cygwin': | 
|  | 1414 | """ | 
| Jason Tishler | 2511594 | 2002-12-05 15:18:15 +0000 | [diff] [blame] | 1415 | test_curses | 
|  | 1416 | test_dbm | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1417 | test_epoll | 
| Jason Tishler | c23f39c | 2003-07-22 18:35:58 +0000 | [diff] [blame] | 1418 | test_ioctl | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1419 | test_kqueue | 
| Jason Tishler | 2511594 | 2002-12-05 15:18:15 +0000 | [diff] [blame] | 1420 | test_largefile | 
| Jason Tishler | 2511594 | 2002-12-05 15:18:15 +0000 | [diff] [blame] | 1421 | test_locale | 
| Jason Tishler | 5c4ded2 | 2003-02-05 16:46:01 +0000 | [diff] [blame] | 1422 | test_ossaudiodev | 
| Jason Tishler | 2511594 | 2002-12-05 15:18:15 +0000 | [diff] [blame] | 1423 | test_socketserver | 
| Jason Tishler | 2511594 | 2002-12-05 15:18:15 +0000 | [diff] [blame] | 1424 | """, | 
| Andrew MacIntyre | fd07e7d | 2002-12-31 11:26:50 +0000 | [diff] [blame] | 1425 | 'os2emx': | 
|  | 1426 | """ | 
| Andrew MacIntyre | fd07e7d | 2002-12-31 11:26:50 +0000 | [diff] [blame] | 1427 | test_audioop | 
| Andrew MacIntyre | fd07e7d | 2002-12-31 11:26:50 +0000 | [diff] [blame] | 1428 | test_curses | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1429 | test_epoll | 
|  | 1430 | test_kqueue | 
| Andrew MacIntyre | fd07e7d | 2002-12-31 11:26:50 +0000 | [diff] [blame] | 1431 | test_largefile | 
| Andrew MacIntyre | fd07e7d | 2002-12-31 11:26:50 +0000 | [diff] [blame] | 1432 | test_mmap | 
| Andrew MacIntyre | fd07e7d | 2002-12-31 11:26:50 +0000 | [diff] [blame] | 1433 | test_openpty | 
|  | 1434 | test_ossaudiodev | 
|  | 1435 | test_pty | 
|  | 1436 | test_resource | 
|  | 1437 | test_signal | 
| Andrew MacIntyre | fd07e7d | 2002-12-31 11:26:50 +0000 | [diff] [blame] | 1438 | """, | 
| Guido van Rossum | 944a6c3 | 2003-11-20 22:11:29 +0000 | [diff] [blame] | 1439 | 'freebsd4': | 
|  | 1440 | """ | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1441 | test_epoll | 
| Benjamin Peterson | 4d48053 | 2008-05-26 19:08:31 +0000 | [diff] [blame] | 1442 | test_dbm_gnu | 
| Guido van Rossum | 944a6c3 | 2003-11-20 22:11:29 +0000 | [diff] [blame] | 1443 | test_locale | 
| Guido van Rossum | 944a6c3 | 2003-11-20 22:11:29 +0000 | [diff] [blame] | 1444 | test_ossaudiodev | 
|  | 1445 | test_pep277 | 
| Hye-Shik Chang | f64700a | 2004-08-18 15:13:41 +0000 | [diff] [blame] | 1446 | test_pty | 
| Guido van Rossum | 944a6c3 | 2003-11-20 22:11:29 +0000 | [diff] [blame] | 1447 | test_socketserver | 
| Hye-Shik Chang | f64700a | 2004-08-18 15:13:41 +0000 | [diff] [blame] | 1448 | test_tcl | 
| Guilherme Polo | a91790a | 2009-02-09 20:40:42 +0000 | [diff] [blame] | 1449 | test_tk | 
| Guilherme Polo | 9de29af | 2009-01-28 20:40:48 +0000 | [diff] [blame] | 1450 | test_ttk_guionly | 
|  | 1451 | test_ttk_textonly | 
| Guido van Rossum | 944a6c3 | 2003-11-20 22:11:29 +0000 | [diff] [blame] | 1452 | test_timeout | 
| Guido van Rossum | 944a6c3 | 2003-11-20 22:11:29 +0000 | [diff] [blame] | 1453 | test_urllibnet | 
| Benjamin Peterson | e5384b0 | 2008-10-04 22:00:42 +0000 | [diff] [blame] | 1454 | test_multiprocessing | 
| Martin v. Löwis | 56f8811 | 2003-06-07 20:01:37 +0000 | [diff] [blame] | 1455 | """, | 
| Guido van Rossum | 8ee3e5a | 2005-09-14 18:09:42 +0000 | [diff] [blame] | 1456 | 'aix5': | 
|  | 1457 | """ | 
| Guido van Rossum | 8ee3e5a | 2005-09-14 18:09:42 +0000 | [diff] [blame] | 1458 | test_bz2 | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1459 | test_epoll | 
| Benjamin Peterson | 4d48053 | 2008-05-26 19:08:31 +0000 | [diff] [blame] | 1460 | test_dbm_gnu | 
| Guido van Rossum | 8ee3e5a | 2005-09-14 18:09:42 +0000 | [diff] [blame] | 1461 | test_gzip | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1462 | test_kqueue | 
| Guido van Rossum | 8ee3e5a | 2005-09-14 18:09:42 +0000 | [diff] [blame] | 1463 | test_ossaudiodev | 
| Guido van Rossum | 8ee3e5a | 2005-09-14 18:09:42 +0000 | [diff] [blame] | 1464 | test_tcl | 
| Guilherme Polo | a91790a | 2009-02-09 20:40:42 +0000 | [diff] [blame] | 1465 | test_tk | 
| Guilherme Polo | 9de29af | 2009-01-28 20:40:48 +0000 | [diff] [blame] | 1466 | test_ttk_guionly | 
|  | 1467 | test_ttk_textonly | 
| Guido van Rossum | 8ee3e5a | 2005-09-14 18:09:42 +0000 | [diff] [blame] | 1468 | test_zipimport | 
|  | 1469 | test_zlib | 
|  | 1470 | """, | 
| Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 1471 | 'openbsd3': | 
|  | 1472 | """ | 
| Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 1473 | test_ctypes | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1474 | test_epoll | 
| Benjamin Peterson | 4d48053 | 2008-05-26 19:08:31 +0000 | [diff] [blame] | 1475 | test_dbm_gnu | 
| Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 1476 | test_locale | 
| Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 1477 | test_normalization | 
|  | 1478 | test_ossaudiodev | 
|  | 1479 | test_pep277 | 
| Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 1480 | test_tcl | 
| Guilherme Polo | a91790a | 2009-02-09 20:40:42 +0000 | [diff] [blame] | 1481 | test_tk | 
| Guilherme Polo | 9de29af | 2009-01-28 20:40:48 +0000 | [diff] [blame] | 1482 | test_ttk_guionly | 
|  | 1483 | test_ttk_textonly | 
| Benjamin Peterson | e5384b0 | 2008-10-04 22:00:42 +0000 | [diff] [blame] | 1484 | test_multiprocessing | 
| Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 1485 | """, | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1486 | 'netbsd3': | 
|  | 1487 | """ | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1488 | test_ctypes | 
|  | 1489 | test_curses | 
| Christian Heimes | 4fbc72b | 2008-03-22 00:47:35 +0000 | [diff] [blame] | 1490 | test_epoll | 
| Benjamin Peterson | 4d48053 | 2008-05-26 19:08:31 +0000 | [diff] [blame] | 1491 | test_dbm_gnu | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1492 | test_locale | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1493 | test_ossaudiodev | 
|  | 1494 | test_pep277 | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1495 | test_tcl | 
| Guilherme Polo | a91790a | 2009-02-09 20:40:42 +0000 | [diff] [blame] | 1496 | test_tk | 
| Guilherme Polo | 9de29af | 2009-01-28 20:40:48 +0000 | [diff] [blame] | 1497 | test_ttk_guionly | 
|  | 1498 | test_ttk_textonly | 
| Benjamin Peterson | e5384b0 | 2008-10-04 22:00:42 +0000 | [diff] [blame] | 1499 | test_multiprocessing | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1500 | """, | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1501 | } | 
| Martin v. Löwis | 32d0c1b | 2004-07-26 12:09:13 +0000 | [diff] [blame] | 1502 | _expectations['freebsd5'] = _expectations['freebsd4'] | 
| Hye-Shik Chang | f64700a | 2004-08-18 15:13:41 +0000 | [diff] [blame] | 1503 | _expectations['freebsd6'] = _expectations['freebsd4'] | 
| Hye-Shik Chang | 4e42281 | 2005-07-17 02:36:59 +0000 | [diff] [blame] | 1504 | _expectations['freebsd7'] = _expectations['freebsd4'] | 
| Guido van Rossum | 8ce8a78 | 2007-11-01 19:42:39 +0000 | [diff] [blame] | 1505 | _expectations['freebsd8'] = _expectations['freebsd4'] | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1506 |  | 
| Tim Peters | b5b7b78 | 2001-08-12 01:20:39 +0000 | [diff] [blame] | 1507 | class _ExpectedSkips: | 
|  | 1508 | def __init__(self): | 
| Tim Peters | 2a182db | 2002-10-09 01:07:11 +0000 | [diff] [blame] | 1509 | import os.path | 
| Neal Norwitz | 55b61d2 | 2003-02-28 19:57:03 +0000 | [diff] [blame] | 1510 | from test import test_timeout | 
| Tim Peters | 1b445d3 | 2002-11-24 18:53:11 +0000 | [diff] [blame] | 1511 |  | 
| Tim Peters | 7c7efe9 | 2002-08-23 17:55:54 +0000 | [diff] [blame] | 1512 | self.valid = False | 
| Tim Peters | de14a30 | 2002-04-01 05:04:46 +0000 | [diff] [blame] | 1513 | if sys.platform in _expectations: | 
| Guido van Rossum | f73e30c | 2001-08-12 02:22:19 +0000 | [diff] [blame] | 1514 | s = _expectations[sys.platform] | 
| Raymond Hettinger | a690a99 | 2003-11-16 16:17:49 +0000 | [diff] [blame] | 1515 | self.expected = set(s.split()) | 
| Tim Peters | 1b445d3 | 2002-11-24 18:53:11 +0000 | [diff] [blame] | 1516 |  | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1517 | # These are broken tests, for now skipped on every platform. | 
|  | 1518 | # XXX Fix these! | 
| Benjamin Peterson | e9ea19e | 2008-08-19 23:02:38 +0000 | [diff] [blame] | 1519 | self.expected.add('test_nis') | 
| Benjamin Peterson | 4fde0c4 | 2008-03-31 02:36:22 +0000 | [diff] [blame] | 1520 |  | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1521 | # expected to be skipped on every platform, even Linux | 
| Tim Peters | 2a182db | 2002-10-09 01:07:11 +0000 | [diff] [blame] | 1522 | if not os.path.supports_unicode_filenames: | 
|  | 1523 | self.expected.add('test_pep277') | 
| Tim Peters | 1b445d3 | 2002-11-24 18:53:11 +0000 | [diff] [blame] | 1524 |  | 
| Neal Norwitz | 94fa2ee | 2008-03-31 02:55:15 +0000 | [diff] [blame] | 1525 | # doctest, profile and cProfile tests fail when the codec for the | 
|  | 1526 | # fs encoding isn't built in because PyUnicode_Decode() adds two | 
|  | 1527 | # calls into Python. | 
|  | 1528 | encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32") | 
|  | 1529 | if sys.getfilesystemencoding().lower() not in encs: | 
|  | 1530 | self.expected.add('test_profile') | 
|  | 1531 | self.expected.add('test_cProfile') | 
|  | 1532 | self.expected.add('test_doctest') | 
| Guido van Rossum | daa251c | 2007-10-25 23:47:33 +0000 | [diff] [blame] | 1533 |  | 
| Neal Norwitz | 55b61d2 | 2003-02-28 19:57:03 +0000 | [diff] [blame] | 1534 | if test_timeout.skip_expected: | 
|  | 1535 | self.expected.add('test_timeout') | 
|  | 1536 |  | 
| Tim Peters | ecd79eb | 2003-01-29 00:35:32 +0000 | [diff] [blame] | 1537 | if sys.platform != "win32": | 
| Guido van Rossum | b5a755e | 2007-07-18 18:15:48 +0000 | [diff] [blame] | 1538 | # test_sqlite is only reliable on Windows where the library | 
|  | 1539 | # is distributed with Python | 
| Benjamin Peterson | 54ad0be | 2010-10-29 21:33:10 +0000 | [diff] [blame] | 1540 | WIN_ONLY = {"test_unicode_file", "test_winreg", | 
| Guido van Rossum | b5a755e | 2007-07-18 18:15:48 +0000 | [diff] [blame] | 1541 | "test_winsound", "test_startfile", | 
| Ross Lagerwall | 226580e | 2011-04-09 20:05:04 +0200 | [diff] [blame] | 1542 | "test_sqlite", "test_msilib"} | 
| Benjamin Peterson | 54ad0be | 2010-10-29 21:33:10 +0000 | [diff] [blame] | 1543 | self.expected |= WIN_ONLY | 
| Tim Peters | f2715e0 | 2003-02-19 02:35:07 +0000 | [diff] [blame] | 1544 |  | 
| Guido van Rossum | b5a755e | 2007-07-18 18:15:48 +0000 | [diff] [blame] | 1545 | if sys.platform != 'sunos5': | 
| Guido van Rossum | b5a755e | 2007-07-18 18:15:48 +0000 | [diff] [blame] | 1546 | self.expected.add('test_nis') | 
| Guido van Rossum | d59da4b | 2007-05-22 18:11:13 +0000 | [diff] [blame] | 1547 |  | 
| Benjamin Peterson | 65c66ab | 2010-10-29 21:31:35 +0000 | [diff] [blame] | 1548 | if support.python_is_optimized(): | 
|  | 1549 | self.expected.add("test_gdb") | 
|  | 1550 |  | 
| Tim Peters | 7c7efe9 | 2002-08-23 17:55:54 +0000 | [diff] [blame] | 1551 | self.valid = True | 
| Tim Peters | b5b7b78 | 2001-08-12 01:20:39 +0000 | [diff] [blame] | 1552 |  | 
|  | 1553 | def isvalid(self): | 
|  | 1554 | "Return true iff _ExpectedSkips knows about the current platform." | 
|  | 1555 | return self.valid | 
|  | 1556 |  | 
|  | 1557 | def getexpected(self): | 
|  | 1558 | """Return set of test names we expect to skip on current platform. | 
|  | 1559 |  | 
|  | 1560 | self.isvalid() must be true. | 
|  | 1561 | """ | 
|  | 1562 |  | 
|  | 1563 | assert self.isvalid() | 
|  | 1564 | return self.expected | 
|  | 1565 |  | 
| Michael Foord | 3ab34cc | 2010-12-03 12:27:40 +0000 | [diff] [blame] | 1566 | def _make_temp_dir_for_build(TEMPDIR): | 
|  | 1567 | # When tests are run from the Python build directory, it is best practice | 
|  | 1568 | # to keep the test files in a subfolder.  It eases the cleanup of leftover | 
|  | 1569 | # files using command "make distclean". | 
|  | 1570 | if sysconfig.is_python_build(): | 
|  | 1571 | TEMPDIR = os.path.join(sysconfig.get_config_var('srcdir'), 'build') | 
|  | 1572 | TEMPDIR = os.path.abspath(TEMPDIR) | 
| Antoine Pitrou | ee42934 | 2011-04-16 18:53:59 +0200 | [diff] [blame] | 1573 | try: | 
| Michael Foord | 3ab34cc | 2010-12-03 12:27:40 +0000 | [diff] [blame] | 1574 | os.mkdir(TEMPDIR) | 
| Antoine Pitrou | ee42934 | 2011-04-16 18:53:59 +0200 | [diff] [blame] | 1575 | except OSError as e: | 
|  | 1576 | if e.errno != errno.EEXIST: | 
|  | 1577 | raise | 
| Michael Foord | 3ab34cc | 2010-12-03 12:27:40 +0000 | [diff] [blame] | 1578 |  | 
|  | 1579 | # Define a writable temp dir that will be used as cwd while running | 
|  | 1580 | # the tests. The name of the dir includes the pid to allow parallel | 
|  | 1581 | # testing (see the -j option). | 
|  | 1582 | TESTCWD = 'test_python_{}'.format(os.getpid()) | 
|  | 1583 |  | 
|  | 1584 | TESTCWD = os.path.join(TEMPDIR, TESTCWD) | 
|  | 1585 | return TEMPDIR, TESTCWD | 
| Nick Coghlan | 4c4c0f2 | 2010-12-03 07:44:33 +0000 | [diff] [blame] | 1586 |  | 
| Guido van Rossum | 152494a | 1996-12-20 03:12:20 +0000 | [diff] [blame] | 1587 | if __name__ == '__main__': | 
| Victor Stinner | 024e37a | 2011-03-31 01:31:06 +0200 | [diff] [blame] | 1588 | # Display the Python traceback on segfault and division by zero | 
|  | 1589 | faulthandler.enable() | 
|  | 1590 |  | 
| Nick Coghlan | 4c4c0f2 | 2010-12-03 07:44:33 +0000 | [diff] [blame] | 1591 | # Remove regrtest.py's own directory from the module search path. Despite | 
|  | 1592 | # the elimination of implicit relative imports, this is still needed to | 
|  | 1593 | # ensure that submodules of the test package do not inappropriately appear | 
|  | 1594 | # as top-level modules even when people (or buildbots!) invoke regrtest.py | 
|  | 1595 | # directly instead of using the -m switch | 
|  | 1596 | mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0]))) | 
|  | 1597 | i = len(sys.path) | 
|  | 1598 | while i >= 0: | 
|  | 1599 | i -= 1 | 
|  | 1600 | if os.path.abspath(os.path.normpath(sys.path[i])) == mydir: | 
|  | 1601 | del sys.path[i] | 
|  | 1602 |  | 
| Florent Xicluna | dc69e72 | 2010-09-13 16:35:02 +0000 | [diff] [blame] | 1603 | # findtestdir() gets the dirname out of __file__, so we have to make it | 
|  | 1604 | # absolute before changing the working directory. | 
|  | 1605 | # For example __file__ may be relative when running trace or profile. | 
|  | 1606 | # See issue #9323. | 
|  | 1607 | __file__ = os.path.abspath(__file__) | 
|  | 1608 |  | 
|  | 1609 | # sanity check | 
| Florent Xicluna | da7bfd5 | 2010-03-06 11:43:55 +0000 | [diff] [blame] | 1610 | assert __file__ == os.path.abspath(sys.argv[0]) | 
| Ezio Melotti | 184bdfb | 2010-02-18 09:37:05 +0000 | [diff] [blame] | 1611 |  | 
| Michael Foord | 3ab34cc | 2010-12-03 12:27:40 +0000 | [diff] [blame] | 1612 | TEMPDIR, TESTCWD = _make_temp_dir_for_build(TEMPDIR) | 
| Ezio Melotti | 184bdfb | 2010-02-18 09:37:05 +0000 | [diff] [blame] | 1613 |  | 
|  | 1614 | # Run the tests in a context manager that temporary changes the CWD to a | 
|  | 1615 | # temporary and writable directory. If it's not possible to create or | 
|  | 1616 | # change the CWD, the original CWD will be used. The original CWD is | 
|  | 1617 | # available from support.SAVEDCWD. | 
|  | 1618 | with support.temp_cwd(TESTCWD, quiet=True): | 
|  | 1619 | main() |