blob: e4f779bd83eb87087fbf99b4d56a479678b0d07e [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
Victor Stinner24c62582019-10-30 12:41:43 +0100250.. data:: LOOPBACK_TIMEOUT
251
252 Timeout in seconds for tests using a network server listening on the network
253 local loopback interface like ``127.0.0.1``.
254
255 The timeout is long enough to prevent test failure: it takes into account
256 that the client and the server can run in different threads or even
257 different processes.
258
259 The timeout should be long enough for :meth:`~socket.socket.connect`,
260 :meth:`~socket.socket.recv` and :meth:`~socket.socket.send` methods of
261 :class:`socket.socket`.
262
263 Its default value is 5 seconds.
264
265 See also :data:`INTERNET_TIMEOUT`.
266
267
268.. data:: INTERNET_TIMEOUT
269
270 Timeout in seconds for network requests going to the Internet.
271
272 The timeout is short enough to prevent a test to wait for too long if the
273 Internet request is blocked for whatever reason.
274
275 Usually, a timeout using :data:`INTERNET_TIMEOUT` should not mark a test as
276 failed, but skip the test instead: see
Serhiy Storchakabfb1cf42020-04-29 10:36:20 +0300277 :func:`~test.support.socket_helper.transient_internet`.
Victor Stinner24c62582019-10-30 12:41:43 +0100278
279 Its default value is 1 minute.
280
281 See also :data:`LOOPBACK_TIMEOUT`.
282
283
284.. data:: SHORT_TIMEOUT
285
286 Timeout in seconds to mark a test as failed if the test takes "too long".
287
288 The timeout value depends on the regrtest ``--timeout`` command line option.
289
290 If a test using :data:`SHORT_TIMEOUT` starts to fail randomly on slow
291 buildbots, use :data:`LONG_TIMEOUT` instead.
292
293 Its default value is 30 seconds.
294
295
296.. data:: LONG_TIMEOUT
297
298 Timeout in seconds to detect when a test hangs.
299
300 It is long enough to reduce the risk of test failure on the slowest Python
301 buildbots. It should not be used to mark a test as failed if the test takes
302 "too long". The timeout value depends on the regrtest ``--timeout`` command
303 line option.
304
305 Its default value is 5 minutes.
306
307 See also :data:`LOOPBACK_TIMEOUT`, :data:`INTERNET_TIMEOUT` and
308 :data:`SHORT_TIMEOUT`.
309
310
Cheryl Sabella988fb282018-02-11 08:10:42 -0500311.. data:: PGO
312
313 Set when tests can be skipped when they are not useful for PGO.
314
315
316.. data:: PIPE_MAX_SIZE
317
318 A constant that is likely larger than the underlying OS pipe buffer size,
319 to make writes blocking.
320
321
322.. data:: SOCK_MAX_SIZE
323
324 A constant that is likely larger than the underlying OS socket buffer size,
325 to make writes blocking.
326
327
328.. data:: TEST_SUPPORT_DIR
329
330 Set to the top level directory that contains :mod:`test.support`.
331
332
333.. data:: TEST_HOME_DIR
334
335 Set to the top level directory for the test package.
336
337
338.. data:: TEST_DATA_DIR
339
340 Set to the ``data`` directory within the test package.
341
342
343.. data:: MAX_Py_ssize_t
344
345 Set to :data:`sys.maxsize` for big memory tests.
346
347
348.. data:: max_memuse
349
350 Set by :func:`set_memlimit` as the memory limit for big memory tests.
351 Limited by :data:`MAX_Py_ssize_t`.
352
353
354.. data:: real_max_memuse
355
356 Set by :func:`set_memlimit` as the memory limit for big memory tests. Not
357 limited by :data:`MAX_Py_ssize_t`.
358
359
360.. data:: MISSING_C_DOCSTRINGS
361
362 Return ``True`` if running on CPython, not on Windows, and configuration
363 not set with ``WITH_DOC_STRINGS``.
364
365
366.. data:: HAVE_DOCSTRINGS
367
368 Check for presence of docstrings.
369
Serhiy Storchaka17e52642019-08-04 12:38:46 +0300370
Stéphane Wirtela40681d2019-02-22 14:45:36 +0100371.. data:: TEST_HTTP_URL
372
373 Define the URL of a dedicated HTTP server for the network tests.
374
Cheryl Sabella988fb282018-02-11 08:10:42 -0500375
Serhiy Storchaka17e52642019-08-04 12:38:46 +0300376.. data:: ALWAYS_EQ
377
378 Object that is equal to anything. Used to test mixed type comparison.
379
380
Serhiy Storchaka18b711c2019-08-04 14:12:48 +0300381.. data:: NEVER_EQ
382
383 Object that is not equal to anything (even to :data:`ALWAYS_EQ`).
384 Used to test mixed type comparison.
385
386
Serhiy Storchaka17e52642019-08-04 12:38:46 +0300387.. data:: LARGEST
388
389 Object that is greater than anything (except itself).
390 Used to test mixed type comparison.
391
392
393.. data:: SMALLEST
394
395 Object that is less than anything (except itself).
396 Used to test mixed type comparison.
397
Cheryl Sabella988fb282018-02-11 08:10:42 -0500398
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000399The :mod:`test.support` module defines the following functions:
Georg Brandl116aa622007-08-15 14:28:22 +0000400
Georg Brandl116aa622007-08-15 14:28:22 +0000401.. function:: is_resource_enabled(resource)
402
Eli Benderskye1689652011-05-06 09:29:27 +0300403 Return ``True`` if *resource* is enabled and available. The list of
Georg Brandl116aa622007-08-15 14:28:22 +0000404 available resources is only set when :mod:`test.regrtest` is executing the
405 tests.
406
407
Cheryl Sabella988fb282018-02-11 08:10:42 -0500408.. function:: python_is_optimized()
409
410 Return ``True`` if Python was not built with ``-O0`` or ``-Og``.
411
412
413.. function:: with_pymalloc()
414
415 Return :data:`_testcapi.WITH_PYMALLOC`.
416
417
Georg Brandl7f01a132009-09-16 15:58:14 +0000418.. function:: requires(resource, msg=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000419
Florent Xiclunab14930c2010-03-13 15:26:44 +0000420 Raise :exc:`ResourceDenied` if *resource* is not available. *msg* is the
Florent Xicluna53b506be2010-03-18 20:00:57 +0000421 argument to :exc:`ResourceDenied` if it is raised. Always returns
Eli Benderskye1689652011-05-06 09:29:27 +0300422 ``True`` if called by a function whose ``__name__`` is ``'__main__'``.
Florent Xicluna53b506be2010-03-18 20:00:57 +0000423 Used when tests are executed by :mod:`test.regrtest`.
Georg Brandl116aa622007-08-15 14:28:22 +0000424
425
Cheryl Sabella988fb282018-02-11 08:10:42 -0500426.. function:: system_must_validate_cert(f)
427
428 Raise :exc:`unittest.SkipTest` on TLS certification validation failures.
429
430
431.. function:: sortdict(dict)
432
433 Return a repr of *dict* with keys sorted.
434
435
Nick Coghlan0494c2a2013-09-08 11:40:34 +1000436.. function:: findfile(filename, subdir=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000437
Florent Xicluna53b506be2010-03-18 20:00:57 +0000438 Return the path to the file named *filename*. If no match is found
439 *filename* is returned. This does not equal a failure since it could be the
440 path to the file.
Georg Brandl116aa622007-08-15 14:28:22 +0000441
Cheryl Sabella988fb282018-02-11 08:10:42 -0500442 Setting *subdir* indicates a relative path to use to find the file
443 rather than looking directly in the path directories.
444
445
Cheryl Sabella988fb282018-02-11 08:10:42 -0500446.. function:: match_test(test)
447
448 Match *test* to patterns set in :func:`set_match_tests`.
449
450
451.. function:: set_match_tests(patterns)
452
453 Define match test with regular expression *patterns*.
Nick Coghlan0494c2a2013-09-08 11:40:34 +1000454
Georg Brandl116aa622007-08-15 14:28:22 +0000455
Andre Delfinodcc997c2020-12-16 22:37:28 -0300456.. function:: run_unittest(*classes)
Georg Brandl116aa622007-08-15 14:28:22 +0000457
458 Execute :class:`unittest.TestCase` subclasses passed to the function. The
Florent Xicluna53b506be2010-03-18 20:00:57 +0000459 function scans the classes for methods starting with the prefix ``test_``
460 and executes the tests individually.
Georg Brandl116aa622007-08-15 14:28:22 +0000461
462 It is also legal to pass strings as parameters; these should be keys in
463 ``sys.modules``. Each associated module will be scanned by
464 ``unittest.TestLoader.loadTestsFromModule()``. This is usually seen in the
465 following :func:`test_main` function::
466
467 def test_main():
Nick Coghlan47384702009-04-22 16:13:36 +0000468 support.run_unittest(__name__)
Georg Brandl116aa622007-08-15 14:28:22 +0000469
470 This will run all tests defined in the named module.
471
Georg Brandl116aa622007-08-15 14:28:22 +0000472
Cheryl Sabella988fb282018-02-11 08:10:42 -0500473.. function:: run_doctest(module, verbosity=None, optionflags=0)
Eli Benderskye1689652011-05-06 09:29:27 +0300474
475 Run :func:`doctest.testmod` on the given *module*. Return
476 ``(failure_count, test_count)``.
477
478 If *verbosity* is ``None``, :func:`doctest.testmod` is run with verbosity
479 set to :data:`verbose`. Otherwise, it is run with verbosity set to
Cheryl Sabella988fb282018-02-11 08:10:42 -0500480 ``None``. *optionflags* is passed as ``optionflags`` to
481 :func:`doctest.testmod`.
482
483
484.. function:: setswitchinterval(interval)
485
486 Set the :func:`sys.setswitchinterval` to the given *interval*. Defines
487 a minimum interval for Android systems to prevent the system from hanging.
488
489
490.. function:: check_impl_detail(**guards)
491
492 Use this check to guard CPython's implementation-specific tests or to
493 run them only on the implementations guarded by the arguments::
494
495 check_impl_detail() # Only on CPython (default).
496 check_impl_detail(jython=True) # Only on Jython.
497 check_impl_detail(cpython=False) # Everywhere except CPython.
498
Eli Benderskye1689652011-05-06 09:29:27 +0300499
Cheryl Sabella988fb282018-02-11 08:10:42 -0500500.. function:: set_memlimit(limit)
501
502 Set the values for :data:`max_memuse` and :data:`real_max_memuse` for big
503 memory tests.
504
505
506.. function:: record_original_stdout(stdout)
507
508 Store the value from *stdout*. It is meant to hold the stdout at the
509 time the regrtest began.
510
511
512.. function:: get_original_stdout
513
514 Return the original stdout set by :func:`record_original_stdout` or
515 ``sys.stdout`` if it's not set.
516
517
Cheryl Sabella988fb282018-02-11 08:10:42 -0500518.. function:: args_from_interpreter_flags()
519
520 Return a list of command line arguments reproducing the current settings
521 in ``sys.flags`` and ``sys.warnoptions``.
522
523
524.. function:: optim_args_from_interpreter_flags()
525
526 Return a list of command line arguments reproducing the current
527 optimization settings in ``sys.flags``.
528
529
R David Murray5a33f812013-07-11 12:28:40 -0400530.. function:: captured_stdin()
531 captured_stdout()
532 captured_stderr()
Thomas Woutersed03b412007-08-28 21:37:11 +0000533
R David Murray5a33f812013-07-11 12:28:40 -0400534 A context managers that temporarily replaces the named stream with
535 :class:`io.StringIO` object.
Thomas Woutersed03b412007-08-28 21:37:11 +0000536
R David Murray5a33f812013-07-11 12:28:40 -0400537 Example use with output streams::
Thomas Woutersed03b412007-08-28 21:37:11 +0000538
R David Murray5a33f812013-07-11 12:28:40 -0400539 with captured_stdout() as stdout, captured_stderr() as stderr:
Collin Winterc79461b2007-09-01 23:34:30 +0000540 print("hello")
R David Murray5a33f812013-07-11 12:28:40 -0400541 print("error", file=sys.stderr)
542 assert stdout.getvalue() == "hello\n"
543 assert stderr.getvalue() == "error\n"
544
545 Example use with input stream::
546
547 with captured_stdin() as stdin:
548 stdin.write('hello\n')
549 stdin.seek(0)
550 # call test code that consumes from sys.stdin
551 captured = input()
552 self.assertEqual(captured, "hello")
Thomas Woutersed03b412007-08-28 21:37:11 +0000553
Thomas Woutersed03b412007-08-28 21:37:11 +0000554
Cheryl Sabella988fb282018-02-11 08:10:42 -0500555.. function:: disable_faulthandler()
556
557 A context manager that replaces ``sys.stderr`` with ``sys.__stderr__``.
558
559
560.. function:: gc_collect()
561
562 Force as many objects as possible to be collected. This is needed because
563 timely deallocation is not guaranteed by the garbage collector. This means
564 that ``__del__`` methods may be called later than expected and weakrefs
565 may remain alive for longer than expected.
566
567
568.. function:: disable_gc()
569
570 A context manager that disables the garbage collector upon entry and
571 reenables it upon exit.
572
573
574.. function:: swap_attr(obj, attr, new_val)
575
576 Context manager to swap out an attribute with a new object.
577
578 Usage::
579
580 with swap_attr(obj, "attr", 5):
581 ...
582
583 This will set ``obj.attr`` to 5 for the duration of the ``with`` block,
584 restoring the old value at the end of the block. If ``attr`` doesn't
585 exist on ``obj``, it will be created and then deleted at the end of the
586 block.
587
588 The old value (or ``None`` if it doesn't exist) will be assigned to the
589 target of the "as" clause, if there is one.
590
591
592.. function:: swap_item(obj, attr, new_val)
593
594 Context manager to swap out an item with a new object.
595
596 Usage::
597
598 with swap_item(obj, "item", 5):
599 ...
600
601 This will set ``obj["item"]`` to 5 for the duration of the ``with`` block,
602 restoring the old value at the end of the block. If ``item`` doesn't
603 exist on ``obj``, it will be created and then deleted at the end of the
604 block.
605
606 The old value (or ``None`` if it doesn't exist) will be assigned to the
607 target of the "as" clause, if there is one.
608
609
Victor Stinnerd663d342020-04-23 19:03:52 +0200610.. function:: print_warning(msg)
611
612 Print a warning into :data:`sys.__stderr__`. Format the message as:
613 ``f"Warning -- {msg}"``. If *msg* is made of multiple lines, add
614 ``"Warning -- "`` prefix to each line.
615
616 .. versionadded:: 3.9
617
618
Victor Stinner278c1e12020-03-31 20:08:12 +0200619.. function:: wait_process(pid, *, exitcode, timeout=None)
620
621 Wait until process *pid* completes and check that the process exit code is
622 *exitcode*.
623
624 Raise an :exc:`AssertionError` if the process exit code is not equal to
625 *exitcode*.
626
627 If the process runs longer than *timeout* seconds (:data:`SHORT_TIMEOUT` by
628 default), kill the process and raise an :exc:`AssertionError`. The timeout
629 feature is not available on Windows.
630
631 .. versionadded:: 3.9
632
633
Cheryl Sabella988fb282018-02-11 08:10:42 -0500634.. function:: calcobjsize(fmt)
635
636 Return :func:`struct.calcsize` for ``nP{fmt}0n`` or, if ``gettotalrefcount``
637 exists, ``2PnP{fmt}0P``.
638
639
640.. function:: calcvobjsize(fmt)
641
642 Return :func:`struct.calcsize` for ``nPn{fmt}0n`` or, if ``gettotalrefcount``
643 exists, ``2PnPn{fmt}0P``.
644
645
646.. function:: checksizeof(test, o, size)
647
648 For testcase *test*, assert that the ``sys.getsizeof`` for *o* plus the GC
649 header size equals *size*.
650
651
Nick Coghlan2496f332011-09-19 20:26:31 +1000652.. decorator:: anticipate_failure(condition)
653
654 A decorator to conditionally mark tests with
655 :func:`unittest.expectedFailure`. Any use of this decorator should
656 have an associated comment identifying the relevant tracker issue.
657
658
659.. decorator:: run_with_locale(catstr, *locales)
Eli Benderskye1689652011-05-06 09:29:27 +0300660
661 A decorator for running a function in a different locale, correctly
662 resetting it after it has finished. *catstr* is the locale category as
663 a string (for example ``"LC_ALL"``). The *locales* passed will be tried
664 sequentially, and the first valid locale will be used.
665
666
Cheryl Sabella988fb282018-02-11 08:10:42 -0500667.. decorator:: run_with_tz(tz)
668
669 A decorator for running a function in a specific timezone, correctly
670 resetting it after it has finished.
671
672
673.. decorator:: requires_freebsd_version(*min_version)
674
675 Decorator for the minimum version when running test on FreeBSD. If the
676 FreeBSD version is less than the minimum, raise :exc:`unittest.SkipTest`.
677
678
679.. decorator:: requires_linux_version(*min_version)
680
681 Decorator for the minimum version when running test on Linux. If the
682 Linux version is less than the minimum, raise :exc:`unittest.SkipTest`.
683
684
685.. decorator:: requires_mac_version(*min_version)
686
687 Decorator for the minimum version when running test on Mac OS X. If the
688 MAC OS X version is less than the minimum, raise :exc:`unittest.SkipTest`.
689
690
691.. decorator:: requires_IEEE_754
692
693 Decorator for skipping tests on non-IEEE 754 platforms.
694
695
696.. decorator:: requires_zlib
697
698 Decorator for skipping tests if :mod:`zlib` doesn't exist.
699
700
701.. decorator:: requires_gzip
702
703 Decorator for skipping tests if :mod:`gzip` doesn't exist.
704
705
706.. decorator:: requires_bz2
707
708 Decorator for skipping tests if :mod:`bz2` doesn't exist.
709
710
711.. decorator:: requires_lzma
712
713 Decorator for skipping tests if :mod:`lzma` doesn't exist.
714
715
716.. decorator:: requires_resource(resource)
717
718 Decorator for skipping tests if *resource* is not available.
719
720
721.. decorator:: requires_docstrings
722
723 Decorator for only running the test if :data:`HAVE_DOCSTRINGS`.
724
725
726.. decorator:: cpython_only(test)
727
728 Decorator for tests only applicable to CPython.
729
730
731.. decorator:: impl_detail(msg=None, **guards)
732
733 Decorator for invoking :func:`check_impl_detail` on *guards*. If that
734 returns ``False``, then uses *msg* as the reason for skipping the test.
735
736
737.. decorator:: no_tracing(func)
738
739 Decorator to temporarily turn off tracing for the duration of the test.
740
741
742.. decorator:: refcount_test(test)
743
744 Decorator for tests which involve reference counting. The decorator does
745 not run the test if it is not run by CPython. Any trace function is unset
746 for the duration of the test to prevent unexpected refcounts caused by
747 the trace function.
748
749
Cheryl Sabella988fb282018-02-11 08:10:42 -0500750.. decorator:: bigmemtest(size, memuse, dry_run=True)
751
752 Decorator for bigmem tests.
753
754 *size* is a requested size for the test (in arbitrary, test-interpreted
755 units.) *memuse* is the number of bytes per unit for the test, or a good
756 estimate of it. For example, a test that needs two byte buffers, of 4 GiB
757 each, could be decorated with ``@bigmemtest(size=_4G, memuse=2)``.
758
759 The *size* argument is normally passed to the decorated test method as an
760 extra argument. If *dry_run* is ``True``, the value passed to the test
761 method may be less than the requested value. If *dry_run* is ``False``, it
762 means the test doesn't support dummy runs when ``-M`` is not specified.
763
764
765.. decorator:: bigaddrspacetest(f)
766
767 Decorator for tests that fill the address space. *f* is the function to
768 wrap.
769
770
Cheryl Sabella988fb282018-02-11 08:10:42 -0500771.. function:: check_syntax_error(testcase, statement, errtext='', *, lineno=None, offset=None)
772
773 Test for syntax errors in *statement* by attempting to compile *statement*.
774 *testcase* is the :mod:`unittest` instance for the test. *errtext* is the
Serhiy Storchakae7a4bb52019-02-19 08:30:15 +0200775 regular expression which should match the string representation of the
776 raised :exc:`SyntaxError`. If *lineno* is not ``None``, compares to
777 the line of the exception. If *offset* is not ``None``, compares to
778 the offset of the exception.
779
780
Cheryl Sabella988fb282018-02-11 08:10:42 -0500781.. function:: open_urlresource(url, *args, **kw)
782
783 Open *url*. If open fails, raises :exc:`TestFailed`.
784
785
Cheryl Sabella988fb282018-02-11 08:10:42 -0500786.. function:: reap_children()
787
788 Use this at the end of ``test_main`` whenever sub-processes are started.
789 This will help ensure that no extra children (zombies) stick around to
790 hog resources and create problems when looking for refleaks.
791
792
793.. function:: get_attribute(obj, name)
794
795 Get an attribute, raising :exc:`unittest.SkipTest` if :exc:`AttributeError`
796 is raised.
797
798
Victor Stinner6dbbe742019-05-25 00:09:38 +0200799.. function:: catch_unraisable_exception()
800
801 Context manager catching unraisable exception using
802 :func:`sys.unraisablehook`.
803
Victor Stinner6d22cc82019-06-13 14:44:54 +0200804 Storing the exception value (``cm.unraisable.exc_value``) creates a
805 reference cycle. The reference cycle is broken explicitly when the context
806 manager exits.
807
Victor Stinner212646c2019-06-14 18:03:22 +0200808 Storing the object (``cm.unraisable.object``) can resurrect it if it is set
809 to an object which is being finalized. Exiting the context manager clears
810 the stored object.
Victor Stinner6d22cc82019-06-13 14:44:54 +0200811
Victor Stinner6dbbe742019-05-25 00:09:38 +0200812 Usage::
813
814 with support.catch_unraisable_exception() as cm:
815 # code creating an "unraisable exception"
816 ...
817
818 # check the unraisable exception: use cm.unraisable
819 ...
820
821 # cm.unraisable attribute no longer exists at this point
822 # (to break a reference cycle)
823
824 .. versionadded:: 3.8
825
826
Zachary Waref012ba42014-07-23 12:00:29 -0500827.. function:: load_package_tests(pkg_dir, loader, standard_tests, pattern)
828
829 Generic implementation of the :mod:`unittest` ``load_tests`` protocol for
830 use in test packages. *pkg_dir* is the root directory of the package;
831 *loader*, *standard_tests*, and *pattern* are the arguments expected by
832 ``load_tests``. In simple cases, the test package's ``__init__.py``
833 can be the following::
834
835 import os
836 from test.support import load_package_tests
837
838 def load_tests(*args):
839 return load_package_tests(os.path.dirname(__file__), *args)
840
Louie Lu7fae81e2017-04-22 14:46:18 +0800841
842.. function:: detect_api_mismatch(ref_api, other_api, *, ignore=())
Gregory P. Smith4e72cce2015-04-14 13:26:06 -0700843
Zachary Ware3d3aedc2015-07-07 00:07:25 -0500844 Returns the set of attributes, functions or methods of *ref_api* not
845 found on *other_api*, except for a defined list of items to be
846 ignored in this check specified in *ignore*.
Gregory P. Smith4e72cce2015-04-14 13:26:06 -0700847
848 By default this skips private attributes beginning with '_' but
849 includes all magic methods, i.e. those starting and ending in '__'.
850
Gregory P. Smith7c63fd32015-04-14 15:25:01 -0700851 .. versionadded:: 3.5
852
Zachary Waref012ba42014-07-23 12:00:29 -0500853
Cheryl Sabella988fb282018-02-11 08:10:42 -0500854.. function:: patch(test_instance, object_to_patch, attr_name, new_value)
855
856 Override *object_to_patch.attr_name* with *new_value*. Also add
857 cleanup procedure to *test_instance* to restore *object_to_patch* for
858 *attr_name*. The *attr_name* should be a valid attribute for
859 *object_to_patch*.
860
861
862.. function:: run_in_subinterp(code)
863
864 Run *code* in subinterpreter. Raise :exc:`unittest.SkipTest` if
865 :mod:`tracemalloc` is enabled.
866
867
868.. function:: check_free_after_iterating(test, iter, cls, args=())
869
870 Assert that *iter* is deallocated after iterating.
871
872
873.. function:: missing_compiler_executable(cmd_names=[])
874
875 Check for the existence of the compiler executables whose names are listed
876 in *cmd_names* or all the compiler executables when *cmd_names* is empty
877 and return the first missing executable or ``None`` when none is found
878 missing.
879
880
Victor Stinnerfbf43f02020-08-17 07:20:40 +0200881.. function:: check__all__(test_case, module, name_of_module=None, extra=(), not_exported=())
Martin Panterd226d302015-11-14 11:47:00 +0000882
883 Assert that the ``__all__`` variable of *module* contains all public names.
884
885 The module's public names (its API) are detected automatically
886 based on whether they match the public name convention and were defined in
887 *module*.
888
889 The *name_of_module* argument can specify (as a string or tuple thereof) what
Serhiy Storchakabac2d5b2018-03-28 22:14:26 +0300890 module(s) an API could be defined in order to be detected as a public
Martin Panterd226d302015-11-14 11:47:00 +0000891 API. One case for this is when *module* imports part of its public API from
892 other modules, possibly a C backend (like ``csv`` and its ``_csv``).
893
894 The *extra* argument can be a set of names that wouldn't otherwise be automatically
895 detected as "public", like objects without a proper ``__module__``
896 attribute. If provided, it will be added to the automatically detected ones.
897
Victor Stinnerfbf43f02020-08-17 07:20:40 +0200898 The *not_exported* argument can be a set of names that must not be treated
899 as part of the public API even though their names indicate otherwise.
Martin Panterd226d302015-11-14 11:47:00 +0000900
901 Example use::
902
903 import bar
904 import foo
905 import unittest
906 from test import support
907
908 class MiscTestCase(unittest.TestCase):
909 def test__all__(self):
910 support.check__all__(self, foo)
911
912 class OtherTestCase(unittest.TestCase):
913 def test__all__(self):
914 extra = {'BAR_CONST', 'FOO_CONST'}
Victor Stinnerfbf43f02020-08-17 07:20:40 +0200915 not_exported = {'baz'} # Undocumented name.
Martin Panterd226d302015-11-14 11:47:00 +0000916 # bar imports part of its API from _bar.
917 support.check__all__(self, bar, ('bar', '_bar'),
Victor Stinnerfbf43f02020-08-17 07:20:40 +0200918 extra=extra, not_exported=not_exported)
Martin Panterd226d302015-11-14 11:47:00 +0000919
920 .. versionadded:: 3.6
921
Victor Stinnerddbeb2f2020-06-18 14:53:19 +0200922.. function:: skip_if_broken_multiprocessing_synchronize()
923
924 Skip tests if the :mod:`multiprocessing.synchronize` module is missing, if
925 there is no available semaphore implementation, or if creating a lock raises
926 an :exc:`OSError`.
927
928 .. versionadded:: 3.10
929
Martin Panterd226d302015-11-14 11:47:00 +0000930
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000931The :mod:`test.support` module defines the following classes:
Georg Brandl116aa622007-08-15 14:28:22 +0000932
Antoine Pitrou77e904e2013-10-08 23:04:32 +0200933.. class:: SuppressCrashReport()
934
935 A context manager used to try to prevent crash dialog popups on tests that
936 are expected to crash a subprocess.
937
938 On Windows, it disables Windows Error Reporting dialogs using
Georg Brandl5d941342016-02-26 19:37:12 +0100939 `SetErrorMode <https://msdn.microsoft.com/en-us/library/windows/desktop/ms680621.aspx>`_.
Antoine Pitrou77e904e2013-10-08 23:04:32 +0200940
941 On UNIX, :func:`resource.setrlimit` is used to set
942 :attr:`resource.RLIMIT_CORE`'s soft limit to 0 to prevent coredump file
943 creation.
944
945 On both platforms, the old value is restored by :meth:`__exit__`.
946
947
Cheryl Sabella988fb282018-02-11 08:10:42 -0500948.. class:: SaveSignals()
949
950 Class to save and restore signal handlers registered by the Python signal
951 handler.
952
953
954.. class:: Matcher()
955
956 .. method:: matches(self, d, **kwargs)
957
958 Try to match a single dict with the supplied arguments.
959
960
961 .. method:: match_value(self, k, dv, v)
962
963 Try to match a single stored value (*dv*) with a supplied value (*v*).
964
965
Cheryl Sabella988fb282018-02-11 08:10:42 -0500966.. class:: BasicTestRunner()
967
968 .. method:: run(test)
969
970 Run *test* and return the result.
971
972
Serhiy Storchaka16994912020-04-25 10:06:29 +0300973:mod:`test.support.socket_helper` --- Utilities for socket tests
974================================================================
975
976.. module:: test.support.socket_helper
977 :synopsis: Support for socket tests.
978
979
980The :mod:`test.support.socket_helper` module provides support for socket tests.
981
982.. versionadded:: 3.9
983
984
985.. data:: IPV6_ENABLED
986
987 Set to ``True`` if IPv6 is enabled on this host, ``False`` otherwise.
988
989
990.. function:: find_unused_port(family=socket.AF_INET, socktype=socket.SOCK_STREAM)
991
992 Returns an unused port that should be suitable for binding. This is
993 achieved by creating a temporary socket with the same family and type as
994 the ``sock`` parameter (default is :const:`~socket.AF_INET`,
995 :const:`~socket.SOCK_STREAM`),
996 and binding it to the specified host address (defaults to ``0.0.0.0``)
997 with the port set to 0, eliciting an unused ephemeral port from the OS.
998 The temporary socket is then closed and deleted, and the ephemeral port is
999 returned.
1000
1001 Either this method or :func:`bind_port` should be used for any tests
1002 where a server socket needs to be bound to a particular port for the
1003 duration of the test.
1004 Which one to use depends on whether the calling code is creating a Python
1005 socket, or if an unused port needs to be provided in a constructor
1006 or passed to an external program (i.e. the ``-accept`` argument to
1007 openssl's s_server mode). Always prefer :func:`bind_port` over
1008 :func:`find_unused_port` where possible. Using a hard coded port is
1009 discouraged since it can make multiple instances of the test impossible to
1010 run simultaneously, which is a problem for buildbots.
1011
1012
1013.. function:: bind_port(sock, host=HOST)
1014
1015 Bind the socket to a free port and return the port number. Relies on
1016 ephemeral ports in order to ensure we are using an unbound port. This is
1017 important as many tests may be running simultaneously, especially in a
1018 buildbot environment. This method raises an exception if the
1019 ``sock.family`` is :const:`~socket.AF_INET` and ``sock.type`` is
1020 :const:`~socket.SOCK_STREAM`, and the socket has
1021 :const:`~socket.SO_REUSEADDR` or :const:`~socket.SO_REUSEPORT` set on it.
1022 Tests should never set these socket options for TCP/IP sockets.
1023 The only case for setting these options is testing multicasting via
1024 multiple UDP sockets.
1025
1026 Additionally, if the :const:`~socket.SO_EXCLUSIVEADDRUSE` socket option is
1027 available (i.e. on Windows), it will be set on the socket. This will
1028 prevent anyone else from binding to our host/port for the duration of the
1029 test.
1030
1031
1032.. function:: bind_unix_socket(sock, addr)
1033
1034 Bind a unix socket, raising :exc:`unittest.SkipTest` if
1035 :exc:`PermissionError` is raised.
1036
1037
1038.. decorator:: skip_unless_bind_unix_socket
1039
1040 A decorator for running tests that require a functional ``bind()`` for Unix
1041 sockets.
1042
1043
Serhiy Storchakabfb1cf42020-04-29 10:36:20 +03001044.. function:: transient_internet(resource_name, *, timeout=30.0, errnos=())
1045
1046 A context manager that raises :exc:`~test.support.ResourceDenied` when
1047 various issues with the internet connection manifest themselves as
1048 exceptions.
1049
1050
Cheryl Sabella988fb282018-02-11 08:10:42 -05001051:mod:`test.support.script_helper` --- Utilities for the Python execution tests
1052==============================================================================
1053
1054.. module:: test.support.script_helper
1055 :synopsis: Support for Python's script execution tests.
1056
1057
1058The :mod:`test.support.script_helper` module provides support for Python's
1059script execution tests.
1060
1061.. function:: interpreter_requires_environment()
1062
1063 Return ``True`` if ``sys.executable interpreter`` requires environment
1064 variables in order to be able to run at all.
1065
1066 This is designed to be used with ``@unittest.skipIf()`` to annotate tests
1067 that need to use an ``assert_python*()`` function to launch an isolated
1068 mode (``-I``) or no environment mode (``-E``) sub-interpreter process.
1069
1070 A normal build & test does not run into this situation but it can happen
1071 when trying to run the standard library test suite from an interpreter that
1072 doesn't have an obvious home with Python's current home finding logic.
1073
1074 Setting :envvar:`PYTHONHOME` is one way to get most of the testsuite to run
1075 in that situation. :envvar:`PYTHONPATH` or :envvar:`PYTHONUSERSITE` are
1076 other common environment variables that might impact whether or not the
1077 interpreter can start.
1078
1079
1080.. function:: run_python_until_end(*args, **env_vars)
1081
1082 Set up the environment based on *env_vars* for running the interpreter
Nick Coghland5d9e022018-03-25 23:03:10 +10001083 in a subprocess. The values can include ``__isolated``, ``__cleanenv``,
1084 ``__cwd``, and ``TERM``.
Cheryl Sabella988fb282018-02-11 08:10:42 -05001085
Victor Stinner6cac1132019-12-08 08:38:16 +01001086 .. versionchanged:: 3.9
1087 The function no longer strips whitespaces from *stderr*.
1088
Cheryl Sabella988fb282018-02-11 08:10:42 -05001089
1090.. function:: assert_python_ok(*args, **env_vars)
1091
1092 Assert that running the interpreter with *args* and optional environment
1093 variables *env_vars* succeeds (``rc == 0``) and return a ``(return code,
1094 stdout, stderr)`` tuple.
1095
1096 If the ``__cleanenv`` keyword is set, *env_vars* is used as a fresh
1097 environment.
1098
1099 Python is started in isolated mode (command line option ``-I``),
1100 except if the ``__isolated`` keyword is set to ``False``.
1101
Victor Stinner6cac1132019-12-08 08:38:16 +01001102 .. versionchanged:: 3.9
1103 The function no longer strips whitespaces from *stderr*.
1104
Cheryl Sabella988fb282018-02-11 08:10:42 -05001105
1106.. function:: assert_python_failure(*args, **env_vars)
1107
1108 Assert that running the interpreter with *args* and optional environment
1109 variables *env_vars* fails (``rc != 0``) and return a ``(return code,
1110 stdout, stderr)`` tuple.
1111
1112 See :func:`assert_python_ok` for more options.
1113
Victor Stinner6cac1132019-12-08 08:38:16 +01001114 .. versionchanged:: 3.9
1115 The function no longer strips whitespaces from *stderr*.
1116
Cheryl Sabella988fb282018-02-11 08:10:42 -05001117
1118.. function:: spawn_python(*args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, **kw)
1119
1120 Run a Python subprocess with the given arguments.
1121
1122 *kw* is extra keyword args to pass to :func:`subprocess.Popen`. Returns a
1123 :class:`subprocess.Popen` object.
1124
1125
1126.. function:: kill_python(p)
1127
1128 Run the given :class:`subprocess.Popen` process until completion and return
1129 stdout.
1130
1131
1132.. function:: make_script(script_dir, script_basename, source, omit_suffix=False)
1133
1134 Create script containing *source* in path *script_dir* and *script_basename*.
1135 If *omit_suffix* is ``False``, append ``.py`` to the name. Return the full
1136 script path.
1137
1138
1139.. function:: make_zip_script(zip_dir, zip_basename, script_name, name_in_zip=None)
1140
1141 Create zip file at *zip_dir* and *zip_basename* with extension ``zip`` which
1142 contains the files in *script_name*. *name_in_zip* is the archive name.
1143 Return a tuple containing ``(full path, full path of archive name)``.
1144
1145
1146.. function:: make_pkg(pkg_dir, init_source='')
1147
1148 Create a directory named *pkg_dir* containing an ``__init__`` file with
1149 *init_source* as its contents.
1150
1151
1152.. function:: make_zip_pkg(zip_dir, zip_basename, pkg_name, script_basename, \
1153 source, depth=1, compiled=False)
1154
1155 Create a zip package directory with a path of *zip_dir* and *zip_basename*
1156 containing an empty ``__init__`` file and a file *script_basename*
1157 containing the *source*. If *compiled* is ``True``, both source files will
1158 be compiled and added to the zip package. Return a tuple of the full zip
1159 path and the archive name for the zip file.
Joannah Nanjekye92777d52019-09-12 10:02:59 +01001160
1161
1162:mod:`test.support.bytecode_helper` --- Support tools for testing correct bytecode generation
1163=============================================================================================
1164
1165.. module:: test.support.bytecode_helper
1166 :synopsis: Support tools for testing correct bytecode generation.
1167
1168The :mod:`test.support.bytecode_helper` module provides support for testing
1169and inspecting bytecode generation.
1170
Saiyang Goucf693e52020-10-15 12:06:23 -07001171.. versionadded:: 3.9
1172
Gurupad Hegde6c7bb382019-12-28 17:16:02 -05001173The module defines the following class:
Joannah Nanjekye92777d52019-09-12 10:02:59 +01001174
1175.. class:: BytecodeTestCase(unittest.TestCase)
1176
1177 This class has custom assertion methods for inspecting bytecode.
1178
1179.. method:: BytecodeTestCase.get_disassembly_as_string(co)
1180
1181 Return the disassembly of *co* as string.
1182
1183
1184.. method:: BytecodeTestCase.assertInBytecode(x, opname, argval=_UNSPECIFIED)
1185
1186 Return instr if *opname* is found, otherwise throws :exc:`AssertionError`.
1187
1188
1189.. method:: BytecodeTestCase.assertNotInBytecode(x, opname, argval=_UNSPECIFIED)
1190
1191 Throws :exc:`AssertionError` if *opname* is found.
Hai Shie80697d2020-05-28 06:10:27 +08001192
1193
1194:mod:`test.support.threading_helper` --- Utilities for threading tests
1195======================================================================
1196
1197.. module:: test.support.threading_helper
1198 :synopsis: Support for threading tests.
1199
1200The :mod:`test.support.threading_helper` module provides support for threading tests.
1201
1202.. versionadded:: 3.10
1203
1204
1205.. function:: join_thread(thread, timeout=None)
1206
1207 Join a *thread* within *timeout*. Raise an :exc:`AssertionError` if thread
1208 is still alive after *timeout* seconds.
1209
1210
1211.. decorator:: reap_threads(func)
1212
1213 Decorator to ensure the threads are cleaned up even if the test fails.
1214
1215
1216.. function:: start_threads(threads, unlock=None)
1217
1218 Context manager to start *threads*. It attempts to join the threads upon
1219 exit.
1220
1221
1222.. function:: threading_cleanup(*original_values)
1223
1224 Cleanup up threads not specified in *original_values*. Designed to emit
1225 a warning if a test leaves running threads in the background.
1226
1227
1228.. function:: threading_setup()
1229
1230 Return current thread count and copy of dangling threads.
1231
1232
1233.. function:: wait_threads_exit(timeout=None)
1234
1235 Context manager to wait until all threads created in the ``with`` statement
1236 exit.
1237
1238
1239.. function:: catch_threading_exception()
1240
1241 Context manager catching :class:`threading.Thread` exception using
1242 :func:`threading.excepthook`.
1243
1244 Attributes set when an exception is catched:
1245
1246 * ``exc_type``
1247 * ``exc_value``
1248 * ``exc_traceback``
1249 * ``thread``
1250
1251 See :func:`threading.excepthook` documentation.
1252
1253 These attributes are deleted at the context manager exit.
1254
1255 Usage::
1256
1257 with threading_helper.catch_threading_exception() as cm:
1258 # code spawning a thread which raises an exception
1259 ...
1260
1261 # check the thread exception, use cm attributes:
1262 # exc_type, exc_value, exc_traceback, thread
1263 ...
1264
1265 # exc_type, exc_value, exc_traceback, thread attributes of cm no longer
1266 # exists at this point
1267 # (to avoid reference cycles)
1268
1269 .. versionadded:: 3.8
Hai Shi0d00b2a2020-06-10 20:29:02 +08001270
1271
1272:mod:`test.support.os_helper` --- Utilities for os tests
1273========================================================================
1274
1275.. module:: test.support.os_helper
1276 :synopsis: Support for os tests.
1277
1278The :mod:`test.support.os_helper` module provides support for os tests.
1279
1280.. versionadded:: 3.10
1281
1282
1283.. data:: FS_NONASCII
1284
1285 A non-ASCII character encodable by :func:`os.fsencode`.
1286
1287
1288.. data:: SAVEDCWD
1289
1290 Set to :func:`os.getcwd`.
1291
1292
1293.. data:: TESTFN
1294
1295 Set to a name that is safe to use as the name of a temporary file. Any
1296 temporary file that is created should be closed and unlinked (removed).
1297
1298
1299.. data:: TESTFN_NONASCII
1300
1301 Set to a filename containing the :data:`FS_NONASCII` character.
1302
1303
1304.. data:: TESTFN_UNENCODABLE
1305
1306 Set to a filename (str type) that should not be able to be encoded by file
1307 system encoding in strict mode. It may be ``None`` if it's not possible to
1308 generate such a filename.
1309
1310
1311.. data:: TESTFN_UNDECODABLE
1312
1313 Set to a filename (bytes type) that should not be able to be decoded by
1314 file system encoding in strict mode. It may be ``None`` if it's not
1315 possible to generate such a filename.
1316
1317
1318.. data:: TESTFN_UNICODE
1319
1320 Set to a non-ASCII name for a temporary file.
1321
1322
1323.. class:: EnvironmentVarGuard()
1324
1325 Class used to temporarily set or unset environment variables. Instances can
1326 be used as a context manager and have a complete dictionary interface for
1327 querying/modifying the underlying ``os.environ``. After exit from the
1328 context manager all changes to environment variables done through this
1329 instance will be rolled back.
1330
1331 .. versionchanged:: 3.1
1332 Added dictionary interface.
1333
1334
1335.. 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
1342.. method:: EnvironmentVarGuard.set(envvar, value)
1343
1344 Temporarily set the environment variable ``envvar`` to the value of
1345 ``value``.
1346
1347
1348.. method:: EnvironmentVarGuard.unset(envvar)
1349
1350 Temporarily unset the environment variable ``envvar``.
1351
1352
1353.. function:: can_symlink()
1354
1355 Return ``True`` if the OS supports symbolic links, ``False``
1356 otherwise.
1357
1358
1359.. function:: can_xattr()
1360
1361 Return ``True`` if the OS supports xattr, ``False``
1362 otherwise.
1363
1364
1365.. function:: change_cwd(path, quiet=False)
1366
1367 A context manager that temporarily changes the current working
1368 directory to *path* and yields the directory.
1369
1370 If *quiet* is ``False``, the context manager raises an exception
1371 on error. Otherwise, it issues only a warning and keeps the current
1372 working directory the same.
1373
1374
1375.. function:: create_empty_file(filename)
1376
1377 Create an empty file with *filename*. If it already exists, truncate it.
1378
1379
1380.. function:: fd_count()
1381
1382 Count the number of open file descriptors.
1383
1384
1385.. function:: fs_is_case_insensitive(directory)
1386
1387 Return ``True`` if the file system for *directory* is case-insensitive.
1388
1389
1390.. function:: make_bad_fd()
1391
1392 Create an invalid file descriptor by opening and closing a temporary file,
1393 and returning its descriptor.
1394
1395
1396.. function:: rmdir(filename)
1397
1398 Call :func:`os.rmdir` on *filename*. On Windows platforms, this is
1399 wrapped with a wait loop that checks for the existence of the file.
1400
1401
1402.. function:: rmtree(path)
1403
1404 Call :func:`shutil.rmtree` on *path* or call :func:`os.lstat` and
1405 :func:`os.rmdir` to remove a path and its contents. On Windows platforms,
1406 this is wrapped with a wait loop that checks for the existence of the files.
1407
1408
1409.. decorator:: skip_unless_symlink
1410
1411 A decorator for running tests that require support for symbolic links.
1412
1413
1414.. decorator:: skip_unless_xattr
1415
1416 A decorator for running tests that require support for xattr.
1417
1418
1419.. function:: temp_cwd(name='tempcwd', quiet=False)
1420
1421 A context manager that temporarily creates a new directory and
1422 changes the current working directory (CWD).
1423
1424 The context manager creates a temporary directory in the current
1425 directory with name *name* before temporarily changing the current
1426 working directory. If *name* is ``None``, the temporary directory is
1427 created using :func:`tempfile.mkdtemp`.
1428
1429 If *quiet* is ``False`` and it is not possible to create or change
1430 the CWD, an error is raised. Otherwise, only a warning is raised
1431 and the original CWD is used.
1432
1433
1434.. function:: temp_dir(path=None, quiet=False)
1435
1436 A context manager that creates a temporary directory at *path* and
1437 yields the directory.
1438
1439 If *path* is ``None``, the temporary directory is created using
1440 :func:`tempfile.mkdtemp`. If *quiet* is ``False``, the context manager
1441 raises an exception on error. Otherwise, if *path* is specified and
1442 cannot be created, only a warning is issued.
1443
1444
1445.. function:: temp_umask(umask)
1446
1447 A context manager that temporarily sets the process umask.
1448
1449
1450.. function:: unlink(filename)
1451
1452 Call :func:`os.unlink` on *filename*. On Windows platforms, this is
1453 wrapped with a wait loop that checks for the existence fo the file.
Hai Shi7f888c72020-06-11 07:51:18 +08001454
1455
1456:mod:`test.support.import_helper` --- Utilities for import tests
1457================================================================
1458
1459.. module:: test.support.import_helper
1460 :synopsis: Support for import tests.
1461
1462The :mod:`test.support.import_helper` module provides support for import tests.
1463
1464.. versionadded:: 3.10
1465
1466
1467.. function:: forget(module_name)
1468
1469 Remove the module named *module_name* from ``sys.modules`` and delete any
1470 byte-compiled files of the module.
1471
1472
1473.. function:: import_fresh_module(name, fresh=(), blocked=(), deprecated=False)
1474
1475 This function imports and returns a fresh copy of the named Python module
1476 by removing the named module from ``sys.modules`` before doing the import.
1477 Note that unlike :func:`reload`, the original module is not affected by
1478 this operation.
1479
1480 *fresh* is an iterable of additional module names that are also removed
1481 from the ``sys.modules`` cache before doing the import.
1482
1483 *blocked* is an iterable of module names that are replaced with ``None``
1484 in the module cache during the import to ensure that attempts to import
1485 them raise :exc:`ImportError`.
1486
1487 The named module and any modules named in the *fresh* and *blocked*
1488 parameters are saved before starting the import and then reinserted into
1489 ``sys.modules`` when the fresh import is complete.
1490
1491 Module and package deprecation messages are suppressed during this import
1492 if *deprecated* is ``True``.
1493
1494 This function will raise :exc:`ImportError` if the named module cannot be
1495 imported.
1496
1497 Example use::
1498
1499 # Get copies of the warnings module for testing without affecting the
1500 # version being used by the rest of the test suite. One copy uses the
1501 # C implementation, the other is forced to use the pure Python fallback
1502 # implementation
1503 py_warnings = import_fresh_module('warnings', blocked=['_warnings'])
1504 c_warnings = import_fresh_module('warnings', fresh=['_warnings'])
1505
1506 .. versionadded:: 3.1
1507
1508
1509.. function:: import_module(name, deprecated=False, *, required_on())
1510
1511 This function imports and returns the named module. Unlike a normal
1512 import, this function raises :exc:`unittest.SkipTest` if the module
1513 cannot be imported.
1514
1515 Module and package deprecation messages are suppressed during this import
1516 if *deprecated* is ``True``. If a module is required on a platform but
1517 optional for others, set *required_on* to an iterable of platform prefixes
1518 which will be compared against :data:`sys.platform`.
1519
1520 .. versionadded:: 3.1
1521
1522
1523.. function:: modules_setup()
1524
1525 Return a copy of :data:`sys.modules`.
1526
1527
1528.. function:: modules_cleanup(oldmodules)
1529
1530 Remove modules except for *oldmodules* and ``encodings`` in order to
1531 preserve internal cache.
1532
1533
1534.. function:: unload(name)
1535
1536 Delete *name* from ``sys.modules``.
1537
1538
1539.. function:: make_legacy_pyc(source)
1540
1541 Move a :pep:`3147`/:pep:`488` pyc file to its legacy pyc location and return the file
1542 system path to the legacy pyc file. The *source* value is the file system
1543 path to the source file. It does not need to exist, however the PEP
1544 3147/488 pyc file must exist.
1545
1546
1547.. class:: CleanImport(*module_names)
1548
1549 A context manager to force import to return a new module reference. This
1550 is useful for testing module-level behaviors, such as the emission of a
1551 DeprecationWarning on import. Example usage::
1552
1553 with CleanImport('foo'):
1554 importlib.import_module('foo') # New reference.
1555
1556
1557.. class:: DirsOnSysPath(*paths)
1558
1559 A context manager to temporarily add directories to sys.path.
1560
1561 This makes a copy of :data:`sys.path`, appends any directories given
1562 as positional arguments, then reverts :data:`sys.path` to the copied
1563 settings when the context ends.
1564
1565 Note that *all* :data:`sys.path` modifications in the body of the
1566 context manager, including replacement of the object,
1567 will be reverted at the end of the block.
1568
1569
Hai Shi10e65062020-06-11 23:36:06 +08001570:mod:`test.support.warnings_helper` --- Utilities for warnings tests
1571====================================================================
1572
1573.. module:: test.support.warnings_helper
1574 :synopsis: Support for warnings tests.
1575
1576The :mod:`test.support.warnings_helper` module provides support for warnings tests.
1577
1578.. versionadded:: 3.10
1579
1580
1581.. function:: check_no_resource_warning(testcase)
1582
1583 Context manager to check that no :exc:`ResourceWarning` was raised. You
1584 must remove the object which may emit :exc:`ResourceWarning` before the
1585 end of the context manager.
1586
1587
1588.. function:: check_syntax_warning(testcase, statement, errtext='', *, lineno=1, offset=None)
1589
1590 Test for syntax warning in *statement* by attempting to compile *statement*.
1591 Test also that the :exc:`SyntaxWarning` is emitted only once, and that it
1592 will be converted to a :exc:`SyntaxError` when turned into error.
1593 *testcase* is the :mod:`unittest` instance for the test. *errtext* is the
1594 regular expression which should match the string representation of the
1595 emitted :exc:`SyntaxWarning` and raised :exc:`SyntaxError`. If *lineno*
1596 is not ``None``, compares to the line of the warning and exception.
1597 If *offset* is not ``None``, compares to the offset of the exception.
1598
1599 .. versionadded:: 3.8
1600
1601
Andre Delfinodcc997c2020-12-16 22:37:28 -03001602.. function:: check_warnings(*filters, quiet=True)
Hai Shi10e65062020-06-11 23:36:06 +08001603
1604 A convenience wrapper for :func:`warnings.catch_warnings()` that makes it
1605 easier to test that a warning was correctly raised. It is approximately
1606 equivalent to calling ``warnings.catch_warnings(record=True)`` with
1607 :meth:`warnings.simplefilter` set to ``always`` and with the option to
1608 automatically validate the results that are recorded.
1609
1610 ``check_warnings`` accepts 2-tuples of the form ``("message regexp",
1611 WarningCategory)`` as positional arguments. If one or more *filters* are
1612 provided, or if the optional keyword argument *quiet* is ``False``,
1613 it checks to make sure the warnings are as expected: each specified filter
1614 must match at least one of the warnings raised by the enclosed code or the
1615 test fails, and if any warnings are raised that do not match any of the
1616 specified filters the test fails. To disable the first of these checks,
1617 set *quiet* to ``True``.
1618
1619 If no arguments are specified, it defaults to::
1620
1621 check_warnings(("", Warning), quiet=True)
1622
1623 In this case all warnings are caught and no errors are raised.
1624
1625 On entry to the context manager, a :class:`WarningRecorder` instance is
1626 returned. The underlying warnings list from
1627 :func:`~warnings.catch_warnings` is available via the recorder object's
1628 :attr:`warnings` attribute. As a convenience, the attributes of the object
1629 representing the most recent warning can also be accessed directly through
1630 the recorder object (see example below). If no warning has been raised,
1631 then any of the attributes that would otherwise be expected on an object
1632 representing a warning will return ``None``.
1633
1634 The recorder object also has a :meth:`reset` method, which clears the
1635 warnings list.
1636
1637 The context manager is designed to be used like this::
1638
1639 with check_warnings(("assertion is always true", SyntaxWarning),
1640 ("", UserWarning)):
1641 exec('assert(False, "Hey!")')
1642 warnings.warn(UserWarning("Hide me!"))
1643
1644 In this case if either warning was not raised, or some other warning was
1645 raised, :func:`check_warnings` would raise an error.
1646
1647 When a test needs to look more deeply into the warnings, rather than
1648 just checking whether or not they occurred, code like this can be used::
1649
1650 with check_warnings(quiet=True) as w:
1651 warnings.warn("foo")
1652 assert str(w.args[0]) == "foo"
1653 warnings.warn("bar")
1654 assert str(w.args[0]) == "bar"
1655 assert str(w.warnings[0].args[0]) == "foo"
1656 assert str(w.warnings[1].args[0]) == "bar"
1657 w.reset()
1658 assert len(w.warnings) == 0
1659
1660
1661 Here all warnings will be caught, and the test code tests the captured
1662 warnings directly.
1663
1664 .. versionchanged:: 3.2
1665 New optional arguments *filters* and *quiet*.
1666
1667
1668.. class:: WarningsRecorder()
1669
1670 Class used to record warnings for unit tests. See documentation of
1671 :func:`check_warnings` above for more details.