blob: bcb951713940eb7d44ed135cab414c9365a0c599 [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
Georg Brandl116aa622007-08-15 14:28:22 +000014The Python unit testing framework, sometimes referred to as "PyUnit," is a
15Python language version of JUnit, by Kent Beck and Erich Gamma. JUnit is, in
16turn, a Java version of Kent's Smalltalk testing framework. Each is the de
17facto standard unit testing framework for its respective language.
18
19:mod:`unittest` supports test automation, sharing of setup and shutdown code for
20tests, aggregation of tests into collections, and independence of the tests from
21the reporting framework. The :mod:`unittest` module provides classes that make
22it easy to support these qualities for a set of tests.
23
24To achieve this, :mod:`unittest` supports some important concepts:
25
26test fixture
27 A :dfn:`test fixture` represents the preparation needed to perform one or more
28 tests, and any associate cleanup actions. This may involve, for example,
29 creating temporary or proxy databases, directories, or starting a server
30 process.
31
32test case
33 A :dfn:`test case` is the smallest unit of testing. It checks for a specific
34 response to a particular set of inputs. :mod:`unittest` provides a base class,
35 :class:`TestCase`, which may be used to create new test cases.
36
37test suite
38 A :dfn:`test suite` is a collection of test cases, test suites, or both. It is
39 used to aggregate tests that should be executed together.
40
41test runner
42 A :dfn:`test runner` is a component which orchestrates the execution of tests
43 and provides the outcome to the user. The runner may use a graphical interface,
44 a textual interface, or return a special value to indicate the results of
45 executing the tests.
46
47The test case and test fixture concepts are supported through the
48:class:`TestCase` and :class:`FunctionTestCase` classes; the former should be
49used when creating new tests, and the latter can be used when integrating
50existing test code with a :mod:`unittest`\ -driven framework. When building test
Benjamin Peterson52baa292009-03-24 00:56:30 +000051fixtures using :class:`TestCase`, the :meth:`~TestCase.setUp` and
52:meth:`~TestCase.tearDown` methods can be overridden to provide initialization
53and cleanup for the fixture. With :class:`FunctionTestCase`, existing functions
54can be passed to the constructor for these purposes. When the test is run, the
55fixture initialization is run first; if it succeeds, the cleanup method is run
56after the test has been executed, regardless of the outcome of the test. Each
57instance of the :class:`TestCase` will only be used to run a single test method,
58so a new fixture is created for each test.
Georg Brandl116aa622007-08-15 14:28:22 +000059
60Test suites are implemented by the :class:`TestSuite` class. This class allows
61individual tests and test suites to be aggregated; when the suite is executed,
Benjamin Peterson14a3dd72009-05-25 00:51:58 +000062all tests added directly to the suite and in "child" test suites are run.
Georg Brandl116aa622007-08-15 14:28:22 +000063
Benjamin Peterson52baa292009-03-24 00:56:30 +000064A test runner is an object that provides a single method,
65:meth:`~TestRunner.run`, which accepts a :class:`TestCase` or :class:`TestSuite`
66object as a parameter, and returns a result object. The class
67:class:`TestResult` is provided for use as the result object. :mod:`unittest`
68provides the :class:`TextTestRunner` as an example test runner which reports
69test results on the standard error stream by default. Alternate runners can be
70implemented for other environments (such as graphical environments) without any
71need to derive from a specific class.
Georg Brandl116aa622007-08-15 14:28:22 +000072
73
74.. seealso::
75
76 Module :mod:`doctest`
77 Another test-support module with a very different flavor.
78
Benjamin Petersonb48af542010-04-11 20:43:16 +000079 `unittest2: A backport of new unittest features for Python 2.4-2.6 <http://pypi.python.org/pypi/unittest2>`_
80 Many new features were added to unittest in Python 2.7, including test
81 discovery. unittest2 allows you to use these features with earlier
82 versions of Python.
83
Georg Brandl116aa622007-08-15 14:28:22 +000084 `Simple Smalltalk Testing: With Patterns <http://www.XProgramming.com/testfram.htm>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000085 Kent Beck's original paper on testing frameworks using the pattern shared
86 by :mod:`unittest`.
Georg Brandl116aa622007-08-15 14:28:22 +000087
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000088 `Nose <http://code.google.com/p/python-nose/>`_ and `py.test <http://pytest.org>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000089 Third-party unittest frameworks with a lighter-weight syntax for writing
90 tests. For example, ``assert func(10) == 42``.
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000091
Benjamin Petersonb48af542010-04-11 20:43:16 +000092 `The Python Testing Tools Taxonomy <http://pycheesecake.org/wiki/PythonTestingToolsTaxonomy>`_
93 An extensive list of Python testing tools including functional testing
94 frameworks and mock object libraries.
Benjamin Petersond2397752009-06-27 23:45:02 +000095
Benjamin Petersonb48af542010-04-11 20:43:16 +000096 `Testing in Python Mailing List <http://lists.idyll.org/listinfo/testing-in-python>`_
97 A special-interest-group for discussion of testing, and testing tools,
98 in Python.
Benjamin Petersond2397752009-06-27 23:45:02 +000099
Georg Brandl116aa622007-08-15 14:28:22 +0000100.. _unittest-minimal-example:
101
102Basic example
103-------------
104
105The :mod:`unittest` module provides a rich set of tools for constructing and
106running tests. This section demonstrates that a small subset of the tools
107suffice to meet the needs of most users.
108
109Here is a short script to test three functions from the :mod:`random` module::
110
111 import random
112 import unittest
113
114 class TestSequenceFunctions(unittest.TestCase):
115
116 def setUp(self):
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000117 self.seq = list(range(10))
Georg Brandl116aa622007-08-15 14:28:22 +0000118
Benjamin Peterson52baa292009-03-24 00:56:30 +0000119 def test_shuffle(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000120 # make sure the shuffled sequence does not lose any elements
121 random.shuffle(self.seq)
122 self.seq.sort()
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000123 self.assertEqual(self.seq, list(range(10)))
Georg Brandl116aa622007-08-15 14:28:22 +0000124
Benjamin Peterson847a4112010-03-14 15:04:17 +0000125 # should raise an exception for an immutable sequence
126 self.assertRaises(TypeError, random.shuffle, (1,2,3))
127
Benjamin Peterson52baa292009-03-24 00:56:30 +0000128 def test_choice(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000129 element = random.choice(self.seq)
Benjamin Peterson847a4112010-03-14 15:04:17 +0000130 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000131
Benjamin Peterson52baa292009-03-24 00:56:30 +0000132 def test_sample(self):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000133 with self.assertRaises(ValueError):
134 random.sample(self.seq, 20)
Georg Brandl116aa622007-08-15 14:28:22 +0000135 for element in random.sample(self.seq, 5):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000136 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000137
138 if __name__ == '__main__':
139 unittest.main()
140
Benjamin Peterson52baa292009-03-24 00:56:30 +0000141A testcase is created by subclassing :class:`unittest.TestCase`. The three
Georg Brandl116aa622007-08-15 14:28:22 +0000142individual tests are defined with methods whose names start with the letters
143``test``. This naming convention informs the test runner about which methods
144represent tests.
145
Benjamin Peterson52baa292009-03-24 00:56:30 +0000146The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
Michael Foord34c94622010-02-10 15:51:42 +0000147expected result; :meth:`~TestCase.assertTrue` to verify a condition; or
Benjamin Peterson52baa292009-03-24 00:56:30 +0000148:meth:`~TestCase.assertRaises` to verify that an expected exception gets raised.
149These methods are used instead of the :keyword:`assert` statement so the test
150runner can accumulate all test results and produce a report.
Georg Brandl116aa622007-08-15 14:28:22 +0000151
Benjamin Peterson52baa292009-03-24 00:56:30 +0000152When a :meth:`~TestCase.setUp` method is defined, the test runner will run that
153method prior to each test. Likewise, if a :meth:`~TestCase.tearDown` method is
154defined, the test runner will invoke that method after each test. In the
155example, :meth:`~TestCase.setUp` was used to create a fresh sequence for each
156test.
Georg Brandl116aa622007-08-15 14:28:22 +0000157
158The final block shows a simple way to run the tests. :func:`unittest.main`
Éric Araujo713d3032010-11-18 16:38:46 +0000159provides a command-line interface to the test script. When run from the command
Georg Brandl116aa622007-08-15 14:28:22 +0000160line, the above script produces an output that looks like this::
161
162 ...
163 ----------------------------------------------------------------------
164 Ran 3 tests in 0.000s
165
166 OK
167
168Instead of :func:`unittest.main`, there are other ways to run the tests with a
169finer level of control, less terse output, and no requirement to be run from the
170command line. For example, the last two lines may be replaced with::
171
172 suite = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)
173 unittest.TextTestRunner(verbosity=2).run(suite)
174
175Running the revised script from the interpreter or another script produces the
176following output::
177
Ezio Melottid59e44a2010-02-28 03:46:13 +0000178 test_choice (__main__.TestSequenceFunctions) ... ok
179 test_sample (__main__.TestSequenceFunctions) ... ok
180 test_shuffle (__main__.TestSequenceFunctions) ... ok
Georg Brandl116aa622007-08-15 14:28:22 +0000181
182 ----------------------------------------------------------------------
183 Ran 3 tests in 0.110s
184
185 OK
186
187The above examples show the most commonly used :mod:`unittest` features which
188are sufficient to meet many everyday testing needs. The remainder of the
189documentation explores the full feature set from first principles.
190
Benjamin Petersonb48af542010-04-11 20:43:16 +0000191
192.. _unittest-command-line-interface:
193
Éric Araujo76338ec2010-11-26 23:46:18 +0000194Command-Line Interface
Benjamin Petersonb48af542010-04-11 20:43:16 +0000195----------------------
196
197The unittest module can be used from the command line to run tests from
198modules, classes or even individual test methods::
199
200 python -m unittest test_module1 test_module2
201 python -m unittest test_module.TestClass
202 python -m unittest test_module.TestClass.test_method
203
204You can pass in a list with any combination of module names, and fully
205qualified class or method names.
206
Michael Foord37d120a2010-12-04 01:11:21 +0000207Test modules can be specified by file path as well::
208
209 python -m unittest tests/test_something.py
210
211This allows you to use the shell filename completion to specify the test module.
212The file specified must still be importable as a module. The path is converted
213to a module name by removing the '.py' and converting path separators into '.'.
214If you want to execute a test file that isn't importable as a module you should
215execute the file directly instead.
216
Benjamin Petersonb48af542010-04-11 20:43:16 +0000217You can run tests with more detail (higher verbosity) by passing in the -v flag::
218
219 python -m unittest -v test_module
220
Michael Foord086f3082010-11-21 21:28:01 +0000221When executed without arguments :ref:`unittest-test-discovery` is started::
222
223 python -m unittest
224
Éric Araujo713d3032010-11-18 16:38:46 +0000225For a list of all the command-line options::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000226
227 python -m unittest -h
228
Georg Brandl67b21b72010-08-17 15:07:14 +0000229.. versionchanged:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000230 In earlier versions it was only possible to run individual test methods and
231 not modules or classes.
232
233
Éric Araujo76338ec2010-11-26 23:46:18 +0000234Command-line options
235~~~~~~~~~~~~~~~~~~~~
Benjamin Petersonb48af542010-04-11 20:43:16 +0000236
Éric Araujod3309df2010-11-21 03:09:17 +0000237:program:`unittest` supports these command-line options:
Benjamin Petersonb48af542010-04-11 20:43:16 +0000238
Éric Araujo713d3032010-11-18 16:38:46 +0000239.. program:: unittest
Benjamin Petersonb48af542010-04-11 20:43:16 +0000240
Éric Araujo713d3032010-11-18 16:38:46 +0000241.. cmdoption:: -b, --buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +0000242
Éric Araujo713d3032010-11-18 16:38:46 +0000243 The standard output and standard error streams are buffered during the test
244 run. Output during a passing test is discarded. Output is echoed normally
245 on test fail or error and is added to the failure messages.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000246
Éric Araujo713d3032010-11-18 16:38:46 +0000247.. cmdoption:: -c, --catch
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000248
Éric Araujo713d3032010-11-18 16:38:46 +0000249 Control-C during the test run waits for the current test to end and then
250 reports all the results so far. A second control-C raises the normal
251 :exc:`KeyboardInterrupt` exception.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000252
Éric Araujo713d3032010-11-18 16:38:46 +0000253 See `Signal Handling`_ for the functions that provide this functionality.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000254
Éric Araujo713d3032010-11-18 16:38:46 +0000255.. cmdoption:: -f, --failfast
256
257 Stop the test run on the first error or failure.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000258
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000259.. versionadded:: 3.2
Éric Araujod6c5f742010-12-16 00:07:01 +0000260 The command-line options ``-b``, ``-c`` and ``-f`` were added.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000261
262The command line can also be used for test discovery, for running all of the
263tests in a project or just a subset.
264
265
266.. _unittest-test-discovery:
267
268Test Discovery
269--------------
270
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000271.. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000272
273Unittest supports simple test discovery. For a project's tests to be
274compatible with test discovery they must all be importable from the top level
275directory of the project (in other words, they must all be in Python packages).
276
277Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
Éric Araujo713d3032010-11-18 16:38:46 +0000278used from the command line. The basic command-line usage is::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000279
280 cd project_directory
281 python -m unittest discover
282
Michael Foord086f3082010-11-21 21:28:01 +0000283.. note::
284
285 As a shortcut, ``python -m unittest`` is the equivalent of
286 ``python -m unittest discover``. If you want to pass arguments to test
287 discovery the `discover` sub-command must be used explicitly.
288
Benjamin Petersonb48af542010-04-11 20:43:16 +0000289The ``discover`` sub-command has the following options:
290
Éric Araujo713d3032010-11-18 16:38:46 +0000291.. program:: unittest discover
292
293.. cmdoption:: -v, --verbose
294
295 Verbose output
296
297.. cmdoption:: -s directory
298
299 Directory to start discovery ('.' default)
300
301.. cmdoption:: -p pattern
302
303 Pattern to match test files ('test*.py' default)
304
305.. cmdoption:: -t directory
306
307 Top level directory of project (defaults to start directory)
Benjamin Petersonb48af542010-04-11 20:43:16 +0000308
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000309The :option:`-s`, :option:`-p`, and :option:`-t` options can be passed in
310as positional arguments in that order. The following two command lines
311are equivalent::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000312
313 python -m unittest discover -s project_directory -p '*_test.py'
314 python -m unittest discover project_directory '*_test.py'
315
Michael Foord16f3e902010-05-08 15:13:42 +0000316As well as being a path it is possible to pass a package name, for example
317``myproject.subpackage.test``, as the start directory. The package name you
318supply will then be imported and its location on the filesystem will be used
319as the start directory.
320
321.. caution::
322
Senthil Kumaran916bd382010-10-15 12:55:19 +0000323 Test discovery loads tests by importing them. Once test discovery has found
324 all the test files from the start directory you specify it turns the paths
325 into package names to import. For example :file:`foo/bar/baz.py` will be
Michael Foord16f3e902010-05-08 15:13:42 +0000326 imported as ``foo.bar.baz``.
327
328 If you have a package installed globally and attempt test discovery on
329 a different copy of the package then the import *could* happen from the
330 wrong place. If this happens test discovery will warn you and exit.
331
332 If you supply the start directory as a package name rather than a
333 path to a directory then discover assumes that whichever location it
334 imports from is the location you intended, so you will not get the
335 warning.
336
Benjamin Petersonb48af542010-04-11 20:43:16 +0000337Test modules and packages can customize test loading and discovery by through
338the `load_tests protocol`_.
339
340
Georg Brandl116aa622007-08-15 14:28:22 +0000341.. _organizing-tests:
342
343Organizing test code
344--------------------
345
346The basic building blocks of unit testing are :dfn:`test cases` --- single
347scenarios that must be set up and checked for correctness. In :mod:`unittest`,
348test cases are represented by instances of :mod:`unittest`'s :class:`TestCase`
349class. To make your own test cases you must write subclasses of
350:class:`TestCase`, or use :class:`FunctionTestCase`.
351
352An instance of a :class:`TestCase`\ -derived class is an object that can
353completely run a single test method, together with optional set-up and tidy-up
354code.
355
356The testing code of a :class:`TestCase` instance should be entirely self
357contained, such that it can be run either in isolation or in arbitrary
358combination with any number of other test cases.
359
Benjamin Peterson52baa292009-03-24 00:56:30 +0000360The simplest :class:`TestCase` subclass will simply override the
361:meth:`~TestCase.runTest` method in order to perform specific testing code::
Georg Brandl116aa622007-08-15 14:28:22 +0000362
363 import unittest
364
365 class DefaultWidgetSizeTestCase(unittest.TestCase):
366 def runTest(self):
367 widget = Widget('The widget')
368 self.assertEqual(widget.size(), (50, 50), 'incorrect default size')
369
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000370Note that in order to test something, we use the one of the :meth:`assert\*`
Benjamin Petersond2397752009-06-27 23:45:02 +0000371methods provided by the :class:`TestCase` base class. If the test fails, an
372exception will be raised, and :mod:`unittest` will identify the test case as a
373:dfn:`failure`. Any other exceptions will be treated as :dfn:`errors`. This
374helps you identify where the problem is: :dfn:`failures` are caused by incorrect
375results - a 5 where you expected a 6. :dfn:`Errors` are caused by incorrect
376code - e.g., a :exc:`TypeError` caused by an incorrect function call.
Georg Brandl116aa622007-08-15 14:28:22 +0000377
378The way to run a test case will be described later. For now, note that to
379construct an instance of such a test case, we call its constructor without
380arguments::
381
382 testCase = DefaultWidgetSizeTestCase()
383
384Now, such test cases can be numerous, and their set-up can be repetitive. In
385the above case, constructing a :class:`Widget` in each of 100 Widget test case
386subclasses would mean unsightly duplication.
387
388Luckily, we can factor out such set-up code by implementing a method called
Benjamin Peterson52baa292009-03-24 00:56:30 +0000389:meth:`~TestCase.setUp`, which the testing framework will automatically call for
390us when we run the test::
Georg Brandl116aa622007-08-15 14:28:22 +0000391
392 import unittest
393
394 class SimpleWidgetTestCase(unittest.TestCase):
395 def setUp(self):
396 self.widget = Widget('The widget')
397
398 class DefaultWidgetSizeTestCase(SimpleWidgetTestCase):
399 def runTest(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000400 self.assertEqual(self.widget.size(), (50,50),
401 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000402
403 class WidgetResizeTestCase(SimpleWidgetTestCase):
404 def runTest(self):
405 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000406 self.assertEqual(self.widget.size(), (100,150),
407 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000408
Benjamin Peterson52baa292009-03-24 00:56:30 +0000409If the :meth:`~TestCase.setUp` method raises an exception while the test is
410running, the framework will consider the test to have suffered an error, and the
411:meth:`~TestCase.runTest` method will not be executed.
Georg Brandl116aa622007-08-15 14:28:22 +0000412
Benjamin Peterson52baa292009-03-24 00:56:30 +0000413Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
414after the :meth:`~TestCase.runTest` method has been run::
Georg Brandl116aa622007-08-15 14:28:22 +0000415
416 import unittest
417
418 class SimpleWidgetTestCase(unittest.TestCase):
419 def setUp(self):
420 self.widget = Widget('The widget')
421
422 def tearDown(self):
423 self.widget.dispose()
424 self.widget = None
425
Benjamin Peterson52baa292009-03-24 00:56:30 +0000426If :meth:`~TestCase.setUp` succeeded, the :meth:`~TestCase.tearDown` method will
427be run whether :meth:`~TestCase.runTest` succeeded or not.
Georg Brandl116aa622007-08-15 14:28:22 +0000428
429Such a working environment for the testing code is called a :dfn:`fixture`.
430
431Often, many small test cases will use the same fixture. In this case, we would
432end up subclassing :class:`SimpleWidgetTestCase` into many small one-method
433classes such as :class:`DefaultWidgetSizeTestCase`. This is time-consuming and
Georg Brandl116aa622007-08-15 14:28:22 +0000434discouraging, so in the same vein as JUnit, :mod:`unittest` provides a simpler
435mechanism::
436
437 import unittest
438
439 class WidgetTestCase(unittest.TestCase):
440 def setUp(self):
441 self.widget = Widget('The widget')
442
443 def tearDown(self):
444 self.widget.dispose()
445 self.widget = None
446
Ezio Melottid59e44a2010-02-28 03:46:13 +0000447 def test_default_size(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000448 self.assertEqual(self.widget.size(), (50,50),
449 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000450
Ezio Melottid59e44a2010-02-28 03:46:13 +0000451 def test_resize(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000452 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000453 self.assertEqual(self.widget.size(), (100,150),
454 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000455
Benjamin Peterson52baa292009-03-24 00:56:30 +0000456Here we have not provided a :meth:`~TestCase.runTest` method, but have instead
457provided two different test methods. Class instances will now each run one of
Ezio Melottid59e44a2010-02-28 03:46:13 +0000458the :meth:`test_\*` methods, with ``self.widget`` created and destroyed
Benjamin Peterson52baa292009-03-24 00:56:30 +0000459separately for each instance. When creating an instance we must specify the
460test method it is to run. We do this by passing the method name in the
461constructor::
Georg Brandl116aa622007-08-15 14:28:22 +0000462
Ezio Melottid59e44a2010-02-28 03:46:13 +0000463 defaultSizeTestCase = WidgetTestCase('test_default_size')
464 resizeTestCase = WidgetTestCase('test_resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000465
466Test case instances are grouped together according to the features they test.
467:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
468represented by :mod:`unittest`'s :class:`TestSuite` class::
469
470 widgetTestSuite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000471 widgetTestSuite.addTest(WidgetTestCase('test_default_size'))
472 widgetTestSuite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000473
474For the ease of running tests, as we will see later, it is a good idea to
475provide in each test module a callable object that returns a pre-built test
476suite::
477
478 def suite():
479 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000480 suite.addTest(WidgetTestCase('test_default_size'))
481 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000482 return suite
483
484or even::
485
486 def suite():
Ezio Melottid59e44a2010-02-28 03:46:13 +0000487 tests = ['test_default_size', 'test_resize']
Georg Brandl116aa622007-08-15 14:28:22 +0000488
489 return unittest.TestSuite(map(WidgetTestCase, tests))
490
491Since it is a common pattern to create a :class:`TestCase` subclass with many
492similarly named test functions, :mod:`unittest` provides a :class:`TestLoader`
493class that can be used to automate the process of creating a test suite and
494populating it with individual tests. For example, ::
495
496 suite = unittest.TestLoader().loadTestsFromTestCase(WidgetTestCase)
497
Ezio Melottid59e44a2010-02-28 03:46:13 +0000498will create a test suite that will run ``WidgetTestCase.test_default_size()`` and
499``WidgetTestCase.test_resize``. :class:`TestLoader` uses the ``'test'`` method
Georg Brandl116aa622007-08-15 14:28:22 +0000500name prefix to identify test methods automatically.
501
Mark Dickinsonc48d8342009-02-01 14:18:10 +0000502Note that the order in which the various test cases will be run is
503determined by sorting the test function names with respect to the
504built-in ordering for strings.
Georg Brandl116aa622007-08-15 14:28:22 +0000505
506Often it is desirable to group suites of test cases together, so as to run tests
507for the whole system at once. This is easy, since :class:`TestSuite` instances
508can be added to a :class:`TestSuite` just as :class:`TestCase` instances can be
509added to a :class:`TestSuite`::
510
511 suite1 = module1.TheTestSuite()
512 suite2 = module2.TheTestSuite()
513 alltests = unittest.TestSuite([suite1, suite2])
514
515You can place the definitions of test cases and test suites in the same modules
516as the code they are to test (such as :file:`widget.py`), but there are several
517advantages to placing the test code in a separate module, such as
518:file:`test_widget.py`:
519
520* The test module can be run standalone from the command line.
521
522* The test code can more easily be separated from shipped code.
523
524* There is less temptation to change test code to fit the code it tests without
525 a good reason.
526
527* Test code should be modified much less frequently than the code it tests.
528
529* Tested code can be refactored more easily.
530
531* Tests for modules written in C must be in separate modules anyway, so why not
532 be consistent?
533
534* If the testing strategy changes, there is no need to change the source code.
535
536
537.. _legacy-unit-tests:
538
539Re-using old test code
540----------------------
541
542Some users will find that they have existing test code that they would like to
543run from :mod:`unittest`, without converting every old test function to a
544:class:`TestCase` subclass.
545
546For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
547This subclass of :class:`TestCase` can be used to wrap an existing test
548function. Set-up and tear-down functions can also be provided.
549
550Given the following test function::
551
552 def testSomething():
553 something = makeSomething()
554 assert something.name is not None
555 # ...
556
557one can create an equivalent test case instance as follows::
558
559 testcase = unittest.FunctionTestCase(testSomething)
560
561If there are additional set-up and tear-down methods that should be called as
562part of the test case's operation, they can also be provided like so::
563
564 testcase = unittest.FunctionTestCase(testSomething,
565 setUp=makeSomethingDB,
566 tearDown=deleteSomethingDB)
567
568To make migrating existing test suites easier, :mod:`unittest` supports tests
569raising :exc:`AssertionError` to indicate test failure. However, it is
570recommended that you use the explicit :meth:`TestCase.fail\*` and
571:meth:`TestCase.assert\*` methods instead, as future versions of :mod:`unittest`
572may treat :exc:`AssertionError` differently.
573
574.. note::
575
Benjamin Petersond2397752009-06-27 23:45:02 +0000576 Even though :class:`FunctionTestCase` can be used to quickly convert an
577 existing test base over to a :mod:`unittest`\ -based system, this approach is
578 not recommended. Taking the time to set up proper :class:`TestCase`
579 subclasses will make future test refactorings infinitely easier.
Georg Brandl116aa622007-08-15 14:28:22 +0000580
Benjamin Peterson52baa292009-03-24 00:56:30 +0000581In some cases, the existing tests may have been written using the :mod:`doctest`
582module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
583automatically build :class:`unittest.TestSuite` instances from the existing
584:mod:`doctest`\ -based tests.
585
Georg Brandl116aa622007-08-15 14:28:22 +0000586
Benjamin Peterson5254c042009-03-23 22:25:03 +0000587.. _unittest-skipping:
588
589Skipping tests and expected failures
590------------------------------------
591
Michael Foordf5c851a2010-02-05 21:48:03 +0000592.. versionadded:: 3.1
593
Benjamin Peterson5254c042009-03-23 22:25:03 +0000594Unittest supports skipping individual test methods and even whole classes of
595tests. In addition, it supports marking a test as a "expected failure," a test
596that is broken and will fail, but shouldn't be counted as a failure on a
597:class:`TestResult`.
598
599Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
600or one of its conditional variants.
601
602Basic skipping looks like this: ::
603
604 class MyTestCase(unittest.TestCase):
605
606 @unittest.skip("demonstrating skipping")
607 def test_nothing(self):
608 self.fail("shouldn't happen")
609
Benjamin Petersond2397752009-06-27 23:45:02 +0000610 @unittest.skipIf(mylib.__version__ < (1, 3),
611 "not supported in this library version")
Benjamin Petersonded31c42009-03-30 15:04:16 +0000612 def test_format(self):
613 # Tests that work for only a certain version of the library.
614 pass
615
616 @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
617 def test_windows_support(self):
618 # windows specific testing code
619 pass
620
Benjamin Peterson5254c042009-03-23 22:25:03 +0000621This is the output of running the example above in verbose mode: ::
622
Benjamin Petersonded31c42009-03-30 15:04:16 +0000623 test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000624 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
Benjamin Petersonded31c42009-03-30 15:04:16 +0000625 test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000626
627 ----------------------------------------------------------------------
Benjamin Petersonded31c42009-03-30 15:04:16 +0000628 Ran 3 tests in 0.005s
629
630 OK (skipped=3)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000631
632Classes can be skipped just like methods: ::
633
634 @skip("showing class skipping")
635 class MySkippedTestCase(unittest.TestCase):
636 def test_not_run(self):
637 pass
638
Benjamin Peterson52baa292009-03-24 00:56:30 +0000639:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
640that needs to be set up is not available.
641
Benjamin Peterson5254c042009-03-23 22:25:03 +0000642Expected failures use the :func:`expectedFailure` decorator. ::
643
644 class ExpectedFailureTestCase(unittest.TestCase):
645 @unittest.expectedFailure
646 def test_fail(self):
647 self.assertEqual(1, 0, "broken")
648
649It's easy to roll your own skipping decorators by making a decorator that calls
650:func:`skip` on the test when it wants it to be skipped. This decorator skips
651the test unless the passed object has a certain attribute: ::
652
653 def skipUnlessHasattr(obj, attr):
654 if hasattr(obj, attr):
655 return lambda func: func
656 return unittest.skip("{0!r} doesn't have {1!r}".format(obj, attr))
657
658The following decorators implement test skipping and expected failures:
659
Georg Brandl8a1caa22010-07-29 16:01:11 +0000660.. decorator:: skip(reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000661
662 Unconditionally skip the decorated test. *reason* should describe why the
663 test is being skipped.
664
Georg Brandl8a1caa22010-07-29 16:01:11 +0000665.. decorator:: skipIf(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000666
667 Skip the decorated test if *condition* is true.
668
Georg Brandl8a1caa22010-07-29 16:01:11 +0000669.. decorator:: skipUnless(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000670
Georg Brandl6faee4e2010-09-21 14:48:28 +0000671 Skip the decorated test unless *condition* is true.
Benjamin Peterson5254c042009-03-23 22:25:03 +0000672
Georg Brandl8a1caa22010-07-29 16:01:11 +0000673.. decorator:: expectedFailure
Benjamin Peterson5254c042009-03-23 22:25:03 +0000674
675 Mark the test as an expected failure. If the test fails when run, the test
676 is not counted as a failure.
677
Benjamin Petersonb48af542010-04-11 20:43:16 +0000678Skipped tests will not have :meth:`setUp` or :meth:`tearDown` run around them.
679Skipped classes will not have :meth:`setUpClass` or :meth:`tearDownClass` run.
680
Benjamin Peterson5254c042009-03-23 22:25:03 +0000681
Georg Brandl116aa622007-08-15 14:28:22 +0000682.. _unittest-contents:
683
684Classes and functions
685---------------------
686
Benjamin Peterson52baa292009-03-24 00:56:30 +0000687This section describes in depth the API of :mod:`unittest`.
688
689
690.. _testcase-objects:
691
692Test cases
693~~~~~~~~~~
Georg Brandl116aa622007-08-15 14:28:22 +0000694
Georg Brandl7f01a132009-09-16 15:58:14 +0000695.. class:: TestCase(methodName='runTest')
Georg Brandl116aa622007-08-15 14:28:22 +0000696
697 Instances of the :class:`TestCase` class represent the smallest testable units
698 in the :mod:`unittest` universe. This class is intended to be used as a base
699 class, with specific tests being implemented by concrete subclasses. This class
700 implements the interface needed by the test runner to allow it to drive the
701 test, and methods that the test code can use to check for and report various
702 kinds of failure.
703
704 Each instance of :class:`TestCase` will run a single test method: the method
705 named *methodName*. If you remember, we had an earlier example that went
706 something like this::
707
708 def suite():
709 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000710 suite.addTest(WidgetTestCase('test_default_size'))
711 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000712 return suite
713
714 Here, we create two instances of :class:`WidgetTestCase`, each of which runs a
715 single test.
716
Benjamin Peterson52baa292009-03-24 00:56:30 +0000717 *methodName* defaults to :meth:`runTest`.
718
719 :class:`TestCase` instances provide three groups of methods: one group used
720 to run the test, another used by the test implementation to check conditions
721 and report failures, and some inquiry methods allowing information about the
722 test itself to be gathered.
723
724 Methods in the first group (running the test) are:
725
726
727 .. method:: setUp()
728
729 Method called to prepare the test fixture. This is called immediately
730 before calling the test method; any exception raised by this method will
731 be considered an error rather than a test failure. The default
732 implementation does nothing.
733
734
735 .. method:: tearDown()
736
737 Method called immediately after the test method has been called and the
738 result recorded. This is called even if the test method raised an
739 exception, so the implementation in subclasses may need to be particularly
740 careful about checking internal state. Any exception raised by this
741 method will be considered an error rather than a test failure. This
742 method will only be called if the :meth:`setUp` succeeds, regardless of
743 the outcome of the test method. The default implementation does nothing.
744
745
Benjamin Petersonb48af542010-04-11 20:43:16 +0000746 .. method:: setUpClass()
747
748 A class method called before tests in an individual class run.
749 ``setUpClass`` is called with the class as the only argument
750 and must be decorated as a :func:`classmethod`::
751
752 @classmethod
753 def setUpClass(cls):
754 ...
755
756 See `Class and Module Fixtures`_ for more details.
757
758 .. versionadded:: 3.2
759
760
761 .. method:: tearDownClass()
762
763 A class method called after tests in an individual class have run.
764 ``tearDownClass`` is called with the class as the only argument
765 and must be decorated as a :meth:`classmethod`::
766
767 @classmethod
768 def tearDownClass(cls):
769 ...
770
771 See `Class and Module Fixtures`_ for more details.
772
773 .. versionadded:: 3.2
774
775
Georg Brandl7f01a132009-09-16 15:58:14 +0000776 .. method:: run(result=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000777
778 Run the test, collecting the result into the test result object passed as
Ezio Melotti75b2a5e2010-11-20 10:13:45 +0000779 *result*. If *result* is omitted or ``None``, a temporary result
Alexandre Vassalotti260484d2009-07-17 11:43:26 +0000780 object is created (by calling the :meth:`defaultTestResult` method) and
781 used. The result object is not returned to :meth:`run`'s caller.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000782
783 The same effect may be had by simply calling the :class:`TestCase`
784 instance.
785
786
Benjamin Petersone549ead2009-03-28 21:42:05 +0000787 .. method:: skipTest(reason)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000788
Stefan Kraha5bf3f52010-05-19 16:09:41 +0000789 Calling this during a test method or :meth:`setUp` skips the current
Benjamin Peterson52baa292009-03-24 00:56:30 +0000790 test. See :ref:`unittest-skipping` for more information.
791
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000792 .. versionadded:: 3.1
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000793
Benjamin Peterson52baa292009-03-24 00:56:30 +0000794
795 .. method:: debug()
796
797 Run the test without collecting the result. This allows exceptions raised
798 by the test to be propagated to the caller, and can be used to support
799 running tests under a debugger.
800
Ezio Melotti22170ed2010-11-20 09:57:27 +0000801 .. _assert-methods:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000802
Ezio Melotti4370b302010-11-03 20:39:14 +0000803 The :class:`TestCase` class provides a number of methods to check for and
804 report failures, such as:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000805
Ezio Melotti4370b302010-11-03 20:39:14 +0000806 +-----------------------------------------+-----------------------------+---------------+
807 | Method | Checks that | New in |
808 +=========================================+=============================+===============+
809 | :meth:`assertEqual(a, b) | ``a == b`` | |
810 | <TestCase.assertEqual>` | | |
811 +-----------------------------------------+-----------------------------+---------------+
812 | :meth:`assertNotEqual(a, b) | ``a != b`` | |
813 | <TestCase.assertNotEqual>` | | |
814 +-----------------------------------------+-----------------------------+---------------+
815 | :meth:`assertTrue(x) | ``bool(x) is True`` | |
816 | <TestCase.assertTrue>` | | |
817 +-----------------------------------------+-----------------------------+---------------+
818 | :meth:`assertFalse(x) | ``bool(x) is False`` | |
819 | <TestCase.assertFalse>` | | |
820 +-----------------------------------------+-----------------------------+---------------+
821 | :meth:`assertIs(a, b) | ``a is b`` | 3.1 |
822 | <TestCase.assertIs>` | | |
823 +-----------------------------------------+-----------------------------+---------------+
824 | :meth:`assertIsNot(a, b) | ``a is not b`` | 3.1 |
825 | <TestCase.assertIsNot>` | | |
826 +-----------------------------------------+-----------------------------+---------------+
827 | :meth:`assertIsNone(x) | ``x is None`` | 3.1 |
828 | <TestCase.assertIsNone>` | | |
829 +-----------------------------------------+-----------------------------+---------------+
830 | :meth:`assertIsNotNone(x) | ``x is not None`` | 3.1 |
831 | <TestCase.assertIsNotNone>` | | |
832 +-----------------------------------------+-----------------------------+---------------+
833 | :meth:`assertIn(a, b) | ``a in b`` | 3.1 |
834 | <TestCase.assertIn>` | | |
835 +-----------------------------------------+-----------------------------+---------------+
836 | :meth:`assertNotIn(a, b) | ``a not in b`` | 3.1 |
837 | <TestCase.assertNotIn>` | | |
838 +-----------------------------------------+-----------------------------+---------------+
839 | :meth:`assertIsInstance(a, b) | ``isinstance(a, b)`` | 3.2 |
840 | <TestCase.assertIsInstance>` | | |
841 +-----------------------------------------+-----------------------------+---------------+
842 | :meth:`assertNotIsInstance(a, b) | ``not isinstance(a, b)`` | 3.2 |
843 | <TestCase.assertNotIsInstance>` | | |
844 +-----------------------------------------+-----------------------------+---------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000845
Ezio Melotti22170ed2010-11-20 09:57:27 +0000846 All the assert methods (except :meth:`assertRaises`,
Ezio Melottied3a7d22010-12-01 02:32:32 +0000847 :meth:`assertRaisesRegex`, :meth:`assertWarns`, :meth:`assertWarnsRegex`)
Ezio Melotti22170ed2010-11-20 09:57:27 +0000848 accept a *msg* argument that, if specified, is used as the error message on
849 failure (see also :data:`longMessage`).
Benjamin Peterson52baa292009-03-24 00:56:30 +0000850
Georg Brandl7f01a132009-09-16 15:58:14 +0000851 .. method:: assertEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000852
853 Test that *first* and *second* are equal. If the values do not compare
Ezio Melotti4841fd62010-11-05 15:43:40 +0000854 equal, the test will fail.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000855
856 In addition, if *first* and *second* are the exact same type and one of
Michael Foord02834952010-02-08 23:10:39 +0000857 list, tuple, dict, set, frozenset or str or any type that a subclass
858 registers with :meth:`addTypeEqualityFunc` the type specific equality
859 function will be called in order to generate a more useful default
Ezio Melotti22170ed2010-11-20 09:57:27 +0000860 error message (see also the :ref:`list of type-specific methods
861 <type-specific-methods>`).
Ezio Melotti4841fd62010-11-05 15:43:40 +0000862
Raymond Hettinger35a88362009-04-09 00:08:24 +0000863 .. versionchanged:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000864 Added the automatic calling of type specific equality function.
865
Michael Foord28a817e2010-02-09 00:03:57 +0000866 .. versionchanged:: 3.2
867 :meth:`assertMultiLineEqual` added as the default type equality
868 function for comparing strings.
Michael Foord02834952010-02-08 23:10:39 +0000869
Benjamin Peterson52baa292009-03-24 00:56:30 +0000870
Georg Brandl7f01a132009-09-16 15:58:14 +0000871 .. method:: assertNotEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000872
873 Test that *first* and *second* are not equal. If the values do compare
Ezio Melotti4841fd62010-11-05 15:43:40 +0000874 equal, the test will fail.
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000875
Ezio Melotti4370b302010-11-03 20:39:14 +0000876 .. method:: assertTrue(expr, msg=None)
Ezio Melotti4841fd62010-11-05 15:43:40 +0000877 assertFalse(expr, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000878
Ezio Melotti4841fd62010-11-05 15:43:40 +0000879 Test that *expr* is true (or false).
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000880
Ezio Melotti4841fd62010-11-05 15:43:40 +0000881 Note that this is equivalent to ``bool(expr) is True`` and not to ``expr
882 is True`` (use ``assertIs(expr, True)`` for the latter). This method
883 should also be avoided when more specific methods are available (e.g.
884 ``assertEqual(a, b)`` instead of ``assertTrue(a == b)``), because they
885 provide a better error message in case of failure.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000886
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000887
Ezio Melotti9794a262010-11-04 14:52:13 +0000888 .. method:: assertIs(first, second, msg=None)
889 assertIsNot(first, second, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000890
Ezio Melotti9794a262010-11-04 14:52:13 +0000891 Test that *first* and *second* evaluate (or don't evaluate) to the same object.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000892
Raymond Hettinger35a88362009-04-09 00:08:24 +0000893 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000894
895
Ezio Melotti4370b302010-11-03 20:39:14 +0000896 .. method:: assertIsNone(expr, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000897 assertIsNotNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000898
Ezio Melotti9794a262010-11-04 14:52:13 +0000899 Test that *expr* is (or is not) None.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000900
Ezio Melotti4370b302010-11-03 20:39:14 +0000901 .. versionadded:: 3.1
Benjamin Petersonb48af542010-04-11 20:43:16 +0000902
903
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000904 .. method:: assertIn(first, second, msg=None)
905 assertNotIn(first, second, msg=None)
906
Ezio Melotti4841fd62010-11-05 15:43:40 +0000907 Test that *first* is (or is not) in *second*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000908
Raymond Hettinger35a88362009-04-09 00:08:24 +0000909 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000910
911
Ezio Melotti9c02c2f2010-11-03 20:45:31 +0000912 .. method:: assertIsInstance(obj, cls, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000913 assertNotIsInstance(obj, cls, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000914
Ezio Melotti9794a262010-11-04 14:52:13 +0000915 Test that *obj* is (or is not) an instance of *cls* (which can be a
916 class or a tuple of classes, as supported by :func:`isinstance`).
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000917
Ezio Melotti4370b302010-11-03 20:39:14 +0000918 .. versionadded:: 3.2
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000919
920
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000921
Ezio Melotti4370b302010-11-03 20:39:14 +0000922 It is also possible to check that exceptions and warnings are raised using
923 the following methods:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000924
Ezio Melotti4370b302010-11-03 20:39:14 +0000925 +---------------------------------------------------------+--------------------------------------+------------+
926 | Method | Checks that | New in |
927 +=========================================================+======================================+============+
928 | :meth:`assertRaises(exc, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `exc` | |
929 | <TestCase.assertRaises>` | | |
930 +---------------------------------------------------------+--------------------------------------+------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +0000931 | :meth:`assertRaisesRegex(exc, re, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `exc` | 3.1 |
932 | <TestCase.assertRaisesRegex>` | and the message matches `re` | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000933 +---------------------------------------------------------+--------------------------------------+------------+
934 | :meth:`assertWarns(warn, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `warn` | 3.2 |
935 | <TestCase.assertWarns>` | | |
936 +---------------------------------------------------------+--------------------------------------+------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +0000937 | :meth:`assertWarnsRegex(warn, re, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `warn` | 3.2 |
938 | <TestCase.assertWarnsRegex>` | and the message matches `re` | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000939 +---------------------------------------------------------+--------------------------------------+------------+
Benjamin Peterson52baa292009-03-24 00:56:30 +0000940
Georg Brandl7f01a132009-09-16 15:58:14 +0000941 .. method:: assertRaises(exception, callable, *args, **kwds)
Georg Brandl7f01a132009-09-16 15:58:14 +0000942 assertRaises(exception)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000943
944 Test that an exception is raised when *callable* is called with any
945 positional or keyword arguments that are also passed to
946 :meth:`assertRaises`. The test passes if *exception* is raised, is an
947 error if another exception is raised, or fails if no exception is raised.
948 To catch any of a group of exceptions, a tuple containing the exception
949 classes may be passed as *exception*.
950
Georg Brandl7f01a132009-09-16 15:58:14 +0000951 If only the *exception* argument is given, returns a context manager so
952 that the code under test can be written inline rather than as a function::
Benjamin Petersonded31c42009-03-30 15:04:16 +0000953
Michael Foord41531f22010-02-05 21:13:40 +0000954 with self.assertRaises(SomeException):
Benjamin Petersonded31c42009-03-30 15:04:16 +0000955 do_something()
956
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000957 The context manager will store the caught exception object in its
Ezio Melotti49008232010-02-08 21:57:48 +0000958 :attr:`exception` attribute. This can be useful if the intention
Michael Foord41531f22010-02-05 21:13:40 +0000959 is to perform additional checks on the exception raised::
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000960
Georg Brandl8a1caa22010-07-29 16:01:11 +0000961 with self.assertRaises(SomeException) as cm:
962 do_something()
Michael Foord41531f22010-02-05 21:13:40 +0000963
Georg Brandl8a1caa22010-07-29 16:01:11 +0000964 the_exception = cm.exception
965 self.assertEqual(the_exception.error_code, 3)
Michael Foord41531f22010-02-05 21:13:40 +0000966
Ezio Melotti49008232010-02-08 21:57:48 +0000967 .. versionchanged:: 3.1
Benjamin Petersonded31c42009-03-30 15:04:16 +0000968 Added the ability to use :meth:`assertRaises` as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000969
Ezio Melotti49008232010-02-08 21:57:48 +0000970 .. versionchanged:: 3.2
971 Added the :attr:`exception` attribute.
972
Benjamin Peterson52baa292009-03-24 00:56:30 +0000973
Ezio Melottied3a7d22010-12-01 02:32:32 +0000974 .. method:: assertRaisesRegex(exception, regex, callable, *args, **kwds)
975 assertRaisesRegex(exception, regex)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000976
Ezio Melottied3a7d22010-12-01 02:32:32 +0000977 Like :meth:`assertRaises` but also tests that *regex* matches
978 on the string representation of the raised exception. *regex* may be
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000979 a regular expression object or a string containing a regular expression
980 suitable for use by :func:`re.search`. Examples::
981
Ezio Melottied3a7d22010-12-01 02:32:32 +0000982 self.assertRaisesRegex(ValueError, 'invalid literal for.*XYZ$',
983 int, 'XYZ')
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000984
985 or::
986
Ezio Melottied3a7d22010-12-01 02:32:32 +0000987 with self.assertRaisesRegex(ValueError, 'literal'):
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000988 int('XYZ')
989
Georg Brandl419e3de2010-12-01 15:44:25 +0000990 .. versionadded:: 3.1
991 under the name ``assertRaisesRegexp``.
Ezio Melottied3a7d22010-12-01 02:32:32 +0000992 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +0000993 Renamed to :meth:`assertRaisesRegex`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000994
995
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000996 .. method:: assertWarns(warning, callable, *args, **kwds)
997 assertWarns(warning)
998
999 Test that a warning is triggered when *callable* is called with any
1000 positional or keyword arguments that are also passed to
1001 :meth:`assertWarns`. The test passes if *warning* is triggered and
1002 fails if it isn't. Also, any unexpected exception is an error.
1003 To catch any of a group of warnings, a tuple containing the warning
1004 classes may be passed as *warnings*.
1005
1006 If only the *warning* argument is given, returns a context manager so
1007 that the code under test can be written inline rather than as a function::
1008
1009 with self.assertWarns(SomeWarning):
1010 do_something()
1011
1012 The context manager will store the caught warning object in its
1013 :attr:`warning` attribute, and the source line which triggered the
1014 warnings in the :attr:`filename` and :attr:`lineno` attributes.
1015 This can be useful if the intention is to perform additional checks
1016 on the exception raised::
1017
1018 with self.assertWarns(SomeWarning) as cm:
1019 do_something()
1020
1021 self.assertIn('myfile.py', cm.filename)
1022 self.assertEqual(320, cm.lineno)
1023
1024 This method works regardless of the warning filters in place when it
1025 is called.
1026
1027 .. versionadded:: 3.2
1028
1029
Ezio Melottied3a7d22010-12-01 02:32:32 +00001030 .. method:: assertWarnsRegex(warning, regex, callable, *args, **kwds)
1031 assertWarnsRegex(warning, regex)
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001032
Ezio Melottied3a7d22010-12-01 02:32:32 +00001033 Like :meth:`assertWarns` but also tests that *regex* matches on the
1034 message of the triggered warning. *regex* may be a regular expression
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001035 object or a string containing a regular expression suitable for use
1036 by :func:`re.search`. Example::
1037
Ezio Melottied3a7d22010-12-01 02:32:32 +00001038 self.assertWarnsRegex(DeprecationWarning,
1039 r'legacy_function\(\) is deprecated',
1040 legacy_function, 'XYZ')
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001041
1042 or::
1043
Ezio Melottied3a7d22010-12-01 02:32:32 +00001044 with self.assertWarnsRegex(RuntimeWarning, 'unsafe frobnicating'):
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001045 frobnicate('/etc/passwd')
1046
1047 .. versionadded:: 3.2
1048
1049
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001050
Ezio Melotti4370b302010-11-03 20:39:14 +00001051 There are also other methods used to perform more specific checks, such as:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001052
Ezio Melotti4370b302010-11-03 20:39:14 +00001053 +---------------------------------------+--------------------------------+--------------+
1054 | Method | Checks that | New in |
1055 +=======================================+================================+==============+
1056 | :meth:`assertAlmostEqual(a, b) | ``round(a-b, 7) == 0`` | |
1057 | <TestCase.assertAlmostEqual>` | | |
1058 +---------------------------------------+--------------------------------+--------------+
1059 | :meth:`assertNotAlmostEqual(a, b) | ``round(a-b, 7) != 0`` | |
1060 | <TestCase.assertNotAlmostEqual>` | | |
1061 +---------------------------------------+--------------------------------+--------------+
1062 | :meth:`assertGreater(a, b) | ``a > b`` | 3.1 |
1063 | <TestCase.assertGreater>` | | |
1064 +---------------------------------------+--------------------------------+--------------+
1065 | :meth:`assertGreaterEqual(a, b) | ``a >= b`` | 3.1 |
1066 | <TestCase.assertGreaterEqual>` | | |
1067 +---------------------------------------+--------------------------------+--------------+
1068 | :meth:`assertLess(a, b) | ``a < b`` | 3.1 |
1069 | <TestCase.assertLess>` | | |
1070 +---------------------------------------+--------------------------------+--------------+
1071 | :meth:`assertLessEqual(a, b) | ``a <= b`` | 3.1 |
1072 | <TestCase.assertLessEqual>` | | |
1073 +---------------------------------------+--------------------------------+--------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +00001074 | :meth:`assertRegex(s, re) | ``regex.search(s)`` | 3.1 |
1075 | <TestCase.assertRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001076 +---------------------------------------+--------------------------------+--------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +00001077 | :meth:`assertNotRegex(s, re) | ``not regex.search(s)`` | 3.2 |
1078 | <TestCase.assertNotRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001079 +---------------------------------------+--------------------------------+--------------+
1080 | :meth:`assertDictContainsSubset(a, b) | all the key/value pairs | 3.1 |
1081 | <TestCase.assertDictContainsSubset>` | in `a` exist in `b` | |
1082 +---------------------------------------+--------------------------------+--------------+
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001083 | :meth:`assertCountEqual(a, b) | `a` and `b` have the same | 3.2 |
1084 | <TestCase.assertCountEqual>` | elements in the same number, | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001085 | | regardless of their order | |
1086 +---------------------------------------+--------------------------------+--------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001087
1088
Ezio Melotti4370b302010-11-03 20:39:14 +00001089 .. method:: assertAlmostEqual(first, second, places=7, msg=None, delta=None)
Ezio Melotti4841fd62010-11-05 15:43:40 +00001090 assertNotAlmostEqual(first, second, places=7, msg=None, delta=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001091
Ezio Melotti4841fd62010-11-05 15:43:40 +00001092 Test that *first* and *second* are approximately (or not approximately)
1093 equal by computing the difference, rounding to the given number of
1094 decimal *places* (default 7), and comparing to zero. Note that these
1095 methods round the values to the given number of *decimal places* (i.e.
1096 like the :func:`round` function) and not *significant digits*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001097
Ezio Melotti4370b302010-11-03 20:39:14 +00001098 If *delta* is supplied instead of *places* then the difference
Ezio Melotti4841fd62010-11-05 15:43:40 +00001099 between *first* and *second* must be less (or more) than *delta*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001100
Ezio Melotti4370b302010-11-03 20:39:14 +00001101 Supplying both *delta* and *places* raises a ``TypeError``.
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001102
Ezio Melotti4370b302010-11-03 20:39:14 +00001103 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001104 :meth:`assertAlmostEqual` automatically considers almost equal objects
1105 that compare equal. :meth:`assertNotAlmostEqual` automatically fails
1106 if the objects compare equal. Added the *delta* keyword argument.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001107
Ezio Melotti4370b302010-11-03 20:39:14 +00001108
Ezio Melotti4370b302010-11-03 20:39:14 +00001109 .. method:: assertGreater(first, second, msg=None)
1110 assertGreaterEqual(first, second, msg=None)
1111 assertLess(first, second, msg=None)
1112 assertLessEqual(first, second, msg=None)
1113
1114 Test that *first* is respectively >, >=, < or <= than *second* depending
Ezio Melotti4841fd62010-11-05 15:43:40 +00001115 on the method name. If not, the test will fail::
Ezio Melotti4370b302010-11-03 20:39:14 +00001116
1117 >>> self.assertGreaterEqual(3, 4)
1118 AssertionError: "3" unexpectedly not greater than or equal to "4"
1119
1120 .. versionadded:: 3.1
1121
1122
Ezio Melottied3a7d22010-12-01 02:32:32 +00001123 .. method:: assertRegex(text, regex, msg=None)
1124 assertNotRegex(text, regex, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001125
Ezio Melottied3a7d22010-12-01 02:32:32 +00001126 Test that a *regex* search matches (or does not match) *text*. In case
Ezio Melotti4841fd62010-11-05 15:43:40 +00001127 of failure, the error message will include the pattern and the *text* (or
Ezio Melottied3a7d22010-12-01 02:32:32 +00001128 the pattern and the part of *text* that unexpectedly matched). *regex*
Ezio Melotti4370b302010-11-03 20:39:14 +00001129 may be a regular expression object or a string containing a regular
1130 expression suitable for use by :func:`re.search`.
1131
Georg Brandl419e3de2010-12-01 15:44:25 +00001132 .. versionadded:: 3.1
1133 under the name ``assertRegexpMatches``.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001134 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001135 The method ``assertRegexpMatches()`` has been renamed to
1136 :meth:`.assertRegex`.
1137 .. versionadded:: 3.2
1138 :meth:`.assertNotRegex`.
Ezio Melotti4370b302010-11-03 20:39:14 +00001139
1140
1141 .. method:: assertDictContainsSubset(expected, actual, msg=None)
1142
1143 Tests whether the key/value pairs in dictionary *actual* are a
1144 superset of those in *expected*. If not, an error message listing
1145 the missing keys and mismatched values is generated.
1146
Ezio Melotti4370b302010-11-03 20:39:14 +00001147 .. versionadded:: 3.1
1148
1149
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001150 .. method:: assertCountEqual(expected, actual, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001151
1152 Test that sequence *expected* contains the same elements as *actual*,
1153 regardless of their order. When they don't, an error message listing the
1154 differences between the sequences will be generated.
1155
1156 Duplicate elements are *not* ignored when comparing *actual* and
1157 *expected*. It verifies if each element has the same count in both
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001158 sequences. Equivalent to:
1159 ``assertEqual(Counter(iter(expected)), Counter(iter(actual)))``
1160 but works with sequences of unhashable objects as well.
Ezio Melotti4370b302010-11-03 20:39:14 +00001161
Ezio Melotti4370b302010-11-03 20:39:14 +00001162 .. versionadded:: 3.2
1163
Ezio Melotti4370b302010-11-03 20:39:14 +00001164 .. method:: assertSameElements(actual, expected, msg=None)
1165
1166 Test that sequence *expected* contains the same elements as *actual*,
1167 regardless of their order. When they don't, an error message listing
1168 the differences between the sequences will be generated.
1169
1170 Duplicate elements are ignored when comparing *actual* and *expected*.
1171 It is the equivalent of ``assertEqual(set(expected), set(actual))``
1172 but it works with sequences of unhashable objects as well. Because
1173 duplicates are ignored, this method has been deprecated in favour of
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001174 :meth:`assertCountEqual`.
Ezio Melotti4370b302010-11-03 20:39:14 +00001175
Ezio Melotti4370b302010-11-03 20:39:14 +00001176 .. versionadded:: 3.1
1177 .. deprecated:: 3.2
1178
1179
Ezio Melotti22170ed2010-11-20 09:57:27 +00001180 .. _type-specific-methods:
Ezio Melotti4370b302010-11-03 20:39:14 +00001181
Ezio Melotti22170ed2010-11-20 09:57:27 +00001182 The :meth:`assertEqual` method dispatches the equality check for objects of
1183 the same type to different type-specific methods. These methods are already
1184 implemented for most of the built-in types, but it's also possible to
1185 register new methods using :meth:`addTypeEqualityFunc`:
1186
1187 .. method:: addTypeEqualityFunc(typeobj, function)
1188
1189 Registers a type-specific method called by :meth:`assertEqual` to check
1190 if two objects of exactly the same *typeobj* (not subclasses) compare
1191 equal. *function* must take two positional arguments and a third msg=None
1192 keyword argument just as :meth:`assertEqual` does. It must raise
1193 :data:`self.failureException(msg) <failureException>` when inequality
1194 between the first two parameters is detected -- possibly providing useful
1195 information and explaining the inequalities in details in the error
1196 message.
1197
1198 .. versionadded:: 3.1
1199
1200 The list of type-specific methods automatically used by
1201 :meth:`~TestCase.assertEqual` are summarized in the following table. Note
1202 that it's usually not necessary to invoke these methods directly.
Ezio Melotti4370b302010-11-03 20:39:14 +00001203
1204 +-----------------------------------------+-----------------------------+--------------+
1205 | Method | Used to compare | New in |
1206 +=========================================+=============================+==============+
1207 | :meth:`assertMultiLineEqual(a, b) | strings | 3.1 |
1208 | <TestCase.assertMultiLineEqual>` | | |
1209 +-----------------------------------------+-----------------------------+--------------+
1210 | :meth:`assertSequenceEqual(a, b) | sequences | 3.1 |
1211 | <TestCase.assertSequenceEqual>` | | |
1212 +-----------------------------------------+-----------------------------+--------------+
1213 | :meth:`assertListEqual(a, b) | lists | 3.1 |
1214 | <TestCase.assertListEqual>` | | |
1215 +-----------------------------------------+-----------------------------+--------------+
1216 | :meth:`assertTupleEqual(a, b) | tuples | 3.1 |
1217 | <TestCase.assertTupleEqual>` | | |
1218 +-----------------------------------------+-----------------------------+--------------+
1219 | :meth:`assertSetEqual(a, b) | sets or frozensets | 3.1 |
1220 | <TestCase.assertSetEqual>` | | |
1221 +-----------------------------------------+-----------------------------+--------------+
1222 | :meth:`assertDictEqual(a, b) | dicts | 3.1 |
1223 | <TestCase.assertDictEqual>` | | |
1224 +-----------------------------------------+-----------------------------+--------------+
1225
1226
1227
Ezio Melotti9c02c2f2010-11-03 20:45:31 +00001228 .. method:: assertMultiLineEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001229
1230 Test that the multiline string *first* is equal to the string *second*.
1231 When not equal a diff of the two strings highlighting the differences
1232 will be included in the error message. This method is used by default
1233 when comparing strings with :meth:`assertEqual`.
1234
Ezio Melotti4370b302010-11-03 20:39:14 +00001235 .. versionadded:: 3.1
1236
1237
1238 .. method:: assertSequenceEqual(seq1, seq2, msg=None, seq_type=None)
1239
1240 Tests that two sequences are equal. If a *seq_type* is supplied, both
1241 *seq1* and *seq2* must be instances of *seq_type* or a failure will
1242 be raised. If the sequences are different an error message is
1243 constructed that shows the difference between the two.
1244
Ezio Melotti22170ed2010-11-20 09:57:27 +00001245 This method is not called directly by :meth:`assertEqual`, but
1246 it's used to implement :meth:`assertListEqual` and
Ezio Melotti4370b302010-11-03 20:39:14 +00001247 :meth:`assertTupleEqual`.
1248
1249 .. versionadded:: 3.1
1250
1251
1252 .. method:: assertListEqual(list1, list2, msg=None)
1253 assertTupleEqual(tuple1, tuple2, msg=None)
1254
1255 Tests that two lists or tuples are equal. If not an error message is
1256 constructed that shows only the differences between the two. An error
1257 is also raised if either of the parameters are of the wrong type.
1258 These methods are used by default when comparing lists or tuples with
1259 :meth:`assertEqual`.
1260
Ezio Melotti4370b302010-11-03 20:39:14 +00001261 .. versionadded:: 3.1
1262
1263
1264 .. method:: assertSetEqual(set1, set2, msg=None)
1265
1266 Tests that two sets are equal. If not, an error message is constructed
1267 that lists the differences between the sets. This method is used by
1268 default when comparing sets or frozensets with :meth:`assertEqual`.
1269
1270 Fails if either of *set1* or *set2* does not have a :meth:`set.difference`
1271 method.
1272
Ezio Melotti4370b302010-11-03 20:39:14 +00001273 .. versionadded:: 3.1
1274
1275
1276 .. method:: assertDictEqual(expected, actual, msg=None)
1277
1278 Test that two dictionaries are equal. If not, an error message is
1279 constructed that shows the differences in the dictionaries. This
1280 method will be used by default to compare dictionaries in
1281 calls to :meth:`assertEqual`.
1282
Ezio Melotti4370b302010-11-03 20:39:14 +00001283 .. versionadded:: 3.1
1284
1285
1286
Ezio Melotti22170ed2010-11-20 09:57:27 +00001287 .. _other-methods-and-attrs:
1288
Ezio Melotti4370b302010-11-03 20:39:14 +00001289 Finally the :class:`TestCase` provides the following methods and attributes:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001290
Benjamin Peterson52baa292009-03-24 00:56:30 +00001291
Georg Brandl7f01a132009-09-16 15:58:14 +00001292 .. method:: fail(msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001293
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001294 Signals a test failure unconditionally, with *msg* or ``None`` for
Benjamin Peterson52baa292009-03-24 00:56:30 +00001295 the error message.
1296
1297
1298 .. attribute:: failureException
1299
1300 This class attribute gives the exception raised by the test method. If a
1301 test framework needs to use a specialized exception, possibly to carry
1302 additional information, it must subclass this exception in order to "play
1303 fair" with the framework. The initial value of this attribute is
1304 :exc:`AssertionError`.
1305
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001306
1307 .. attribute:: longMessage
1308
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001309 If set to ``True`` then any explicit failure message you pass in to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001310 :ref:`assert methods <assert-methods>` will be appended to the end of the
1311 normal failure message. The normal messages contain useful information
1312 about the objects involved, for example the message from assertEqual
1313 shows you the repr of the two unequal objects. Setting this attribute
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001314 to ``True`` allows you to have a custom error message in addition to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001315 normal one.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001316
Michael Foord5074df62010-12-03 00:53:09 +00001317 This attribute defaults to ``True``. If set to False then a custom message
1318 passed to an assert method will silence the normal message.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001319
1320 The class setting can be overridden in individual tests by assigning an
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001321 instance attribute to ``True`` or ``False`` before calling the assert methods.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001322
Raymond Hettinger35a88362009-04-09 00:08:24 +00001323 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001324
1325
Michael Foord98b3e762010-06-05 21:59:55 +00001326 .. attribute:: maxDiff
1327
1328 This attribute controls the maximum length of diffs output by assert
1329 methods that report diffs on failure. It defaults to 80*8 characters.
1330 Assert methods affected by this attribute are
1331 :meth:`assertSequenceEqual` (including all the sequence comparison
1332 methods that delegate to it), :meth:`assertDictEqual` and
1333 :meth:`assertMultiLineEqual`.
1334
1335 Setting ``maxDiff`` to None means that there is no maximum length of
1336 diffs.
1337
1338 .. versionadded:: 3.2
1339
1340
Benjamin Peterson52baa292009-03-24 00:56:30 +00001341 Testing frameworks can use the following methods to collect information on
1342 the test:
1343
1344
1345 .. method:: countTestCases()
1346
1347 Return the number of tests represented by this test object. For
1348 :class:`TestCase` instances, this will always be ``1``.
1349
1350
1351 .. method:: defaultTestResult()
1352
1353 Return an instance of the test result class that should be used for this
1354 test case class (if no other result instance is provided to the
1355 :meth:`run` method).
1356
1357 For :class:`TestCase` instances, this will always be an instance of
1358 :class:`TestResult`; subclasses of :class:`TestCase` should override this
1359 as necessary.
1360
1361
1362 .. method:: id()
1363
1364 Return a string identifying the specific test case. This is usually the
1365 full name of the test method, including the module and class name.
1366
1367
1368 .. method:: shortDescription()
1369
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001370 Returns a description of the test, or ``None`` if no description
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001371 has been provided. The default implementation of this method
1372 returns the first line of the test method's docstring, if available,
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001373 or ``None``.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001374
Georg Brandl419e3de2010-12-01 15:44:25 +00001375 .. versionchanged:: 3.1
Michael Foord34c94622010-02-10 15:51:42 +00001376 In 3.1 this was changed to add the test name to the short description
Georg Brandl419e3de2010-12-01 15:44:25 +00001377 even in the presence of a docstring. This caused compatibility issues
Michael Foord34c94622010-02-10 15:51:42 +00001378 with unittest extensions and adding the test name was moved to the
Georg Brandl419e3de2010-12-01 15:44:25 +00001379 :class:`TextTestResult` in Python 3.2.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001380
Georg Brandl116aa622007-08-15 14:28:22 +00001381
Georg Brandl7f01a132009-09-16 15:58:14 +00001382 .. method:: addCleanup(function, *args, **kwargs)
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001383
1384 Add a function to be called after :meth:`tearDown` to cleanup resources
1385 used during the test. Functions will be called in reverse order to the
1386 order they are added (LIFO). They are called with any arguments and
1387 keyword arguments passed into :meth:`addCleanup` when they are
1388 added.
1389
1390 If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
1391 then any cleanup functions added will still be called.
1392
Georg Brandl853947a2010-01-31 18:53:23 +00001393 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001394
1395
1396 .. method:: doCleanups()
1397
Barry Warsaw0c9fd632010-04-12 14:50:57 +00001398 This method is called unconditionally after :meth:`tearDown`, or
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001399 after :meth:`setUp` if :meth:`setUp` raises an exception.
1400
1401 It is responsible for calling all the cleanup functions added by
1402 :meth:`addCleanup`. If you need cleanup functions to be called
1403 *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
1404 yourself.
1405
1406 :meth:`doCleanups` pops methods off the stack of cleanup
1407 functions one at a time, so it can be called at any time.
1408
Georg Brandl853947a2010-01-31 18:53:23 +00001409 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001410
1411
Georg Brandl7f01a132009-09-16 15:58:14 +00001412.. class:: FunctionTestCase(testFunc, setUp=None, tearDown=None, description=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001413
1414 This class implements the portion of the :class:`TestCase` interface which
Benjamin Petersond2397752009-06-27 23:45:02 +00001415 allows the test runner to drive the test, but does not provide the methods
1416 which test code can use to check and report errors. This is used to create
1417 test cases using legacy test code, allowing it to be integrated into a
1418 :mod:`unittest`-based test framework.
Georg Brandl116aa622007-08-15 14:28:22 +00001419
1420
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001421.. _deprecated-aliases:
1422
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001423Deprecated aliases
1424##################
1425
1426For historical reasons, some of the :class:`TestCase` methods had one or more
1427aliases that are now deprecated. The following table lists the correct names
1428along with their deprecated aliases:
1429
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001430 ============================== ====================== ======================
1431 Method Name Deprecated alias Deprecated alias
1432 ============================== ====================== ======================
1433 :meth:`.assertEqual` failUnlessEqual assertEquals
1434 :meth:`.assertNotEqual` failIfEqual assertNotEquals
1435 :meth:`.assertTrue` failUnless assert\_
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001436 :meth:`.assertFalse` failIf
1437 :meth:`.assertRaises` failUnlessRaises
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001438 :meth:`.assertAlmostEqual` failUnlessAlmostEqual assertAlmostEquals
1439 :meth:`.assertNotAlmostEqual` failIfAlmostEqual assertNotAlmostEquals
Ezio Melottied3a7d22010-12-01 02:32:32 +00001440 :meth:`.assertRegex` assertRegexpMatches
1441 :meth:`.assertRaisesRegex` assertRaisesRegexp
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001442 ============================== ====================== ======================
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001443
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001444 .. deprecated-removed:: 3.1 3.3
1445 the fail* aliases listed in the second column.
1446 .. deprecated:: 3.2
1447 the assert* aliases listed in the third column.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001448 .. deprecated:: 3.2
1449 ``assertRegexpMatches`` and ``assertRaisesRegexp`` have been renamed to
1450 :meth:`.assertRegex` and :meth:`.assertRaisesRegex`
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001451
1452
Benjamin Peterson52baa292009-03-24 00:56:30 +00001453.. _testsuite-objects:
1454
Benjamin Peterson52baa292009-03-24 00:56:30 +00001455Grouping tests
1456~~~~~~~~~~~~~~
1457
Georg Brandl7f01a132009-09-16 15:58:14 +00001458.. class:: TestSuite(tests=())
Georg Brandl116aa622007-08-15 14:28:22 +00001459
1460 This class represents an aggregation of individual tests cases and test suites.
1461 The class presents the interface needed by the test runner to allow it to be run
1462 as any other test case. Running a :class:`TestSuite` instance is the same as
1463 iterating over the suite, running each test individually.
1464
1465 If *tests* is given, it must be an iterable of individual test cases or other
1466 test suites that will be used to build the suite initially. Additional methods
1467 are provided to add test cases and suites to the collection later on.
1468
Benjamin Peterson14a3dd72009-05-25 00:51:58 +00001469 :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
1470 they do not actually implement a test. Instead, they are used to aggregate
1471 tests into groups of tests that should be run together. Some additional
1472 methods are available to add tests to :class:`TestSuite` instances:
Benjamin Peterson52baa292009-03-24 00:56:30 +00001473
1474
1475 .. method:: TestSuite.addTest(test)
1476
1477 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1478
1479
1480 .. method:: TestSuite.addTests(tests)
1481
1482 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
1483 instances to this test suite.
1484
Benjamin Petersond2397752009-06-27 23:45:02 +00001485 This is equivalent to iterating over *tests*, calling :meth:`addTest` for
1486 each element.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001487
1488 :class:`TestSuite` shares the following methods with :class:`TestCase`:
1489
1490
1491 .. method:: run(result)
1492
1493 Run the tests associated with this suite, collecting the result into the
1494 test result object passed as *result*. Note that unlike
1495 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
1496 be passed in.
1497
1498
1499 .. method:: debug()
1500
1501 Run the tests associated with this suite without collecting the
1502 result. This allows exceptions raised by the test to be propagated to the
1503 caller and can be used to support running tests under a debugger.
1504
1505
1506 .. method:: countTestCases()
1507
1508 Return the number of tests represented by this test object, including all
1509 individual tests and sub-suites.
1510
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001511
1512 .. method:: __iter__()
1513
1514 Tests grouped by a :class:`TestSuite` are always accessed by iteration.
1515 Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
1516 that this method maybe called several times on a single suite
1517 (for example when counting tests or comparing for equality)
1518 so the tests returned must be the same for repeated iterations.
1519
Georg Brandl853947a2010-01-31 18:53:23 +00001520 .. versionchanged:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001521 In earlier versions the :class:`TestSuite` accessed tests directly rather
1522 than through iteration, so overriding :meth:`__iter__` wasn't sufficient
1523 for providing tests.
1524
Benjamin Peterson52baa292009-03-24 00:56:30 +00001525 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
1526 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1527
1528
Benjamin Peterson52baa292009-03-24 00:56:30 +00001529Loading and running tests
1530~~~~~~~~~~~~~~~~~~~~~~~~~
1531
Georg Brandl116aa622007-08-15 14:28:22 +00001532.. class:: TestLoader()
1533
Benjamin Peterson52baa292009-03-24 00:56:30 +00001534 The :class:`TestLoader` class is used to create test suites from classes and
1535 modules. Normally, there is no need to create an instance of this class; the
1536 :mod:`unittest` module provides an instance that can be shared as
1537 ``unittest.defaultTestLoader``. Using a subclass or instance, however, allows
1538 customization of some configurable properties.
1539
1540 :class:`TestLoader` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001541
Ezio Melotti9c02c2f2010-11-03 20:45:31 +00001542
Benjamin Peterson52baa292009-03-24 00:56:30 +00001543 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl116aa622007-08-15 14:28:22 +00001544
Benjamin Peterson52baa292009-03-24 00:56:30 +00001545 Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
1546 :class:`testCaseClass`.
1547
1548
1549 .. method:: loadTestsFromModule(module)
1550
1551 Return a suite of all tests cases contained in the given module. This
1552 method searches *module* for classes derived from :class:`TestCase` and
1553 creates an instance of the class for each test method defined for the
1554 class.
1555
Georg Brandle720c0a2009-04-27 16:20:50 +00001556 .. note::
Benjamin Peterson52baa292009-03-24 00:56:30 +00001557
1558 While using a hierarchy of :class:`TestCase`\ -derived classes can be
1559 convenient in sharing fixtures and helper functions, defining test
1560 methods on base classes that are not intended to be instantiated
1561 directly does not play well with this method. Doing so, however, can
1562 be useful when the fixtures are different and defined in subclasses.
1563
Benjamin Petersond2397752009-06-27 23:45:02 +00001564 If a module provides a ``load_tests`` function it will be called to
1565 load the tests. This allows modules to customize test loading.
1566 This is the `load_tests protocol`_.
1567
Georg Brandl853947a2010-01-31 18:53:23 +00001568 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001569 Support for ``load_tests`` added.
1570
Benjamin Peterson52baa292009-03-24 00:56:30 +00001571
Georg Brandl7f01a132009-09-16 15:58:14 +00001572 .. method:: loadTestsFromName(name, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001573
1574 Return a suite of all tests cases given a string specifier.
1575
1576 The specifier *name* is a "dotted name" that may resolve either to a
1577 module, a test case class, a test method within a test case class, a
1578 :class:`TestSuite` instance, or a callable object which returns a
1579 :class:`TestCase` or :class:`TestSuite` instance. These checks are
1580 applied in the order listed here; that is, a method on a possible test
1581 case class will be picked up as "a test method within a test case class",
1582 rather than "a callable object".
1583
1584 For example, if you have a module :mod:`SampleTests` containing a
1585 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
1586 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
Benjamin Petersond2397752009-06-27 23:45:02 +00001587 specifier ``'SampleTests.SampleTestCase'`` would cause this method to
1588 return a suite which will run all three test methods. Using the specifier
1589 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
1590 suite which will run only the :meth:`test_two` test method. The specifier
1591 can refer to modules and packages which have not been imported; they will
1592 be imported as a side-effect.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001593
1594 The method optionally resolves *name* relative to the given *module*.
1595
1596
Georg Brandl7f01a132009-09-16 15:58:14 +00001597 .. method:: loadTestsFromNames(names, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001598
1599 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
1600 than a single name. The return value is a test suite which supports all
1601 the tests defined for each name.
1602
1603
1604 .. method:: getTestCaseNames(testCaseClass)
1605
1606 Return a sorted sequence of method names found within *testCaseClass*;
1607 this should be a subclass of :class:`TestCase`.
1608
Benjamin Petersond2397752009-06-27 23:45:02 +00001609
1610 .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)
1611
1612 Find and return all test modules from the specified start directory,
1613 recursing into subdirectories to find them. Only test files that match
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001614 *pattern* will be loaded. (Using shell style pattern matching.) Only
1615 module names that are importable (i.e. are valid Python identifiers) will
1616 be loaded.
Benjamin Petersond2397752009-06-27 23:45:02 +00001617
1618 All test modules must be importable from the top level of the project. If
1619 the start directory is not the top level directory then the top level
1620 directory must be specified separately.
1621
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001622 If importing a module fails, for example due to a syntax error, then this
1623 will be recorded as a single error and discovery will continue.
1624
Benjamin Petersond2397752009-06-27 23:45:02 +00001625 If a test package name (directory with :file:`__init__.py`) matches the
1626 pattern then the package will be checked for a ``load_tests``
1627 function. If this exists then it will be called with *loader*, *tests*,
1628 *pattern*.
1629
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001630 If load_tests exists then discovery does *not* recurse into the package,
Benjamin Petersond2397752009-06-27 23:45:02 +00001631 ``load_tests`` is responsible for loading all tests in the package.
1632
1633 The pattern is deliberately not stored as a loader attribute so that
1634 packages can continue discovery themselves. *top_level_dir* is stored so
1635 ``load_tests`` does not need to pass this argument in to
1636 ``loader.discover()``.
1637
Benjamin Petersonb48af542010-04-11 20:43:16 +00001638 *start_dir* can be a dotted module name as well as a directory.
1639
Georg Brandl853947a2010-01-31 18:53:23 +00001640 .. versionadded:: 3.2
1641
Benjamin Petersond2397752009-06-27 23:45:02 +00001642
Benjamin Peterson52baa292009-03-24 00:56:30 +00001643 The following attributes of a :class:`TestLoader` can be configured either by
1644 subclassing or assignment on an instance:
1645
1646
1647 .. attribute:: testMethodPrefix
1648
1649 String giving the prefix of method names which will be interpreted as test
1650 methods. The default value is ``'test'``.
1651
1652 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
1653 methods.
1654
1655
1656 .. attribute:: sortTestMethodsUsing
1657
1658 Function to be used to compare method names when sorting them in
1659 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
1660
1661
1662 .. attribute:: suiteClass
1663
1664 Callable object that constructs a test suite from a list of tests. No
1665 methods on the resulting object are needed. The default value is the
1666 :class:`TestSuite` class.
1667
1668 This affects all the :meth:`loadTestsFrom\*` methods.
1669
1670
Benjamin Peterson52baa292009-03-24 00:56:30 +00001671.. class:: TestResult
1672
1673 This class is used to compile information about which tests have succeeded
1674 and which have failed.
1675
1676 A :class:`TestResult` object stores the results of a set of tests. The
1677 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
1678 properly recorded; test authors do not need to worry about recording the
1679 outcome of tests.
1680
1681 Testing frameworks built on top of :mod:`unittest` may want access to the
1682 :class:`TestResult` object generated by running a set of tests for reporting
1683 purposes; a :class:`TestResult` instance is returned by the
1684 :meth:`TestRunner.run` method for this purpose.
1685
1686 :class:`TestResult` instances have the following attributes that will be of
1687 interest when inspecting the results of running a set of tests:
1688
1689
1690 .. attribute:: errors
1691
1692 A list containing 2-tuples of :class:`TestCase` instances and strings
1693 holding formatted tracebacks. Each tuple represents a test which raised an
1694 unexpected exception.
1695
Benjamin Peterson52baa292009-03-24 00:56:30 +00001696 .. attribute:: failures
1697
1698 A list containing 2-tuples of :class:`TestCase` instances and strings
1699 holding formatted tracebacks. Each tuple represents a test where a failure
1700 was explicitly signalled using the :meth:`TestCase.fail\*` or
1701 :meth:`TestCase.assert\*` methods.
1702
Benjamin Peterson52baa292009-03-24 00:56:30 +00001703 .. attribute:: skipped
1704
1705 A list containing 2-tuples of :class:`TestCase` instances and strings
1706 holding the reason for skipping the test.
1707
Benjamin Peterson70e32c82009-03-24 01:00:11 +00001708 .. versionadded:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +00001709
1710 .. attribute:: expectedFailures
1711
Georg Brandl6faee4e2010-09-21 14:48:28 +00001712 A list containing 2-tuples of :class:`TestCase` instances and strings
1713 holding formatted tracebacks. Each tuple represents an expected failure
Benjamin Peterson52baa292009-03-24 00:56:30 +00001714 of the test case.
1715
1716 .. attribute:: unexpectedSuccesses
1717
1718 A list containing :class:`TestCase` instances that were marked as expected
1719 failures, but succeeded.
1720
1721 .. attribute:: shouldStop
1722
1723 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1724
1725
1726 .. attribute:: testsRun
1727
1728 The total number of tests run so far.
1729
1730
Georg Brandl12037202010-12-02 22:35:25 +00001731 .. attribute:: buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +00001732
1733 If set to true, ``sys.stdout`` and ``sys.stderr`` will be buffered in between
1734 :meth:`startTest` and :meth:`stopTest` being called. Collected output will
1735 only be echoed onto the real ``sys.stdout`` and ``sys.stderr`` if the test
1736 fails or errors. Any output is also attached to the failure / error message.
1737
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001738 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001739
1740
1741 .. attribute:: failfast
1742
1743 If set to true :meth:`stop` will be called on the first failure or error,
1744 halting the test run.
1745
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001746 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001747
1748
Benjamin Peterson52baa292009-03-24 00:56:30 +00001749 .. method:: wasSuccessful()
1750
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001751 Return ``True`` if all tests run so far have passed, otherwise returns
1752 ``False``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001753
1754
1755 .. method:: stop()
1756
1757 This method can be called to signal that the set of tests being run should
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001758 be aborted by setting the :attr:`shouldStop` attribute to ``True``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001759 :class:`TestRunner` objects should respect this flag and return without
1760 running any additional tests.
1761
1762 For example, this feature is used by the :class:`TextTestRunner` class to
1763 stop the test framework when the user signals an interrupt from the
1764 keyboard. Interactive tools which provide :class:`TestRunner`
1765 implementations can use this in a similar manner.
1766
1767 The following methods of the :class:`TestResult` class are used to maintain
1768 the internal data structures, and may be extended in subclasses to support
1769 additional reporting requirements. This is particularly useful in building
1770 tools which support interactive reporting while tests are being run.
1771
1772
1773 .. method:: startTest(test)
1774
1775 Called when the test case *test* is about to be run.
1776
Benjamin Peterson52baa292009-03-24 00:56:30 +00001777 .. method:: stopTest(test)
1778
1779 Called after the test case *test* has been executed, regardless of the
1780 outcome.
1781
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001782 .. method:: startTestRun(test)
1783
1784 Called once before any tests are executed.
1785
Georg Brandl853947a2010-01-31 18:53:23 +00001786 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001787
1788
1789 .. method:: stopTestRun(test)
1790
Ezio Melotti176d6c42010-01-27 20:58:07 +00001791 Called once after all tests are executed.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001792
Georg Brandl853947a2010-01-31 18:53:23 +00001793 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001794
1795
Benjamin Peterson52baa292009-03-24 00:56:30 +00001796 .. method:: addError(test, err)
1797
1798 Called when the test case *test* raises an unexpected exception *err* is a
1799 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1800 traceback)``.
1801
1802 The default implementation appends a tuple ``(test, formatted_err)`` to
1803 the instance's :attr:`errors` attribute, where *formatted_err* is a
1804 formatted traceback derived from *err*.
1805
1806
1807 .. method:: addFailure(test, err)
1808
Benjamin Petersond2397752009-06-27 23:45:02 +00001809 Called when the test case *test* signals a failure. *err* is a tuple of
1810 the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001811
1812 The default implementation appends a tuple ``(test, formatted_err)`` to
1813 the instance's :attr:`failures` attribute, where *formatted_err* is a
1814 formatted traceback derived from *err*.
1815
1816
1817 .. method:: addSuccess(test)
1818
1819 Called when the test case *test* succeeds.
1820
1821 The default implementation does nothing.
1822
1823
1824 .. method:: addSkip(test, reason)
1825
1826 Called when the test case *test* is skipped. *reason* is the reason the
1827 test gave for skipping.
1828
1829 The default implementation appends a tuple ``(test, reason)`` to the
1830 instance's :attr:`skipped` attribute.
1831
1832
1833 .. method:: addExpectedFailure(test, err)
1834
1835 Called when the test case *test* fails, but was marked with the
1836 :func:`expectedFailure` decorator.
1837
1838 The default implementation appends a tuple ``(test, formatted_err)`` to
1839 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1840 is a formatted traceback derived from *err*.
1841
1842
1843 .. method:: addUnexpectedSuccess(test)
1844
1845 Called when the test case *test* was marked with the
1846 :func:`expectedFailure` decorator, but succeeded.
1847
1848 The default implementation appends the test to the instance's
1849 :attr:`unexpectedSuccesses` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +00001850
Georg Brandl67b21b72010-08-17 15:07:14 +00001851
Michael Foord34c94622010-02-10 15:51:42 +00001852.. class:: TextTestResult(stream, descriptions, verbosity)
1853
Georg Brandl67b21b72010-08-17 15:07:14 +00001854 A concrete implementation of :class:`TestResult` used by the
1855 :class:`TextTestRunner`.
Michael Foord34c94622010-02-10 15:51:42 +00001856
Georg Brandl67b21b72010-08-17 15:07:14 +00001857 .. versionadded:: 3.2
1858 This class was previously named ``_TextTestResult``. The old name still
1859 exists as an alias but is deprecated.
1860
Georg Brandl116aa622007-08-15 14:28:22 +00001861
1862.. data:: defaultTestLoader
1863
1864 Instance of the :class:`TestLoader` class intended to be shared. If no
1865 customization of the :class:`TestLoader` is needed, this instance can be used
1866 instead of repeatedly creating new instances.
1867
1868
Ezio Melotti60901872010-12-01 00:56:10 +00001869.. class:: TextTestRunner(stream=sys.stderr, descriptions=True, verbosity=1, runnerclass=None, warnings=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001870
1871 A basic test runner implementation which prints results on standard error. It
1872 has a few configurable parameters, but is essentially very simple. Graphical
1873 applications which run test suites should provide alternate implementations.
1874
Ezio Melotti60901872010-12-01 00:56:10 +00001875 By default this runner shows :exc:`DeprecationWarning`,
1876 :exc:`PendingDeprecationWarning`, and :exc:`ImportWarning` even if they are
1877 :ref:`ignored by default <warning-ignored>`. Deprecation warnings caused by
1878 :ref:`deprecated unittest methods <deprecated-aliases>` are also
1879 special-cased and, when the warning filters are ``'default'`` or ``'always'``,
1880 they will appear only once per-module, in order to avoid too many warning
Georg Brandl46402372010-12-04 19:06:18 +00001881 messages. This behavior can be overridden using the :option:`-Wd` or
Ezio Melotti60901872010-12-01 00:56:10 +00001882 :option:`-Wa` options and leaving *warnings* to ``None``.
1883
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001884 .. method:: _makeResult()
Georg Brandl116aa622007-08-15 14:28:22 +00001885
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001886 This method returns the instance of ``TestResult`` used by :meth:`run`.
1887 It is not intended to be called directly, but can be overridden in
1888 subclasses to provide a custom ``TestResult``.
1889
Michael Foord34c94622010-02-10 15:51:42 +00001890 ``_makeResult()`` instantiates the class or callable passed in the
1891 ``TextTestRunner`` constructor as the ``resultclass`` argument. It
Benjamin Petersonb48af542010-04-11 20:43:16 +00001892 defaults to :class:`TextTestResult` if no ``resultclass`` is provided.
Michael Foord34c94622010-02-10 15:51:42 +00001893 The result class is instantiated with the following arguments::
1894
1895 stream, descriptions, verbosity
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001896
Georg Brandl419e3de2010-12-01 15:44:25 +00001897 .. versionchanged:: 3.2
1898 Added the ``warnings`` argument.
Ezio Melotti60901872010-12-01 00:56:10 +00001899
Georg Brandl419e3de2010-12-01 15:44:25 +00001900.. function:: main(module='__main__', defaultTest=None, argv=None, testRunner=None, \
1901 testLoader=unittest.loader.defaultTestLoader, exit=True, verbosity=1, \
1902 failfast=None, catchbreak=None, buffer=None, warnings=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001903
1904 A command-line program that runs a set of tests; this is primarily for making
1905 test modules conveniently executable. The simplest use for this function is to
1906 include the following line at the end of a test script::
1907
1908 if __name__ == '__main__':
1909 unittest.main()
1910
Benjamin Petersond2397752009-06-27 23:45:02 +00001911 You can run tests with more detailed information by passing in the verbosity
1912 argument::
1913
1914 if __name__ == '__main__':
1915 unittest.main(verbosity=2)
1916
Georg Brandl116aa622007-08-15 14:28:22 +00001917 The *testRunner* argument can either be a test runner class or an already
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001918 created instance of it. By default ``main`` calls :func:`sys.exit` with
1919 an exit code indicating success or failure of the tests run.
1920
1921 ``main`` supports being used from the interactive interpreter by passing in the
1922 argument ``exit=False``. This displays the result on standard output without
1923 calling :func:`sys.exit`::
1924
1925 >>> from unittest import main
1926 >>> main(module='test_module', exit=False)
1927
Benjamin Petersonb48af542010-04-11 20:43:16 +00001928 The ``failfast``, ``catchbreak`` and ``buffer`` parameters have the same
Éric Araujo76338ec2010-11-26 23:46:18 +00001929 effect as the same-name `command-line options`_.
Benjamin Petersonb48af542010-04-11 20:43:16 +00001930
Ezio Melotti60901872010-12-01 00:56:10 +00001931 The *warning* argument specifies the :ref:`warning filter <warning-filter>`
1932 that should be used while running the tests. If it's not specified, it will
1933 remain ``None`` if a :option:`-W` option is passed to :program:`python`,
1934 otherwise it will be set to ``'default'``.
1935
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001936 Calling ``main`` actually returns an instance of the ``TestProgram`` class.
1937 This stores the result of the tests run as the ``result`` attribute.
1938
Georg Brandl853947a2010-01-31 18:53:23 +00001939 .. versionchanged:: 3.2
Ezio Melotti60901872010-12-01 00:56:10 +00001940 The ``exit``, ``verbosity``, ``failfast``, ``catchbreak``, ``buffer``,
1941 and ``warnings`` parameters were added.
Benjamin Petersond2397752009-06-27 23:45:02 +00001942
1943
1944load_tests Protocol
1945###################
1946
Georg Brandl853947a2010-01-31 18:53:23 +00001947.. versionadded:: 3.2
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001948
Benjamin Petersond2397752009-06-27 23:45:02 +00001949Modules or packages can customize how tests are loaded from them during normal
1950test runs or test discovery by implementing a function called ``load_tests``.
1951
1952If a test module defines ``load_tests`` it will be called by
1953:meth:`TestLoader.loadTestsFromModule` with the following arguments::
1954
1955 load_tests(loader, standard_tests, None)
1956
1957It should return a :class:`TestSuite`.
1958
1959*loader* is the instance of :class:`TestLoader` doing the loading.
1960*standard_tests* are the tests that would be loaded by default from the
1961module. It is common for test modules to only want to add or remove tests
1962from the standard set of tests.
1963The third argument is used when loading packages as part of test discovery.
1964
1965A typical ``load_tests`` function that loads tests from a specific set of
1966:class:`TestCase` classes may look like::
1967
1968 test_cases = (TestCase1, TestCase2, TestCase3)
1969
1970 def load_tests(loader, tests, pattern):
1971 suite = TestSuite()
1972 for test_class in test_cases:
1973 tests = loader.loadTestsFromTestCase(test_class)
1974 suite.addTests(tests)
1975 return suite
1976
1977If discovery is started, either from the command line or by calling
1978:meth:`TestLoader.discover`, with a pattern that matches a package
1979name then the package :file:`__init__.py` will be checked for ``load_tests``.
1980
1981.. note::
1982
Ezio Melotti0639d5a2009-12-19 23:26:38 +00001983 The default pattern is 'test*.py'. This matches all Python files
Benjamin Petersond2397752009-06-27 23:45:02 +00001984 that start with 'test' but *won't* match any test directories.
1985
1986 A pattern like 'test*' will match test packages as well as
1987 modules.
1988
1989If the package :file:`__init__.py` defines ``load_tests`` then it will be
1990called and discovery not continued into the package. ``load_tests``
1991is called with the following arguments::
1992
1993 load_tests(loader, standard_tests, pattern)
1994
1995This should return a :class:`TestSuite` representing all the tests
1996from the package. (``standard_tests`` will only contain tests
1997collected from :file:`__init__.py`.)
1998
1999Because the pattern is passed into ``load_tests`` the package is free to
2000continue (and potentially modify) test discovery. A 'do nothing'
2001``load_tests`` function for a test package would look like::
2002
2003 def load_tests(loader, standard_tests, pattern):
2004 # top level directory cached on loader instance
2005 this_dir = os.path.dirname(__file__)
2006 package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
2007 standard_tests.addTests(package_tests)
2008 return standard_tests
Benjamin Petersonb48af542010-04-11 20:43:16 +00002009
2010
2011Class and Module Fixtures
2012-------------------------
2013
2014Class and module level fixtures are implemented in :class:`TestSuite`. When
2015the test suite encounters a test from a new class then :meth:`tearDownClass`
2016from the previous class (if there is one) is called, followed by
2017:meth:`setUpClass` from the new class.
2018
2019Similarly if a test is from a different module from the previous test then
2020``tearDownModule`` from the previous module is run, followed by
2021``setUpModule`` from the new module.
2022
2023After all the tests have run the final ``tearDownClass`` and
2024``tearDownModule`` are run.
2025
2026Note that shared fixtures do not play well with [potential] features like test
2027parallelization and they break test isolation. They should be used with care.
2028
2029The default ordering of tests created by the unittest test loaders is to group
2030all tests from the same modules and classes together. This will lead to
2031``setUpClass`` / ``setUpModule`` (etc) being called exactly once per class and
2032module. If you randomize the order, so that tests from different modules and
2033classes are adjacent to each other, then these shared fixture functions may be
2034called multiple times in a single test run.
2035
2036Shared fixtures are not intended to work with suites with non-standard
2037ordering. A ``BaseTestSuite`` still exists for frameworks that don't want to
2038support shared fixtures.
2039
2040If there are any exceptions raised during one of the shared fixture functions
2041the test is reported as an error. Because there is no corresponding test
2042instance an ``_ErrorHolder`` object (that has the same interface as a
2043:class:`TestCase`) is created to represent the error. If you are just using
2044the standard unittest test runner then this detail doesn't matter, but if you
2045are a framework author it may be relevant.
2046
2047
2048setUpClass and tearDownClass
2049~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2050
2051These must be implemented as class methods::
2052
2053 import unittest
2054
2055 class Test(unittest.TestCase):
2056 @classmethod
2057 def setUpClass(cls):
2058 cls._connection = createExpensiveConnectionObject()
2059
2060 @classmethod
2061 def tearDownClass(cls):
2062 cls._connection.destroy()
2063
2064If you want the ``setUpClass`` and ``tearDownClass`` on base classes called
2065then you must call up to them yourself. The implementations in
2066:class:`TestCase` are empty.
2067
2068If an exception is raised during a ``setUpClass`` then the tests in the class
2069are not run and the ``tearDownClass`` is not run. Skipped classes will not
Michael Foord98b3e762010-06-05 21:59:55 +00002070have ``setUpClass`` or ``tearDownClass`` run. If the exception is a
2071``SkipTest`` exception then the class will be reported as having been skipped
2072instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002073
2074
2075setUpModule and tearDownModule
2076~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2077
2078These should be implemented as functions::
2079
2080 def setUpModule():
2081 createConnection()
2082
2083 def tearDownModule():
2084 closeConnection()
2085
2086If an exception is raised in a ``setUpModule`` then none of the tests in the
Michael Foord98b3e762010-06-05 21:59:55 +00002087module will be run and the ``tearDownModule`` will not be run. If the exception is a
2088``SkipTest`` exception then the module will be reported as having been skipped
2089instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002090
2091
2092Signal Handling
2093---------------
2094
Georg Brandl419e3de2010-12-01 15:44:25 +00002095.. versionadded:: 3.2
2096
Éric Araujo8acb67c2010-11-26 23:31:07 +00002097The :option:`-c/--catch <unittest -c>` command-line option to unittest,
Éric Araujo76338ec2010-11-26 23:46:18 +00002098along with the ``catchbreak`` parameter to :func:`unittest.main()`, provide
2099more friendly handling of control-C during a test run. With catch break
2100behavior enabled control-C will allow the currently running test to complete,
2101and the test run will then end and report all the results so far. A second
2102control-c will raise a :exc:`KeyboardInterrupt` in the usual way.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002103
Michael Foordde4ceab2010-04-25 19:53:49 +00002104The control-c handling signal handler attempts to remain compatible with code or
2105tests that install their own :const:`signal.SIGINT` handler. If the ``unittest``
2106handler is called but *isn't* the installed :const:`signal.SIGINT` handler,
2107i.e. it has been replaced by the system under test and delegated to, then it
2108calls the default handler. This will normally be the expected behavior by code
2109that replaces an installed handler and delegates to it. For individual tests
2110that need ``unittest`` control-c handling disabled the :func:`removeHandler`
2111decorator can be used.
2112
2113There are a few utility functions for framework authors to enable control-c
2114handling functionality within test frameworks.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002115
2116.. function:: installHandler()
2117
2118 Install the control-c handler. When a :const:`signal.SIGINT` is received
2119 (usually in response to the user pressing control-c) all registered results
2120 have :meth:`~TestResult.stop` called.
2121
Michael Foord469b1f02010-04-26 23:41:26 +00002122
Benjamin Petersonb48af542010-04-11 20:43:16 +00002123.. function:: registerResult(result)
2124
2125 Register a :class:`TestResult` object for control-c handling. Registering a
2126 result stores a weak reference to it, so it doesn't prevent the result from
2127 being garbage collected.
2128
Michael Foordde4ceab2010-04-25 19:53:49 +00002129 Registering a :class:`TestResult` object has no side-effects if control-c
2130 handling is not enabled, so test frameworks can unconditionally register
2131 all results they create independently of whether or not handling is enabled.
2132
Michael Foord469b1f02010-04-26 23:41:26 +00002133
Benjamin Petersonb48af542010-04-11 20:43:16 +00002134.. function:: removeResult(result)
2135
2136 Remove a registered result. Once a result has been removed then
2137 :meth:`~TestResult.stop` will no longer be called on that result object in
2138 response to a control-c.
2139
Michael Foord469b1f02010-04-26 23:41:26 +00002140
Michael Foordde4ceab2010-04-25 19:53:49 +00002141.. function:: removeHandler(function=None)
2142
2143 When called without arguments this function removes the control-c handler
2144 if it has been installed. This function can also be used as a test decorator
2145 to temporarily remove the handler whilst the test is being executed::
2146
2147 @unittest.removeHandler
2148 def test_signal_handling(self):
2149 ...