blob: 7580fb5e9b174f22c0caece367e81382a003c435 [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001:mod:`test` --- Regression tests package for Python
2===================================================
3
4.. module:: test
5 :synopsis: Regression tests package containing the testing suite for Python.
Terry Jan Reedyfa089b92016-06-11 15:02:54 -04006
Georg Brandl116aa622007-08-15 14:28:22 +00007.. sectionauthor:: Brett Cannon <brett@python.org>
8
Antoine Pitrou197c9c92010-12-18 12:33:06 +00009.. note::
Antoine Pitrou36730e82010-12-12 18:25:25 +000010 The :mod:`test` package is meant for internal use by Python only. It is
11 documented for the benefit of the core developers of Python. Any use of
12 this package outside of Python's standard library is discouraged as code
13 mentioned here can change or be removed without notice between releases of
14 Python.
Brett Cannon3a4e50c2010-07-23 11:31:31 +000015
Terry Jan Reedyfa089b92016-06-11 15:02:54 -040016--------------
Georg Brandl116aa622007-08-15 14:28:22 +000017
18The :mod:`test` package contains all regression tests for Python as well as the
Nick Coghlan47384702009-04-22 16:13:36 +000019modules :mod:`test.support` and :mod:`test.regrtest`.
20:mod:`test.support` is used to enhance your tests while
Georg Brandl116aa622007-08-15 14:28:22 +000021:mod:`test.regrtest` drives the testing suite.
22
23Each module in the :mod:`test` package whose name starts with ``test_`` is a
24testing suite for a specific module or feature. All new tests should be written
25using the :mod:`unittest` or :mod:`doctest` module. Some older tests are
26written using a "traditional" testing style that compares output printed to
27``sys.stdout``; this style of test is considered deprecated.
28
29
30.. seealso::
31
32 Module :mod:`unittest`
33 Writing PyUnit regression tests.
34
35 Module :mod:`doctest`
36 Tests embedded in documentation strings.
37
38
39.. _writing-tests:
40
41Writing Unit Tests for the :mod:`test` package
42----------------------------------------------
43
Georg Brandl116aa622007-08-15 14:28:22 +000044It is preferred that tests that use the :mod:`unittest` module follow a few
45guidelines. One is to name the test module by starting it with ``test_`` and end
46it with the name of the module being tested. The test methods in the test module
47should start with ``test_`` and end with a description of what the method is
48testing. This is needed so that the methods are recognized by the test driver as
49test methods. Also, no documentation string for the method should be included. A
50comment (such as ``# Tests function returns only True or False``) should be used
51to provide documentation for test methods. This is done because documentation
52strings get printed out if they exist and thus what test is being run is not
53stated.
54
55A basic boilerplate is often used::
56
57 import unittest
Nick Coghlan47384702009-04-22 16:13:36 +000058 from test import support
Georg Brandl116aa622007-08-15 14:28:22 +000059
60 class MyTestCase1(unittest.TestCase):
61
62 # Only use setUp() and tearDown() if necessary
63
64 def setUp(self):
65 ... code to execute in preparation for tests ...
66
67 def tearDown(self):
68 ... code to execute to clean up after tests ...
69
70 def test_feature_one(self):
71 # Test feature one.
72 ... testing code ...
73
74 def test_feature_two(self):
75 # Test feature two.
76 ... testing code ...
77
78 ... more test methods ...
79
80 class MyTestCase2(unittest.TestCase):
81 ... same structure as MyTestCase1 ...
82
83 ... more test classes ...
84
Georg Brandl116aa622007-08-15 14:28:22 +000085 if __name__ == '__main__':
R David Murray78fc25c2012-04-09 08:55:42 -040086 unittest.main()
Georg Brandl116aa622007-08-15 14:28:22 +000087
R David Murray78fc25c2012-04-09 08:55:42 -040088This code pattern allows the testing suite to be run by :mod:`test.regrtest`,
89on its own as a script that supports the :mod:`unittest` CLI, or via the
Georg Brandl93a56cd2014-10-30 22:25:41 +010090``python -m unittest`` CLI.
Georg Brandl116aa622007-08-15 14:28:22 +000091
92The goal for regression testing is to try to break code. This leads to a few
93guidelines to be followed:
94
95* The testing suite should exercise all classes, functions, and constants. This
Florent Xicluna53b506be2010-03-18 20:00:57 +000096 includes not just the external API that is to be presented to the outside
97 world but also "private" code.
Georg Brandl116aa622007-08-15 14:28:22 +000098
99* Whitebox testing (examining the code being tested when the tests are being
100 written) is preferred. Blackbox testing (testing only the published user
Florent Xicluna53b506be2010-03-18 20:00:57 +0000101 interface) is not complete enough to make sure all boundary and edge cases
102 are tested.
Georg Brandl116aa622007-08-15 14:28:22 +0000103
104* Make sure all possible values are tested including invalid ones. This makes
Florent Xicluna53b506be2010-03-18 20:00:57 +0000105 sure that not only all valid values are acceptable but also that improper
106 values are handled correctly.
Georg Brandl116aa622007-08-15 14:28:22 +0000107
108* Exhaust as many code paths as possible. Test where branching occurs and thus
109 tailor input to make sure as many different paths through the code are taken.
110
111* Add an explicit test for any bugs discovered for the tested code. This will
112 make sure that the error does not crop up again if the code is changed in the
113 future.
114
115* Make sure to clean up after your tests (such as close and remove all temporary
116 files).
117
118* If a test is dependent on a specific condition of the operating system then
119 verify the condition already exists before attempting the test.
120
121* Import as few modules as possible and do it as soon as possible. This
122 minimizes external dependencies of tests and also minimizes possible anomalous
123 behavior from side-effects of importing a module.
124
125* Try to maximize code reuse. On occasion, tests will vary by something as small
Florent Xicluna53b506be2010-03-18 20:00:57 +0000126 as what type of input is used. Minimize code duplication by subclassing a
127 basic test class with a class that specifies the input::
Georg Brandl116aa622007-08-15 14:28:22 +0000128
R David Murray78fc25c2012-04-09 08:55:42 -0400129 class TestFuncAcceptsSequencesMixin:
Georg Brandl116aa622007-08-15 14:28:22 +0000130
131 func = mySuperWhammyFunction
132
133 def test_func(self):
134 self.func(self.arg)
135
R David Murray78fc25c2012-04-09 08:55:42 -0400136 class AcceptLists(TestFuncAcceptsSequencesMixin, unittest.TestCase):
Florent Xiclunab14930c2010-03-13 15:26:44 +0000137 arg = [1, 2, 3]
Georg Brandl116aa622007-08-15 14:28:22 +0000138
R David Murray78fc25c2012-04-09 08:55:42 -0400139 class AcceptStrings(TestFuncAcceptsSequencesMixin, unittest.TestCase):
Georg Brandl116aa622007-08-15 14:28:22 +0000140 arg = 'abc'
141
R David Murray78fc25c2012-04-09 08:55:42 -0400142 class AcceptTuples(TestFuncAcceptsSequencesMixin, unittest.TestCase):
Florent Xiclunab14930c2010-03-13 15:26:44 +0000143 arg = (1, 2, 3)
Georg Brandl116aa622007-08-15 14:28:22 +0000144
R David Murray78fc25c2012-04-09 08:55:42 -0400145 When using this pattern, remember that all classes that inherit from
Georg Brandl93a56cd2014-10-30 22:25:41 +0100146 :class:`unittest.TestCase` are run as tests. The :class:`Mixin` class in the example above
R David Murray78fc25c2012-04-09 08:55:42 -0400147 does not have any data and so can't be run by itself, thus it does not
Georg Brandl93a56cd2014-10-30 22:25:41 +0100148 inherit from :class:`unittest.TestCase`.
R David Murray78fc25c2012-04-09 08:55:42 -0400149
Georg Brandl116aa622007-08-15 14:28:22 +0000150
151.. seealso::
152
153 Test Driven Development
154 A book by Kent Beck on writing tests before code.
155
156
157.. _regrtest:
158
Éric Araujo1d55c7e2010-12-16 01:40:26 +0000159Running tests using the command-line interface
160----------------------------------------------
Georg Brandl116aa622007-08-15 14:28:22 +0000161
Éric Araujo1d55c7e2010-12-16 01:40:26 +0000162The :mod:`test` package can be run as a script to drive Python's regression
163test suite, thanks to the :option:`-m` option: :program:`python -m test`. Under
164the hood, it uses :mod:`test.regrtest`; the call :program:`python -m
Martin Panter9955a372015-10-07 10:26:23 +0000165test.regrtest` used in previous Python versions still works. Running the
R David Murrayf1cdb602012-04-09 09:12:57 -0400166script by itself automatically starts running all regression tests in the
167:mod:`test` package. It does this by finding all modules in the package whose
168name starts with ``test_``, importing them, and executing the function
169:func:`test_main` if present or loading the tests via
170unittest.TestLoader.loadTestsFromModule if ``test_main`` does not exist. The
171names of tests to execute may also be passed to the script. Specifying a single
172regression test (:program:`python -m test test_spam`) will minimize output and
173only print whether the test passed or failed.
Georg Brandl116aa622007-08-15 14:28:22 +0000174
Éric Araujo1d55c7e2010-12-16 01:40:26 +0000175Running :mod:`test` directly allows what resources are available for
Éric Araujo713d3032010-11-18 16:38:46 +0000176tests to use to be set. You do this by using the ``-u`` command-line
R David Murray644cabe2012-04-11 20:11:53 -0400177option. Specifying ``all`` as the value for the ``-u`` option enables all
178possible resources: :program:`python -m test -uall`.
179If all but one resource is desired (a more common case), a
Georg Brandl116aa622007-08-15 14:28:22 +0000180comma-separated list of resources that are not desired may be listed after
Éric Araujo1d55c7e2010-12-16 01:40:26 +0000181``all``. The command :program:`python -m test -uall,-audio,-largefile`
182will run :mod:`test` with all resources except the ``audio`` and
Éric Araujo713d3032010-11-18 16:38:46 +0000183``largefile`` resources. For a list of all resources and more command-line
Éric Araujo1d55c7e2010-12-16 01:40:26 +0000184options, run :program:`python -m test -h`.
Georg Brandl116aa622007-08-15 14:28:22 +0000185
186Some other ways to execute the regression tests depend on what platform the
Éric Araujo713d3032010-11-18 16:38:46 +0000187tests are being executed on. On Unix, you can run :program:`make test` at the
188top-level directory where Python was built. On Windows,
Stefan Grönkef1502d02017-09-25 18:58:10 +0200189executing :program:`rt.bat` from your :file:`PCbuild` directory will run all
Florent Xicluna53b506be2010-03-18 20:00:57 +0000190regression tests.
Georg Brandl116aa622007-08-15 14:28:22 +0000191
192
Georg Brandleea6cda2011-07-30 09:00:48 +0200193:mod:`test.support` --- Utilities for the Python test suite
194===========================================================
Georg Brandl116aa622007-08-15 14:28:22 +0000195
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000196.. module:: test.support
Georg Brandleea6cda2011-07-30 09:00:48 +0200197 :synopsis: Support for Python's regression test suite.
Georg Brandl116aa622007-08-15 14:28:22 +0000198
199
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000200The :mod:`test.support` module provides support for Python's regression
Georg Brandleea6cda2011-07-30 09:00:48 +0200201test suite.
202
203.. note::
Larry Hastings3732ed22014-03-15 21:13:56 -0700204
Georg Brandleea6cda2011-07-30 09:00:48 +0200205 :mod:`test.support` is not a public module. It is documented here to help
206 Python developers write tests. The API of this module is subject to change
207 without backwards compatibility concerns between releases.
208
Georg Brandl116aa622007-08-15 14:28:22 +0000209
210This module defines the following exceptions:
211
Georg Brandl116aa622007-08-15 14:28:22 +0000212.. exception:: TestFailed
213
214 Exception to be raised when a test fails. This is deprecated in favor of
215 :mod:`unittest`\ -based tests and :class:`unittest.TestCase`'s assertion
216 methods.
217
218
Georg Brandl116aa622007-08-15 14:28:22 +0000219.. exception:: ResourceDenied
220
Florent Xicluna53b506be2010-03-18 20:00:57 +0000221 Subclass of :exc:`unittest.SkipTest`. Raised when a resource (such as a
222 network connection) is not available. Raised by the :func:`requires`
223 function.
Georg Brandl116aa622007-08-15 14:28:22 +0000224
Georg Brandl116aa622007-08-15 14:28:22 +0000225
Georg Brandleea6cda2011-07-30 09:00:48 +0200226The :mod:`test.support` module defines the following constants:
Georg Brandl116aa622007-08-15 14:28:22 +0000227
228.. data:: verbose
229
Eli Benderskye1689652011-05-06 09:29:27 +0300230 ``True`` when verbose output is enabled. Should be checked when more
Georg Brandl116aa622007-08-15 14:28:22 +0000231 detailed information is desired about a running test. *verbose* is set by
232 :mod:`test.regrtest`.
233
234
Georg Brandl116aa622007-08-15 14:28:22 +0000235.. data:: is_jython
236
Eli Benderskye1689652011-05-06 09:29:27 +0300237 ``True`` if the running interpreter is Jython.
Georg Brandl116aa622007-08-15 14:28:22 +0000238
239
Cheryl Sabella988fb282018-02-11 08:10:42 -0500240.. data:: is_android
241
242 ``True`` if the system is Android.
243
244
245.. data:: unix_shell
246
247 Path for shell if not on Windows; otherwise ``None``.
248
249
250.. data:: FS_NONASCII
251
252 A non-ASCII character encodable by :func:`os.fsencode`.
253
254
Georg Brandl116aa622007-08-15 14:28:22 +0000255.. data:: TESTFN
256
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000257 Set to a name that is safe to use as the name of a temporary file. Any
258 temporary file that is created should be closed and unlinked (removed).
Georg Brandl116aa622007-08-15 14:28:22 +0000259
Eli Benderskye1689652011-05-06 09:29:27 +0300260
Cheryl Sabella988fb282018-02-11 08:10:42 -0500261.. data:: TESTFN_UNICODE
262
263 Set to a non-ASCII name for a temporary file.
264
265
Cheryl Sabella988fb282018-02-11 08:10:42 -0500266.. data:: TESTFN_UNENCODABLE
267
268 Set to a filename (str type) that should not be able to be encoded by file
269 system encoding in strict mode. It may be ``None`` if it's not possible to
270 generate such a filename.
271
272
273.. data:: TESTFN_UNDECODABLE
274
275 Set to a filename (bytes type) that should not be able to be decoded by
276 file system encoding in strict mode. It may be ``None`` if it's not
277 possible to generate such a filename.
278
279
280.. data:: TESTFN_NONASCII
281
282 Set to a filename containing the :data:`FS_NONASCII` character.
283
284
Victor Stinner24c62582019-10-30 12:41:43 +0100285.. data:: LOOPBACK_TIMEOUT
286
287 Timeout in seconds for tests using a network server listening on the network
288 local loopback interface like ``127.0.0.1``.
289
290 The timeout is long enough to prevent test failure: it takes into account
291 that the client and the server can run in different threads or even
292 different processes.
293
294 The timeout should be long enough for :meth:`~socket.socket.connect`,
295 :meth:`~socket.socket.recv` and :meth:`~socket.socket.send` methods of
296 :class:`socket.socket`.
297
298 Its default value is 5 seconds.
299
300 See also :data:`INTERNET_TIMEOUT`.
301
302
303.. data:: INTERNET_TIMEOUT
304
305 Timeout in seconds for network requests going to the Internet.
306
307 The timeout is short enough to prevent a test to wait for too long if the
308 Internet request is blocked for whatever reason.
309
310 Usually, a timeout using :data:`INTERNET_TIMEOUT` should not mark a test as
311 failed, but skip the test instead: see
Serhiy Storchakabfb1cf42020-04-29 10:36:20 +0300312 :func:`~test.support.socket_helper.transient_internet`.
Victor Stinner24c62582019-10-30 12:41:43 +0100313
314 Its default value is 1 minute.
315
316 See also :data:`LOOPBACK_TIMEOUT`.
317
318
319.. data:: SHORT_TIMEOUT
320
321 Timeout in seconds to mark a test as failed if the test takes "too long".
322
323 The timeout value depends on the regrtest ``--timeout`` command line option.
324
325 If a test using :data:`SHORT_TIMEOUT` starts to fail randomly on slow
326 buildbots, use :data:`LONG_TIMEOUT` instead.
327
328 Its default value is 30 seconds.
329
330
331.. data:: LONG_TIMEOUT
332
333 Timeout in seconds to detect when a test hangs.
334
335 It is long enough to reduce the risk of test failure on the slowest Python
336 buildbots. It should not be used to mark a test as failed if the test takes
337 "too long". The timeout value depends on the regrtest ``--timeout`` command
338 line option.
339
340 Its default value is 5 minutes.
341
342 See also :data:`LOOPBACK_TIMEOUT`, :data:`INTERNET_TIMEOUT` and
343 :data:`SHORT_TIMEOUT`.
344
345
Cheryl Sabella988fb282018-02-11 08:10:42 -0500346.. data:: SAVEDCWD
347
348 Set to :func:`os.getcwd`.
349
350
351.. data:: PGO
352
353 Set when tests can be skipped when they are not useful for PGO.
354
355
356.. data:: PIPE_MAX_SIZE
357
358 A constant that is likely larger than the underlying OS pipe buffer size,
359 to make writes blocking.
360
361
362.. data:: SOCK_MAX_SIZE
363
364 A constant that is likely larger than the underlying OS socket buffer size,
365 to make writes blocking.
366
367
368.. data:: TEST_SUPPORT_DIR
369
370 Set to the top level directory that contains :mod:`test.support`.
371
372
373.. data:: TEST_HOME_DIR
374
375 Set to the top level directory for the test package.
376
377
378.. data:: TEST_DATA_DIR
379
380 Set to the ``data`` directory within the test package.
381
382
383.. data:: MAX_Py_ssize_t
384
385 Set to :data:`sys.maxsize` for big memory tests.
386
387
388.. data:: max_memuse
389
390 Set by :func:`set_memlimit` as the memory limit for big memory tests.
391 Limited by :data:`MAX_Py_ssize_t`.
392
393
394.. data:: real_max_memuse
395
396 Set by :func:`set_memlimit` as the memory limit for big memory tests. Not
397 limited by :data:`MAX_Py_ssize_t`.
398
399
400.. data:: MISSING_C_DOCSTRINGS
401
402 Return ``True`` if running on CPython, not on Windows, and configuration
403 not set with ``WITH_DOC_STRINGS``.
404
405
406.. data:: HAVE_DOCSTRINGS
407
408 Check for presence of docstrings.
409
Serhiy Storchaka17e52642019-08-04 12:38:46 +0300410
Stéphane Wirtela40681d2019-02-22 14:45:36 +0100411.. data:: TEST_HTTP_URL
412
413 Define the URL of a dedicated HTTP server for the network tests.
414
Cheryl Sabella988fb282018-02-11 08:10:42 -0500415
Serhiy Storchaka17e52642019-08-04 12:38:46 +0300416.. data:: ALWAYS_EQ
417
418 Object that is equal to anything. Used to test mixed type comparison.
419
420
Serhiy Storchaka18b711c2019-08-04 14:12:48 +0300421.. data:: NEVER_EQ
422
423 Object that is not equal to anything (even to :data:`ALWAYS_EQ`).
424 Used to test mixed type comparison.
425
426
Serhiy Storchaka17e52642019-08-04 12:38:46 +0300427.. data:: LARGEST
428
429 Object that is greater than anything (except itself).
430 Used to test mixed type comparison.
431
432
433.. data:: SMALLEST
434
435 Object that is less than anything (except itself).
436 Used to test mixed type comparison.
437
Cheryl Sabella988fb282018-02-11 08:10:42 -0500438
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000439The :mod:`test.support` module defines the following functions:
Georg Brandl116aa622007-08-15 14:28:22 +0000440
Georg Brandl116aa622007-08-15 14:28:22 +0000441.. function:: forget(module_name)
442
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000443 Remove the module named *module_name* from ``sys.modules`` and delete any
Georg Brandl116aa622007-08-15 14:28:22 +0000444 byte-compiled files of the module.
445
446
Cheryl Sabella988fb282018-02-11 08:10:42 -0500447.. function:: unload(name)
448
449 Delete *name* from ``sys.modules``.
450
451
452.. function:: unlink(filename)
453
454 Call :func:`os.unlink` on *filename*. On Windows platforms, this is
455 wrapped with a wait loop that checks for the existence fo the file.
456
457
458.. function:: rmdir(filename)
459
460 Call :func:`os.rmdir` on *filename*. On Windows platforms, this is
461 wrapped with a wait loop that checks for the existence of the file.
462
463
464.. function:: rmtree(path)
465
466 Call :func:`shutil.rmtree` on *path* or call :func:`os.lstat` and
467 :func:`os.rmdir` to remove a path and its contents. On Windows platforms,
468 this is wrapped with a wait loop that checks for the existence of the files.
469
470
471.. function:: make_legacy_pyc(source)
472
Stéphane Wirtel12e696b2018-10-27 00:58:26 +0200473 Move a :pep:`3147`/:pep:`488` pyc file to its legacy pyc location and return the file
Cheryl Sabella988fb282018-02-11 08:10:42 -0500474 system path to the legacy pyc file. The *source* value is the file system
475 path to the source file. It does not need to exist, however the PEP
476 3147/488 pyc file must exist.
477
478
Georg Brandl116aa622007-08-15 14:28:22 +0000479.. function:: is_resource_enabled(resource)
480
Eli Benderskye1689652011-05-06 09:29:27 +0300481 Return ``True`` if *resource* is enabled and available. The list of
Georg Brandl116aa622007-08-15 14:28:22 +0000482 available resources is only set when :mod:`test.regrtest` is executing the
483 tests.
484
485
Cheryl Sabella988fb282018-02-11 08:10:42 -0500486.. function:: python_is_optimized()
487
488 Return ``True`` if Python was not built with ``-O0`` or ``-Og``.
489
490
491.. function:: with_pymalloc()
492
493 Return :data:`_testcapi.WITH_PYMALLOC`.
494
495
Georg Brandl7f01a132009-09-16 15:58:14 +0000496.. function:: requires(resource, msg=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000497
Florent Xiclunab14930c2010-03-13 15:26:44 +0000498 Raise :exc:`ResourceDenied` if *resource* is not available. *msg* is the
Florent Xicluna53b506be2010-03-18 20:00:57 +0000499 argument to :exc:`ResourceDenied` if it is raised. Always returns
Eli Benderskye1689652011-05-06 09:29:27 +0300500 ``True`` if called by a function whose ``__name__`` is ``'__main__'``.
Florent Xicluna53b506be2010-03-18 20:00:57 +0000501 Used when tests are executed by :mod:`test.regrtest`.
Georg Brandl116aa622007-08-15 14:28:22 +0000502
503
Cheryl Sabella988fb282018-02-11 08:10:42 -0500504.. function:: system_must_validate_cert(f)
505
506 Raise :exc:`unittest.SkipTest` on TLS certification validation failures.
507
508
509.. function:: sortdict(dict)
510
511 Return a repr of *dict* with keys sorted.
512
513
Nick Coghlan0494c2a2013-09-08 11:40:34 +1000514.. function:: findfile(filename, subdir=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000515
Florent Xicluna53b506be2010-03-18 20:00:57 +0000516 Return the path to the file named *filename*. If no match is found
517 *filename* is returned. This does not equal a failure since it could be the
518 path to the file.
Georg Brandl116aa622007-08-15 14:28:22 +0000519
Cheryl Sabella988fb282018-02-11 08:10:42 -0500520 Setting *subdir* indicates a relative path to use to find the file
521 rather than looking directly in the path directories.
522
523
524.. function:: create_empty_file(filename)
525
526 Create an empty file with *filename*. If it already exists, truncate it.
527
528
529.. function:: fd_count()
530
531 Count the number of open file descriptors.
532
533
534.. function:: match_test(test)
535
536 Match *test* to patterns set in :func:`set_match_tests`.
537
538
539.. function:: set_match_tests(patterns)
540
541 Define match test with regular expression *patterns*.
Nick Coghlan0494c2a2013-09-08 11:40:34 +1000542
Georg Brandl116aa622007-08-15 14:28:22 +0000543
Senthil Kumaran279b56d2010-10-15 15:21:19 +0000544.. function:: run_unittest(\*classes)
Georg Brandl116aa622007-08-15 14:28:22 +0000545
546 Execute :class:`unittest.TestCase` subclasses passed to the function. The
Florent Xicluna53b506be2010-03-18 20:00:57 +0000547 function scans the classes for methods starting with the prefix ``test_``
548 and executes the tests individually.
Georg Brandl116aa622007-08-15 14:28:22 +0000549
550 It is also legal to pass strings as parameters; these should be keys in
551 ``sys.modules``. Each associated module will be scanned by
552 ``unittest.TestLoader.loadTestsFromModule()``. This is usually seen in the
553 following :func:`test_main` function::
554
555 def test_main():
Nick Coghlan47384702009-04-22 16:13:36 +0000556 support.run_unittest(__name__)
Georg Brandl116aa622007-08-15 14:28:22 +0000557
558 This will run all tests defined in the named module.
559
Georg Brandl116aa622007-08-15 14:28:22 +0000560
Cheryl Sabella988fb282018-02-11 08:10:42 -0500561.. function:: run_doctest(module, verbosity=None, optionflags=0)
Eli Benderskye1689652011-05-06 09:29:27 +0300562
563 Run :func:`doctest.testmod` on the given *module*. Return
564 ``(failure_count, test_count)``.
565
566 If *verbosity* is ``None``, :func:`doctest.testmod` is run with verbosity
567 set to :data:`verbose`. Otherwise, it is run with verbosity set to
Cheryl Sabella988fb282018-02-11 08:10:42 -0500568 ``None``. *optionflags* is passed as ``optionflags`` to
569 :func:`doctest.testmod`.
570
571
572.. function:: setswitchinterval(interval)
573
574 Set the :func:`sys.setswitchinterval` to the given *interval*. Defines
575 a minimum interval for Android systems to prevent the system from hanging.
576
577
578.. function:: check_impl_detail(**guards)
579
580 Use this check to guard CPython's implementation-specific tests or to
581 run them only on the implementations guarded by the arguments::
582
583 check_impl_detail() # Only on CPython (default).
584 check_impl_detail(jython=True) # Only on Jython.
585 check_impl_detail(cpython=False) # Everywhere except CPython.
586
Eli Benderskye1689652011-05-06 09:29:27 +0300587
Senthil Kumaran279b56d2010-10-15 15:21:19 +0000588.. function:: check_warnings(\*filters, quiet=True)
Thomas Woutersed03b412007-08-28 21:37:11 +0000589
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000590 A convenience wrapper for :func:`warnings.catch_warnings()` that makes it
591 easier to test that a warning was correctly raised. It is approximately
592 equivalent to calling ``warnings.catch_warnings(record=True)`` with
593 :meth:`warnings.simplefilter` set to ``always`` and with the option to
594 automatically validate the results that are recorded.
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000595
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000596 ``check_warnings`` accepts 2-tuples of the form ``("message regexp",
597 WarningCategory)`` as positional arguments. If one or more *filters* are
Eli Benderskye1689652011-05-06 09:29:27 +0300598 provided, or if the optional keyword argument *quiet* is ``False``,
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000599 it checks to make sure the warnings are as expected: each specified filter
600 must match at least one of the warnings raised by the enclosed code or the
601 test fails, and if any warnings are raised that do not match any of the
602 specified filters the test fails. To disable the first of these checks,
Eli Benderskye1689652011-05-06 09:29:27 +0300603 set *quiet* to ``True``.
Florent Xiclunab14930c2010-03-13 15:26:44 +0000604
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000605 If no arguments are specified, it defaults to::
Florent Xiclunab14930c2010-03-13 15:26:44 +0000606
Florent Xicluna53b506be2010-03-18 20:00:57 +0000607 check_warnings(("", Warning), quiet=True)
Florent Xiclunab14930c2010-03-13 15:26:44 +0000608
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000609 In this case all warnings are caught and no errors are raised.
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000610
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000611 On entry to the context manager, a :class:`WarningRecorder` instance is
612 returned. The underlying warnings list from
613 :func:`~warnings.catch_warnings` is available via the recorder object's
614 :attr:`warnings` attribute. As a convenience, the attributes of the object
615 representing the most recent warning can also be accessed directly through
616 the recorder object (see example below). If no warning has been raised,
617 then any of the attributes that would otherwise be expected on an object
Eli Benderskye1689652011-05-06 09:29:27 +0300618 representing a warning will return ``None``.
Thomas Woutersed03b412007-08-28 21:37:11 +0000619
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000620 The recorder object also has a :meth:`reset` method, which clears the
621 warnings list.
Thomas Woutersed03b412007-08-28 21:37:11 +0000622
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000623 The context manager is designed to be used like this::
Florent Xiclunab14930c2010-03-13 15:26:44 +0000624
625 with check_warnings(("assertion is always true", SyntaxWarning),
626 ("", UserWarning)):
627 exec('assert(False, "Hey!")')
628 warnings.warn(UserWarning("Hide me!"))
629
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000630 In this case if either warning was not raised, or some other warning was
631 raised, :func:`check_warnings` would raise an error.
632
633 When a test needs to look more deeply into the warnings, rather than
634 just checking whether or not they occurred, code like this can be used::
635
Florent Xiclunab14930c2010-03-13 15:26:44 +0000636 with check_warnings(quiet=True) as w:
Thomas Woutersed03b412007-08-28 21:37:11 +0000637 warnings.warn("foo")
Florent Xiclunab14930c2010-03-13 15:26:44 +0000638 assert str(w.args[0]) == "foo"
Nick Coghlanb1304932008-07-13 12:25:08 +0000639 warnings.warn("bar")
Florent Xiclunab14930c2010-03-13 15:26:44 +0000640 assert str(w.args[0]) == "bar"
641 assert str(w.warnings[0].args[0]) == "foo"
642 assert str(w.warnings[1].args[0]) == "bar"
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000643 w.reset()
644 assert len(w.warnings) == 0
Thomas Woutersed03b412007-08-28 21:37:11 +0000645
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000646
647 Here all warnings will be caught, and the test code tests the captured
648 warnings directly.
649
Ezio Melottif8754a62010-03-21 07:16:43 +0000650 .. versionchanged:: 3.2
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000651 New optional arguments *filters* and *quiet*.
Florent Xiclunab14930c2010-03-13 15:26:44 +0000652
Thomas Woutersed03b412007-08-28 21:37:11 +0000653
Cheryl Sabella988fb282018-02-11 08:10:42 -0500654.. function:: check_no_resource_warning(testcase)
655
656 Context manager to check that no :exc:`ResourceWarning` was raised. You
657 must remove the object which may emit :exc:`ResourceWarning` before the
658 end of the context manager.
659
660
661.. function:: set_memlimit(limit)
662
663 Set the values for :data:`max_memuse` and :data:`real_max_memuse` for big
664 memory tests.
665
666
667.. function:: record_original_stdout(stdout)
668
669 Store the value from *stdout*. It is meant to hold the stdout at the
670 time the regrtest began.
671
672
673.. function:: get_original_stdout
674
675 Return the original stdout set by :func:`record_original_stdout` or
676 ``sys.stdout`` if it's not set.
677
678
Cheryl Sabella988fb282018-02-11 08:10:42 -0500679.. function:: args_from_interpreter_flags()
680
681 Return a list of command line arguments reproducing the current settings
682 in ``sys.flags`` and ``sys.warnoptions``.
683
684
685.. function:: optim_args_from_interpreter_flags()
686
687 Return a list of command line arguments reproducing the current
688 optimization settings in ``sys.flags``.
689
690
R David Murray5a33f812013-07-11 12:28:40 -0400691.. function:: captured_stdin()
692 captured_stdout()
693 captured_stderr()
Thomas Woutersed03b412007-08-28 21:37:11 +0000694
R David Murray5a33f812013-07-11 12:28:40 -0400695 A context managers that temporarily replaces the named stream with
696 :class:`io.StringIO` object.
Thomas Woutersed03b412007-08-28 21:37:11 +0000697
R David Murray5a33f812013-07-11 12:28:40 -0400698 Example use with output streams::
Thomas Woutersed03b412007-08-28 21:37:11 +0000699
R David Murray5a33f812013-07-11 12:28:40 -0400700 with captured_stdout() as stdout, captured_stderr() as stderr:
Collin Winterc79461b2007-09-01 23:34:30 +0000701 print("hello")
R David Murray5a33f812013-07-11 12:28:40 -0400702 print("error", file=sys.stderr)
703 assert stdout.getvalue() == "hello\n"
704 assert stderr.getvalue() == "error\n"
705
706 Example use with input stream::
707
708 with captured_stdin() as stdin:
709 stdin.write('hello\n')
710 stdin.seek(0)
711 # call test code that consumes from sys.stdin
712 captured = input()
713 self.assertEqual(captured, "hello")
Thomas Woutersed03b412007-08-28 21:37:11 +0000714
Thomas Woutersed03b412007-08-28 21:37:11 +0000715
Nick Coghlan55175962013-07-28 22:11:50 +1000716.. function:: temp_dir(path=None, quiet=False)
717
718 A context manager that creates a temporary directory at *path* and
719 yields the directory.
720
Serhiy Storchakaecf41da2016-10-19 16:29:26 +0300721 If *path* is ``None``, the temporary directory is created using
Nick Coghlan55175962013-07-28 22:11:50 +1000722 :func:`tempfile.mkdtemp`. If *quiet* is ``False``, the context manager
723 raises an exception on error. Otherwise, if *path* is specified and
724 cannot be created, only a warning is issued.
725
726
727.. function:: change_cwd(path, quiet=False)
Eli Benderskye1689652011-05-06 09:29:27 +0300728
729 A context manager that temporarily changes the current working
Nick Coghlan55175962013-07-28 22:11:50 +1000730 directory to *path* and yields the directory.
Eli Benderskye1689652011-05-06 09:29:27 +0300731
Nick Coghlan55175962013-07-28 22:11:50 +1000732 If *quiet* is ``False``, the context manager raises an exception
733 on error. Otherwise, it issues only a warning and keeps the current
734 working directory the same.
Eli Benderskye1689652011-05-06 09:29:27 +0300735
Nick Coghlan55175962013-07-28 22:11:50 +1000736
737.. function:: temp_cwd(name='tempcwd', quiet=False)
738
739 A context manager that temporarily creates a new directory and
740 changes the current working directory (CWD).
741
742 The context manager creates a temporary directory in the current
743 directory with name *name* before temporarily changing the current
Serhiy Storchakaecf41da2016-10-19 16:29:26 +0300744 working directory. If *name* is ``None``, the temporary directory is
Nick Coghlan55175962013-07-28 22:11:50 +1000745 created using :func:`tempfile.mkdtemp`.
746
747 If *quiet* is ``False`` and it is not possible to create or change
748 the CWD, an error is raised. Otherwise, only a warning is raised
749 and the original CWD is used.
Eli Benderskye1689652011-05-06 09:29:27 +0300750
751
752.. function:: temp_umask(umask)
753
754 A context manager that temporarily sets the process umask.
755
756
Cheryl Sabella988fb282018-02-11 08:10:42 -0500757.. function:: disable_faulthandler()
758
759 A context manager that replaces ``sys.stderr`` with ``sys.__stderr__``.
760
761
762.. function:: gc_collect()
763
764 Force as many objects as possible to be collected. This is needed because
765 timely deallocation is not guaranteed by the garbage collector. This means
766 that ``__del__`` methods may be called later than expected and weakrefs
767 may remain alive for longer than expected.
768
769
770.. function:: disable_gc()
771
772 A context manager that disables the garbage collector upon entry and
773 reenables it upon exit.
774
775
776.. function:: swap_attr(obj, attr, new_val)
777
778 Context manager to swap out an attribute with a new object.
779
780 Usage::
781
782 with swap_attr(obj, "attr", 5):
783 ...
784
785 This will set ``obj.attr`` to 5 for the duration of the ``with`` block,
786 restoring the old value at the end of the block. If ``attr`` doesn't
787 exist on ``obj``, it will be created and then deleted at the end of the
788 block.
789
790 The old value (or ``None`` if it doesn't exist) will be assigned to the
791 target of the "as" clause, if there is one.
792
793
794.. function:: swap_item(obj, attr, new_val)
795
796 Context manager to swap out an item with a new object.
797
798 Usage::
799
800 with swap_item(obj, "item", 5):
801 ...
802
803 This will set ``obj["item"]`` to 5 for the duration of the ``with`` block,
804 restoring the old value at the end of the block. If ``item`` doesn't
805 exist on ``obj``, it will be created and then deleted at the end of the
806 block.
807
808 The old value (or ``None`` if it doesn't exist) will be assigned to the
809 target of the "as" clause, if there is one.
810
811
Victor Stinnerd663d342020-04-23 19:03:52 +0200812.. function:: print_warning(msg)
813
814 Print a warning into :data:`sys.__stderr__`. Format the message as:
815 ``f"Warning -- {msg}"``. If *msg* is made of multiple lines, add
816 ``"Warning -- "`` prefix to each line.
817
818 .. versionadded:: 3.9
819
820
Victor Stinner278c1e12020-03-31 20:08:12 +0200821.. function:: wait_process(pid, *, exitcode, timeout=None)
822
823 Wait until process *pid* completes and check that the process exit code is
824 *exitcode*.
825
826 Raise an :exc:`AssertionError` if the process exit code is not equal to
827 *exitcode*.
828
829 If the process runs longer than *timeout* seconds (:data:`SHORT_TIMEOUT` by
830 default), kill the process and raise an :exc:`AssertionError`. The timeout
831 feature is not available on Windows.
832
833 .. versionadded:: 3.9
834
835
Cheryl Sabella988fb282018-02-11 08:10:42 -0500836.. function:: calcobjsize(fmt)
837
838 Return :func:`struct.calcsize` for ``nP{fmt}0n`` or, if ``gettotalrefcount``
839 exists, ``2PnP{fmt}0P``.
840
841
842.. function:: calcvobjsize(fmt)
843
844 Return :func:`struct.calcsize` for ``nPn{fmt}0n`` or, if ``gettotalrefcount``
845 exists, ``2PnPn{fmt}0P``.
846
847
848.. function:: checksizeof(test, o, size)
849
850 For testcase *test*, assert that the ``sys.getsizeof`` for *o* plus the GC
851 header size equals *size*.
852
853
Eli Benderskye1689652011-05-06 09:29:27 +0300854.. function:: can_symlink()
855
856 Return ``True`` if the OS supports symbolic links, ``False``
857 otherwise.
858
859
Cheryl Sabella988fb282018-02-11 08:10:42 -0500860.. function:: can_xattr()
861
862 Return ``True`` if the OS supports xattr, ``False``
863 otherwise.
864
865
Daisuke Miyakawa0e61e672017-10-12 23:39:43 +0900866.. decorator:: skip_unless_symlink
Eli Benderskye1689652011-05-06 09:29:27 +0300867
868 A decorator for running tests that require support for symbolic links.
869
870
Cheryl Sabella988fb282018-02-11 08:10:42 -0500871.. decorator:: skip_unless_xattr
872
873 A decorator for running tests that require support for xattr.
874
875
Nick Coghlan2496f332011-09-19 20:26:31 +1000876.. decorator:: anticipate_failure(condition)
877
878 A decorator to conditionally mark tests with
879 :func:`unittest.expectedFailure`. Any use of this decorator should
880 have an associated comment identifying the relevant tracker issue.
881
882
883.. decorator:: run_with_locale(catstr, *locales)
Eli Benderskye1689652011-05-06 09:29:27 +0300884
885 A decorator for running a function in a different locale, correctly
886 resetting it after it has finished. *catstr* is the locale category as
887 a string (for example ``"LC_ALL"``). The *locales* passed will be tried
888 sequentially, and the first valid locale will be used.
889
890
Cheryl Sabella988fb282018-02-11 08:10:42 -0500891.. decorator:: run_with_tz(tz)
892
893 A decorator for running a function in a specific timezone, correctly
894 resetting it after it has finished.
895
896
897.. decorator:: requires_freebsd_version(*min_version)
898
899 Decorator for the minimum version when running test on FreeBSD. If the
900 FreeBSD version is less than the minimum, raise :exc:`unittest.SkipTest`.
901
902
903.. decorator:: requires_linux_version(*min_version)
904
905 Decorator for the minimum version when running test on Linux. If the
906 Linux version is less than the minimum, raise :exc:`unittest.SkipTest`.
907
908
909.. decorator:: requires_mac_version(*min_version)
910
911 Decorator for the minimum version when running test on Mac OS X. If the
912 MAC OS X version is less than the minimum, raise :exc:`unittest.SkipTest`.
913
914
915.. decorator:: requires_IEEE_754
916
917 Decorator for skipping tests on non-IEEE 754 platforms.
918
919
920.. decorator:: requires_zlib
921
922 Decorator for skipping tests if :mod:`zlib` doesn't exist.
923
924
925.. decorator:: requires_gzip
926
927 Decorator for skipping tests if :mod:`gzip` doesn't exist.
928
929
930.. decorator:: requires_bz2
931
932 Decorator for skipping tests if :mod:`bz2` doesn't exist.
933
934
935.. decorator:: requires_lzma
936
937 Decorator for skipping tests if :mod:`lzma` doesn't exist.
938
939
940.. decorator:: requires_resource(resource)
941
942 Decorator for skipping tests if *resource* is not available.
943
944
945.. decorator:: requires_docstrings
946
947 Decorator for only running the test if :data:`HAVE_DOCSTRINGS`.
948
949
950.. decorator:: cpython_only(test)
951
952 Decorator for tests only applicable to CPython.
953
954
955.. decorator:: impl_detail(msg=None, **guards)
956
957 Decorator for invoking :func:`check_impl_detail` on *guards*. If that
958 returns ``False``, then uses *msg* as the reason for skipping the test.
959
960
961.. decorator:: no_tracing(func)
962
963 Decorator to temporarily turn off tracing for the duration of the test.
964
965
966.. decorator:: refcount_test(test)
967
968 Decorator for tests which involve reference counting. The decorator does
969 not run the test if it is not run by CPython. Any trace function is unset
970 for the duration of the test to prevent unexpected refcounts caused by
971 the trace function.
972
973
Cheryl Sabella988fb282018-02-11 08:10:42 -0500974.. decorator:: bigmemtest(size, memuse, dry_run=True)
975
976 Decorator for bigmem tests.
977
978 *size* is a requested size for the test (in arbitrary, test-interpreted
979 units.) *memuse* is the number of bytes per unit for the test, or a good
980 estimate of it. For example, a test that needs two byte buffers, of 4 GiB
981 each, could be decorated with ``@bigmemtest(size=_4G, memuse=2)``.
982
983 The *size* argument is normally passed to the decorated test method as an
984 extra argument. If *dry_run* is ``True``, the value passed to the test
985 method may be less than the requested value. If *dry_run* is ``False``, it
986 means the test doesn't support dummy runs when ``-M`` is not specified.
987
988
989.. decorator:: bigaddrspacetest(f)
990
991 Decorator for tests that fill the address space. *f* is the function to
992 wrap.
993
994
Eli Benderskye1689652011-05-06 09:29:27 +0300995.. function:: make_bad_fd()
996
997 Create an invalid file descriptor by opening and closing a temporary file,
Zachary Waref012ba42014-07-23 12:00:29 -0500998 and returning its descriptor.
Eli Benderskye1689652011-05-06 09:29:27 +0300999
1000
Cheryl Sabella988fb282018-02-11 08:10:42 -05001001.. function:: check_syntax_error(testcase, statement, errtext='', *, lineno=None, offset=None)
1002
1003 Test for syntax errors in *statement* by attempting to compile *statement*.
1004 *testcase* is the :mod:`unittest` instance for the test. *errtext* is the
Serhiy Storchakae7a4bb52019-02-19 08:30:15 +02001005 regular expression which should match the string representation of the
1006 raised :exc:`SyntaxError`. If *lineno* is not ``None``, compares to
1007 the line of the exception. If *offset* is not ``None``, compares to
1008 the offset of the exception.
1009
1010
1011.. function:: check_syntax_warning(testcase, statement, errtext='', *, lineno=1, offset=None)
1012
1013 Test for syntax warning in *statement* by attempting to compile *statement*.
1014 Test also that the :exc:`SyntaxWarning` is emitted only once, and that it
1015 will be converted to a :exc:`SyntaxError` when turned into error.
1016 *testcase* is the :mod:`unittest` instance for the test. *errtext* is the
1017 regular expression which should match the string representation of the
1018 emitted :exc:`SyntaxWarning` and raised :exc:`SyntaxError`. If *lineno*
1019 is not ``None``, compares to the line of the warning and exception.
1020 If *offset* is not ``None``, compares to the offset of the exception.
1021
1022 .. versionadded:: 3.8
Cheryl Sabella988fb282018-02-11 08:10:42 -05001023
1024
1025.. function:: open_urlresource(url, *args, **kw)
1026
1027 Open *url*. If open fails, raises :exc:`TestFailed`.
1028
1029
1030.. function:: import_module(name, deprecated=False, *, required_on())
Nick Coghlanfce769e2009-04-11 14:30:59 +00001031
1032 This function imports and returns the named module. Unlike a normal
1033 import, this function raises :exc:`unittest.SkipTest` if the module
1034 cannot be imported.
1035
1036 Module and package deprecation messages are suppressed during this import
Cheryl Sabella988fb282018-02-11 08:10:42 -05001037 if *deprecated* is ``True``. If a module is required on a platform but
1038 optional for others, set *required_on* to an iterable of platform prefixes
1039 which will be compared against :data:`sys.platform`.
Nick Coghlanfce769e2009-04-11 14:30:59 +00001040
1041 .. versionadded:: 3.1
1042
1043
Nick Coghlan47384702009-04-22 16:13:36 +00001044.. function:: import_fresh_module(name, fresh=(), blocked=(), deprecated=False)
Nick Coghlanfce769e2009-04-11 14:30:59 +00001045
Nick Coghlan47384702009-04-22 16:13:36 +00001046 This function imports and returns a fresh copy of the named Python module
1047 by removing the named module from ``sys.modules`` before doing the import.
1048 Note that unlike :func:`reload`, the original module is not affected by
1049 this operation.
1050
1051 *fresh* is an iterable of additional module names that are also removed
1052 from the ``sys.modules`` cache before doing the import.
1053
Eli Benderskyba5517d2013-08-11 15:38:08 -07001054 *blocked* is an iterable of module names that are replaced with ``None``
Nick Coghlan47384702009-04-22 16:13:36 +00001055 in the module cache during the import to ensure that attempts to import
1056 them raise :exc:`ImportError`.
1057
1058 The named module and any modules named in the *fresh* and *blocked*
1059 parameters are saved before starting the import and then reinserted into
1060 ``sys.modules`` when the fresh import is complete.
Nick Coghlanfce769e2009-04-11 14:30:59 +00001061
1062 Module and package deprecation messages are suppressed during this import
Eli Benderskye1689652011-05-06 09:29:27 +03001063 if *deprecated* is ``True``.
Nick Coghlanfce769e2009-04-11 14:30:59 +00001064
Eli Benderskyba5517d2013-08-11 15:38:08 -07001065 This function will raise :exc:`ImportError` if the named module cannot be
1066 imported.
Nick Coghlan47384702009-04-22 16:13:36 +00001067
1068 Example use::
1069
Eli Benderskyba5517d2013-08-11 15:38:08 -07001070 # Get copies of the warnings module for testing without affecting the
1071 # version being used by the rest of the test suite. One copy uses the
1072 # C implementation, the other is forced to use the pure Python fallback
1073 # implementation
Nick Coghlan47384702009-04-22 16:13:36 +00001074 py_warnings = import_fresh_module('warnings', blocked=['_warnings'])
1075 c_warnings = import_fresh_module('warnings', fresh=['_warnings'])
1076
Nick Coghlanfce769e2009-04-11 14:30:59 +00001077 .. versionadded:: 3.1
1078
1079
Cheryl Sabella988fb282018-02-11 08:10:42 -05001080.. function:: modules_setup()
1081
1082 Return a copy of :data:`sys.modules`.
1083
1084
1085.. function:: modules_cleanup(oldmodules)
1086
1087 Remove modules except for *oldmodules* and ``encodings`` in order to
1088 preserve internal cache.
1089
1090
Cheryl Sabella988fb282018-02-11 08:10:42 -05001091.. function:: reap_children()
1092
1093 Use this at the end of ``test_main`` whenever sub-processes are started.
1094 This will help ensure that no extra children (zombies) stick around to
1095 hog resources and create problems when looking for refleaks.
1096
1097
1098.. function:: get_attribute(obj, name)
1099
1100 Get an attribute, raising :exc:`unittest.SkipTest` if :exc:`AttributeError`
1101 is raised.
1102
1103
Victor Stinner6dbbe742019-05-25 00:09:38 +02001104.. function:: catch_unraisable_exception()
1105
1106 Context manager catching unraisable exception using
1107 :func:`sys.unraisablehook`.
1108
Victor Stinner6d22cc82019-06-13 14:44:54 +02001109 Storing the exception value (``cm.unraisable.exc_value``) creates a
1110 reference cycle. The reference cycle is broken explicitly when the context
1111 manager exits.
1112
Victor Stinner212646c2019-06-14 18:03:22 +02001113 Storing the object (``cm.unraisable.object``) can resurrect it if it is set
1114 to an object which is being finalized. Exiting the context manager clears
1115 the stored object.
Victor Stinner6d22cc82019-06-13 14:44:54 +02001116
Victor Stinner6dbbe742019-05-25 00:09:38 +02001117 Usage::
1118
1119 with support.catch_unraisable_exception() as cm:
1120 # code creating an "unraisable exception"
1121 ...
1122
1123 # check the unraisable exception: use cm.unraisable
1124 ...
1125
1126 # cm.unraisable attribute no longer exists at this point
1127 # (to break a reference cycle)
1128
1129 .. versionadded:: 3.8
1130
1131
Zachary Waref012ba42014-07-23 12:00:29 -05001132.. function:: load_package_tests(pkg_dir, loader, standard_tests, pattern)
1133
1134 Generic implementation of the :mod:`unittest` ``load_tests`` protocol for
1135 use in test packages. *pkg_dir* is the root directory of the package;
1136 *loader*, *standard_tests*, and *pattern* are the arguments expected by
1137 ``load_tests``. In simple cases, the test package's ``__init__.py``
1138 can be the following::
1139
1140 import os
1141 from test.support import load_package_tests
1142
1143 def load_tests(*args):
1144 return load_package_tests(os.path.dirname(__file__), *args)
1145
Louie Lu7fae81e2017-04-22 14:46:18 +08001146
Cheryl Sabella988fb282018-02-11 08:10:42 -05001147.. function:: fs_is_case_insensitive(directory)
1148
1149 Return ``True`` if the file system for *directory* is case-insensitive.
1150
1151
Louie Lu7fae81e2017-04-22 14:46:18 +08001152.. function:: detect_api_mismatch(ref_api, other_api, *, ignore=())
Gregory P. Smith4e72cce2015-04-14 13:26:06 -07001153
Zachary Ware3d3aedc2015-07-07 00:07:25 -05001154 Returns the set of attributes, functions or methods of *ref_api* not
1155 found on *other_api*, except for a defined list of items to be
1156 ignored in this check specified in *ignore*.
Gregory P. Smith4e72cce2015-04-14 13:26:06 -07001157
1158 By default this skips private attributes beginning with '_' but
1159 includes all magic methods, i.e. those starting and ending in '__'.
1160
Gregory P. Smith7c63fd32015-04-14 15:25:01 -07001161 .. versionadded:: 3.5
1162
Zachary Waref012ba42014-07-23 12:00:29 -05001163
Cheryl Sabella988fb282018-02-11 08:10:42 -05001164.. function:: patch(test_instance, object_to_patch, attr_name, new_value)
1165
1166 Override *object_to_patch.attr_name* with *new_value*. Also add
1167 cleanup procedure to *test_instance* to restore *object_to_patch* for
1168 *attr_name*. The *attr_name* should be a valid attribute for
1169 *object_to_patch*.
1170
1171
1172.. function:: run_in_subinterp(code)
1173
1174 Run *code* in subinterpreter. Raise :exc:`unittest.SkipTest` if
1175 :mod:`tracemalloc` is enabled.
1176
1177
1178.. function:: check_free_after_iterating(test, iter, cls, args=())
1179
1180 Assert that *iter* is deallocated after iterating.
1181
1182
1183.. function:: missing_compiler_executable(cmd_names=[])
1184
1185 Check for the existence of the compiler executables whose names are listed
1186 in *cmd_names* or all the compiler executables when *cmd_names* is empty
1187 and return the first missing executable or ``None`` when none is found
1188 missing.
1189
1190
Martin Panterd226d302015-11-14 11:47:00 +00001191.. function:: check__all__(test_case, module, name_of_module=None, extra=(), blacklist=())
1192
1193 Assert that the ``__all__`` variable of *module* contains all public names.
1194
1195 The module's public names (its API) are detected automatically
1196 based on whether they match the public name convention and were defined in
1197 *module*.
1198
1199 The *name_of_module* argument can specify (as a string or tuple thereof) what
Serhiy Storchakabac2d5b2018-03-28 22:14:26 +03001200 module(s) an API could be defined in order to be detected as a public
Martin Panterd226d302015-11-14 11:47:00 +00001201 API. One case for this is when *module* imports part of its public API from
1202 other modules, possibly a C backend (like ``csv`` and its ``_csv``).
1203
1204 The *extra* argument can be a set of names that wouldn't otherwise be automatically
1205 detected as "public", like objects without a proper ``__module__``
1206 attribute. If provided, it will be added to the automatically detected ones.
1207
1208 The *blacklist* argument can be a set of names that must not be treated as part of
1209 the public API even though their names indicate otherwise.
1210
1211 Example use::
1212
1213 import bar
1214 import foo
1215 import unittest
1216 from test import support
1217
1218 class MiscTestCase(unittest.TestCase):
1219 def test__all__(self):
1220 support.check__all__(self, foo)
1221
1222 class OtherTestCase(unittest.TestCase):
1223 def test__all__(self):
1224 extra = {'BAR_CONST', 'FOO_CONST'}
1225 blacklist = {'baz'} # Undocumented name.
1226 # bar imports part of its API from _bar.
1227 support.check__all__(self, bar, ('bar', '_bar'),
1228 extra=extra, blacklist=blacklist)
1229
1230 .. versionadded:: 3.6
1231
1232
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001233The :mod:`test.support` module defines the following classes:
Georg Brandl116aa622007-08-15 14:28:22 +00001234
Georg Brandl7f01a132009-09-16 15:58:14 +00001235.. class:: TransientResource(exc, **kwargs)
Georg Brandl116aa622007-08-15 14:28:22 +00001236
1237 Instances are a context manager that raises :exc:`ResourceDenied` if the
1238 specified exception type is raised. Any keyword arguments are treated as
1239 attribute/value pairs to be compared against any exception raised within the
1240 :keyword:`with` statement. Only if all pairs match properly against
1241 attributes on the exception is :exc:`ResourceDenied` raised.
1242
Georg Brandl116aa622007-08-15 14:28:22 +00001243
1244.. class:: EnvironmentVarGuard()
1245
Florent Xicluna53b506be2010-03-18 20:00:57 +00001246 Class used to temporarily set or unset environment variables. Instances can
1247 be used as a context manager and have a complete dictionary interface for
1248 querying/modifying the underlying ``os.environ``. After exit from the
1249 context manager all changes to environment variables done through this
1250 instance will be rolled back.
Georg Brandl116aa622007-08-15 14:28:22 +00001251
Georg Brandl705d9d52009-05-05 09:29:50 +00001252 .. versionchanged:: 3.1
Walter Dörwald155374d2009-05-01 19:58:58 +00001253 Added dictionary interface.
Georg Brandl116aa622007-08-15 14:28:22 +00001254
1255.. method:: EnvironmentVarGuard.set(envvar, value)
1256
Florent Xicluna53b506be2010-03-18 20:00:57 +00001257 Temporarily set the environment variable ``envvar`` to the value of
1258 ``value``.
Georg Brandl116aa622007-08-15 14:28:22 +00001259
1260
1261.. method:: EnvironmentVarGuard.unset(envvar)
1262
1263 Temporarily unset the environment variable ``envvar``.
Nick Coghlanb1304932008-07-13 12:25:08 +00001264
Walter Dörwald155374d2009-05-01 19:58:58 +00001265
Antoine Pitrou77e904e2013-10-08 23:04:32 +02001266.. class:: SuppressCrashReport()
1267
1268 A context manager used to try to prevent crash dialog popups on tests that
1269 are expected to crash a subprocess.
1270
1271 On Windows, it disables Windows Error Reporting dialogs using
Georg Brandl5d941342016-02-26 19:37:12 +01001272 `SetErrorMode <https://msdn.microsoft.com/en-us/library/windows/desktop/ms680621.aspx>`_.
Antoine Pitrou77e904e2013-10-08 23:04:32 +02001273
1274 On UNIX, :func:`resource.setrlimit` is used to set
1275 :attr:`resource.RLIMIT_CORE`'s soft limit to 0 to prevent coredump file
1276 creation.
1277
1278 On both platforms, the old value is restored by :meth:`__exit__`.
1279
1280
Cheryl Sabella988fb282018-02-11 08:10:42 -05001281.. class:: CleanImport(*module_names)
1282
1283 A context manager to force import to return a new module reference. This
1284 is useful for testing module-level behaviors, such as the emission of a
1285 DeprecationWarning on import. Example usage::
1286
1287 with CleanImport('foo'):
1288 importlib.import_module('foo') # New reference.
1289
1290
1291.. class:: DirsOnSysPath(*paths)
1292
1293 A context manager to temporarily add directories to sys.path.
1294
1295 This makes a copy of :data:`sys.path`, appends any directories given
1296 as positional arguments, then reverts :data:`sys.path` to the copied
1297 settings when the context ends.
1298
1299 Note that *all* :data:`sys.path` modifications in the body of the
1300 context manager, including replacement of the object,
1301 will be reverted at the end of the block.
1302
1303
1304.. class:: SaveSignals()
1305
1306 Class to save and restore signal handlers registered by the Python signal
1307 handler.
1308
1309
1310.. class:: Matcher()
1311
1312 .. method:: matches(self, d, **kwargs)
1313
1314 Try to match a single dict with the supplied arguments.
1315
1316
1317 .. method:: match_value(self, k, dv, v)
1318
1319 Try to match a single stored value (*dv*) with a supplied value (*v*).
1320
1321
Benjamin Petersonfcf5d632008-10-16 23:24:44 +00001322.. class:: WarningsRecorder()
1323
1324 Class used to record warnings for unit tests. See documentation of
1325 :func:`check_warnings` above for more details.
Cheryl Sabella988fb282018-02-11 08:10:42 -05001326
1327
1328.. class:: BasicTestRunner()
1329
1330 .. method:: run(test)
1331
1332 Run *test* and return the result.
1333
1334
Serhiy Storchakab21d1552018-03-02 11:53:51 +02001335.. class:: FakePath(path)
1336
1337 Simple :term:`path-like object`. It implements the :meth:`__fspath__`
1338 method which just returns the *path* argument. If *path* is an exception,
1339 it will be raised in :meth:`!__fspath__`.
1340
1341
Serhiy Storchaka16994912020-04-25 10:06:29 +03001342:mod:`test.support.socket_helper` --- Utilities for socket tests
1343================================================================
1344
1345.. module:: test.support.socket_helper
1346 :synopsis: Support for socket tests.
1347
1348
1349The :mod:`test.support.socket_helper` module provides support for socket tests.
1350
1351.. versionadded:: 3.9
1352
1353
1354.. data:: IPV6_ENABLED
1355
1356 Set to ``True`` if IPv6 is enabled on this host, ``False`` otherwise.
1357
1358
1359.. function:: find_unused_port(family=socket.AF_INET, socktype=socket.SOCK_STREAM)
1360
1361 Returns an unused port that should be suitable for binding. This is
1362 achieved by creating a temporary socket with the same family and type as
1363 the ``sock`` parameter (default is :const:`~socket.AF_INET`,
1364 :const:`~socket.SOCK_STREAM`),
1365 and binding it to the specified host address (defaults to ``0.0.0.0``)
1366 with the port set to 0, eliciting an unused ephemeral port from the OS.
1367 The temporary socket is then closed and deleted, and the ephemeral port is
1368 returned.
1369
1370 Either this method or :func:`bind_port` should be used for any tests
1371 where a server socket needs to be bound to a particular port for the
1372 duration of the test.
1373 Which one to use depends on whether the calling code is creating a Python
1374 socket, or if an unused port needs to be provided in a constructor
1375 or passed to an external program (i.e. the ``-accept`` argument to
1376 openssl's s_server mode). Always prefer :func:`bind_port` over
1377 :func:`find_unused_port` where possible. Using a hard coded port is
1378 discouraged since it can make multiple instances of the test impossible to
1379 run simultaneously, which is a problem for buildbots.
1380
1381
1382.. function:: bind_port(sock, host=HOST)
1383
1384 Bind the socket to a free port and return the port number. Relies on
1385 ephemeral ports in order to ensure we are using an unbound port. This is
1386 important as many tests may be running simultaneously, especially in a
1387 buildbot environment. This method raises an exception if the
1388 ``sock.family`` is :const:`~socket.AF_INET` and ``sock.type`` is
1389 :const:`~socket.SOCK_STREAM`, and the socket has
1390 :const:`~socket.SO_REUSEADDR` or :const:`~socket.SO_REUSEPORT` set on it.
1391 Tests should never set these socket options for TCP/IP sockets.
1392 The only case for setting these options is testing multicasting via
1393 multiple UDP sockets.
1394
1395 Additionally, if the :const:`~socket.SO_EXCLUSIVEADDRUSE` socket option is
1396 available (i.e. on Windows), it will be set on the socket. This will
1397 prevent anyone else from binding to our host/port for the duration of the
1398 test.
1399
1400
1401.. function:: bind_unix_socket(sock, addr)
1402
1403 Bind a unix socket, raising :exc:`unittest.SkipTest` if
1404 :exc:`PermissionError` is raised.
1405
1406
1407.. decorator:: skip_unless_bind_unix_socket
1408
1409 A decorator for running tests that require a functional ``bind()`` for Unix
1410 sockets.
1411
1412
Serhiy Storchakabfb1cf42020-04-29 10:36:20 +03001413.. function:: transient_internet(resource_name, *, timeout=30.0, errnos=())
1414
1415 A context manager that raises :exc:`~test.support.ResourceDenied` when
1416 various issues with the internet connection manifest themselves as
1417 exceptions.
1418
1419
Cheryl Sabella988fb282018-02-11 08:10:42 -05001420:mod:`test.support.script_helper` --- Utilities for the Python execution tests
1421==============================================================================
1422
1423.. module:: test.support.script_helper
1424 :synopsis: Support for Python's script execution tests.
1425
1426
1427The :mod:`test.support.script_helper` module provides support for Python's
1428script execution tests.
1429
1430.. function:: interpreter_requires_environment()
1431
1432 Return ``True`` if ``sys.executable interpreter`` requires environment
1433 variables in order to be able to run at all.
1434
1435 This is designed to be used with ``@unittest.skipIf()`` to annotate tests
1436 that need to use an ``assert_python*()`` function to launch an isolated
1437 mode (``-I``) or no environment mode (``-E``) sub-interpreter process.
1438
1439 A normal build & test does not run into this situation but it can happen
1440 when trying to run the standard library test suite from an interpreter that
1441 doesn't have an obvious home with Python's current home finding logic.
1442
1443 Setting :envvar:`PYTHONHOME` is one way to get most of the testsuite to run
1444 in that situation. :envvar:`PYTHONPATH` or :envvar:`PYTHONUSERSITE` are
1445 other common environment variables that might impact whether or not the
1446 interpreter can start.
1447
1448
1449.. function:: run_python_until_end(*args, **env_vars)
1450
1451 Set up the environment based on *env_vars* for running the interpreter
Nick Coghland5d9e022018-03-25 23:03:10 +10001452 in a subprocess. The values can include ``__isolated``, ``__cleanenv``,
1453 ``__cwd``, and ``TERM``.
Cheryl Sabella988fb282018-02-11 08:10:42 -05001454
Victor Stinner6cac1132019-12-08 08:38:16 +01001455 .. versionchanged:: 3.9
1456 The function no longer strips whitespaces from *stderr*.
1457
Cheryl Sabella988fb282018-02-11 08:10:42 -05001458
1459.. function:: assert_python_ok(*args, **env_vars)
1460
1461 Assert that running the interpreter with *args* and optional environment
1462 variables *env_vars* succeeds (``rc == 0``) and return a ``(return code,
1463 stdout, stderr)`` tuple.
1464
1465 If the ``__cleanenv`` keyword is set, *env_vars* is used as a fresh
1466 environment.
1467
1468 Python is started in isolated mode (command line option ``-I``),
1469 except if the ``__isolated`` keyword is set to ``False``.
1470
Victor Stinner6cac1132019-12-08 08:38:16 +01001471 .. versionchanged:: 3.9
1472 The function no longer strips whitespaces from *stderr*.
1473
Cheryl Sabella988fb282018-02-11 08:10:42 -05001474
1475.. function:: assert_python_failure(*args, **env_vars)
1476
1477 Assert that running the interpreter with *args* and optional environment
1478 variables *env_vars* fails (``rc != 0``) and return a ``(return code,
1479 stdout, stderr)`` tuple.
1480
1481 See :func:`assert_python_ok` for more options.
1482
Victor Stinner6cac1132019-12-08 08:38:16 +01001483 .. versionchanged:: 3.9
1484 The function no longer strips whitespaces from *stderr*.
1485
Cheryl Sabella988fb282018-02-11 08:10:42 -05001486
1487.. function:: spawn_python(*args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, **kw)
1488
1489 Run a Python subprocess with the given arguments.
1490
1491 *kw* is extra keyword args to pass to :func:`subprocess.Popen`. Returns a
1492 :class:`subprocess.Popen` object.
1493
1494
1495.. function:: kill_python(p)
1496
1497 Run the given :class:`subprocess.Popen` process until completion and return
1498 stdout.
1499
1500
1501.. function:: make_script(script_dir, script_basename, source, omit_suffix=False)
1502
1503 Create script containing *source* in path *script_dir* and *script_basename*.
1504 If *omit_suffix* is ``False``, append ``.py`` to the name. Return the full
1505 script path.
1506
1507
1508.. function:: make_zip_script(zip_dir, zip_basename, script_name, name_in_zip=None)
1509
1510 Create zip file at *zip_dir* and *zip_basename* with extension ``zip`` which
1511 contains the files in *script_name*. *name_in_zip* is the archive name.
1512 Return a tuple containing ``(full path, full path of archive name)``.
1513
1514
1515.. function:: make_pkg(pkg_dir, init_source='')
1516
1517 Create a directory named *pkg_dir* containing an ``__init__`` file with
1518 *init_source* as its contents.
1519
1520
1521.. function:: make_zip_pkg(zip_dir, zip_basename, pkg_name, script_basename, \
1522 source, depth=1, compiled=False)
1523
1524 Create a zip package directory with a path of *zip_dir* and *zip_basename*
1525 containing an empty ``__init__`` file and a file *script_basename*
1526 containing the *source*. If *compiled* is ``True``, both source files will
1527 be compiled and added to the zip package. Return a tuple of the full zip
1528 path and the archive name for the zip file.
Joannah Nanjekye92777d52019-09-12 10:02:59 +01001529
1530
1531:mod:`test.support.bytecode_helper` --- Support tools for testing correct bytecode generation
1532=============================================================================================
1533
1534.. module:: test.support.bytecode_helper
1535 :synopsis: Support tools for testing correct bytecode generation.
1536
1537The :mod:`test.support.bytecode_helper` module provides support for testing
1538and inspecting bytecode generation.
1539
Gurupad Hegde6c7bb382019-12-28 17:16:02 -05001540The module defines the following class:
Joannah Nanjekye92777d52019-09-12 10:02:59 +01001541
1542.. class:: BytecodeTestCase(unittest.TestCase)
1543
1544 This class has custom assertion methods for inspecting bytecode.
1545
1546.. method:: BytecodeTestCase.get_disassembly_as_string(co)
1547
1548 Return the disassembly of *co* as string.
1549
1550
1551.. method:: BytecodeTestCase.assertInBytecode(x, opname, argval=_UNSPECIFIED)
1552
1553 Return instr if *opname* is found, otherwise throws :exc:`AssertionError`.
1554
1555
1556.. method:: BytecodeTestCase.assertNotInBytecode(x, opname, argval=_UNSPECIFIED)
1557
1558 Throws :exc:`AssertionError` if *opname* is found.
Hai Shie80697d2020-05-28 06:10:27 +08001559
1560
1561:mod:`test.support.threading_helper` --- Utilities for threading tests
1562======================================================================
1563
1564.. module:: test.support.threading_helper
1565 :synopsis: Support for threading tests.
1566
1567The :mod:`test.support.threading_helper` module provides support for threading tests.
1568
1569.. versionadded:: 3.10
1570
1571
1572.. function:: join_thread(thread, timeout=None)
1573
1574 Join a *thread* within *timeout*. Raise an :exc:`AssertionError` if thread
1575 is still alive after *timeout* seconds.
1576
1577
1578.. decorator:: reap_threads(func)
1579
1580 Decorator to ensure the threads are cleaned up even if the test fails.
1581
1582
1583.. function:: start_threads(threads, unlock=None)
1584
1585 Context manager to start *threads*. It attempts to join the threads upon
1586 exit.
1587
1588
1589.. function:: threading_cleanup(*original_values)
1590
1591 Cleanup up threads not specified in *original_values*. Designed to emit
1592 a warning if a test leaves running threads in the background.
1593
1594
1595.. function:: threading_setup()
1596
1597 Return current thread count and copy of dangling threads.
1598
1599
1600.. function:: wait_threads_exit(timeout=None)
1601
1602 Context manager to wait until all threads created in the ``with`` statement
1603 exit.
1604
1605
1606.. function:: catch_threading_exception()
1607
1608 Context manager catching :class:`threading.Thread` exception using
1609 :func:`threading.excepthook`.
1610
1611 Attributes set when an exception is catched:
1612
1613 * ``exc_type``
1614 * ``exc_value``
1615 * ``exc_traceback``
1616 * ``thread``
1617
1618 See :func:`threading.excepthook` documentation.
1619
1620 These attributes are deleted at the context manager exit.
1621
1622 Usage::
1623
1624 with threading_helper.catch_threading_exception() as cm:
1625 # code spawning a thread which raises an exception
1626 ...
1627
1628 # check the thread exception, use cm attributes:
1629 # exc_type, exc_value, exc_traceback, thread
1630 ...
1631
1632 # exc_type, exc_value, exc_traceback, thread attributes of cm no longer
1633 # exists at this point
1634 # (to avoid reference cycles)
1635
1636 .. versionadded:: 3.8