blob: d1b4f3ba0bbbdae850b0e28361574aa7509a812b [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001:mod:`unittest` --- Unit testing framework
2==========================================
3
4.. module:: unittest
5 :synopsis: Unit testing framework for Python.
Terry Jan Reedyfa089b92016-06-11 15:02:54 -04006
Georg Brandl116aa622007-08-15 14:28:22 +00007.. moduleauthor:: Steve Purcell <stephen_purcell@yahoo.com>
8.. sectionauthor:: Steve Purcell <stephen_purcell@yahoo.com>
9.. sectionauthor:: Fred L. Drake, Jr. <fdrake@acm.org>
10.. sectionauthor:: Raymond Hettinger <python@rcn.com>
11
Terry Jan Reedyfa089b92016-06-11 15:02:54 -040012**Source code:** :source:`Lib/unittest/__init__.py`
13
14--------------
15
Ezio Melotti22170ed2010-11-20 09:57:27 +000016(If you are already familiar with the basic concepts of testing, you might want
17to skip to :ref:`the list of assert methods <assert-methods>`.)
Georg Brandl116aa622007-08-15 14:28:22 +000018
Antoine Pitrou2c5e9502013-01-20 01:29:39 +010019The :mod:`unittest` unit testing framework was originally inspired by JUnit
20and has a similar flavor as major unit testing frameworks in other
21languages. It supports test automation, sharing of setup and shutdown code
22for tests, aggregation of tests into collections, and independence of the
23tests from the reporting framework.
Georg Brandl116aa622007-08-15 14:28:22 +000024
Antoine Pitrou2c5e9502013-01-20 01:29:39 +010025To achieve this, :mod:`unittest` supports some important concepts in an
26object-oriented way:
Georg Brandl116aa622007-08-15 14:28:22 +000027
28test fixture
29 A :dfn:`test fixture` represents the preparation needed to perform one or more
Julien Palard513e9b442019-02-19 15:46:07 +010030 tests, and any associated cleanup actions. This may involve, for example,
Georg Brandl116aa622007-08-15 14:28:22 +000031 creating temporary or proxy databases, directories, or starting a server
32 process.
33
34test case
Antoine Pitrou2c5e9502013-01-20 01:29:39 +010035 A :dfn:`test case` is the individual unit of testing. It checks for a specific
Georg Brandl116aa622007-08-15 14:28:22 +000036 response to a particular set of inputs. :mod:`unittest` provides a base class,
37 :class:`TestCase`, which may be used to create new test cases.
38
39test suite
40 A :dfn:`test suite` is a collection of test cases, test suites, or both. It is
41 used to aggregate tests that should be executed together.
42
43test runner
44 A :dfn:`test runner` is a component which orchestrates the execution of tests
45 and provides the outcome to the user. The runner may use a graphical interface,
46 a textual interface, or return a special value to indicate the results of
47 executing the tests.
48
Georg Brandl116aa622007-08-15 14:28:22 +000049
50.. seealso::
51
52 Module :mod:`doctest`
53 Another test-support module with a very different flavor.
54
R David Murraya1005ed2015-07-04 15:44:14 -040055 `Simple Smalltalk Testing: With Patterns <https://web.archive.org/web/20150315073817/http://www.xprogramming.com/testfram.htm>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000056 Kent Beck's original paper on testing frameworks using the pattern shared
57 by :mod:`unittest`.
Georg Brandl116aa622007-08-15 14:28:22 +000058
Andreas Pelmed855f2f2018-10-27 20:09:12 +020059 `Nose <https://nose.readthedocs.io/>`_ and `pytest <https://docs.pytest.org/>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000060 Third-party unittest frameworks with a lighter-weight syntax for writing
61 tests. For example, ``assert func(10) == 42``.
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000062
Georg Brandle73778c2014-10-29 08:36:35 +010063 `The Python Testing Tools Taxonomy <https://wiki.python.org/moin/PythonTestingToolsTaxonomy>`_
Benjamin Petersonb48af542010-04-11 20:43:16 +000064 An extensive list of Python testing tools including functional testing
65 frameworks and mock object libraries.
Benjamin Petersond2397752009-06-27 23:45:02 +000066
Benjamin Petersonb48af542010-04-11 20:43:16 +000067 `Testing in Python Mailing List <http://lists.idyll.org/listinfo/testing-in-python>`_
68 A special-interest-group for discussion of testing, and testing tools,
69 in Python.
Benjamin Petersond2397752009-06-27 23:45:02 +000070
Michael Foord90efac72011-01-03 15:39:49 +000071 The script :file:`Tools/unittestgui/unittestgui.py` in the Python source distribution is
72 a GUI tool for test discovery and execution. This is intended largely for ease of use
Senthil Kumaran847c33c2012-10-27 11:04:55 -070073 for those new to unit testing. For production environments it is
74 recommended that tests be driven by a continuous integration system such as
Serhiy Storchaka6dff0202016-05-07 10:49:07 +030075 `Buildbot <https://buildbot.net/>`_, `Jenkins <https://jenkins.io/>`_
Senthil Kumaran847c33c2012-10-27 11:04:55 -070076 or `Hudson <http://hudson-ci.org/>`_.
Michael Foord90efac72011-01-03 15:39:49 +000077
78
Georg Brandl116aa622007-08-15 14:28:22 +000079.. _unittest-minimal-example:
80
81Basic example
82-------------
83
84The :mod:`unittest` module provides a rich set of tools for constructing and
85running tests. This section demonstrates that a small subset of the tools
86suffice to meet the needs of most users.
87
Ezio Melotti2e3998f2015-03-24 12:42:41 +020088Here is a short script to test three string methods::
Georg Brandl116aa622007-08-15 14:28:22 +000089
Ezio Melotti2e3998f2015-03-24 12:42:41 +020090 import unittest
Georg Brandl116aa622007-08-15 14:28:22 +000091
Ezio Melotti2e3998f2015-03-24 12:42:41 +020092 class TestStringMethods(unittest.TestCase):
Georg Brandl116aa622007-08-15 14:28:22 +000093
Serhiy Storchakadba90392016-05-10 12:01:23 +030094 def test_upper(self):
95 self.assertEqual('foo'.upper(), 'FOO')
Georg Brandl116aa622007-08-15 14:28:22 +000096
Serhiy Storchakadba90392016-05-10 12:01:23 +030097 def test_isupper(self):
98 self.assertTrue('FOO'.isupper())
99 self.assertFalse('Foo'.isupper())
Georg Brandl116aa622007-08-15 14:28:22 +0000100
Serhiy Storchakadba90392016-05-10 12:01:23 +0300101 def test_split(self):
102 s = 'hello world'
103 self.assertEqual(s.split(), ['hello', 'world'])
104 # check that s.split fails when the separator is not a string
105 with self.assertRaises(TypeError):
106 s.split(2)
Benjamin Peterson847a4112010-03-14 15:04:17 +0000107
Ezio Melotti2e3998f2015-03-24 12:42:41 +0200108 if __name__ == '__main__':
109 unittest.main()
Georg Brandl116aa622007-08-15 14:28:22 +0000110
Georg Brandl116aa622007-08-15 14:28:22 +0000111
Benjamin Peterson52baa292009-03-24 00:56:30 +0000112A testcase is created by subclassing :class:`unittest.TestCase`. The three
Georg Brandl116aa622007-08-15 14:28:22 +0000113individual tests are defined with methods whose names start with the letters
114``test``. This naming convention informs the test runner about which methods
115represent tests.
116
Benjamin Peterson52baa292009-03-24 00:56:30 +0000117The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
Ezio Melotti2e3998f2015-03-24 12:42:41 +0200118expected result; :meth:`~TestCase.assertTrue` or :meth:`~TestCase.assertFalse`
119to verify a condition; or :meth:`~TestCase.assertRaises` to verify that a
120specific exception gets raised. These methods are used instead of the
121:keyword:`assert` statement so the test runner can accumulate all test results
122and produce a report.
Georg Brandl116aa622007-08-15 14:28:22 +0000123
Ezio Melotti2e3998f2015-03-24 12:42:41 +0200124The :meth:`~TestCase.setUp` and :meth:`~TestCase.tearDown` methods allow you
125to define instructions that will be executed before and after each test method.
Benjamin Peterson8a6ddb92016-01-07 22:01:26 -0800126They are covered in more detail in the section :ref:`organizing-tests`.
Georg Brandl116aa622007-08-15 14:28:22 +0000127
128The final block shows a simple way to run the tests. :func:`unittest.main`
Éric Araujo713d3032010-11-18 16:38:46 +0000129provides a command-line interface to the test script. When run from the command
Georg Brandl116aa622007-08-15 14:28:22 +0000130line, the above script produces an output that looks like this::
131
132 ...
133 ----------------------------------------------------------------------
134 Ran 3 tests in 0.000s
135
136 OK
137
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100138Passing the ``-v`` option to your test script will instruct :func:`unittest.main`
139to enable a higher level of verbosity, and produce the following output::
Georg Brandl116aa622007-08-15 14:28:22 +0000140
Ezio Melotti2e3998f2015-03-24 12:42:41 +0200141 test_isupper (__main__.TestStringMethods) ... ok
142 test_split (__main__.TestStringMethods) ... ok
143 test_upper (__main__.TestStringMethods) ... ok
Georg Brandl116aa622007-08-15 14:28:22 +0000144
145 ----------------------------------------------------------------------
Ezio Melotti2e3998f2015-03-24 12:42:41 +0200146 Ran 3 tests in 0.001s
Georg Brandl116aa622007-08-15 14:28:22 +0000147
148 OK
149
150The above examples show the most commonly used :mod:`unittest` features which
151are sufficient to meet many everyday testing needs. The remainder of the
152documentation explores the full feature set from first principles.
153
Benjamin Petersonb48af542010-04-11 20:43:16 +0000154
155.. _unittest-command-line-interface:
156
Éric Araujo76338ec2010-11-26 23:46:18 +0000157Command-Line Interface
Benjamin Petersonb48af542010-04-11 20:43:16 +0000158----------------------
159
160The unittest module can be used from the command line to run tests from
161modules, classes or even individual test methods::
162
163 python -m unittest test_module1 test_module2
164 python -m unittest test_module.TestClass
165 python -m unittest test_module.TestClass.test_method
166
167You can pass in a list with any combination of module names, and fully
168qualified class or method names.
169
Michael Foord37d120a2010-12-04 01:11:21 +0000170Test modules can be specified by file path as well::
171
172 python -m unittest tests/test_something.py
173
174This allows you to use the shell filename completion to specify the test module.
175The file specified must still be importable as a module. The path is converted
176to a module name by removing the '.py' and converting path separators into '.'.
177If you want to execute a test file that isn't importable as a module you should
178execute the file directly instead.
179
Benjamin Petersonb48af542010-04-11 20:43:16 +0000180You can run tests with more detail (higher verbosity) by passing in the -v flag::
181
182 python -m unittest -v test_module
183
Michael Foord086f3082010-11-21 21:28:01 +0000184When executed without arguments :ref:`unittest-test-discovery` is started::
185
186 python -m unittest
187
Éric Araujo713d3032010-11-18 16:38:46 +0000188For a list of all the command-line options::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000189
190 python -m unittest -h
191
Georg Brandl67b21b72010-08-17 15:07:14 +0000192.. versionchanged:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000193 In earlier versions it was only possible to run individual test methods and
194 not modules or classes.
195
196
Éric Araujo76338ec2010-11-26 23:46:18 +0000197Command-line options
198~~~~~~~~~~~~~~~~~~~~
Benjamin Petersonb48af542010-04-11 20:43:16 +0000199
Éric Araujod3309df2010-11-21 03:09:17 +0000200:program:`unittest` supports these command-line options:
Benjamin Petersonb48af542010-04-11 20:43:16 +0000201
Éric Araujo713d3032010-11-18 16:38:46 +0000202.. program:: unittest
Benjamin Petersonb48af542010-04-11 20:43:16 +0000203
Éric Araujo713d3032010-11-18 16:38:46 +0000204.. cmdoption:: -b, --buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +0000205
Éric Araujo713d3032010-11-18 16:38:46 +0000206 The standard output and standard error streams are buffered during the test
207 run. Output during a passing test is discarded. Output is echoed normally
208 on test fail or error and is added to the failure messages.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000209
Éric Araujo713d3032010-11-18 16:38:46 +0000210.. cmdoption:: -c, --catch
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000211
Serhiy Storchaka0424eaf2015-09-12 17:45:25 +0300212 :kbd:`Control-C` during the test run waits for the current test to end and then
213 reports all the results so far. A second :kbd:`Control-C` raises the normal
Éric Araujo713d3032010-11-18 16:38:46 +0000214 :exc:`KeyboardInterrupt` exception.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000215
Éric Araujo713d3032010-11-18 16:38:46 +0000216 See `Signal Handling`_ for the functions that provide this functionality.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000217
Éric Araujo713d3032010-11-18 16:38:46 +0000218.. cmdoption:: -f, --failfast
219
220 Stop the test run on the first error or failure.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000221
Jonas Haag5b48dc62017-11-25 16:23:52 +0100222.. cmdoption:: -k
223
224 Only run test methods and classes that match the pattern or substring.
225 This option may be used multiple times, in which case all test cases that
226 match of the given patterns are included.
227
228 Patterns that contain a wildcard character (``*``) are matched against the
229 test name using :meth:`fnmatch.fnmatchcase`; otherwise simple case-sensitive
230 substring matching is used.
231
232 Patterns are matched against the fully qualified test method name as
233 imported by the test loader.
234
235 For example, ``-k foo`` matches ``foo_tests.SomeTest.test_something``,
236 ``bar_tests.SomeTest.test_foo``, but not ``bar_tests.FooTest.test_something``.
237
Robert Collinsf0c819a2015-03-06 13:46:35 +1300238.. cmdoption:: --locals
239
240 Show local variables in tracebacks.
241
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000242.. versionadded:: 3.2
Éric Araujod6c5f742010-12-16 00:07:01 +0000243 The command-line options ``-b``, ``-c`` and ``-f`` were added.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000244
Robert Collinsf0c819a2015-03-06 13:46:35 +1300245.. versionadded:: 3.5
246 The command-line option ``--locals``.
247
Jonas Haag5b48dc62017-11-25 16:23:52 +0100248.. versionadded:: 3.7
249 The command-line option ``-k``.
250
Benjamin Petersonb48af542010-04-11 20:43:16 +0000251The command line can also be used for test discovery, for running all of the
252tests in a project or just a subset.
253
254
255.. _unittest-test-discovery:
256
257Test Discovery
258--------------
259
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000260.. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000261
Ezio Melotti3d995842011-03-08 16:17:35 +0200262Unittest supports simple test discovery. In order to be compatible with test
263discovery, all of the test files must be :ref:`modules <tut-modules>` or
Larry Hastings3732ed22014-03-15 21:13:56 -0700264:ref:`packages <tut-packages>` (including :term:`namespace packages
265<namespace package>`) importable from the top-level directory of
266the project (this means that their filenames must be valid :ref:`identifiers
267<identifiers>`).
Benjamin Petersonb48af542010-04-11 20:43:16 +0000268
269Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
Éric Araujo713d3032010-11-18 16:38:46 +0000270used from the command line. The basic command-line usage is::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000271
272 cd project_directory
273 python -m unittest discover
274
Michael Foord086f3082010-11-21 21:28:01 +0000275.. note::
276
277 As a shortcut, ``python -m unittest`` is the equivalent of
278 ``python -m unittest discover``. If you want to pass arguments to test
Éric Araujo941afed2011-09-01 02:47:34 +0200279 discovery the ``discover`` sub-command must be used explicitly.
Michael Foord086f3082010-11-21 21:28:01 +0000280
Benjamin Petersonb48af542010-04-11 20:43:16 +0000281The ``discover`` sub-command has the following options:
282
Éric Araujo713d3032010-11-18 16:38:46 +0000283.. program:: unittest discover
284
285.. cmdoption:: -v, --verbose
286
287 Verbose output
288
Chris Jerdonekd69ad552013-02-21 18:54:43 -0800289.. cmdoption:: -s, --start-directory directory
Éric Araujo713d3032010-11-18 16:38:46 +0000290
Éric Araujo941afed2011-09-01 02:47:34 +0200291 Directory to start discovery (``.`` default)
Éric Araujo713d3032010-11-18 16:38:46 +0000292
Chris Jerdonekd69ad552013-02-21 18:54:43 -0800293.. cmdoption:: -p, --pattern pattern
Éric Araujo713d3032010-11-18 16:38:46 +0000294
Éric Araujo941afed2011-09-01 02:47:34 +0200295 Pattern to match test files (``test*.py`` default)
Éric Araujo713d3032010-11-18 16:38:46 +0000296
Chris Jerdonekd69ad552013-02-21 18:54:43 -0800297.. cmdoption:: -t, --top-level-directory directory
Éric Araujo713d3032010-11-18 16:38:46 +0000298
299 Top level directory of project (defaults to start directory)
Benjamin Petersonb48af542010-04-11 20:43:16 +0000300
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000301The :option:`-s`, :option:`-p`, and :option:`-t` options can be passed in
302as positional arguments in that order. The following two command lines
303are equivalent::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000304
Robert Collinsa2b00552015-08-24 12:14:28 +1200305 python -m unittest discover -s project_directory -p "*_test.py"
306 python -m unittest discover project_directory "*_test.py"
Benjamin Petersonb48af542010-04-11 20:43:16 +0000307
Michael Foord16f3e902010-05-08 15:13:42 +0000308As well as being a path it is possible to pass a package name, for example
309``myproject.subpackage.test``, as the start directory. The package name you
310supply will then be imported and its location on the filesystem will be used
311as the start directory.
312
313.. caution::
314
Senthil Kumaran916bd382010-10-15 12:55:19 +0000315 Test discovery loads tests by importing them. Once test discovery has found
316 all the test files from the start directory you specify it turns the paths
317 into package names to import. For example :file:`foo/bar/baz.py` will be
Michael Foord16f3e902010-05-08 15:13:42 +0000318 imported as ``foo.bar.baz``.
319
320 If you have a package installed globally and attempt test discovery on
321 a different copy of the package then the import *could* happen from the
322 wrong place. If this happens test discovery will warn you and exit.
323
324 If you supply the start directory as a package name rather than a
325 path to a directory then discover assumes that whichever location it
326 imports from is the location you intended, so you will not get the
327 warning.
328
Benjamin Petersonb48af542010-04-11 20:43:16 +0000329Test modules and packages can customize test loading and discovery by through
330the `load_tests protocol`_.
331
Larry Hastings3732ed22014-03-15 21:13:56 -0700332.. versionchanged:: 3.4
333 Test discovery supports :term:`namespace packages <namespace package>`.
334
Benjamin Petersonb48af542010-04-11 20:43:16 +0000335
Georg Brandl116aa622007-08-15 14:28:22 +0000336.. _organizing-tests:
337
338Organizing test code
339--------------------
340
341The basic building blocks of unit testing are :dfn:`test cases` --- single
342scenarios that must be set up and checked for correctness. In :mod:`unittest`,
Raymond Hettinger833ad0e2011-02-06 21:00:38 +0000343test cases are represented by :class:`unittest.TestCase` instances.
344To make your own test cases you must write subclasses of
345:class:`TestCase` or use :class:`FunctionTestCase`.
Georg Brandl116aa622007-08-15 14:28:22 +0000346
Georg Brandl116aa622007-08-15 14:28:22 +0000347The testing code of a :class:`TestCase` instance should be entirely self
348contained, such that it can be run either in isolation or in arbitrary
349combination with any number of other test cases.
350
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100351The simplest :class:`TestCase` subclass will simply implement a test method
352(i.e. a method whose name starts with ``test``) in order to perform specific
353testing code::
Georg Brandl116aa622007-08-15 14:28:22 +0000354
355 import unittest
356
357 class DefaultWidgetSizeTestCase(unittest.TestCase):
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100358 def test_default_widget_size(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000359 widget = Widget('The widget')
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100360 self.assertEqual(widget.size(), (50, 50))
Georg Brandl116aa622007-08-15 14:28:22 +0000361
Sandro Tosi41b24042012-01-21 10:59:37 +0100362Note that in order to test something, we use one of the :meth:`assert\*`
Benjamin Petersond2397752009-06-27 23:45:02 +0000363methods provided by the :class:`TestCase` base class. If the test fails, an
Gregory P. Smithdff46752018-05-17 10:08:45 -0500364exception will be raised with an explanatory message, and :mod:`unittest`
365will identify the test case as a :dfn:`failure`. Any other exceptions will be
366treated as :dfn:`errors`.
Georg Brandl116aa622007-08-15 14:28:22 +0000367
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100368Tests can be numerous, and their set-up can be repetitive. Luckily, we
369can factor out set-up code by implementing a method called
370:meth:`~TestCase.setUp`, which the testing framework will automatically
371call for every single test we run::
Georg Brandl116aa622007-08-15 14:28:22 +0000372
373 import unittest
374
Berker Peksagab75e022016-08-06 03:00:03 +0300375 class WidgetTestCase(unittest.TestCase):
Georg Brandl116aa622007-08-15 14:28:22 +0000376 def setUp(self):
377 self.widget = Widget('The widget')
378
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100379 def test_default_widget_size(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000380 self.assertEqual(self.widget.size(), (50,50),
381 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000382
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100383 def test_widget_resize(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000384 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000385 self.assertEqual(self.widget.size(), (100,150),
386 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000387
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100388.. note::
389 The order in which the various tests will be run is determined
390 by sorting the test method names with respect to the built-in
391 ordering for strings.
392
Benjamin Peterson52baa292009-03-24 00:56:30 +0000393If the :meth:`~TestCase.setUp` method raises an exception while the test is
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100394running, the framework will consider the test to have suffered an error, and
395the test method will not be executed.
Georg Brandl116aa622007-08-15 14:28:22 +0000396
Benjamin Peterson52baa292009-03-24 00:56:30 +0000397Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100398after the test method has been run::
Georg Brandl116aa622007-08-15 14:28:22 +0000399
400 import unittest
401
Berker Peksagab75e022016-08-06 03:00:03 +0300402 class WidgetTestCase(unittest.TestCase):
Georg Brandl116aa622007-08-15 14:28:22 +0000403 def setUp(self):
404 self.widget = Widget('The widget')
405
406 def tearDown(self):
407 self.widget.dispose()
Georg Brandl116aa622007-08-15 14:28:22 +0000408
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100409If :meth:`~TestCase.setUp` succeeded, :meth:`~TestCase.tearDown` will be
410run whether the test method succeeded or not.
Georg Brandl116aa622007-08-15 14:28:22 +0000411
Gregory P. Smithdff46752018-05-17 10:08:45 -0500412Such a working environment for the testing code is called a
413:dfn:`test fixture`. A new TestCase instance is created as a unique
414test fixture used to execute each individual test method. Thus
Berker Peksag2e5566d2018-08-04 00:56:55 +0300415:meth:`~TestCase.setUp`, :meth:`~TestCase.tearDown`, and :meth:`~TestCase.__init__`
Gregory P. Smithdff46752018-05-17 10:08:45 -0500416will be called once per test.
Georg Brandl116aa622007-08-15 14:28:22 +0000417
Gregory P. Smithdff46752018-05-17 10:08:45 -0500418It is recommended that you use TestCase implementations to group tests together
419according to the features they test. :mod:`unittest` provides a mechanism for
420this: the :dfn:`test suite`, represented by :mod:`unittest`'s
421:class:`TestSuite` class. In most cases, calling :func:`unittest.main` will do
422the right thing and collect all the module's test cases for you and execute
423them.
Georg Brandl116aa622007-08-15 14:28:22 +0000424
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100425However, should you want to customize the building of your test suite,
426you can do it yourself::
Georg Brandl116aa622007-08-15 14:28:22 +0000427
428 def suite():
429 suite = unittest.TestSuite()
Berker Peksag92551042017-10-13 06:41:57 +0300430 suite.addTest(WidgetTestCase('test_default_widget_size'))
431 suite.addTest(WidgetTestCase('test_widget_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000432 return suite
433
Berker Peksag92551042017-10-13 06:41:57 +0300434 if __name__ == '__main__':
435 runner = unittest.TextTestRunner()
436 runner.run(suite())
437
Georg Brandl116aa622007-08-15 14:28:22 +0000438You can place the definitions of test cases and test suites in the same modules
439as the code they are to test (such as :file:`widget.py`), but there are several
440advantages to placing the test code in a separate module, such as
441:file:`test_widget.py`:
442
443* The test module can be run standalone from the command line.
444
445* The test code can more easily be separated from shipped code.
446
447* There is less temptation to change test code to fit the code it tests without
448 a good reason.
449
450* Test code should be modified much less frequently than the code it tests.
451
452* Tested code can be refactored more easily.
453
454* Tests for modules written in C must be in separate modules anyway, so why not
455 be consistent?
456
457* If the testing strategy changes, there is no need to change the source code.
458
459
460.. _legacy-unit-tests:
461
462Re-using old test code
463----------------------
464
465Some users will find that they have existing test code that they would like to
466run from :mod:`unittest`, without converting every old test function to a
467:class:`TestCase` subclass.
468
469For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
470This subclass of :class:`TestCase` can be used to wrap an existing test
471function. Set-up and tear-down functions can also be provided.
472
473Given the following test function::
474
475 def testSomething():
476 something = makeSomething()
477 assert something.name is not None
478 # ...
479
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100480one can create an equivalent test case instance as follows, with optional
481set-up and tear-down methods::
Georg Brandl116aa622007-08-15 14:28:22 +0000482
483 testcase = unittest.FunctionTestCase(testSomething,
484 setUp=makeSomethingDB,
485 tearDown=deleteSomethingDB)
486
Georg Brandl116aa622007-08-15 14:28:22 +0000487.. note::
488
Benjamin Petersond2397752009-06-27 23:45:02 +0000489 Even though :class:`FunctionTestCase` can be used to quickly convert an
490 existing test base over to a :mod:`unittest`\ -based system, this approach is
491 not recommended. Taking the time to set up proper :class:`TestCase`
492 subclasses will make future test refactorings infinitely easier.
Georg Brandl116aa622007-08-15 14:28:22 +0000493
Benjamin Peterson52baa292009-03-24 00:56:30 +0000494In some cases, the existing tests may have been written using the :mod:`doctest`
495module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
496automatically build :class:`unittest.TestSuite` instances from the existing
497:mod:`doctest`\ -based tests.
498
Georg Brandl116aa622007-08-15 14:28:22 +0000499
Benjamin Peterson5254c042009-03-23 22:25:03 +0000500.. _unittest-skipping:
501
502Skipping tests and expected failures
503------------------------------------
504
Michael Foordf5c851a2010-02-05 21:48:03 +0000505.. versionadded:: 3.1
506
Benjamin Peterson5254c042009-03-23 22:25:03 +0000507Unittest supports skipping individual test methods and even whole classes of
Serhiy Storchakad65c9492015-11-02 14:10:23 +0200508tests. In addition, it supports marking a test as an "expected failure," a test
Benjamin Peterson5254c042009-03-23 22:25:03 +0000509that is broken and will fail, but shouldn't be counted as a failure on a
510:class:`TestResult`.
511
512Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
513or one of its conditional variants.
514
Ezio Melottifed69ba2013-03-01 21:26:04 +0200515Basic skipping looks like this::
Benjamin Peterson5254c042009-03-23 22:25:03 +0000516
517 class MyTestCase(unittest.TestCase):
518
519 @unittest.skip("demonstrating skipping")
520 def test_nothing(self):
521 self.fail("shouldn't happen")
522
Benjamin Petersond2397752009-06-27 23:45:02 +0000523 @unittest.skipIf(mylib.__version__ < (1, 3),
524 "not supported in this library version")
Benjamin Petersonded31c42009-03-30 15:04:16 +0000525 def test_format(self):
526 # Tests that work for only a certain version of the library.
527 pass
528
529 @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
530 def test_windows_support(self):
531 # windows specific testing code
532 pass
533
Ezio Melottifed69ba2013-03-01 21:26:04 +0200534This is the output of running the example above in verbose mode::
Benjamin Peterson5254c042009-03-23 22:25:03 +0000535
Benjamin Petersonded31c42009-03-30 15:04:16 +0000536 test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000537 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
Benjamin Petersonded31c42009-03-30 15:04:16 +0000538 test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000539
540 ----------------------------------------------------------------------
Benjamin Petersonded31c42009-03-30 15:04:16 +0000541 Ran 3 tests in 0.005s
542
543 OK (skipped=3)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000544
Ezio Melottifed69ba2013-03-01 21:26:04 +0200545Classes can be skipped just like methods::
Benjamin Peterson5254c042009-03-23 22:25:03 +0000546
Sandro Tosi317075d2012-03-31 18:34:59 +0200547 @unittest.skip("showing class skipping")
Benjamin Peterson5254c042009-03-23 22:25:03 +0000548 class MySkippedTestCase(unittest.TestCase):
549 def test_not_run(self):
550 pass
551
Benjamin Peterson52baa292009-03-24 00:56:30 +0000552:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
553that needs to be set up is not available.
554
Benjamin Peterson5254c042009-03-23 22:25:03 +0000555Expected failures use the :func:`expectedFailure` decorator. ::
556
557 class ExpectedFailureTestCase(unittest.TestCase):
558 @unittest.expectedFailure
559 def test_fail(self):
560 self.assertEqual(1, 0, "broken")
561
562It's easy to roll your own skipping decorators by making a decorator that calls
563:func:`skip` on the test when it wants it to be skipped. This decorator skips
Ezio Melottifed69ba2013-03-01 21:26:04 +0200564the test unless the passed object has a certain attribute::
Benjamin Peterson5254c042009-03-23 22:25:03 +0000565
566 def skipUnlessHasattr(obj, attr):
567 if hasattr(obj, attr):
568 return lambda func: func
Ezio Melotti265281a2013-03-27 20:11:55 +0200569 return unittest.skip("{!r} doesn't have {!r}".format(obj, attr))
Benjamin Peterson5254c042009-03-23 22:25:03 +0000570
571The following decorators implement test skipping and expected failures:
572
Georg Brandl8a1caa22010-07-29 16:01:11 +0000573.. decorator:: skip(reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000574
575 Unconditionally skip the decorated test. *reason* should describe why the
576 test is being skipped.
577
Georg Brandl8a1caa22010-07-29 16:01:11 +0000578.. decorator:: skipIf(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000579
580 Skip the decorated test if *condition* is true.
581
Georg Brandl8a1caa22010-07-29 16:01:11 +0000582.. decorator:: skipUnless(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000583
Georg Brandl6faee4e2010-09-21 14:48:28 +0000584 Skip the decorated test unless *condition* is true.
Benjamin Peterson5254c042009-03-23 22:25:03 +0000585
Georg Brandl8a1caa22010-07-29 16:01:11 +0000586.. decorator:: expectedFailure
Benjamin Peterson5254c042009-03-23 22:25:03 +0000587
Gregory P. Smith91f259b2018-12-06 12:56:24 -0800588 Mark the test as an expected failure. If the test fails it will be
589 considered a success. If the test passes, it will be considered a failure.
Benjamin Peterson5254c042009-03-23 22:25:03 +0000590
Ezio Melotti265281a2013-03-27 20:11:55 +0200591.. exception:: SkipTest(reason)
592
593 This exception is raised to skip a test.
594
595 Usually you can use :meth:`TestCase.skipTest` or one of the skipping
596 decorators instead of raising this directly.
597
R David Murray42fa1102014-01-03 13:03:36 -0500598Skipped tests will not have :meth:`~TestCase.setUp` or :meth:`~TestCase.tearDown` run around them.
599Skipped classes will not have :meth:`~TestCase.setUpClass` or :meth:`~TestCase.tearDownClass` run.
600Skipped modules will not have :func:`setUpModule` or :func:`tearDownModule` run.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000601
Benjamin Peterson5254c042009-03-23 22:25:03 +0000602
Antoine Pitrouc9b3ef22013-03-20 20:16:47 +0100603.. _subtests:
604
605Distinguishing test iterations using subtests
606---------------------------------------------
607
608.. versionadded:: 3.4
609
Géry Ogam009b2f02018-11-09 20:34:54 +0100610When there are very small differences among your tests, for
Antoine Pitrouc9b3ef22013-03-20 20:16:47 +0100611instance some parameters, unittest allows you to distinguish them inside
612the body of a test method using the :meth:`~TestCase.subTest` context manager.
613
614For example, the following test::
615
616 class NumbersTest(unittest.TestCase):
617
618 def test_even(self):
619 """
620 Test that numbers between 0 and 5 are all even.
621 """
622 for i in range(0, 6):
623 with self.subTest(i=i):
624 self.assertEqual(i % 2, 0)
625
626will produce the following output::
627
628 ======================================================================
629 FAIL: test_even (__main__.NumbersTest) (i=1)
630 ----------------------------------------------------------------------
631 Traceback (most recent call last):
632 File "subtests.py", line 32, in test_even
633 self.assertEqual(i % 2, 0)
634 AssertionError: 1 != 0
635
636 ======================================================================
637 FAIL: test_even (__main__.NumbersTest) (i=3)
638 ----------------------------------------------------------------------
639 Traceback (most recent call last):
640 File "subtests.py", line 32, in test_even
641 self.assertEqual(i % 2, 0)
642 AssertionError: 1 != 0
643
644 ======================================================================
645 FAIL: test_even (__main__.NumbersTest) (i=5)
646 ----------------------------------------------------------------------
647 Traceback (most recent call last):
648 File "subtests.py", line 32, in test_even
649 self.assertEqual(i % 2, 0)
650 AssertionError: 1 != 0
651
652Without using a subtest, execution would stop after the first failure,
653and the error would be less easy to diagnose because the value of ``i``
654wouldn't be displayed::
655
656 ======================================================================
657 FAIL: test_even (__main__.NumbersTest)
658 ----------------------------------------------------------------------
659 Traceback (most recent call last):
660 File "subtests.py", line 32, in test_even
661 self.assertEqual(i % 2, 0)
662 AssertionError: 1 != 0
663
664
Georg Brandl116aa622007-08-15 14:28:22 +0000665.. _unittest-contents:
666
667Classes and functions
668---------------------
669
Benjamin Peterson52baa292009-03-24 00:56:30 +0000670This section describes in depth the API of :mod:`unittest`.
671
672
673.. _testcase-objects:
674
675Test cases
676~~~~~~~~~~
Georg Brandl116aa622007-08-15 14:28:22 +0000677
Georg Brandl7f01a132009-09-16 15:58:14 +0000678.. class:: TestCase(methodName='runTest')
Georg Brandl116aa622007-08-15 14:28:22 +0000679
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100680 Instances of the :class:`TestCase` class represent the logical test units
Georg Brandl116aa622007-08-15 14:28:22 +0000681 in the :mod:`unittest` universe. This class is intended to be used as a base
682 class, with specific tests being implemented by concrete subclasses. This class
683 implements the interface needed by the test runner to allow it to drive the
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100684 tests, and methods that the test code can use to check for and report various
Georg Brandl116aa622007-08-15 14:28:22 +0000685 kinds of failure.
686
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100687 Each instance of :class:`TestCase` will run a single base method: the method
Robert Collinse02f6c22015-07-23 06:37:26 +1200688 named *methodName*.
689 In most uses of :class:`TestCase`, you will neither change
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100690 the *methodName* nor reimplement the default ``runTest()`` method.
Georg Brandl116aa622007-08-15 14:28:22 +0000691
Michael Foord1341bb02011-03-14 19:01:46 -0400692 .. versionchanged:: 3.2
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100693 :class:`TestCase` can be instantiated successfully without providing a
694 *methodName*. This makes it easier to experiment with :class:`TestCase`
695 from the interactive interpreter.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000696
697 :class:`TestCase` instances provide three groups of methods: one group used
698 to run the test, another used by the test implementation to check conditions
699 and report failures, and some inquiry methods allowing information about the
700 test itself to be gathered.
701
702 Methods in the first group (running the test) are:
703
Benjamin Peterson52baa292009-03-24 00:56:30 +0000704 .. method:: setUp()
705
706 Method called to prepare the test fixture. This is called immediately
Terry Jan Reedy7f84d1e2014-04-15 23:44:14 -0400707 before calling the test method; other than :exc:`AssertionError` or :exc:`SkipTest`,
708 any exception raised by this method will be considered an error rather than
Terry Jan Reedy6ac42402014-04-15 23:38:18 -0400709 a test failure. The default implementation does nothing.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000710
711
712 .. method:: tearDown()
713
714 Method called immediately after the test method has been called and the
715 result recorded. This is called even if the test method raised an
716 exception, so the implementation in subclasses may need to be particularly
Ezio Melotti8dea74f2016-03-13 09:40:09 +0200717 careful about checking internal state. Any exception, other than
718 :exc:`AssertionError` or :exc:`SkipTest`, raised by this method will be
719 considered an additional error rather than a test failure (thus increasing
720 the total number of reported errors). This method will only be called if
721 the :meth:`setUp` succeeds, regardless of the outcome of the test method.
722 The default implementation does nothing.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000723
724
Benjamin Petersonb48af542010-04-11 20:43:16 +0000725 .. method:: setUpClass()
726
Ville Skyttäc33bb5d2018-08-23 17:49:18 +0300727 A class method called before tests in an individual class are run.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000728 ``setUpClass`` is called with the class as the only argument
729 and must be decorated as a :func:`classmethod`::
730
731 @classmethod
732 def setUpClass(cls):
733 ...
734
735 See `Class and Module Fixtures`_ for more details.
736
737 .. versionadded:: 3.2
738
739
740 .. method:: tearDownClass()
741
742 A class method called after tests in an individual class have run.
743 ``tearDownClass`` is called with the class as the only argument
744 and must be decorated as a :meth:`classmethod`::
745
746 @classmethod
747 def tearDownClass(cls):
748 ...
749
750 See `Class and Module Fixtures`_ for more details.
751
752 .. versionadded:: 3.2
753
754
Georg Brandl7f01a132009-09-16 15:58:14 +0000755 .. method:: run(result=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000756
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100757 Run the test, collecting the result into the :class:`TestResult` object
758 passed as *result*. If *result* is omitted or ``None``, a temporary
759 result object is created (by calling the :meth:`defaultTestResult`
760 method) and used. The result object is returned to :meth:`run`'s
761 caller.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000762
763 The same effect may be had by simply calling the :class:`TestCase`
764 instance.
765
Michael Foord1341bb02011-03-14 19:01:46 -0400766 .. versionchanged:: 3.3
767 Previous versions of ``run`` did not return the result. Neither did
768 calling an instance.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000769
Benjamin Petersone549ead2009-03-28 21:42:05 +0000770 .. method:: skipTest(reason)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000771
Stefan Kraha5bf3f52010-05-19 16:09:41 +0000772 Calling this during a test method or :meth:`setUp` skips the current
Benjamin Peterson52baa292009-03-24 00:56:30 +0000773 test. See :ref:`unittest-skipping` for more information.
774
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000775 .. versionadded:: 3.1
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000776
Benjamin Peterson52baa292009-03-24 00:56:30 +0000777
Antoine Pitrouc9b3ef22013-03-20 20:16:47 +0100778 .. method:: subTest(msg=None, **params)
779
780 Return a context manager which executes the enclosed code block as a
781 subtest. *msg* and *params* are optional, arbitrary values which are
782 displayed whenever a subtest fails, allowing you to identify them
783 clearly.
784
785 A test case can contain any number of subtest declarations, and
786 they can be arbitrarily nested.
787
788 See :ref:`subtests` for more information.
789
790 .. versionadded:: 3.4
791
792
Benjamin Peterson52baa292009-03-24 00:56:30 +0000793 .. method:: debug()
794
795 Run the test without collecting the result. This allows exceptions raised
796 by the test to be propagated to the caller, and can be used to support
797 running tests under a debugger.
798
Ezio Melotti22170ed2010-11-20 09:57:27 +0000799 .. _assert-methods:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000800
Ezio Melottif418db22016-01-12 11:03:31 +0200801 The :class:`TestCase` class provides several assert methods to check for and
802 report failures. The following table lists the most commonly used methods
803 (see the tables below for more assert methods):
Benjamin Peterson52baa292009-03-24 00:56:30 +0000804
Ezio Melotti4370b302010-11-03 20:39:14 +0000805 +-----------------------------------------+-----------------------------+---------------+
806 | Method | Checks that | New in |
807 +=========================================+=============================+===============+
808 | :meth:`assertEqual(a, b) | ``a == b`` | |
809 | <TestCase.assertEqual>` | | |
810 +-----------------------------------------+-----------------------------+---------------+
811 | :meth:`assertNotEqual(a, b) | ``a != b`` | |
812 | <TestCase.assertNotEqual>` | | |
813 +-----------------------------------------+-----------------------------+---------------+
814 | :meth:`assertTrue(x) | ``bool(x) is True`` | |
815 | <TestCase.assertTrue>` | | |
816 +-----------------------------------------+-----------------------------+---------------+
817 | :meth:`assertFalse(x) | ``bool(x) is False`` | |
818 | <TestCase.assertFalse>` | | |
819 +-----------------------------------------+-----------------------------+---------------+
820 | :meth:`assertIs(a, b) | ``a is b`` | 3.1 |
821 | <TestCase.assertIs>` | | |
822 +-----------------------------------------+-----------------------------+---------------+
823 | :meth:`assertIsNot(a, b) | ``a is not b`` | 3.1 |
824 | <TestCase.assertIsNot>` | | |
825 +-----------------------------------------+-----------------------------+---------------+
826 | :meth:`assertIsNone(x) | ``x is None`` | 3.1 |
827 | <TestCase.assertIsNone>` | | |
828 +-----------------------------------------+-----------------------------+---------------+
829 | :meth:`assertIsNotNone(x) | ``x is not None`` | 3.1 |
830 | <TestCase.assertIsNotNone>` | | |
831 +-----------------------------------------+-----------------------------+---------------+
832 | :meth:`assertIn(a, b) | ``a in b`` | 3.1 |
833 | <TestCase.assertIn>` | | |
834 +-----------------------------------------+-----------------------------+---------------+
835 | :meth:`assertNotIn(a, b) | ``a not in b`` | 3.1 |
836 | <TestCase.assertNotIn>` | | |
837 +-----------------------------------------+-----------------------------+---------------+
838 | :meth:`assertIsInstance(a, b) | ``isinstance(a, b)`` | 3.2 |
839 | <TestCase.assertIsInstance>` | | |
840 +-----------------------------------------+-----------------------------+---------------+
841 | :meth:`assertNotIsInstance(a, b) | ``not isinstance(a, b)`` | 3.2 |
842 | <TestCase.assertNotIsInstance>` | | |
843 +-----------------------------------------+-----------------------------+---------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000844
Ezio Melottib4dc2502011-05-06 15:01:41 +0300845 All the assert methods accept a *msg* argument that, if specified, is used
846 as the error message on failure (see also :data:`longMessage`).
847 Note that the *msg* keyword argument can be passed to :meth:`assertRaises`,
848 :meth:`assertRaisesRegex`, :meth:`assertWarns`, :meth:`assertWarnsRegex`
849 only when they are used as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000850
Michael Foorde180d392011-01-28 19:51:48 +0000851 .. method:: assertEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000852
Michael Foorde180d392011-01-28 19:51:48 +0000853 Test that *first* and *second* are equal. If the values do not
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000854 compare equal, the test will fail.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000855
Michael Foorde180d392011-01-28 19:51:48 +0000856 In addition, if *first* and *second* are the exact same type and one of
Michael Foord02834952010-02-08 23:10:39 +0000857 list, tuple, dict, set, frozenset or str or any type that a subclass
Ezio Melotti9ecb6be2012-01-16 08:28:54 +0200858 registers with :meth:`addTypeEqualityFunc` the type-specific equality
Michael Foord02834952010-02-08 23:10:39 +0000859 function will be called in order to generate a more useful default
Ezio Melotti22170ed2010-11-20 09:57:27 +0000860 error message (see also the :ref:`list of type-specific methods
861 <type-specific-methods>`).
Ezio Melotti4841fd62010-11-05 15:43:40 +0000862
Raymond Hettinger35a88362009-04-09 00:08:24 +0000863 .. versionchanged:: 3.1
Ezio Melotti9ecb6be2012-01-16 08:28:54 +0200864 Added the automatic calling of type-specific equality function.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000865
Michael Foord28a817e2010-02-09 00:03:57 +0000866 .. versionchanged:: 3.2
867 :meth:`assertMultiLineEqual` added as the default type equality
868 function for comparing strings.
Michael Foord02834952010-02-08 23:10:39 +0000869
Benjamin Peterson52baa292009-03-24 00:56:30 +0000870
Michael Foorde180d392011-01-28 19:51:48 +0000871 .. method:: assertNotEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000872
Michael Foorde180d392011-01-28 19:51:48 +0000873 Test that *first* and *second* are not equal. If the values do
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000874 compare equal, the test will fail.
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000875
Ezio Melotti4370b302010-11-03 20:39:14 +0000876 .. method:: assertTrue(expr, msg=None)
Ezio Melotti4841fd62010-11-05 15:43:40 +0000877 assertFalse(expr, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000878
Ezio Melotti4841fd62010-11-05 15:43:40 +0000879 Test that *expr* is true (or false).
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000880
Ezio Melotti4841fd62010-11-05 15:43:40 +0000881 Note that this is equivalent to ``bool(expr) is True`` and not to ``expr
882 is True`` (use ``assertIs(expr, True)`` for the latter). This method
883 should also be avoided when more specific methods are available (e.g.
884 ``assertEqual(a, b)`` instead of ``assertTrue(a == b)``), because they
885 provide a better error message in case of failure.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000886
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000887
Michael Foorde180d392011-01-28 19:51:48 +0000888 .. method:: assertIs(first, second, msg=None)
889 assertIsNot(first, second, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000890
Michael Foorde180d392011-01-28 19:51:48 +0000891 Test that *first* and *second* evaluate (or don't evaluate) to the
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000892 same object.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000893
Raymond Hettinger35a88362009-04-09 00:08:24 +0000894 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000895
896
Ezio Melotti4370b302010-11-03 20:39:14 +0000897 .. method:: assertIsNone(expr, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000898 assertIsNotNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000899
Serhiy Storchakaecf41da2016-10-19 16:29:26 +0300900 Test that *expr* is (or is not) ``None``.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000901
Ezio Melotti4370b302010-11-03 20:39:14 +0000902 .. versionadded:: 3.1
Benjamin Petersonb48af542010-04-11 20:43:16 +0000903
904
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000905 .. method:: assertIn(first, second, msg=None)
906 assertNotIn(first, second, msg=None)
907
Ezio Melotti4841fd62010-11-05 15:43:40 +0000908 Test that *first* is (or is not) in *second*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000909
Raymond Hettinger35a88362009-04-09 00:08:24 +0000910 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000911
912
Ezio Melotti9c02c2f2010-11-03 20:45:31 +0000913 .. method:: assertIsInstance(obj, cls, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000914 assertNotIsInstance(obj, cls, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000915
Ezio Melotti9794a262010-11-04 14:52:13 +0000916 Test that *obj* is (or is not) an instance of *cls* (which can be a
917 class or a tuple of classes, as supported by :func:`isinstance`).
Ezio Melotti80a61e82011-12-19 07:04:48 +0200918 To check for the exact type, use :func:`assertIs(type(obj), cls) <assertIs>`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000919
Ezio Melotti4370b302010-11-03 20:39:14 +0000920 .. versionadded:: 3.2
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000921
922
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000923
Ezio Melottif418db22016-01-12 11:03:31 +0200924 It is also possible to check the production of exceptions, warnings, and
Antoine Pitrou7c89ae22013-09-15 02:01:39 +0200925 log messages using the following methods:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000926
Ezio Melotti4370b302010-11-03 20:39:14 +0000927 +---------------------------------------------------------+--------------------------------------+------------+
928 | Method | Checks that | New in |
929 +=========================================================+======================================+============+
Éric Araujo941afed2011-09-01 02:47:34 +0200930 | :meth:`assertRaises(exc, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *exc* | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000931 | <TestCase.assertRaises>` | | |
932 +---------------------------------------------------------+--------------------------------------+------------+
Ezio Melotti560a7782013-09-13 22:17:40 +0300933 | :meth:`assertRaisesRegex(exc, r, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *exc* | 3.1 |
934 | <TestCase.assertRaisesRegex>` | and the message matches regex *r* | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000935 +---------------------------------------------------------+--------------------------------------+------------+
Éric Araujo941afed2011-09-01 02:47:34 +0200936 | :meth:`assertWarns(warn, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *warn* | 3.2 |
Ezio Melotti4370b302010-11-03 20:39:14 +0000937 | <TestCase.assertWarns>` | | |
938 +---------------------------------------------------------+--------------------------------------+------------+
Ezio Melotti560a7782013-09-13 22:17:40 +0300939 | :meth:`assertWarnsRegex(warn, r, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *warn* | 3.2 |
940 | <TestCase.assertWarnsRegex>` | and the message matches regex *r* | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000941 +---------------------------------------------------------+--------------------------------------+------------+
Georg Brandled007d52013-11-24 16:09:26 +0100942 | :meth:`assertLogs(logger, level) | The ``with`` block logs on *logger* | 3.4 |
943 | <TestCase.assertLogs>` | with minimum *level* | |
Antoine Pitrou7c89ae22013-09-15 02:01:39 +0200944 +---------------------------------------------------------+--------------------------------------+------------+
Benjamin Peterson52baa292009-03-24 00:56:30 +0000945
Georg Brandl7f01a132009-09-16 15:58:14 +0000946 .. method:: assertRaises(exception, callable, *args, **kwds)
Benjamin Petersonbe4e5b82018-10-01 22:18:44 -0700947 assertRaises(exception, *, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000948
949 Test that an exception is raised when *callable* is called with any
950 positional or keyword arguments that are also passed to
951 :meth:`assertRaises`. The test passes if *exception* is raised, is an
952 error if another exception is raised, or fails if no exception is raised.
953 To catch any of a group of exceptions, a tuple containing the exception
954 classes may be passed as *exception*.
955
Ezio Melottib4dc2502011-05-06 15:01:41 +0300956 If only the *exception* and possibly the *msg* arguments are given,
957 return a context manager so that the code under test can be written
958 inline rather than as a function::
Benjamin Petersonded31c42009-03-30 15:04:16 +0000959
Michael Foord41531f22010-02-05 21:13:40 +0000960 with self.assertRaises(SomeException):
Benjamin Petersonded31c42009-03-30 15:04:16 +0000961 do_something()
962
Ezio Melottib4dc2502011-05-06 15:01:41 +0300963 When used as a context manager, :meth:`assertRaises` accepts the
964 additional keyword argument *msg*.
965
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000966 The context manager will store the caught exception object in its
Ezio Melotti49008232010-02-08 21:57:48 +0000967 :attr:`exception` attribute. This can be useful if the intention
Michael Foord41531f22010-02-05 21:13:40 +0000968 is to perform additional checks on the exception raised::
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000969
Georg Brandl8a1caa22010-07-29 16:01:11 +0000970 with self.assertRaises(SomeException) as cm:
971 do_something()
Michael Foord41531f22010-02-05 21:13:40 +0000972
Georg Brandl8a1caa22010-07-29 16:01:11 +0000973 the_exception = cm.exception
974 self.assertEqual(the_exception.error_code, 3)
Michael Foord41531f22010-02-05 21:13:40 +0000975
Ezio Melotti49008232010-02-08 21:57:48 +0000976 .. versionchanged:: 3.1
Benjamin Petersonded31c42009-03-30 15:04:16 +0000977 Added the ability to use :meth:`assertRaises` as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000978
Ezio Melotti49008232010-02-08 21:57:48 +0000979 .. versionchanged:: 3.2
980 Added the :attr:`exception` attribute.
981
Ezio Melottib4dc2502011-05-06 15:01:41 +0300982 .. versionchanged:: 3.3
983 Added the *msg* keyword argument when used as a context manager.
984
Benjamin Peterson52baa292009-03-24 00:56:30 +0000985
Ezio Melottied3a7d22010-12-01 02:32:32 +0000986 .. method:: assertRaisesRegex(exception, regex, callable, *args, **kwds)
Benjamin Petersone006b392018-10-02 21:38:39 -0700987 assertRaisesRegex(exception, regex, *, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000988
Ezio Melottied3a7d22010-12-01 02:32:32 +0000989 Like :meth:`assertRaises` but also tests that *regex* matches
990 on the string representation of the raised exception. *regex* may be
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000991 a regular expression object or a string containing a regular expression
992 suitable for use by :func:`re.search`. Examples::
993
Terry Jan Reedyc4565a92013-06-29 13:15:43 -0400994 self.assertRaisesRegex(ValueError, "invalid literal for.*XYZ'$",
Ezio Melottied3a7d22010-12-01 02:32:32 +0000995 int, 'XYZ')
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000996
997 or::
998
Ezio Melottied3a7d22010-12-01 02:32:32 +0000999 with self.assertRaisesRegex(ValueError, 'literal'):
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001000 int('XYZ')
1001
Georg Brandl419e3de2010-12-01 15:44:25 +00001002 .. versionadded:: 3.1
1003 under the name ``assertRaisesRegexp``.
Ezio Melottib4dc2502011-05-06 15:01:41 +03001004
Ezio Melottied3a7d22010-12-01 02:32:32 +00001005 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001006 Renamed to :meth:`assertRaisesRegex`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001007
Ezio Melottib4dc2502011-05-06 15:01:41 +03001008 .. versionchanged:: 3.3
1009 Added the *msg* keyword argument when used as a context manager.
1010
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001011
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001012 .. method:: assertWarns(warning, callable, *args, **kwds)
Benjamin Petersone006b392018-10-02 21:38:39 -07001013 assertWarns(warning, *, msg=None)
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001014
1015 Test that a warning is triggered when *callable* is called with any
1016 positional or keyword arguments that are also passed to
1017 :meth:`assertWarns`. The test passes if *warning* is triggered and
Terry Jan Reedy778cba72013-07-30 22:31:06 -04001018 fails if it isn't. Any exception is an error.
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001019 To catch any of a group of warnings, a tuple containing the warning
1020 classes may be passed as *warnings*.
1021
Ezio Melottib4dc2502011-05-06 15:01:41 +03001022 If only the *warning* and possibly the *msg* arguments are given,
Terry Jan Reedy778cba72013-07-30 22:31:06 -04001023 return a context manager so that the code under test can be written
Ezio Melottib4dc2502011-05-06 15:01:41 +03001024 inline rather than as a function::
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001025
1026 with self.assertWarns(SomeWarning):
1027 do_something()
1028
Terry Jan Reedy9eda66d2013-07-27 16:15:29 -04001029 When used as a context manager, :meth:`assertWarns` accepts the
Ezio Melottib4dc2502011-05-06 15:01:41 +03001030 additional keyword argument *msg*.
1031
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001032 The context manager will store the caught warning object in its
1033 :attr:`warning` attribute, and the source line which triggered the
1034 warnings in the :attr:`filename` and :attr:`lineno` attributes.
1035 This can be useful if the intention is to perform additional checks
Terry Jan Reedy778cba72013-07-30 22:31:06 -04001036 on the warning caught::
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001037
1038 with self.assertWarns(SomeWarning) as cm:
1039 do_something()
1040
1041 self.assertIn('myfile.py', cm.filename)
1042 self.assertEqual(320, cm.lineno)
1043
1044 This method works regardless of the warning filters in place when it
1045 is called.
1046
1047 .. versionadded:: 3.2
1048
Ezio Melottib4dc2502011-05-06 15:01:41 +03001049 .. versionchanged:: 3.3
1050 Added the *msg* keyword argument when used as a context manager.
1051
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001052
Ezio Melottied3a7d22010-12-01 02:32:32 +00001053 .. method:: assertWarnsRegex(warning, regex, callable, *args, **kwds)
Benjamin Petersone006b392018-10-02 21:38:39 -07001054 assertWarnsRegex(warning, regex, *, msg=None)
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001055
Ezio Melottied3a7d22010-12-01 02:32:32 +00001056 Like :meth:`assertWarns` but also tests that *regex* matches on the
1057 message of the triggered warning. *regex* may be a regular expression
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001058 object or a string containing a regular expression suitable for use
1059 by :func:`re.search`. Example::
1060
Ezio Melottied3a7d22010-12-01 02:32:32 +00001061 self.assertWarnsRegex(DeprecationWarning,
1062 r'legacy_function\(\) is deprecated',
1063 legacy_function, 'XYZ')
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001064
1065 or::
1066
Ezio Melottied3a7d22010-12-01 02:32:32 +00001067 with self.assertWarnsRegex(RuntimeWarning, 'unsafe frobnicating'):
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001068 frobnicate('/etc/passwd')
1069
1070 .. versionadded:: 3.2
1071
Ezio Melottib4dc2502011-05-06 15:01:41 +03001072 .. versionchanged:: 3.3
1073 Added the *msg* keyword argument when used as a context manager.
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001074
Antoine Pitrou0715b9f2013-09-14 19:45:47 +02001075 .. method:: assertLogs(logger=None, level=None)
1076
1077 A context manager to test that at least one message is logged on
1078 the *logger* or one of its children, with at least the given
1079 *level*.
1080
1081 If given, *logger* should be a :class:`logging.Logger` object or a
1082 :class:`str` giving the name of a logger. The default is the root
1083 logger, which will catch all messages.
1084
1085 If given, *level* should be either a numeric logging level or
1086 its string equivalent (for example either ``"ERROR"`` or
1087 :attr:`logging.ERROR`). The default is :attr:`logging.INFO`.
1088
1089 The test passes if at least one message emitted inside the ``with``
1090 block matches the *logger* and *level* conditions, otherwise it fails.
1091
1092 The object returned by the context manager is a recording helper
1093 which keeps tracks of the matching log messages. It has two
1094 attributes:
1095
1096 .. attribute:: records
1097
1098 A list of :class:`logging.LogRecord` objects of the matching
1099 log messages.
1100
1101 .. attribute:: output
1102
1103 A list of :class:`str` objects with the formatted output of
1104 matching messages.
1105
1106 Example::
1107
1108 with self.assertLogs('foo', level='INFO') as cm:
1109 logging.getLogger('foo').info('first message')
1110 logging.getLogger('foo.bar').error('second message')
1111 self.assertEqual(cm.output, ['INFO:foo:first message',
1112 'ERROR:foo.bar:second message'])
1113
1114 .. versionadded:: 3.4
1115
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001116
Ezio Melotti4370b302010-11-03 20:39:14 +00001117 There are also other methods used to perform more specific checks, such as:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001118
Ezio Melotti4370b302010-11-03 20:39:14 +00001119 +---------------------------------------+--------------------------------+--------------+
1120 | Method | Checks that | New in |
1121 +=======================================+================================+==============+
1122 | :meth:`assertAlmostEqual(a, b) | ``round(a-b, 7) == 0`` | |
1123 | <TestCase.assertAlmostEqual>` | | |
1124 +---------------------------------------+--------------------------------+--------------+
1125 | :meth:`assertNotAlmostEqual(a, b) | ``round(a-b, 7) != 0`` | |
1126 | <TestCase.assertNotAlmostEqual>` | | |
1127 +---------------------------------------+--------------------------------+--------------+
1128 | :meth:`assertGreater(a, b) | ``a > b`` | 3.1 |
1129 | <TestCase.assertGreater>` | | |
1130 +---------------------------------------+--------------------------------+--------------+
1131 | :meth:`assertGreaterEqual(a, b) | ``a >= b`` | 3.1 |
1132 | <TestCase.assertGreaterEqual>` | | |
1133 +---------------------------------------+--------------------------------+--------------+
1134 | :meth:`assertLess(a, b) | ``a < b`` | 3.1 |
1135 | <TestCase.assertLess>` | | |
1136 +---------------------------------------+--------------------------------+--------------+
1137 | :meth:`assertLessEqual(a, b) | ``a <= b`` | 3.1 |
1138 | <TestCase.assertLessEqual>` | | |
1139 +---------------------------------------+--------------------------------+--------------+
Ezio Melotti560a7782013-09-13 22:17:40 +03001140 | :meth:`assertRegex(s, r) | ``r.search(s)`` | 3.1 |
Ezio Melottied3a7d22010-12-01 02:32:32 +00001141 | <TestCase.assertRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001142 +---------------------------------------+--------------------------------+--------------+
Ezio Melotti560a7782013-09-13 22:17:40 +03001143 | :meth:`assertNotRegex(s, r) | ``not r.search(s)`` | 3.2 |
Ezio Melottied3a7d22010-12-01 02:32:32 +00001144 | <TestCase.assertNotRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001145 +---------------------------------------+--------------------------------+--------------+
Éric Araujo941afed2011-09-01 02:47:34 +02001146 | :meth:`assertCountEqual(a, b) | *a* and *b* have the same | 3.2 |
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001147 | <TestCase.assertCountEqual>` | elements in the same number, | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001148 | | regardless of their order | |
1149 +---------------------------------------+--------------------------------+--------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001150
1151
Michael Foorde180d392011-01-28 19:51:48 +00001152 .. method:: assertAlmostEqual(first, second, places=7, msg=None, delta=None)
1153 assertNotAlmostEqual(first, second, places=7, msg=None, delta=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001154
Michael Foorde180d392011-01-28 19:51:48 +00001155 Test that *first* and *second* are approximately (or not approximately)
Ezio Melotti4841fd62010-11-05 15:43:40 +00001156 equal by computing the difference, rounding to the given number of
1157 decimal *places* (default 7), and comparing to zero. Note that these
1158 methods round the values to the given number of *decimal places* (i.e.
1159 like the :func:`round` function) and not *significant digits*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001160
Ezio Melotti4370b302010-11-03 20:39:14 +00001161 If *delta* is supplied instead of *places* then the difference
Ezio Melottid51914c2013-08-11 13:04:50 +03001162 between *first* and *second* must be less or equal to (or greater than) *delta*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001163
Stéphane Wirtele483f022018-10-26 12:52:11 +02001164 Supplying both *delta* and *places* raises a :exc:`TypeError`.
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001165
Ezio Melotti4370b302010-11-03 20:39:14 +00001166 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001167 :meth:`assertAlmostEqual` automatically considers almost equal objects
1168 that compare equal. :meth:`assertNotAlmostEqual` automatically fails
1169 if the objects compare equal. Added the *delta* keyword argument.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001170
Ezio Melotti4370b302010-11-03 20:39:14 +00001171
Michael Foorde180d392011-01-28 19:51:48 +00001172 .. method:: assertGreater(first, second, msg=None)
1173 assertGreaterEqual(first, second, msg=None)
1174 assertLess(first, second, msg=None)
1175 assertLessEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001176
Michael Foorde180d392011-01-28 19:51:48 +00001177 Test that *first* is respectively >, >=, < or <= than *second* depending
Ezio Melotti4841fd62010-11-05 15:43:40 +00001178 on the method name. If not, the test will fail::
Ezio Melotti4370b302010-11-03 20:39:14 +00001179
1180 >>> self.assertGreaterEqual(3, 4)
1181 AssertionError: "3" unexpectedly not greater than or equal to "4"
1182
1183 .. versionadded:: 3.1
1184
1185
Ezio Melottied3a7d22010-12-01 02:32:32 +00001186 .. method:: assertRegex(text, regex, msg=None)
1187 assertNotRegex(text, regex, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001188
Ezio Melottied3a7d22010-12-01 02:32:32 +00001189 Test that a *regex* search matches (or does not match) *text*. In case
Ezio Melotti4841fd62010-11-05 15:43:40 +00001190 of failure, the error message will include the pattern and the *text* (or
Ezio Melottied3a7d22010-12-01 02:32:32 +00001191 the pattern and the part of *text* that unexpectedly matched). *regex*
Ezio Melotti4370b302010-11-03 20:39:14 +00001192 may be a regular expression object or a string containing a regular
1193 expression suitable for use by :func:`re.search`.
1194
Georg Brandl419e3de2010-12-01 15:44:25 +00001195 .. versionadded:: 3.1
1196 under the name ``assertRegexpMatches``.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001197 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001198 The method ``assertRegexpMatches()`` has been renamed to
1199 :meth:`.assertRegex`.
1200 .. versionadded:: 3.2
1201 :meth:`.assertNotRegex`.
Jim Fasarakis-Hilliard74921ed2017-06-09 23:27:20 +03001202 .. versionadded:: 3.5
1203 The name ``assertNotRegexpMatches`` is a deprecated alias
1204 for :meth:`.assertNotRegex`.
Ezio Melotti4370b302010-11-03 20:39:14 +00001205
1206
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001207 .. method:: assertCountEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001208
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001209 Test that sequence *first* contains the same elements as *second*,
Ezio Melotti4370b302010-11-03 20:39:14 +00001210 regardless of their order. When they don't, an error message listing the
1211 differences between the sequences will be generated.
1212
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001213 Duplicate elements are *not* ignored when comparing *first* and
1214 *second*. It verifies whether each element has the same count in both
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001215 sequences. Equivalent to:
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001216 ``assertEqual(Counter(list(first)), Counter(list(second)))``
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001217 but works with sequences of unhashable objects as well.
Ezio Melotti4370b302010-11-03 20:39:14 +00001218
Ezio Melotti4370b302010-11-03 20:39:14 +00001219 .. versionadded:: 3.2
1220
Ezio Melotti4370b302010-11-03 20:39:14 +00001221
Ezio Melotti22170ed2010-11-20 09:57:27 +00001222 .. _type-specific-methods:
Ezio Melotti4370b302010-11-03 20:39:14 +00001223
Ezio Melotti22170ed2010-11-20 09:57:27 +00001224 The :meth:`assertEqual` method dispatches the equality check for objects of
1225 the same type to different type-specific methods. These methods are already
1226 implemented for most of the built-in types, but it's also possible to
1227 register new methods using :meth:`addTypeEqualityFunc`:
1228
1229 .. method:: addTypeEqualityFunc(typeobj, function)
1230
1231 Registers a type-specific method called by :meth:`assertEqual` to check
1232 if two objects of exactly the same *typeobj* (not subclasses) compare
1233 equal. *function* must take two positional arguments and a third msg=None
1234 keyword argument just as :meth:`assertEqual` does. It must raise
1235 :data:`self.failureException(msg) <failureException>` when inequality
1236 between the first two parameters is detected -- possibly providing useful
1237 information and explaining the inequalities in details in the error
1238 message.
1239
1240 .. versionadded:: 3.1
1241
1242 The list of type-specific methods automatically used by
1243 :meth:`~TestCase.assertEqual` are summarized in the following table. Note
1244 that it's usually not necessary to invoke these methods directly.
Ezio Melotti4370b302010-11-03 20:39:14 +00001245
1246 +-----------------------------------------+-----------------------------+--------------+
1247 | Method | Used to compare | New in |
1248 +=========================================+=============================+==============+
1249 | :meth:`assertMultiLineEqual(a, b) | strings | 3.1 |
1250 | <TestCase.assertMultiLineEqual>` | | |
1251 +-----------------------------------------+-----------------------------+--------------+
1252 | :meth:`assertSequenceEqual(a, b) | sequences | 3.1 |
1253 | <TestCase.assertSequenceEqual>` | | |
1254 +-----------------------------------------+-----------------------------+--------------+
1255 | :meth:`assertListEqual(a, b) | lists | 3.1 |
1256 | <TestCase.assertListEqual>` | | |
1257 +-----------------------------------------+-----------------------------+--------------+
1258 | :meth:`assertTupleEqual(a, b) | tuples | 3.1 |
1259 | <TestCase.assertTupleEqual>` | | |
1260 +-----------------------------------------+-----------------------------+--------------+
1261 | :meth:`assertSetEqual(a, b) | sets or frozensets | 3.1 |
1262 | <TestCase.assertSetEqual>` | | |
1263 +-----------------------------------------+-----------------------------+--------------+
1264 | :meth:`assertDictEqual(a, b) | dicts | 3.1 |
1265 | <TestCase.assertDictEqual>` | | |
1266 +-----------------------------------------+-----------------------------+--------------+
1267
1268
1269
Michael Foorde180d392011-01-28 19:51:48 +00001270 .. method:: assertMultiLineEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001271
Michael Foorde180d392011-01-28 19:51:48 +00001272 Test that the multiline string *first* is equal to the string *second*.
Ezio Melotti4370b302010-11-03 20:39:14 +00001273 When not equal a diff of the two strings highlighting the differences
1274 will be included in the error message. This method is used by default
1275 when comparing strings with :meth:`assertEqual`.
1276
Ezio Melotti4370b302010-11-03 20:39:14 +00001277 .. versionadded:: 3.1
1278
1279
Michael Foorde180d392011-01-28 19:51:48 +00001280 .. method:: assertSequenceEqual(first, second, msg=None, seq_type=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001281
1282 Tests that two sequences are equal. If a *seq_type* is supplied, both
Michael Foorde180d392011-01-28 19:51:48 +00001283 *first* and *second* must be instances of *seq_type* or a failure will
Ezio Melotti4370b302010-11-03 20:39:14 +00001284 be raised. If the sequences are different an error message is
1285 constructed that shows the difference between the two.
1286
Ezio Melotti22170ed2010-11-20 09:57:27 +00001287 This method is not called directly by :meth:`assertEqual`, but
1288 it's used to implement :meth:`assertListEqual` and
Ezio Melotti4370b302010-11-03 20:39:14 +00001289 :meth:`assertTupleEqual`.
1290
1291 .. versionadded:: 3.1
1292
1293
Michael Foorde180d392011-01-28 19:51:48 +00001294 .. method:: assertListEqual(first, second, msg=None)
1295 assertTupleEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001296
Ezio Melotti49ccd512012-08-29 17:50:42 +03001297 Tests that two lists or tuples are equal. If not, an error message is
Ezio Melotti4370b302010-11-03 20:39:14 +00001298 constructed that shows only the differences between the two. An error
1299 is also raised if either of the parameters are of the wrong type.
1300 These methods are used by default when comparing lists or tuples with
1301 :meth:`assertEqual`.
1302
Ezio Melotti4370b302010-11-03 20:39:14 +00001303 .. versionadded:: 3.1
1304
1305
Michael Foorde180d392011-01-28 19:51:48 +00001306 .. method:: assertSetEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001307
1308 Tests that two sets are equal. If not, an error message is constructed
1309 that lists the differences between the sets. This method is used by
1310 default when comparing sets or frozensets with :meth:`assertEqual`.
1311
Michael Foorde180d392011-01-28 19:51:48 +00001312 Fails if either of *first* or *second* does not have a :meth:`set.difference`
Ezio Melotti4370b302010-11-03 20:39:14 +00001313 method.
1314
Ezio Melotti4370b302010-11-03 20:39:14 +00001315 .. versionadded:: 3.1
1316
1317
Michael Foorde180d392011-01-28 19:51:48 +00001318 .. method:: assertDictEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001319
1320 Test that two dictionaries are equal. If not, an error message is
1321 constructed that shows the differences in the dictionaries. This
1322 method will be used by default to compare dictionaries in
1323 calls to :meth:`assertEqual`.
1324
Ezio Melotti4370b302010-11-03 20:39:14 +00001325 .. versionadded:: 3.1
1326
1327
1328
Ezio Melotti22170ed2010-11-20 09:57:27 +00001329 .. _other-methods-and-attrs:
1330
Ezio Melotti4370b302010-11-03 20:39:14 +00001331 Finally the :class:`TestCase` provides the following methods and attributes:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001332
Benjamin Peterson52baa292009-03-24 00:56:30 +00001333
Georg Brandl7f01a132009-09-16 15:58:14 +00001334 .. method:: fail(msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001335
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001336 Signals a test failure unconditionally, with *msg* or ``None`` for
Benjamin Peterson52baa292009-03-24 00:56:30 +00001337 the error message.
1338
1339
1340 .. attribute:: failureException
1341
1342 This class attribute gives the exception raised by the test method. If a
1343 test framework needs to use a specialized exception, possibly to carry
1344 additional information, it must subclass this exception in order to "play
1345 fair" with the framework. The initial value of this attribute is
1346 :exc:`AssertionError`.
1347
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001348
1349 .. attribute:: longMessage
1350
Guido van Rossum4a452352016-10-13 14:23:01 -07001351 This class attribute determines what happens when a custom failure message
1352 is passed as the msg argument to an assertXYY call that fails.
1353 ``True`` is the default value. In this case, the custom message is appended
1354 to the end of the standard failure message.
1355 When set to ``False``, the custom message replaces the standard message.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001356
Guido van Rossum4a452352016-10-13 14:23:01 -07001357 The class setting can be overridden in individual test methods by assigning
1358 an instance attribute, self.longMessage, to ``True`` or ``False`` before
1359 calling the assert methods.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001360
Guido van Rossum4a452352016-10-13 14:23:01 -07001361 The class setting gets reset before each test call.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001362
Raymond Hettinger35a88362009-04-09 00:08:24 +00001363 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001364
1365
Michael Foord98b3e762010-06-05 21:59:55 +00001366 .. attribute:: maxDiff
1367
1368 This attribute controls the maximum length of diffs output by assert
1369 methods that report diffs on failure. It defaults to 80*8 characters.
1370 Assert methods affected by this attribute are
1371 :meth:`assertSequenceEqual` (including all the sequence comparison
1372 methods that delegate to it), :meth:`assertDictEqual` and
1373 :meth:`assertMultiLineEqual`.
1374
Serhiy Storchakaecf41da2016-10-19 16:29:26 +03001375 Setting ``maxDiff`` to ``None`` means that there is no maximum length of
Michael Foord98b3e762010-06-05 21:59:55 +00001376 diffs.
1377
1378 .. versionadded:: 3.2
1379
1380
Benjamin Peterson52baa292009-03-24 00:56:30 +00001381 Testing frameworks can use the following methods to collect information on
1382 the test:
1383
1384
1385 .. method:: countTestCases()
1386
1387 Return the number of tests represented by this test object. For
1388 :class:`TestCase` instances, this will always be ``1``.
1389
1390
1391 .. method:: defaultTestResult()
1392
1393 Return an instance of the test result class that should be used for this
1394 test case class (if no other result instance is provided to the
1395 :meth:`run` method).
1396
1397 For :class:`TestCase` instances, this will always be an instance of
1398 :class:`TestResult`; subclasses of :class:`TestCase` should override this
1399 as necessary.
1400
1401
1402 .. method:: id()
1403
1404 Return a string identifying the specific test case. This is usually the
1405 full name of the test method, including the module and class name.
1406
1407
1408 .. method:: shortDescription()
1409
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001410 Returns a description of the test, or ``None`` if no description
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001411 has been provided. The default implementation of this method
1412 returns the first line of the test method's docstring, if available,
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001413 or ``None``.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001414
Georg Brandl419e3de2010-12-01 15:44:25 +00001415 .. versionchanged:: 3.1
Michael Foord34c94622010-02-10 15:51:42 +00001416 In 3.1 this was changed to add the test name to the short description
Georg Brandl419e3de2010-12-01 15:44:25 +00001417 even in the presence of a docstring. This caused compatibility issues
Michael Foord34c94622010-02-10 15:51:42 +00001418 with unittest extensions and adding the test name was moved to the
Georg Brandl419e3de2010-12-01 15:44:25 +00001419 :class:`TextTestResult` in Python 3.2.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001420
Georg Brandl116aa622007-08-15 14:28:22 +00001421
Georg Brandl7f01a132009-09-16 15:58:14 +00001422 .. method:: addCleanup(function, *args, **kwargs)
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001423
1424 Add a function to be called after :meth:`tearDown` to cleanup resources
1425 used during the test. Functions will be called in reverse order to the
Serhiy Storchaka4ecfa452016-05-16 09:31:54 +03001426 order they are added (:abbr:`LIFO (last-in, first-out)`). They
1427 are called with any arguments and keyword arguments passed into
1428 :meth:`addCleanup` when they are added.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001429
1430 If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
1431 then any cleanup functions added will still be called.
1432
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001433 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001434
1435
1436 .. method:: doCleanups()
1437
Barry Warsaw0c9fd632010-04-12 14:50:57 +00001438 This method is called unconditionally after :meth:`tearDown`, or
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001439 after :meth:`setUp` if :meth:`setUp` raises an exception.
1440
1441 It is responsible for calling all the cleanup functions added by
1442 :meth:`addCleanup`. If you need cleanup functions to be called
1443 *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
1444 yourself.
1445
1446 :meth:`doCleanups` pops methods off the stack of cleanup
1447 functions one at a time, so it can be called at any time.
1448
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001449 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001450
Lisa Roach0f221d02018-11-08 18:34:33 -08001451 .. classmethod:: addClassCleanup(function, *args, **kwargs)
1452
1453 Add a function to be called after :meth:`tearDownClass` to cleanup
1454 resources used during the test class. Functions will be called in reverse
1455 order to the order they are added (:abbr:`LIFO (last-in, first-out)`).
1456 They are called with any arguments and keyword arguments passed into
1457 :meth:`addClassCleanup` when they are added.
1458
1459 If :meth:`setUpClass` fails, meaning that :meth:`tearDownClass` is not
1460 called, then any cleanup functions added will still be called.
1461
1462 .. versionadded:: 3.8
1463
1464
1465 .. classmethod:: doClassCleanups()
1466
1467 This method is called unconditionally after :meth:`tearDownClass`, or
1468 after :meth:`setUpClass` if :meth:`setUpClass` raises an exception.
1469
1470 It is responsible for calling all the cleanup functions added by
1471 :meth:`addCleanupClass`. If you need cleanup functions to be called
1472 *prior* to :meth:`tearDownClass` then you can call
1473 :meth:`doCleanupsClass` yourself.
1474
1475 :meth:`doCleanupsClass` pops methods off the stack of cleanup
1476 functions one at a time, so it can be called at any time.
1477
1478 .. versionadded:: 3.8
1479
1480
1481
1482
1483
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001484
Georg Brandl7f01a132009-09-16 15:58:14 +00001485.. class:: FunctionTestCase(testFunc, setUp=None, tearDown=None, description=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001486
1487 This class implements the portion of the :class:`TestCase` interface which
Benjamin Petersond2397752009-06-27 23:45:02 +00001488 allows the test runner to drive the test, but does not provide the methods
1489 which test code can use to check and report errors. This is used to create
1490 test cases using legacy test code, allowing it to be integrated into a
1491 :mod:`unittest`-based test framework.
Georg Brandl116aa622007-08-15 14:28:22 +00001492
1493
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001494.. _deprecated-aliases:
1495
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001496Deprecated aliases
1497##################
1498
1499For historical reasons, some of the :class:`TestCase` methods had one or more
1500aliases that are now deprecated. The following table lists the correct names
1501along with their deprecated aliases:
1502
Jim Fasarakis-Hilliard74921ed2017-06-09 23:27:20 +03001503 ============================== ====================== =======================
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001504 Method Name Deprecated alias Deprecated alias
Jim Fasarakis-Hilliard74921ed2017-06-09 23:27:20 +03001505 ============================== ====================== =======================
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001506 :meth:`.assertEqual` failUnlessEqual assertEquals
1507 :meth:`.assertNotEqual` failIfEqual assertNotEquals
1508 :meth:`.assertTrue` failUnless assert\_
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001509 :meth:`.assertFalse` failIf
1510 :meth:`.assertRaises` failUnlessRaises
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001511 :meth:`.assertAlmostEqual` failUnlessAlmostEqual assertAlmostEquals
1512 :meth:`.assertNotAlmostEqual` failIfAlmostEqual assertNotAlmostEquals
Ezio Melottied3a7d22010-12-01 02:32:32 +00001513 :meth:`.assertRegex` assertRegexpMatches
Jim Fasarakis-Hilliard74921ed2017-06-09 23:27:20 +03001514 :meth:`.assertNotRegex` assertNotRegexpMatches
Ezio Melottied3a7d22010-12-01 02:32:32 +00001515 :meth:`.assertRaisesRegex` assertRaisesRegexp
Jim Fasarakis-Hilliard74921ed2017-06-09 23:27:20 +03001516 ============================== ====================== =======================
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001517
Ezio Melotti361467e2011-04-03 17:37:58 +03001518 .. deprecated:: 3.1
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001519 the fail* aliases listed in the second column.
1520 .. deprecated:: 3.2
1521 the assert* aliases listed in the third column.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001522 .. deprecated:: 3.2
1523 ``assertRegexpMatches`` and ``assertRaisesRegexp`` have been renamed to
Jim Fasarakis-Hilliard74921ed2017-06-09 23:27:20 +03001524 :meth:`.assertRegex` and :meth:`.assertRaisesRegex`.
1525 .. deprecated:: 3.5
1526 the ``assertNotRegexpMatches`` name in favor of :meth:`.assertNotRegex`.
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001527
Benjamin Peterson52baa292009-03-24 00:56:30 +00001528.. _testsuite-objects:
1529
Benjamin Peterson52baa292009-03-24 00:56:30 +00001530Grouping tests
1531~~~~~~~~~~~~~~
1532
Georg Brandl7f01a132009-09-16 15:58:14 +00001533.. class:: TestSuite(tests=())
Georg Brandl116aa622007-08-15 14:28:22 +00001534
Martin Panter37f183d2017-01-18 12:06:38 +00001535 This class represents an aggregation of individual test cases and test suites.
Georg Brandl116aa622007-08-15 14:28:22 +00001536 The class presents the interface needed by the test runner to allow it to be run
1537 as any other test case. Running a :class:`TestSuite` instance is the same as
1538 iterating over the suite, running each test individually.
1539
1540 If *tests* is given, it must be an iterable of individual test cases or other
1541 test suites that will be used to build the suite initially. Additional methods
1542 are provided to add test cases and suites to the collection later on.
1543
Benjamin Peterson14a3dd72009-05-25 00:51:58 +00001544 :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
1545 they do not actually implement a test. Instead, they are used to aggregate
1546 tests into groups of tests that should be run together. Some additional
1547 methods are available to add tests to :class:`TestSuite` instances:
Benjamin Peterson52baa292009-03-24 00:56:30 +00001548
1549
1550 .. method:: TestSuite.addTest(test)
1551
1552 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1553
1554
1555 .. method:: TestSuite.addTests(tests)
1556
1557 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
1558 instances to this test suite.
1559
Benjamin Petersond2397752009-06-27 23:45:02 +00001560 This is equivalent to iterating over *tests*, calling :meth:`addTest` for
1561 each element.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001562
1563 :class:`TestSuite` shares the following methods with :class:`TestCase`:
1564
1565
1566 .. method:: run(result)
1567
1568 Run the tests associated with this suite, collecting the result into the
1569 test result object passed as *result*. Note that unlike
1570 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
1571 be passed in.
1572
1573
1574 .. method:: debug()
1575
1576 Run the tests associated with this suite without collecting the
1577 result. This allows exceptions raised by the test to be propagated to the
1578 caller and can be used to support running tests under a debugger.
1579
1580
1581 .. method:: countTestCases()
1582
1583 Return the number of tests represented by this test object, including all
1584 individual tests and sub-suites.
1585
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001586
1587 .. method:: __iter__()
1588
1589 Tests grouped by a :class:`TestSuite` are always accessed by iteration.
1590 Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
Andrew Svetloveb973682013-08-28 21:28:38 +03001591 that this method may be called several times on a single suite (for
1592 example when counting tests or comparing for equality) so the tests
1593 returned by repeated iterations before :meth:`TestSuite.run` must be the
1594 same for each call iteration. After :meth:`TestSuite.run`, callers should
1595 not rely on the tests returned by this method unless the caller uses a
1596 subclass that overrides :meth:`TestSuite._removeTestAtIndex` to preserve
1597 test references.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001598
Georg Brandl853947a2010-01-31 18:53:23 +00001599 .. versionchanged:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001600 In earlier versions the :class:`TestSuite` accessed tests directly rather
1601 than through iteration, so overriding :meth:`__iter__` wasn't sufficient
1602 for providing tests.
1603
Andrew Svetloveb973682013-08-28 21:28:38 +03001604 .. versionchanged:: 3.4
1605 In earlier versions the :class:`TestSuite` held references to each
1606 :class:`TestCase` after :meth:`TestSuite.run`. Subclasses can restore
1607 that behavior by overriding :meth:`TestSuite._removeTestAtIndex`.
1608
Benjamin Peterson52baa292009-03-24 00:56:30 +00001609 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
1610 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1611
1612
Benjamin Peterson52baa292009-03-24 00:56:30 +00001613Loading and running tests
1614~~~~~~~~~~~~~~~~~~~~~~~~~
1615
Georg Brandl116aa622007-08-15 14:28:22 +00001616.. class:: TestLoader()
1617
Benjamin Peterson52baa292009-03-24 00:56:30 +00001618 The :class:`TestLoader` class is used to create test suites from classes and
1619 modules. Normally, there is no need to create an instance of this class; the
1620 :mod:`unittest` module provides an instance that can be shared as
Ezio Melottib8e336b2012-04-29 10:52:18 +03001621 :data:`unittest.defaultTestLoader`. Using a subclass or instance, however,
1622 allows customization of some configurable properties.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001623
Robert Collinsf920c212014-10-20 13:24:05 +13001624 :class:`TestLoader` objects have the following attributes:
1625
1626
1627 .. attribute:: errors
1628
1629 A list of the non-fatal errors encountered while loading tests. Not reset
1630 by the loader at any point. Fatal errors are signalled by the relevant
1631 a method raising an exception to the caller. Non-fatal errors are also
1632 indicated by a synthetic test that will raise the original error when
1633 run.
1634
1635 .. versionadded:: 3.5
1636
1637
Benjamin Peterson52baa292009-03-24 00:56:30 +00001638 :class:`TestLoader` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001639
Ezio Melotti9c02c2f2010-11-03 20:45:31 +00001640
Benjamin Peterson52baa292009-03-24 00:56:30 +00001641 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl116aa622007-08-15 14:28:22 +00001642
Martin Panter37f183d2017-01-18 12:06:38 +00001643 Return a suite of all test cases contained in the :class:`TestCase`\ -derived
Benjamin Peterson52baa292009-03-24 00:56:30 +00001644 :class:`testCaseClass`.
1645
Robert Collinse02f6c22015-07-23 06:37:26 +12001646 A test case instance is created for each method named by
1647 :meth:`getTestCaseNames`. By default these are the method names
1648 beginning with ``test``. If :meth:`getTestCaseNames` returns no
1649 methods, but the :meth:`runTest` method is implemented, a single test
1650 case is created for that method instead.
1651
Benjamin Peterson52baa292009-03-24 00:56:30 +00001652
Barry Warsawd78742a2014-09-08 14:21:37 -04001653 .. method:: loadTestsFromModule(module, pattern=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001654
Martin Panter37f183d2017-01-18 12:06:38 +00001655 Return a suite of all test cases contained in the given module. This
Benjamin Peterson52baa292009-03-24 00:56:30 +00001656 method searches *module* for classes derived from :class:`TestCase` and
1657 creates an instance of the class for each test method defined for the
1658 class.
1659
Georg Brandle720c0a2009-04-27 16:20:50 +00001660 .. note::
Benjamin Peterson52baa292009-03-24 00:56:30 +00001661
1662 While using a hierarchy of :class:`TestCase`\ -derived classes can be
1663 convenient in sharing fixtures and helper functions, defining test
1664 methods on base classes that are not intended to be instantiated
1665 directly does not play well with this method. Doing so, however, can
1666 be useful when the fixtures are different and defined in subclasses.
1667
Benjamin Petersond2397752009-06-27 23:45:02 +00001668 If a module provides a ``load_tests`` function it will be called to
1669 load the tests. This allows modules to customize test loading.
Barry Warsawd78742a2014-09-08 14:21:37 -04001670 This is the `load_tests protocol`_. The *pattern* argument is passed as
1671 the third argument to ``load_tests``.
Benjamin Petersond2397752009-06-27 23:45:02 +00001672
Georg Brandl853947a2010-01-31 18:53:23 +00001673 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001674 Support for ``load_tests`` added.
1675
Barry Warsawd78742a2014-09-08 14:21:37 -04001676 .. versionchanged:: 3.5
1677 The undocumented and unofficial *use_load_tests* default argument is
1678 deprecated and ignored, although it is still accepted for backward
1679 compatibility. The method also now accepts a keyword-only argument
1680 *pattern* which is passed to ``load_tests`` as the third argument.
1681
Benjamin Peterson52baa292009-03-24 00:56:30 +00001682
Georg Brandl7f01a132009-09-16 15:58:14 +00001683 .. method:: loadTestsFromName(name, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001684
Martin Panter37f183d2017-01-18 12:06:38 +00001685 Return a suite of all test cases given a string specifier.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001686
1687 The specifier *name* is a "dotted name" that may resolve either to a
1688 module, a test case class, a test method within a test case class, a
1689 :class:`TestSuite` instance, or a callable object which returns a
1690 :class:`TestCase` or :class:`TestSuite` instance. These checks are
1691 applied in the order listed here; that is, a method on a possible test
1692 case class will be picked up as "a test method within a test case class",
1693 rather than "a callable object".
1694
1695 For example, if you have a module :mod:`SampleTests` containing a
1696 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
1697 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
Benjamin Petersond2397752009-06-27 23:45:02 +00001698 specifier ``'SampleTests.SampleTestCase'`` would cause this method to
1699 return a suite which will run all three test methods. Using the specifier
1700 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
1701 suite which will run only the :meth:`test_two` test method. The specifier
1702 can refer to modules and packages which have not been imported; they will
1703 be imported as a side-effect.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001704
1705 The method optionally resolves *name* relative to the given *module*.
1706
Martin Panter536d70e2017-01-14 08:23:08 +00001707 .. versionchanged:: 3.5
1708 If an :exc:`ImportError` or :exc:`AttributeError` occurs while traversing
1709 *name* then a synthetic test that raises that error when run will be
1710 returned. These errors are included in the errors accumulated by
1711 self.errors.
Robert Collins659dd622014-10-30 08:27:27 +13001712
Benjamin Peterson52baa292009-03-24 00:56:30 +00001713
Georg Brandl7f01a132009-09-16 15:58:14 +00001714 .. method:: loadTestsFromNames(names, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001715
1716 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
1717 than a single name. The return value is a test suite which supports all
1718 the tests defined for each name.
1719
1720
1721 .. method:: getTestCaseNames(testCaseClass)
1722
1723 Return a sorted sequence of method names found within *testCaseClass*;
1724 this should be a subclass of :class:`TestCase`.
1725
Benjamin Petersond2397752009-06-27 23:45:02 +00001726
1727 .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)
1728
Larry Hastings3732ed22014-03-15 21:13:56 -07001729 Find all the test modules by recursing into subdirectories from the
1730 specified start directory, and return a TestSuite object containing them.
1731 Only test files that match *pattern* will be loaded. (Using shell style
1732 pattern matching.) Only module names that are importable (i.e. are valid
1733 Python identifiers) will be loaded.
Benjamin Petersond2397752009-06-27 23:45:02 +00001734
1735 All test modules must be importable from the top level of the project. If
1736 the start directory is not the top level directory then the top level
1737 directory must be specified separately.
1738
Barry Warsawd78742a2014-09-08 14:21:37 -04001739 If importing a module fails, for example due to a syntax error, then
1740 this will be recorded as a single error and discovery will continue. If
1741 the import failure is due to :exc:`SkipTest` being raised, it will be
1742 recorded as a skip instead of an error.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001743
Barry Warsawd78742a2014-09-08 14:21:37 -04001744 If a package (a directory containing a file named :file:`__init__.py`) is
1745 found, the package will be checked for a ``load_tests`` function. If this
Robert Collinsbf2bda32014-11-05 03:09:01 +13001746 exists then it will be called
1747 ``package.load_tests(loader, tests, pattern)``. Test discovery takes care
1748 to ensure that a package is only checked for tests once during an
1749 invocation, even if the load_tests function itself calls
1750 ``loader.discover``.
Benjamin Petersond2397752009-06-27 23:45:02 +00001751
Barry Warsawd78742a2014-09-08 14:21:37 -04001752 If ``load_tests`` exists then discovery does *not* recurse into the
1753 package, ``load_tests`` is responsible for loading all tests in the
1754 package.
Benjamin Petersond2397752009-06-27 23:45:02 +00001755
1756 The pattern is deliberately not stored as a loader attribute so that
1757 packages can continue discovery themselves. *top_level_dir* is stored so
1758 ``load_tests`` does not need to pass this argument in to
1759 ``loader.discover()``.
1760
Benjamin Petersonb48af542010-04-11 20:43:16 +00001761 *start_dir* can be a dotted module name as well as a directory.
1762
Georg Brandl853947a2010-01-31 18:53:23 +00001763 .. versionadded:: 3.2
1764
Ezio Melottieae2b382013-03-01 14:47:50 +02001765 .. versionchanged:: 3.4
Ezio Melotti67ddcca2013-03-27 20:13:59 +02001766 Modules that raise :exc:`SkipTest` on import are recorded as skips,
Larry Hastings3732ed22014-03-15 21:13:56 -07001767 not errors.
1768 Discovery works for :term:`namespace packages <namespace package>`.
1769 Paths are sorted before being imported so that execution order is
1770 the same even if the underlying file system's ordering is not
1771 dependent on file name.
Michael Foord80cbc9e2013-03-18 17:50:12 -07001772
Barry Warsawd78742a2014-09-08 14:21:37 -04001773 .. versionchanged:: 3.5
1774 Found packages are now checked for ``load_tests`` regardless of
1775 whether their path matches *pattern*, because it is impossible for
1776 a package name to match the default pattern.
1777
Benjamin Petersond2397752009-06-27 23:45:02 +00001778
Benjamin Peterson52baa292009-03-24 00:56:30 +00001779 The following attributes of a :class:`TestLoader` can be configured either by
1780 subclassing or assignment on an instance:
1781
1782
1783 .. attribute:: testMethodPrefix
1784
1785 String giving the prefix of method names which will be interpreted as test
1786 methods. The default value is ``'test'``.
1787
1788 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
1789 methods.
1790
1791
1792 .. attribute:: sortTestMethodsUsing
1793
1794 Function to be used to compare method names when sorting them in
1795 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
1796
1797
1798 .. attribute:: suiteClass
1799
1800 Callable object that constructs a test suite from a list of tests. No
1801 methods on the resulting object are needed. The default value is the
1802 :class:`TestSuite` class.
1803
1804 This affects all the :meth:`loadTestsFrom\*` methods.
1805
Jonas Haag5b48dc62017-11-25 16:23:52 +01001806 .. attribute:: testNamePatterns
1807
1808 List of Unix shell-style wildcard test name patterns that test methods
1809 have to match to be included in test suites (see ``-v`` option).
1810
1811 If this attribute is not ``None`` (the default), all test methods to be
1812 included in test suites must match one of the patterns in this list.
1813 Note that matches are always performed using :meth:`fnmatch.fnmatchcase`,
1814 so unlike patterns passed to the ``-v`` option, simple substring patterns
1815 will have to be converted using ``*`` wildcards.
1816
1817 This affects all the :meth:`loadTestsFrom\*` methods.
1818
1819 .. versionadded:: 3.7
1820
Benjamin Peterson52baa292009-03-24 00:56:30 +00001821
Benjamin Peterson52baa292009-03-24 00:56:30 +00001822.. class:: TestResult
1823
1824 This class is used to compile information about which tests have succeeded
1825 and which have failed.
1826
1827 A :class:`TestResult` object stores the results of a set of tests. The
1828 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
1829 properly recorded; test authors do not need to worry about recording the
1830 outcome of tests.
1831
1832 Testing frameworks built on top of :mod:`unittest` may want access to the
1833 :class:`TestResult` object generated by running a set of tests for reporting
1834 purposes; a :class:`TestResult` instance is returned by the
1835 :meth:`TestRunner.run` method for this purpose.
1836
1837 :class:`TestResult` instances have the following attributes that will be of
1838 interest when inspecting the results of running a set of tests:
1839
1840
1841 .. attribute:: errors
1842
1843 A list containing 2-tuples of :class:`TestCase` instances and strings
1844 holding formatted tracebacks. Each tuple represents a test which raised an
1845 unexpected exception.
1846
Benjamin Peterson52baa292009-03-24 00:56:30 +00001847 .. attribute:: failures
1848
1849 A list containing 2-tuples of :class:`TestCase` instances and strings
1850 holding formatted tracebacks. Each tuple represents a test where a failure
Ezio Melottie2202362013-09-07 15:19:30 +03001851 was explicitly signalled using the :meth:`TestCase.assert\*` methods.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001852
Benjamin Peterson52baa292009-03-24 00:56:30 +00001853 .. attribute:: skipped
1854
1855 A list containing 2-tuples of :class:`TestCase` instances and strings
1856 holding the reason for skipping the test.
1857
Benjamin Peterson70e32c82009-03-24 01:00:11 +00001858 .. versionadded:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +00001859
1860 .. attribute:: expectedFailures
1861
Georg Brandl6faee4e2010-09-21 14:48:28 +00001862 A list containing 2-tuples of :class:`TestCase` instances and strings
1863 holding formatted tracebacks. Each tuple represents an expected failure
Benjamin Peterson52baa292009-03-24 00:56:30 +00001864 of the test case.
1865
1866 .. attribute:: unexpectedSuccesses
1867
1868 A list containing :class:`TestCase` instances that were marked as expected
1869 failures, but succeeded.
1870
1871 .. attribute:: shouldStop
1872
1873 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1874
Benjamin Peterson52baa292009-03-24 00:56:30 +00001875 .. attribute:: testsRun
1876
1877 The total number of tests run so far.
1878
Georg Brandl12037202010-12-02 22:35:25 +00001879 .. attribute:: buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +00001880
1881 If set to true, ``sys.stdout`` and ``sys.stderr`` will be buffered in between
1882 :meth:`startTest` and :meth:`stopTest` being called. Collected output will
1883 only be echoed onto the real ``sys.stdout`` and ``sys.stderr`` if the test
1884 fails or errors. Any output is also attached to the failure / error message.
1885
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001886 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001887
Benjamin Petersonb48af542010-04-11 20:43:16 +00001888 .. attribute:: failfast
1889
1890 If set to true :meth:`stop` will be called on the first failure or error,
1891 halting the test run.
1892
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001893 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001894
Robert Collinsf0c819a2015-03-06 13:46:35 +13001895 .. attribute:: tb_locals
1896
1897 If set to true then local variables will be shown in tracebacks.
1898
1899 .. versionadded:: 3.5
Benjamin Petersonb48af542010-04-11 20:43:16 +00001900
Benjamin Peterson52baa292009-03-24 00:56:30 +00001901 .. method:: wasSuccessful()
1902
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001903 Return ``True`` if all tests run so far have passed, otherwise returns
1904 ``False``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001905
Gregory P. Smith5a6d4bf2014-01-20 01:11:18 -08001906 .. versionchanged:: 3.4
1907 Returns ``False`` if there were any :attr:`unexpectedSuccesses`
1908 from tests marked with the :func:`expectedFailure` decorator.
1909
Benjamin Peterson52baa292009-03-24 00:56:30 +00001910 .. method:: stop()
1911
1912 This method can be called to signal that the set of tests being run should
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001913 be aborted by setting the :attr:`shouldStop` attribute to ``True``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001914 :class:`TestRunner` objects should respect this flag and return without
1915 running any additional tests.
1916
1917 For example, this feature is used by the :class:`TextTestRunner` class to
1918 stop the test framework when the user signals an interrupt from the
1919 keyboard. Interactive tools which provide :class:`TestRunner`
1920 implementations can use this in a similar manner.
1921
1922 The following methods of the :class:`TestResult` class are used to maintain
1923 the internal data structures, and may be extended in subclasses to support
1924 additional reporting requirements. This is particularly useful in building
1925 tools which support interactive reporting while tests are being run.
1926
1927
1928 .. method:: startTest(test)
1929
1930 Called when the test case *test* is about to be run.
1931
Benjamin Peterson52baa292009-03-24 00:56:30 +00001932 .. method:: stopTest(test)
1933
1934 Called after the test case *test* has been executed, regardless of the
1935 outcome.
1936
Terry Jan Reedyf98021c2014-04-11 14:11:11 -04001937 .. method:: startTestRun()
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001938
1939 Called once before any tests are executed.
1940
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001941 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001942
1943
Terry Jan Reedyf98021c2014-04-11 14:11:11 -04001944 .. method:: stopTestRun()
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001945
Ezio Melotti176d6c42010-01-27 20:58:07 +00001946 Called once after all tests are executed.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001947
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001948 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001949
1950
Benjamin Peterson52baa292009-03-24 00:56:30 +00001951 .. method:: addError(test, err)
1952
Ezio Melottie64a91a2013-09-07 15:23:36 +03001953 Called when the test case *test* raises an unexpected exception. *err* is a
Benjamin Peterson52baa292009-03-24 00:56:30 +00001954 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1955 traceback)``.
1956
1957 The default implementation appends a tuple ``(test, formatted_err)`` to
1958 the instance's :attr:`errors` attribute, where *formatted_err* is a
1959 formatted traceback derived from *err*.
1960
1961
1962 .. method:: addFailure(test, err)
1963
Benjamin Petersond2397752009-06-27 23:45:02 +00001964 Called when the test case *test* signals a failure. *err* is a tuple of
1965 the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001966
1967 The default implementation appends a tuple ``(test, formatted_err)`` to
1968 the instance's :attr:`failures` attribute, where *formatted_err* is a
1969 formatted traceback derived from *err*.
1970
1971
1972 .. method:: addSuccess(test)
1973
1974 Called when the test case *test* succeeds.
1975
1976 The default implementation does nothing.
1977
1978
1979 .. method:: addSkip(test, reason)
1980
1981 Called when the test case *test* is skipped. *reason* is the reason the
1982 test gave for skipping.
1983
1984 The default implementation appends a tuple ``(test, reason)`` to the
1985 instance's :attr:`skipped` attribute.
1986
1987
1988 .. method:: addExpectedFailure(test, err)
1989
1990 Called when the test case *test* fails, but was marked with the
1991 :func:`expectedFailure` decorator.
1992
1993 The default implementation appends a tuple ``(test, formatted_err)`` to
1994 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1995 is a formatted traceback derived from *err*.
1996
1997
1998 .. method:: addUnexpectedSuccess(test)
1999
2000 Called when the test case *test* was marked with the
2001 :func:`expectedFailure` decorator, but succeeded.
2002
2003 The default implementation appends the test to the instance's
2004 :attr:`unexpectedSuccesses` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +00002005
Georg Brandl67b21b72010-08-17 15:07:14 +00002006
Antoine Pitrouc9b3ef22013-03-20 20:16:47 +01002007 .. method:: addSubTest(test, subtest, outcome)
2008
2009 Called when a subtest finishes. *test* is the test case
2010 corresponding to the test method. *subtest* is a custom
2011 :class:`TestCase` instance describing the subtest.
2012
2013 If *outcome* is :const:`None`, the subtest succeeded. Otherwise,
2014 it failed with an exception where *outcome* is a tuple of the form
2015 returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
2016
2017 The default implementation does nothing when the outcome is a
2018 success, and records subtest failures as normal failures.
2019
2020 .. versionadded:: 3.4
2021
2022
Michael Foord34c94622010-02-10 15:51:42 +00002023.. class:: TextTestResult(stream, descriptions, verbosity)
2024
Georg Brandl67b21b72010-08-17 15:07:14 +00002025 A concrete implementation of :class:`TestResult` used by the
2026 :class:`TextTestRunner`.
Michael Foord34c94622010-02-10 15:51:42 +00002027
Georg Brandl67b21b72010-08-17 15:07:14 +00002028 .. versionadded:: 3.2
2029 This class was previously named ``_TextTestResult``. The old name still
2030 exists as an alias but is deprecated.
2031
Georg Brandl116aa622007-08-15 14:28:22 +00002032
2033.. data:: defaultTestLoader
2034
2035 Instance of the :class:`TestLoader` class intended to be shared. If no
2036 customization of the :class:`TestLoader` is needed, this instance can be used
2037 instead of repeatedly creating new instances.
2038
2039
Ezio Melotti9c939bc2013-05-07 09:46:30 +03002040.. class:: TextTestRunner(stream=None, descriptions=True, verbosity=1, failfast=False, \
Robert Collinsf0c819a2015-03-06 13:46:35 +13002041 buffer=False, resultclass=None, warnings=None, *, tb_locals=False)
Georg Brandl116aa622007-08-15 14:28:22 +00002042
Michael Foordd218e952011-01-03 12:55:11 +00002043 A basic test runner implementation that outputs results to a stream. If *stream*
Éric Araujo941afed2011-09-01 02:47:34 +02002044 is ``None``, the default, :data:`sys.stderr` is used as the output stream. This class
Georg Brandl116aa622007-08-15 14:28:22 +00002045 has a few configurable parameters, but is essentially very simple. Graphical
Robert Collinsf0c819a2015-03-06 13:46:35 +13002046 applications which run test suites should provide alternate implementations. Such
2047 implementations should accept ``**kwargs`` as the interface to construct runners
2048 changes when features are added to unittest.
Georg Brandl116aa622007-08-15 14:28:22 +00002049
Ezio Melotti60901872010-12-01 00:56:10 +00002050 By default this runner shows :exc:`DeprecationWarning`,
Senthil Kumaran409ea5d2014-02-08 14:28:03 -08002051 :exc:`PendingDeprecationWarning`, :exc:`ResourceWarning` and
Larry Hastingsad88d7a2014-02-10 04:26:10 -08002052 :exc:`ImportWarning` even if they are :ref:`ignored by default
2053 <warning-ignored>`. Deprecation warnings caused by :ref:`deprecated unittest
2054 methods <deprecated-aliases>` are also special-cased and, when the warning
2055 filters are ``'default'`` or ``'always'``, they will appear only once
2056 per-module, in order to avoid too many warning messages. This behavior can
Martin Panterb8c5f542016-10-30 04:20:23 +00002057 be overridden using Python's :option:`!-Wd` or :option:`!-Wa` options
2058 (see :ref:`Warning control <using-on-warnings>`) and leaving
Larry Hastingsad88d7a2014-02-10 04:26:10 -08002059 *warnings* to ``None``.
Ezio Melotti60901872010-12-01 00:56:10 +00002060
Michael Foordd218e952011-01-03 12:55:11 +00002061 .. versionchanged:: 3.2
2062 Added the ``warnings`` argument.
2063
2064 .. versionchanged:: 3.2
Éric Araujo941afed2011-09-01 02:47:34 +02002065 The default stream is set to :data:`sys.stderr` at instantiation time rather
Michael Foordd218e952011-01-03 12:55:11 +00002066 than import time.
2067
Robert Collinsf0c819a2015-03-06 13:46:35 +13002068 .. versionchanged:: 3.5
2069 Added the tb_locals parameter.
2070
Benjamin Peterson25c95f12009-05-08 20:42:26 +00002071 .. method:: _makeResult()
Georg Brandl116aa622007-08-15 14:28:22 +00002072
Benjamin Peterson25c95f12009-05-08 20:42:26 +00002073 This method returns the instance of ``TestResult`` used by :meth:`run`.
2074 It is not intended to be called directly, but can be overridden in
2075 subclasses to provide a custom ``TestResult``.
2076
Michael Foord34c94622010-02-10 15:51:42 +00002077 ``_makeResult()`` instantiates the class or callable passed in the
2078 ``TextTestRunner`` constructor as the ``resultclass`` argument. It
Benjamin Petersonb48af542010-04-11 20:43:16 +00002079 defaults to :class:`TextTestResult` if no ``resultclass`` is provided.
Michael Foord34c94622010-02-10 15:51:42 +00002080 The result class is instantiated with the following arguments::
2081
2082 stream, descriptions, verbosity
Benjamin Peterson25c95f12009-05-08 20:42:26 +00002083
Michael Foord4d1639f2013-12-29 23:38:55 +00002084 .. method:: run(test)
2085
Julien Palard6e240dd2019-02-22 09:22:27 +01002086 This method is the main public interface to the ``TextTestRunner``. This
Michael Foord4d1639f2013-12-29 23:38:55 +00002087 method takes a :class:`TestSuite` or :class:`TestCase` instance. A
2088 :class:`TestResult` is created by calling
2089 :func:`_makeResult` and the test(s) are run and the
2090 results printed to stdout.
2091
Ezio Melotti60901872010-12-01 00:56:10 +00002092
Georg Brandl419e3de2010-12-01 15:44:25 +00002093.. function:: main(module='__main__', defaultTest=None, argv=None, testRunner=None, \
Ezio Melotti40dcb1d2011-03-10 13:46:50 +02002094 testLoader=unittest.defaultTestLoader, exit=True, verbosity=1, \
Georg Brandl419e3de2010-12-01 15:44:25 +00002095 failfast=None, catchbreak=None, buffer=None, warnings=None)
Georg Brandl116aa622007-08-15 14:28:22 +00002096
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03002097 A command-line program that loads a set of tests from *module* and runs them;
2098 this is primarily for making test modules conveniently executable.
2099 The simplest use for this function is to include the following line at the
2100 end of a test script::
Georg Brandl116aa622007-08-15 14:28:22 +00002101
2102 if __name__ == '__main__':
2103 unittest.main()
2104
Benjamin Petersond2397752009-06-27 23:45:02 +00002105 You can run tests with more detailed information by passing in the verbosity
2106 argument::
2107
2108 if __name__ == '__main__':
2109 unittest.main(verbosity=2)
2110
R David Murray6e731b02014-01-02 13:43:02 -05002111 The *defaultTest* argument is either the name of a single test or an
2112 iterable of test names to run if no test names are specified via *argv*. If
2113 not specified or ``None`` and no test names are provided via *argv*, all
2114 tests found in *module* are run.
R David Murray12e930f2014-01-02 13:37:26 -05002115
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03002116 The *argv* argument can be a list of options passed to the program, with the
2117 first element being the program name. If not specified or ``None``,
2118 the values of :data:`sys.argv` are used.
2119
Georg Brandl116aa622007-08-15 14:28:22 +00002120 The *testRunner* argument can either be a test runner class or an already
Benjamin Peterson25c95f12009-05-08 20:42:26 +00002121 created instance of it. By default ``main`` calls :func:`sys.exit` with
2122 an exit code indicating success or failure of the tests run.
2123
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03002124 The *testLoader* argument has to be a :class:`TestLoader` instance,
2125 and defaults to :data:`defaultTestLoader`.
2126
Benjamin Peterson25c95f12009-05-08 20:42:26 +00002127 ``main`` supports being used from the interactive interpreter by passing in the
2128 argument ``exit=False``. This displays the result on standard output without
2129 calling :func:`sys.exit`::
2130
2131 >>> from unittest import main
2132 >>> main(module='test_module', exit=False)
2133
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03002134 The *failfast*, *catchbreak* and *buffer* parameters have the same
Éric Araujo76338ec2010-11-26 23:46:18 +00002135 effect as the same-name `command-line options`_.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002136
Zachary Waref0a71cf2016-08-30 00:16:13 -05002137 The *warnings* argument specifies the :ref:`warning filter <warning-filter>`
Ezio Melotti60901872010-12-01 00:56:10 +00002138 that should be used while running the tests. If it's not specified, it will
Martin Panterb8c5f542016-10-30 04:20:23 +00002139 remain ``None`` if a :option:`!-W` option is passed to :program:`python`
2140 (see :ref:`Warning control <using-on-warnings>`),
Ezio Melotti60901872010-12-01 00:56:10 +00002141 otherwise it will be set to ``'default'``.
2142
Benjamin Peterson25c95f12009-05-08 20:42:26 +00002143 Calling ``main`` actually returns an instance of the ``TestProgram`` class.
2144 This stores the result of the tests run as the ``result`` attribute.
2145
Éric Araujo971dc012010-12-16 03:13:05 +00002146 .. versionchanged:: 3.1
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03002147 The *exit* parameter was added.
Éric Araujo971dc012010-12-16 03:13:05 +00002148
Georg Brandl853947a2010-01-31 18:53:23 +00002149 .. versionchanged:: 3.2
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03002150 The *verbosity*, *failfast*, *catchbreak*, *buffer*
2151 and *warnings* parameters were added.
Benjamin Petersond2397752009-06-27 23:45:02 +00002152
Chris Jerdonekccbc26a2013-02-23 15:44:46 -08002153 .. versionchanged:: 3.4
2154 The *defaultTest* parameter was changed to also accept an iterable of
2155 test names.
2156
Benjamin Petersond2397752009-06-27 23:45:02 +00002157
2158load_tests Protocol
2159###################
2160
Georg Brandl853947a2010-01-31 18:53:23 +00002161.. versionadded:: 3.2
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00002162
Benjamin Petersond2397752009-06-27 23:45:02 +00002163Modules or packages can customize how tests are loaded from them during normal
2164test runs or test discovery by implementing a function called ``load_tests``.
2165
2166If a test module defines ``load_tests`` it will be called by
2167:meth:`TestLoader.loadTestsFromModule` with the following arguments::
2168
Barry Warsawd78742a2014-09-08 14:21:37 -04002169 load_tests(loader, standard_tests, pattern)
2170
2171where *pattern* is passed straight through from ``loadTestsFromModule``. It
2172defaults to ``None``.
Benjamin Petersond2397752009-06-27 23:45:02 +00002173
2174It should return a :class:`TestSuite`.
2175
2176*loader* is the instance of :class:`TestLoader` doing the loading.
2177*standard_tests* are the tests that would be loaded by default from the
2178module. It is common for test modules to only want to add or remove tests
2179from the standard set of tests.
2180The third argument is used when loading packages as part of test discovery.
2181
2182A typical ``load_tests`` function that loads tests from a specific set of
2183:class:`TestCase` classes may look like::
2184
2185 test_cases = (TestCase1, TestCase2, TestCase3)
2186
2187 def load_tests(loader, tests, pattern):
2188 suite = TestSuite()
2189 for test_class in test_cases:
2190 tests = loader.loadTestsFromTestCase(test_class)
2191 suite.addTests(tests)
2192 return suite
2193
Barry Warsawd78742a2014-09-08 14:21:37 -04002194If discovery is started in a directory containing a package, either from the
2195command line or by calling :meth:`TestLoader.discover`, then the package
2196:file:`__init__.py` will be checked for ``load_tests``. If that function does
2197not exist, discovery will recurse into the package as though it were just
2198another directory. Otherwise, discovery of the package's tests will be left up
2199to ``load_tests`` which is called with the following arguments::
Benjamin Petersond2397752009-06-27 23:45:02 +00002200
2201 load_tests(loader, standard_tests, pattern)
2202
2203This should return a :class:`TestSuite` representing all the tests
2204from the package. (``standard_tests`` will only contain tests
2205collected from :file:`__init__.py`.)
2206
2207Because the pattern is passed into ``load_tests`` the package is free to
2208continue (and potentially modify) test discovery. A 'do nothing'
2209``load_tests`` function for a test package would look like::
2210
2211 def load_tests(loader, standard_tests, pattern):
2212 # top level directory cached on loader instance
2213 this_dir = os.path.dirname(__file__)
2214 package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
2215 standard_tests.addTests(package_tests)
2216 return standard_tests
Benjamin Petersonb48af542010-04-11 20:43:16 +00002217
Barry Warsawd78742a2014-09-08 14:21:37 -04002218.. versionchanged:: 3.5
2219 Discovery no longer checks package names for matching *pattern* due to the
2220 impossibility of package names matching the default pattern.
2221
2222
Benjamin Petersonb48af542010-04-11 20:43:16 +00002223
2224Class and Module Fixtures
2225-------------------------
2226
2227Class and module level fixtures are implemented in :class:`TestSuite`. When
2228the test suite encounters a test from a new class then :meth:`tearDownClass`
2229from the previous class (if there is one) is called, followed by
2230:meth:`setUpClass` from the new class.
2231
2232Similarly if a test is from a different module from the previous test then
2233``tearDownModule`` from the previous module is run, followed by
2234``setUpModule`` from the new module.
2235
2236After all the tests have run the final ``tearDownClass`` and
2237``tearDownModule`` are run.
2238
2239Note that shared fixtures do not play well with [potential] features like test
2240parallelization and they break test isolation. They should be used with care.
2241
2242The default ordering of tests created by the unittest test loaders is to group
2243all tests from the same modules and classes together. This will lead to
2244``setUpClass`` / ``setUpModule`` (etc) being called exactly once per class and
2245module. If you randomize the order, so that tests from different modules and
2246classes are adjacent to each other, then these shared fixture functions may be
2247called multiple times in a single test run.
2248
2249Shared fixtures are not intended to work with suites with non-standard
2250ordering. A ``BaseTestSuite`` still exists for frameworks that don't want to
2251support shared fixtures.
2252
2253If there are any exceptions raised during one of the shared fixture functions
2254the test is reported as an error. Because there is no corresponding test
2255instance an ``_ErrorHolder`` object (that has the same interface as a
2256:class:`TestCase`) is created to represent the error. If you are just using
2257the standard unittest test runner then this detail doesn't matter, but if you
2258are a framework author it may be relevant.
2259
2260
2261setUpClass and tearDownClass
2262~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2263
2264These must be implemented as class methods::
2265
2266 import unittest
2267
2268 class Test(unittest.TestCase):
2269 @classmethod
2270 def setUpClass(cls):
2271 cls._connection = createExpensiveConnectionObject()
2272
2273 @classmethod
2274 def tearDownClass(cls):
2275 cls._connection.destroy()
2276
2277If you want the ``setUpClass`` and ``tearDownClass`` on base classes called
2278then you must call up to them yourself. The implementations in
2279:class:`TestCase` are empty.
2280
2281If an exception is raised during a ``setUpClass`` then the tests in the class
2282are not run and the ``tearDownClass`` is not run. Skipped classes will not
Michael Foord98b3e762010-06-05 21:59:55 +00002283have ``setUpClass`` or ``tearDownClass`` run. If the exception is a
Ezio Melotti265281a2013-03-27 20:11:55 +02002284:exc:`SkipTest` exception then the class will be reported as having been skipped
Michael Foord98b3e762010-06-05 21:59:55 +00002285instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002286
2287
2288setUpModule and tearDownModule
2289~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2290
2291These should be implemented as functions::
2292
2293 def setUpModule():
2294 createConnection()
2295
2296 def tearDownModule():
2297 closeConnection()
2298
2299If an exception is raised in a ``setUpModule`` then none of the tests in the
Michael Foord98b3e762010-06-05 21:59:55 +00002300module will be run and the ``tearDownModule`` will not be run. If the exception is a
Ezio Melotti265281a2013-03-27 20:11:55 +02002301:exc:`SkipTest` exception then the module will be reported as having been skipped
Michael Foord98b3e762010-06-05 21:59:55 +00002302instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002303
Lisa Roach0f221d02018-11-08 18:34:33 -08002304To add cleanup code that must be run even in the case of an exception, use
2305``addModuleCleanup``:
2306
2307
2308.. function:: addModuleCleanup(function, *args, **kwargs)
2309
2310 Add a function to be called after :func:`tearDownModule` to cleanup
2311 resources used during the test class. Functions will be called in reverse
2312 order to the order they are added (:abbr:`LIFO (last-in, first-out)`).
2313 They are called with any arguments and keyword arguments passed into
2314 :meth:`addModuleCleanup` when they are added.
2315
2316 If :meth:`setUpModule` fails, meaning that :func:`tearDownModule` is not
2317 called, then any cleanup functions added will still be called.
2318
2319 .. versionadded:: 3.8
2320
2321
2322.. function:: doModuleCleanups()
2323
2324 This function is called unconditionally after :func:`tearDownModule`, or
2325 after :func:`setUpModule` if :func:`setUpModule` raises an exception.
2326
2327 It is responsible for calling all the cleanup functions added by
2328 :func:`addCleanupModule`. If you need cleanup functions to be called
2329 *prior* to :func:`tearDownModule` then you can call
2330 :func:`doModuleCleanups` yourself.
2331
2332 :func:`doModuleCleanups` pops methods off the stack of cleanup
2333 functions one at a time, so it can be called at any time.
2334
2335 .. versionadded:: 3.8
Benjamin Petersonb48af542010-04-11 20:43:16 +00002336
2337Signal Handling
2338---------------
2339
Georg Brandl419e3de2010-12-01 15:44:25 +00002340.. versionadded:: 3.2
2341
Éric Araujo8acb67c2010-11-26 23:31:07 +00002342The :option:`-c/--catch <unittest -c>` command-line option to unittest,
Éric Araujo76338ec2010-11-26 23:46:18 +00002343along with the ``catchbreak`` parameter to :func:`unittest.main()`, provide
2344more friendly handling of control-C during a test run. With catch break
2345behavior enabled control-C will allow the currently running test to complete,
2346and the test run will then end and report all the results so far. A second
2347control-c will raise a :exc:`KeyboardInterrupt` in the usual way.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002348
Michael Foordde4ceab2010-04-25 19:53:49 +00002349The control-c handling signal handler attempts to remain compatible with code or
2350tests that install their own :const:`signal.SIGINT` handler. If the ``unittest``
2351handler is called but *isn't* the installed :const:`signal.SIGINT` handler,
2352i.e. it has been replaced by the system under test and delegated to, then it
2353calls the default handler. This will normally be the expected behavior by code
2354that replaces an installed handler and delegates to it. For individual tests
2355that need ``unittest`` control-c handling disabled the :func:`removeHandler`
2356decorator can be used.
2357
2358There are a few utility functions for framework authors to enable control-c
2359handling functionality within test frameworks.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002360
2361.. function:: installHandler()
2362
2363 Install the control-c handler. When a :const:`signal.SIGINT` is received
2364 (usually in response to the user pressing control-c) all registered results
2365 have :meth:`~TestResult.stop` called.
2366
Michael Foord469b1f02010-04-26 23:41:26 +00002367
Benjamin Petersonb48af542010-04-11 20:43:16 +00002368.. function:: registerResult(result)
2369
2370 Register a :class:`TestResult` object for control-c handling. Registering a
2371 result stores a weak reference to it, so it doesn't prevent the result from
2372 being garbage collected.
2373
Michael Foordde4ceab2010-04-25 19:53:49 +00002374 Registering a :class:`TestResult` object has no side-effects if control-c
2375 handling is not enabled, so test frameworks can unconditionally register
2376 all results they create independently of whether or not handling is enabled.
2377
Michael Foord469b1f02010-04-26 23:41:26 +00002378
Benjamin Petersonb48af542010-04-11 20:43:16 +00002379.. function:: removeResult(result)
2380
2381 Remove a registered result. Once a result has been removed then
2382 :meth:`~TestResult.stop` will no longer be called on that result object in
2383 response to a control-c.
2384
Michael Foord469b1f02010-04-26 23:41:26 +00002385
Michael Foordde4ceab2010-04-25 19:53:49 +00002386.. function:: removeHandler(function=None)
2387
2388 When called without arguments this function removes the control-c handler
2389 if it has been installed. This function can also be used as a test decorator
Mariatta98f42aa2018-02-23 09:51:11 -08002390 to temporarily remove the handler while the test is being executed::
Michael Foordde4ceab2010-04-25 19:53:49 +00002391
2392 @unittest.removeHandler
2393 def test_signal_handling(self):
2394 ...