blob: 9647b4b665367aad6978d99920f6f4e422c879db [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.
6.. moduleauthor:: Steve Purcell <stephen_purcell@yahoo.com>
7.. sectionauthor:: Steve Purcell <stephen_purcell@yahoo.com>
8.. sectionauthor:: Fred L. Drake, Jr. <fdrake@acm.org>
9.. sectionauthor:: Raymond Hettinger <python@rcn.com>
10
Ezio Melotti22170ed2010-11-20 09:57:27 +000011(If you are already familiar with the basic concepts of testing, you might want
12to skip to :ref:`the list of assert methods <assert-methods>`.)
Georg Brandl116aa622007-08-15 14:28:22 +000013
Antoine Pitrou2c5e9502013-01-20 01:29:39 +010014The :mod:`unittest` unit testing framework was originally inspired by JUnit
15and has a similar flavor as major unit testing frameworks in other
16languages. It supports test automation, sharing of setup and shutdown code
17for tests, aggregation of tests into collections, and independence of the
18tests from the reporting framework.
Georg Brandl116aa622007-08-15 14:28:22 +000019
Antoine Pitrou2c5e9502013-01-20 01:29:39 +010020To achieve this, :mod:`unittest` supports some important concepts in an
21object-oriented way:
Georg Brandl116aa622007-08-15 14:28:22 +000022
23test fixture
24 A :dfn:`test fixture` represents the preparation needed to perform one or more
25 tests, and any associate cleanup actions. This may involve, for example,
26 creating temporary or proxy databases, directories, or starting a server
27 process.
28
29test case
Antoine Pitrou2c5e9502013-01-20 01:29:39 +010030 A :dfn:`test case` is the individual unit of testing. It checks for a specific
Georg Brandl116aa622007-08-15 14:28:22 +000031 response to a particular set of inputs. :mod:`unittest` provides a base class,
32 :class:`TestCase`, which may be used to create new test cases.
33
34test suite
35 A :dfn:`test suite` is a collection of test cases, test suites, or both. It is
36 used to aggregate tests that should be executed together.
37
38test runner
39 A :dfn:`test runner` is a component which orchestrates the execution of tests
40 and provides the outcome to the user. The runner may use a graphical interface,
41 a textual interface, or return a special value to indicate the results of
42 executing the tests.
43
Georg Brandl116aa622007-08-15 14:28:22 +000044
45.. seealso::
46
47 Module :mod:`doctest`
48 Another test-support module with a very different flavor.
49
R David Murraya1005ed2015-07-04 15:44:14 -040050 `Simple Smalltalk Testing: With Patterns <https://web.archive.org/web/20150315073817/http://www.xprogramming.com/testfram.htm>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000051 Kent Beck's original paper on testing frameworks using the pattern shared
52 by :mod:`unittest`.
Georg Brandl116aa622007-08-15 14:28:22 +000053
Berker Peksaga1a14092014-12-28 18:48:33 +020054 `Nose <https://nose.readthedocs.org/en/latest/>`_ and `py.test <http://pytest.org>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000055 Third-party unittest frameworks with a lighter-weight syntax for writing
56 tests. For example, ``assert func(10) == 42``.
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000057
Georg Brandle73778c2014-10-29 08:36:35 +010058 `The Python Testing Tools Taxonomy <https://wiki.python.org/moin/PythonTestingToolsTaxonomy>`_
Benjamin Petersonb48af542010-04-11 20:43:16 +000059 An extensive list of Python testing tools including functional testing
60 frameworks and mock object libraries.
Benjamin Petersond2397752009-06-27 23:45:02 +000061
Benjamin Petersonb48af542010-04-11 20:43:16 +000062 `Testing in Python Mailing List <http://lists.idyll.org/listinfo/testing-in-python>`_
63 A special-interest-group for discussion of testing, and testing tools,
64 in Python.
Benjamin Petersond2397752009-06-27 23:45:02 +000065
Michael Foord90efac72011-01-03 15:39:49 +000066 The script :file:`Tools/unittestgui/unittestgui.py` in the Python source distribution is
67 a GUI tool for test discovery and execution. This is intended largely for ease of use
Senthil Kumaran847c33c2012-10-27 11:04:55 -070068 for those new to unit testing. For production environments it is
69 recommended that tests be driven by a continuous integration system such as
Georg Brandl525d3552014-10-29 10:26:56 +010070 `Buildbot <http://buildbot.net/>`_, `Jenkins <http://jenkins-ci.org/>`_
Senthil Kumaran847c33c2012-10-27 11:04:55 -070071 or `Hudson <http://hudson-ci.org/>`_.
Michael Foord90efac72011-01-03 15:39:49 +000072
73
Georg Brandl116aa622007-08-15 14:28:22 +000074.. _unittest-minimal-example:
75
76Basic example
77-------------
78
79The :mod:`unittest` module provides a rich set of tools for constructing and
80running tests. This section demonstrates that a small subset of the tools
81suffice to meet the needs of most users.
82
Ezio Melotti2e3998f2015-03-24 12:42:41 +020083Here is a short script to test three string methods::
Georg Brandl116aa622007-08-15 14:28:22 +000084
Ezio Melotti2e3998f2015-03-24 12:42:41 +020085 import unittest
Georg Brandl116aa622007-08-15 14:28:22 +000086
Ezio Melotti2e3998f2015-03-24 12:42:41 +020087 class TestStringMethods(unittest.TestCase):
Georg Brandl116aa622007-08-15 14:28:22 +000088
Ezio Melotti2e3998f2015-03-24 12:42:41 +020089 def test_upper(self):
90 self.assertEqual('foo'.upper(), 'FOO')
Georg Brandl116aa622007-08-15 14:28:22 +000091
Ezio Melotti2e3998f2015-03-24 12:42:41 +020092 def test_isupper(self):
93 self.assertTrue('FOO'.isupper())
94 self.assertFalse('Foo'.isupper())
Georg Brandl116aa622007-08-15 14:28:22 +000095
Ezio Melotti2e3998f2015-03-24 12:42:41 +020096 def test_split(self):
97 s = 'hello world'
98 self.assertEqual(s.split(), ['hello', 'world'])
99 # check that s.split fails when the separator is not a string
100 with self.assertRaises(TypeError):
101 s.split(2)
Benjamin Peterson847a4112010-03-14 15:04:17 +0000102
Ezio Melotti2e3998f2015-03-24 12:42:41 +0200103 if __name__ == '__main__':
104 unittest.main()
Georg Brandl116aa622007-08-15 14:28:22 +0000105
Georg Brandl116aa622007-08-15 14:28:22 +0000106
Benjamin Peterson52baa292009-03-24 00:56:30 +0000107A testcase is created by subclassing :class:`unittest.TestCase`. The three
Georg Brandl116aa622007-08-15 14:28:22 +0000108individual tests are defined with methods whose names start with the letters
109``test``. This naming convention informs the test runner about which methods
110represent tests.
111
Benjamin Peterson52baa292009-03-24 00:56:30 +0000112The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
Ezio Melotti2e3998f2015-03-24 12:42:41 +0200113expected result; :meth:`~TestCase.assertTrue` or :meth:`~TestCase.assertFalse`
114to verify a condition; or :meth:`~TestCase.assertRaises` to verify that a
115specific exception gets raised. These methods are used instead of the
116:keyword:`assert` statement so the test runner can accumulate all test results
117and produce a report.
Georg Brandl116aa622007-08-15 14:28:22 +0000118
Ezio Melotti2e3998f2015-03-24 12:42:41 +0200119The :meth:`~TestCase.setUp` and :meth:`~TestCase.tearDown` methods allow you
120to define instructions that will be executed before and after each test method.
Benjamin Peterson8a6ddb92016-01-07 22:01:26 -0800121They are covered in more detail in the section :ref:`organizing-tests`.
Georg Brandl116aa622007-08-15 14:28:22 +0000122
123The final block shows a simple way to run the tests. :func:`unittest.main`
Éric Araujo713d3032010-11-18 16:38:46 +0000124provides a command-line interface to the test script. When run from the command
Georg Brandl116aa622007-08-15 14:28:22 +0000125line, the above script produces an output that looks like this::
126
127 ...
128 ----------------------------------------------------------------------
129 Ran 3 tests in 0.000s
130
131 OK
132
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100133Passing the ``-v`` option to your test script will instruct :func:`unittest.main`
134to enable a higher level of verbosity, and produce the following output::
Georg Brandl116aa622007-08-15 14:28:22 +0000135
Ezio Melotti2e3998f2015-03-24 12:42:41 +0200136 test_isupper (__main__.TestStringMethods) ... ok
137 test_split (__main__.TestStringMethods) ... ok
138 test_upper (__main__.TestStringMethods) ... ok
Georg Brandl116aa622007-08-15 14:28:22 +0000139
140 ----------------------------------------------------------------------
Ezio Melotti2e3998f2015-03-24 12:42:41 +0200141 Ran 3 tests in 0.001s
Georg Brandl116aa622007-08-15 14:28:22 +0000142
143 OK
144
145The above examples show the most commonly used :mod:`unittest` features which
146are sufficient to meet many everyday testing needs. The remainder of the
147documentation explores the full feature set from first principles.
148
Benjamin Petersonb48af542010-04-11 20:43:16 +0000149
150.. _unittest-command-line-interface:
151
Éric Araujo76338ec2010-11-26 23:46:18 +0000152Command-Line Interface
Benjamin Petersonb48af542010-04-11 20:43:16 +0000153----------------------
154
155The unittest module can be used from the command line to run tests from
156modules, classes or even individual test methods::
157
158 python -m unittest test_module1 test_module2
159 python -m unittest test_module.TestClass
160 python -m unittest test_module.TestClass.test_method
161
162You can pass in a list with any combination of module names, and fully
163qualified class or method names.
164
Michael Foord37d120a2010-12-04 01:11:21 +0000165Test modules can be specified by file path as well::
166
167 python -m unittest tests/test_something.py
168
169This allows you to use the shell filename completion to specify the test module.
170The file specified must still be importable as a module. The path is converted
171to a module name by removing the '.py' and converting path separators into '.'.
172If you want to execute a test file that isn't importable as a module you should
173execute the file directly instead.
174
Benjamin Petersonb48af542010-04-11 20:43:16 +0000175You can run tests with more detail (higher verbosity) by passing in the -v flag::
176
177 python -m unittest -v test_module
178
Michael Foord086f3082010-11-21 21:28:01 +0000179When executed without arguments :ref:`unittest-test-discovery` is started::
180
181 python -m unittest
182
Éric Araujo713d3032010-11-18 16:38:46 +0000183For a list of all the command-line options::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000184
185 python -m unittest -h
186
Georg Brandl67b21b72010-08-17 15:07:14 +0000187.. versionchanged:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000188 In earlier versions it was only possible to run individual test methods and
189 not modules or classes.
190
191
Éric Araujo76338ec2010-11-26 23:46:18 +0000192Command-line options
193~~~~~~~~~~~~~~~~~~~~
Benjamin Petersonb48af542010-04-11 20:43:16 +0000194
Éric Araujod3309df2010-11-21 03:09:17 +0000195:program:`unittest` supports these command-line options:
Benjamin Petersonb48af542010-04-11 20:43:16 +0000196
Éric Araujo713d3032010-11-18 16:38:46 +0000197.. program:: unittest
Benjamin Petersonb48af542010-04-11 20:43:16 +0000198
Éric Araujo713d3032010-11-18 16:38:46 +0000199.. cmdoption:: -b, --buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +0000200
Éric Araujo713d3032010-11-18 16:38:46 +0000201 The standard output and standard error streams are buffered during the test
202 run. Output during a passing test is discarded. Output is echoed normally
203 on test fail or error and is added to the failure messages.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000204
Éric Araujo713d3032010-11-18 16:38:46 +0000205.. cmdoption:: -c, --catch
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000206
Serhiy Storchaka0424eaf2015-09-12 17:45:25 +0300207 :kbd:`Control-C` during the test run waits for the current test to end and then
208 reports all the results so far. A second :kbd:`Control-C` raises the normal
Éric Araujo713d3032010-11-18 16:38:46 +0000209 :exc:`KeyboardInterrupt` exception.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000210
Éric Araujo713d3032010-11-18 16:38:46 +0000211 See `Signal Handling`_ for the functions that provide this functionality.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000212
Éric Araujo713d3032010-11-18 16:38:46 +0000213.. cmdoption:: -f, --failfast
214
215 Stop the test run on the first error or failure.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000216
Robert Collinsf0c819a2015-03-06 13:46:35 +1300217.. cmdoption:: --locals
218
219 Show local variables in tracebacks.
220
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000221.. versionadded:: 3.2
Éric Araujod6c5f742010-12-16 00:07:01 +0000222 The command-line options ``-b``, ``-c`` and ``-f`` were added.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000223
Robert Collinsf0c819a2015-03-06 13:46:35 +1300224.. versionadded:: 3.5
225 The command-line option ``--locals``.
226
Benjamin Petersonb48af542010-04-11 20:43:16 +0000227The command line can also be used for test discovery, for running all of the
228tests in a project or just a subset.
229
230
231.. _unittest-test-discovery:
232
233Test Discovery
234--------------
235
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000236.. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000237
Ezio Melotti3d995842011-03-08 16:17:35 +0200238Unittest supports simple test discovery. In order to be compatible with test
239discovery, all of the test files must be :ref:`modules <tut-modules>` or
Larry Hastings3732ed22014-03-15 21:13:56 -0700240:ref:`packages <tut-packages>` (including :term:`namespace packages
241<namespace package>`) importable from the top-level directory of
242the project (this means that their filenames must be valid :ref:`identifiers
243<identifiers>`).
Benjamin Petersonb48af542010-04-11 20:43:16 +0000244
245Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
Éric Araujo713d3032010-11-18 16:38:46 +0000246used from the command line. The basic command-line usage is::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000247
248 cd project_directory
249 python -m unittest discover
250
Michael Foord086f3082010-11-21 21:28:01 +0000251.. note::
252
253 As a shortcut, ``python -m unittest`` is the equivalent of
254 ``python -m unittest discover``. If you want to pass arguments to test
Éric Araujo941afed2011-09-01 02:47:34 +0200255 discovery the ``discover`` sub-command must be used explicitly.
Michael Foord086f3082010-11-21 21:28:01 +0000256
Benjamin Petersonb48af542010-04-11 20:43:16 +0000257The ``discover`` sub-command has the following options:
258
Éric Araujo713d3032010-11-18 16:38:46 +0000259.. program:: unittest discover
260
261.. cmdoption:: -v, --verbose
262
263 Verbose output
264
Chris Jerdonekd69ad552013-02-21 18:54:43 -0800265.. cmdoption:: -s, --start-directory directory
Éric Araujo713d3032010-11-18 16:38:46 +0000266
Éric Araujo941afed2011-09-01 02:47:34 +0200267 Directory to start discovery (``.`` default)
Éric Araujo713d3032010-11-18 16:38:46 +0000268
Chris Jerdonekd69ad552013-02-21 18:54:43 -0800269.. cmdoption:: -p, --pattern pattern
Éric Araujo713d3032010-11-18 16:38:46 +0000270
Éric Araujo941afed2011-09-01 02:47:34 +0200271 Pattern to match test files (``test*.py`` default)
Éric Araujo713d3032010-11-18 16:38:46 +0000272
Chris Jerdonekd69ad552013-02-21 18:54:43 -0800273.. cmdoption:: -t, --top-level-directory directory
Éric Araujo713d3032010-11-18 16:38:46 +0000274
275 Top level directory of project (defaults to start directory)
Benjamin Petersonb48af542010-04-11 20:43:16 +0000276
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000277The :option:`-s`, :option:`-p`, and :option:`-t` options can be passed in
278as positional arguments in that order. The following two command lines
279are equivalent::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000280
Robert Collinsa2b00552015-08-24 12:14:28 +1200281 python -m unittest discover -s project_directory -p "*_test.py"
282 python -m unittest discover project_directory "*_test.py"
Benjamin Petersonb48af542010-04-11 20:43:16 +0000283
Michael Foord16f3e902010-05-08 15:13:42 +0000284As well as being a path it is possible to pass a package name, for example
285``myproject.subpackage.test``, as the start directory. The package name you
286supply will then be imported and its location on the filesystem will be used
287as the start directory.
288
289.. caution::
290
Senthil Kumaran916bd382010-10-15 12:55:19 +0000291 Test discovery loads tests by importing them. Once test discovery has found
292 all the test files from the start directory you specify it turns the paths
293 into package names to import. For example :file:`foo/bar/baz.py` will be
Michael Foord16f3e902010-05-08 15:13:42 +0000294 imported as ``foo.bar.baz``.
295
296 If you have a package installed globally and attempt test discovery on
297 a different copy of the package then the import *could* happen from the
298 wrong place. If this happens test discovery will warn you and exit.
299
300 If you supply the start directory as a package name rather than a
301 path to a directory then discover assumes that whichever location it
302 imports from is the location you intended, so you will not get the
303 warning.
304
Benjamin Petersonb48af542010-04-11 20:43:16 +0000305Test modules and packages can customize test loading and discovery by through
306the `load_tests protocol`_.
307
Larry Hastings3732ed22014-03-15 21:13:56 -0700308.. versionchanged:: 3.4
309 Test discovery supports :term:`namespace packages <namespace package>`.
310
Benjamin Petersonb48af542010-04-11 20:43:16 +0000311
Georg Brandl116aa622007-08-15 14:28:22 +0000312.. _organizing-tests:
313
314Organizing test code
315--------------------
316
317The basic building blocks of unit testing are :dfn:`test cases` --- single
318scenarios that must be set up and checked for correctness. In :mod:`unittest`,
Raymond Hettinger833ad0e2011-02-06 21:00:38 +0000319test cases are represented by :class:`unittest.TestCase` instances.
320To make your own test cases you must write subclasses of
321:class:`TestCase` or use :class:`FunctionTestCase`.
Georg Brandl116aa622007-08-15 14:28:22 +0000322
Georg Brandl116aa622007-08-15 14:28:22 +0000323The testing code of a :class:`TestCase` instance should be entirely self
324contained, such that it can be run either in isolation or in arbitrary
325combination with any number of other test cases.
326
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100327The simplest :class:`TestCase` subclass will simply implement a test method
328(i.e. a method whose name starts with ``test``) in order to perform specific
329testing code::
Georg Brandl116aa622007-08-15 14:28:22 +0000330
331 import unittest
332
333 class DefaultWidgetSizeTestCase(unittest.TestCase):
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100334 def test_default_widget_size(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000335 widget = Widget('The widget')
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100336 self.assertEqual(widget.size(), (50, 50))
Georg Brandl116aa622007-08-15 14:28:22 +0000337
Sandro Tosi41b24042012-01-21 10:59:37 +0100338Note that in order to test something, we use one of the :meth:`assert\*`
Benjamin Petersond2397752009-06-27 23:45:02 +0000339methods provided by the :class:`TestCase` base class. If the test fails, an
340exception will be raised, and :mod:`unittest` will identify the test case as a
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100341:dfn:`failure`. Any other exceptions will be treated as :dfn:`errors`.
Georg Brandl116aa622007-08-15 14:28:22 +0000342
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100343Tests can be numerous, and their set-up can be repetitive. Luckily, we
344can factor out set-up code by implementing a method called
345:meth:`~TestCase.setUp`, which the testing framework will automatically
346call for every single test we run::
Georg Brandl116aa622007-08-15 14:28:22 +0000347
348 import unittest
349
350 class SimpleWidgetTestCase(unittest.TestCase):
351 def setUp(self):
352 self.widget = Widget('The widget')
353
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100354 def test_default_widget_size(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000355 self.assertEqual(self.widget.size(), (50,50),
356 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000357
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100358 def test_widget_resize(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000359 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000360 self.assertEqual(self.widget.size(), (100,150),
361 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000362
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100363.. note::
364 The order in which the various tests will be run is determined
365 by sorting the test method names with respect to the built-in
366 ordering for strings.
367
Benjamin Peterson52baa292009-03-24 00:56:30 +0000368If the :meth:`~TestCase.setUp` method raises an exception while the test is
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100369running, the framework will consider the test to have suffered an error, and
370the test method will not be executed.
Georg Brandl116aa622007-08-15 14:28:22 +0000371
Benjamin Peterson52baa292009-03-24 00:56:30 +0000372Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100373after the test method has been run::
Georg Brandl116aa622007-08-15 14:28:22 +0000374
375 import unittest
376
377 class SimpleWidgetTestCase(unittest.TestCase):
378 def setUp(self):
379 self.widget = Widget('The widget')
380
381 def tearDown(self):
382 self.widget.dispose()
Georg Brandl116aa622007-08-15 14:28:22 +0000383
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100384If :meth:`~TestCase.setUp` succeeded, :meth:`~TestCase.tearDown` will be
385run whether the test method succeeded or not.
Georg Brandl116aa622007-08-15 14:28:22 +0000386
387Such a working environment for the testing code is called a :dfn:`fixture`.
388
Georg Brandl116aa622007-08-15 14:28:22 +0000389Test case instances are grouped together according to the features they test.
390:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100391represented by :mod:`unittest`'s :class:`TestSuite` class. In most cases,
392calling :func:`unittest.main` will do the right thing and collect all the
393module's test cases for you, and then execute them.
Georg Brandl116aa622007-08-15 14:28:22 +0000394
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100395However, should you want to customize the building of your test suite,
396you can do it yourself::
Georg Brandl116aa622007-08-15 14:28:22 +0000397
398 def suite():
399 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000400 suite.addTest(WidgetTestCase('test_default_size'))
401 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000402 return suite
403
Georg Brandl116aa622007-08-15 14:28:22 +0000404You can place the definitions of test cases and test suites in the same modules
405as the code they are to test (such as :file:`widget.py`), but there are several
406advantages to placing the test code in a separate module, such as
407:file:`test_widget.py`:
408
409* The test module can be run standalone from the command line.
410
411* The test code can more easily be separated from shipped code.
412
413* There is less temptation to change test code to fit the code it tests without
414 a good reason.
415
416* Test code should be modified much less frequently than the code it tests.
417
418* Tested code can be refactored more easily.
419
420* Tests for modules written in C must be in separate modules anyway, so why not
421 be consistent?
422
423* If the testing strategy changes, there is no need to change the source code.
424
425
426.. _legacy-unit-tests:
427
428Re-using old test code
429----------------------
430
431Some users will find that they have existing test code that they would like to
432run from :mod:`unittest`, without converting every old test function to a
433:class:`TestCase` subclass.
434
435For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
436This subclass of :class:`TestCase` can be used to wrap an existing test
437function. Set-up and tear-down functions can also be provided.
438
439Given the following test function::
440
441 def testSomething():
442 something = makeSomething()
443 assert something.name is not None
444 # ...
445
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100446one can create an equivalent test case instance as follows, with optional
447set-up and tear-down methods::
Georg Brandl116aa622007-08-15 14:28:22 +0000448
449 testcase = unittest.FunctionTestCase(testSomething,
450 setUp=makeSomethingDB,
451 tearDown=deleteSomethingDB)
452
Georg Brandl116aa622007-08-15 14:28:22 +0000453.. note::
454
Benjamin Petersond2397752009-06-27 23:45:02 +0000455 Even though :class:`FunctionTestCase` can be used to quickly convert an
456 existing test base over to a :mod:`unittest`\ -based system, this approach is
457 not recommended. Taking the time to set up proper :class:`TestCase`
458 subclasses will make future test refactorings infinitely easier.
Georg Brandl116aa622007-08-15 14:28:22 +0000459
Benjamin Peterson52baa292009-03-24 00:56:30 +0000460In some cases, the existing tests may have been written using the :mod:`doctest`
461module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
462automatically build :class:`unittest.TestSuite` instances from the existing
463:mod:`doctest`\ -based tests.
464
Georg Brandl116aa622007-08-15 14:28:22 +0000465
Benjamin Peterson5254c042009-03-23 22:25:03 +0000466.. _unittest-skipping:
467
468Skipping tests and expected failures
469------------------------------------
470
Michael Foordf5c851a2010-02-05 21:48:03 +0000471.. versionadded:: 3.1
472
Benjamin Peterson5254c042009-03-23 22:25:03 +0000473Unittest supports skipping individual test methods and even whole classes of
Serhiy Storchakad65c9492015-11-02 14:10:23 +0200474tests. In addition, it supports marking a test as an "expected failure," a test
Benjamin Peterson5254c042009-03-23 22:25:03 +0000475that is broken and will fail, but shouldn't be counted as a failure on a
476:class:`TestResult`.
477
478Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
479or one of its conditional variants.
480
Ezio Melottifed69ba2013-03-01 21:26:04 +0200481Basic skipping looks like this::
Benjamin Peterson5254c042009-03-23 22:25:03 +0000482
483 class MyTestCase(unittest.TestCase):
484
485 @unittest.skip("demonstrating skipping")
486 def test_nothing(self):
487 self.fail("shouldn't happen")
488
Benjamin Petersond2397752009-06-27 23:45:02 +0000489 @unittest.skipIf(mylib.__version__ < (1, 3),
490 "not supported in this library version")
Benjamin Petersonded31c42009-03-30 15:04:16 +0000491 def test_format(self):
492 # Tests that work for only a certain version of the library.
493 pass
494
495 @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
496 def test_windows_support(self):
497 # windows specific testing code
498 pass
499
Ezio Melottifed69ba2013-03-01 21:26:04 +0200500This is the output of running the example above in verbose mode::
Benjamin Peterson5254c042009-03-23 22:25:03 +0000501
Benjamin Petersonded31c42009-03-30 15:04:16 +0000502 test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000503 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
Benjamin Petersonded31c42009-03-30 15:04:16 +0000504 test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000505
506 ----------------------------------------------------------------------
Benjamin Petersonded31c42009-03-30 15:04:16 +0000507 Ran 3 tests in 0.005s
508
509 OK (skipped=3)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000510
Ezio Melottifed69ba2013-03-01 21:26:04 +0200511Classes can be skipped just like methods::
Benjamin Peterson5254c042009-03-23 22:25:03 +0000512
Sandro Tosi317075d2012-03-31 18:34:59 +0200513 @unittest.skip("showing class skipping")
Benjamin Peterson5254c042009-03-23 22:25:03 +0000514 class MySkippedTestCase(unittest.TestCase):
515 def test_not_run(self):
516 pass
517
Benjamin Peterson52baa292009-03-24 00:56:30 +0000518:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
519that needs to be set up is not available.
520
Benjamin Peterson5254c042009-03-23 22:25:03 +0000521Expected failures use the :func:`expectedFailure` decorator. ::
522
523 class ExpectedFailureTestCase(unittest.TestCase):
524 @unittest.expectedFailure
525 def test_fail(self):
526 self.assertEqual(1, 0, "broken")
527
528It's easy to roll your own skipping decorators by making a decorator that calls
529:func:`skip` on the test when it wants it to be skipped. This decorator skips
Ezio Melottifed69ba2013-03-01 21:26:04 +0200530the test unless the passed object has a certain attribute::
Benjamin Peterson5254c042009-03-23 22:25:03 +0000531
532 def skipUnlessHasattr(obj, attr):
533 if hasattr(obj, attr):
534 return lambda func: func
Ezio Melotti265281a2013-03-27 20:11:55 +0200535 return unittest.skip("{!r} doesn't have {!r}".format(obj, attr))
Benjamin Peterson5254c042009-03-23 22:25:03 +0000536
537The following decorators implement test skipping and expected failures:
538
Georg Brandl8a1caa22010-07-29 16:01:11 +0000539.. decorator:: skip(reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000540
541 Unconditionally skip the decorated test. *reason* should describe why the
542 test is being skipped.
543
Georg Brandl8a1caa22010-07-29 16:01:11 +0000544.. decorator:: skipIf(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000545
546 Skip the decorated test if *condition* is true.
547
Georg Brandl8a1caa22010-07-29 16:01:11 +0000548.. decorator:: skipUnless(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000549
Georg Brandl6faee4e2010-09-21 14:48:28 +0000550 Skip the decorated test unless *condition* is true.
Benjamin Peterson5254c042009-03-23 22:25:03 +0000551
Georg Brandl8a1caa22010-07-29 16:01:11 +0000552.. decorator:: expectedFailure
Benjamin Peterson5254c042009-03-23 22:25:03 +0000553
554 Mark the test as an expected failure. If the test fails when run, the test
555 is not counted as a failure.
556
Ezio Melotti265281a2013-03-27 20:11:55 +0200557.. exception:: SkipTest(reason)
558
559 This exception is raised to skip a test.
560
561 Usually you can use :meth:`TestCase.skipTest` or one of the skipping
562 decorators instead of raising this directly.
563
R David Murray42fa1102014-01-03 13:03:36 -0500564Skipped tests will not have :meth:`~TestCase.setUp` or :meth:`~TestCase.tearDown` run around them.
565Skipped classes will not have :meth:`~TestCase.setUpClass` or :meth:`~TestCase.tearDownClass` run.
566Skipped modules will not have :func:`setUpModule` or :func:`tearDownModule` run.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000567
Benjamin Peterson5254c042009-03-23 22:25:03 +0000568
Antoine Pitrouc9b3ef22013-03-20 20:16:47 +0100569.. _subtests:
570
571Distinguishing test iterations using subtests
572---------------------------------------------
573
574.. versionadded:: 3.4
575
576When some of your tests differ only by a some very small differences, for
577instance some parameters, unittest allows you to distinguish them inside
578the body of a test method using the :meth:`~TestCase.subTest` context manager.
579
580For example, the following test::
581
582 class NumbersTest(unittest.TestCase):
583
584 def test_even(self):
585 """
586 Test that numbers between 0 and 5 are all even.
587 """
588 for i in range(0, 6):
589 with self.subTest(i=i):
590 self.assertEqual(i % 2, 0)
591
592will produce the following output::
593
594 ======================================================================
595 FAIL: test_even (__main__.NumbersTest) (i=1)
596 ----------------------------------------------------------------------
597 Traceback (most recent call last):
598 File "subtests.py", line 32, in test_even
599 self.assertEqual(i % 2, 0)
600 AssertionError: 1 != 0
601
602 ======================================================================
603 FAIL: test_even (__main__.NumbersTest) (i=3)
604 ----------------------------------------------------------------------
605 Traceback (most recent call last):
606 File "subtests.py", line 32, in test_even
607 self.assertEqual(i % 2, 0)
608 AssertionError: 1 != 0
609
610 ======================================================================
611 FAIL: test_even (__main__.NumbersTest) (i=5)
612 ----------------------------------------------------------------------
613 Traceback (most recent call last):
614 File "subtests.py", line 32, in test_even
615 self.assertEqual(i % 2, 0)
616 AssertionError: 1 != 0
617
618Without using a subtest, execution would stop after the first failure,
619and the error would be less easy to diagnose because the value of ``i``
620wouldn't be displayed::
621
622 ======================================================================
623 FAIL: test_even (__main__.NumbersTest)
624 ----------------------------------------------------------------------
625 Traceback (most recent call last):
626 File "subtests.py", line 32, in test_even
627 self.assertEqual(i % 2, 0)
628 AssertionError: 1 != 0
629
630
Georg Brandl116aa622007-08-15 14:28:22 +0000631.. _unittest-contents:
632
633Classes and functions
634---------------------
635
Benjamin Peterson52baa292009-03-24 00:56:30 +0000636This section describes in depth the API of :mod:`unittest`.
637
638
639.. _testcase-objects:
640
641Test cases
642~~~~~~~~~~
Georg Brandl116aa622007-08-15 14:28:22 +0000643
Georg Brandl7f01a132009-09-16 15:58:14 +0000644.. class:: TestCase(methodName='runTest')
Georg Brandl116aa622007-08-15 14:28:22 +0000645
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100646 Instances of the :class:`TestCase` class represent the logical test units
Georg Brandl116aa622007-08-15 14:28:22 +0000647 in the :mod:`unittest` universe. This class is intended to be used as a base
648 class, with specific tests being implemented by concrete subclasses. This class
649 implements the interface needed by the test runner to allow it to drive the
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100650 tests, and methods that the test code can use to check for and report various
Georg Brandl116aa622007-08-15 14:28:22 +0000651 kinds of failure.
652
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100653 Each instance of :class:`TestCase` will run a single base method: the method
Robert Collinse02f6c22015-07-23 06:37:26 +1200654 named *methodName*.
655 In most uses of :class:`TestCase`, you will neither change
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100656 the *methodName* nor reimplement the default ``runTest()`` method.
Georg Brandl116aa622007-08-15 14:28:22 +0000657
Michael Foord1341bb02011-03-14 19:01:46 -0400658 .. versionchanged:: 3.2
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100659 :class:`TestCase` can be instantiated successfully without providing a
660 *methodName*. This makes it easier to experiment with :class:`TestCase`
661 from the interactive interpreter.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000662
663 :class:`TestCase` instances provide three groups of methods: one group used
664 to run the test, another used by the test implementation to check conditions
665 and report failures, and some inquiry methods allowing information about the
666 test itself to be gathered.
667
668 Methods in the first group (running the test) are:
669
Benjamin Peterson52baa292009-03-24 00:56:30 +0000670 .. method:: setUp()
671
672 Method called to prepare the test fixture. This is called immediately
Terry Jan Reedy7f84d1e2014-04-15 23:44:14 -0400673 before calling the test method; other than :exc:`AssertionError` or :exc:`SkipTest`,
674 any exception raised by this method will be considered an error rather than
Terry Jan Reedy6ac42402014-04-15 23:38:18 -0400675 a test failure. The default implementation does nothing.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000676
677
678 .. method:: tearDown()
679
680 Method called immediately after the test method has been called and the
681 result recorded. This is called even if the test method raised an
682 exception, so the implementation in subclasses may need to be particularly
Terry Jan Reedy7f84d1e2014-04-15 23:44:14 -0400683 careful about checking internal state. Any exception, other than :exc:`AssertionError`
684 or :exc:`SkipTest`, raised by this method will be considered an error rather than a
685 test failure. This method will only be called if the :meth:`setUp` succeeds,
Terry Jan Reedy6ac42402014-04-15 23:38:18 -0400686 regardless of the outcome of the test method. The default implementation does nothing.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000687
688
Benjamin Petersonb48af542010-04-11 20:43:16 +0000689 .. method:: setUpClass()
690
691 A class method called before tests in an individual class run.
692 ``setUpClass`` is called with the class as the only argument
693 and must be decorated as a :func:`classmethod`::
694
695 @classmethod
696 def setUpClass(cls):
697 ...
698
699 See `Class and Module Fixtures`_ for more details.
700
701 .. versionadded:: 3.2
702
703
704 .. method:: tearDownClass()
705
706 A class method called after tests in an individual class have run.
707 ``tearDownClass`` is called with the class as the only argument
708 and must be decorated as a :meth:`classmethod`::
709
710 @classmethod
711 def tearDownClass(cls):
712 ...
713
714 See `Class and Module Fixtures`_ for more details.
715
716 .. versionadded:: 3.2
717
718
Georg Brandl7f01a132009-09-16 15:58:14 +0000719 .. method:: run(result=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000720
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100721 Run the test, collecting the result into the :class:`TestResult` object
722 passed as *result*. If *result* is omitted or ``None``, a temporary
723 result object is created (by calling the :meth:`defaultTestResult`
724 method) and used. The result object is returned to :meth:`run`'s
725 caller.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000726
727 The same effect may be had by simply calling the :class:`TestCase`
728 instance.
729
Michael Foord1341bb02011-03-14 19:01:46 -0400730 .. versionchanged:: 3.3
731 Previous versions of ``run`` did not return the result. Neither did
732 calling an instance.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000733
Benjamin Petersone549ead2009-03-28 21:42:05 +0000734 .. method:: skipTest(reason)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000735
Stefan Kraha5bf3f52010-05-19 16:09:41 +0000736 Calling this during a test method or :meth:`setUp` skips the current
Benjamin Peterson52baa292009-03-24 00:56:30 +0000737 test. See :ref:`unittest-skipping` for more information.
738
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000739 .. versionadded:: 3.1
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000740
Benjamin Peterson52baa292009-03-24 00:56:30 +0000741
Antoine Pitrouc9b3ef22013-03-20 20:16:47 +0100742 .. method:: subTest(msg=None, **params)
743
744 Return a context manager which executes the enclosed code block as a
745 subtest. *msg* and *params* are optional, arbitrary values which are
746 displayed whenever a subtest fails, allowing you to identify them
747 clearly.
748
749 A test case can contain any number of subtest declarations, and
750 they can be arbitrarily nested.
751
752 See :ref:`subtests` for more information.
753
754 .. versionadded:: 3.4
755
756
Benjamin Peterson52baa292009-03-24 00:56:30 +0000757 .. method:: debug()
758
759 Run the test without collecting the result. This allows exceptions raised
760 by the test to be propagated to the caller, and can be used to support
761 running tests under a debugger.
762
Ezio Melotti22170ed2010-11-20 09:57:27 +0000763 .. _assert-methods:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000764
Ezio Melottif418db22016-01-12 11:03:31 +0200765 The :class:`TestCase` class provides several assert methods to check for and
766 report failures. The following table lists the most commonly used methods
767 (see the tables below for more assert methods):
Benjamin Peterson52baa292009-03-24 00:56:30 +0000768
Ezio Melotti4370b302010-11-03 20:39:14 +0000769 +-----------------------------------------+-----------------------------+---------------+
770 | Method | Checks that | New in |
771 +=========================================+=============================+===============+
772 | :meth:`assertEqual(a, b) | ``a == b`` | |
773 | <TestCase.assertEqual>` | | |
774 +-----------------------------------------+-----------------------------+---------------+
775 | :meth:`assertNotEqual(a, b) | ``a != b`` | |
776 | <TestCase.assertNotEqual>` | | |
777 +-----------------------------------------+-----------------------------+---------------+
778 | :meth:`assertTrue(x) | ``bool(x) is True`` | |
779 | <TestCase.assertTrue>` | | |
780 +-----------------------------------------+-----------------------------+---------------+
781 | :meth:`assertFalse(x) | ``bool(x) is False`` | |
782 | <TestCase.assertFalse>` | | |
783 +-----------------------------------------+-----------------------------+---------------+
784 | :meth:`assertIs(a, b) | ``a is b`` | 3.1 |
785 | <TestCase.assertIs>` | | |
786 +-----------------------------------------+-----------------------------+---------------+
787 | :meth:`assertIsNot(a, b) | ``a is not b`` | 3.1 |
788 | <TestCase.assertIsNot>` | | |
789 +-----------------------------------------+-----------------------------+---------------+
790 | :meth:`assertIsNone(x) | ``x is None`` | 3.1 |
791 | <TestCase.assertIsNone>` | | |
792 +-----------------------------------------+-----------------------------+---------------+
793 | :meth:`assertIsNotNone(x) | ``x is not None`` | 3.1 |
794 | <TestCase.assertIsNotNone>` | | |
795 +-----------------------------------------+-----------------------------+---------------+
796 | :meth:`assertIn(a, b) | ``a in b`` | 3.1 |
797 | <TestCase.assertIn>` | | |
798 +-----------------------------------------+-----------------------------+---------------+
799 | :meth:`assertNotIn(a, b) | ``a not in b`` | 3.1 |
800 | <TestCase.assertNotIn>` | | |
801 +-----------------------------------------+-----------------------------+---------------+
802 | :meth:`assertIsInstance(a, b) | ``isinstance(a, b)`` | 3.2 |
803 | <TestCase.assertIsInstance>` | | |
804 +-----------------------------------------+-----------------------------+---------------+
805 | :meth:`assertNotIsInstance(a, b) | ``not isinstance(a, b)`` | 3.2 |
806 | <TestCase.assertNotIsInstance>` | | |
807 +-----------------------------------------+-----------------------------+---------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000808
Ezio Melottib4dc2502011-05-06 15:01:41 +0300809 All the assert methods accept a *msg* argument that, if specified, is used
810 as the error message on failure (see also :data:`longMessage`).
811 Note that the *msg* keyword argument can be passed to :meth:`assertRaises`,
812 :meth:`assertRaisesRegex`, :meth:`assertWarns`, :meth:`assertWarnsRegex`
813 only when they are used as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000814
Michael Foorde180d392011-01-28 19:51:48 +0000815 .. method:: assertEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000816
Michael Foorde180d392011-01-28 19:51:48 +0000817 Test that *first* and *second* are equal. If the values do not
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000818 compare equal, the test will fail.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000819
Michael Foorde180d392011-01-28 19:51:48 +0000820 In addition, if *first* and *second* are the exact same type and one of
Michael Foord02834952010-02-08 23:10:39 +0000821 list, tuple, dict, set, frozenset or str or any type that a subclass
Ezio Melotti9ecb6be2012-01-16 08:28:54 +0200822 registers with :meth:`addTypeEqualityFunc` the type-specific equality
Michael Foord02834952010-02-08 23:10:39 +0000823 function will be called in order to generate a more useful default
Ezio Melotti22170ed2010-11-20 09:57:27 +0000824 error message (see also the :ref:`list of type-specific methods
825 <type-specific-methods>`).
Ezio Melotti4841fd62010-11-05 15:43:40 +0000826
Raymond Hettinger35a88362009-04-09 00:08:24 +0000827 .. versionchanged:: 3.1
Ezio Melotti9ecb6be2012-01-16 08:28:54 +0200828 Added the automatic calling of type-specific equality function.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000829
Michael Foord28a817e2010-02-09 00:03:57 +0000830 .. versionchanged:: 3.2
831 :meth:`assertMultiLineEqual` added as the default type equality
832 function for comparing strings.
Michael Foord02834952010-02-08 23:10:39 +0000833
Benjamin Peterson52baa292009-03-24 00:56:30 +0000834
Michael Foorde180d392011-01-28 19:51:48 +0000835 .. method:: assertNotEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000836
Michael Foorde180d392011-01-28 19:51:48 +0000837 Test that *first* and *second* are not equal. If the values do
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000838 compare equal, the test will fail.
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000839
Ezio Melotti4370b302010-11-03 20:39:14 +0000840 .. method:: assertTrue(expr, msg=None)
Ezio Melotti4841fd62010-11-05 15:43:40 +0000841 assertFalse(expr, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000842
Ezio Melotti4841fd62010-11-05 15:43:40 +0000843 Test that *expr* is true (or false).
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000844
Ezio Melotti4841fd62010-11-05 15:43:40 +0000845 Note that this is equivalent to ``bool(expr) is True`` and not to ``expr
846 is True`` (use ``assertIs(expr, True)`` for the latter). This method
847 should also be avoided when more specific methods are available (e.g.
848 ``assertEqual(a, b)`` instead of ``assertTrue(a == b)``), because they
849 provide a better error message in case of failure.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000850
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000851
Michael Foorde180d392011-01-28 19:51:48 +0000852 .. method:: assertIs(first, second, msg=None)
853 assertIsNot(first, second, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000854
Michael Foorde180d392011-01-28 19:51:48 +0000855 Test that *first* and *second* evaluate (or don't evaluate) to the
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000856 same object.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000857
Raymond Hettinger35a88362009-04-09 00:08:24 +0000858 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000859
860
Ezio Melotti4370b302010-11-03 20:39:14 +0000861 .. method:: assertIsNone(expr, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000862 assertIsNotNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000863
Ezio Melotti9794a262010-11-04 14:52:13 +0000864 Test that *expr* is (or is not) None.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000865
Ezio Melotti4370b302010-11-03 20:39:14 +0000866 .. versionadded:: 3.1
Benjamin Petersonb48af542010-04-11 20:43:16 +0000867
868
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000869 .. method:: assertIn(first, second, msg=None)
870 assertNotIn(first, second, msg=None)
871
Ezio Melotti4841fd62010-11-05 15:43:40 +0000872 Test that *first* is (or is not) in *second*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000873
Raymond Hettinger35a88362009-04-09 00:08:24 +0000874 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000875
876
Ezio Melotti9c02c2f2010-11-03 20:45:31 +0000877 .. method:: assertIsInstance(obj, cls, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000878 assertNotIsInstance(obj, cls, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000879
Ezio Melotti9794a262010-11-04 14:52:13 +0000880 Test that *obj* is (or is not) an instance of *cls* (which can be a
881 class or a tuple of classes, as supported by :func:`isinstance`).
Ezio Melotti80a61e82011-12-19 07:04:48 +0200882 To check for the exact type, use :func:`assertIs(type(obj), cls) <assertIs>`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000883
Ezio Melotti4370b302010-11-03 20:39:14 +0000884 .. versionadded:: 3.2
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000885
886
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000887
Ezio Melottif418db22016-01-12 11:03:31 +0200888 It is also possible to check the production of exceptions, warnings, and
Antoine Pitrou7c89ae22013-09-15 02:01:39 +0200889 log messages using the following methods:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000890
Ezio Melotti4370b302010-11-03 20:39:14 +0000891 +---------------------------------------------------------+--------------------------------------+------------+
892 | Method | Checks that | New in |
893 +=========================================================+======================================+============+
Éric Araujo941afed2011-09-01 02:47:34 +0200894 | :meth:`assertRaises(exc, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *exc* | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000895 | <TestCase.assertRaises>` | | |
896 +---------------------------------------------------------+--------------------------------------+------------+
Ezio Melotti560a7782013-09-13 22:17:40 +0300897 | :meth:`assertRaisesRegex(exc, r, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *exc* | 3.1 |
898 | <TestCase.assertRaisesRegex>` | and the message matches regex *r* | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000899 +---------------------------------------------------------+--------------------------------------+------------+
Éric Araujo941afed2011-09-01 02:47:34 +0200900 | :meth:`assertWarns(warn, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *warn* | 3.2 |
Ezio Melotti4370b302010-11-03 20:39:14 +0000901 | <TestCase.assertWarns>` | | |
902 +---------------------------------------------------------+--------------------------------------+------------+
Ezio Melotti560a7782013-09-13 22:17:40 +0300903 | :meth:`assertWarnsRegex(warn, r, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *warn* | 3.2 |
904 | <TestCase.assertWarnsRegex>` | and the message matches regex *r* | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000905 +---------------------------------------------------------+--------------------------------------+------------+
Georg Brandled007d52013-11-24 16:09:26 +0100906 | :meth:`assertLogs(logger, level) | The ``with`` block logs on *logger* | 3.4 |
907 | <TestCase.assertLogs>` | with minimum *level* | |
Antoine Pitrou7c89ae22013-09-15 02:01:39 +0200908 +---------------------------------------------------------+--------------------------------------+------------+
Benjamin Peterson52baa292009-03-24 00:56:30 +0000909
Georg Brandl7f01a132009-09-16 15:58:14 +0000910 .. method:: assertRaises(exception, callable, *args, **kwds)
Ezio Melottib4dc2502011-05-06 15:01:41 +0300911 assertRaises(exception, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000912
913 Test that an exception is raised when *callable* is called with any
914 positional or keyword arguments that are also passed to
915 :meth:`assertRaises`. The test passes if *exception* is raised, is an
916 error if another exception is raised, or fails if no exception is raised.
917 To catch any of a group of exceptions, a tuple containing the exception
918 classes may be passed as *exception*.
919
Ezio Melottib4dc2502011-05-06 15:01:41 +0300920 If only the *exception* and possibly the *msg* arguments are given,
921 return a context manager so that the code under test can be written
922 inline rather than as a function::
Benjamin Petersonded31c42009-03-30 15:04:16 +0000923
Michael Foord41531f22010-02-05 21:13:40 +0000924 with self.assertRaises(SomeException):
Benjamin Petersonded31c42009-03-30 15:04:16 +0000925 do_something()
926
Ezio Melottib4dc2502011-05-06 15:01:41 +0300927 When used as a context manager, :meth:`assertRaises` accepts the
928 additional keyword argument *msg*.
929
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000930 The context manager will store the caught exception object in its
Ezio Melotti49008232010-02-08 21:57:48 +0000931 :attr:`exception` attribute. This can be useful if the intention
Michael Foord41531f22010-02-05 21:13:40 +0000932 is to perform additional checks on the exception raised::
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000933
Georg Brandl8a1caa22010-07-29 16:01:11 +0000934 with self.assertRaises(SomeException) as cm:
935 do_something()
Michael Foord41531f22010-02-05 21:13:40 +0000936
Georg Brandl8a1caa22010-07-29 16:01:11 +0000937 the_exception = cm.exception
938 self.assertEqual(the_exception.error_code, 3)
Michael Foord41531f22010-02-05 21:13:40 +0000939
Ezio Melotti49008232010-02-08 21:57:48 +0000940 .. versionchanged:: 3.1
Benjamin Petersonded31c42009-03-30 15:04:16 +0000941 Added the ability to use :meth:`assertRaises` as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000942
Ezio Melotti49008232010-02-08 21:57:48 +0000943 .. versionchanged:: 3.2
944 Added the :attr:`exception` attribute.
945
Ezio Melottib4dc2502011-05-06 15:01:41 +0300946 .. versionchanged:: 3.3
947 Added the *msg* keyword argument when used as a context manager.
948
Benjamin Peterson52baa292009-03-24 00:56:30 +0000949
Ezio Melottied3a7d22010-12-01 02:32:32 +0000950 .. method:: assertRaisesRegex(exception, regex, callable, *args, **kwds)
Ezio Melottib4dc2502011-05-06 15:01:41 +0300951 assertRaisesRegex(exception, regex, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000952
Ezio Melottied3a7d22010-12-01 02:32:32 +0000953 Like :meth:`assertRaises` but also tests that *regex* matches
954 on the string representation of the raised exception. *regex* may be
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000955 a regular expression object or a string containing a regular expression
956 suitable for use by :func:`re.search`. Examples::
957
Terry Jan Reedyc4565a92013-06-29 13:15:43 -0400958 self.assertRaisesRegex(ValueError, "invalid literal for.*XYZ'$",
Ezio Melottied3a7d22010-12-01 02:32:32 +0000959 int, 'XYZ')
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000960
961 or::
962
Ezio Melottied3a7d22010-12-01 02:32:32 +0000963 with self.assertRaisesRegex(ValueError, 'literal'):
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000964 int('XYZ')
965
Georg Brandl419e3de2010-12-01 15:44:25 +0000966 .. versionadded:: 3.1
967 under the name ``assertRaisesRegexp``.
Ezio Melottib4dc2502011-05-06 15:01:41 +0300968
Ezio Melottied3a7d22010-12-01 02:32:32 +0000969 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +0000970 Renamed to :meth:`assertRaisesRegex`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000971
Ezio Melottib4dc2502011-05-06 15:01:41 +0300972 .. versionchanged:: 3.3
973 Added the *msg* keyword argument when used as a context manager.
974
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000975
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000976 .. method:: assertWarns(warning, callable, *args, **kwds)
Ezio Melottib4dc2502011-05-06 15:01:41 +0300977 assertWarns(warning, msg=None)
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000978
979 Test that a warning is triggered when *callable* is called with any
980 positional or keyword arguments that are also passed to
981 :meth:`assertWarns`. The test passes if *warning* is triggered and
Terry Jan Reedy778cba72013-07-30 22:31:06 -0400982 fails if it isn't. Any exception is an error.
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000983 To catch any of a group of warnings, a tuple containing the warning
984 classes may be passed as *warnings*.
985
Ezio Melottib4dc2502011-05-06 15:01:41 +0300986 If only the *warning* and possibly the *msg* arguments are given,
Terry Jan Reedy778cba72013-07-30 22:31:06 -0400987 return a context manager so that the code under test can be written
Ezio Melottib4dc2502011-05-06 15:01:41 +0300988 inline rather than as a function::
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000989
990 with self.assertWarns(SomeWarning):
991 do_something()
992
Terry Jan Reedy9eda66d2013-07-27 16:15:29 -0400993 When used as a context manager, :meth:`assertWarns` accepts the
Ezio Melottib4dc2502011-05-06 15:01:41 +0300994 additional keyword argument *msg*.
995
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000996 The context manager will store the caught warning object in its
997 :attr:`warning` attribute, and the source line which triggered the
998 warnings in the :attr:`filename` and :attr:`lineno` attributes.
999 This can be useful if the intention is to perform additional checks
Terry Jan Reedy778cba72013-07-30 22:31:06 -04001000 on the warning caught::
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001001
1002 with self.assertWarns(SomeWarning) as cm:
1003 do_something()
1004
1005 self.assertIn('myfile.py', cm.filename)
1006 self.assertEqual(320, cm.lineno)
1007
1008 This method works regardless of the warning filters in place when it
1009 is called.
1010
1011 .. versionadded:: 3.2
1012
Ezio Melottib4dc2502011-05-06 15:01:41 +03001013 .. versionchanged:: 3.3
1014 Added the *msg* keyword argument when used as a context manager.
1015
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001016
Ezio Melottied3a7d22010-12-01 02:32:32 +00001017 .. method:: assertWarnsRegex(warning, regex, callable, *args, **kwds)
Ezio Melottib4dc2502011-05-06 15:01:41 +03001018 assertWarnsRegex(warning, regex, msg=None)
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001019
Ezio Melottied3a7d22010-12-01 02:32:32 +00001020 Like :meth:`assertWarns` but also tests that *regex* matches on the
1021 message of the triggered warning. *regex* may be a regular expression
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001022 object or a string containing a regular expression suitable for use
1023 by :func:`re.search`. Example::
1024
Ezio Melottied3a7d22010-12-01 02:32:32 +00001025 self.assertWarnsRegex(DeprecationWarning,
1026 r'legacy_function\(\) is deprecated',
1027 legacy_function, 'XYZ')
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001028
1029 or::
1030
Ezio Melottied3a7d22010-12-01 02:32:32 +00001031 with self.assertWarnsRegex(RuntimeWarning, 'unsafe frobnicating'):
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001032 frobnicate('/etc/passwd')
1033
1034 .. versionadded:: 3.2
1035
Ezio Melottib4dc2502011-05-06 15:01:41 +03001036 .. versionchanged:: 3.3
1037 Added the *msg* keyword argument when used as a context manager.
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001038
Antoine Pitrou0715b9f2013-09-14 19:45:47 +02001039 .. method:: assertLogs(logger=None, level=None)
1040
1041 A context manager to test that at least one message is logged on
1042 the *logger* or one of its children, with at least the given
1043 *level*.
1044
1045 If given, *logger* should be a :class:`logging.Logger` object or a
1046 :class:`str` giving the name of a logger. The default is the root
1047 logger, which will catch all messages.
1048
1049 If given, *level* should be either a numeric logging level or
1050 its string equivalent (for example either ``"ERROR"`` or
1051 :attr:`logging.ERROR`). The default is :attr:`logging.INFO`.
1052
1053 The test passes if at least one message emitted inside the ``with``
1054 block matches the *logger* and *level* conditions, otherwise it fails.
1055
1056 The object returned by the context manager is a recording helper
1057 which keeps tracks of the matching log messages. It has two
1058 attributes:
1059
1060 .. attribute:: records
1061
1062 A list of :class:`logging.LogRecord` objects of the matching
1063 log messages.
1064
1065 .. attribute:: output
1066
1067 A list of :class:`str` objects with the formatted output of
1068 matching messages.
1069
1070 Example::
1071
1072 with self.assertLogs('foo', level='INFO') as cm:
1073 logging.getLogger('foo').info('first message')
1074 logging.getLogger('foo.bar').error('second message')
1075 self.assertEqual(cm.output, ['INFO:foo:first message',
1076 'ERROR:foo.bar:second message'])
1077
1078 .. versionadded:: 3.4
1079
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001080
Ezio Melotti4370b302010-11-03 20:39:14 +00001081 There are also other methods used to perform more specific checks, such as:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001082
Ezio Melotti4370b302010-11-03 20:39:14 +00001083 +---------------------------------------+--------------------------------+--------------+
1084 | Method | Checks that | New in |
1085 +=======================================+================================+==============+
1086 | :meth:`assertAlmostEqual(a, b) | ``round(a-b, 7) == 0`` | |
1087 | <TestCase.assertAlmostEqual>` | | |
1088 +---------------------------------------+--------------------------------+--------------+
1089 | :meth:`assertNotAlmostEqual(a, b) | ``round(a-b, 7) != 0`` | |
1090 | <TestCase.assertNotAlmostEqual>` | | |
1091 +---------------------------------------+--------------------------------+--------------+
1092 | :meth:`assertGreater(a, b) | ``a > b`` | 3.1 |
1093 | <TestCase.assertGreater>` | | |
1094 +---------------------------------------+--------------------------------+--------------+
1095 | :meth:`assertGreaterEqual(a, b) | ``a >= b`` | 3.1 |
1096 | <TestCase.assertGreaterEqual>` | | |
1097 +---------------------------------------+--------------------------------+--------------+
1098 | :meth:`assertLess(a, b) | ``a < b`` | 3.1 |
1099 | <TestCase.assertLess>` | | |
1100 +---------------------------------------+--------------------------------+--------------+
1101 | :meth:`assertLessEqual(a, b) | ``a <= b`` | 3.1 |
1102 | <TestCase.assertLessEqual>` | | |
1103 +---------------------------------------+--------------------------------+--------------+
Ezio Melotti560a7782013-09-13 22:17:40 +03001104 | :meth:`assertRegex(s, r) | ``r.search(s)`` | 3.1 |
Ezio Melottied3a7d22010-12-01 02:32:32 +00001105 | <TestCase.assertRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001106 +---------------------------------------+--------------------------------+--------------+
Ezio Melotti560a7782013-09-13 22:17:40 +03001107 | :meth:`assertNotRegex(s, r) | ``not r.search(s)`` | 3.2 |
Ezio Melottied3a7d22010-12-01 02:32:32 +00001108 | <TestCase.assertNotRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001109 +---------------------------------------+--------------------------------+--------------+
Éric Araujo941afed2011-09-01 02:47:34 +02001110 | :meth:`assertCountEqual(a, b) | *a* and *b* have the same | 3.2 |
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001111 | <TestCase.assertCountEqual>` | elements in the same number, | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001112 | | regardless of their order | |
1113 +---------------------------------------+--------------------------------+--------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001114
1115
Michael Foorde180d392011-01-28 19:51:48 +00001116 .. method:: assertAlmostEqual(first, second, places=7, msg=None, delta=None)
1117 assertNotAlmostEqual(first, second, places=7, msg=None, delta=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001118
Michael Foorde180d392011-01-28 19:51:48 +00001119 Test that *first* and *second* are approximately (or not approximately)
Ezio Melotti4841fd62010-11-05 15:43:40 +00001120 equal by computing the difference, rounding to the given number of
1121 decimal *places* (default 7), and comparing to zero. Note that these
1122 methods round the values to the given number of *decimal places* (i.e.
1123 like the :func:`round` function) and not *significant digits*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001124
Ezio Melotti4370b302010-11-03 20:39:14 +00001125 If *delta* is supplied instead of *places* then the difference
Ezio Melottid51914c2013-08-11 13:04:50 +03001126 between *first* and *second* must be less or equal to (or greater than) *delta*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001127
Ezio Melotti4370b302010-11-03 20:39:14 +00001128 Supplying both *delta* and *places* raises a ``TypeError``.
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001129
Ezio Melotti4370b302010-11-03 20:39:14 +00001130 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001131 :meth:`assertAlmostEqual` automatically considers almost equal objects
1132 that compare equal. :meth:`assertNotAlmostEqual` automatically fails
1133 if the objects compare equal. Added the *delta* keyword argument.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001134
Ezio Melotti4370b302010-11-03 20:39:14 +00001135
Michael Foorde180d392011-01-28 19:51:48 +00001136 .. method:: assertGreater(first, second, msg=None)
1137 assertGreaterEqual(first, second, msg=None)
1138 assertLess(first, second, msg=None)
1139 assertLessEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001140
Michael Foorde180d392011-01-28 19:51:48 +00001141 Test that *first* is respectively >, >=, < or <= than *second* depending
Ezio Melotti4841fd62010-11-05 15:43:40 +00001142 on the method name. If not, the test will fail::
Ezio Melotti4370b302010-11-03 20:39:14 +00001143
1144 >>> self.assertGreaterEqual(3, 4)
1145 AssertionError: "3" unexpectedly not greater than or equal to "4"
1146
1147 .. versionadded:: 3.1
1148
1149
Ezio Melottied3a7d22010-12-01 02:32:32 +00001150 .. method:: assertRegex(text, regex, msg=None)
1151 assertNotRegex(text, regex, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001152
Ezio Melottied3a7d22010-12-01 02:32:32 +00001153 Test that a *regex* search matches (or does not match) *text*. In case
Ezio Melotti4841fd62010-11-05 15:43:40 +00001154 of failure, the error message will include the pattern and the *text* (or
Ezio Melottied3a7d22010-12-01 02:32:32 +00001155 the pattern and the part of *text* that unexpectedly matched). *regex*
Ezio Melotti4370b302010-11-03 20:39:14 +00001156 may be a regular expression object or a string containing a regular
1157 expression suitable for use by :func:`re.search`.
1158
Georg Brandl419e3de2010-12-01 15:44:25 +00001159 .. versionadded:: 3.1
1160 under the name ``assertRegexpMatches``.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001161 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001162 The method ``assertRegexpMatches()`` has been renamed to
1163 :meth:`.assertRegex`.
1164 .. versionadded:: 3.2
1165 :meth:`.assertNotRegex`.
Ezio Melotti4370b302010-11-03 20:39:14 +00001166
1167
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001168 .. method:: assertCountEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001169
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001170 Test that sequence *first* contains the same elements as *second*,
Ezio Melotti4370b302010-11-03 20:39:14 +00001171 regardless of their order. When they don't, an error message listing the
1172 differences between the sequences will be generated.
1173
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001174 Duplicate elements are *not* ignored when comparing *first* and
1175 *second*. It verifies whether each element has the same count in both
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001176 sequences. Equivalent to:
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001177 ``assertEqual(Counter(list(first)), Counter(list(second)))``
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001178 but works with sequences of unhashable objects as well.
Ezio Melotti4370b302010-11-03 20:39:14 +00001179
Ezio Melotti4370b302010-11-03 20:39:14 +00001180 .. versionadded:: 3.2
1181
Ezio Melotti4370b302010-11-03 20:39:14 +00001182
Ezio Melotti22170ed2010-11-20 09:57:27 +00001183 .. _type-specific-methods:
Ezio Melotti4370b302010-11-03 20:39:14 +00001184
Ezio Melotti22170ed2010-11-20 09:57:27 +00001185 The :meth:`assertEqual` method dispatches the equality check for objects of
1186 the same type to different type-specific methods. These methods are already
1187 implemented for most of the built-in types, but it's also possible to
1188 register new methods using :meth:`addTypeEqualityFunc`:
1189
1190 .. method:: addTypeEqualityFunc(typeobj, function)
1191
1192 Registers a type-specific method called by :meth:`assertEqual` to check
1193 if two objects of exactly the same *typeobj* (not subclasses) compare
1194 equal. *function* must take two positional arguments and a third msg=None
1195 keyword argument just as :meth:`assertEqual` does. It must raise
1196 :data:`self.failureException(msg) <failureException>` when inequality
1197 between the first two parameters is detected -- possibly providing useful
1198 information and explaining the inequalities in details in the error
1199 message.
1200
1201 .. versionadded:: 3.1
1202
1203 The list of type-specific methods automatically used by
1204 :meth:`~TestCase.assertEqual` are summarized in the following table. Note
1205 that it's usually not necessary to invoke these methods directly.
Ezio Melotti4370b302010-11-03 20:39:14 +00001206
1207 +-----------------------------------------+-----------------------------+--------------+
1208 | Method | Used to compare | New in |
1209 +=========================================+=============================+==============+
1210 | :meth:`assertMultiLineEqual(a, b) | strings | 3.1 |
1211 | <TestCase.assertMultiLineEqual>` | | |
1212 +-----------------------------------------+-----------------------------+--------------+
1213 | :meth:`assertSequenceEqual(a, b) | sequences | 3.1 |
1214 | <TestCase.assertSequenceEqual>` | | |
1215 +-----------------------------------------+-----------------------------+--------------+
1216 | :meth:`assertListEqual(a, b) | lists | 3.1 |
1217 | <TestCase.assertListEqual>` | | |
1218 +-----------------------------------------+-----------------------------+--------------+
1219 | :meth:`assertTupleEqual(a, b) | tuples | 3.1 |
1220 | <TestCase.assertTupleEqual>` | | |
1221 +-----------------------------------------+-----------------------------+--------------+
1222 | :meth:`assertSetEqual(a, b) | sets or frozensets | 3.1 |
1223 | <TestCase.assertSetEqual>` | | |
1224 +-----------------------------------------+-----------------------------+--------------+
1225 | :meth:`assertDictEqual(a, b) | dicts | 3.1 |
1226 | <TestCase.assertDictEqual>` | | |
1227 +-----------------------------------------+-----------------------------+--------------+
1228
1229
1230
Michael Foorde180d392011-01-28 19:51:48 +00001231 .. method:: assertMultiLineEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001232
Michael Foorde180d392011-01-28 19:51:48 +00001233 Test that the multiline string *first* is equal to the string *second*.
Ezio Melotti4370b302010-11-03 20:39:14 +00001234 When not equal a diff of the two strings highlighting the differences
1235 will be included in the error message. This method is used by default
1236 when comparing strings with :meth:`assertEqual`.
1237
Ezio Melotti4370b302010-11-03 20:39:14 +00001238 .. versionadded:: 3.1
1239
1240
Michael Foorde180d392011-01-28 19:51:48 +00001241 .. method:: assertSequenceEqual(first, second, msg=None, seq_type=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001242
1243 Tests that two sequences are equal. If a *seq_type* is supplied, both
Michael Foorde180d392011-01-28 19:51:48 +00001244 *first* and *second* must be instances of *seq_type* or a failure will
Ezio Melotti4370b302010-11-03 20:39:14 +00001245 be raised. If the sequences are different an error message is
1246 constructed that shows the difference between the two.
1247
Ezio Melotti22170ed2010-11-20 09:57:27 +00001248 This method is not called directly by :meth:`assertEqual`, but
1249 it's used to implement :meth:`assertListEqual` and
Ezio Melotti4370b302010-11-03 20:39:14 +00001250 :meth:`assertTupleEqual`.
1251
1252 .. versionadded:: 3.1
1253
1254
Michael Foorde180d392011-01-28 19:51:48 +00001255 .. method:: assertListEqual(first, second, msg=None)
1256 assertTupleEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001257
Ezio Melotti49ccd512012-08-29 17:50:42 +03001258 Tests that two lists or tuples are equal. If not, an error message is
Ezio Melotti4370b302010-11-03 20:39:14 +00001259 constructed that shows only the differences between the two. An error
1260 is also raised if either of the parameters are of the wrong type.
1261 These methods are used by default when comparing lists or tuples with
1262 :meth:`assertEqual`.
1263
Ezio Melotti4370b302010-11-03 20:39:14 +00001264 .. versionadded:: 3.1
1265
1266
Michael Foorde180d392011-01-28 19:51:48 +00001267 .. method:: assertSetEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001268
1269 Tests that two sets are equal. If not, an error message is constructed
1270 that lists the differences between the sets. This method is used by
1271 default when comparing sets or frozensets with :meth:`assertEqual`.
1272
Michael Foorde180d392011-01-28 19:51:48 +00001273 Fails if either of *first* or *second* does not have a :meth:`set.difference`
Ezio Melotti4370b302010-11-03 20:39:14 +00001274 method.
1275
Ezio Melotti4370b302010-11-03 20:39:14 +00001276 .. versionadded:: 3.1
1277
1278
Michael Foorde180d392011-01-28 19:51:48 +00001279 .. method:: assertDictEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001280
1281 Test that two dictionaries are equal. If not, an error message is
1282 constructed that shows the differences in the dictionaries. This
1283 method will be used by default to compare dictionaries in
1284 calls to :meth:`assertEqual`.
1285
Ezio Melotti4370b302010-11-03 20:39:14 +00001286 .. versionadded:: 3.1
1287
1288
1289
Ezio Melotti22170ed2010-11-20 09:57:27 +00001290 .. _other-methods-and-attrs:
1291
Ezio Melotti4370b302010-11-03 20:39:14 +00001292 Finally the :class:`TestCase` provides the following methods and attributes:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001293
Benjamin Peterson52baa292009-03-24 00:56:30 +00001294
Georg Brandl7f01a132009-09-16 15:58:14 +00001295 .. method:: fail(msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001296
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001297 Signals a test failure unconditionally, with *msg* or ``None`` for
Benjamin Peterson52baa292009-03-24 00:56:30 +00001298 the error message.
1299
1300
1301 .. attribute:: failureException
1302
1303 This class attribute gives the exception raised by the test method. If a
1304 test framework needs to use a specialized exception, possibly to carry
1305 additional information, it must subclass this exception in order to "play
1306 fair" with the framework. The initial value of this attribute is
1307 :exc:`AssertionError`.
1308
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001309
1310 .. attribute:: longMessage
1311
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001312 If set to ``True`` then any explicit failure message you pass in to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001313 :ref:`assert methods <assert-methods>` will be appended to the end of the
1314 normal failure message. The normal messages contain useful information
1315 about the objects involved, for example the message from assertEqual
1316 shows you the repr of the two unequal objects. Setting this attribute
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001317 to ``True`` allows you to have a custom error message in addition to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001318 normal one.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001319
Michael Foord5074df62010-12-03 00:53:09 +00001320 This attribute defaults to ``True``. If set to False then a custom message
1321 passed to an assert method will silence the normal message.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001322
1323 The class setting can be overridden in individual tests by assigning an
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001324 instance attribute to ``True`` or ``False`` before calling the assert methods.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001325
Raymond Hettinger35a88362009-04-09 00:08:24 +00001326 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001327
1328
Michael Foord98b3e762010-06-05 21:59:55 +00001329 .. attribute:: maxDiff
1330
1331 This attribute controls the maximum length of diffs output by assert
1332 methods that report diffs on failure. It defaults to 80*8 characters.
1333 Assert methods affected by this attribute are
1334 :meth:`assertSequenceEqual` (including all the sequence comparison
1335 methods that delegate to it), :meth:`assertDictEqual` and
1336 :meth:`assertMultiLineEqual`.
1337
1338 Setting ``maxDiff`` to None means that there is no maximum length of
1339 diffs.
1340
1341 .. versionadded:: 3.2
1342
1343
Benjamin Peterson52baa292009-03-24 00:56:30 +00001344 Testing frameworks can use the following methods to collect information on
1345 the test:
1346
1347
1348 .. method:: countTestCases()
1349
1350 Return the number of tests represented by this test object. For
1351 :class:`TestCase` instances, this will always be ``1``.
1352
1353
1354 .. method:: defaultTestResult()
1355
1356 Return an instance of the test result class that should be used for this
1357 test case class (if no other result instance is provided to the
1358 :meth:`run` method).
1359
1360 For :class:`TestCase` instances, this will always be an instance of
1361 :class:`TestResult`; subclasses of :class:`TestCase` should override this
1362 as necessary.
1363
1364
1365 .. method:: id()
1366
1367 Return a string identifying the specific test case. This is usually the
1368 full name of the test method, including the module and class name.
1369
1370
1371 .. method:: shortDescription()
1372
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001373 Returns a description of the test, or ``None`` if no description
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001374 has been provided. The default implementation of this method
1375 returns the first line of the test method's docstring, if available,
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001376 or ``None``.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001377
Georg Brandl419e3de2010-12-01 15:44:25 +00001378 .. versionchanged:: 3.1
Michael Foord34c94622010-02-10 15:51:42 +00001379 In 3.1 this was changed to add the test name to the short description
Georg Brandl419e3de2010-12-01 15:44:25 +00001380 even in the presence of a docstring. This caused compatibility issues
Michael Foord34c94622010-02-10 15:51:42 +00001381 with unittest extensions and adding the test name was moved to the
Georg Brandl419e3de2010-12-01 15:44:25 +00001382 :class:`TextTestResult` in Python 3.2.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001383
Georg Brandl116aa622007-08-15 14:28:22 +00001384
Georg Brandl7f01a132009-09-16 15:58:14 +00001385 .. method:: addCleanup(function, *args, **kwargs)
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001386
1387 Add a function to be called after :meth:`tearDown` to cleanup resources
1388 used during the test. Functions will be called in reverse order to the
1389 order they are added (LIFO). They are called with any arguments and
1390 keyword arguments passed into :meth:`addCleanup` when they are
1391 added.
1392
1393 If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
1394 then any cleanup functions added will still be called.
1395
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001396 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001397
1398
1399 .. method:: doCleanups()
1400
Barry Warsaw0c9fd632010-04-12 14:50:57 +00001401 This method is called unconditionally after :meth:`tearDown`, or
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001402 after :meth:`setUp` if :meth:`setUp` raises an exception.
1403
1404 It is responsible for calling all the cleanup functions added by
1405 :meth:`addCleanup`. If you need cleanup functions to be called
1406 *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
1407 yourself.
1408
1409 :meth:`doCleanups` pops methods off the stack of cleanup
1410 functions one at a time, so it can be called at any time.
1411
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001412 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001413
1414
Georg Brandl7f01a132009-09-16 15:58:14 +00001415.. class:: FunctionTestCase(testFunc, setUp=None, tearDown=None, description=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001416
1417 This class implements the portion of the :class:`TestCase` interface which
Benjamin Petersond2397752009-06-27 23:45:02 +00001418 allows the test runner to drive the test, but does not provide the methods
1419 which test code can use to check and report errors. This is used to create
1420 test cases using legacy test code, allowing it to be integrated into a
1421 :mod:`unittest`-based test framework.
Georg Brandl116aa622007-08-15 14:28:22 +00001422
1423
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001424.. _deprecated-aliases:
1425
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001426Deprecated aliases
1427##################
1428
1429For historical reasons, some of the :class:`TestCase` methods had one or more
1430aliases that are now deprecated. The following table lists the correct names
1431along with their deprecated aliases:
1432
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001433 ============================== ====================== ======================
1434 Method Name Deprecated alias Deprecated alias
1435 ============================== ====================== ======================
1436 :meth:`.assertEqual` failUnlessEqual assertEquals
1437 :meth:`.assertNotEqual` failIfEqual assertNotEquals
1438 :meth:`.assertTrue` failUnless assert\_
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001439 :meth:`.assertFalse` failIf
1440 :meth:`.assertRaises` failUnlessRaises
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001441 :meth:`.assertAlmostEqual` failUnlessAlmostEqual assertAlmostEquals
1442 :meth:`.assertNotAlmostEqual` failIfAlmostEqual assertNotAlmostEquals
Ezio Melottied3a7d22010-12-01 02:32:32 +00001443 :meth:`.assertRegex` assertRegexpMatches
1444 :meth:`.assertRaisesRegex` assertRaisesRegexp
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001445 ============================== ====================== ======================
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001446
Ezio Melotti361467e2011-04-03 17:37:58 +03001447 .. deprecated:: 3.1
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001448 the fail* aliases listed in the second column.
1449 .. deprecated:: 3.2
1450 the assert* aliases listed in the third column.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001451 .. deprecated:: 3.2
1452 ``assertRegexpMatches`` and ``assertRaisesRegexp`` have been renamed to
1453 :meth:`.assertRegex` and :meth:`.assertRaisesRegex`
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001454
1455
Benjamin Peterson52baa292009-03-24 00:56:30 +00001456.. _testsuite-objects:
1457
Benjamin Peterson52baa292009-03-24 00:56:30 +00001458Grouping tests
1459~~~~~~~~~~~~~~
1460
Georg Brandl7f01a132009-09-16 15:58:14 +00001461.. class:: TestSuite(tests=())
Georg Brandl116aa622007-08-15 14:28:22 +00001462
1463 This class represents an aggregation of individual tests cases and test suites.
1464 The class presents the interface needed by the test runner to allow it to be run
1465 as any other test case. Running a :class:`TestSuite` instance is the same as
1466 iterating over the suite, running each test individually.
1467
1468 If *tests* is given, it must be an iterable of individual test cases or other
1469 test suites that will be used to build the suite initially. Additional methods
1470 are provided to add test cases and suites to the collection later on.
1471
Benjamin Peterson14a3dd72009-05-25 00:51:58 +00001472 :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
1473 they do not actually implement a test. Instead, they are used to aggregate
1474 tests into groups of tests that should be run together. Some additional
1475 methods are available to add tests to :class:`TestSuite` instances:
Benjamin Peterson52baa292009-03-24 00:56:30 +00001476
1477
1478 .. method:: TestSuite.addTest(test)
1479
1480 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1481
1482
1483 .. method:: TestSuite.addTests(tests)
1484
1485 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
1486 instances to this test suite.
1487
Benjamin Petersond2397752009-06-27 23:45:02 +00001488 This is equivalent to iterating over *tests*, calling :meth:`addTest` for
1489 each element.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001490
1491 :class:`TestSuite` shares the following methods with :class:`TestCase`:
1492
1493
1494 .. method:: run(result)
1495
1496 Run the tests associated with this suite, collecting the result into the
1497 test result object passed as *result*. Note that unlike
1498 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
1499 be passed in.
1500
1501
1502 .. method:: debug()
1503
1504 Run the tests associated with this suite without collecting the
1505 result. This allows exceptions raised by the test to be propagated to the
1506 caller and can be used to support running tests under a debugger.
1507
1508
1509 .. method:: countTestCases()
1510
1511 Return the number of tests represented by this test object, including all
1512 individual tests and sub-suites.
1513
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001514
1515 .. method:: __iter__()
1516
1517 Tests grouped by a :class:`TestSuite` are always accessed by iteration.
1518 Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
Andrew Svetloveb973682013-08-28 21:28:38 +03001519 that this method may be called several times on a single suite (for
1520 example when counting tests or comparing for equality) so the tests
1521 returned by repeated iterations before :meth:`TestSuite.run` must be the
1522 same for each call iteration. After :meth:`TestSuite.run`, callers should
1523 not rely on the tests returned by this method unless the caller uses a
1524 subclass that overrides :meth:`TestSuite._removeTestAtIndex` to preserve
1525 test references.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001526
Georg Brandl853947a2010-01-31 18:53:23 +00001527 .. versionchanged:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001528 In earlier versions the :class:`TestSuite` accessed tests directly rather
1529 than through iteration, so overriding :meth:`__iter__` wasn't sufficient
1530 for providing tests.
1531
Andrew Svetloveb973682013-08-28 21:28:38 +03001532 .. versionchanged:: 3.4
1533 In earlier versions the :class:`TestSuite` held references to each
1534 :class:`TestCase` after :meth:`TestSuite.run`. Subclasses can restore
1535 that behavior by overriding :meth:`TestSuite._removeTestAtIndex`.
1536
Benjamin Peterson52baa292009-03-24 00:56:30 +00001537 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
1538 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1539
1540
Benjamin Peterson52baa292009-03-24 00:56:30 +00001541Loading and running tests
1542~~~~~~~~~~~~~~~~~~~~~~~~~
1543
Georg Brandl116aa622007-08-15 14:28:22 +00001544.. class:: TestLoader()
1545
Benjamin Peterson52baa292009-03-24 00:56:30 +00001546 The :class:`TestLoader` class is used to create test suites from classes and
1547 modules. Normally, there is no need to create an instance of this class; the
1548 :mod:`unittest` module provides an instance that can be shared as
Ezio Melottib8e336b2012-04-29 10:52:18 +03001549 :data:`unittest.defaultTestLoader`. Using a subclass or instance, however,
1550 allows customization of some configurable properties.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001551
Robert Collinsf920c212014-10-20 13:24:05 +13001552 :class:`TestLoader` objects have the following attributes:
1553
1554
1555 .. attribute:: errors
1556
1557 A list of the non-fatal errors encountered while loading tests. Not reset
1558 by the loader at any point. Fatal errors are signalled by the relevant
1559 a method raising an exception to the caller. Non-fatal errors are also
1560 indicated by a synthetic test that will raise the original error when
1561 run.
1562
1563 .. versionadded:: 3.5
1564
1565
Benjamin Peterson52baa292009-03-24 00:56:30 +00001566 :class:`TestLoader` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001567
Ezio Melotti9c02c2f2010-11-03 20:45:31 +00001568
Benjamin Peterson52baa292009-03-24 00:56:30 +00001569 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl116aa622007-08-15 14:28:22 +00001570
Benjamin Peterson52baa292009-03-24 00:56:30 +00001571 Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
1572 :class:`testCaseClass`.
1573
Robert Collinse02f6c22015-07-23 06:37:26 +12001574 A test case instance is created for each method named by
1575 :meth:`getTestCaseNames`. By default these are the method names
1576 beginning with ``test``. If :meth:`getTestCaseNames` returns no
1577 methods, but the :meth:`runTest` method is implemented, a single test
1578 case is created for that method instead.
1579
Benjamin Peterson52baa292009-03-24 00:56:30 +00001580
Barry Warsawd78742a2014-09-08 14:21:37 -04001581 .. method:: loadTestsFromModule(module, pattern=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001582
1583 Return a suite of all tests cases contained in the given module. This
1584 method searches *module* for classes derived from :class:`TestCase` and
1585 creates an instance of the class for each test method defined for the
1586 class.
1587
Georg Brandle720c0a2009-04-27 16:20:50 +00001588 .. note::
Benjamin Peterson52baa292009-03-24 00:56:30 +00001589
1590 While using a hierarchy of :class:`TestCase`\ -derived classes can be
1591 convenient in sharing fixtures and helper functions, defining test
1592 methods on base classes that are not intended to be instantiated
1593 directly does not play well with this method. Doing so, however, can
1594 be useful when the fixtures are different and defined in subclasses.
1595
Benjamin Petersond2397752009-06-27 23:45:02 +00001596 If a module provides a ``load_tests`` function it will be called to
1597 load the tests. This allows modules to customize test loading.
Barry Warsawd78742a2014-09-08 14:21:37 -04001598 This is the `load_tests protocol`_. The *pattern* argument is passed as
1599 the third argument to ``load_tests``.
Benjamin Petersond2397752009-06-27 23:45:02 +00001600
Georg Brandl853947a2010-01-31 18:53:23 +00001601 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001602 Support for ``load_tests`` added.
1603
Barry Warsawd78742a2014-09-08 14:21:37 -04001604 .. versionchanged:: 3.5
1605 The undocumented and unofficial *use_load_tests* default argument is
1606 deprecated and ignored, although it is still accepted for backward
1607 compatibility. The method also now accepts a keyword-only argument
1608 *pattern* which is passed to ``load_tests`` as the third argument.
1609
Benjamin Peterson52baa292009-03-24 00:56:30 +00001610
Georg Brandl7f01a132009-09-16 15:58:14 +00001611 .. method:: loadTestsFromName(name, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001612
1613 Return a suite of all tests cases given a string specifier.
1614
1615 The specifier *name* is a "dotted name" that may resolve either to a
1616 module, a test case class, a test method within a test case class, a
1617 :class:`TestSuite` instance, or a callable object which returns a
1618 :class:`TestCase` or :class:`TestSuite` instance. These checks are
1619 applied in the order listed here; that is, a method on a possible test
1620 case class will be picked up as "a test method within a test case class",
1621 rather than "a callable object".
1622
1623 For example, if you have a module :mod:`SampleTests` containing a
1624 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
1625 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
Benjamin Petersond2397752009-06-27 23:45:02 +00001626 specifier ``'SampleTests.SampleTestCase'`` would cause this method to
1627 return a suite which will run all three test methods. Using the specifier
1628 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
1629 suite which will run only the :meth:`test_two` test method. The specifier
1630 can refer to modules and packages which have not been imported; they will
1631 be imported as a side-effect.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001632
1633 The method optionally resolves *name* relative to the given *module*.
1634
Robert Collins659dd622014-10-30 08:27:27 +13001635 .. versionchanged:: 3.5
1636 If an :exc:`ImportError` or :exc:`AttributeError` occurs while traversing
1637 *name* then a synthetic test that raises that error when run will be
1638 returned. These errors are included in the errors accumulated by
1639 self.errors.
1640
Benjamin Peterson52baa292009-03-24 00:56:30 +00001641
Georg Brandl7f01a132009-09-16 15:58:14 +00001642 .. method:: loadTestsFromNames(names, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001643
1644 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
1645 than a single name. The return value is a test suite which supports all
1646 the tests defined for each name.
1647
1648
1649 .. method:: getTestCaseNames(testCaseClass)
1650
1651 Return a sorted sequence of method names found within *testCaseClass*;
1652 this should be a subclass of :class:`TestCase`.
1653
Benjamin Petersond2397752009-06-27 23:45:02 +00001654
1655 .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)
1656
Larry Hastings3732ed22014-03-15 21:13:56 -07001657 Find all the test modules by recursing into subdirectories from the
1658 specified start directory, and return a TestSuite object containing them.
1659 Only test files that match *pattern* will be loaded. (Using shell style
1660 pattern matching.) Only module names that are importable (i.e. are valid
1661 Python identifiers) will be loaded.
Benjamin Petersond2397752009-06-27 23:45:02 +00001662
1663 All test modules must be importable from the top level of the project. If
1664 the start directory is not the top level directory then the top level
1665 directory must be specified separately.
1666
Barry Warsawd78742a2014-09-08 14:21:37 -04001667 If importing a module fails, for example due to a syntax error, then
1668 this will be recorded as a single error and discovery will continue. If
1669 the import failure is due to :exc:`SkipTest` being raised, it will be
1670 recorded as a skip instead of an error.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001671
Barry Warsawd78742a2014-09-08 14:21:37 -04001672 If a package (a directory containing a file named :file:`__init__.py`) is
1673 found, the package will be checked for a ``load_tests`` function. If this
Robert Collinsbf2bda32014-11-05 03:09:01 +13001674 exists then it will be called
1675 ``package.load_tests(loader, tests, pattern)``. Test discovery takes care
1676 to ensure that a package is only checked for tests once during an
1677 invocation, even if the load_tests function itself calls
1678 ``loader.discover``.
Benjamin Petersond2397752009-06-27 23:45:02 +00001679
Barry Warsawd78742a2014-09-08 14:21:37 -04001680 If ``load_tests`` exists then discovery does *not* recurse into the
1681 package, ``load_tests`` is responsible for loading all tests in the
1682 package.
Benjamin Petersond2397752009-06-27 23:45:02 +00001683
1684 The pattern is deliberately not stored as a loader attribute so that
1685 packages can continue discovery themselves. *top_level_dir* is stored so
1686 ``load_tests`` does not need to pass this argument in to
1687 ``loader.discover()``.
1688
Benjamin Petersonb48af542010-04-11 20:43:16 +00001689 *start_dir* can be a dotted module name as well as a directory.
1690
Georg Brandl853947a2010-01-31 18:53:23 +00001691 .. versionadded:: 3.2
1692
Ezio Melottieae2b382013-03-01 14:47:50 +02001693 .. versionchanged:: 3.4
Ezio Melotti67ddcca2013-03-27 20:13:59 +02001694 Modules that raise :exc:`SkipTest` on import are recorded as skips,
Larry Hastings3732ed22014-03-15 21:13:56 -07001695 not errors.
1696 Discovery works for :term:`namespace packages <namespace package>`.
1697 Paths are sorted before being imported so that execution order is
1698 the same even if the underlying file system's ordering is not
1699 dependent on file name.
Michael Foord80cbc9e2013-03-18 17:50:12 -07001700
Barry Warsawd78742a2014-09-08 14:21:37 -04001701 .. versionchanged:: 3.5
1702 Found packages are now checked for ``load_tests`` regardless of
1703 whether their path matches *pattern*, because it is impossible for
1704 a package name to match the default pattern.
1705
Benjamin Petersond2397752009-06-27 23:45:02 +00001706
Benjamin Peterson52baa292009-03-24 00:56:30 +00001707 The following attributes of a :class:`TestLoader` can be configured either by
1708 subclassing or assignment on an instance:
1709
1710
1711 .. attribute:: testMethodPrefix
1712
1713 String giving the prefix of method names which will be interpreted as test
1714 methods. The default value is ``'test'``.
1715
1716 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
1717 methods.
1718
1719
1720 .. attribute:: sortTestMethodsUsing
1721
1722 Function to be used to compare method names when sorting them in
1723 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
1724
1725
1726 .. attribute:: suiteClass
1727
1728 Callable object that constructs a test suite from a list of tests. No
1729 methods on the resulting object are needed. The default value is the
1730 :class:`TestSuite` class.
1731
1732 This affects all the :meth:`loadTestsFrom\*` methods.
1733
1734
Benjamin Peterson52baa292009-03-24 00:56:30 +00001735.. class:: TestResult
1736
1737 This class is used to compile information about which tests have succeeded
1738 and which have failed.
1739
1740 A :class:`TestResult` object stores the results of a set of tests. The
1741 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
1742 properly recorded; test authors do not need to worry about recording the
1743 outcome of tests.
1744
1745 Testing frameworks built on top of :mod:`unittest` may want access to the
1746 :class:`TestResult` object generated by running a set of tests for reporting
1747 purposes; a :class:`TestResult` instance is returned by the
1748 :meth:`TestRunner.run` method for this purpose.
1749
1750 :class:`TestResult` instances have the following attributes that will be of
1751 interest when inspecting the results of running a set of tests:
1752
1753
1754 .. attribute:: errors
1755
1756 A list containing 2-tuples of :class:`TestCase` instances and strings
1757 holding formatted tracebacks. Each tuple represents a test which raised an
1758 unexpected exception.
1759
Benjamin Peterson52baa292009-03-24 00:56:30 +00001760 .. attribute:: failures
1761
1762 A list containing 2-tuples of :class:`TestCase` instances and strings
1763 holding formatted tracebacks. Each tuple represents a test where a failure
Ezio Melottie2202362013-09-07 15:19:30 +03001764 was explicitly signalled using the :meth:`TestCase.assert\*` methods.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001765
Benjamin Peterson52baa292009-03-24 00:56:30 +00001766 .. attribute:: skipped
1767
1768 A list containing 2-tuples of :class:`TestCase` instances and strings
1769 holding the reason for skipping the test.
1770
Benjamin Peterson70e32c82009-03-24 01:00:11 +00001771 .. versionadded:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +00001772
1773 .. attribute:: expectedFailures
1774
Georg Brandl6faee4e2010-09-21 14:48:28 +00001775 A list containing 2-tuples of :class:`TestCase` instances and strings
1776 holding formatted tracebacks. Each tuple represents an expected failure
Benjamin Peterson52baa292009-03-24 00:56:30 +00001777 of the test case.
1778
1779 .. attribute:: unexpectedSuccesses
1780
1781 A list containing :class:`TestCase` instances that were marked as expected
1782 failures, but succeeded.
1783
1784 .. attribute:: shouldStop
1785
1786 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1787
Benjamin Peterson52baa292009-03-24 00:56:30 +00001788 .. attribute:: testsRun
1789
1790 The total number of tests run so far.
1791
Georg Brandl12037202010-12-02 22:35:25 +00001792 .. attribute:: buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +00001793
1794 If set to true, ``sys.stdout`` and ``sys.stderr`` will be buffered in between
1795 :meth:`startTest` and :meth:`stopTest` being called. Collected output will
1796 only be echoed onto the real ``sys.stdout`` and ``sys.stderr`` if the test
1797 fails or errors. Any output is also attached to the failure / error message.
1798
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001799 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001800
Benjamin Petersonb48af542010-04-11 20:43:16 +00001801 .. attribute:: failfast
1802
1803 If set to true :meth:`stop` will be called on the first failure or error,
1804 halting the test run.
1805
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001806 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001807
Robert Collinsf0c819a2015-03-06 13:46:35 +13001808 .. attribute:: tb_locals
1809
1810 If set to true then local variables will be shown in tracebacks.
1811
1812 .. versionadded:: 3.5
Benjamin Petersonb48af542010-04-11 20:43:16 +00001813
Benjamin Peterson52baa292009-03-24 00:56:30 +00001814 .. method:: wasSuccessful()
1815
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001816 Return ``True`` if all tests run so far have passed, otherwise returns
1817 ``False``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001818
Gregory P. Smith5a6d4bf2014-01-20 01:11:18 -08001819 .. versionchanged:: 3.4
1820 Returns ``False`` if there were any :attr:`unexpectedSuccesses`
1821 from tests marked with the :func:`expectedFailure` decorator.
1822
Benjamin Peterson52baa292009-03-24 00:56:30 +00001823 .. method:: stop()
1824
1825 This method can be called to signal that the set of tests being run should
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001826 be aborted by setting the :attr:`shouldStop` attribute to ``True``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001827 :class:`TestRunner` objects should respect this flag and return without
1828 running any additional tests.
1829
1830 For example, this feature is used by the :class:`TextTestRunner` class to
1831 stop the test framework when the user signals an interrupt from the
1832 keyboard. Interactive tools which provide :class:`TestRunner`
1833 implementations can use this in a similar manner.
1834
1835 The following methods of the :class:`TestResult` class are used to maintain
1836 the internal data structures, and may be extended in subclasses to support
1837 additional reporting requirements. This is particularly useful in building
1838 tools which support interactive reporting while tests are being run.
1839
1840
1841 .. method:: startTest(test)
1842
1843 Called when the test case *test* is about to be run.
1844
Benjamin Peterson52baa292009-03-24 00:56:30 +00001845 .. method:: stopTest(test)
1846
1847 Called after the test case *test* has been executed, regardless of the
1848 outcome.
1849
Terry Jan Reedyf98021c2014-04-11 14:11:11 -04001850 .. method:: startTestRun()
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001851
1852 Called once before any tests are executed.
1853
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001854 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001855
1856
Terry Jan Reedyf98021c2014-04-11 14:11:11 -04001857 .. method:: stopTestRun()
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001858
Ezio Melotti176d6c42010-01-27 20:58:07 +00001859 Called once after all tests are executed.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001860
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001861 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001862
1863
Benjamin Peterson52baa292009-03-24 00:56:30 +00001864 .. method:: addError(test, err)
1865
Ezio Melottie64a91a2013-09-07 15:23:36 +03001866 Called when the test case *test* raises an unexpected exception. *err* is a
Benjamin Peterson52baa292009-03-24 00:56:30 +00001867 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1868 traceback)``.
1869
1870 The default implementation appends a tuple ``(test, formatted_err)`` to
1871 the instance's :attr:`errors` attribute, where *formatted_err* is a
1872 formatted traceback derived from *err*.
1873
1874
1875 .. method:: addFailure(test, err)
1876
Benjamin Petersond2397752009-06-27 23:45:02 +00001877 Called when the test case *test* signals a failure. *err* is a tuple of
1878 the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001879
1880 The default implementation appends a tuple ``(test, formatted_err)`` to
1881 the instance's :attr:`failures` attribute, where *formatted_err* is a
1882 formatted traceback derived from *err*.
1883
1884
1885 .. method:: addSuccess(test)
1886
1887 Called when the test case *test* succeeds.
1888
1889 The default implementation does nothing.
1890
1891
1892 .. method:: addSkip(test, reason)
1893
1894 Called when the test case *test* is skipped. *reason* is the reason the
1895 test gave for skipping.
1896
1897 The default implementation appends a tuple ``(test, reason)`` to the
1898 instance's :attr:`skipped` attribute.
1899
1900
1901 .. method:: addExpectedFailure(test, err)
1902
1903 Called when the test case *test* fails, but was marked with the
1904 :func:`expectedFailure` decorator.
1905
1906 The default implementation appends a tuple ``(test, formatted_err)`` to
1907 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1908 is a formatted traceback derived from *err*.
1909
1910
1911 .. method:: addUnexpectedSuccess(test)
1912
1913 Called when the test case *test* was marked with the
1914 :func:`expectedFailure` decorator, but succeeded.
1915
1916 The default implementation appends the test to the instance's
1917 :attr:`unexpectedSuccesses` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +00001918
Georg Brandl67b21b72010-08-17 15:07:14 +00001919
Antoine Pitrouc9b3ef22013-03-20 20:16:47 +01001920 .. method:: addSubTest(test, subtest, outcome)
1921
1922 Called when a subtest finishes. *test* is the test case
1923 corresponding to the test method. *subtest* is a custom
1924 :class:`TestCase` instance describing the subtest.
1925
1926 If *outcome* is :const:`None`, the subtest succeeded. Otherwise,
1927 it failed with an exception where *outcome* is a tuple of the form
1928 returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
1929
1930 The default implementation does nothing when the outcome is a
1931 success, and records subtest failures as normal failures.
1932
1933 .. versionadded:: 3.4
1934
1935
Michael Foord34c94622010-02-10 15:51:42 +00001936.. class:: TextTestResult(stream, descriptions, verbosity)
1937
Georg Brandl67b21b72010-08-17 15:07:14 +00001938 A concrete implementation of :class:`TestResult` used by the
1939 :class:`TextTestRunner`.
Michael Foord34c94622010-02-10 15:51:42 +00001940
Georg Brandl67b21b72010-08-17 15:07:14 +00001941 .. versionadded:: 3.2
1942 This class was previously named ``_TextTestResult``. The old name still
1943 exists as an alias but is deprecated.
1944
Georg Brandl116aa622007-08-15 14:28:22 +00001945
1946.. data:: defaultTestLoader
1947
1948 Instance of the :class:`TestLoader` class intended to be shared. If no
1949 customization of the :class:`TestLoader` is needed, this instance can be used
1950 instead of repeatedly creating new instances.
1951
1952
Ezio Melotti9c939bc2013-05-07 09:46:30 +03001953.. class:: TextTestRunner(stream=None, descriptions=True, verbosity=1, failfast=False, \
Robert Collinsf0c819a2015-03-06 13:46:35 +13001954 buffer=False, resultclass=None, warnings=None, *, tb_locals=False)
Georg Brandl116aa622007-08-15 14:28:22 +00001955
Michael Foordd218e952011-01-03 12:55:11 +00001956 A basic test runner implementation that outputs results to a stream. If *stream*
Éric Araujo941afed2011-09-01 02:47:34 +02001957 is ``None``, the default, :data:`sys.stderr` is used as the output stream. This class
Georg Brandl116aa622007-08-15 14:28:22 +00001958 has a few configurable parameters, but is essentially very simple. Graphical
Robert Collinsf0c819a2015-03-06 13:46:35 +13001959 applications which run test suites should provide alternate implementations. Such
1960 implementations should accept ``**kwargs`` as the interface to construct runners
1961 changes when features are added to unittest.
Georg Brandl116aa622007-08-15 14:28:22 +00001962
Ezio Melotti60901872010-12-01 00:56:10 +00001963 By default this runner shows :exc:`DeprecationWarning`,
Senthil Kumaran409ea5d2014-02-08 14:28:03 -08001964 :exc:`PendingDeprecationWarning`, :exc:`ResourceWarning` and
Larry Hastingsad88d7a2014-02-10 04:26:10 -08001965 :exc:`ImportWarning` even if they are :ref:`ignored by default
1966 <warning-ignored>`. Deprecation warnings caused by :ref:`deprecated unittest
1967 methods <deprecated-aliases>` are also special-cased and, when the warning
1968 filters are ``'default'`` or ``'always'``, they will appear only once
1969 per-module, in order to avoid too many warning messages. This behavior can
1970 be overridden using the :option:`-Wd` or :option:`-Wa` options and leaving
1971 *warnings* to ``None``.
Ezio Melotti60901872010-12-01 00:56:10 +00001972
Michael Foordd218e952011-01-03 12:55:11 +00001973 .. versionchanged:: 3.2
1974 Added the ``warnings`` argument.
1975
1976 .. versionchanged:: 3.2
Éric Araujo941afed2011-09-01 02:47:34 +02001977 The default stream is set to :data:`sys.stderr` at instantiation time rather
Michael Foordd218e952011-01-03 12:55:11 +00001978 than import time.
1979
Robert Collinsf0c819a2015-03-06 13:46:35 +13001980 .. versionchanged:: 3.5
1981 Added the tb_locals parameter.
1982
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001983 .. method:: _makeResult()
Georg Brandl116aa622007-08-15 14:28:22 +00001984
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001985 This method returns the instance of ``TestResult`` used by :meth:`run`.
1986 It is not intended to be called directly, but can be overridden in
1987 subclasses to provide a custom ``TestResult``.
1988
Michael Foord34c94622010-02-10 15:51:42 +00001989 ``_makeResult()`` instantiates the class or callable passed in the
1990 ``TextTestRunner`` constructor as the ``resultclass`` argument. It
Benjamin Petersonb48af542010-04-11 20:43:16 +00001991 defaults to :class:`TextTestResult` if no ``resultclass`` is provided.
Michael Foord34c94622010-02-10 15:51:42 +00001992 The result class is instantiated with the following arguments::
1993
1994 stream, descriptions, verbosity
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001995
Michael Foord4d1639f2013-12-29 23:38:55 +00001996 .. method:: run(test)
1997
1998 This method is the main public interface to the `TextTestRunner`. This
1999 method takes a :class:`TestSuite` or :class:`TestCase` instance. A
2000 :class:`TestResult` is created by calling
2001 :func:`_makeResult` and the test(s) are run and the
2002 results printed to stdout.
2003
Ezio Melotti60901872010-12-01 00:56:10 +00002004
Georg Brandl419e3de2010-12-01 15:44:25 +00002005.. function:: main(module='__main__', defaultTest=None, argv=None, testRunner=None, \
Ezio Melotti40dcb1d2011-03-10 13:46:50 +02002006 testLoader=unittest.defaultTestLoader, exit=True, verbosity=1, \
Georg Brandl419e3de2010-12-01 15:44:25 +00002007 failfast=None, catchbreak=None, buffer=None, warnings=None)
Georg Brandl116aa622007-08-15 14:28:22 +00002008
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03002009 A command-line program that loads a set of tests from *module* and runs them;
2010 this is primarily for making test modules conveniently executable.
2011 The simplest use for this function is to include the following line at the
2012 end of a test script::
Georg Brandl116aa622007-08-15 14:28:22 +00002013
2014 if __name__ == '__main__':
2015 unittest.main()
2016
Benjamin Petersond2397752009-06-27 23:45:02 +00002017 You can run tests with more detailed information by passing in the verbosity
2018 argument::
2019
2020 if __name__ == '__main__':
2021 unittest.main(verbosity=2)
2022
R David Murray6e731b02014-01-02 13:43:02 -05002023 The *defaultTest* argument is either the name of a single test or an
2024 iterable of test names to run if no test names are specified via *argv*. If
2025 not specified or ``None`` and no test names are provided via *argv*, all
2026 tests found in *module* are run.
R David Murray12e930f2014-01-02 13:37:26 -05002027
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03002028 The *argv* argument can be a list of options passed to the program, with the
2029 first element being the program name. If not specified or ``None``,
2030 the values of :data:`sys.argv` are used.
2031
Georg Brandl116aa622007-08-15 14:28:22 +00002032 The *testRunner* argument can either be a test runner class or an already
Benjamin Peterson25c95f12009-05-08 20:42:26 +00002033 created instance of it. By default ``main`` calls :func:`sys.exit` with
2034 an exit code indicating success or failure of the tests run.
2035
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03002036 The *testLoader* argument has to be a :class:`TestLoader` instance,
2037 and defaults to :data:`defaultTestLoader`.
2038
Benjamin Peterson25c95f12009-05-08 20:42:26 +00002039 ``main`` supports being used from the interactive interpreter by passing in the
2040 argument ``exit=False``. This displays the result on standard output without
2041 calling :func:`sys.exit`::
2042
2043 >>> from unittest import main
2044 >>> main(module='test_module', exit=False)
2045
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03002046 The *failfast*, *catchbreak* and *buffer* parameters have the same
Éric Araujo76338ec2010-11-26 23:46:18 +00002047 effect as the same-name `command-line options`_.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002048
Ezio Melotti60901872010-12-01 00:56:10 +00002049 The *warning* argument specifies the :ref:`warning filter <warning-filter>`
2050 that should be used while running the tests. If it's not specified, it will
2051 remain ``None`` if a :option:`-W` option is passed to :program:`python`,
2052 otherwise it will be set to ``'default'``.
2053
Benjamin Peterson25c95f12009-05-08 20:42:26 +00002054 Calling ``main`` actually returns an instance of the ``TestProgram`` class.
2055 This stores the result of the tests run as the ``result`` attribute.
2056
Éric Araujo971dc012010-12-16 03:13:05 +00002057 .. versionchanged:: 3.1
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03002058 The *exit* parameter was added.
Éric Araujo971dc012010-12-16 03:13:05 +00002059
Georg Brandl853947a2010-01-31 18:53:23 +00002060 .. versionchanged:: 3.2
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03002061 The *verbosity*, *failfast*, *catchbreak*, *buffer*
2062 and *warnings* parameters were added.
Benjamin Petersond2397752009-06-27 23:45:02 +00002063
Chris Jerdonekccbc26a2013-02-23 15:44:46 -08002064 .. versionchanged:: 3.4
2065 The *defaultTest* parameter was changed to also accept an iterable of
2066 test names.
2067
Benjamin Petersond2397752009-06-27 23:45:02 +00002068
2069load_tests Protocol
2070###################
2071
Georg Brandl853947a2010-01-31 18:53:23 +00002072.. versionadded:: 3.2
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00002073
Benjamin Petersond2397752009-06-27 23:45:02 +00002074Modules or packages can customize how tests are loaded from them during normal
2075test runs or test discovery by implementing a function called ``load_tests``.
2076
2077If a test module defines ``load_tests`` it will be called by
2078:meth:`TestLoader.loadTestsFromModule` with the following arguments::
2079
Barry Warsawd78742a2014-09-08 14:21:37 -04002080 load_tests(loader, standard_tests, pattern)
2081
2082where *pattern* is passed straight through from ``loadTestsFromModule``. It
2083defaults to ``None``.
Benjamin Petersond2397752009-06-27 23:45:02 +00002084
2085It should return a :class:`TestSuite`.
2086
2087*loader* is the instance of :class:`TestLoader` doing the loading.
2088*standard_tests* are the tests that would be loaded by default from the
2089module. It is common for test modules to only want to add or remove tests
2090from the standard set of tests.
2091The third argument is used when loading packages as part of test discovery.
2092
2093A typical ``load_tests`` function that loads tests from a specific set of
2094:class:`TestCase` classes may look like::
2095
2096 test_cases = (TestCase1, TestCase2, TestCase3)
2097
2098 def load_tests(loader, tests, pattern):
2099 suite = TestSuite()
2100 for test_class in test_cases:
2101 tests = loader.loadTestsFromTestCase(test_class)
2102 suite.addTests(tests)
2103 return suite
2104
Barry Warsawd78742a2014-09-08 14:21:37 -04002105If discovery is started in a directory containing a package, either from the
2106command line or by calling :meth:`TestLoader.discover`, then the package
2107:file:`__init__.py` will be checked for ``load_tests``. If that function does
2108not exist, discovery will recurse into the package as though it were just
2109another directory. Otherwise, discovery of the package's tests will be left up
2110to ``load_tests`` which is called with the following arguments::
Benjamin Petersond2397752009-06-27 23:45:02 +00002111
2112 load_tests(loader, standard_tests, pattern)
2113
2114This should return a :class:`TestSuite` representing all the tests
2115from the package. (``standard_tests`` will only contain tests
2116collected from :file:`__init__.py`.)
2117
2118Because the pattern is passed into ``load_tests`` the package is free to
2119continue (and potentially modify) test discovery. A 'do nothing'
2120``load_tests`` function for a test package would look like::
2121
2122 def load_tests(loader, standard_tests, pattern):
2123 # top level directory cached on loader instance
2124 this_dir = os.path.dirname(__file__)
2125 package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
2126 standard_tests.addTests(package_tests)
2127 return standard_tests
Benjamin Petersonb48af542010-04-11 20:43:16 +00002128
Barry Warsawd78742a2014-09-08 14:21:37 -04002129.. versionchanged:: 3.5
2130 Discovery no longer checks package names for matching *pattern* due to the
2131 impossibility of package names matching the default pattern.
2132
2133
Benjamin Petersonb48af542010-04-11 20:43:16 +00002134
2135Class and Module Fixtures
2136-------------------------
2137
2138Class and module level fixtures are implemented in :class:`TestSuite`. When
2139the test suite encounters a test from a new class then :meth:`tearDownClass`
2140from the previous class (if there is one) is called, followed by
2141:meth:`setUpClass` from the new class.
2142
2143Similarly if a test is from a different module from the previous test then
2144``tearDownModule`` from the previous module is run, followed by
2145``setUpModule`` from the new module.
2146
2147After all the tests have run the final ``tearDownClass`` and
2148``tearDownModule`` are run.
2149
2150Note that shared fixtures do not play well with [potential] features like test
2151parallelization and they break test isolation. They should be used with care.
2152
2153The default ordering of tests created by the unittest test loaders is to group
2154all tests from the same modules and classes together. This will lead to
2155``setUpClass`` / ``setUpModule`` (etc) being called exactly once per class and
2156module. If you randomize the order, so that tests from different modules and
2157classes are adjacent to each other, then these shared fixture functions may be
2158called multiple times in a single test run.
2159
2160Shared fixtures are not intended to work with suites with non-standard
2161ordering. A ``BaseTestSuite`` still exists for frameworks that don't want to
2162support shared fixtures.
2163
2164If there are any exceptions raised during one of the shared fixture functions
2165the test is reported as an error. Because there is no corresponding test
2166instance an ``_ErrorHolder`` object (that has the same interface as a
2167:class:`TestCase`) is created to represent the error. If you are just using
2168the standard unittest test runner then this detail doesn't matter, but if you
2169are a framework author it may be relevant.
2170
2171
2172setUpClass and tearDownClass
2173~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2174
2175These must be implemented as class methods::
2176
2177 import unittest
2178
2179 class Test(unittest.TestCase):
2180 @classmethod
2181 def setUpClass(cls):
2182 cls._connection = createExpensiveConnectionObject()
2183
2184 @classmethod
2185 def tearDownClass(cls):
2186 cls._connection.destroy()
2187
2188If you want the ``setUpClass`` and ``tearDownClass`` on base classes called
2189then you must call up to them yourself. The implementations in
2190:class:`TestCase` are empty.
2191
2192If an exception is raised during a ``setUpClass`` then the tests in the class
2193are not run and the ``tearDownClass`` is not run. Skipped classes will not
Michael Foord98b3e762010-06-05 21:59:55 +00002194have ``setUpClass`` or ``tearDownClass`` run. If the exception is a
Ezio Melotti265281a2013-03-27 20:11:55 +02002195:exc:`SkipTest` exception then the class will be reported as having been skipped
Michael Foord98b3e762010-06-05 21:59:55 +00002196instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002197
2198
2199setUpModule and tearDownModule
2200~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2201
2202These should be implemented as functions::
2203
2204 def setUpModule():
2205 createConnection()
2206
2207 def tearDownModule():
2208 closeConnection()
2209
2210If an exception is raised in a ``setUpModule`` then none of the tests in the
Michael Foord98b3e762010-06-05 21:59:55 +00002211module will be run and the ``tearDownModule`` will not be run. If the exception is a
Ezio Melotti265281a2013-03-27 20:11:55 +02002212:exc:`SkipTest` exception then the module will be reported as having been skipped
Michael Foord98b3e762010-06-05 21:59:55 +00002213instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002214
2215
2216Signal Handling
2217---------------
2218
Georg Brandl419e3de2010-12-01 15:44:25 +00002219.. versionadded:: 3.2
2220
Éric Araujo8acb67c2010-11-26 23:31:07 +00002221The :option:`-c/--catch <unittest -c>` command-line option to unittest,
Éric Araujo76338ec2010-11-26 23:46:18 +00002222along with the ``catchbreak`` parameter to :func:`unittest.main()`, provide
2223more friendly handling of control-C during a test run. With catch break
2224behavior enabled control-C will allow the currently running test to complete,
2225and the test run will then end and report all the results so far. A second
2226control-c will raise a :exc:`KeyboardInterrupt` in the usual way.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002227
Michael Foordde4ceab2010-04-25 19:53:49 +00002228The control-c handling signal handler attempts to remain compatible with code or
2229tests that install their own :const:`signal.SIGINT` handler. If the ``unittest``
2230handler is called but *isn't* the installed :const:`signal.SIGINT` handler,
2231i.e. it has been replaced by the system under test and delegated to, then it
2232calls the default handler. This will normally be the expected behavior by code
2233that replaces an installed handler and delegates to it. For individual tests
2234that need ``unittest`` control-c handling disabled the :func:`removeHandler`
2235decorator can be used.
2236
2237There are a few utility functions for framework authors to enable control-c
2238handling functionality within test frameworks.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002239
2240.. function:: installHandler()
2241
2242 Install the control-c handler. When a :const:`signal.SIGINT` is received
2243 (usually in response to the user pressing control-c) all registered results
2244 have :meth:`~TestResult.stop` called.
2245
Michael Foord469b1f02010-04-26 23:41:26 +00002246
Benjamin Petersonb48af542010-04-11 20:43:16 +00002247.. function:: registerResult(result)
2248
2249 Register a :class:`TestResult` object for control-c handling. Registering a
2250 result stores a weak reference to it, so it doesn't prevent the result from
2251 being garbage collected.
2252
Michael Foordde4ceab2010-04-25 19:53:49 +00002253 Registering a :class:`TestResult` object has no side-effects if control-c
2254 handling is not enabled, so test frameworks can unconditionally register
2255 all results they create independently of whether or not handling is enabled.
2256
Michael Foord469b1f02010-04-26 23:41:26 +00002257
Benjamin Petersonb48af542010-04-11 20:43:16 +00002258.. function:: removeResult(result)
2259
2260 Remove a registered result. Once a result has been removed then
2261 :meth:`~TestResult.stop` will no longer be called on that result object in
2262 response to a control-c.
2263
Michael Foord469b1f02010-04-26 23:41:26 +00002264
Michael Foordde4ceab2010-04-25 19:53:49 +00002265.. function:: removeHandler(function=None)
2266
2267 When called without arguments this function removes the control-c handler
2268 if it has been installed. This function can also be used as a test decorator
2269 to temporarily remove the handler whilst the test is being executed::
2270
2271 @unittest.removeHandler
2272 def test_signal_handling(self):
2273 ...