blob: 7bee6e8031a05fbadb26ccea7fb32550e1e8c6e2 [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
266.. data:: TESTFN_ENCODING
267
268 Set to :func:`sys.getfilesystemencoding`.
269
270
271.. data:: TESTFN_UNENCODABLE
272
273 Set to a filename (str type) that should not be able to be encoded by file
274 system encoding in strict mode. It may be ``None`` if it's not possible to
275 generate such a filename.
276
277
278.. data:: TESTFN_UNDECODABLE
279
280 Set to a filename (bytes type) that should not be able to be decoded by
281 file system encoding in strict mode. It may be ``None`` if it's not
282 possible to generate such a filename.
283
284
285.. data:: TESTFN_NONASCII
286
287 Set to a filename containing the :data:`FS_NONASCII` character.
288
289
Victor Stinner24c62582019-10-30 12:41:43 +0100290.. data:: LOOPBACK_TIMEOUT
291
292 Timeout in seconds for tests using a network server listening on the network
293 local loopback interface like ``127.0.0.1``.
294
295 The timeout is long enough to prevent test failure: it takes into account
296 that the client and the server can run in different threads or even
297 different processes.
298
299 The timeout should be long enough for :meth:`~socket.socket.connect`,
300 :meth:`~socket.socket.recv` and :meth:`~socket.socket.send` methods of
301 :class:`socket.socket`.
302
303 Its default value is 5 seconds.
304
305 See also :data:`INTERNET_TIMEOUT`.
306
307
308.. data:: INTERNET_TIMEOUT
309
310 Timeout in seconds for network requests going to the Internet.
311
312 The timeout is short enough to prevent a test to wait for too long if the
313 Internet request is blocked for whatever reason.
314
315 Usually, a timeout using :data:`INTERNET_TIMEOUT` should not mark a test as
316 failed, but skip the test instead: see
Serhiy Storchakabfb1cf42020-04-29 10:36:20 +0300317 :func:`~test.support.socket_helper.transient_internet`.
Victor Stinner24c62582019-10-30 12:41:43 +0100318
319 Its default value is 1 minute.
320
321 See also :data:`LOOPBACK_TIMEOUT`.
322
323
324.. data:: SHORT_TIMEOUT
325
326 Timeout in seconds to mark a test as failed if the test takes "too long".
327
328 The timeout value depends on the regrtest ``--timeout`` command line option.
329
330 If a test using :data:`SHORT_TIMEOUT` starts to fail randomly on slow
331 buildbots, use :data:`LONG_TIMEOUT` instead.
332
333 Its default value is 30 seconds.
334
335
336.. data:: LONG_TIMEOUT
337
338 Timeout in seconds to detect when a test hangs.
339
340 It is long enough to reduce the risk of test failure on the slowest Python
341 buildbots. It should not be used to mark a test as failed if the test takes
342 "too long". The timeout value depends on the regrtest ``--timeout`` command
343 line option.
344
345 Its default value is 5 minutes.
346
347 See also :data:`LOOPBACK_TIMEOUT`, :data:`INTERNET_TIMEOUT` and
348 :data:`SHORT_TIMEOUT`.
349
350
Cheryl Sabella988fb282018-02-11 08:10:42 -0500351.. data:: SAVEDCWD
352
353 Set to :func:`os.getcwd`.
354
355
356.. data:: PGO
357
358 Set when tests can be skipped when they are not useful for PGO.
359
360
361.. data:: PIPE_MAX_SIZE
362
363 A constant that is likely larger than the underlying OS pipe buffer size,
364 to make writes blocking.
365
366
367.. data:: SOCK_MAX_SIZE
368
369 A constant that is likely larger than the underlying OS socket buffer size,
370 to make writes blocking.
371
372
373.. data:: TEST_SUPPORT_DIR
374
375 Set to the top level directory that contains :mod:`test.support`.
376
377
378.. data:: TEST_HOME_DIR
379
380 Set to the top level directory for the test package.
381
382
383.. data:: TEST_DATA_DIR
384
385 Set to the ``data`` directory within the test package.
386
387
388.. data:: MAX_Py_ssize_t
389
390 Set to :data:`sys.maxsize` for big memory tests.
391
392
393.. data:: max_memuse
394
395 Set by :func:`set_memlimit` as the memory limit for big memory tests.
396 Limited by :data:`MAX_Py_ssize_t`.
397
398
399.. data:: real_max_memuse
400
401 Set by :func:`set_memlimit` as the memory limit for big memory tests. Not
402 limited by :data:`MAX_Py_ssize_t`.
403
404
405.. data:: MISSING_C_DOCSTRINGS
406
407 Return ``True`` if running on CPython, not on Windows, and configuration
408 not set with ``WITH_DOC_STRINGS``.
409
410
411.. data:: HAVE_DOCSTRINGS
412
413 Check for presence of docstrings.
414
Serhiy Storchaka17e52642019-08-04 12:38:46 +0300415
Stéphane Wirtela40681d2019-02-22 14:45:36 +0100416.. data:: TEST_HTTP_URL
417
418 Define the URL of a dedicated HTTP server for the network tests.
419
Cheryl Sabella988fb282018-02-11 08:10:42 -0500420
Serhiy Storchaka17e52642019-08-04 12:38:46 +0300421.. data:: ALWAYS_EQ
422
423 Object that is equal to anything. Used to test mixed type comparison.
424
425
Serhiy Storchaka18b711c2019-08-04 14:12:48 +0300426.. data:: NEVER_EQ
427
428 Object that is not equal to anything (even to :data:`ALWAYS_EQ`).
429 Used to test mixed type comparison.
430
431
Serhiy Storchaka17e52642019-08-04 12:38:46 +0300432.. data:: LARGEST
433
434 Object that is greater than anything (except itself).
435 Used to test mixed type comparison.
436
437
438.. data:: SMALLEST
439
440 Object that is less than anything (except itself).
441 Used to test mixed type comparison.
442
Cheryl Sabella988fb282018-02-11 08:10:42 -0500443
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000444The :mod:`test.support` module defines the following functions:
Georg Brandl116aa622007-08-15 14:28:22 +0000445
Georg Brandl116aa622007-08-15 14:28:22 +0000446.. function:: forget(module_name)
447
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000448 Remove the module named *module_name* from ``sys.modules`` and delete any
Georg Brandl116aa622007-08-15 14:28:22 +0000449 byte-compiled files of the module.
450
451
Cheryl Sabella988fb282018-02-11 08:10:42 -0500452.. function:: unload(name)
453
454 Delete *name* from ``sys.modules``.
455
456
457.. function:: unlink(filename)
458
459 Call :func:`os.unlink` on *filename*. On Windows platforms, this is
460 wrapped with a wait loop that checks for the existence fo the file.
461
462
463.. function:: rmdir(filename)
464
465 Call :func:`os.rmdir` on *filename*. On Windows platforms, this is
466 wrapped with a wait loop that checks for the existence of the file.
467
468
469.. function:: rmtree(path)
470
471 Call :func:`shutil.rmtree` on *path* or call :func:`os.lstat` and
472 :func:`os.rmdir` to remove a path and its contents. On Windows platforms,
473 this is wrapped with a wait loop that checks for the existence of the files.
474
475
476.. function:: make_legacy_pyc(source)
477
Stéphane Wirtel12e696b2018-10-27 00:58:26 +0200478 Move a :pep:`3147`/:pep:`488` pyc file to its legacy pyc location and return the file
Cheryl Sabella988fb282018-02-11 08:10:42 -0500479 system path to the legacy pyc file. The *source* value is the file system
480 path to the source file. It does not need to exist, however the PEP
481 3147/488 pyc file must exist.
482
483
Georg Brandl116aa622007-08-15 14:28:22 +0000484.. function:: is_resource_enabled(resource)
485
Eli Benderskye1689652011-05-06 09:29:27 +0300486 Return ``True`` if *resource* is enabled and available. The list of
Georg Brandl116aa622007-08-15 14:28:22 +0000487 available resources is only set when :mod:`test.regrtest` is executing the
488 tests.
489
490
Cheryl Sabella988fb282018-02-11 08:10:42 -0500491.. function:: python_is_optimized()
492
493 Return ``True`` if Python was not built with ``-O0`` or ``-Og``.
494
495
496.. function:: with_pymalloc()
497
498 Return :data:`_testcapi.WITH_PYMALLOC`.
499
500
Georg Brandl7f01a132009-09-16 15:58:14 +0000501.. function:: requires(resource, msg=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000502
Florent Xiclunab14930c2010-03-13 15:26:44 +0000503 Raise :exc:`ResourceDenied` if *resource* is not available. *msg* is the
Florent Xicluna53b506be2010-03-18 20:00:57 +0000504 argument to :exc:`ResourceDenied` if it is raised. Always returns
Eli Benderskye1689652011-05-06 09:29:27 +0300505 ``True`` if called by a function whose ``__name__`` is ``'__main__'``.
Florent Xicluna53b506be2010-03-18 20:00:57 +0000506 Used when tests are executed by :mod:`test.regrtest`.
Georg Brandl116aa622007-08-15 14:28:22 +0000507
508
Cheryl Sabella988fb282018-02-11 08:10:42 -0500509.. function:: system_must_validate_cert(f)
510
511 Raise :exc:`unittest.SkipTest` on TLS certification validation failures.
512
513
514.. function:: sortdict(dict)
515
516 Return a repr of *dict* with keys sorted.
517
518
Nick Coghlan0494c2a2013-09-08 11:40:34 +1000519.. function:: findfile(filename, subdir=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000520
Florent Xicluna53b506be2010-03-18 20:00:57 +0000521 Return the path to the file named *filename*. If no match is found
522 *filename* is returned. This does not equal a failure since it could be the
523 path to the file.
Georg Brandl116aa622007-08-15 14:28:22 +0000524
Cheryl Sabella988fb282018-02-11 08:10:42 -0500525 Setting *subdir* indicates a relative path to use to find the file
526 rather than looking directly in the path directories.
527
528
529.. function:: create_empty_file(filename)
530
531 Create an empty file with *filename*. If it already exists, truncate it.
532
533
534.. function:: fd_count()
535
536 Count the number of open file descriptors.
537
538
539.. function:: match_test(test)
540
541 Match *test* to patterns set in :func:`set_match_tests`.
542
543
544.. function:: set_match_tests(patterns)
545
546 Define match test with regular expression *patterns*.
Nick Coghlan0494c2a2013-09-08 11:40:34 +1000547
Georg Brandl116aa622007-08-15 14:28:22 +0000548
Senthil Kumaran279b56d2010-10-15 15:21:19 +0000549.. function:: run_unittest(\*classes)
Georg Brandl116aa622007-08-15 14:28:22 +0000550
551 Execute :class:`unittest.TestCase` subclasses passed to the function. The
Florent Xicluna53b506be2010-03-18 20:00:57 +0000552 function scans the classes for methods starting with the prefix ``test_``
553 and executes the tests individually.
Georg Brandl116aa622007-08-15 14:28:22 +0000554
555 It is also legal to pass strings as parameters; these should be keys in
556 ``sys.modules``. Each associated module will be scanned by
557 ``unittest.TestLoader.loadTestsFromModule()``. This is usually seen in the
558 following :func:`test_main` function::
559
560 def test_main():
Nick Coghlan47384702009-04-22 16:13:36 +0000561 support.run_unittest(__name__)
Georg Brandl116aa622007-08-15 14:28:22 +0000562
563 This will run all tests defined in the named module.
564
Georg Brandl116aa622007-08-15 14:28:22 +0000565
Cheryl Sabella988fb282018-02-11 08:10:42 -0500566.. function:: run_doctest(module, verbosity=None, optionflags=0)
Eli Benderskye1689652011-05-06 09:29:27 +0300567
568 Run :func:`doctest.testmod` on the given *module*. Return
569 ``(failure_count, test_count)``.
570
571 If *verbosity* is ``None``, :func:`doctest.testmod` is run with verbosity
572 set to :data:`verbose`. Otherwise, it is run with verbosity set to
Cheryl Sabella988fb282018-02-11 08:10:42 -0500573 ``None``. *optionflags* is passed as ``optionflags`` to
574 :func:`doctest.testmod`.
575
576
577.. function:: setswitchinterval(interval)
578
579 Set the :func:`sys.setswitchinterval` to the given *interval*. Defines
580 a minimum interval for Android systems to prevent the system from hanging.
581
582
583.. function:: check_impl_detail(**guards)
584
585 Use this check to guard CPython's implementation-specific tests or to
586 run them only on the implementations guarded by the arguments::
587
588 check_impl_detail() # Only on CPython (default).
589 check_impl_detail(jython=True) # Only on Jython.
590 check_impl_detail(cpython=False) # Everywhere except CPython.
591
Eli Benderskye1689652011-05-06 09:29:27 +0300592
Senthil Kumaran279b56d2010-10-15 15:21:19 +0000593.. function:: check_warnings(\*filters, quiet=True)
Thomas Woutersed03b412007-08-28 21:37:11 +0000594
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000595 A convenience wrapper for :func:`warnings.catch_warnings()` that makes it
596 easier to test that a warning was correctly raised. It is approximately
597 equivalent to calling ``warnings.catch_warnings(record=True)`` with
598 :meth:`warnings.simplefilter` set to ``always`` and with the option to
599 automatically validate the results that are recorded.
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000600
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000601 ``check_warnings`` accepts 2-tuples of the form ``("message regexp",
602 WarningCategory)`` as positional arguments. If one or more *filters* are
Eli Benderskye1689652011-05-06 09:29:27 +0300603 provided, or if the optional keyword argument *quiet* is ``False``,
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000604 it checks to make sure the warnings are as expected: each specified filter
605 must match at least one of the warnings raised by the enclosed code or the
606 test fails, and if any warnings are raised that do not match any of the
607 specified filters the test fails. To disable the first of these checks,
Eli Benderskye1689652011-05-06 09:29:27 +0300608 set *quiet* to ``True``.
Florent Xiclunab14930c2010-03-13 15:26:44 +0000609
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000610 If no arguments are specified, it defaults to::
Florent Xiclunab14930c2010-03-13 15:26:44 +0000611
Florent Xicluna53b506be2010-03-18 20:00:57 +0000612 check_warnings(("", Warning), quiet=True)
Florent Xiclunab14930c2010-03-13 15:26:44 +0000613
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000614 In this case all warnings are caught and no errors are raised.
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000615
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000616 On entry to the context manager, a :class:`WarningRecorder` instance is
617 returned. The underlying warnings list from
618 :func:`~warnings.catch_warnings` is available via the recorder object's
619 :attr:`warnings` attribute. As a convenience, the attributes of the object
620 representing the most recent warning can also be accessed directly through
621 the recorder object (see example below). If no warning has been raised,
622 then any of the attributes that would otherwise be expected on an object
Eli Benderskye1689652011-05-06 09:29:27 +0300623 representing a warning will return ``None``.
Thomas Woutersed03b412007-08-28 21:37:11 +0000624
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000625 The recorder object also has a :meth:`reset` method, which clears the
626 warnings list.
Thomas Woutersed03b412007-08-28 21:37:11 +0000627
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000628 The context manager is designed to be used like this::
Florent Xiclunab14930c2010-03-13 15:26:44 +0000629
630 with check_warnings(("assertion is always true", SyntaxWarning),
631 ("", UserWarning)):
632 exec('assert(False, "Hey!")')
633 warnings.warn(UserWarning("Hide me!"))
634
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000635 In this case if either warning was not raised, or some other warning was
636 raised, :func:`check_warnings` would raise an error.
637
638 When a test needs to look more deeply into the warnings, rather than
639 just checking whether or not they occurred, code like this can be used::
640
Florent Xiclunab14930c2010-03-13 15:26:44 +0000641 with check_warnings(quiet=True) as w:
Thomas Woutersed03b412007-08-28 21:37:11 +0000642 warnings.warn("foo")
Florent Xiclunab14930c2010-03-13 15:26:44 +0000643 assert str(w.args[0]) == "foo"
Nick Coghlanb1304932008-07-13 12:25:08 +0000644 warnings.warn("bar")
Florent Xiclunab14930c2010-03-13 15:26:44 +0000645 assert str(w.args[0]) == "bar"
646 assert str(w.warnings[0].args[0]) == "foo"
647 assert str(w.warnings[1].args[0]) == "bar"
Benjamin Petersonfcf5d632008-10-16 23:24:44 +0000648 w.reset()
649 assert len(w.warnings) == 0
Thomas Woutersed03b412007-08-28 21:37:11 +0000650
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000651
652 Here all warnings will be caught, and the test code tests the captured
653 warnings directly.
654
Ezio Melottif8754a62010-03-21 07:16:43 +0000655 .. versionchanged:: 3.2
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000656 New optional arguments *filters* and *quiet*.
Florent Xiclunab14930c2010-03-13 15:26:44 +0000657
Thomas Woutersed03b412007-08-28 21:37:11 +0000658
Cheryl Sabella988fb282018-02-11 08:10:42 -0500659.. function:: check_no_resource_warning(testcase)
660
661 Context manager to check that no :exc:`ResourceWarning` was raised. You
662 must remove the object which may emit :exc:`ResourceWarning` before the
663 end of the context manager.
664
665
666.. function:: set_memlimit(limit)
667
668 Set the values for :data:`max_memuse` and :data:`real_max_memuse` for big
669 memory tests.
670
671
672.. function:: record_original_stdout(stdout)
673
674 Store the value from *stdout*. It is meant to hold the stdout at the
675 time the regrtest began.
676
677
678.. function:: get_original_stdout
679
680 Return the original stdout set by :func:`record_original_stdout` or
681 ``sys.stdout`` if it's not set.
682
683
Cheryl Sabella988fb282018-02-11 08:10:42 -0500684.. function:: args_from_interpreter_flags()
685
686 Return a list of command line arguments reproducing the current settings
687 in ``sys.flags`` and ``sys.warnoptions``.
688
689
690.. function:: optim_args_from_interpreter_flags()
691
692 Return a list of command line arguments reproducing the current
693 optimization settings in ``sys.flags``.
694
695
R David Murray5a33f812013-07-11 12:28:40 -0400696.. function:: captured_stdin()
697 captured_stdout()
698 captured_stderr()
Thomas Woutersed03b412007-08-28 21:37:11 +0000699
R David Murray5a33f812013-07-11 12:28:40 -0400700 A context managers that temporarily replaces the named stream with
701 :class:`io.StringIO` object.
Thomas Woutersed03b412007-08-28 21:37:11 +0000702
R David Murray5a33f812013-07-11 12:28:40 -0400703 Example use with output streams::
Thomas Woutersed03b412007-08-28 21:37:11 +0000704
R David Murray5a33f812013-07-11 12:28:40 -0400705 with captured_stdout() as stdout, captured_stderr() as stderr:
Collin Winterc79461b2007-09-01 23:34:30 +0000706 print("hello")
R David Murray5a33f812013-07-11 12:28:40 -0400707 print("error", file=sys.stderr)
708 assert stdout.getvalue() == "hello\n"
709 assert stderr.getvalue() == "error\n"
710
711 Example use with input stream::
712
713 with captured_stdin() as stdin:
714 stdin.write('hello\n')
715 stdin.seek(0)
716 # call test code that consumes from sys.stdin
717 captured = input()
718 self.assertEqual(captured, "hello")
Thomas Woutersed03b412007-08-28 21:37:11 +0000719
Thomas Woutersed03b412007-08-28 21:37:11 +0000720
Nick Coghlan55175962013-07-28 22:11:50 +1000721.. function:: temp_dir(path=None, quiet=False)
722
723 A context manager that creates a temporary directory at *path* and
724 yields the directory.
725
Serhiy Storchakaecf41da2016-10-19 16:29:26 +0300726 If *path* is ``None``, the temporary directory is created using
Nick Coghlan55175962013-07-28 22:11:50 +1000727 :func:`tempfile.mkdtemp`. If *quiet* is ``False``, the context manager
728 raises an exception on error. Otherwise, if *path* is specified and
729 cannot be created, only a warning is issued.
730
731
732.. function:: change_cwd(path, quiet=False)
Eli Benderskye1689652011-05-06 09:29:27 +0300733
734 A context manager that temporarily changes the current working
Nick Coghlan55175962013-07-28 22:11:50 +1000735 directory to *path* and yields the directory.
Eli Benderskye1689652011-05-06 09:29:27 +0300736
Nick Coghlan55175962013-07-28 22:11:50 +1000737 If *quiet* is ``False``, the context manager raises an exception
738 on error. Otherwise, it issues only a warning and keeps the current
739 working directory the same.
Eli Benderskye1689652011-05-06 09:29:27 +0300740
Nick Coghlan55175962013-07-28 22:11:50 +1000741
742.. function:: temp_cwd(name='tempcwd', quiet=False)
743
744 A context manager that temporarily creates a new directory and
745 changes the current working directory (CWD).
746
747 The context manager creates a temporary directory in the current
748 directory with name *name* before temporarily changing the current
Serhiy Storchakaecf41da2016-10-19 16:29:26 +0300749 working directory. If *name* is ``None``, the temporary directory is
Nick Coghlan55175962013-07-28 22:11:50 +1000750 created using :func:`tempfile.mkdtemp`.
751
752 If *quiet* is ``False`` and it is not possible to create or change
753 the CWD, an error is raised. Otherwise, only a warning is raised
754 and the original CWD is used.
Eli Benderskye1689652011-05-06 09:29:27 +0300755
756
757.. function:: temp_umask(umask)
758
759 A context manager that temporarily sets the process umask.
760
761
Cheryl Sabella988fb282018-02-11 08:10:42 -0500762.. function:: disable_faulthandler()
763
764 A context manager that replaces ``sys.stderr`` with ``sys.__stderr__``.
765
766
767.. function:: gc_collect()
768
769 Force as many objects as possible to be collected. This is needed because
770 timely deallocation is not guaranteed by the garbage collector. This means
771 that ``__del__`` methods may be called later than expected and weakrefs
772 may remain alive for longer than expected.
773
774
775.. function:: disable_gc()
776
777 A context manager that disables the garbage collector upon entry and
778 reenables it upon exit.
779
780
781.. function:: swap_attr(obj, attr, new_val)
782
783 Context manager to swap out an attribute with a new object.
784
785 Usage::
786
787 with swap_attr(obj, "attr", 5):
788 ...
789
790 This will set ``obj.attr`` to 5 for the duration of the ``with`` block,
791 restoring the old value at the end of the block. If ``attr`` doesn't
792 exist on ``obj``, it will be created and then deleted at the end of the
793 block.
794
795 The old value (or ``None`` if it doesn't exist) will be assigned to the
796 target of the "as" clause, if there is one.
797
798
799.. function:: swap_item(obj, attr, new_val)
800
801 Context manager to swap out an item with a new object.
802
803 Usage::
804
805 with swap_item(obj, "item", 5):
806 ...
807
808 This will set ``obj["item"]`` to 5 for the duration of the ``with`` block,
809 restoring the old value at the end of the block. If ``item`` doesn't
810 exist on ``obj``, it will be created and then deleted at the end of the
811 block.
812
813 The old value (or ``None`` if it doesn't exist) will be assigned to the
814 target of the "as" clause, if there is one.
815
816
Victor Stinnerd663d342020-04-23 19:03:52 +0200817.. function:: print_warning(msg)
818
819 Print a warning into :data:`sys.__stderr__`. Format the message as:
820 ``f"Warning -- {msg}"``. If *msg* is made of multiple lines, add
821 ``"Warning -- "`` prefix to each line.
822
823 .. versionadded:: 3.9
824
825
Victor Stinner278c1e12020-03-31 20:08:12 +0200826.. function:: wait_process(pid, *, exitcode, timeout=None)
827
828 Wait until process *pid* completes and check that the process exit code is
829 *exitcode*.
830
831 Raise an :exc:`AssertionError` if the process exit code is not equal to
832 *exitcode*.
833
834 If the process runs longer than *timeout* seconds (:data:`SHORT_TIMEOUT` by
835 default), kill the process and raise an :exc:`AssertionError`. The timeout
836 feature is not available on Windows.
837
838 .. versionadded:: 3.9
839
840
Cheryl Sabella988fb282018-02-11 08:10:42 -0500841.. function:: calcobjsize(fmt)
842
843 Return :func:`struct.calcsize` for ``nP{fmt}0n`` or, if ``gettotalrefcount``
844 exists, ``2PnP{fmt}0P``.
845
846
847.. function:: calcvobjsize(fmt)
848
849 Return :func:`struct.calcsize` for ``nPn{fmt}0n`` or, if ``gettotalrefcount``
850 exists, ``2PnPn{fmt}0P``.
851
852
853.. function:: checksizeof(test, o, size)
854
855 For testcase *test*, assert that the ``sys.getsizeof`` for *o* plus the GC
856 header size equals *size*.
857
858
Eli Benderskye1689652011-05-06 09:29:27 +0300859.. function:: can_symlink()
860
861 Return ``True`` if the OS supports symbolic links, ``False``
862 otherwise.
863
864
Cheryl Sabella988fb282018-02-11 08:10:42 -0500865.. function:: can_xattr()
866
867 Return ``True`` if the OS supports xattr, ``False``
868 otherwise.
869
870
Daisuke Miyakawa0e61e672017-10-12 23:39:43 +0900871.. decorator:: skip_unless_symlink
Eli Benderskye1689652011-05-06 09:29:27 +0300872
873 A decorator for running tests that require support for symbolic links.
874
875
Cheryl Sabella988fb282018-02-11 08:10:42 -0500876.. decorator:: skip_unless_xattr
877
878 A decorator for running tests that require support for xattr.
879
880
Nick Coghlan2496f332011-09-19 20:26:31 +1000881.. decorator:: anticipate_failure(condition)
882
883 A decorator to conditionally mark tests with
884 :func:`unittest.expectedFailure`. Any use of this decorator should
885 have an associated comment identifying the relevant tracker issue.
886
887
888.. decorator:: run_with_locale(catstr, *locales)
Eli Benderskye1689652011-05-06 09:29:27 +0300889
890 A decorator for running a function in a different locale, correctly
891 resetting it after it has finished. *catstr* is the locale category as
892 a string (for example ``"LC_ALL"``). The *locales* passed will be tried
893 sequentially, and the first valid locale will be used.
894
895
Cheryl Sabella988fb282018-02-11 08:10:42 -0500896.. decorator:: run_with_tz(tz)
897
898 A decorator for running a function in a specific timezone, correctly
899 resetting it after it has finished.
900
901
902.. decorator:: requires_freebsd_version(*min_version)
903
904 Decorator for the minimum version when running test on FreeBSD. If the
905 FreeBSD version is less than the minimum, raise :exc:`unittest.SkipTest`.
906
907
908.. decorator:: requires_linux_version(*min_version)
909
910 Decorator for the minimum version when running test on Linux. If the
911 Linux version is less than the minimum, raise :exc:`unittest.SkipTest`.
912
913
914.. decorator:: requires_mac_version(*min_version)
915
916 Decorator for the minimum version when running test on Mac OS X. If the
917 MAC OS X version is less than the minimum, raise :exc:`unittest.SkipTest`.
918
919
920.. decorator:: requires_IEEE_754
921
922 Decorator for skipping tests on non-IEEE 754 platforms.
923
924
925.. decorator:: requires_zlib
926
927 Decorator for skipping tests if :mod:`zlib` doesn't exist.
928
929
930.. decorator:: requires_gzip
931
932 Decorator for skipping tests if :mod:`gzip` doesn't exist.
933
934
935.. decorator:: requires_bz2
936
937 Decorator for skipping tests if :mod:`bz2` doesn't exist.
938
939
940.. decorator:: requires_lzma
941
942 Decorator for skipping tests if :mod:`lzma` doesn't exist.
943
944
945.. decorator:: requires_resource(resource)
946
947 Decorator for skipping tests if *resource* is not available.
948
949
950.. decorator:: requires_docstrings
951
952 Decorator for only running the test if :data:`HAVE_DOCSTRINGS`.
953
954
955.. decorator:: cpython_only(test)
956
957 Decorator for tests only applicable to CPython.
958
959
960.. decorator:: impl_detail(msg=None, **guards)
961
962 Decorator for invoking :func:`check_impl_detail` on *guards*. If that
963 returns ``False``, then uses *msg* as the reason for skipping the test.
964
965
966.. decorator:: no_tracing(func)
967
968 Decorator to temporarily turn off tracing for the duration of the test.
969
970
971.. decorator:: refcount_test(test)
972
973 Decorator for tests which involve reference counting. The decorator does
974 not run the test if it is not run by CPython. Any trace function is unset
975 for the duration of the test to prevent unexpected refcounts caused by
976 the trace function.
977
978
Cheryl Sabella988fb282018-02-11 08:10:42 -0500979.. decorator:: bigmemtest(size, memuse, dry_run=True)
980
981 Decorator for bigmem tests.
982
983 *size* is a requested size for the test (in arbitrary, test-interpreted
984 units.) *memuse* is the number of bytes per unit for the test, or a good
985 estimate of it. For example, a test that needs two byte buffers, of 4 GiB
986 each, could be decorated with ``@bigmemtest(size=_4G, memuse=2)``.
987
988 The *size* argument is normally passed to the decorated test method as an
989 extra argument. If *dry_run* is ``True``, the value passed to the test
990 method may be less than the requested value. If *dry_run* is ``False``, it
991 means the test doesn't support dummy runs when ``-M`` is not specified.
992
993
994.. decorator:: bigaddrspacetest(f)
995
996 Decorator for tests that fill the address space. *f* is the function to
997 wrap.
998
999
Eli Benderskye1689652011-05-06 09:29:27 +03001000.. function:: make_bad_fd()
1001
1002 Create an invalid file descriptor by opening and closing a temporary file,
Zachary Waref012ba42014-07-23 12:00:29 -05001003 and returning its descriptor.
Eli Benderskye1689652011-05-06 09:29:27 +03001004
1005
Cheryl Sabella988fb282018-02-11 08:10:42 -05001006.. function:: check_syntax_error(testcase, statement, errtext='', *, lineno=None, offset=None)
1007
1008 Test for syntax errors in *statement* by attempting to compile *statement*.
1009 *testcase* is the :mod:`unittest` instance for the test. *errtext* is the
Serhiy Storchakae7a4bb52019-02-19 08:30:15 +02001010 regular expression which should match the string representation of the
1011 raised :exc:`SyntaxError`. If *lineno* is not ``None``, compares to
1012 the line of the exception. If *offset* is not ``None``, compares to
1013 the offset of the exception.
1014
1015
1016.. function:: check_syntax_warning(testcase, statement, errtext='', *, lineno=1, offset=None)
1017
1018 Test for syntax warning in *statement* by attempting to compile *statement*.
1019 Test also that the :exc:`SyntaxWarning` is emitted only once, and that it
1020 will be converted to a :exc:`SyntaxError` when turned into error.
1021 *testcase* is the :mod:`unittest` instance for the test. *errtext* is the
1022 regular expression which should match the string representation of the
1023 emitted :exc:`SyntaxWarning` and raised :exc:`SyntaxError`. If *lineno*
1024 is not ``None``, compares to the line of the warning and exception.
1025 If *offset* is not ``None``, compares to the offset of the exception.
1026
1027 .. versionadded:: 3.8
Cheryl Sabella988fb282018-02-11 08:10:42 -05001028
1029
1030.. function:: open_urlresource(url, *args, **kw)
1031
1032 Open *url*. If open fails, raises :exc:`TestFailed`.
1033
1034
1035.. function:: import_module(name, deprecated=False, *, required_on())
Nick Coghlanfce769e2009-04-11 14:30:59 +00001036
1037 This function imports and returns the named module. Unlike a normal
1038 import, this function raises :exc:`unittest.SkipTest` if the module
1039 cannot be imported.
1040
1041 Module and package deprecation messages are suppressed during this import
Cheryl Sabella988fb282018-02-11 08:10:42 -05001042 if *deprecated* is ``True``. If a module is required on a platform but
1043 optional for others, set *required_on* to an iterable of platform prefixes
1044 which will be compared against :data:`sys.platform`.
Nick Coghlanfce769e2009-04-11 14:30:59 +00001045
1046 .. versionadded:: 3.1
1047
1048
Nick Coghlan47384702009-04-22 16:13:36 +00001049.. function:: import_fresh_module(name, fresh=(), blocked=(), deprecated=False)
Nick Coghlanfce769e2009-04-11 14:30:59 +00001050
Nick Coghlan47384702009-04-22 16:13:36 +00001051 This function imports and returns a fresh copy of the named Python module
1052 by removing the named module from ``sys.modules`` before doing the import.
1053 Note that unlike :func:`reload`, the original module is not affected by
1054 this operation.
1055
1056 *fresh* is an iterable of additional module names that are also removed
1057 from the ``sys.modules`` cache before doing the import.
1058
Eli Benderskyba5517d2013-08-11 15:38:08 -07001059 *blocked* is an iterable of module names that are replaced with ``None``
Nick Coghlan47384702009-04-22 16:13:36 +00001060 in the module cache during the import to ensure that attempts to import
1061 them raise :exc:`ImportError`.
1062
1063 The named module and any modules named in the *fresh* and *blocked*
1064 parameters are saved before starting the import and then reinserted into
1065 ``sys.modules`` when the fresh import is complete.
Nick Coghlanfce769e2009-04-11 14:30:59 +00001066
1067 Module and package deprecation messages are suppressed during this import
Eli Benderskye1689652011-05-06 09:29:27 +03001068 if *deprecated* is ``True``.
Nick Coghlanfce769e2009-04-11 14:30:59 +00001069
Eli Benderskyba5517d2013-08-11 15:38:08 -07001070 This function will raise :exc:`ImportError` if the named module cannot be
1071 imported.
Nick Coghlan47384702009-04-22 16:13:36 +00001072
1073 Example use::
1074
Eli Benderskyba5517d2013-08-11 15:38:08 -07001075 # Get copies of the warnings module for testing without affecting the
1076 # version being used by the rest of the test suite. One copy uses the
1077 # C implementation, the other is forced to use the pure Python fallback
1078 # implementation
Nick Coghlan47384702009-04-22 16:13:36 +00001079 py_warnings = import_fresh_module('warnings', blocked=['_warnings'])
1080 c_warnings = import_fresh_module('warnings', fresh=['_warnings'])
1081
Nick Coghlanfce769e2009-04-11 14:30:59 +00001082 .. versionadded:: 3.1
1083
1084
Cheryl Sabella988fb282018-02-11 08:10:42 -05001085.. function:: modules_setup()
1086
1087 Return a copy of :data:`sys.modules`.
1088
1089
1090.. function:: modules_cleanup(oldmodules)
1091
1092 Remove modules except for *oldmodules* and ``encodings`` in order to
1093 preserve internal cache.
1094
1095
Cheryl Sabella988fb282018-02-11 08:10:42 -05001096.. function:: reap_children()
1097
1098 Use this at the end of ``test_main`` whenever sub-processes are started.
1099 This will help ensure that no extra children (zombies) stick around to
1100 hog resources and create problems when looking for refleaks.
1101
1102
1103.. function:: get_attribute(obj, name)
1104
1105 Get an attribute, raising :exc:`unittest.SkipTest` if :exc:`AttributeError`
1106 is raised.
1107
1108
Victor Stinner6dbbe742019-05-25 00:09:38 +02001109.. function:: catch_unraisable_exception()
1110
1111 Context manager catching unraisable exception using
1112 :func:`sys.unraisablehook`.
1113
Victor Stinner6d22cc82019-06-13 14:44:54 +02001114 Storing the exception value (``cm.unraisable.exc_value``) creates a
1115 reference cycle. The reference cycle is broken explicitly when the context
1116 manager exits.
1117
Victor Stinner212646c2019-06-14 18:03:22 +02001118 Storing the object (``cm.unraisable.object``) can resurrect it if it is set
1119 to an object which is being finalized. Exiting the context manager clears
1120 the stored object.
Victor Stinner6d22cc82019-06-13 14:44:54 +02001121
Victor Stinner6dbbe742019-05-25 00:09:38 +02001122 Usage::
1123
1124 with support.catch_unraisable_exception() as cm:
1125 # code creating an "unraisable exception"
1126 ...
1127
1128 # check the unraisable exception: use cm.unraisable
1129 ...
1130
1131 # cm.unraisable attribute no longer exists at this point
1132 # (to break a reference cycle)
1133
1134 .. versionadded:: 3.8
1135
1136
Zachary Waref012ba42014-07-23 12:00:29 -05001137.. function:: load_package_tests(pkg_dir, loader, standard_tests, pattern)
1138
1139 Generic implementation of the :mod:`unittest` ``load_tests`` protocol for
1140 use in test packages. *pkg_dir* is the root directory of the package;
1141 *loader*, *standard_tests*, and *pattern* are the arguments expected by
1142 ``load_tests``. In simple cases, the test package's ``__init__.py``
1143 can be the following::
1144
1145 import os
1146 from test.support import load_package_tests
1147
1148 def load_tests(*args):
1149 return load_package_tests(os.path.dirname(__file__), *args)
1150
Louie Lu7fae81e2017-04-22 14:46:18 +08001151
Cheryl Sabella988fb282018-02-11 08:10:42 -05001152.. function:: fs_is_case_insensitive(directory)
1153
1154 Return ``True`` if the file system for *directory* is case-insensitive.
1155
1156
Louie Lu7fae81e2017-04-22 14:46:18 +08001157.. function:: detect_api_mismatch(ref_api, other_api, *, ignore=())
Gregory P. Smith4e72cce2015-04-14 13:26:06 -07001158
Zachary Ware3d3aedc2015-07-07 00:07:25 -05001159 Returns the set of attributes, functions or methods of *ref_api* not
1160 found on *other_api*, except for a defined list of items to be
1161 ignored in this check specified in *ignore*.
Gregory P. Smith4e72cce2015-04-14 13:26:06 -07001162
1163 By default this skips private attributes beginning with '_' but
1164 includes all magic methods, i.e. those starting and ending in '__'.
1165
Gregory P. Smith7c63fd32015-04-14 15:25:01 -07001166 .. versionadded:: 3.5
1167
Zachary Waref012ba42014-07-23 12:00:29 -05001168
Cheryl Sabella988fb282018-02-11 08:10:42 -05001169.. function:: patch(test_instance, object_to_patch, attr_name, new_value)
1170
1171 Override *object_to_patch.attr_name* with *new_value*. Also add
1172 cleanup procedure to *test_instance* to restore *object_to_patch* for
1173 *attr_name*. The *attr_name* should be a valid attribute for
1174 *object_to_patch*.
1175
1176
1177.. function:: run_in_subinterp(code)
1178
1179 Run *code* in subinterpreter. Raise :exc:`unittest.SkipTest` if
1180 :mod:`tracemalloc` is enabled.
1181
1182
1183.. function:: check_free_after_iterating(test, iter, cls, args=())
1184
1185 Assert that *iter* is deallocated after iterating.
1186
1187
1188.. function:: missing_compiler_executable(cmd_names=[])
1189
1190 Check for the existence of the compiler executables whose names are listed
1191 in *cmd_names* or all the compiler executables when *cmd_names* is empty
1192 and return the first missing executable or ``None`` when none is found
1193 missing.
1194
1195
Martin Panterd226d302015-11-14 11:47:00 +00001196.. function:: check__all__(test_case, module, name_of_module=None, extra=(), blacklist=())
1197
1198 Assert that the ``__all__`` variable of *module* contains all public names.
1199
1200 The module's public names (its API) are detected automatically
1201 based on whether they match the public name convention and were defined in
1202 *module*.
1203
1204 The *name_of_module* argument can specify (as a string or tuple thereof) what
Serhiy Storchakabac2d5b2018-03-28 22:14:26 +03001205 module(s) an API could be defined in order to be detected as a public
Martin Panterd226d302015-11-14 11:47:00 +00001206 API. One case for this is when *module* imports part of its public API from
1207 other modules, possibly a C backend (like ``csv`` and its ``_csv``).
1208
1209 The *extra* argument can be a set of names that wouldn't otherwise be automatically
1210 detected as "public", like objects without a proper ``__module__``
1211 attribute. If provided, it will be added to the automatically detected ones.
1212
1213 The *blacklist* argument can be a set of names that must not be treated as part of
1214 the public API even though their names indicate otherwise.
1215
1216 Example use::
1217
1218 import bar
1219 import foo
1220 import unittest
1221 from test import support
1222
1223 class MiscTestCase(unittest.TestCase):
1224 def test__all__(self):
1225 support.check__all__(self, foo)
1226
1227 class OtherTestCase(unittest.TestCase):
1228 def test__all__(self):
1229 extra = {'BAR_CONST', 'FOO_CONST'}
1230 blacklist = {'baz'} # Undocumented name.
1231 # bar imports part of its API from _bar.
1232 support.check__all__(self, bar, ('bar', '_bar'),
1233 extra=extra, blacklist=blacklist)
1234
1235 .. versionadded:: 3.6
1236
1237
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001238The :mod:`test.support` module defines the following classes:
Georg Brandl116aa622007-08-15 14:28:22 +00001239
Georg Brandl7f01a132009-09-16 15:58:14 +00001240.. class:: TransientResource(exc, **kwargs)
Georg Brandl116aa622007-08-15 14:28:22 +00001241
1242 Instances are a context manager that raises :exc:`ResourceDenied` if the
1243 specified exception type is raised. Any keyword arguments are treated as
1244 attribute/value pairs to be compared against any exception raised within the
1245 :keyword:`with` statement. Only if all pairs match properly against
1246 attributes on the exception is :exc:`ResourceDenied` raised.
1247
Georg Brandl116aa622007-08-15 14:28:22 +00001248
1249.. class:: EnvironmentVarGuard()
1250
Florent Xicluna53b506be2010-03-18 20:00:57 +00001251 Class used to temporarily set or unset environment variables. Instances can
1252 be used as a context manager and have a complete dictionary interface for
1253 querying/modifying the underlying ``os.environ``. After exit from the
1254 context manager all changes to environment variables done through this
1255 instance will be rolled back.
Georg Brandl116aa622007-08-15 14:28:22 +00001256
Georg Brandl705d9d52009-05-05 09:29:50 +00001257 .. versionchanged:: 3.1
Walter Dörwald155374d2009-05-01 19:58:58 +00001258 Added dictionary interface.
Georg Brandl116aa622007-08-15 14:28:22 +00001259
1260.. method:: EnvironmentVarGuard.set(envvar, value)
1261
Florent Xicluna53b506be2010-03-18 20:00:57 +00001262 Temporarily set the environment variable ``envvar`` to the value of
1263 ``value``.
Georg Brandl116aa622007-08-15 14:28:22 +00001264
1265
1266.. method:: EnvironmentVarGuard.unset(envvar)
1267
1268 Temporarily unset the environment variable ``envvar``.
Nick Coghlanb1304932008-07-13 12:25:08 +00001269
Walter Dörwald155374d2009-05-01 19:58:58 +00001270
Antoine Pitrou77e904e2013-10-08 23:04:32 +02001271.. class:: SuppressCrashReport()
1272
1273 A context manager used to try to prevent crash dialog popups on tests that
1274 are expected to crash a subprocess.
1275
1276 On Windows, it disables Windows Error Reporting dialogs using
Georg Brandl5d941342016-02-26 19:37:12 +01001277 `SetErrorMode <https://msdn.microsoft.com/en-us/library/windows/desktop/ms680621.aspx>`_.
Antoine Pitrou77e904e2013-10-08 23:04:32 +02001278
1279 On UNIX, :func:`resource.setrlimit` is used to set
1280 :attr:`resource.RLIMIT_CORE`'s soft limit to 0 to prevent coredump file
1281 creation.
1282
1283 On both platforms, the old value is restored by :meth:`__exit__`.
1284
1285
Cheryl Sabella988fb282018-02-11 08:10:42 -05001286.. class:: CleanImport(*module_names)
1287
1288 A context manager to force import to return a new module reference. This
1289 is useful for testing module-level behaviors, such as the emission of a
1290 DeprecationWarning on import. Example usage::
1291
1292 with CleanImport('foo'):
1293 importlib.import_module('foo') # New reference.
1294
1295
1296.. class:: DirsOnSysPath(*paths)
1297
1298 A context manager to temporarily add directories to sys.path.
1299
1300 This makes a copy of :data:`sys.path`, appends any directories given
1301 as positional arguments, then reverts :data:`sys.path` to the copied
1302 settings when the context ends.
1303
1304 Note that *all* :data:`sys.path` modifications in the body of the
1305 context manager, including replacement of the object,
1306 will be reverted at the end of the block.
1307
1308
1309.. class:: SaveSignals()
1310
1311 Class to save and restore signal handlers registered by the Python signal
1312 handler.
1313
1314
1315.. class:: Matcher()
1316
1317 .. method:: matches(self, d, **kwargs)
1318
1319 Try to match a single dict with the supplied arguments.
1320
1321
1322 .. method:: match_value(self, k, dv, v)
1323
1324 Try to match a single stored value (*dv*) with a supplied value (*v*).
1325
1326
Benjamin Petersonfcf5d632008-10-16 23:24:44 +00001327.. class:: WarningsRecorder()
1328
1329 Class used to record warnings for unit tests. See documentation of
1330 :func:`check_warnings` above for more details.
Cheryl Sabella988fb282018-02-11 08:10:42 -05001331
1332
1333.. class:: BasicTestRunner()
1334
1335 .. method:: run(test)
1336
1337 Run *test* and return the result.
1338
1339
Serhiy Storchakab21d1552018-03-02 11:53:51 +02001340.. class:: FakePath(path)
1341
1342 Simple :term:`path-like object`. It implements the :meth:`__fspath__`
1343 method which just returns the *path* argument. If *path* is an exception,
1344 it will be raised in :meth:`!__fspath__`.
1345
1346
Serhiy Storchaka16994912020-04-25 10:06:29 +03001347:mod:`test.support.socket_helper` --- Utilities for socket tests
1348================================================================
1349
1350.. module:: test.support.socket_helper
1351 :synopsis: Support for socket tests.
1352
1353
1354The :mod:`test.support.socket_helper` module provides support for socket tests.
1355
1356.. versionadded:: 3.9
1357
1358
1359.. data:: IPV6_ENABLED
1360
1361 Set to ``True`` if IPv6 is enabled on this host, ``False`` otherwise.
1362
1363
1364.. function:: find_unused_port(family=socket.AF_INET, socktype=socket.SOCK_STREAM)
1365
1366 Returns an unused port that should be suitable for binding. This is
1367 achieved by creating a temporary socket with the same family and type as
1368 the ``sock`` parameter (default is :const:`~socket.AF_INET`,
1369 :const:`~socket.SOCK_STREAM`),
1370 and binding it to the specified host address (defaults to ``0.0.0.0``)
1371 with the port set to 0, eliciting an unused ephemeral port from the OS.
1372 The temporary socket is then closed and deleted, and the ephemeral port is
1373 returned.
1374
1375 Either this method or :func:`bind_port` should be used for any tests
1376 where a server socket needs to be bound to a particular port for the
1377 duration of the test.
1378 Which one to use depends on whether the calling code is creating a Python
1379 socket, or if an unused port needs to be provided in a constructor
1380 or passed to an external program (i.e. the ``-accept`` argument to
1381 openssl's s_server mode). Always prefer :func:`bind_port` over
1382 :func:`find_unused_port` where possible. Using a hard coded port is
1383 discouraged since it can make multiple instances of the test impossible to
1384 run simultaneously, which is a problem for buildbots.
1385
1386
1387.. function:: bind_port(sock, host=HOST)
1388
1389 Bind the socket to a free port and return the port number. Relies on
1390 ephemeral ports in order to ensure we are using an unbound port. This is
1391 important as many tests may be running simultaneously, especially in a
1392 buildbot environment. This method raises an exception if the
1393 ``sock.family`` is :const:`~socket.AF_INET` and ``sock.type`` is
1394 :const:`~socket.SOCK_STREAM`, and the socket has
1395 :const:`~socket.SO_REUSEADDR` or :const:`~socket.SO_REUSEPORT` set on it.
1396 Tests should never set these socket options for TCP/IP sockets.
1397 The only case for setting these options is testing multicasting via
1398 multiple UDP sockets.
1399
1400 Additionally, if the :const:`~socket.SO_EXCLUSIVEADDRUSE` socket option is
1401 available (i.e. on Windows), it will be set on the socket. This will
1402 prevent anyone else from binding to our host/port for the duration of the
1403 test.
1404
1405
1406.. function:: bind_unix_socket(sock, addr)
1407
1408 Bind a unix socket, raising :exc:`unittest.SkipTest` if
1409 :exc:`PermissionError` is raised.
1410
1411
1412.. decorator:: skip_unless_bind_unix_socket
1413
1414 A decorator for running tests that require a functional ``bind()`` for Unix
1415 sockets.
1416
1417
Serhiy Storchakabfb1cf42020-04-29 10:36:20 +03001418.. function:: transient_internet(resource_name, *, timeout=30.0, errnos=())
1419
1420 A context manager that raises :exc:`~test.support.ResourceDenied` when
1421 various issues with the internet connection manifest themselves as
1422 exceptions.
1423
1424
Cheryl Sabella988fb282018-02-11 08:10:42 -05001425:mod:`test.support.script_helper` --- Utilities for the Python execution tests
1426==============================================================================
1427
1428.. module:: test.support.script_helper
1429 :synopsis: Support for Python's script execution tests.
1430
1431
1432The :mod:`test.support.script_helper` module provides support for Python's
1433script execution tests.
1434
1435.. function:: interpreter_requires_environment()
1436
1437 Return ``True`` if ``sys.executable interpreter`` requires environment
1438 variables in order to be able to run at all.
1439
1440 This is designed to be used with ``@unittest.skipIf()`` to annotate tests
1441 that need to use an ``assert_python*()`` function to launch an isolated
1442 mode (``-I``) or no environment mode (``-E``) sub-interpreter process.
1443
1444 A normal build & test does not run into this situation but it can happen
1445 when trying to run the standard library test suite from an interpreter that
1446 doesn't have an obvious home with Python's current home finding logic.
1447
1448 Setting :envvar:`PYTHONHOME` is one way to get most of the testsuite to run
1449 in that situation. :envvar:`PYTHONPATH` or :envvar:`PYTHONUSERSITE` are
1450 other common environment variables that might impact whether or not the
1451 interpreter can start.
1452
1453
1454.. function:: run_python_until_end(*args, **env_vars)
1455
1456 Set up the environment based on *env_vars* for running the interpreter
Nick Coghland5d9e022018-03-25 23:03:10 +10001457 in a subprocess. The values can include ``__isolated``, ``__cleanenv``,
1458 ``__cwd``, and ``TERM``.
Cheryl Sabella988fb282018-02-11 08:10:42 -05001459
Victor Stinner6cac1132019-12-08 08:38:16 +01001460 .. versionchanged:: 3.9
1461 The function no longer strips whitespaces from *stderr*.
1462
Cheryl Sabella988fb282018-02-11 08:10:42 -05001463
1464.. function:: assert_python_ok(*args, **env_vars)
1465
1466 Assert that running the interpreter with *args* and optional environment
1467 variables *env_vars* succeeds (``rc == 0``) and return a ``(return code,
1468 stdout, stderr)`` tuple.
1469
1470 If the ``__cleanenv`` keyword is set, *env_vars* is used as a fresh
1471 environment.
1472
1473 Python is started in isolated mode (command line option ``-I``),
1474 except if the ``__isolated`` keyword is set to ``False``.
1475
Victor Stinner6cac1132019-12-08 08:38:16 +01001476 .. versionchanged:: 3.9
1477 The function no longer strips whitespaces from *stderr*.
1478
Cheryl Sabella988fb282018-02-11 08:10:42 -05001479
1480.. function:: assert_python_failure(*args, **env_vars)
1481
1482 Assert that running the interpreter with *args* and optional environment
1483 variables *env_vars* fails (``rc != 0``) and return a ``(return code,
1484 stdout, stderr)`` tuple.
1485
1486 See :func:`assert_python_ok` for more options.
1487
Victor Stinner6cac1132019-12-08 08:38:16 +01001488 .. versionchanged:: 3.9
1489 The function no longer strips whitespaces from *stderr*.
1490
Cheryl Sabella988fb282018-02-11 08:10:42 -05001491
1492.. function:: spawn_python(*args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, **kw)
1493
1494 Run a Python subprocess with the given arguments.
1495
1496 *kw* is extra keyword args to pass to :func:`subprocess.Popen`. Returns a
1497 :class:`subprocess.Popen` object.
1498
1499
1500.. function:: kill_python(p)
1501
1502 Run the given :class:`subprocess.Popen` process until completion and return
1503 stdout.
1504
1505
1506.. function:: make_script(script_dir, script_basename, source, omit_suffix=False)
1507
1508 Create script containing *source* in path *script_dir* and *script_basename*.
1509 If *omit_suffix* is ``False``, append ``.py`` to the name. Return the full
1510 script path.
1511
1512
1513.. function:: make_zip_script(zip_dir, zip_basename, script_name, name_in_zip=None)
1514
1515 Create zip file at *zip_dir* and *zip_basename* with extension ``zip`` which
1516 contains the files in *script_name*. *name_in_zip* is the archive name.
1517 Return a tuple containing ``(full path, full path of archive name)``.
1518
1519
1520.. function:: make_pkg(pkg_dir, init_source='')
1521
1522 Create a directory named *pkg_dir* containing an ``__init__`` file with
1523 *init_source* as its contents.
1524
1525
1526.. function:: make_zip_pkg(zip_dir, zip_basename, pkg_name, script_basename, \
1527 source, depth=1, compiled=False)
1528
1529 Create a zip package directory with a path of *zip_dir* and *zip_basename*
1530 containing an empty ``__init__`` file and a file *script_basename*
1531 containing the *source*. If *compiled* is ``True``, both source files will
1532 be compiled and added to the zip package. Return a tuple of the full zip
1533 path and the archive name for the zip file.
Joannah Nanjekye92777d52019-09-12 10:02:59 +01001534
1535
1536:mod:`test.support.bytecode_helper` --- Support tools for testing correct bytecode generation
1537=============================================================================================
1538
1539.. module:: test.support.bytecode_helper
1540 :synopsis: Support tools for testing correct bytecode generation.
1541
1542The :mod:`test.support.bytecode_helper` module provides support for testing
1543and inspecting bytecode generation.
1544
Gurupad Hegde6c7bb382019-12-28 17:16:02 -05001545The module defines the following class:
Joannah Nanjekye92777d52019-09-12 10:02:59 +01001546
1547.. class:: BytecodeTestCase(unittest.TestCase)
1548
1549 This class has custom assertion methods for inspecting bytecode.
1550
1551.. method:: BytecodeTestCase.get_disassembly_as_string(co)
1552
1553 Return the disassembly of *co* as string.
1554
1555
1556.. method:: BytecodeTestCase.assertInBytecode(x, opname, argval=_UNSPECIFIED)
1557
1558 Return instr if *opname* is found, otherwise throws :exc:`AssertionError`.
1559
1560
1561.. method:: BytecodeTestCase.assertNotInBytecode(x, opname, argval=_UNSPECIFIED)
1562
1563 Throws :exc:`AssertionError` if *opname* is found.
Hai Shie80697d2020-05-28 06:10:27 +08001564
1565
1566:mod:`test.support.threading_helper` --- Utilities for threading tests
1567======================================================================
1568
1569.. module:: test.support.threading_helper
1570 :synopsis: Support for threading tests.
1571
1572The :mod:`test.support.threading_helper` module provides support for threading tests.
1573
1574.. versionadded:: 3.10
1575
1576
1577.. function:: join_thread(thread, timeout=None)
1578
1579 Join a *thread* within *timeout*. Raise an :exc:`AssertionError` if thread
1580 is still alive after *timeout* seconds.
1581
1582
1583.. decorator:: reap_threads(func)
1584
1585 Decorator to ensure the threads are cleaned up even if the test fails.
1586
1587
1588.. function:: start_threads(threads, unlock=None)
1589
1590 Context manager to start *threads*. It attempts to join the threads upon
1591 exit.
1592
1593
1594.. function:: threading_cleanup(*original_values)
1595
1596 Cleanup up threads not specified in *original_values*. Designed to emit
1597 a warning if a test leaves running threads in the background.
1598
1599
1600.. function:: threading_setup()
1601
1602 Return current thread count and copy of dangling threads.
1603
1604
1605.. function:: wait_threads_exit(timeout=None)
1606
1607 Context manager to wait until all threads created in the ``with`` statement
1608 exit.
1609
1610
1611.. function:: catch_threading_exception()
1612
1613 Context manager catching :class:`threading.Thread` exception using
1614 :func:`threading.excepthook`.
1615
1616 Attributes set when an exception is catched:
1617
1618 * ``exc_type``
1619 * ``exc_value``
1620 * ``exc_traceback``
1621 * ``thread``
1622
1623 See :func:`threading.excepthook` documentation.
1624
1625 These attributes are deleted at the context manager exit.
1626
1627 Usage::
1628
1629 with threading_helper.catch_threading_exception() as cm:
1630 # code spawning a thread which raises an exception
1631 ...
1632
1633 # check the thread exception, use cm attributes:
1634 # exc_type, exc_value, exc_traceback, thread
1635 ...
1636
1637 # exc_type, exc_value, exc_traceback, thread attributes of cm no longer
1638 # exists at this point
1639 # (to avoid reference cycles)
1640
1641 .. versionadded:: 3.8