blob: 63106242129fa6e6a391124941d3a9efa489d99b [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
50 `Simple Smalltalk Testing: With Patterns <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
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000054 `Nose <http://code.google.com/p/python-nose/>`_ 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
Antoine Pitrou2c5e9502013-01-20 01:29:39 +010058 `The Python Testing Tools Taxonomy <http://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
70 `Buildbot <http://buildbot.net/trac>`_, `Jenkins <http://jenkins-ci.org>`_
71 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
83Here is a short script to test three functions from the :mod:`random` module::
84
85 import random
86 import unittest
87
88 class TestSequenceFunctions(unittest.TestCase):
89
90 def setUp(self):
Benjamin Petersonbe0e1772009-07-25 01:02:01 +000091 self.seq = list(range(10))
Georg Brandl116aa622007-08-15 14:28:22 +000092
Benjamin Peterson52baa292009-03-24 00:56:30 +000093 def test_shuffle(self):
Georg Brandl116aa622007-08-15 14:28:22 +000094 # make sure the shuffled sequence does not lose any elements
95 random.shuffle(self.seq)
96 self.seq.sort()
Benjamin Petersonbe0e1772009-07-25 01:02:01 +000097 self.assertEqual(self.seq, list(range(10)))
Georg Brandl116aa622007-08-15 14:28:22 +000098
Benjamin Peterson847a4112010-03-14 15:04:17 +000099 # should raise an exception for an immutable sequence
100 self.assertRaises(TypeError, random.shuffle, (1,2,3))
101
Benjamin Peterson52baa292009-03-24 00:56:30 +0000102 def test_choice(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000103 element = random.choice(self.seq)
Benjamin Peterson847a4112010-03-14 15:04:17 +0000104 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000105
Benjamin Peterson52baa292009-03-24 00:56:30 +0000106 def test_sample(self):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000107 with self.assertRaises(ValueError):
108 random.sample(self.seq, 20)
Georg Brandl116aa622007-08-15 14:28:22 +0000109 for element in random.sample(self.seq, 5):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000110 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000111
112 if __name__ == '__main__':
113 unittest.main()
114
Benjamin Peterson52baa292009-03-24 00:56:30 +0000115A testcase is created by subclassing :class:`unittest.TestCase`. The three
Georg Brandl116aa622007-08-15 14:28:22 +0000116individual tests are defined with methods whose names start with the letters
117``test``. This naming convention informs the test runner about which methods
118represent tests.
119
Benjamin Peterson52baa292009-03-24 00:56:30 +0000120The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
Michael Foord34c94622010-02-10 15:51:42 +0000121expected result; :meth:`~TestCase.assertTrue` to verify a condition; or
Benjamin Peterson52baa292009-03-24 00:56:30 +0000122:meth:`~TestCase.assertRaises` to verify that an expected exception gets raised.
123These methods are used instead of the :keyword:`assert` statement so the test
124runner can accumulate all test results and produce a report.
Georg Brandl116aa622007-08-15 14:28:22 +0000125
Benjamin Peterson52baa292009-03-24 00:56:30 +0000126When a :meth:`~TestCase.setUp` method is defined, the test runner will run that
127method prior to each test. Likewise, if a :meth:`~TestCase.tearDown` method is
128defined, the test runner will invoke that method after each test. In the
129example, :meth:`~TestCase.setUp` was used to create a fresh sequence for each
130test.
Georg Brandl116aa622007-08-15 14:28:22 +0000131
132The final block shows a simple way to run the tests. :func:`unittest.main`
Éric Araujo713d3032010-11-18 16:38:46 +0000133provides a command-line interface to the test script. When run from the command
Georg Brandl116aa622007-08-15 14:28:22 +0000134line, the above script produces an output that looks like this::
135
136 ...
137 ----------------------------------------------------------------------
138 Ran 3 tests in 0.000s
139
140 OK
141
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100142Passing the ``-v`` option to your test script will instruct :func:`unittest.main`
143to enable a higher level of verbosity, and produce the following output::
Georg Brandl116aa622007-08-15 14:28:22 +0000144
Ezio Melottid59e44a2010-02-28 03:46:13 +0000145 test_choice (__main__.TestSequenceFunctions) ... ok
146 test_sample (__main__.TestSequenceFunctions) ... ok
147 test_shuffle (__main__.TestSequenceFunctions) ... ok
Georg Brandl116aa622007-08-15 14:28:22 +0000148
149 ----------------------------------------------------------------------
150 Ran 3 tests in 0.110s
151
152 OK
153
154The above examples show the most commonly used :mod:`unittest` features which
155are sufficient to meet many everyday testing needs. The remainder of the
156documentation explores the full feature set from first principles.
157
Benjamin Petersonb48af542010-04-11 20:43:16 +0000158
159.. _unittest-command-line-interface:
160
Éric Araujo76338ec2010-11-26 23:46:18 +0000161Command-Line Interface
Benjamin Petersonb48af542010-04-11 20:43:16 +0000162----------------------
163
164The unittest module can be used from the command line to run tests from
165modules, classes or even individual test methods::
166
167 python -m unittest test_module1 test_module2
168 python -m unittest test_module.TestClass
169 python -m unittest test_module.TestClass.test_method
170
171You can pass in a list with any combination of module names, and fully
172qualified class or method names.
173
Michael Foord37d120a2010-12-04 01:11:21 +0000174Test modules can be specified by file path as well::
175
176 python -m unittest tests/test_something.py
177
178This allows you to use the shell filename completion to specify the test module.
179The file specified must still be importable as a module. The path is converted
180to a module name by removing the '.py' and converting path separators into '.'.
181If you want to execute a test file that isn't importable as a module you should
182execute the file directly instead.
183
Benjamin Petersonb48af542010-04-11 20:43:16 +0000184You can run tests with more detail (higher verbosity) by passing in the -v flag::
185
186 python -m unittest -v test_module
187
Michael Foord086f3082010-11-21 21:28:01 +0000188When executed without arguments :ref:`unittest-test-discovery` is started::
189
190 python -m unittest
191
Éric Araujo713d3032010-11-18 16:38:46 +0000192For a list of all the command-line options::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000193
194 python -m unittest -h
195
Georg Brandl67b21b72010-08-17 15:07:14 +0000196.. versionchanged:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000197 In earlier versions it was only possible to run individual test methods and
198 not modules or classes.
199
200
Éric Araujo76338ec2010-11-26 23:46:18 +0000201Command-line options
202~~~~~~~~~~~~~~~~~~~~
Benjamin Petersonb48af542010-04-11 20:43:16 +0000203
Éric Araujod3309df2010-11-21 03:09:17 +0000204:program:`unittest` supports these command-line options:
Benjamin Petersonb48af542010-04-11 20:43:16 +0000205
Éric Araujo713d3032010-11-18 16:38:46 +0000206.. program:: unittest
Benjamin Petersonb48af542010-04-11 20:43:16 +0000207
Éric Araujo713d3032010-11-18 16:38:46 +0000208.. cmdoption:: -b, --buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +0000209
Éric Araujo713d3032010-11-18 16:38:46 +0000210 The standard output and standard error streams are buffered during the test
211 run. Output during a passing test is discarded. Output is echoed normally
212 on test fail or error and is added to the failure messages.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000213
Éric Araujo713d3032010-11-18 16:38:46 +0000214.. cmdoption:: -c, --catch
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000215
Éric Araujo713d3032010-11-18 16:38:46 +0000216 Control-C during the test run waits for the current test to end and then
217 reports all the results so far. A second control-C raises the normal
218 :exc:`KeyboardInterrupt` exception.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000219
Éric Araujo713d3032010-11-18 16:38:46 +0000220 See `Signal Handling`_ for the functions that provide this functionality.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000221
Éric Araujo713d3032010-11-18 16:38:46 +0000222.. cmdoption:: -f, --failfast
223
224 Stop the test run on the first error or failure.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000225
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000226.. versionadded:: 3.2
Éric Araujod6c5f742010-12-16 00:07:01 +0000227 The command-line options ``-b``, ``-c`` and ``-f`` were added.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000228
229The command line can also be used for test discovery, for running all of the
230tests in a project or just a subset.
231
232
233.. _unittest-test-discovery:
234
235Test Discovery
236--------------
237
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000238.. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000239
Ezio Melotti3d995842011-03-08 16:17:35 +0200240Unittest supports simple test discovery. In order to be compatible with test
241discovery, all of the test files must be :ref:`modules <tut-modules>` or
242:ref:`packages <tut-packages>` importable from the top-level directory of
243the project (this means that their filenames must be valid
244:ref:`identifiers <identifiers>`).
Benjamin Petersonb48af542010-04-11 20:43:16 +0000245
246Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
Éric Araujo713d3032010-11-18 16:38:46 +0000247used from the command line. The basic command-line usage is::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000248
249 cd project_directory
250 python -m unittest discover
251
Michael Foord086f3082010-11-21 21:28:01 +0000252.. note::
253
254 As a shortcut, ``python -m unittest`` is the equivalent of
255 ``python -m unittest discover``. If you want to pass arguments to test
Éric Araujo941afed2011-09-01 02:47:34 +0200256 discovery the ``discover`` sub-command must be used explicitly.
Michael Foord086f3082010-11-21 21:28:01 +0000257
Benjamin Petersonb48af542010-04-11 20:43:16 +0000258The ``discover`` sub-command has the following options:
259
Éric Araujo713d3032010-11-18 16:38:46 +0000260.. program:: unittest discover
261
262.. cmdoption:: -v, --verbose
263
264 Verbose output
265
Chris Jerdonekd69ad552013-02-21 18:54:43 -0800266.. cmdoption:: -s, --start-directory directory
Éric Araujo713d3032010-11-18 16:38:46 +0000267
Éric Araujo941afed2011-09-01 02:47:34 +0200268 Directory to start discovery (``.`` default)
Éric Araujo713d3032010-11-18 16:38:46 +0000269
Chris Jerdonekd69ad552013-02-21 18:54:43 -0800270.. cmdoption:: -p, --pattern pattern
Éric Araujo713d3032010-11-18 16:38:46 +0000271
Éric Araujo941afed2011-09-01 02:47:34 +0200272 Pattern to match test files (``test*.py`` default)
Éric Araujo713d3032010-11-18 16:38:46 +0000273
Chris Jerdonekd69ad552013-02-21 18:54:43 -0800274.. cmdoption:: -t, --top-level-directory directory
Éric Araujo713d3032010-11-18 16:38:46 +0000275
276 Top level directory of project (defaults to start directory)
Benjamin Petersonb48af542010-04-11 20:43:16 +0000277
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000278The :option:`-s`, :option:`-p`, and :option:`-t` options can be passed in
279as positional arguments in that order. The following two command lines
280are equivalent::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000281
282 python -m unittest discover -s project_directory -p '*_test.py'
283 python -m unittest discover project_directory '*_test.py'
284
Michael Foord16f3e902010-05-08 15:13:42 +0000285As well as being a path it is possible to pass a package name, for example
286``myproject.subpackage.test``, as the start directory. The package name you
287supply will then be imported and its location on the filesystem will be used
288as the start directory.
289
290.. caution::
291
Senthil Kumaran916bd382010-10-15 12:55:19 +0000292 Test discovery loads tests by importing them. Once test discovery has found
293 all the test files from the start directory you specify it turns the paths
294 into package names to import. For example :file:`foo/bar/baz.py` will be
Michael Foord16f3e902010-05-08 15:13:42 +0000295 imported as ``foo.bar.baz``.
296
297 If you have a package installed globally and attempt test discovery on
298 a different copy of the package then the import *could* happen from the
299 wrong place. If this happens test discovery will warn you and exit.
300
301 If you supply the start directory as a package name rather than a
302 path to a directory then discover assumes that whichever location it
303 imports from is the location you intended, so you will not get the
304 warning.
305
Benjamin Petersonb48af542010-04-11 20:43:16 +0000306Test modules and packages can customize test loading and discovery by through
307the `load_tests protocol`_.
308
309
Georg Brandl116aa622007-08-15 14:28:22 +0000310.. _organizing-tests:
311
312Organizing test code
313--------------------
314
315The basic building blocks of unit testing are :dfn:`test cases` --- single
316scenarios that must be set up and checked for correctness. In :mod:`unittest`,
Raymond Hettinger833ad0e2011-02-06 21:00:38 +0000317test cases are represented by :class:`unittest.TestCase` instances.
318To make your own test cases you must write subclasses of
319:class:`TestCase` or use :class:`FunctionTestCase`.
Georg Brandl116aa622007-08-15 14:28:22 +0000320
Georg Brandl116aa622007-08-15 14:28:22 +0000321The testing code of a :class:`TestCase` instance should be entirely self
322contained, such that it can be run either in isolation or in arbitrary
323combination with any number of other test cases.
324
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100325The simplest :class:`TestCase` subclass will simply implement a test method
326(i.e. a method whose name starts with ``test``) in order to perform specific
327testing code::
Georg Brandl116aa622007-08-15 14:28:22 +0000328
329 import unittest
330
331 class DefaultWidgetSizeTestCase(unittest.TestCase):
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100332 def test_default_widget_size(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000333 widget = Widget('The widget')
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100334 self.assertEqual(widget.size(), (50, 50))
Georg Brandl116aa622007-08-15 14:28:22 +0000335
Sandro Tosi41b24042012-01-21 10:59:37 +0100336Note that in order to test something, we use one of the :meth:`assert\*`
Benjamin Petersond2397752009-06-27 23:45:02 +0000337methods provided by the :class:`TestCase` base class. If the test fails, an
338exception will be raised, and :mod:`unittest` will identify the test case as a
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100339:dfn:`failure`. Any other exceptions will be treated as :dfn:`errors`.
Georg Brandl116aa622007-08-15 14:28:22 +0000340
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100341Tests can be numerous, and their set-up can be repetitive. Luckily, we
342can factor out set-up code by implementing a method called
343:meth:`~TestCase.setUp`, which the testing framework will automatically
344call for every single test we run::
Georg Brandl116aa622007-08-15 14:28:22 +0000345
346 import unittest
347
348 class SimpleWidgetTestCase(unittest.TestCase):
349 def setUp(self):
350 self.widget = Widget('The widget')
351
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100352 def test_default_widget_size(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000353 self.assertEqual(self.widget.size(), (50,50),
354 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000355
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100356 def test_widget_resize(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000357 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000358 self.assertEqual(self.widget.size(), (100,150),
359 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000360
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100361.. note::
362 The order in which the various tests will be run is determined
363 by sorting the test method names with respect to the built-in
364 ordering for strings.
365
Benjamin Peterson52baa292009-03-24 00:56:30 +0000366If the :meth:`~TestCase.setUp` method raises an exception while the test is
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100367running, the framework will consider the test to have suffered an error, and
368the test method will not be executed.
Georg Brandl116aa622007-08-15 14:28:22 +0000369
Benjamin Peterson52baa292009-03-24 00:56:30 +0000370Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100371after the test method has been run::
Georg Brandl116aa622007-08-15 14:28:22 +0000372
373 import unittest
374
375 class SimpleWidgetTestCase(unittest.TestCase):
376 def setUp(self):
377 self.widget = Widget('The widget')
378
379 def tearDown(self):
380 self.widget.dispose()
Georg Brandl116aa622007-08-15 14:28:22 +0000381
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100382If :meth:`~TestCase.setUp` succeeded, :meth:`~TestCase.tearDown` will be
383run whether the test method succeeded or not.
Georg Brandl116aa622007-08-15 14:28:22 +0000384
385Such a working environment for the testing code is called a :dfn:`fixture`.
386
Georg Brandl116aa622007-08-15 14:28:22 +0000387Test case instances are grouped together according to the features they test.
388:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100389represented by :mod:`unittest`'s :class:`TestSuite` class. In most cases,
390calling :func:`unittest.main` will do the right thing and collect all the
391module's test cases for you, and then execute them.
Georg Brandl116aa622007-08-15 14:28:22 +0000392
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100393However, should you want to customize the building of your test suite,
394you can do it yourself::
Georg Brandl116aa622007-08-15 14:28:22 +0000395
396 def suite():
397 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000398 suite.addTest(WidgetTestCase('test_default_size'))
399 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000400 return suite
401
Georg Brandl116aa622007-08-15 14:28:22 +0000402You can place the definitions of test cases and test suites in the same modules
403as the code they are to test (such as :file:`widget.py`), but there are several
404advantages to placing the test code in a separate module, such as
405:file:`test_widget.py`:
406
407* The test module can be run standalone from the command line.
408
409* The test code can more easily be separated from shipped code.
410
411* There is less temptation to change test code to fit the code it tests without
412 a good reason.
413
414* Test code should be modified much less frequently than the code it tests.
415
416* Tested code can be refactored more easily.
417
418* Tests for modules written in C must be in separate modules anyway, so why not
419 be consistent?
420
421* If the testing strategy changes, there is no need to change the source code.
422
423
424.. _legacy-unit-tests:
425
426Re-using old test code
427----------------------
428
429Some users will find that they have existing test code that they would like to
430run from :mod:`unittest`, without converting every old test function to a
431:class:`TestCase` subclass.
432
433For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
434This subclass of :class:`TestCase` can be used to wrap an existing test
435function. Set-up and tear-down functions can also be provided.
436
437Given the following test function::
438
439 def testSomething():
440 something = makeSomething()
441 assert something.name is not None
442 # ...
443
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100444one can create an equivalent test case instance as follows, with optional
445set-up and tear-down methods::
Georg Brandl116aa622007-08-15 14:28:22 +0000446
447 testcase = unittest.FunctionTestCase(testSomething,
448 setUp=makeSomethingDB,
449 tearDown=deleteSomethingDB)
450
Georg Brandl116aa622007-08-15 14:28:22 +0000451.. note::
452
Benjamin Petersond2397752009-06-27 23:45:02 +0000453 Even though :class:`FunctionTestCase` can be used to quickly convert an
454 existing test base over to a :mod:`unittest`\ -based system, this approach is
455 not recommended. Taking the time to set up proper :class:`TestCase`
456 subclasses will make future test refactorings infinitely easier.
Georg Brandl116aa622007-08-15 14:28:22 +0000457
Benjamin Peterson52baa292009-03-24 00:56:30 +0000458In some cases, the existing tests may have been written using the :mod:`doctest`
459module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
460automatically build :class:`unittest.TestSuite` instances from the existing
461:mod:`doctest`\ -based tests.
462
Georg Brandl116aa622007-08-15 14:28:22 +0000463
Benjamin Peterson5254c042009-03-23 22:25:03 +0000464.. _unittest-skipping:
465
466Skipping tests and expected failures
467------------------------------------
468
Michael Foordf5c851a2010-02-05 21:48:03 +0000469.. versionadded:: 3.1
470
Benjamin Peterson5254c042009-03-23 22:25:03 +0000471Unittest supports skipping individual test methods and even whole classes of
472tests. In addition, it supports marking a test as a "expected failure," a test
473that is broken and will fail, but shouldn't be counted as a failure on a
474:class:`TestResult`.
475
476Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
477or one of its conditional variants.
478
Ezio Melottifed69ba2013-03-01 21:26:04 +0200479Basic skipping looks like this::
Benjamin Peterson5254c042009-03-23 22:25:03 +0000480
481 class MyTestCase(unittest.TestCase):
482
483 @unittest.skip("demonstrating skipping")
484 def test_nothing(self):
485 self.fail("shouldn't happen")
486
Benjamin Petersond2397752009-06-27 23:45:02 +0000487 @unittest.skipIf(mylib.__version__ < (1, 3),
488 "not supported in this library version")
Benjamin Petersonded31c42009-03-30 15:04:16 +0000489 def test_format(self):
490 # Tests that work for only a certain version of the library.
491 pass
492
493 @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
494 def test_windows_support(self):
495 # windows specific testing code
496 pass
497
Ezio Melottifed69ba2013-03-01 21:26:04 +0200498This is the output of running the example above in verbose mode::
Benjamin Peterson5254c042009-03-23 22:25:03 +0000499
Benjamin Petersonded31c42009-03-30 15:04:16 +0000500 test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000501 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
Benjamin Petersonded31c42009-03-30 15:04:16 +0000502 test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000503
504 ----------------------------------------------------------------------
Benjamin Petersonded31c42009-03-30 15:04:16 +0000505 Ran 3 tests in 0.005s
506
507 OK (skipped=3)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000508
Ezio Melottifed69ba2013-03-01 21:26:04 +0200509Classes can be skipped just like methods::
Benjamin Peterson5254c042009-03-23 22:25:03 +0000510
Sandro Tosi317075d2012-03-31 18:34:59 +0200511 @unittest.skip("showing class skipping")
Benjamin Peterson5254c042009-03-23 22:25:03 +0000512 class MySkippedTestCase(unittest.TestCase):
513 def test_not_run(self):
514 pass
515
Benjamin Peterson52baa292009-03-24 00:56:30 +0000516:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
517that needs to be set up is not available.
518
Benjamin Peterson5254c042009-03-23 22:25:03 +0000519Expected failures use the :func:`expectedFailure` decorator. ::
520
521 class ExpectedFailureTestCase(unittest.TestCase):
522 @unittest.expectedFailure
523 def test_fail(self):
524 self.assertEqual(1, 0, "broken")
525
526It's easy to roll your own skipping decorators by making a decorator that calls
527:func:`skip` on the test when it wants it to be skipped. This decorator skips
Ezio Melottifed69ba2013-03-01 21:26:04 +0200528the test unless the passed object has a certain attribute::
Benjamin Peterson5254c042009-03-23 22:25:03 +0000529
530 def skipUnlessHasattr(obj, attr):
531 if hasattr(obj, attr):
532 return lambda func: func
Ezio Melotti265281a2013-03-27 20:11:55 +0200533 return unittest.skip("{!r} doesn't have {!r}".format(obj, attr))
Benjamin Peterson5254c042009-03-23 22:25:03 +0000534
535The following decorators implement test skipping and expected failures:
536
Georg Brandl8a1caa22010-07-29 16:01:11 +0000537.. decorator:: skip(reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000538
539 Unconditionally skip the decorated test. *reason* should describe why the
540 test is being skipped.
541
Georg Brandl8a1caa22010-07-29 16:01:11 +0000542.. decorator:: skipIf(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000543
544 Skip the decorated test if *condition* is true.
545
Georg Brandl8a1caa22010-07-29 16:01:11 +0000546.. decorator:: skipUnless(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000547
Georg Brandl6faee4e2010-09-21 14:48:28 +0000548 Skip the decorated test unless *condition* is true.
Benjamin Peterson5254c042009-03-23 22:25:03 +0000549
Georg Brandl8a1caa22010-07-29 16:01:11 +0000550.. decorator:: expectedFailure
Benjamin Peterson5254c042009-03-23 22:25:03 +0000551
552 Mark the test as an expected failure. If the test fails when run, the test
553 is not counted as a failure.
554
Ezio Melotti265281a2013-03-27 20:11:55 +0200555.. exception:: SkipTest(reason)
556
557 This exception is raised to skip a test.
558
559 Usually you can use :meth:`TestCase.skipTest` or one of the skipping
560 decorators instead of raising this directly.
561
Benjamin Petersonb48af542010-04-11 20:43:16 +0000562Skipped tests will not have :meth:`setUp` or :meth:`tearDown` run around them.
563Skipped classes will not have :meth:`setUpClass` or :meth:`tearDownClass` run.
564
Benjamin Peterson5254c042009-03-23 22:25:03 +0000565
Antoine Pitrouc9b3ef22013-03-20 20:16:47 +0100566.. _subtests:
567
568Distinguishing test iterations using subtests
569---------------------------------------------
570
571.. versionadded:: 3.4
572
573When some of your tests differ only by a some very small differences, for
574instance some parameters, unittest allows you to distinguish them inside
575the body of a test method using the :meth:`~TestCase.subTest` context manager.
576
577For example, the following test::
578
579 class NumbersTest(unittest.TestCase):
580
581 def test_even(self):
582 """
583 Test that numbers between 0 and 5 are all even.
584 """
585 for i in range(0, 6):
586 with self.subTest(i=i):
587 self.assertEqual(i % 2, 0)
588
589will produce the following output::
590
591 ======================================================================
592 FAIL: test_even (__main__.NumbersTest) (i=1)
593 ----------------------------------------------------------------------
594 Traceback (most recent call last):
595 File "subtests.py", line 32, in test_even
596 self.assertEqual(i % 2, 0)
597 AssertionError: 1 != 0
598
599 ======================================================================
600 FAIL: test_even (__main__.NumbersTest) (i=3)
601 ----------------------------------------------------------------------
602 Traceback (most recent call last):
603 File "subtests.py", line 32, in test_even
604 self.assertEqual(i % 2, 0)
605 AssertionError: 1 != 0
606
607 ======================================================================
608 FAIL: test_even (__main__.NumbersTest) (i=5)
609 ----------------------------------------------------------------------
610 Traceback (most recent call last):
611 File "subtests.py", line 32, in test_even
612 self.assertEqual(i % 2, 0)
613 AssertionError: 1 != 0
614
615Without using a subtest, execution would stop after the first failure,
616and the error would be less easy to diagnose because the value of ``i``
617wouldn't be displayed::
618
619 ======================================================================
620 FAIL: test_even (__main__.NumbersTest)
621 ----------------------------------------------------------------------
622 Traceback (most recent call last):
623 File "subtests.py", line 32, in test_even
624 self.assertEqual(i % 2, 0)
625 AssertionError: 1 != 0
626
627
Georg Brandl116aa622007-08-15 14:28:22 +0000628.. _unittest-contents:
629
630Classes and functions
631---------------------
632
Benjamin Peterson52baa292009-03-24 00:56:30 +0000633This section describes in depth the API of :mod:`unittest`.
634
635
636.. _testcase-objects:
637
638Test cases
639~~~~~~~~~~
Georg Brandl116aa622007-08-15 14:28:22 +0000640
Georg Brandl7f01a132009-09-16 15:58:14 +0000641.. class:: TestCase(methodName='runTest')
Georg Brandl116aa622007-08-15 14:28:22 +0000642
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100643 Instances of the :class:`TestCase` class represent the logical test units
Georg Brandl116aa622007-08-15 14:28:22 +0000644 in the :mod:`unittest` universe. This class is intended to be used as a base
645 class, with specific tests being implemented by concrete subclasses. This class
646 implements the interface needed by the test runner to allow it to drive the
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100647 tests, and methods that the test code can use to check for and report various
Georg Brandl116aa622007-08-15 14:28:22 +0000648 kinds of failure.
649
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100650 Each instance of :class:`TestCase` will run a single base method: the method
651 named *methodName*. However, the standard implementation of the default
652 *methodName*, ``runTest()``, will run every method starting with ``test``
653 as an individual test, and count successes and failures accordingly.
654 Therefore, in most uses of :class:`TestCase`, you will neither change
655 the *methodName* nor reimplement the default ``runTest()`` method.
Georg Brandl116aa622007-08-15 14:28:22 +0000656
Michael Foord1341bb02011-03-14 19:01:46 -0400657 .. versionchanged:: 3.2
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100658 :class:`TestCase` can be instantiated successfully without providing a
659 *methodName*. This makes it easier to experiment with :class:`TestCase`
660 from the interactive interpreter.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000661
662 :class:`TestCase` instances provide three groups of methods: one group used
663 to run the test, another used by the test implementation to check conditions
664 and report failures, and some inquiry methods allowing information about the
665 test itself to be gathered.
666
667 Methods in the first group (running the test) are:
668
Benjamin Peterson52baa292009-03-24 00:56:30 +0000669 .. method:: setUp()
670
671 Method called to prepare the test fixture. This is called immediately
672 before calling the test method; any exception raised by this method will
673 be considered an error rather than a test failure. The default
674 implementation does nothing.
675
676
677 .. method:: tearDown()
678
679 Method called immediately after the test method has been called and the
680 result recorded. This is called even if the test method raised an
681 exception, so the implementation in subclasses may need to be particularly
682 careful about checking internal state. Any exception raised by this
683 method will be considered an error rather than a test failure. This
684 method will only be called if the :meth:`setUp` succeeds, regardless of
685 the outcome of the test method. The default implementation does nothing.
686
687
Benjamin Petersonb48af542010-04-11 20:43:16 +0000688 .. method:: setUpClass()
689
690 A class method called before tests in an individual class run.
691 ``setUpClass`` is called with the class as the only argument
692 and must be decorated as a :func:`classmethod`::
693
694 @classmethod
695 def setUpClass(cls):
696 ...
697
698 See `Class and Module Fixtures`_ for more details.
699
700 .. versionadded:: 3.2
701
702
703 .. method:: tearDownClass()
704
705 A class method called after tests in an individual class have run.
706 ``tearDownClass`` is called with the class as the only argument
707 and must be decorated as a :meth:`classmethod`::
708
709 @classmethod
710 def tearDownClass(cls):
711 ...
712
713 See `Class and Module Fixtures`_ for more details.
714
715 .. versionadded:: 3.2
716
717
Georg Brandl7f01a132009-09-16 15:58:14 +0000718 .. method:: run(result=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000719
Antoine Pitrou2c5e9502013-01-20 01:29:39 +0100720 Run the test, collecting the result into the :class:`TestResult` object
721 passed as *result*. If *result* is omitted or ``None``, a temporary
722 result object is created (by calling the :meth:`defaultTestResult`
723 method) and used. The result object is returned to :meth:`run`'s
724 caller.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000725
726 The same effect may be had by simply calling the :class:`TestCase`
727 instance.
728
Michael Foord1341bb02011-03-14 19:01:46 -0400729 .. versionchanged:: 3.3
730 Previous versions of ``run`` did not return the result. Neither did
731 calling an instance.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000732
Benjamin Petersone549ead2009-03-28 21:42:05 +0000733 .. method:: skipTest(reason)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000734
Stefan Kraha5bf3f52010-05-19 16:09:41 +0000735 Calling this during a test method or :meth:`setUp` skips the current
Benjamin Peterson52baa292009-03-24 00:56:30 +0000736 test. See :ref:`unittest-skipping` for more information.
737
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000738 .. versionadded:: 3.1
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000739
Benjamin Peterson52baa292009-03-24 00:56:30 +0000740
Antoine Pitrouc9b3ef22013-03-20 20:16:47 +0100741 .. method:: subTest(msg=None, **params)
742
743 Return a context manager which executes the enclosed code block as a
744 subtest. *msg* and *params* are optional, arbitrary values which are
745 displayed whenever a subtest fails, allowing you to identify them
746 clearly.
747
748 A test case can contain any number of subtest declarations, and
749 they can be arbitrarily nested.
750
751 See :ref:`subtests` for more information.
752
753 .. versionadded:: 3.4
754
755
Benjamin Peterson52baa292009-03-24 00:56:30 +0000756 .. method:: debug()
757
758 Run the test without collecting the result. This allows exceptions raised
759 by the test to be propagated to the caller, and can be used to support
760 running tests under a debugger.
761
Ezio Melotti22170ed2010-11-20 09:57:27 +0000762 .. _assert-methods:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000763
Ezio Melotti4370b302010-11-03 20:39:14 +0000764 The :class:`TestCase` class provides a number of methods to check for and
765 report failures, such as:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000766
Ezio Melotti4370b302010-11-03 20:39:14 +0000767 +-----------------------------------------+-----------------------------+---------------+
768 | Method | Checks that | New in |
769 +=========================================+=============================+===============+
770 | :meth:`assertEqual(a, b) | ``a == b`` | |
771 | <TestCase.assertEqual>` | | |
772 +-----------------------------------------+-----------------------------+---------------+
773 | :meth:`assertNotEqual(a, b) | ``a != b`` | |
774 | <TestCase.assertNotEqual>` | | |
775 +-----------------------------------------+-----------------------------+---------------+
776 | :meth:`assertTrue(x) | ``bool(x) is True`` | |
777 | <TestCase.assertTrue>` | | |
778 +-----------------------------------------+-----------------------------+---------------+
779 | :meth:`assertFalse(x) | ``bool(x) is False`` | |
780 | <TestCase.assertFalse>` | | |
781 +-----------------------------------------+-----------------------------+---------------+
782 | :meth:`assertIs(a, b) | ``a is b`` | 3.1 |
783 | <TestCase.assertIs>` | | |
784 +-----------------------------------------+-----------------------------+---------------+
785 | :meth:`assertIsNot(a, b) | ``a is not b`` | 3.1 |
786 | <TestCase.assertIsNot>` | | |
787 +-----------------------------------------+-----------------------------+---------------+
788 | :meth:`assertIsNone(x) | ``x is None`` | 3.1 |
789 | <TestCase.assertIsNone>` | | |
790 +-----------------------------------------+-----------------------------+---------------+
791 | :meth:`assertIsNotNone(x) | ``x is not None`` | 3.1 |
792 | <TestCase.assertIsNotNone>` | | |
793 +-----------------------------------------+-----------------------------+---------------+
794 | :meth:`assertIn(a, b) | ``a in b`` | 3.1 |
795 | <TestCase.assertIn>` | | |
796 +-----------------------------------------+-----------------------------+---------------+
797 | :meth:`assertNotIn(a, b) | ``a not in b`` | 3.1 |
798 | <TestCase.assertNotIn>` | | |
799 +-----------------------------------------+-----------------------------+---------------+
800 | :meth:`assertIsInstance(a, b) | ``isinstance(a, b)`` | 3.2 |
801 | <TestCase.assertIsInstance>` | | |
802 +-----------------------------------------+-----------------------------+---------------+
803 | :meth:`assertNotIsInstance(a, b) | ``not isinstance(a, b)`` | 3.2 |
804 | <TestCase.assertNotIsInstance>` | | |
805 +-----------------------------------------+-----------------------------+---------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000806
Ezio Melottib4dc2502011-05-06 15:01:41 +0300807 All the assert methods accept a *msg* argument that, if specified, is used
808 as the error message on failure (see also :data:`longMessage`).
809 Note that the *msg* keyword argument can be passed to :meth:`assertRaises`,
810 :meth:`assertRaisesRegex`, :meth:`assertWarns`, :meth:`assertWarnsRegex`
811 only when they are used as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000812
Michael Foorde180d392011-01-28 19:51:48 +0000813 .. method:: assertEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000814
Michael Foorde180d392011-01-28 19:51:48 +0000815 Test that *first* and *second* are equal. If the values do not
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000816 compare equal, the test will fail.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000817
Michael Foorde180d392011-01-28 19:51:48 +0000818 In addition, if *first* and *second* are the exact same type and one of
Michael Foord02834952010-02-08 23:10:39 +0000819 list, tuple, dict, set, frozenset or str or any type that a subclass
Ezio Melotti9ecb6be2012-01-16 08:28:54 +0200820 registers with :meth:`addTypeEqualityFunc` the type-specific equality
Michael Foord02834952010-02-08 23:10:39 +0000821 function will be called in order to generate a more useful default
Ezio Melotti22170ed2010-11-20 09:57:27 +0000822 error message (see also the :ref:`list of type-specific methods
823 <type-specific-methods>`).
Ezio Melotti4841fd62010-11-05 15:43:40 +0000824
Raymond Hettinger35a88362009-04-09 00:08:24 +0000825 .. versionchanged:: 3.1
Ezio Melotti9ecb6be2012-01-16 08:28:54 +0200826 Added the automatic calling of type-specific equality function.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000827
Michael Foord28a817e2010-02-09 00:03:57 +0000828 .. versionchanged:: 3.2
829 :meth:`assertMultiLineEqual` added as the default type equality
830 function for comparing strings.
Michael Foord02834952010-02-08 23:10:39 +0000831
Benjamin Peterson52baa292009-03-24 00:56:30 +0000832
Michael Foorde180d392011-01-28 19:51:48 +0000833 .. method:: assertNotEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000834
Michael Foorde180d392011-01-28 19:51:48 +0000835 Test that *first* and *second* are not equal. If the values do
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000836 compare equal, the test will fail.
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000837
Ezio Melotti4370b302010-11-03 20:39:14 +0000838 .. method:: assertTrue(expr, msg=None)
Ezio Melotti4841fd62010-11-05 15:43:40 +0000839 assertFalse(expr, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000840
Ezio Melotti4841fd62010-11-05 15:43:40 +0000841 Test that *expr* is true (or false).
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000842
Ezio Melotti4841fd62010-11-05 15:43:40 +0000843 Note that this is equivalent to ``bool(expr) is True`` and not to ``expr
844 is True`` (use ``assertIs(expr, True)`` for the latter). This method
845 should also be avoided when more specific methods are available (e.g.
846 ``assertEqual(a, b)`` instead of ``assertTrue(a == b)``), because they
847 provide a better error message in case of failure.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000848
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000849
Michael Foorde180d392011-01-28 19:51:48 +0000850 .. method:: assertIs(first, second, msg=None)
851 assertIsNot(first, second, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000852
Michael Foorde180d392011-01-28 19:51:48 +0000853 Test that *first* and *second* evaluate (or don't evaluate) to the
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000854 same object.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000855
Raymond Hettinger35a88362009-04-09 00:08:24 +0000856 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000857
858
Ezio Melotti4370b302010-11-03 20:39:14 +0000859 .. method:: assertIsNone(expr, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000860 assertIsNotNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000861
Ezio Melotti9794a262010-11-04 14:52:13 +0000862 Test that *expr* is (or is not) None.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000863
Ezio Melotti4370b302010-11-03 20:39:14 +0000864 .. versionadded:: 3.1
Benjamin Petersonb48af542010-04-11 20:43:16 +0000865
866
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000867 .. method:: assertIn(first, second, msg=None)
868 assertNotIn(first, second, msg=None)
869
Ezio Melotti4841fd62010-11-05 15:43:40 +0000870 Test that *first* is (or is not) in *second*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000871
Raymond Hettinger35a88362009-04-09 00:08:24 +0000872 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000873
874
Ezio Melotti9c02c2f2010-11-03 20:45:31 +0000875 .. method:: assertIsInstance(obj, cls, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000876 assertNotIsInstance(obj, cls, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000877
Ezio Melotti9794a262010-11-04 14:52:13 +0000878 Test that *obj* is (or is not) an instance of *cls* (which can be a
879 class or a tuple of classes, as supported by :func:`isinstance`).
Ezio Melotti80a61e82011-12-19 07:04:48 +0200880 To check for the exact type, use :func:`assertIs(type(obj), cls) <assertIs>`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000881
Ezio Melotti4370b302010-11-03 20:39:14 +0000882 .. versionadded:: 3.2
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000883
884
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000885
Antoine Pitrou7c89ae22013-09-15 02:01:39 +0200886 It is also possible to check the production of exceptions, warnings and
887 log messages using the following methods:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000888
Ezio Melotti4370b302010-11-03 20:39:14 +0000889 +---------------------------------------------------------+--------------------------------------+------------+
890 | Method | Checks that | New in |
891 +=========================================================+======================================+============+
Éric Araujo941afed2011-09-01 02:47:34 +0200892 | :meth:`assertRaises(exc, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *exc* | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000893 | <TestCase.assertRaises>` | | |
894 +---------------------------------------------------------+--------------------------------------+------------+
Ezio Melotti560a7782013-09-13 22:17:40 +0300895 | :meth:`assertRaisesRegex(exc, r, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *exc* | 3.1 |
896 | <TestCase.assertRaisesRegex>` | and the message matches regex *r* | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000897 +---------------------------------------------------------+--------------------------------------+------------+
Éric Araujo941afed2011-09-01 02:47:34 +0200898 | :meth:`assertWarns(warn, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *warn* | 3.2 |
Ezio Melotti4370b302010-11-03 20:39:14 +0000899 | <TestCase.assertWarns>` | | |
900 +---------------------------------------------------------+--------------------------------------+------------+
Ezio Melotti560a7782013-09-13 22:17:40 +0300901 | :meth:`assertWarnsRegex(warn, r, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *warn* | 3.2 |
902 | <TestCase.assertWarnsRegex>` | and the message matches regex *r* | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000903 +---------------------------------------------------------+--------------------------------------+------------+
Georg Brandled007d52013-11-24 16:09:26 +0100904 | :meth:`assertLogs(logger, level) | The ``with`` block logs on *logger* | 3.4 |
905 | <TestCase.assertLogs>` | with minimum *level* | |
Antoine Pitrou7c89ae22013-09-15 02:01:39 +0200906 +---------------------------------------------------------+--------------------------------------+------------+
Benjamin Peterson52baa292009-03-24 00:56:30 +0000907
Georg Brandl7f01a132009-09-16 15:58:14 +0000908 .. method:: assertRaises(exception, callable, *args, **kwds)
Ezio Melottib4dc2502011-05-06 15:01:41 +0300909 assertRaises(exception, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000910
911 Test that an exception is raised when *callable* is called with any
912 positional or keyword arguments that are also passed to
913 :meth:`assertRaises`. The test passes if *exception* is raised, is an
914 error if another exception is raised, or fails if no exception is raised.
915 To catch any of a group of exceptions, a tuple containing the exception
916 classes may be passed as *exception*.
917
Ezio Melottib4dc2502011-05-06 15:01:41 +0300918 If only the *exception* and possibly the *msg* arguments are given,
919 return a context manager so that the code under test can be written
920 inline rather than as a function::
Benjamin Petersonded31c42009-03-30 15:04:16 +0000921
Michael Foord41531f22010-02-05 21:13:40 +0000922 with self.assertRaises(SomeException):
Benjamin Petersonded31c42009-03-30 15:04:16 +0000923 do_something()
924
Ezio Melottib4dc2502011-05-06 15:01:41 +0300925 When used as a context manager, :meth:`assertRaises` accepts the
926 additional keyword argument *msg*.
927
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000928 The context manager will store the caught exception object in its
Ezio Melotti49008232010-02-08 21:57:48 +0000929 :attr:`exception` attribute. This can be useful if the intention
Michael Foord41531f22010-02-05 21:13:40 +0000930 is to perform additional checks on the exception raised::
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000931
Georg Brandl8a1caa22010-07-29 16:01:11 +0000932 with self.assertRaises(SomeException) as cm:
933 do_something()
Michael Foord41531f22010-02-05 21:13:40 +0000934
Georg Brandl8a1caa22010-07-29 16:01:11 +0000935 the_exception = cm.exception
936 self.assertEqual(the_exception.error_code, 3)
Michael Foord41531f22010-02-05 21:13:40 +0000937
Ezio Melotti49008232010-02-08 21:57:48 +0000938 .. versionchanged:: 3.1
Benjamin Petersonded31c42009-03-30 15:04:16 +0000939 Added the ability to use :meth:`assertRaises` as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000940
Ezio Melotti49008232010-02-08 21:57:48 +0000941 .. versionchanged:: 3.2
942 Added the :attr:`exception` attribute.
943
Ezio Melottib4dc2502011-05-06 15:01:41 +0300944 .. versionchanged:: 3.3
945 Added the *msg* keyword argument when used as a context manager.
946
Benjamin Peterson52baa292009-03-24 00:56:30 +0000947
Ezio Melottied3a7d22010-12-01 02:32:32 +0000948 .. method:: assertRaisesRegex(exception, regex, callable, *args, **kwds)
Ezio Melottib4dc2502011-05-06 15:01:41 +0300949 assertRaisesRegex(exception, regex, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000950
Ezio Melottied3a7d22010-12-01 02:32:32 +0000951 Like :meth:`assertRaises` but also tests that *regex* matches
952 on the string representation of the raised exception. *regex* may be
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000953 a regular expression object or a string containing a regular expression
954 suitable for use by :func:`re.search`. Examples::
955
Terry Jan Reedyc4565a92013-06-29 13:15:43 -0400956 self.assertRaisesRegex(ValueError, "invalid literal for.*XYZ'$",
Ezio Melottied3a7d22010-12-01 02:32:32 +0000957 int, 'XYZ')
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000958
959 or::
960
Ezio Melottied3a7d22010-12-01 02:32:32 +0000961 with self.assertRaisesRegex(ValueError, 'literal'):
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000962 int('XYZ')
963
Georg Brandl419e3de2010-12-01 15:44:25 +0000964 .. versionadded:: 3.1
965 under the name ``assertRaisesRegexp``.
Ezio Melottib4dc2502011-05-06 15:01:41 +0300966
Ezio Melottied3a7d22010-12-01 02:32:32 +0000967 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +0000968 Renamed to :meth:`assertRaisesRegex`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000969
Ezio Melottib4dc2502011-05-06 15:01:41 +0300970 .. versionchanged:: 3.3
971 Added the *msg* keyword argument when used as a context manager.
972
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000973
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000974 .. method:: assertWarns(warning, callable, *args, **kwds)
Ezio Melottib4dc2502011-05-06 15:01:41 +0300975 assertWarns(warning, msg=None)
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000976
977 Test that a warning is triggered when *callable* is called with any
978 positional or keyword arguments that are also passed to
979 :meth:`assertWarns`. The test passes if *warning* is triggered and
Terry Jan Reedy778cba72013-07-30 22:31:06 -0400980 fails if it isn't. Any exception is an error.
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000981 To catch any of a group of warnings, a tuple containing the warning
982 classes may be passed as *warnings*.
983
Ezio Melottib4dc2502011-05-06 15:01:41 +0300984 If only the *warning* and possibly the *msg* arguments are given,
Terry Jan Reedy778cba72013-07-30 22:31:06 -0400985 return a context manager so that the code under test can be written
Ezio Melottib4dc2502011-05-06 15:01:41 +0300986 inline rather than as a function::
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000987
988 with self.assertWarns(SomeWarning):
989 do_something()
990
Terry Jan Reedy9eda66d2013-07-27 16:15:29 -0400991 When used as a context manager, :meth:`assertWarns` accepts the
Ezio Melottib4dc2502011-05-06 15:01:41 +0300992 additional keyword argument *msg*.
993
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000994 The context manager will store the caught warning object in its
995 :attr:`warning` attribute, and the source line which triggered the
996 warnings in the :attr:`filename` and :attr:`lineno` attributes.
997 This can be useful if the intention is to perform additional checks
Terry Jan Reedy778cba72013-07-30 22:31:06 -0400998 on the warning caught::
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000999
1000 with self.assertWarns(SomeWarning) as cm:
1001 do_something()
1002
1003 self.assertIn('myfile.py', cm.filename)
1004 self.assertEqual(320, cm.lineno)
1005
1006 This method works regardless of the warning filters in place when it
1007 is called.
1008
1009 .. versionadded:: 3.2
1010
Ezio Melottib4dc2502011-05-06 15:01:41 +03001011 .. versionchanged:: 3.3
1012 Added the *msg* keyword argument when used as a context manager.
1013
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001014
Ezio Melottied3a7d22010-12-01 02:32:32 +00001015 .. method:: assertWarnsRegex(warning, regex, callable, *args, **kwds)
Ezio Melottib4dc2502011-05-06 15:01:41 +03001016 assertWarnsRegex(warning, regex, msg=None)
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001017
Ezio Melottied3a7d22010-12-01 02:32:32 +00001018 Like :meth:`assertWarns` but also tests that *regex* matches on the
1019 message of the triggered warning. *regex* may be a regular expression
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001020 object or a string containing a regular expression suitable for use
1021 by :func:`re.search`. Example::
1022
Ezio Melottied3a7d22010-12-01 02:32:32 +00001023 self.assertWarnsRegex(DeprecationWarning,
1024 r'legacy_function\(\) is deprecated',
1025 legacy_function, 'XYZ')
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001026
1027 or::
1028
Ezio Melottied3a7d22010-12-01 02:32:32 +00001029 with self.assertWarnsRegex(RuntimeWarning, 'unsafe frobnicating'):
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001030 frobnicate('/etc/passwd')
1031
1032 .. versionadded:: 3.2
1033
Ezio Melottib4dc2502011-05-06 15:01:41 +03001034 .. versionchanged:: 3.3
1035 Added the *msg* keyword argument when used as a context manager.
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001036
Antoine Pitrou0715b9f2013-09-14 19:45:47 +02001037 .. method:: assertLogs(logger=None, level=None)
1038
1039 A context manager to test that at least one message is logged on
1040 the *logger* or one of its children, with at least the given
1041 *level*.
1042
1043 If given, *logger* should be a :class:`logging.Logger` object or a
1044 :class:`str` giving the name of a logger. The default is the root
1045 logger, which will catch all messages.
1046
1047 If given, *level* should be either a numeric logging level or
1048 its string equivalent (for example either ``"ERROR"`` or
1049 :attr:`logging.ERROR`). The default is :attr:`logging.INFO`.
1050
1051 The test passes if at least one message emitted inside the ``with``
1052 block matches the *logger* and *level* conditions, otherwise it fails.
1053
1054 The object returned by the context manager is a recording helper
1055 which keeps tracks of the matching log messages. It has two
1056 attributes:
1057
1058 .. attribute:: records
1059
1060 A list of :class:`logging.LogRecord` objects of the matching
1061 log messages.
1062
1063 .. attribute:: output
1064
1065 A list of :class:`str` objects with the formatted output of
1066 matching messages.
1067
1068 Example::
1069
1070 with self.assertLogs('foo', level='INFO') as cm:
1071 logging.getLogger('foo').info('first message')
1072 logging.getLogger('foo.bar').error('second message')
1073 self.assertEqual(cm.output, ['INFO:foo:first message',
1074 'ERROR:foo.bar:second message'])
1075
1076 .. versionadded:: 3.4
1077
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001078
Ezio Melotti4370b302010-11-03 20:39:14 +00001079 There are also other methods used to perform more specific checks, such as:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001080
Ezio Melotti4370b302010-11-03 20:39:14 +00001081 +---------------------------------------+--------------------------------+--------------+
1082 | Method | Checks that | New in |
1083 +=======================================+================================+==============+
1084 | :meth:`assertAlmostEqual(a, b) | ``round(a-b, 7) == 0`` | |
1085 | <TestCase.assertAlmostEqual>` | | |
1086 +---------------------------------------+--------------------------------+--------------+
1087 | :meth:`assertNotAlmostEqual(a, b) | ``round(a-b, 7) != 0`` | |
1088 | <TestCase.assertNotAlmostEqual>` | | |
1089 +---------------------------------------+--------------------------------+--------------+
1090 | :meth:`assertGreater(a, b) | ``a > b`` | 3.1 |
1091 | <TestCase.assertGreater>` | | |
1092 +---------------------------------------+--------------------------------+--------------+
1093 | :meth:`assertGreaterEqual(a, b) | ``a >= b`` | 3.1 |
1094 | <TestCase.assertGreaterEqual>` | | |
1095 +---------------------------------------+--------------------------------+--------------+
1096 | :meth:`assertLess(a, b) | ``a < b`` | 3.1 |
1097 | <TestCase.assertLess>` | | |
1098 +---------------------------------------+--------------------------------+--------------+
1099 | :meth:`assertLessEqual(a, b) | ``a <= b`` | 3.1 |
1100 | <TestCase.assertLessEqual>` | | |
1101 +---------------------------------------+--------------------------------+--------------+
Ezio Melotti560a7782013-09-13 22:17:40 +03001102 | :meth:`assertRegex(s, r) | ``r.search(s)`` | 3.1 |
Ezio Melottied3a7d22010-12-01 02:32:32 +00001103 | <TestCase.assertRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001104 +---------------------------------------+--------------------------------+--------------+
Ezio Melotti560a7782013-09-13 22:17:40 +03001105 | :meth:`assertNotRegex(s, r) | ``not r.search(s)`` | 3.2 |
Ezio Melottied3a7d22010-12-01 02:32:32 +00001106 | <TestCase.assertNotRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001107 +---------------------------------------+--------------------------------+--------------+
Éric Araujo941afed2011-09-01 02:47:34 +02001108 | :meth:`assertCountEqual(a, b) | *a* and *b* have the same | 3.2 |
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001109 | <TestCase.assertCountEqual>` | elements in the same number, | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001110 | | regardless of their order | |
1111 +---------------------------------------+--------------------------------+--------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001112
1113
Michael Foorde180d392011-01-28 19:51:48 +00001114 .. method:: assertAlmostEqual(first, second, places=7, msg=None, delta=None)
1115 assertNotAlmostEqual(first, second, places=7, msg=None, delta=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001116
Michael Foorde180d392011-01-28 19:51:48 +00001117 Test that *first* and *second* are approximately (or not approximately)
Ezio Melotti4841fd62010-11-05 15:43:40 +00001118 equal by computing the difference, rounding to the given number of
1119 decimal *places* (default 7), and comparing to zero. Note that these
1120 methods round the values to the given number of *decimal places* (i.e.
1121 like the :func:`round` function) and not *significant digits*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001122
Ezio Melotti4370b302010-11-03 20:39:14 +00001123 If *delta* is supplied instead of *places* then the difference
Ezio Melottid51914c2013-08-11 13:04:50 +03001124 between *first* and *second* must be less or equal to (or greater than) *delta*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001125
Ezio Melotti4370b302010-11-03 20:39:14 +00001126 Supplying both *delta* and *places* raises a ``TypeError``.
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001127
Ezio Melotti4370b302010-11-03 20:39:14 +00001128 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001129 :meth:`assertAlmostEqual` automatically considers almost equal objects
1130 that compare equal. :meth:`assertNotAlmostEqual` automatically fails
1131 if the objects compare equal. Added the *delta* keyword argument.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001132
Ezio Melotti4370b302010-11-03 20:39:14 +00001133
Michael Foorde180d392011-01-28 19:51:48 +00001134 .. method:: assertGreater(first, second, msg=None)
1135 assertGreaterEqual(first, second, msg=None)
1136 assertLess(first, second, msg=None)
1137 assertLessEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001138
Michael Foorde180d392011-01-28 19:51:48 +00001139 Test that *first* is respectively >, >=, < or <= than *second* depending
Ezio Melotti4841fd62010-11-05 15:43:40 +00001140 on the method name. If not, the test will fail::
Ezio Melotti4370b302010-11-03 20:39:14 +00001141
1142 >>> self.assertGreaterEqual(3, 4)
1143 AssertionError: "3" unexpectedly not greater than or equal to "4"
1144
1145 .. versionadded:: 3.1
1146
1147
Ezio Melottied3a7d22010-12-01 02:32:32 +00001148 .. method:: assertRegex(text, regex, msg=None)
1149 assertNotRegex(text, regex, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001150
Ezio Melottied3a7d22010-12-01 02:32:32 +00001151 Test that a *regex* search matches (or does not match) *text*. In case
Ezio Melotti4841fd62010-11-05 15:43:40 +00001152 of failure, the error message will include the pattern and the *text* (or
Ezio Melottied3a7d22010-12-01 02:32:32 +00001153 the pattern and the part of *text* that unexpectedly matched). *regex*
Ezio Melotti4370b302010-11-03 20:39:14 +00001154 may be a regular expression object or a string containing a regular
1155 expression suitable for use by :func:`re.search`.
1156
Georg Brandl419e3de2010-12-01 15:44:25 +00001157 .. versionadded:: 3.1
1158 under the name ``assertRegexpMatches``.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001159 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001160 The method ``assertRegexpMatches()`` has been renamed to
1161 :meth:`.assertRegex`.
1162 .. versionadded:: 3.2
1163 :meth:`.assertNotRegex`.
Ezio Melotti4370b302010-11-03 20:39:14 +00001164
1165
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001166 .. method:: assertCountEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001167
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001168 Test that sequence *first* contains the same elements as *second*,
Ezio Melotti4370b302010-11-03 20:39:14 +00001169 regardless of their order. When they don't, an error message listing the
1170 differences between the sequences will be generated.
1171
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001172 Duplicate elements are *not* ignored when comparing *first* and
1173 *second*. It verifies whether each element has the same count in both
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001174 sequences. Equivalent to:
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001175 ``assertEqual(Counter(list(first)), Counter(list(second)))``
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001176 but works with sequences of unhashable objects as well.
Ezio Melotti4370b302010-11-03 20:39:14 +00001177
Ezio Melotti4370b302010-11-03 20:39:14 +00001178 .. versionadded:: 3.2
1179
Ezio Melotti4370b302010-11-03 20:39:14 +00001180
Ezio Melotti22170ed2010-11-20 09:57:27 +00001181 .. _type-specific-methods:
Ezio Melotti4370b302010-11-03 20:39:14 +00001182
Ezio Melotti22170ed2010-11-20 09:57:27 +00001183 The :meth:`assertEqual` method dispatches the equality check for objects of
1184 the same type to different type-specific methods. These methods are already
1185 implemented for most of the built-in types, but it's also possible to
1186 register new methods using :meth:`addTypeEqualityFunc`:
1187
1188 .. method:: addTypeEqualityFunc(typeobj, function)
1189
1190 Registers a type-specific method called by :meth:`assertEqual` to check
1191 if two objects of exactly the same *typeobj* (not subclasses) compare
1192 equal. *function* must take two positional arguments and a third msg=None
1193 keyword argument just as :meth:`assertEqual` does. It must raise
1194 :data:`self.failureException(msg) <failureException>` when inequality
1195 between the first two parameters is detected -- possibly providing useful
1196 information and explaining the inequalities in details in the error
1197 message.
1198
1199 .. versionadded:: 3.1
1200
1201 The list of type-specific methods automatically used by
1202 :meth:`~TestCase.assertEqual` are summarized in the following table. Note
1203 that it's usually not necessary to invoke these methods directly.
Ezio Melotti4370b302010-11-03 20:39:14 +00001204
1205 +-----------------------------------------+-----------------------------+--------------+
1206 | Method | Used to compare | New in |
1207 +=========================================+=============================+==============+
1208 | :meth:`assertMultiLineEqual(a, b) | strings | 3.1 |
1209 | <TestCase.assertMultiLineEqual>` | | |
1210 +-----------------------------------------+-----------------------------+--------------+
1211 | :meth:`assertSequenceEqual(a, b) | sequences | 3.1 |
1212 | <TestCase.assertSequenceEqual>` | | |
1213 +-----------------------------------------+-----------------------------+--------------+
1214 | :meth:`assertListEqual(a, b) | lists | 3.1 |
1215 | <TestCase.assertListEqual>` | | |
1216 +-----------------------------------------+-----------------------------+--------------+
1217 | :meth:`assertTupleEqual(a, b) | tuples | 3.1 |
1218 | <TestCase.assertTupleEqual>` | | |
1219 +-----------------------------------------+-----------------------------+--------------+
1220 | :meth:`assertSetEqual(a, b) | sets or frozensets | 3.1 |
1221 | <TestCase.assertSetEqual>` | | |
1222 +-----------------------------------------+-----------------------------+--------------+
1223 | :meth:`assertDictEqual(a, b) | dicts | 3.1 |
1224 | <TestCase.assertDictEqual>` | | |
1225 +-----------------------------------------+-----------------------------+--------------+
1226
1227
1228
Michael Foorde180d392011-01-28 19:51:48 +00001229 .. method:: assertMultiLineEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001230
Michael Foorde180d392011-01-28 19:51:48 +00001231 Test that the multiline string *first* is equal to the string *second*.
Ezio Melotti4370b302010-11-03 20:39:14 +00001232 When not equal a diff of the two strings highlighting the differences
1233 will be included in the error message. This method is used by default
1234 when comparing strings with :meth:`assertEqual`.
1235
Ezio Melotti4370b302010-11-03 20:39:14 +00001236 .. versionadded:: 3.1
1237
1238
Michael Foorde180d392011-01-28 19:51:48 +00001239 .. method:: assertSequenceEqual(first, second, msg=None, seq_type=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001240
1241 Tests that two sequences are equal. If a *seq_type* is supplied, both
Michael Foorde180d392011-01-28 19:51:48 +00001242 *first* and *second* must be instances of *seq_type* or a failure will
Ezio Melotti4370b302010-11-03 20:39:14 +00001243 be raised. If the sequences are different an error message is
1244 constructed that shows the difference between the two.
1245
Ezio Melotti22170ed2010-11-20 09:57:27 +00001246 This method is not called directly by :meth:`assertEqual`, but
1247 it's used to implement :meth:`assertListEqual` and
Ezio Melotti4370b302010-11-03 20:39:14 +00001248 :meth:`assertTupleEqual`.
1249
1250 .. versionadded:: 3.1
1251
1252
Michael Foorde180d392011-01-28 19:51:48 +00001253 .. method:: assertListEqual(first, second, msg=None)
1254 assertTupleEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001255
Ezio Melotti49ccd512012-08-29 17:50:42 +03001256 Tests that two lists or tuples are equal. If not, an error message is
Ezio Melotti4370b302010-11-03 20:39:14 +00001257 constructed that shows only the differences between the two. An error
1258 is also raised if either of the parameters are of the wrong type.
1259 These methods are used by default when comparing lists or tuples with
1260 :meth:`assertEqual`.
1261
Ezio Melotti4370b302010-11-03 20:39:14 +00001262 .. versionadded:: 3.1
1263
1264
Michael Foorde180d392011-01-28 19:51:48 +00001265 .. method:: assertSetEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001266
1267 Tests that two sets are equal. If not, an error message is constructed
1268 that lists the differences between the sets. This method is used by
1269 default when comparing sets or frozensets with :meth:`assertEqual`.
1270
Michael Foorde180d392011-01-28 19:51:48 +00001271 Fails if either of *first* or *second* does not have a :meth:`set.difference`
Ezio Melotti4370b302010-11-03 20:39:14 +00001272 method.
1273
Ezio Melotti4370b302010-11-03 20:39:14 +00001274 .. versionadded:: 3.1
1275
1276
Michael Foorde180d392011-01-28 19:51:48 +00001277 .. method:: assertDictEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001278
1279 Test that two dictionaries are equal. If not, an error message is
1280 constructed that shows the differences in the dictionaries. This
1281 method will be used by default to compare dictionaries in
1282 calls to :meth:`assertEqual`.
1283
Ezio Melotti4370b302010-11-03 20:39:14 +00001284 .. versionadded:: 3.1
1285
1286
1287
Ezio Melotti22170ed2010-11-20 09:57:27 +00001288 .. _other-methods-and-attrs:
1289
Ezio Melotti4370b302010-11-03 20:39:14 +00001290 Finally the :class:`TestCase` provides the following methods and attributes:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001291
Benjamin Peterson52baa292009-03-24 00:56:30 +00001292
Georg Brandl7f01a132009-09-16 15:58:14 +00001293 .. method:: fail(msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001294
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001295 Signals a test failure unconditionally, with *msg* or ``None`` for
Benjamin Peterson52baa292009-03-24 00:56:30 +00001296 the error message.
1297
1298
1299 .. attribute:: failureException
1300
1301 This class attribute gives the exception raised by the test method. If a
1302 test framework needs to use a specialized exception, possibly to carry
1303 additional information, it must subclass this exception in order to "play
1304 fair" with the framework. The initial value of this attribute is
1305 :exc:`AssertionError`.
1306
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001307
1308 .. attribute:: longMessage
1309
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001310 If set to ``True`` then any explicit failure message you pass in to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001311 :ref:`assert methods <assert-methods>` will be appended to the end of the
1312 normal failure message. The normal messages contain useful information
1313 about the objects involved, for example the message from assertEqual
1314 shows you the repr of the two unequal objects. Setting this attribute
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001315 to ``True`` allows you to have a custom error message in addition to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001316 normal one.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001317
Michael Foord5074df62010-12-03 00:53:09 +00001318 This attribute defaults to ``True``. If set to False then a custom message
1319 passed to an assert method will silence the normal message.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001320
1321 The class setting can be overridden in individual tests by assigning an
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001322 instance attribute to ``True`` or ``False`` before calling the assert methods.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001323
Raymond Hettinger35a88362009-04-09 00:08:24 +00001324 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001325
1326
Michael Foord98b3e762010-06-05 21:59:55 +00001327 .. attribute:: maxDiff
1328
1329 This attribute controls the maximum length of diffs output by assert
1330 methods that report diffs on failure. It defaults to 80*8 characters.
1331 Assert methods affected by this attribute are
1332 :meth:`assertSequenceEqual` (including all the sequence comparison
1333 methods that delegate to it), :meth:`assertDictEqual` and
1334 :meth:`assertMultiLineEqual`.
1335
1336 Setting ``maxDiff`` to None means that there is no maximum length of
1337 diffs.
1338
1339 .. versionadded:: 3.2
1340
1341
Benjamin Peterson52baa292009-03-24 00:56:30 +00001342 Testing frameworks can use the following methods to collect information on
1343 the test:
1344
1345
1346 .. method:: countTestCases()
1347
1348 Return the number of tests represented by this test object. For
1349 :class:`TestCase` instances, this will always be ``1``.
1350
1351
1352 .. method:: defaultTestResult()
1353
1354 Return an instance of the test result class that should be used for this
1355 test case class (if no other result instance is provided to the
1356 :meth:`run` method).
1357
1358 For :class:`TestCase` instances, this will always be an instance of
1359 :class:`TestResult`; subclasses of :class:`TestCase` should override this
1360 as necessary.
1361
1362
1363 .. method:: id()
1364
1365 Return a string identifying the specific test case. This is usually the
1366 full name of the test method, including the module and class name.
1367
1368
1369 .. method:: shortDescription()
1370
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001371 Returns a description of the test, or ``None`` if no description
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001372 has been provided. The default implementation of this method
1373 returns the first line of the test method's docstring, if available,
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001374 or ``None``.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001375
Georg Brandl419e3de2010-12-01 15:44:25 +00001376 .. versionchanged:: 3.1
Michael Foord34c94622010-02-10 15:51:42 +00001377 In 3.1 this was changed to add the test name to the short description
Georg Brandl419e3de2010-12-01 15:44:25 +00001378 even in the presence of a docstring. This caused compatibility issues
Michael Foord34c94622010-02-10 15:51:42 +00001379 with unittest extensions and adding the test name was moved to the
Georg Brandl419e3de2010-12-01 15:44:25 +00001380 :class:`TextTestResult` in Python 3.2.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001381
Georg Brandl116aa622007-08-15 14:28:22 +00001382
Georg Brandl7f01a132009-09-16 15:58:14 +00001383 .. method:: addCleanup(function, *args, **kwargs)
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001384
1385 Add a function to be called after :meth:`tearDown` to cleanup resources
1386 used during the test. Functions will be called in reverse order to the
1387 order they are added (LIFO). They are called with any arguments and
1388 keyword arguments passed into :meth:`addCleanup` when they are
1389 added.
1390
1391 If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
1392 then any cleanup functions added will still be called.
1393
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001394 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001395
1396
1397 .. method:: doCleanups()
1398
Barry Warsaw0c9fd632010-04-12 14:50:57 +00001399 This method is called unconditionally after :meth:`tearDown`, or
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001400 after :meth:`setUp` if :meth:`setUp` raises an exception.
1401
1402 It is responsible for calling all the cleanup functions added by
1403 :meth:`addCleanup`. If you need cleanup functions to be called
1404 *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
1405 yourself.
1406
1407 :meth:`doCleanups` pops methods off the stack of cleanup
1408 functions one at a time, so it can be called at any time.
1409
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001410 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001411
1412
Georg Brandl7f01a132009-09-16 15:58:14 +00001413.. class:: FunctionTestCase(testFunc, setUp=None, tearDown=None, description=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001414
1415 This class implements the portion of the :class:`TestCase` interface which
Benjamin Petersond2397752009-06-27 23:45:02 +00001416 allows the test runner to drive the test, but does not provide the methods
1417 which test code can use to check and report errors. This is used to create
1418 test cases using legacy test code, allowing it to be integrated into a
1419 :mod:`unittest`-based test framework.
Georg Brandl116aa622007-08-15 14:28:22 +00001420
1421
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001422.. _deprecated-aliases:
1423
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001424Deprecated aliases
1425##################
1426
1427For historical reasons, some of the :class:`TestCase` methods had one or more
1428aliases that are now deprecated. The following table lists the correct names
1429along with their deprecated aliases:
1430
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001431 ============================== ====================== ======================
1432 Method Name Deprecated alias Deprecated alias
1433 ============================== ====================== ======================
1434 :meth:`.assertEqual` failUnlessEqual assertEquals
1435 :meth:`.assertNotEqual` failIfEqual assertNotEquals
1436 :meth:`.assertTrue` failUnless assert\_
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001437 :meth:`.assertFalse` failIf
1438 :meth:`.assertRaises` failUnlessRaises
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001439 :meth:`.assertAlmostEqual` failUnlessAlmostEqual assertAlmostEquals
1440 :meth:`.assertNotAlmostEqual` failIfAlmostEqual assertNotAlmostEquals
Ezio Melottied3a7d22010-12-01 02:32:32 +00001441 :meth:`.assertRegex` assertRegexpMatches
1442 :meth:`.assertRaisesRegex` assertRaisesRegexp
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001443 ============================== ====================== ======================
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001444
Ezio Melotti361467e2011-04-03 17:37:58 +03001445 .. deprecated:: 3.1
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001446 the fail* aliases listed in the second column.
1447 .. deprecated:: 3.2
1448 the assert* aliases listed in the third column.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001449 .. deprecated:: 3.2
1450 ``assertRegexpMatches`` and ``assertRaisesRegexp`` have been renamed to
1451 :meth:`.assertRegex` and :meth:`.assertRaisesRegex`
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001452
1453
Benjamin Peterson52baa292009-03-24 00:56:30 +00001454.. _testsuite-objects:
1455
Benjamin Peterson52baa292009-03-24 00:56:30 +00001456Grouping tests
1457~~~~~~~~~~~~~~
1458
Georg Brandl7f01a132009-09-16 15:58:14 +00001459.. class:: TestSuite(tests=())
Georg Brandl116aa622007-08-15 14:28:22 +00001460
1461 This class represents an aggregation of individual tests cases and test suites.
1462 The class presents the interface needed by the test runner to allow it to be run
1463 as any other test case. Running a :class:`TestSuite` instance is the same as
1464 iterating over the suite, running each test individually.
1465
1466 If *tests* is given, it must be an iterable of individual test cases or other
1467 test suites that will be used to build the suite initially. Additional methods
1468 are provided to add test cases and suites to the collection later on.
1469
Benjamin Peterson14a3dd72009-05-25 00:51:58 +00001470 :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
1471 they do not actually implement a test. Instead, they are used to aggregate
1472 tests into groups of tests that should be run together. Some additional
1473 methods are available to add tests to :class:`TestSuite` instances:
Benjamin Peterson52baa292009-03-24 00:56:30 +00001474
1475
1476 .. method:: TestSuite.addTest(test)
1477
1478 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1479
1480
1481 .. method:: TestSuite.addTests(tests)
1482
1483 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
1484 instances to this test suite.
1485
Benjamin Petersond2397752009-06-27 23:45:02 +00001486 This is equivalent to iterating over *tests*, calling :meth:`addTest` for
1487 each element.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001488
1489 :class:`TestSuite` shares the following methods with :class:`TestCase`:
1490
1491
1492 .. method:: run(result)
1493
1494 Run the tests associated with this suite, collecting the result into the
1495 test result object passed as *result*. Note that unlike
1496 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
1497 be passed in.
1498
1499
1500 .. method:: debug()
1501
1502 Run the tests associated with this suite without collecting the
1503 result. This allows exceptions raised by the test to be propagated to the
1504 caller and can be used to support running tests under a debugger.
1505
1506
1507 .. method:: countTestCases()
1508
1509 Return the number of tests represented by this test object, including all
1510 individual tests and sub-suites.
1511
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001512
1513 .. method:: __iter__()
1514
1515 Tests grouped by a :class:`TestSuite` are always accessed by iteration.
1516 Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
Andrew Svetloveb973682013-08-28 21:28:38 +03001517 that this method may be called several times on a single suite (for
1518 example when counting tests or comparing for equality) so the tests
1519 returned by repeated iterations before :meth:`TestSuite.run` must be the
1520 same for each call iteration. After :meth:`TestSuite.run`, callers should
1521 not rely on the tests returned by this method unless the caller uses a
1522 subclass that overrides :meth:`TestSuite._removeTestAtIndex` to preserve
1523 test references.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001524
Georg Brandl853947a2010-01-31 18:53:23 +00001525 .. versionchanged:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001526 In earlier versions the :class:`TestSuite` accessed tests directly rather
1527 than through iteration, so overriding :meth:`__iter__` wasn't sufficient
1528 for providing tests.
1529
Andrew Svetloveb973682013-08-28 21:28:38 +03001530 .. versionchanged:: 3.4
1531 In earlier versions the :class:`TestSuite` held references to each
1532 :class:`TestCase` after :meth:`TestSuite.run`. Subclasses can restore
1533 that behavior by overriding :meth:`TestSuite._removeTestAtIndex`.
1534
Benjamin Peterson52baa292009-03-24 00:56:30 +00001535 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
1536 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1537
1538
Benjamin Peterson52baa292009-03-24 00:56:30 +00001539Loading and running tests
1540~~~~~~~~~~~~~~~~~~~~~~~~~
1541
Georg Brandl116aa622007-08-15 14:28:22 +00001542.. class:: TestLoader()
1543
Benjamin Peterson52baa292009-03-24 00:56:30 +00001544 The :class:`TestLoader` class is used to create test suites from classes and
1545 modules. Normally, there is no need to create an instance of this class; the
1546 :mod:`unittest` module provides an instance that can be shared as
Ezio Melottib8e336b2012-04-29 10:52:18 +03001547 :data:`unittest.defaultTestLoader`. Using a subclass or instance, however,
1548 allows customization of some configurable properties.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001549
1550 :class:`TestLoader` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001551
Ezio Melotti9c02c2f2010-11-03 20:45:31 +00001552
Benjamin Peterson52baa292009-03-24 00:56:30 +00001553 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl116aa622007-08-15 14:28:22 +00001554
Benjamin Peterson52baa292009-03-24 00:56:30 +00001555 Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
1556 :class:`testCaseClass`.
1557
1558
1559 .. method:: loadTestsFromModule(module)
1560
1561 Return a suite of all tests cases contained in the given module. This
1562 method searches *module* for classes derived from :class:`TestCase` and
1563 creates an instance of the class for each test method defined for the
1564 class.
1565
Georg Brandle720c0a2009-04-27 16:20:50 +00001566 .. note::
Benjamin Peterson52baa292009-03-24 00:56:30 +00001567
1568 While using a hierarchy of :class:`TestCase`\ -derived classes can be
1569 convenient in sharing fixtures and helper functions, defining test
1570 methods on base classes that are not intended to be instantiated
1571 directly does not play well with this method. Doing so, however, can
1572 be useful when the fixtures are different and defined in subclasses.
1573
Benjamin Petersond2397752009-06-27 23:45:02 +00001574 If a module provides a ``load_tests`` function it will be called to
1575 load the tests. This allows modules to customize test loading.
1576 This is the `load_tests protocol`_.
1577
Georg Brandl853947a2010-01-31 18:53:23 +00001578 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001579 Support for ``load_tests`` added.
1580
Benjamin Peterson52baa292009-03-24 00:56:30 +00001581
Georg Brandl7f01a132009-09-16 15:58:14 +00001582 .. method:: loadTestsFromName(name, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001583
1584 Return a suite of all tests cases given a string specifier.
1585
1586 The specifier *name* is a "dotted name" that may resolve either to a
1587 module, a test case class, a test method within a test case class, a
1588 :class:`TestSuite` instance, or a callable object which returns a
1589 :class:`TestCase` or :class:`TestSuite` instance. These checks are
1590 applied in the order listed here; that is, a method on a possible test
1591 case class will be picked up as "a test method within a test case class",
1592 rather than "a callable object".
1593
1594 For example, if you have a module :mod:`SampleTests` containing a
1595 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
1596 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
Benjamin Petersond2397752009-06-27 23:45:02 +00001597 specifier ``'SampleTests.SampleTestCase'`` would cause this method to
1598 return a suite which will run all three test methods. Using the specifier
1599 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
1600 suite which will run only the :meth:`test_two` test method. The specifier
1601 can refer to modules and packages which have not been imported; they will
1602 be imported as a side-effect.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001603
1604 The method optionally resolves *name* relative to the given *module*.
1605
1606
Georg Brandl7f01a132009-09-16 15:58:14 +00001607 .. method:: loadTestsFromNames(names, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001608
1609 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
1610 than a single name. The return value is a test suite which supports all
1611 the tests defined for each name.
1612
1613
1614 .. method:: getTestCaseNames(testCaseClass)
1615
1616 Return a sorted sequence of method names found within *testCaseClass*;
1617 this should be a subclass of :class:`TestCase`.
1618
Benjamin Petersond2397752009-06-27 23:45:02 +00001619
1620 .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)
1621
1622 Find and return all test modules from the specified start directory,
1623 recursing into subdirectories to find them. Only test files that match
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001624 *pattern* will be loaded. (Using shell style pattern matching.) Only
1625 module names that are importable (i.e. are valid Python identifiers) will
1626 be loaded.
Benjamin Petersond2397752009-06-27 23:45:02 +00001627
1628 All test modules must be importable from the top level of the project. If
1629 the start directory is not the top level directory then the top level
1630 directory must be specified separately.
1631
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001632 If importing a module fails, for example due to a syntax error, then this
Ezio Melottieae2b382013-03-01 14:47:50 +02001633 will be recorded as a single error and discovery will continue. If the
Ezio Melotti67ddcca2013-03-27 20:13:59 +02001634 import failure is due to :exc:`SkipTest` being raised, it will be recorded
Ezio Melottieae2b382013-03-01 14:47:50 +02001635 as a skip instead of an error.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001636
Benjamin Petersond2397752009-06-27 23:45:02 +00001637 If a test package name (directory with :file:`__init__.py`) matches the
1638 pattern then the package will be checked for a ``load_tests``
1639 function. If this exists then it will be called with *loader*, *tests*,
1640 *pattern*.
1641
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001642 If load_tests exists then discovery does *not* recurse into the package,
Benjamin Petersond2397752009-06-27 23:45:02 +00001643 ``load_tests`` is responsible for loading all tests in the package.
1644
1645 The pattern is deliberately not stored as a loader attribute so that
1646 packages can continue discovery themselves. *top_level_dir* is stored so
1647 ``load_tests`` does not need to pass this argument in to
1648 ``loader.discover()``.
1649
Benjamin Petersonb48af542010-04-11 20:43:16 +00001650 *start_dir* can be a dotted module name as well as a directory.
1651
Georg Brandl853947a2010-01-31 18:53:23 +00001652 .. versionadded:: 3.2
1653
Ezio Melottieae2b382013-03-01 14:47:50 +02001654 .. versionchanged:: 3.4
Ezio Melotti67ddcca2013-03-27 20:13:59 +02001655 Modules that raise :exc:`SkipTest` on import are recorded as skips,
1656 not errors.
Ezio Melottieae2b382013-03-01 14:47:50 +02001657
Michael Foord80cbc9e2013-03-18 17:50:12 -07001658 .. versionchanged:: 3.4
1659 Paths are sorted before being imported to ensure execution order for a
1660 given test suite is the same even if the underlying file system's ordering
1661 is not dependent on file name like in ext3/4.
1662
Benjamin Petersond2397752009-06-27 23:45:02 +00001663
Benjamin Peterson52baa292009-03-24 00:56:30 +00001664 The following attributes of a :class:`TestLoader` can be configured either by
1665 subclassing or assignment on an instance:
1666
1667
1668 .. attribute:: testMethodPrefix
1669
1670 String giving the prefix of method names which will be interpreted as test
1671 methods. The default value is ``'test'``.
1672
1673 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
1674 methods.
1675
1676
1677 .. attribute:: sortTestMethodsUsing
1678
1679 Function to be used to compare method names when sorting them in
1680 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
1681
1682
1683 .. attribute:: suiteClass
1684
1685 Callable object that constructs a test suite from a list of tests. No
1686 methods on the resulting object are needed. The default value is the
1687 :class:`TestSuite` class.
1688
1689 This affects all the :meth:`loadTestsFrom\*` methods.
1690
1691
Benjamin Peterson52baa292009-03-24 00:56:30 +00001692.. class:: TestResult
1693
1694 This class is used to compile information about which tests have succeeded
1695 and which have failed.
1696
1697 A :class:`TestResult` object stores the results of a set of tests. The
1698 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
1699 properly recorded; test authors do not need to worry about recording the
1700 outcome of tests.
1701
1702 Testing frameworks built on top of :mod:`unittest` may want access to the
1703 :class:`TestResult` object generated by running a set of tests for reporting
1704 purposes; a :class:`TestResult` instance is returned by the
1705 :meth:`TestRunner.run` method for this purpose.
1706
1707 :class:`TestResult` instances have the following attributes that will be of
1708 interest when inspecting the results of running a set of tests:
1709
1710
1711 .. attribute:: errors
1712
1713 A list containing 2-tuples of :class:`TestCase` instances and strings
1714 holding formatted tracebacks. Each tuple represents a test which raised an
1715 unexpected exception.
1716
Benjamin Peterson52baa292009-03-24 00:56:30 +00001717 .. attribute:: failures
1718
1719 A list containing 2-tuples of :class:`TestCase` instances and strings
1720 holding formatted tracebacks. Each tuple represents a test where a failure
Ezio Melottie2202362013-09-07 15:19:30 +03001721 was explicitly signalled using the :meth:`TestCase.assert\*` methods.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001722
Benjamin Peterson52baa292009-03-24 00:56:30 +00001723 .. attribute:: skipped
1724
1725 A list containing 2-tuples of :class:`TestCase` instances and strings
1726 holding the reason for skipping the test.
1727
Benjamin Peterson70e32c82009-03-24 01:00:11 +00001728 .. versionadded:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +00001729
1730 .. attribute:: expectedFailures
1731
Georg Brandl6faee4e2010-09-21 14:48:28 +00001732 A list containing 2-tuples of :class:`TestCase` instances and strings
1733 holding formatted tracebacks. Each tuple represents an expected failure
Benjamin Peterson52baa292009-03-24 00:56:30 +00001734 of the test case.
1735
1736 .. attribute:: unexpectedSuccesses
1737
1738 A list containing :class:`TestCase` instances that were marked as expected
1739 failures, but succeeded.
1740
1741 .. attribute:: shouldStop
1742
1743 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1744
1745
1746 .. attribute:: testsRun
1747
1748 The total number of tests run so far.
1749
1750
Georg Brandl12037202010-12-02 22:35:25 +00001751 .. attribute:: buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +00001752
1753 If set to true, ``sys.stdout`` and ``sys.stderr`` will be buffered in between
1754 :meth:`startTest` and :meth:`stopTest` being called. Collected output will
1755 only be echoed onto the real ``sys.stdout`` and ``sys.stderr`` if the test
1756 fails or errors. Any output is also attached to the failure / error message.
1757
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001758 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001759
1760
1761 .. attribute:: failfast
1762
1763 If set to true :meth:`stop` will be called on the first failure or error,
1764 halting the test run.
1765
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001766 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001767
1768
Benjamin Peterson52baa292009-03-24 00:56:30 +00001769 .. method:: wasSuccessful()
1770
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001771 Return ``True`` if all tests run so far have passed, otherwise returns
1772 ``False``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001773
1774
1775 .. method:: stop()
1776
1777 This method can be called to signal that the set of tests being run should
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001778 be aborted by setting the :attr:`shouldStop` attribute to ``True``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001779 :class:`TestRunner` objects should respect this flag and return without
1780 running any additional tests.
1781
1782 For example, this feature is used by the :class:`TextTestRunner` class to
1783 stop the test framework when the user signals an interrupt from the
1784 keyboard. Interactive tools which provide :class:`TestRunner`
1785 implementations can use this in a similar manner.
1786
1787 The following methods of the :class:`TestResult` class are used to maintain
1788 the internal data structures, and may be extended in subclasses to support
1789 additional reporting requirements. This is particularly useful in building
1790 tools which support interactive reporting while tests are being run.
1791
1792
1793 .. method:: startTest(test)
1794
1795 Called when the test case *test* is about to be run.
1796
Benjamin Peterson52baa292009-03-24 00:56:30 +00001797 .. method:: stopTest(test)
1798
1799 Called after the test case *test* has been executed, regardless of the
1800 outcome.
1801
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001802 .. method:: startTestRun(test)
1803
1804 Called once before any tests are executed.
1805
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001806 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001807
1808
1809 .. method:: stopTestRun(test)
1810
Ezio Melotti176d6c42010-01-27 20:58:07 +00001811 Called once after all tests are executed.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001812
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001813 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001814
1815
Benjamin Peterson52baa292009-03-24 00:56:30 +00001816 .. method:: addError(test, err)
1817
Ezio Melottie64a91a2013-09-07 15:23:36 +03001818 Called when the test case *test* raises an unexpected exception. *err* is a
Benjamin Peterson52baa292009-03-24 00:56:30 +00001819 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1820 traceback)``.
1821
1822 The default implementation appends a tuple ``(test, formatted_err)`` to
1823 the instance's :attr:`errors` attribute, where *formatted_err* is a
1824 formatted traceback derived from *err*.
1825
1826
1827 .. method:: addFailure(test, err)
1828
Benjamin Petersond2397752009-06-27 23:45:02 +00001829 Called when the test case *test* signals a failure. *err* is a tuple of
1830 the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001831
1832 The default implementation appends a tuple ``(test, formatted_err)`` to
1833 the instance's :attr:`failures` attribute, where *formatted_err* is a
1834 formatted traceback derived from *err*.
1835
1836
1837 .. method:: addSuccess(test)
1838
1839 Called when the test case *test* succeeds.
1840
1841 The default implementation does nothing.
1842
1843
1844 .. method:: addSkip(test, reason)
1845
1846 Called when the test case *test* is skipped. *reason* is the reason the
1847 test gave for skipping.
1848
1849 The default implementation appends a tuple ``(test, reason)`` to the
1850 instance's :attr:`skipped` attribute.
1851
1852
1853 .. method:: addExpectedFailure(test, err)
1854
1855 Called when the test case *test* fails, but was marked with the
1856 :func:`expectedFailure` decorator.
1857
1858 The default implementation appends a tuple ``(test, formatted_err)`` to
1859 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1860 is a formatted traceback derived from *err*.
1861
1862
1863 .. method:: addUnexpectedSuccess(test)
1864
1865 Called when the test case *test* was marked with the
1866 :func:`expectedFailure` decorator, but succeeded.
1867
1868 The default implementation appends the test to the instance's
1869 :attr:`unexpectedSuccesses` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +00001870
Georg Brandl67b21b72010-08-17 15:07:14 +00001871
Antoine Pitrouc9b3ef22013-03-20 20:16:47 +01001872 .. method:: addSubTest(test, subtest, outcome)
1873
1874 Called when a subtest finishes. *test* is the test case
1875 corresponding to the test method. *subtest* is a custom
1876 :class:`TestCase` instance describing the subtest.
1877
1878 If *outcome* is :const:`None`, the subtest succeeded. Otherwise,
1879 it failed with an exception where *outcome* is a tuple of the form
1880 returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
1881
1882 The default implementation does nothing when the outcome is a
1883 success, and records subtest failures as normal failures.
1884
1885 .. versionadded:: 3.4
1886
1887
Michael Foord34c94622010-02-10 15:51:42 +00001888.. class:: TextTestResult(stream, descriptions, verbosity)
1889
Georg Brandl67b21b72010-08-17 15:07:14 +00001890 A concrete implementation of :class:`TestResult` used by the
1891 :class:`TextTestRunner`.
Michael Foord34c94622010-02-10 15:51:42 +00001892
Georg Brandl67b21b72010-08-17 15:07:14 +00001893 .. versionadded:: 3.2
1894 This class was previously named ``_TextTestResult``. The old name still
1895 exists as an alias but is deprecated.
1896
Georg Brandl116aa622007-08-15 14:28:22 +00001897
1898.. data:: defaultTestLoader
1899
1900 Instance of the :class:`TestLoader` class intended to be shared. If no
1901 customization of the :class:`TestLoader` is needed, this instance can be used
1902 instead of repeatedly creating new instances.
1903
1904
Ezio Melotti9c939bc2013-05-07 09:46:30 +03001905.. class:: TextTestRunner(stream=None, descriptions=True, verbosity=1, failfast=False, \
1906 buffer=False, resultclass=None, warnings=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001907
Michael Foordd218e952011-01-03 12:55:11 +00001908 A basic test runner implementation that outputs results to a stream. If *stream*
Éric Araujo941afed2011-09-01 02:47:34 +02001909 is ``None``, the default, :data:`sys.stderr` is used as the output stream. This class
Georg Brandl116aa622007-08-15 14:28:22 +00001910 has a few configurable parameters, but is essentially very simple. Graphical
1911 applications which run test suites should provide alternate implementations.
1912
Ezio Melotti60901872010-12-01 00:56:10 +00001913 By default this runner shows :exc:`DeprecationWarning`,
1914 :exc:`PendingDeprecationWarning`, and :exc:`ImportWarning` even if they are
1915 :ref:`ignored by default <warning-ignored>`. Deprecation warnings caused by
1916 :ref:`deprecated unittest methods <deprecated-aliases>` are also
1917 special-cased and, when the warning filters are ``'default'`` or ``'always'``,
1918 they will appear only once per-module, in order to avoid too many warning
Georg Brandl46402372010-12-04 19:06:18 +00001919 messages. This behavior can be overridden using the :option:`-Wd` or
Ezio Melotti60901872010-12-01 00:56:10 +00001920 :option:`-Wa` options and leaving *warnings* to ``None``.
1921
Michael Foordd218e952011-01-03 12:55:11 +00001922 .. versionchanged:: 3.2
1923 Added the ``warnings`` argument.
1924
1925 .. versionchanged:: 3.2
Éric Araujo941afed2011-09-01 02:47:34 +02001926 The default stream is set to :data:`sys.stderr` at instantiation time rather
Michael Foordd218e952011-01-03 12:55:11 +00001927 than import time.
1928
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001929 .. method:: _makeResult()
Georg Brandl116aa622007-08-15 14:28:22 +00001930
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001931 This method returns the instance of ``TestResult`` used by :meth:`run`.
1932 It is not intended to be called directly, but can be overridden in
1933 subclasses to provide a custom ``TestResult``.
1934
Michael Foord34c94622010-02-10 15:51:42 +00001935 ``_makeResult()`` instantiates the class or callable passed in the
1936 ``TextTestRunner`` constructor as the ``resultclass`` argument. It
Benjamin Petersonb48af542010-04-11 20:43:16 +00001937 defaults to :class:`TextTestResult` if no ``resultclass`` is provided.
Michael Foord34c94622010-02-10 15:51:42 +00001938 The result class is instantiated with the following arguments::
1939
1940 stream, descriptions, verbosity
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001941
Ezio Melotti60901872010-12-01 00:56:10 +00001942
Georg Brandl419e3de2010-12-01 15:44:25 +00001943.. function:: main(module='__main__', defaultTest=None, argv=None, testRunner=None, \
Ezio Melotti40dcb1d2011-03-10 13:46:50 +02001944 testLoader=unittest.defaultTestLoader, exit=True, verbosity=1, \
Georg Brandl419e3de2010-12-01 15:44:25 +00001945 failfast=None, catchbreak=None, buffer=None, warnings=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001946
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03001947 A command-line program that loads a set of tests from *module* and runs them;
1948 this is primarily for making test modules conveniently executable.
1949 The simplest use for this function is to include the following line at the
1950 end of a test script::
Georg Brandl116aa622007-08-15 14:28:22 +00001951
1952 if __name__ == '__main__':
1953 unittest.main()
1954
Benjamin Petersond2397752009-06-27 23:45:02 +00001955 You can run tests with more detailed information by passing in the verbosity
1956 argument::
1957
1958 if __name__ == '__main__':
1959 unittest.main(verbosity=2)
1960
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03001961 The *argv* argument can be a list of options passed to the program, with the
1962 first element being the program name. If not specified or ``None``,
1963 the values of :data:`sys.argv` are used.
1964
Georg Brandl116aa622007-08-15 14:28:22 +00001965 The *testRunner* argument can either be a test runner class or an already
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001966 created instance of it. By default ``main`` calls :func:`sys.exit` with
1967 an exit code indicating success or failure of the tests run.
1968
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03001969 The *testLoader* argument has to be a :class:`TestLoader` instance,
1970 and defaults to :data:`defaultTestLoader`.
1971
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001972 ``main`` supports being used from the interactive interpreter by passing in the
1973 argument ``exit=False``. This displays the result on standard output without
1974 calling :func:`sys.exit`::
1975
1976 >>> from unittest import main
1977 >>> main(module='test_module', exit=False)
1978
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03001979 The *failfast*, *catchbreak* and *buffer* parameters have the same
Éric Araujo76338ec2010-11-26 23:46:18 +00001980 effect as the same-name `command-line options`_.
Benjamin Petersonb48af542010-04-11 20:43:16 +00001981
Ezio Melotti60901872010-12-01 00:56:10 +00001982 The *warning* argument specifies the :ref:`warning filter <warning-filter>`
1983 that should be used while running the tests. If it's not specified, it will
1984 remain ``None`` if a :option:`-W` option is passed to :program:`python`,
1985 otherwise it will be set to ``'default'``.
1986
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001987 Calling ``main`` actually returns an instance of the ``TestProgram`` class.
1988 This stores the result of the tests run as the ``result`` attribute.
1989
Éric Araujo971dc012010-12-16 03:13:05 +00001990 .. versionchanged:: 3.1
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03001991 The *exit* parameter was added.
Éric Araujo971dc012010-12-16 03:13:05 +00001992
Georg Brandl853947a2010-01-31 18:53:23 +00001993 .. versionchanged:: 3.2
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03001994 The *verbosity*, *failfast*, *catchbreak*, *buffer*
1995 and *warnings* parameters were added.
Benjamin Petersond2397752009-06-27 23:45:02 +00001996
Chris Jerdonekccbc26a2013-02-23 15:44:46 -08001997 .. versionchanged:: 3.4
1998 The *defaultTest* parameter was changed to also accept an iterable of
1999 test names.
2000
Benjamin Petersond2397752009-06-27 23:45:02 +00002001
2002load_tests Protocol
2003###################
2004
Georg Brandl853947a2010-01-31 18:53:23 +00002005.. versionadded:: 3.2
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00002006
Benjamin Petersond2397752009-06-27 23:45:02 +00002007Modules or packages can customize how tests are loaded from them during normal
2008test runs or test discovery by implementing a function called ``load_tests``.
2009
2010If a test module defines ``load_tests`` it will be called by
2011:meth:`TestLoader.loadTestsFromModule` with the following arguments::
2012
2013 load_tests(loader, standard_tests, None)
2014
2015It should return a :class:`TestSuite`.
2016
2017*loader* is the instance of :class:`TestLoader` doing the loading.
2018*standard_tests* are the tests that would be loaded by default from the
2019module. It is common for test modules to only want to add or remove tests
2020from the standard set of tests.
2021The third argument is used when loading packages as part of test discovery.
2022
2023A typical ``load_tests`` function that loads tests from a specific set of
2024:class:`TestCase` classes may look like::
2025
2026 test_cases = (TestCase1, TestCase2, TestCase3)
2027
2028 def load_tests(loader, tests, pattern):
2029 suite = TestSuite()
2030 for test_class in test_cases:
2031 tests = loader.loadTestsFromTestCase(test_class)
2032 suite.addTests(tests)
2033 return suite
2034
2035If discovery is started, either from the command line or by calling
2036:meth:`TestLoader.discover`, with a pattern that matches a package
2037name then the package :file:`__init__.py` will be checked for ``load_tests``.
2038
2039.. note::
2040
Ezio Melotti4d6cb0f2013-02-28 08:28:11 +02002041 The default pattern is ``'test*.py'``. This matches all Python files
2042 that start with ``'test'`` but *won't* match any test directories.
Benjamin Petersond2397752009-06-27 23:45:02 +00002043
Ezio Melotti4d6cb0f2013-02-28 08:28:11 +02002044 A pattern like ``'test*'`` will match test packages as well as
Benjamin Petersond2397752009-06-27 23:45:02 +00002045 modules.
2046
2047If the package :file:`__init__.py` defines ``load_tests`` then it will be
2048called and discovery not continued into the package. ``load_tests``
2049is called with the following arguments::
2050
2051 load_tests(loader, standard_tests, pattern)
2052
2053This should return a :class:`TestSuite` representing all the tests
2054from the package. (``standard_tests`` will only contain tests
2055collected from :file:`__init__.py`.)
2056
2057Because the pattern is passed into ``load_tests`` the package is free to
2058continue (and potentially modify) test discovery. A 'do nothing'
2059``load_tests`` function for a test package would look like::
2060
2061 def load_tests(loader, standard_tests, pattern):
2062 # top level directory cached on loader instance
2063 this_dir = os.path.dirname(__file__)
2064 package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
2065 standard_tests.addTests(package_tests)
2066 return standard_tests
Benjamin Petersonb48af542010-04-11 20:43:16 +00002067
2068
2069Class and Module Fixtures
2070-------------------------
2071
2072Class and module level fixtures are implemented in :class:`TestSuite`. When
2073the test suite encounters a test from a new class then :meth:`tearDownClass`
2074from the previous class (if there is one) is called, followed by
2075:meth:`setUpClass` from the new class.
2076
2077Similarly if a test is from a different module from the previous test then
2078``tearDownModule`` from the previous module is run, followed by
2079``setUpModule`` from the new module.
2080
2081After all the tests have run the final ``tearDownClass`` and
2082``tearDownModule`` are run.
2083
2084Note that shared fixtures do not play well with [potential] features like test
2085parallelization and they break test isolation. They should be used with care.
2086
2087The default ordering of tests created by the unittest test loaders is to group
2088all tests from the same modules and classes together. This will lead to
2089``setUpClass`` / ``setUpModule`` (etc) being called exactly once per class and
2090module. If you randomize the order, so that tests from different modules and
2091classes are adjacent to each other, then these shared fixture functions may be
2092called multiple times in a single test run.
2093
2094Shared fixtures are not intended to work with suites with non-standard
2095ordering. A ``BaseTestSuite`` still exists for frameworks that don't want to
2096support shared fixtures.
2097
2098If there are any exceptions raised during one of the shared fixture functions
2099the test is reported as an error. Because there is no corresponding test
2100instance an ``_ErrorHolder`` object (that has the same interface as a
2101:class:`TestCase`) is created to represent the error. If you are just using
2102the standard unittest test runner then this detail doesn't matter, but if you
2103are a framework author it may be relevant.
2104
2105
2106setUpClass and tearDownClass
2107~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2108
2109These must be implemented as class methods::
2110
2111 import unittest
2112
2113 class Test(unittest.TestCase):
2114 @classmethod
2115 def setUpClass(cls):
2116 cls._connection = createExpensiveConnectionObject()
2117
2118 @classmethod
2119 def tearDownClass(cls):
2120 cls._connection.destroy()
2121
2122If you want the ``setUpClass`` and ``tearDownClass`` on base classes called
2123then you must call up to them yourself. The implementations in
2124:class:`TestCase` are empty.
2125
2126If an exception is raised during a ``setUpClass`` then the tests in the class
2127are not run and the ``tearDownClass`` is not run. Skipped classes will not
Michael Foord98b3e762010-06-05 21:59:55 +00002128have ``setUpClass`` or ``tearDownClass`` run. If the exception is a
Ezio Melotti265281a2013-03-27 20:11:55 +02002129:exc:`SkipTest` exception then the class will be reported as having been skipped
Michael Foord98b3e762010-06-05 21:59:55 +00002130instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002131
2132
2133setUpModule and tearDownModule
2134~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2135
2136These should be implemented as functions::
2137
2138 def setUpModule():
2139 createConnection()
2140
2141 def tearDownModule():
2142 closeConnection()
2143
2144If an exception is raised in a ``setUpModule`` then none of the tests in the
Michael Foord98b3e762010-06-05 21:59:55 +00002145module will be run and the ``tearDownModule`` will not be run. If the exception is a
Ezio Melotti265281a2013-03-27 20:11:55 +02002146:exc:`SkipTest` exception then the module will be reported as having been skipped
Michael Foord98b3e762010-06-05 21:59:55 +00002147instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002148
2149
2150Signal Handling
2151---------------
2152
Georg Brandl419e3de2010-12-01 15:44:25 +00002153.. versionadded:: 3.2
2154
Éric Araujo8acb67c2010-11-26 23:31:07 +00002155The :option:`-c/--catch <unittest -c>` command-line option to unittest,
Éric Araujo76338ec2010-11-26 23:46:18 +00002156along with the ``catchbreak`` parameter to :func:`unittest.main()`, provide
2157more friendly handling of control-C during a test run. With catch break
2158behavior enabled control-C will allow the currently running test to complete,
2159and the test run will then end and report all the results so far. A second
2160control-c will raise a :exc:`KeyboardInterrupt` in the usual way.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002161
Michael Foordde4ceab2010-04-25 19:53:49 +00002162The control-c handling signal handler attempts to remain compatible with code or
2163tests that install their own :const:`signal.SIGINT` handler. If the ``unittest``
2164handler is called but *isn't* the installed :const:`signal.SIGINT` handler,
2165i.e. it has been replaced by the system under test and delegated to, then it
2166calls the default handler. This will normally be the expected behavior by code
2167that replaces an installed handler and delegates to it. For individual tests
2168that need ``unittest`` control-c handling disabled the :func:`removeHandler`
2169decorator can be used.
2170
2171There are a few utility functions for framework authors to enable control-c
2172handling functionality within test frameworks.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002173
2174.. function:: installHandler()
2175
2176 Install the control-c handler. When a :const:`signal.SIGINT` is received
2177 (usually in response to the user pressing control-c) all registered results
2178 have :meth:`~TestResult.stop` called.
2179
Michael Foord469b1f02010-04-26 23:41:26 +00002180
Benjamin Petersonb48af542010-04-11 20:43:16 +00002181.. function:: registerResult(result)
2182
2183 Register a :class:`TestResult` object for control-c handling. Registering a
2184 result stores a weak reference to it, so it doesn't prevent the result from
2185 being garbage collected.
2186
Michael Foordde4ceab2010-04-25 19:53:49 +00002187 Registering a :class:`TestResult` object has no side-effects if control-c
2188 handling is not enabled, so test frameworks can unconditionally register
2189 all results they create independently of whether or not handling is enabled.
2190
Michael Foord469b1f02010-04-26 23:41:26 +00002191
Benjamin Petersonb48af542010-04-11 20:43:16 +00002192.. function:: removeResult(result)
2193
2194 Remove a registered result. Once a result has been removed then
2195 :meth:`~TestResult.stop` will no longer be called on that result object in
2196 response to a control-c.
2197
Michael Foord469b1f02010-04-26 23:41:26 +00002198
Michael Foordde4ceab2010-04-25 19:53:49 +00002199.. function:: removeHandler(function=None)
2200
2201 When called without arguments this function removes the control-c handler
2202 if it has been installed. This function can also be used as a test decorator
2203 to temporarily remove the handler whilst the test is being executed::
2204
2205 @unittest.removeHandler
2206 def test_signal_handling(self):
2207 ...