blob: 63ad705220edf09d9b09de1bf5e0cb5a5d0b54e5 [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 Araujo76338ec2010-11-26 23:46:18 +0000260 The command-line options :option:`-b`, :option:`-c` and :option:`-f`
261 were added.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000262
263The command line can also be used for test discovery, for running all of the
264tests in a project or just a subset.
265
266
267.. _unittest-test-discovery:
268
269Test Discovery
270--------------
271
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000272.. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000273
274Unittest supports simple test discovery. For a project's tests to be
275compatible with test discovery they must all be importable from the top level
276directory of the project (in other words, they must all be in Python packages).
277
278Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
Éric Araujo713d3032010-11-18 16:38:46 +0000279used from the command line. The basic command-line usage is::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000280
281 cd project_directory
282 python -m unittest discover
283
Michael Foord086f3082010-11-21 21:28:01 +0000284.. note::
285
286 As a shortcut, ``python -m unittest`` is the equivalent of
287 ``python -m unittest discover``. If you want to pass arguments to test
288 discovery the `discover` sub-command must be used explicitly.
289
Benjamin Petersonb48af542010-04-11 20:43:16 +0000290The ``discover`` sub-command has the following options:
291
Éric Araujo713d3032010-11-18 16:38:46 +0000292.. program:: unittest discover
293
294.. cmdoption:: -v, --verbose
295
296 Verbose output
297
298.. cmdoption:: -s directory
299
300 Directory to start discovery ('.' default)
301
302.. cmdoption:: -p pattern
303
304 Pattern to match test files ('test*.py' default)
305
306.. cmdoption:: -t directory
307
308 Top level directory of project (defaults to start directory)
Benjamin Petersonb48af542010-04-11 20:43:16 +0000309
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000310The :option:`-s`, :option:`-p`, and :option:`-t` options can be passed in
311as positional arguments in that order. The following two command lines
312are equivalent::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000313
314 python -m unittest discover -s project_directory -p '*_test.py'
315 python -m unittest discover project_directory '*_test.py'
316
Michael Foord16f3e902010-05-08 15:13:42 +0000317As well as being a path it is possible to pass a package name, for example
318``myproject.subpackage.test``, as the start directory. The package name you
319supply will then be imported and its location on the filesystem will be used
320as the start directory.
321
322.. caution::
323
Senthil Kumaran916bd382010-10-15 12:55:19 +0000324 Test discovery loads tests by importing them. Once test discovery has found
325 all the test files from the start directory you specify it turns the paths
326 into package names to import. For example :file:`foo/bar/baz.py` will be
Michael Foord16f3e902010-05-08 15:13:42 +0000327 imported as ``foo.bar.baz``.
328
329 If you have a package installed globally and attempt test discovery on
330 a different copy of the package then the import *could* happen from the
331 wrong place. If this happens test discovery will warn you and exit.
332
333 If you supply the start directory as a package name rather than a
334 path to a directory then discover assumes that whichever location it
335 imports from is the location you intended, so you will not get the
336 warning.
337
Benjamin Petersonb48af542010-04-11 20:43:16 +0000338Test modules and packages can customize test loading and discovery by through
339the `load_tests protocol`_.
340
341
Georg Brandl116aa622007-08-15 14:28:22 +0000342.. _organizing-tests:
343
344Organizing test code
345--------------------
346
347The basic building blocks of unit testing are :dfn:`test cases` --- single
348scenarios that must be set up and checked for correctness. In :mod:`unittest`,
349test cases are represented by instances of :mod:`unittest`'s :class:`TestCase`
350class. To make your own test cases you must write subclasses of
351:class:`TestCase`, or use :class:`FunctionTestCase`.
352
353An instance of a :class:`TestCase`\ -derived class is an object that can
354completely run a single test method, together with optional set-up and tidy-up
355code.
356
357The testing code of a :class:`TestCase` instance should be entirely self
358contained, such that it can be run either in isolation or in arbitrary
359combination with any number of other test cases.
360
Benjamin Peterson52baa292009-03-24 00:56:30 +0000361The simplest :class:`TestCase` subclass will simply override the
362:meth:`~TestCase.runTest` method in order to perform specific testing code::
Georg Brandl116aa622007-08-15 14:28:22 +0000363
364 import unittest
365
366 class DefaultWidgetSizeTestCase(unittest.TestCase):
367 def runTest(self):
368 widget = Widget('The widget')
369 self.assertEqual(widget.size(), (50, 50), 'incorrect default size')
370
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000371Note that in order to test something, we use the one of the :meth:`assert\*`
Benjamin Petersond2397752009-06-27 23:45:02 +0000372methods provided by the :class:`TestCase` base class. If the test fails, an
373exception will be raised, and :mod:`unittest` will identify the test case as a
374:dfn:`failure`. Any other exceptions will be treated as :dfn:`errors`. This
375helps you identify where the problem is: :dfn:`failures` are caused by incorrect
376results - a 5 where you expected a 6. :dfn:`Errors` are caused by incorrect
377code - e.g., a :exc:`TypeError` caused by an incorrect function call.
Georg Brandl116aa622007-08-15 14:28:22 +0000378
379The way to run a test case will be described later. For now, note that to
380construct an instance of such a test case, we call its constructor without
381arguments::
382
383 testCase = DefaultWidgetSizeTestCase()
384
385Now, such test cases can be numerous, and their set-up can be repetitive. In
386the above case, constructing a :class:`Widget` in each of 100 Widget test case
387subclasses would mean unsightly duplication.
388
389Luckily, we can factor out such set-up code by implementing a method called
Benjamin Peterson52baa292009-03-24 00:56:30 +0000390:meth:`~TestCase.setUp`, which the testing framework will automatically call for
391us when we run the test::
Georg Brandl116aa622007-08-15 14:28:22 +0000392
393 import unittest
394
395 class SimpleWidgetTestCase(unittest.TestCase):
396 def setUp(self):
397 self.widget = Widget('The widget')
398
399 class DefaultWidgetSizeTestCase(SimpleWidgetTestCase):
400 def runTest(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000401 self.assertEqual(self.widget.size(), (50,50),
402 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000403
404 class WidgetResizeTestCase(SimpleWidgetTestCase):
405 def runTest(self):
406 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000407 self.assertEqual(self.widget.size(), (100,150),
408 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000409
Benjamin Peterson52baa292009-03-24 00:56:30 +0000410If the :meth:`~TestCase.setUp` method raises an exception while the test is
411running, the framework will consider the test to have suffered an error, and the
412:meth:`~TestCase.runTest` method will not be executed.
Georg Brandl116aa622007-08-15 14:28:22 +0000413
Benjamin Peterson52baa292009-03-24 00:56:30 +0000414Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
415after the :meth:`~TestCase.runTest` method has been run::
Georg Brandl116aa622007-08-15 14:28:22 +0000416
417 import unittest
418
419 class SimpleWidgetTestCase(unittest.TestCase):
420 def setUp(self):
421 self.widget = Widget('The widget')
422
423 def tearDown(self):
424 self.widget.dispose()
425 self.widget = None
426
Benjamin Peterson52baa292009-03-24 00:56:30 +0000427If :meth:`~TestCase.setUp` succeeded, the :meth:`~TestCase.tearDown` method will
428be run whether :meth:`~TestCase.runTest` succeeded or not.
Georg Brandl116aa622007-08-15 14:28:22 +0000429
430Such a working environment for the testing code is called a :dfn:`fixture`.
431
432Often, many small test cases will use the same fixture. In this case, we would
433end up subclassing :class:`SimpleWidgetTestCase` into many small one-method
434classes such as :class:`DefaultWidgetSizeTestCase`. This is time-consuming and
Georg Brandl116aa622007-08-15 14:28:22 +0000435discouraging, so in the same vein as JUnit, :mod:`unittest` provides a simpler
436mechanism::
437
438 import unittest
439
440 class WidgetTestCase(unittest.TestCase):
441 def setUp(self):
442 self.widget = Widget('The widget')
443
444 def tearDown(self):
445 self.widget.dispose()
446 self.widget = None
447
Ezio Melottid59e44a2010-02-28 03:46:13 +0000448 def test_default_size(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000449 self.assertEqual(self.widget.size(), (50,50),
450 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000451
Ezio Melottid59e44a2010-02-28 03:46:13 +0000452 def test_resize(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000453 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000454 self.assertEqual(self.widget.size(), (100,150),
455 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000456
Benjamin Peterson52baa292009-03-24 00:56:30 +0000457Here we have not provided a :meth:`~TestCase.runTest` method, but have instead
458provided two different test methods. Class instances will now each run one of
Ezio Melottid59e44a2010-02-28 03:46:13 +0000459the :meth:`test_\*` methods, with ``self.widget`` created and destroyed
Benjamin Peterson52baa292009-03-24 00:56:30 +0000460separately for each instance. When creating an instance we must specify the
461test method it is to run. We do this by passing the method name in the
462constructor::
Georg Brandl116aa622007-08-15 14:28:22 +0000463
Ezio Melottid59e44a2010-02-28 03:46:13 +0000464 defaultSizeTestCase = WidgetTestCase('test_default_size')
465 resizeTestCase = WidgetTestCase('test_resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000466
467Test case instances are grouped together according to the features they test.
468:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
469represented by :mod:`unittest`'s :class:`TestSuite` class::
470
471 widgetTestSuite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000472 widgetTestSuite.addTest(WidgetTestCase('test_default_size'))
473 widgetTestSuite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000474
475For the ease of running tests, as we will see later, it is a good idea to
476provide in each test module a callable object that returns a pre-built test
477suite::
478
479 def suite():
480 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000481 suite.addTest(WidgetTestCase('test_default_size'))
482 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000483 return suite
484
485or even::
486
487 def suite():
Ezio Melottid59e44a2010-02-28 03:46:13 +0000488 tests = ['test_default_size', 'test_resize']
Georg Brandl116aa622007-08-15 14:28:22 +0000489
490 return unittest.TestSuite(map(WidgetTestCase, tests))
491
492Since it is a common pattern to create a :class:`TestCase` subclass with many
493similarly named test functions, :mod:`unittest` provides a :class:`TestLoader`
494class that can be used to automate the process of creating a test suite and
495populating it with individual tests. For example, ::
496
497 suite = unittest.TestLoader().loadTestsFromTestCase(WidgetTestCase)
498
Ezio Melottid59e44a2010-02-28 03:46:13 +0000499will create a test suite that will run ``WidgetTestCase.test_default_size()`` and
500``WidgetTestCase.test_resize``. :class:`TestLoader` uses the ``'test'`` method
Georg Brandl116aa622007-08-15 14:28:22 +0000501name prefix to identify test methods automatically.
502
Mark Dickinsonc48d8342009-02-01 14:18:10 +0000503Note that the order in which the various test cases will be run is
504determined by sorting the test function names with respect to the
505built-in ordering for strings.
Georg Brandl116aa622007-08-15 14:28:22 +0000506
507Often it is desirable to group suites of test cases together, so as to run tests
508for the whole system at once. This is easy, since :class:`TestSuite` instances
509can be added to a :class:`TestSuite` just as :class:`TestCase` instances can be
510added to a :class:`TestSuite`::
511
512 suite1 = module1.TheTestSuite()
513 suite2 = module2.TheTestSuite()
514 alltests = unittest.TestSuite([suite1, suite2])
515
516You can place the definitions of test cases and test suites in the same modules
517as the code they are to test (such as :file:`widget.py`), but there are several
518advantages to placing the test code in a separate module, such as
519:file:`test_widget.py`:
520
521* The test module can be run standalone from the command line.
522
523* The test code can more easily be separated from shipped code.
524
525* There is less temptation to change test code to fit the code it tests without
526 a good reason.
527
528* Test code should be modified much less frequently than the code it tests.
529
530* Tested code can be refactored more easily.
531
532* Tests for modules written in C must be in separate modules anyway, so why not
533 be consistent?
534
535* If the testing strategy changes, there is no need to change the source code.
536
537
538.. _legacy-unit-tests:
539
540Re-using old test code
541----------------------
542
543Some users will find that they have existing test code that they would like to
544run from :mod:`unittest`, without converting every old test function to a
545:class:`TestCase` subclass.
546
547For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
548This subclass of :class:`TestCase` can be used to wrap an existing test
549function. Set-up and tear-down functions can also be provided.
550
551Given the following test function::
552
553 def testSomething():
554 something = makeSomething()
555 assert something.name is not None
556 # ...
557
558one can create an equivalent test case instance as follows::
559
560 testcase = unittest.FunctionTestCase(testSomething)
561
562If there are additional set-up and tear-down methods that should be called as
563part of the test case's operation, they can also be provided like so::
564
565 testcase = unittest.FunctionTestCase(testSomething,
566 setUp=makeSomethingDB,
567 tearDown=deleteSomethingDB)
568
569To make migrating existing test suites easier, :mod:`unittest` supports tests
570raising :exc:`AssertionError` to indicate test failure. However, it is
571recommended that you use the explicit :meth:`TestCase.fail\*` and
572:meth:`TestCase.assert\*` methods instead, as future versions of :mod:`unittest`
573may treat :exc:`AssertionError` differently.
574
575.. note::
576
Benjamin Petersond2397752009-06-27 23:45:02 +0000577 Even though :class:`FunctionTestCase` can be used to quickly convert an
578 existing test base over to a :mod:`unittest`\ -based system, this approach is
579 not recommended. Taking the time to set up proper :class:`TestCase`
580 subclasses will make future test refactorings infinitely easier.
Georg Brandl116aa622007-08-15 14:28:22 +0000581
Benjamin Peterson52baa292009-03-24 00:56:30 +0000582In some cases, the existing tests may have been written using the :mod:`doctest`
583module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
584automatically build :class:`unittest.TestSuite` instances from the existing
585:mod:`doctest`\ -based tests.
586
Georg Brandl116aa622007-08-15 14:28:22 +0000587
Benjamin Peterson5254c042009-03-23 22:25:03 +0000588.. _unittest-skipping:
589
590Skipping tests and expected failures
591------------------------------------
592
Michael Foordf5c851a2010-02-05 21:48:03 +0000593.. versionadded:: 3.1
594
Benjamin Peterson5254c042009-03-23 22:25:03 +0000595Unittest supports skipping individual test methods and even whole classes of
596tests. In addition, it supports marking a test as a "expected failure," a test
597that is broken and will fail, but shouldn't be counted as a failure on a
598:class:`TestResult`.
599
600Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
601or one of its conditional variants.
602
603Basic skipping looks like this: ::
604
605 class MyTestCase(unittest.TestCase):
606
607 @unittest.skip("demonstrating skipping")
608 def test_nothing(self):
609 self.fail("shouldn't happen")
610
Benjamin Petersond2397752009-06-27 23:45:02 +0000611 @unittest.skipIf(mylib.__version__ < (1, 3),
612 "not supported in this library version")
Benjamin Petersonded31c42009-03-30 15:04:16 +0000613 def test_format(self):
614 # Tests that work for only a certain version of the library.
615 pass
616
617 @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
618 def test_windows_support(self):
619 # windows specific testing code
620 pass
621
Benjamin Peterson5254c042009-03-23 22:25:03 +0000622This is the output of running the example above in verbose mode: ::
623
Benjamin Petersonded31c42009-03-30 15:04:16 +0000624 test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000625 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
Benjamin Petersonded31c42009-03-30 15:04:16 +0000626 test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000627
628 ----------------------------------------------------------------------
Benjamin Petersonded31c42009-03-30 15:04:16 +0000629 Ran 3 tests in 0.005s
630
631 OK (skipped=3)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000632
633Classes can be skipped just like methods: ::
634
635 @skip("showing class skipping")
636 class MySkippedTestCase(unittest.TestCase):
637 def test_not_run(self):
638 pass
639
Benjamin Peterson52baa292009-03-24 00:56:30 +0000640:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
641that needs to be set up is not available.
642
Benjamin Peterson5254c042009-03-23 22:25:03 +0000643Expected failures use the :func:`expectedFailure` decorator. ::
644
645 class ExpectedFailureTestCase(unittest.TestCase):
646 @unittest.expectedFailure
647 def test_fail(self):
648 self.assertEqual(1, 0, "broken")
649
650It's easy to roll your own skipping decorators by making a decorator that calls
651:func:`skip` on the test when it wants it to be skipped. This decorator skips
652the test unless the passed object has a certain attribute: ::
653
654 def skipUnlessHasattr(obj, attr):
655 if hasattr(obj, attr):
656 return lambda func: func
657 return unittest.skip("{0!r} doesn't have {1!r}".format(obj, attr))
658
659The following decorators implement test skipping and expected failures:
660
Georg Brandl8a1caa22010-07-29 16:01:11 +0000661.. decorator:: skip(reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000662
663 Unconditionally skip the decorated test. *reason* should describe why the
664 test is being skipped.
665
Georg Brandl8a1caa22010-07-29 16:01:11 +0000666.. decorator:: skipIf(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000667
668 Skip the decorated test if *condition* is true.
669
Georg Brandl8a1caa22010-07-29 16:01:11 +0000670.. decorator:: skipUnless(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000671
Georg Brandl6faee4e2010-09-21 14:48:28 +0000672 Skip the decorated test unless *condition* is true.
Benjamin Peterson5254c042009-03-23 22:25:03 +0000673
Georg Brandl8a1caa22010-07-29 16:01:11 +0000674.. decorator:: expectedFailure
Benjamin Peterson5254c042009-03-23 22:25:03 +0000675
676 Mark the test as an expected failure. If the test fails when run, the test
677 is not counted as a failure.
678
Benjamin Petersonb48af542010-04-11 20:43:16 +0000679Skipped tests will not have :meth:`setUp` or :meth:`tearDown` run around them.
680Skipped classes will not have :meth:`setUpClass` or :meth:`tearDownClass` run.
681
Benjamin Peterson5254c042009-03-23 22:25:03 +0000682
Georg Brandl116aa622007-08-15 14:28:22 +0000683.. _unittest-contents:
684
685Classes and functions
686---------------------
687
Benjamin Peterson52baa292009-03-24 00:56:30 +0000688This section describes in depth the API of :mod:`unittest`.
689
690
691.. _testcase-objects:
692
693Test cases
694~~~~~~~~~~
Georg Brandl116aa622007-08-15 14:28:22 +0000695
Georg Brandl7f01a132009-09-16 15:58:14 +0000696.. class:: TestCase(methodName='runTest')
Georg Brandl116aa622007-08-15 14:28:22 +0000697
698 Instances of the :class:`TestCase` class represent the smallest testable units
699 in the :mod:`unittest` universe. This class is intended to be used as a base
700 class, with specific tests being implemented by concrete subclasses. This class
701 implements the interface needed by the test runner to allow it to drive the
702 test, and methods that the test code can use to check for and report various
703 kinds of failure.
704
705 Each instance of :class:`TestCase` will run a single test method: the method
706 named *methodName*. If you remember, we had an earlier example that went
707 something like this::
708
709 def suite():
710 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000711 suite.addTest(WidgetTestCase('test_default_size'))
712 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000713 return suite
714
715 Here, we create two instances of :class:`WidgetTestCase`, each of which runs a
716 single test.
717
Benjamin Peterson52baa292009-03-24 00:56:30 +0000718 *methodName* defaults to :meth:`runTest`.
719
720 :class:`TestCase` instances provide three groups of methods: one group used
721 to run the test, another used by the test implementation to check conditions
722 and report failures, and some inquiry methods allowing information about the
723 test itself to be gathered.
724
725 Methods in the first group (running the test) are:
726
727
728 .. method:: setUp()
729
730 Method called to prepare the test fixture. This is called immediately
731 before calling the test method; any exception raised by this method will
732 be considered an error rather than a test failure. The default
733 implementation does nothing.
734
735
736 .. method:: tearDown()
737
738 Method called immediately after the test method has been called and the
739 result recorded. This is called even if the test method raised an
740 exception, so the implementation in subclasses may need to be particularly
741 careful about checking internal state. Any exception raised by this
742 method will be considered an error rather than a test failure. This
743 method will only be called if the :meth:`setUp` succeeds, regardless of
744 the outcome of the test method. The default implementation does nothing.
745
746
Benjamin Petersonb48af542010-04-11 20:43:16 +0000747 .. method:: setUpClass()
748
749 A class method called before tests in an individual class run.
750 ``setUpClass`` is called with the class as the only argument
751 and must be decorated as a :func:`classmethod`::
752
753 @classmethod
754 def setUpClass(cls):
755 ...
756
757 See `Class and Module Fixtures`_ for more details.
758
759 .. versionadded:: 3.2
760
761
762 .. method:: tearDownClass()
763
764 A class method called after tests in an individual class have run.
765 ``tearDownClass`` is called with the class as the only argument
766 and must be decorated as a :meth:`classmethod`::
767
768 @classmethod
769 def tearDownClass(cls):
770 ...
771
772 See `Class and Module Fixtures`_ for more details.
773
774 .. versionadded:: 3.2
775
776
Georg Brandl7f01a132009-09-16 15:58:14 +0000777 .. method:: run(result=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000778
779 Run the test, collecting the result into the test result object passed as
Ezio Melotti75b2a5e2010-11-20 10:13:45 +0000780 *result*. If *result* is omitted or ``None``, a temporary result
Alexandre Vassalotti260484d2009-07-17 11:43:26 +0000781 object is created (by calling the :meth:`defaultTestResult` method) and
782 used. The result object is not returned to :meth:`run`'s caller.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000783
784 The same effect may be had by simply calling the :class:`TestCase`
785 instance.
786
787
Benjamin Petersone549ead2009-03-28 21:42:05 +0000788 .. method:: skipTest(reason)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000789
Stefan Kraha5bf3f52010-05-19 16:09:41 +0000790 Calling this during a test method or :meth:`setUp` skips the current
Benjamin Peterson52baa292009-03-24 00:56:30 +0000791 test. See :ref:`unittest-skipping` for more information.
792
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000793 .. versionadded:: 3.1
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000794
Benjamin Peterson52baa292009-03-24 00:56:30 +0000795
796 .. method:: debug()
797
798 Run the test without collecting the result. This allows exceptions raised
799 by the test to be propagated to the caller, and can be used to support
800 running tests under a debugger.
801
Ezio Melotti22170ed2010-11-20 09:57:27 +0000802 .. _assert-methods:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000803
Ezio Melotti4370b302010-11-03 20:39:14 +0000804 The :class:`TestCase` class provides a number of methods to check for and
805 report failures, such as:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000806
Ezio Melotti4370b302010-11-03 20:39:14 +0000807 +-----------------------------------------+-----------------------------+---------------+
808 | Method | Checks that | New in |
809 +=========================================+=============================+===============+
810 | :meth:`assertEqual(a, b) | ``a == b`` | |
811 | <TestCase.assertEqual>` | | |
812 +-----------------------------------------+-----------------------------+---------------+
813 | :meth:`assertNotEqual(a, b) | ``a != b`` | |
814 | <TestCase.assertNotEqual>` | | |
815 +-----------------------------------------+-----------------------------+---------------+
816 | :meth:`assertTrue(x) | ``bool(x) is True`` | |
817 | <TestCase.assertTrue>` | | |
818 +-----------------------------------------+-----------------------------+---------------+
819 | :meth:`assertFalse(x) | ``bool(x) is False`` | |
820 | <TestCase.assertFalse>` | | |
821 +-----------------------------------------+-----------------------------+---------------+
822 | :meth:`assertIs(a, b) | ``a is b`` | 3.1 |
823 | <TestCase.assertIs>` | | |
824 +-----------------------------------------+-----------------------------+---------------+
825 | :meth:`assertIsNot(a, b) | ``a is not b`` | 3.1 |
826 | <TestCase.assertIsNot>` | | |
827 +-----------------------------------------+-----------------------------+---------------+
828 | :meth:`assertIsNone(x) | ``x is None`` | 3.1 |
829 | <TestCase.assertIsNone>` | | |
830 +-----------------------------------------+-----------------------------+---------------+
831 | :meth:`assertIsNotNone(x) | ``x is not None`` | 3.1 |
832 | <TestCase.assertIsNotNone>` | | |
833 +-----------------------------------------+-----------------------------+---------------+
834 | :meth:`assertIn(a, b) | ``a in b`` | 3.1 |
835 | <TestCase.assertIn>` | | |
836 +-----------------------------------------+-----------------------------+---------------+
837 | :meth:`assertNotIn(a, b) | ``a not in b`` | 3.1 |
838 | <TestCase.assertNotIn>` | | |
839 +-----------------------------------------+-----------------------------+---------------+
840 | :meth:`assertIsInstance(a, b) | ``isinstance(a, b)`` | 3.2 |
841 | <TestCase.assertIsInstance>` | | |
842 +-----------------------------------------+-----------------------------+---------------+
843 | :meth:`assertNotIsInstance(a, b) | ``not isinstance(a, b)`` | 3.2 |
844 | <TestCase.assertNotIsInstance>` | | |
845 +-----------------------------------------+-----------------------------+---------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000846
Ezio Melotti22170ed2010-11-20 09:57:27 +0000847 All the assert methods (except :meth:`assertRaises`,
Ezio Melottied3a7d22010-12-01 02:32:32 +0000848 :meth:`assertRaisesRegex`, :meth:`assertWarns`, :meth:`assertWarnsRegex`)
Ezio Melotti22170ed2010-11-20 09:57:27 +0000849 accept a *msg* argument that, if specified, is used as the error message on
850 failure (see also :data:`longMessage`).
Benjamin Peterson52baa292009-03-24 00:56:30 +0000851
Georg Brandl7f01a132009-09-16 15:58:14 +0000852 .. method:: assertEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000853
854 Test that *first* and *second* are equal. If the values do not compare
Ezio Melotti4841fd62010-11-05 15:43:40 +0000855 equal, the test will fail.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000856
857 In addition, if *first* and *second* are the exact same type and one of
Michael Foord02834952010-02-08 23:10:39 +0000858 list, tuple, dict, set, frozenset or str or any type that a subclass
859 registers with :meth:`addTypeEqualityFunc` the type specific equality
860 function will be called in order to generate a more useful default
Ezio Melotti22170ed2010-11-20 09:57:27 +0000861 error message (see also the :ref:`list of type-specific methods
862 <type-specific-methods>`).
Ezio Melotti4841fd62010-11-05 15:43:40 +0000863
Raymond Hettinger35a88362009-04-09 00:08:24 +0000864 .. versionchanged:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000865 Added the automatic calling of type specific equality function.
866
Michael Foord28a817e2010-02-09 00:03:57 +0000867 .. versionchanged:: 3.2
868 :meth:`assertMultiLineEqual` added as the default type equality
869 function for comparing strings.
Michael Foord02834952010-02-08 23:10:39 +0000870
Benjamin Peterson52baa292009-03-24 00:56:30 +0000871
Georg Brandl7f01a132009-09-16 15:58:14 +0000872 .. method:: assertNotEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000873
874 Test that *first* and *second* are not equal. If the values do compare
Ezio Melotti4841fd62010-11-05 15:43:40 +0000875 equal, the test will fail.
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000876
Ezio Melotti4370b302010-11-03 20:39:14 +0000877 .. method:: assertTrue(expr, msg=None)
Ezio Melotti4841fd62010-11-05 15:43:40 +0000878 assertFalse(expr, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000879
Ezio Melotti4841fd62010-11-05 15:43:40 +0000880 Test that *expr* is true (or false).
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000881
Ezio Melotti4841fd62010-11-05 15:43:40 +0000882 Note that this is equivalent to ``bool(expr) is True`` and not to ``expr
883 is True`` (use ``assertIs(expr, True)`` for the latter). This method
884 should also be avoided when more specific methods are available (e.g.
885 ``assertEqual(a, b)`` instead of ``assertTrue(a == b)``), because they
886 provide a better error message in case of failure.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000887
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000888
Ezio Melotti9794a262010-11-04 14:52:13 +0000889 .. method:: assertIs(first, second, msg=None)
890 assertIsNot(first, second, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000891
Ezio Melotti9794a262010-11-04 14:52:13 +0000892 Test that *first* and *second* evaluate (or don't evaluate) to the same object.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000893
Raymond Hettinger35a88362009-04-09 00:08:24 +0000894 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000895
896
Ezio Melotti4370b302010-11-03 20:39:14 +0000897 .. method:: assertIsNone(expr, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000898 assertIsNotNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000899
Ezio Melotti9794a262010-11-04 14:52:13 +0000900 Test that *expr* is (or is not) None.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000901
Ezio Melotti4370b302010-11-03 20:39:14 +0000902 .. versionadded:: 3.1
Benjamin Petersonb48af542010-04-11 20:43:16 +0000903
904
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000905 .. method:: assertIn(first, second, msg=None)
906 assertNotIn(first, second, msg=None)
907
Ezio Melotti4841fd62010-11-05 15:43:40 +0000908 Test that *first* is (or is not) in *second*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000909
Raymond Hettinger35a88362009-04-09 00:08:24 +0000910 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000911
912
Ezio Melotti9c02c2f2010-11-03 20:45:31 +0000913 .. method:: assertIsInstance(obj, cls, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000914 assertNotIsInstance(obj, cls, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000915
Ezio Melotti9794a262010-11-04 14:52:13 +0000916 Test that *obj* is (or is not) an instance of *cls* (which can be a
917 class or a tuple of classes, as supported by :func:`isinstance`).
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000918
Ezio Melotti4370b302010-11-03 20:39:14 +0000919 .. versionadded:: 3.2
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000920
921
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000922
Ezio Melotti4370b302010-11-03 20:39:14 +0000923 It is also possible to check that exceptions and warnings are raised using
924 the following methods:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000925
Ezio Melotti4370b302010-11-03 20:39:14 +0000926 +---------------------------------------------------------+--------------------------------------+------------+
927 | Method | Checks that | New in |
928 +=========================================================+======================================+============+
929 | :meth:`assertRaises(exc, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `exc` | |
930 | <TestCase.assertRaises>` | | |
931 +---------------------------------------------------------+--------------------------------------+------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +0000932 | :meth:`assertRaisesRegex(exc, re, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `exc` | 3.1 |
933 | <TestCase.assertRaisesRegex>` | and the message matches `re` | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000934 +---------------------------------------------------------+--------------------------------------+------------+
935 | :meth:`assertWarns(warn, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `warn` | 3.2 |
936 | <TestCase.assertWarns>` | | |
937 +---------------------------------------------------------+--------------------------------------+------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +0000938 | :meth:`assertWarnsRegex(warn, re, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `warn` | 3.2 |
939 | <TestCase.assertWarnsRegex>` | and the message matches `re` | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000940 +---------------------------------------------------------+--------------------------------------+------------+
Benjamin Peterson52baa292009-03-24 00:56:30 +0000941
Georg Brandl7f01a132009-09-16 15:58:14 +0000942 .. method:: assertRaises(exception, callable, *args, **kwds)
Georg Brandl7f01a132009-09-16 15:58:14 +0000943 assertRaises(exception)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000944
945 Test that an exception is raised when *callable* is called with any
946 positional or keyword arguments that are also passed to
947 :meth:`assertRaises`. The test passes if *exception* is raised, is an
948 error if another exception is raised, or fails if no exception is raised.
949 To catch any of a group of exceptions, a tuple containing the exception
950 classes may be passed as *exception*.
951
Georg Brandl7f01a132009-09-16 15:58:14 +0000952 If only the *exception* argument is given, returns a context manager so
953 that the code under test can be written inline rather than as a function::
Benjamin Petersonded31c42009-03-30 15:04:16 +0000954
Michael Foord41531f22010-02-05 21:13:40 +0000955 with self.assertRaises(SomeException):
Benjamin Petersonded31c42009-03-30 15:04:16 +0000956 do_something()
957
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000958 The context manager will store the caught exception object in its
Ezio Melotti49008232010-02-08 21:57:48 +0000959 :attr:`exception` attribute. This can be useful if the intention
Michael Foord41531f22010-02-05 21:13:40 +0000960 is to perform additional checks on the exception raised::
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000961
Georg Brandl8a1caa22010-07-29 16:01:11 +0000962 with self.assertRaises(SomeException) as cm:
963 do_something()
Michael Foord41531f22010-02-05 21:13:40 +0000964
Georg Brandl8a1caa22010-07-29 16:01:11 +0000965 the_exception = cm.exception
966 self.assertEqual(the_exception.error_code, 3)
Michael Foord41531f22010-02-05 21:13:40 +0000967
Ezio Melotti49008232010-02-08 21:57:48 +0000968 .. versionchanged:: 3.1
Benjamin Petersonded31c42009-03-30 15:04:16 +0000969 Added the ability to use :meth:`assertRaises` as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000970
Ezio Melotti49008232010-02-08 21:57:48 +0000971 .. versionchanged:: 3.2
972 Added the :attr:`exception` attribute.
973
Benjamin Peterson52baa292009-03-24 00:56:30 +0000974
Ezio Melottied3a7d22010-12-01 02:32:32 +0000975 .. method:: assertRaisesRegex(exception, regex, callable, *args, **kwds)
976 assertRaisesRegex(exception, regex)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000977
Ezio Melottied3a7d22010-12-01 02:32:32 +0000978 Like :meth:`assertRaises` but also tests that *regex* matches
979 on the string representation of the raised exception. *regex* may be
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000980 a regular expression object or a string containing a regular expression
981 suitable for use by :func:`re.search`. Examples::
982
Ezio Melottied3a7d22010-12-01 02:32:32 +0000983 self.assertRaisesRegex(ValueError, 'invalid literal for.*XYZ$',
984 int, 'XYZ')
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000985
986 or::
987
Ezio Melottied3a7d22010-12-01 02:32:32 +0000988 with self.assertRaisesRegex(ValueError, 'literal'):
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000989 int('XYZ')
990
Georg Brandl419e3de2010-12-01 15:44:25 +0000991 .. versionadded:: 3.1
992 under the name ``assertRaisesRegexp``.
Ezio Melottied3a7d22010-12-01 02:32:32 +0000993 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +0000994 Renamed to :meth:`assertRaisesRegex`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000995
996
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000997 .. method:: assertWarns(warning, callable, *args, **kwds)
998 assertWarns(warning)
999
1000 Test that a warning is triggered when *callable* is called with any
1001 positional or keyword arguments that are also passed to
1002 :meth:`assertWarns`. The test passes if *warning* is triggered and
1003 fails if it isn't. Also, any unexpected exception is an error.
1004 To catch any of a group of warnings, a tuple containing the warning
1005 classes may be passed as *warnings*.
1006
1007 If only the *warning* argument is given, returns a context manager so
1008 that the code under test can be written inline rather than as a function::
1009
1010 with self.assertWarns(SomeWarning):
1011 do_something()
1012
1013 The context manager will store the caught warning object in its
1014 :attr:`warning` attribute, and the source line which triggered the
1015 warnings in the :attr:`filename` and :attr:`lineno` attributes.
1016 This can be useful if the intention is to perform additional checks
1017 on the exception raised::
1018
1019 with self.assertWarns(SomeWarning) as cm:
1020 do_something()
1021
1022 self.assertIn('myfile.py', cm.filename)
1023 self.assertEqual(320, cm.lineno)
1024
1025 This method works regardless of the warning filters in place when it
1026 is called.
1027
1028 .. versionadded:: 3.2
1029
1030
Ezio Melottied3a7d22010-12-01 02:32:32 +00001031 .. method:: assertWarnsRegex(warning, regex, callable, *args, **kwds)
1032 assertWarnsRegex(warning, regex)
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001033
Ezio Melottied3a7d22010-12-01 02:32:32 +00001034 Like :meth:`assertWarns` but also tests that *regex* matches on the
1035 message of the triggered warning. *regex* may be a regular expression
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001036 object or a string containing a regular expression suitable for use
1037 by :func:`re.search`. Example::
1038
Ezio Melottied3a7d22010-12-01 02:32:32 +00001039 self.assertWarnsRegex(DeprecationWarning,
1040 r'legacy_function\(\) is deprecated',
1041 legacy_function, 'XYZ')
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001042
1043 or::
1044
Ezio Melottied3a7d22010-12-01 02:32:32 +00001045 with self.assertWarnsRegex(RuntimeWarning, 'unsafe frobnicating'):
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001046 frobnicate('/etc/passwd')
1047
1048 .. versionadded:: 3.2
1049
1050
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001051
Ezio Melotti4370b302010-11-03 20:39:14 +00001052 There are also other methods used to perform more specific checks, such as:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001053
Ezio Melotti4370b302010-11-03 20:39:14 +00001054 +---------------------------------------+--------------------------------+--------------+
1055 | Method | Checks that | New in |
1056 +=======================================+================================+==============+
1057 | :meth:`assertAlmostEqual(a, b) | ``round(a-b, 7) == 0`` | |
1058 | <TestCase.assertAlmostEqual>` | | |
1059 +---------------------------------------+--------------------------------+--------------+
1060 | :meth:`assertNotAlmostEqual(a, b) | ``round(a-b, 7) != 0`` | |
1061 | <TestCase.assertNotAlmostEqual>` | | |
1062 +---------------------------------------+--------------------------------+--------------+
1063 | :meth:`assertGreater(a, b) | ``a > b`` | 3.1 |
1064 | <TestCase.assertGreater>` | | |
1065 +---------------------------------------+--------------------------------+--------------+
1066 | :meth:`assertGreaterEqual(a, b) | ``a >= b`` | 3.1 |
1067 | <TestCase.assertGreaterEqual>` | | |
1068 +---------------------------------------+--------------------------------+--------------+
1069 | :meth:`assertLess(a, b) | ``a < b`` | 3.1 |
1070 | <TestCase.assertLess>` | | |
1071 +---------------------------------------+--------------------------------+--------------+
1072 | :meth:`assertLessEqual(a, b) | ``a <= b`` | 3.1 |
1073 | <TestCase.assertLessEqual>` | | |
1074 +---------------------------------------+--------------------------------+--------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +00001075 | :meth:`assertRegex(s, re) | ``regex.search(s)`` | 3.1 |
1076 | <TestCase.assertRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001077 +---------------------------------------+--------------------------------+--------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +00001078 | :meth:`assertNotRegex(s, re) | ``not regex.search(s)`` | 3.2 |
1079 | <TestCase.assertNotRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001080 +---------------------------------------+--------------------------------+--------------+
1081 | :meth:`assertDictContainsSubset(a, b) | all the key/value pairs | 3.1 |
1082 | <TestCase.assertDictContainsSubset>` | in `a` exist in `b` | |
1083 +---------------------------------------+--------------------------------+--------------+
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001084 | :meth:`assertCountEqual(a, b) | `a` and `b` have the same | 3.2 |
1085 | <TestCase.assertCountEqual>` | elements in the same number, | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001086 | | regardless of their order | |
1087 +---------------------------------------+--------------------------------+--------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001088
1089
Ezio Melotti4370b302010-11-03 20:39:14 +00001090 .. method:: assertAlmostEqual(first, second, places=7, msg=None, delta=None)
Ezio Melotti4841fd62010-11-05 15:43:40 +00001091 assertNotAlmostEqual(first, second, places=7, msg=None, delta=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001092
Ezio Melotti4841fd62010-11-05 15:43:40 +00001093 Test that *first* and *second* are approximately (or not approximately)
1094 equal by computing the difference, rounding to the given number of
1095 decimal *places* (default 7), and comparing to zero. Note that these
1096 methods round the values to the given number of *decimal places* (i.e.
1097 like the :func:`round` function) and not *significant digits*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001098
Ezio Melotti4370b302010-11-03 20:39:14 +00001099 If *delta* is supplied instead of *places* then the difference
Ezio Melotti4841fd62010-11-05 15:43:40 +00001100 between *first* and *second* must be less (or more) than *delta*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001101
Ezio Melotti4370b302010-11-03 20:39:14 +00001102 Supplying both *delta* and *places* raises a ``TypeError``.
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001103
Ezio Melotti4370b302010-11-03 20:39:14 +00001104 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001105 :meth:`assertAlmostEqual` automatically considers almost equal objects
1106 that compare equal. :meth:`assertNotAlmostEqual` automatically fails
1107 if the objects compare equal. Added the *delta* keyword argument.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001108
Ezio Melotti4370b302010-11-03 20:39:14 +00001109
Ezio Melotti4370b302010-11-03 20:39:14 +00001110 .. method:: assertGreater(first, second, msg=None)
1111 assertGreaterEqual(first, second, msg=None)
1112 assertLess(first, second, msg=None)
1113 assertLessEqual(first, second, msg=None)
1114
1115 Test that *first* is respectively >, >=, < or <= than *second* depending
Ezio Melotti4841fd62010-11-05 15:43:40 +00001116 on the method name. If not, the test will fail::
Ezio Melotti4370b302010-11-03 20:39:14 +00001117
1118 >>> self.assertGreaterEqual(3, 4)
1119 AssertionError: "3" unexpectedly not greater than or equal to "4"
1120
1121 .. versionadded:: 3.1
1122
1123
Ezio Melottied3a7d22010-12-01 02:32:32 +00001124 .. method:: assertRegex(text, regex, msg=None)
1125 assertNotRegex(text, regex, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001126
Ezio Melottied3a7d22010-12-01 02:32:32 +00001127 Test that a *regex* search matches (or does not match) *text*. In case
Ezio Melotti4841fd62010-11-05 15:43:40 +00001128 of failure, the error message will include the pattern and the *text* (or
Ezio Melottied3a7d22010-12-01 02:32:32 +00001129 the pattern and the part of *text* that unexpectedly matched). *regex*
Ezio Melotti4370b302010-11-03 20:39:14 +00001130 may be a regular expression object or a string containing a regular
1131 expression suitable for use by :func:`re.search`.
1132
Georg Brandl419e3de2010-12-01 15:44:25 +00001133 .. versionadded:: 3.1
1134 under the name ``assertRegexpMatches``.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001135 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001136 The method ``assertRegexpMatches()`` has been renamed to
1137 :meth:`.assertRegex`.
1138 .. versionadded:: 3.2
1139 :meth:`.assertNotRegex`.
Ezio Melotti4370b302010-11-03 20:39:14 +00001140
1141
1142 .. method:: assertDictContainsSubset(expected, actual, msg=None)
1143
1144 Tests whether the key/value pairs in dictionary *actual* are a
1145 superset of those in *expected*. If not, an error message listing
1146 the missing keys and mismatched values is generated.
1147
Ezio Melotti4370b302010-11-03 20:39:14 +00001148 .. versionadded:: 3.1
1149
1150
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001151 .. method:: assertCountEqual(expected, actual, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001152
1153 Test that sequence *expected* contains the same elements as *actual*,
1154 regardless of their order. When they don't, an error message listing the
1155 differences between the sequences will be generated.
1156
1157 Duplicate elements are *not* ignored when comparing *actual* and
1158 *expected*. It verifies if each element has the same count in both
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001159 sequences. Equivalent to:
1160 ``assertEqual(Counter(iter(expected)), Counter(iter(actual)))``
1161 but works with sequences of unhashable objects as well.
Ezio Melotti4370b302010-11-03 20:39:14 +00001162
Ezio Melotti4370b302010-11-03 20:39:14 +00001163 .. versionadded:: 3.2
1164
Ezio Melotti4370b302010-11-03 20:39:14 +00001165 .. method:: assertSameElements(actual, expected, msg=None)
1166
1167 Test that sequence *expected* contains the same elements as *actual*,
1168 regardless of their order. When they don't, an error message listing
1169 the differences between the sequences will be generated.
1170
1171 Duplicate elements are ignored when comparing *actual* and *expected*.
1172 It is the equivalent of ``assertEqual(set(expected), set(actual))``
1173 but it works with sequences of unhashable objects as well. Because
1174 duplicates are ignored, this method has been deprecated in favour of
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001175 :meth:`assertCountEqual`.
Ezio Melotti4370b302010-11-03 20:39:14 +00001176
Ezio Melotti4370b302010-11-03 20:39:14 +00001177 .. versionadded:: 3.1
1178 .. deprecated:: 3.2
1179
1180
Ezio Melotti22170ed2010-11-20 09:57:27 +00001181 .. _type-specific-methods:
Ezio Melotti4370b302010-11-03 20:39:14 +00001182
Ezio Melotti22170ed2010-11-20 09:57:27 +00001183 The :meth:`assertEqual` method dispatches the equality check for objects of
1184 the same type to different type-specific methods. These methods are already
1185 implemented for most of the built-in types, but it's also possible to
1186 register new methods using :meth:`addTypeEqualityFunc`:
1187
1188 .. method:: addTypeEqualityFunc(typeobj, function)
1189
1190 Registers a type-specific method called by :meth:`assertEqual` to check
1191 if two objects of exactly the same *typeobj* (not subclasses) compare
1192 equal. *function* must take two positional arguments and a third msg=None
1193 keyword argument just as :meth:`assertEqual` does. It must raise
1194 :data:`self.failureException(msg) <failureException>` when inequality
1195 between the first two parameters is detected -- possibly providing useful
1196 information and explaining the inequalities in details in the error
1197 message.
1198
1199 .. versionadded:: 3.1
1200
1201 The list of type-specific methods automatically used by
1202 :meth:`~TestCase.assertEqual` are summarized in the following table. Note
1203 that it's usually not necessary to invoke these methods directly.
Ezio Melotti4370b302010-11-03 20:39:14 +00001204
1205 +-----------------------------------------+-----------------------------+--------------+
1206 | Method | Used to compare | New in |
1207 +=========================================+=============================+==============+
1208 | :meth:`assertMultiLineEqual(a, b) | strings | 3.1 |
1209 | <TestCase.assertMultiLineEqual>` | | |
1210 +-----------------------------------------+-----------------------------+--------------+
1211 | :meth:`assertSequenceEqual(a, b) | sequences | 3.1 |
1212 | <TestCase.assertSequenceEqual>` | | |
1213 +-----------------------------------------+-----------------------------+--------------+
1214 | :meth:`assertListEqual(a, b) | lists | 3.1 |
1215 | <TestCase.assertListEqual>` | | |
1216 +-----------------------------------------+-----------------------------+--------------+
1217 | :meth:`assertTupleEqual(a, b) | tuples | 3.1 |
1218 | <TestCase.assertTupleEqual>` | | |
1219 +-----------------------------------------+-----------------------------+--------------+
1220 | :meth:`assertSetEqual(a, b) | sets or frozensets | 3.1 |
1221 | <TestCase.assertSetEqual>` | | |
1222 +-----------------------------------------+-----------------------------+--------------+
1223 | :meth:`assertDictEqual(a, b) | dicts | 3.1 |
1224 | <TestCase.assertDictEqual>` | | |
1225 +-----------------------------------------+-----------------------------+--------------+
1226
1227
1228
Ezio Melotti9c02c2f2010-11-03 20:45:31 +00001229 .. method:: assertMultiLineEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001230
1231 Test that the multiline string *first* is equal to the string *second*.
1232 When not equal a diff of the two strings highlighting the differences
1233 will be included in the error message. This method is used by default
1234 when comparing strings with :meth:`assertEqual`.
1235
Ezio Melotti4370b302010-11-03 20:39:14 +00001236 .. versionadded:: 3.1
1237
1238
1239 .. method:: assertSequenceEqual(seq1, seq2, msg=None, seq_type=None)
1240
1241 Tests that two sequences are equal. If a *seq_type* is supplied, both
1242 *seq1* and *seq2* must be instances of *seq_type* or a failure will
1243 be raised. If the sequences are different an error message is
1244 constructed that shows the difference between the two.
1245
Ezio Melotti22170ed2010-11-20 09:57:27 +00001246 This method is not called directly by :meth:`assertEqual`, but
1247 it's used to implement :meth:`assertListEqual` and
Ezio Melotti4370b302010-11-03 20:39:14 +00001248 :meth:`assertTupleEqual`.
1249
1250 .. versionadded:: 3.1
1251
1252
1253 .. method:: assertListEqual(list1, list2, msg=None)
1254 assertTupleEqual(tuple1, tuple2, msg=None)
1255
1256 Tests that two lists or tuples are equal. If not an error message is
1257 constructed that shows only the differences between the two. An error
1258 is also raised if either of the parameters are of the wrong type.
1259 These methods are used by default when comparing lists or tuples with
1260 :meth:`assertEqual`.
1261
Ezio Melotti4370b302010-11-03 20:39:14 +00001262 .. versionadded:: 3.1
1263
1264
1265 .. method:: assertSetEqual(set1, set2, msg=None)
1266
1267 Tests that two sets are equal. If not, an error message is constructed
1268 that lists the differences between the sets. This method is used by
1269 default when comparing sets or frozensets with :meth:`assertEqual`.
1270
1271 Fails if either of *set1* or *set2* does not have a :meth:`set.difference`
1272 method.
1273
Ezio Melotti4370b302010-11-03 20:39:14 +00001274 .. versionadded:: 3.1
1275
1276
1277 .. method:: assertDictEqual(expected, actual, msg=None)
1278
1279 Test that two dictionaries are equal. If not, an error message is
1280 constructed that shows the differences in the dictionaries. This
1281 method will be used by default to compare dictionaries in
1282 calls to :meth:`assertEqual`.
1283
Ezio Melotti4370b302010-11-03 20:39:14 +00001284 .. versionadded:: 3.1
1285
1286
1287
Ezio Melotti22170ed2010-11-20 09:57:27 +00001288 .. _other-methods-and-attrs:
1289
Ezio Melotti4370b302010-11-03 20:39:14 +00001290 Finally the :class:`TestCase` provides the following methods and attributes:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001291
Benjamin Peterson52baa292009-03-24 00:56:30 +00001292
Georg Brandl7f01a132009-09-16 15:58:14 +00001293 .. method:: fail(msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001294
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001295 Signals a test failure unconditionally, with *msg* or ``None`` for
Benjamin Peterson52baa292009-03-24 00:56:30 +00001296 the error message.
1297
1298
1299 .. attribute:: failureException
1300
1301 This class attribute gives the exception raised by the test method. If a
1302 test framework needs to use a specialized exception, possibly to carry
1303 additional information, it must subclass this exception in order to "play
1304 fair" with the framework. The initial value of this attribute is
1305 :exc:`AssertionError`.
1306
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001307
1308 .. attribute:: longMessage
1309
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001310 If set to ``True`` then any explicit failure message you pass in to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001311 :ref:`assert methods <assert-methods>` will be appended to the end of the
1312 normal failure message. The normal messages contain useful information
1313 about the objects involved, for example the message from assertEqual
1314 shows you the repr of the two unequal objects. Setting this attribute
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001315 to ``True`` allows you to have a custom error message in addition to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001316 normal one.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001317
Michael Foord5074df62010-12-03 00:53:09 +00001318 This attribute defaults to ``True``. If set to False then a custom message
1319 passed to an assert method will silence the normal message.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001320
1321 The class setting can be overridden in individual tests by assigning an
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001322 instance attribute to ``True`` or ``False`` before calling the assert methods.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001323
Raymond Hettinger35a88362009-04-09 00:08:24 +00001324 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001325
1326
Michael Foord98b3e762010-06-05 21:59:55 +00001327 .. attribute:: maxDiff
1328
1329 This attribute controls the maximum length of diffs output by assert
1330 methods that report diffs on failure. It defaults to 80*8 characters.
1331 Assert methods affected by this attribute are
1332 :meth:`assertSequenceEqual` (including all the sequence comparison
1333 methods that delegate to it), :meth:`assertDictEqual` and
1334 :meth:`assertMultiLineEqual`.
1335
1336 Setting ``maxDiff`` to None means that there is no maximum length of
1337 diffs.
1338
1339 .. versionadded:: 3.2
1340
1341
Benjamin Peterson52baa292009-03-24 00:56:30 +00001342 Testing frameworks can use the following methods to collect information on
1343 the test:
1344
1345
1346 .. method:: countTestCases()
1347
1348 Return the number of tests represented by this test object. For
1349 :class:`TestCase` instances, this will always be ``1``.
1350
1351
1352 .. method:: defaultTestResult()
1353
1354 Return an instance of the test result class that should be used for this
1355 test case class (if no other result instance is provided to the
1356 :meth:`run` method).
1357
1358 For :class:`TestCase` instances, this will always be an instance of
1359 :class:`TestResult`; subclasses of :class:`TestCase` should override this
1360 as necessary.
1361
1362
1363 .. method:: id()
1364
1365 Return a string identifying the specific test case. This is usually the
1366 full name of the test method, including the module and class name.
1367
1368
1369 .. method:: shortDescription()
1370
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001371 Returns a description of the test, or ``None`` if no description
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001372 has been provided. The default implementation of this method
1373 returns the first line of the test method's docstring, if available,
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001374 or ``None``.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001375
Georg Brandl419e3de2010-12-01 15:44:25 +00001376 .. versionchanged:: 3.1
Michael Foord34c94622010-02-10 15:51:42 +00001377 In 3.1 this was changed to add the test name to the short description
Georg Brandl419e3de2010-12-01 15:44:25 +00001378 even in the presence of a docstring. This caused compatibility issues
Michael Foord34c94622010-02-10 15:51:42 +00001379 with unittest extensions and adding the test name was moved to the
Georg Brandl419e3de2010-12-01 15:44:25 +00001380 :class:`TextTestResult` in Python 3.2.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001381
Georg Brandl116aa622007-08-15 14:28:22 +00001382
Georg Brandl7f01a132009-09-16 15:58:14 +00001383 .. method:: addCleanup(function, *args, **kwargs)
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001384
1385 Add a function to be called after :meth:`tearDown` to cleanup resources
1386 used during the test. Functions will be called in reverse order to the
1387 order they are added (LIFO). They are called with any arguments and
1388 keyword arguments passed into :meth:`addCleanup` when they are
1389 added.
1390
1391 If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
1392 then any cleanup functions added will still be called.
1393
Georg Brandl853947a2010-01-31 18:53:23 +00001394 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001395
1396
1397 .. method:: doCleanups()
1398
Barry Warsaw0c9fd632010-04-12 14:50:57 +00001399 This method is called unconditionally after :meth:`tearDown`, or
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001400 after :meth:`setUp` if :meth:`setUp` raises an exception.
1401
1402 It is responsible for calling all the cleanup functions added by
1403 :meth:`addCleanup`. If you need cleanup functions to be called
1404 *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
1405 yourself.
1406
1407 :meth:`doCleanups` pops methods off the stack of cleanup
1408 functions one at a time, so it can be called at any time.
1409
Georg Brandl853947a2010-01-31 18:53:23 +00001410 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001411
1412
Georg Brandl7f01a132009-09-16 15:58:14 +00001413.. class:: FunctionTestCase(testFunc, setUp=None, tearDown=None, description=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001414
1415 This class implements the portion of the :class:`TestCase` interface which
Benjamin Petersond2397752009-06-27 23:45:02 +00001416 allows the test runner to drive the test, but does not provide the methods
1417 which test code can use to check and report errors. This is used to create
1418 test cases using legacy test code, allowing it to be integrated into a
1419 :mod:`unittest`-based test framework.
Georg Brandl116aa622007-08-15 14:28:22 +00001420
1421
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001422.. _deprecated-aliases:
1423
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001424Deprecated aliases
1425##################
1426
1427For historical reasons, some of the :class:`TestCase` methods had one or more
1428aliases that are now deprecated. The following table lists the correct names
1429along with their deprecated aliases:
1430
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001431 ============================== ====================== ======================
1432 Method Name Deprecated alias Deprecated alias
1433 ============================== ====================== ======================
1434 :meth:`.assertEqual` failUnlessEqual assertEquals
1435 :meth:`.assertNotEqual` failIfEqual assertNotEquals
1436 :meth:`.assertTrue` failUnless assert\_
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001437 :meth:`.assertFalse` failIf
1438 :meth:`.assertRaises` failUnlessRaises
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001439 :meth:`.assertAlmostEqual` failUnlessAlmostEqual assertAlmostEquals
1440 :meth:`.assertNotAlmostEqual` failIfAlmostEqual assertNotAlmostEquals
Ezio Melottied3a7d22010-12-01 02:32:32 +00001441 :meth:`.assertRegex` assertRegexpMatches
1442 :meth:`.assertRaisesRegex` assertRaisesRegexp
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001443 ============================== ====================== ======================
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001444
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001445 .. deprecated-removed:: 3.1 3.3
1446 the fail* aliases listed in the second column.
1447 .. deprecated:: 3.2
1448 the assert* aliases listed in the third column.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001449 .. deprecated:: 3.2
1450 ``assertRegexpMatches`` and ``assertRaisesRegexp`` have been renamed to
1451 :meth:`.assertRegex` and :meth:`.assertRaisesRegex`
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001452
1453
Benjamin Peterson52baa292009-03-24 00:56:30 +00001454.. _testsuite-objects:
1455
Benjamin Peterson52baa292009-03-24 00:56:30 +00001456Grouping tests
1457~~~~~~~~~~~~~~
1458
Georg Brandl7f01a132009-09-16 15:58:14 +00001459.. class:: TestSuite(tests=())
Georg Brandl116aa622007-08-15 14:28:22 +00001460
1461 This class represents an aggregation of individual tests cases and test suites.
1462 The class presents the interface needed by the test runner to allow it to be run
1463 as any other test case. Running a :class:`TestSuite` instance is the same as
1464 iterating over the suite, running each test individually.
1465
1466 If *tests* is given, it must be an iterable of individual test cases or other
1467 test suites that will be used to build the suite initially. Additional methods
1468 are provided to add test cases and suites to the collection later on.
1469
Benjamin Peterson14a3dd72009-05-25 00:51:58 +00001470 :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
1471 they do not actually implement a test. Instead, they are used to aggregate
1472 tests into groups of tests that should be run together. Some additional
1473 methods are available to add tests to :class:`TestSuite` instances:
Benjamin Peterson52baa292009-03-24 00:56:30 +00001474
1475
1476 .. method:: TestSuite.addTest(test)
1477
1478 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1479
1480
1481 .. method:: TestSuite.addTests(tests)
1482
1483 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
1484 instances to this test suite.
1485
Benjamin Petersond2397752009-06-27 23:45:02 +00001486 This is equivalent to iterating over *tests*, calling :meth:`addTest` for
1487 each element.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001488
1489 :class:`TestSuite` shares the following methods with :class:`TestCase`:
1490
1491
1492 .. method:: run(result)
1493
1494 Run the tests associated with this suite, collecting the result into the
1495 test result object passed as *result*. Note that unlike
1496 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
1497 be passed in.
1498
1499
1500 .. method:: debug()
1501
1502 Run the tests associated with this suite without collecting the
1503 result. This allows exceptions raised by the test to be propagated to the
1504 caller and can be used to support running tests under a debugger.
1505
1506
1507 .. method:: countTestCases()
1508
1509 Return the number of tests represented by this test object, including all
1510 individual tests and sub-suites.
1511
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001512
1513 .. method:: __iter__()
1514
1515 Tests grouped by a :class:`TestSuite` are always accessed by iteration.
1516 Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
1517 that this method maybe called several times on a single suite
1518 (for example when counting tests or comparing for equality)
1519 so the tests returned must be the same for repeated iterations.
1520
Georg Brandl853947a2010-01-31 18:53:23 +00001521 .. versionchanged:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001522 In earlier versions the :class:`TestSuite` accessed tests directly rather
1523 than through iteration, so overriding :meth:`__iter__` wasn't sufficient
1524 for providing tests.
1525
Benjamin Peterson52baa292009-03-24 00:56:30 +00001526 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
1527 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1528
1529
Benjamin Peterson52baa292009-03-24 00:56:30 +00001530Loading and running tests
1531~~~~~~~~~~~~~~~~~~~~~~~~~
1532
Georg Brandl116aa622007-08-15 14:28:22 +00001533.. class:: TestLoader()
1534
Benjamin Peterson52baa292009-03-24 00:56:30 +00001535 The :class:`TestLoader` class is used to create test suites from classes and
1536 modules. Normally, there is no need to create an instance of this class; the
1537 :mod:`unittest` module provides an instance that can be shared as
1538 ``unittest.defaultTestLoader``. Using a subclass or instance, however, allows
1539 customization of some configurable properties.
1540
1541 :class:`TestLoader` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001542
Ezio Melotti9c02c2f2010-11-03 20:45:31 +00001543
Benjamin Peterson52baa292009-03-24 00:56:30 +00001544 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl116aa622007-08-15 14:28:22 +00001545
Benjamin Peterson52baa292009-03-24 00:56:30 +00001546 Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
1547 :class:`testCaseClass`.
1548
1549
1550 .. method:: loadTestsFromModule(module)
1551
1552 Return a suite of all tests cases contained in the given module. This
1553 method searches *module* for classes derived from :class:`TestCase` and
1554 creates an instance of the class for each test method defined for the
1555 class.
1556
Georg Brandle720c0a2009-04-27 16:20:50 +00001557 .. note::
Benjamin Peterson52baa292009-03-24 00:56:30 +00001558
1559 While using a hierarchy of :class:`TestCase`\ -derived classes can be
1560 convenient in sharing fixtures and helper functions, defining test
1561 methods on base classes that are not intended to be instantiated
1562 directly does not play well with this method. Doing so, however, can
1563 be useful when the fixtures are different and defined in subclasses.
1564
Benjamin Petersond2397752009-06-27 23:45:02 +00001565 If a module provides a ``load_tests`` function it will be called to
1566 load the tests. This allows modules to customize test loading.
1567 This is the `load_tests protocol`_.
1568
Georg Brandl853947a2010-01-31 18:53:23 +00001569 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001570 Support for ``load_tests`` added.
1571
Benjamin Peterson52baa292009-03-24 00:56:30 +00001572
Georg Brandl7f01a132009-09-16 15:58:14 +00001573 .. method:: loadTestsFromName(name, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001574
1575 Return a suite of all tests cases given a string specifier.
1576
1577 The specifier *name* is a "dotted name" that may resolve either to a
1578 module, a test case class, a test method within a test case class, a
1579 :class:`TestSuite` instance, or a callable object which returns a
1580 :class:`TestCase` or :class:`TestSuite` instance. These checks are
1581 applied in the order listed here; that is, a method on a possible test
1582 case class will be picked up as "a test method within a test case class",
1583 rather than "a callable object".
1584
1585 For example, if you have a module :mod:`SampleTests` containing a
1586 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
1587 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
Benjamin Petersond2397752009-06-27 23:45:02 +00001588 specifier ``'SampleTests.SampleTestCase'`` would cause this method to
1589 return a suite which will run all three test methods. Using the specifier
1590 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
1591 suite which will run only the :meth:`test_two` test method. The specifier
1592 can refer to modules and packages which have not been imported; they will
1593 be imported as a side-effect.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001594
1595 The method optionally resolves *name* relative to the given *module*.
1596
1597
Georg Brandl7f01a132009-09-16 15:58:14 +00001598 .. method:: loadTestsFromNames(names, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001599
1600 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
1601 than a single name. The return value is a test suite which supports all
1602 the tests defined for each name.
1603
1604
1605 .. method:: getTestCaseNames(testCaseClass)
1606
1607 Return a sorted sequence of method names found within *testCaseClass*;
1608 this should be a subclass of :class:`TestCase`.
1609
Benjamin Petersond2397752009-06-27 23:45:02 +00001610
1611 .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)
1612
1613 Find and return all test modules from the specified start directory,
1614 recursing into subdirectories to find them. Only test files that match
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001615 *pattern* will be loaded. (Using shell style pattern matching.) Only
1616 module names that are importable (i.e. are valid Python identifiers) will
1617 be loaded.
Benjamin Petersond2397752009-06-27 23:45:02 +00001618
1619 All test modules must be importable from the top level of the project. If
1620 the start directory is not the top level directory then the top level
1621 directory must be specified separately.
1622
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001623 If importing a module fails, for example due to a syntax error, then this
1624 will be recorded as a single error and discovery will continue.
1625
Benjamin Petersond2397752009-06-27 23:45:02 +00001626 If a test package name (directory with :file:`__init__.py`) matches the
1627 pattern then the package will be checked for a ``load_tests``
1628 function. If this exists then it will be called with *loader*, *tests*,
1629 *pattern*.
1630
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001631 If load_tests exists then discovery does *not* recurse into the package,
Benjamin Petersond2397752009-06-27 23:45:02 +00001632 ``load_tests`` is responsible for loading all tests in the package.
1633
1634 The pattern is deliberately not stored as a loader attribute so that
1635 packages can continue discovery themselves. *top_level_dir* is stored so
1636 ``load_tests`` does not need to pass this argument in to
1637 ``loader.discover()``.
1638
Benjamin Petersonb48af542010-04-11 20:43:16 +00001639 *start_dir* can be a dotted module name as well as a directory.
1640
Georg Brandl853947a2010-01-31 18:53:23 +00001641 .. versionadded:: 3.2
1642
Benjamin Petersond2397752009-06-27 23:45:02 +00001643
Benjamin Peterson52baa292009-03-24 00:56:30 +00001644 The following attributes of a :class:`TestLoader` can be configured either by
1645 subclassing or assignment on an instance:
1646
1647
1648 .. attribute:: testMethodPrefix
1649
1650 String giving the prefix of method names which will be interpreted as test
1651 methods. The default value is ``'test'``.
1652
1653 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
1654 methods.
1655
1656
1657 .. attribute:: sortTestMethodsUsing
1658
1659 Function to be used to compare method names when sorting them in
1660 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
1661
1662
1663 .. attribute:: suiteClass
1664
1665 Callable object that constructs a test suite from a list of tests. No
1666 methods on the resulting object are needed. The default value is the
1667 :class:`TestSuite` class.
1668
1669 This affects all the :meth:`loadTestsFrom\*` methods.
1670
1671
Benjamin Peterson52baa292009-03-24 00:56:30 +00001672.. class:: TestResult
1673
1674 This class is used to compile information about which tests have succeeded
1675 and which have failed.
1676
1677 A :class:`TestResult` object stores the results of a set of tests. The
1678 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
1679 properly recorded; test authors do not need to worry about recording the
1680 outcome of tests.
1681
1682 Testing frameworks built on top of :mod:`unittest` may want access to the
1683 :class:`TestResult` object generated by running a set of tests for reporting
1684 purposes; a :class:`TestResult` instance is returned by the
1685 :meth:`TestRunner.run` method for this purpose.
1686
1687 :class:`TestResult` instances have the following attributes that will be of
1688 interest when inspecting the results of running a set of tests:
1689
1690
1691 .. attribute:: errors
1692
1693 A list containing 2-tuples of :class:`TestCase` instances and strings
1694 holding formatted tracebacks. Each tuple represents a test which raised an
1695 unexpected exception.
1696
Benjamin Peterson52baa292009-03-24 00:56:30 +00001697 .. attribute:: failures
1698
1699 A list containing 2-tuples of :class:`TestCase` instances and strings
1700 holding formatted tracebacks. Each tuple represents a test where a failure
1701 was explicitly signalled using the :meth:`TestCase.fail\*` or
1702 :meth:`TestCase.assert\*` methods.
1703
Benjamin Peterson52baa292009-03-24 00:56:30 +00001704 .. attribute:: skipped
1705
1706 A list containing 2-tuples of :class:`TestCase` instances and strings
1707 holding the reason for skipping the test.
1708
Benjamin Peterson70e32c82009-03-24 01:00:11 +00001709 .. versionadded:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +00001710
1711 .. attribute:: expectedFailures
1712
Georg Brandl6faee4e2010-09-21 14:48:28 +00001713 A list containing 2-tuples of :class:`TestCase` instances and strings
1714 holding formatted tracebacks. Each tuple represents an expected failure
Benjamin Peterson52baa292009-03-24 00:56:30 +00001715 of the test case.
1716
1717 .. attribute:: unexpectedSuccesses
1718
1719 A list containing :class:`TestCase` instances that were marked as expected
1720 failures, but succeeded.
1721
1722 .. attribute:: shouldStop
1723
1724 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1725
1726
1727 .. attribute:: testsRun
1728
1729 The total number of tests run so far.
1730
1731
Georg Brandl12037202010-12-02 22:35:25 +00001732 .. attribute:: buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +00001733
1734 If set to true, ``sys.stdout`` and ``sys.stderr`` will be buffered in between
1735 :meth:`startTest` and :meth:`stopTest` being called. Collected output will
1736 only be echoed onto the real ``sys.stdout`` and ``sys.stderr`` if the test
1737 fails or errors. Any output is also attached to the failure / error message.
1738
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001739 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001740
1741
1742 .. attribute:: failfast
1743
1744 If set to true :meth:`stop` will be called on the first failure or error,
1745 halting the test run.
1746
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001747 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001748
1749
Benjamin Peterson52baa292009-03-24 00:56:30 +00001750 .. method:: wasSuccessful()
1751
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001752 Return ``True`` if all tests run so far have passed, otherwise returns
1753 ``False``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001754
1755
1756 .. method:: stop()
1757
1758 This method can be called to signal that the set of tests being run should
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001759 be aborted by setting the :attr:`shouldStop` attribute to ``True``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001760 :class:`TestRunner` objects should respect this flag and return without
1761 running any additional tests.
1762
1763 For example, this feature is used by the :class:`TextTestRunner` class to
1764 stop the test framework when the user signals an interrupt from the
1765 keyboard. Interactive tools which provide :class:`TestRunner`
1766 implementations can use this in a similar manner.
1767
1768 The following methods of the :class:`TestResult` class are used to maintain
1769 the internal data structures, and may be extended in subclasses to support
1770 additional reporting requirements. This is particularly useful in building
1771 tools which support interactive reporting while tests are being run.
1772
1773
1774 .. method:: startTest(test)
1775
1776 Called when the test case *test* is about to be run.
1777
Benjamin Peterson52baa292009-03-24 00:56:30 +00001778 .. method:: stopTest(test)
1779
1780 Called after the test case *test* has been executed, regardless of the
1781 outcome.
1782
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001783 .. method:: startTestRun(test)
1784
1785 Called once before any tests are executed.
1786
Georg Brandl853947a2010-01-31 18:53:23 +00001787 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001788
1789
1790 .. method:: stopTestRun(test)
1791
Ezio Melotti176d6c42010-01-27 20:58:07 +00001792 Called once after all tests are executed.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001793
Georg Brandl853947a2010-01-31 18:53:23 +00001794 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001795
1796
Benjamin Peterson52baa292009-03-24 00:56:30 +00001797 .. method:: addError(test, err)
1798
1799 Called when the test case *test* raises an unexpected exception *err* is a
1800 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1801 traceback)``.
1802
1803 The default implementation appends a tuple ``(test, formatted_err)`` to
1804 the instance's :attr:`errors` attribute, where *formatted_err* is a
1805 formatted traceback derived from *err*.
1806
1807
1808 .. method:: addFailure(test, err)
1809
Benjamin Petersond2397752009-06-27 23:45:02 +00001810 Called when the test case *test* signals a failure. *err* is a tuple of
1811 the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001812
1813 The default implementation appends a tuple ``(test, formatted_err)`` to
1814 the instance's :attr:`failures` attribute, where *formatted_err* is a
1815 formatted traceback derived from *err*.
1816
1817
1818 .. method:: addSuccess(test)
1819
1820 Called when the test case *test* succeeds.
1821
1822 The default implementation does nothing.
1823
1824
1825 .. method:: addSkip(test, reason)
1826
1827 Called when the test case *test* is skipped. *reason* is the reason the
1828 test gave for skipping.
1829
1830 The default implementation appends a tuple ``(test, reason)`` to the
1831 instance's :attr:`skipped` attribute.
1832
1833
1834 .. method:: addExpectedFailure(test, err)
1835
1836 Called when the test case *test* fails, but was marked with the
1837 :func:`expectedFailure` decorator.
1838
1839 The default implementation appends a tuple ``(test, formatted_err)`` to
1840 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1841 is a formatted traceback derived from *err*.
1842
1843
1844 .. method:: addUnexpectedSuccess(test)
1845
1846 Called when the test case *test* was marked with the
1847 :func:`expectedFailure` decorator, but succeeded.
1848
1849 The default implementation appends the test to the instance's
1850 :attr:`unexpectedSuccesses` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +00001851
Georg Brandl67b21b72010-08-17 15:07:14 +00001852
Michael Foord34c94622010-02-10 15:51:42 +00001853.. class:: TextTestResult(stream, descriptions, verbosity)
1854
Georg Brandl67b21b72010-08-17 15:07:14 +00001855 A concrete implementation of :class:`TestResult` used by the
1856 :class:`TextTestRunner`.
Michael Foord34c94622010-02-10 15:51:42 +00001857
Georg Brandl67b21b72010-08-17 15:07:14 +00001858 .. versionadded:: 3.2
1859 This class was previously named ``_TextTestResult``. The old name still
1860 exists as an alias but is deprecated.
1861
Georg Brandl116aa622007-08-15 14:28:22 +00001862
1863.. data:: defaultTestLoader
1864
1865 Instance of the :class:`TestLoader` class intended to be shared. If no
1866 customization of the :class:`TestLoader` is needed, this instance can be used
1867 instead of repeatedly creating new instances.
1868
1869
Ezio Melotti60901872010-12-01 00:56:10 +00001870.. class:: TextTestRunner(stream=sys.stderr, descriptions=True, verbosity=1, runnerclass=None, warnings=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001871
1872 A basic test runner implementation which prints results on standard error. It
1873 has a few configurable parameters, but is essentially very simple. Graphical
1874 applications which run test suites should provide alternate implementations.
1875
Ezio Melotti60901872010-12-01 00:56:10 +00001876 By default this runner shows :exc:`DeprecationWarning`,
1877 :exc:`PendingDeprecationWarning`, and :exc:`ImportWarning` even if they are
1878 :ref:`ignored by default <warning-ignored>`. Deprecation warnings caused by
1879 :ref:`deprecated unittest methods <deprecated-aliases>` are also
1880 special-cased and, when the warning filters are ``'default'`` or ``'always'``,
1881 they will appear only once per-module, in order to avoid too many warning
1882 messages. This behavior can be overridden using the :option`-Wd` or
1883 :option:`-Wa` options and leaving *warnings* to ``None``.
1884
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001885 .. method:: _makeResult()
Georg Brandl116aa622007-08-15 14:28:22 +00001886
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001887 This method returns the instance of ``TestResult`` used by :meth:`run`.
1888 It is not intended to be called directly, but can be overridden in
1889 subclasses to provide a custom ``TestResult``.
1890
Michael Foord34c94622010-02-10 15:51:42 +00001891 ``_makeResult()`` instantiates the class or callable passed in the
1892 ``TextTestRunner`` constructor as the ``resultclass`` argument. It
Benjamin Petersonb48af542010-04-11 20:43:16 +00001893 defaults to :class:`TextTestResult` if no ``resultclass`` is provided.
Michael Foord34c94622010-02-10 15:51:42 +00001894 The result class is instantiated with the following arguments::
1895
1896 stream, descriptions, verbosity
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001897
Georg Brandl419e3de2010-12-01 15:44:25 +00001898 .. versionchanged:: 3.2
1899 Added the ``warnings`` argument.
Ezio Melotti60901872010-12-01 00:56:10 +00001900
Georg Brandl419e3de2010-12-01 15:44:25 +00001901.. function:: main(module='__main__', defaultTest=None, argv=None, testRunner=None, \
1902 testLoader=unittest.loader.defaultTestLoader, exit=True, verbosity=1, \
1903 failfast=None, catchbreak=None, buffer=None, warnings=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001904
1905 A command-line program that runs a set of tests; this is primarily for making
1906 test modules conveniently executable. The simplest use for this function is to
1907 include the following line at the end of a test script::
1908
1909 if __name__ == '__main__':
1910 unittest.main()
1911
Benjamin Petersond2397752009-06-27 23:45:02 +00001912 You can run tests with more detailed information by passing in the verbosity
1913 argument::
1914
1915 if __name__ == '__main__':
1916 unittest.main(verbosity=2)
1917
Georg Brandl116aa622007-08-15 14:28:22 +00001918 The *testRunner* argument can either be a test runner class or an already
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001919 created instance of it. By default ``main`` calls :func:`sys.exit` with
1920 an exit code indicating success or failure of the tests run.
1921
1922 ``main`` supports being used from the interactive interpreter by passing in the
1923 argument ``exit=False``. This displays the result on standard output without
1924 calling :func:`sys.exit`::
1925
1926 >>> from unittest import main
1927 >>> main(module='test_module', exit=False)
1928
Benjamin Petersonb48af542010-04-11 20:43:16 +00001929 The ``failfast``, ``catchbreak`` and ``buffer`` parameters have the same
Éric Araujo76338ec2010-11-26 23:46:18 +00001930 effect as the same-name `command-line options`_.
Benjamin Petersonb48af542010-04-11 20:43:16 +00001931
Ezio Melotti60901872010-12-01 00:56:10 +00001932 The *warning* argument specifies the :ref:`warning filter <warning-filter>`
1933 that should be used while running the tests. If it's not specified, it will
1934 remain ``None`` if a :option:`-W` option is passed to :program:`python`,
1935 otherwise it will be set to ``'default'``.
1936
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001937 Calling ``main`` actually returns an instance of the ``TestProgram`` class.
1938 This stores the result of the tests run as the ``result`` attribute.
1939
Georg Brandl853947a2010-01-31 18:53:23 +00001940 .. versionchanged:: 3.2
Ezio Melotti60901872010-12-01 00:56:10 +00001941 The ``exit``, ``verbosity``, ``failfast``, ``catchbreak``, ``buffer``,
1942 and ``warnings`` parameters were added.
Benjamin Petersond2397752009-06-27 23:45:02 +00001943
1944
1945load_tests Protocol
1946###################
1947
Georg Brandl853947a2010-01-31 18:53:23 +00001948.. versionadded:: 3.2
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001949
Benjamin Petersond2397752009-06-27 23:45:02 +00001950Modules or packages can customize how tests are loaded from them during normal
1951test runs or test discovery by implementing a function called ``load_tests``.
1952
1953If a test module defines ``load_tests`` it will be called by
1954:meth:`TestLoader.loadTestsFromModule` with the following arguments::
1955
1956 load_tests(loader, standard_tests, None)
1957
1958It should return a :class:`TestSuite`.
1959
1960*loader* is the instance of :class:`TestLoader` doing the loading.
1961*standard_tests* are the tests that would be loaded by default from the
1962module. It is common for test modules to only want to add or remove tests
1963from the standard set of tests.
1964The third argument is used when loading packages as part of test discovery.
1965
1966A typical ``load_tests`` function that loads tests from a specific set of
1967:class:`TestCase` classes may look like::
1968
1969 test_cases = (TestCase1, TestCase2, TestCase3)
1970
1971 def load_tests(loader, tests, pattern):
1972 suite = TestSuite()
1973 for test_class in test_cases:
1974 tests = loader.loadTestsFromTestCase(test_class)
1975 suite.addTests(tests)
1976 return suite
1977
1978If discovery is started, either from the command line or by calling
1979:meth:`TestLoader.discover`, with a pattern that matches a package
1980name then the package :file:`__init__.py` will be checked for ``load_tests``.
1981
1982.. note::
1983
Ezio Melotti0639d5a2009-12-19 23:26:38 +00001984 The default pattern is 'test*.py'. This matches all Python files
Benjamin Petersond2397752009-06-27 23:45:02 +00001985 that start with 'test' but *won't* match any test directories.
1986
1987 A pattern like 'test*' will match test packages as well as
1988 modules.
1989
1990If the package :file:`__init__.py` defines ``load_tests`` then it will be
1991called and discovery not continued into the package. ``load_tests``
1992is called with the following arguments::
1993
1994 load_tests(loader, standard_tests, pattern)
1995
1996This should return a :class:`TestSuite` representing all the tests
1997from the package. (``standard_tests`` will only contain tests
1998collected from :file:`__init__.py`.)
1999
2000Because the pattern is passed into ``load_tests`` the package is free to
2001continue (and potentially modify) test discovery. A 'do nothing'
2002``load_tests`` function for a test package would look like::
2003
2004 def load_tests(loader, standard_tests, pattern):
2005 # top level directory cached on loader instance
2006 this_dir = os.path.dirname(__file__)
2007 package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
2008 standard_tests.addTests(package_tests)
2009 return standard_tests
Benjamin Petersonb48af542010-04-11 20:43:16 +00002010
2011
2012Class and Module Fixtures
2013-------------------------
2014
2015Class and module level fixtures are implemented in :class:`TestSuite`. When
2016the test suite encounters a test from a new class then :meth:`tearDownClass`
2017from the previous class (if there is one) is called, followed by
2018:meth:`setUpClass` from the new class.
2019
2020Similarly if a test is from a different module from the previous test then
2021``tearDownModule`` from the previous module is run, followed by
2022``setUpModule`` from the new module.
2023
2024After all the tests have run the final ``tearDownClass`` and
2025``tearDownModule`` are run.
2026
2027Note that shared fixtures do not play well with [potential] features like test
2028parallelization and they break test isolation. They should be used with care.
2029
2030The default ordering of tests created by the unittest test loaders is to group
2031all tests from the same modules and classes together. This will lead to
2032``setUpClass`` / ``setUpModule`` (etc) being called exactly once per class and
2033module. If you randomize the order, so that tests from different modules and
2034classes are adjacent to each other, then these shared fixture functions may be
2035called multiple times in a single test run.
2036
2037Shared fixtures are not intended to work with suites with non-standard
2038ordering. A ``BaseTestSuite`` still exists for frameworks that don't want to
2039support shared fixtures.
2040
2041If there are any exceptions raised during one of the shared fixture functions
2042the test is reported as an error. Because there is no corresponding test
2043instance an ``_ErrorHolder`` object (that has the same interface as a
2044:class:`TestCase`) is created to represent the error. If you are just using
2045the standard unittest test runner then this detail doesn't matter, but if you
2046are a framework author it may be relevant.
2047
2048
2049setUpClass and tearDownClass
2050~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2051
2052These must be implemented as class methods::
2053
2054 import unittest
2055
2056 class Test(unittest.TestCase):
2057 @classmethod
2058 def setUpClass(cls):
2059 cls._connection = createExpensiveConnectionObject()
2060
2061 @classmethod
2062 def tearDownClass(cls):
2063 cls._connection.destroy()
2064
2065If you want the ``setUpClass`` and ``tearDownClass`` on base classes called
2066then you must call up to them yourself. The implementations in
2067:class:`TestCase` are empty.
2068
2069If an exception is raised during a ``setUpClass`` then the tests in the class
2070are not run and the ``tearDownClass`` is not run. Skipped classes will not
Michael Foord98b3e762010-06-05 21:59:55 +00002071have ``setUpClass`` or ``tearDownClass`` run. If the exception is a
2072``SkipTest`` exception then the class will be reported as having been skipped
2073instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002074
2075
2076setUpModule and tearDownModule
2077~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2078
2079These should be implemented as functions::
2080
2081 def setUpModule():
2082 createConnection()
2083
2084 def tearDownModule():
2085 closeConnection()
2086
2087If an exception is raised in a ``setUpModule`` then none of the tests in the
Michael Foord98b3e762010-06-05 21:59:55 +00002088module will be run and the ``tearDownModule`` will not be run. If the exception is a
2089``SkipTest`` exception then the module will be reported as having been skipped
2090instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002091
2092
2093Signal Handling
2094---------------
2095
Georg Brandl419e3de2010-12-01 15:44:25 +00002096.. versionadded:: 3.2
2097
Éric Araujo8acb67c2010-11-26 23:31:07 +00002098The :option:`-c/--catch <unittest -c>` command-line option to unittest,
Éric Araujo76338ec2010-11-26 23:46:18 +00002099along with the ``catchbreak`` parameter to :func:`unittest.main()`, provide
2100more friendly handling of control-C during a test run. With catch break
2101behavior enabled control-C will allow the currently running test to complete,
2102and the test run will then end and report all the results so far. A second
2103control-c will raise a :exc:`KeyboardInterrupt` in the usual way.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002104
Michael Foordde4ceab2010-04-25 19:53:49 +00002105The control-c handling signal handler attempts to remain compatible with code or
2106tests that install their own :const:`signal.SIGINT` handler. If the ``unittest``
2107handler is called but *isn't* the installed :const:`signal.SIGINT` handler,
2108i.e. it has been replaced by the system under test and delegated to, then it
2109calls the default handler. This will normally be the expected behavior by code
2110that replaces an installed handler and delegates to it. For individual tests
2111that need ``unittest`` control-c handling disabled the :func:`removeHandler`
2112decorator can be used.
2113
2114There are a few utility functions for framework authors to enable control-c
2115handling functionality within test frameworks.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002116
2117.. function:: installHandler()
2118
2119 Install the control-c handler. When a :const:`signal.SIGINT` is received
2120 (usually in response to the user pressing control-c) all registered results
2121 have :meth:`~TestResult.stop` called.
2122
Michael Foord469b1f02010-04-26 23:41:26 +00002123
Benjamin Petersonb48af542010-04-11 20:43:16 +00002124.. function:: registerResult(result)
2125
2126 Register a :class:`TestResult` object for control-c handling. Registering a
2127 result stores a weak reference to it, so it doesn't prevent the result from
2128 being garbage collected.
2129
Michael Foordde4ceab2010-04-25 19:53:49 +00002130 Registering a :class:`TestResult` object has no side-effects if control-c
2131 handling is not enabled, so test frameworks can unconditionally register
2132 all results they create independently of whether or not handling is enabled.
2133
Michael Foord469b1f02010-04-26 23:41:26 +00002134
Benjamin Petersonb48af542010-04-11 20:43:16 +00002135.. function:: removeResult(result)
2136
2137 Remove a registered result. Once a result has been removed then
2138 :meth:`~TestResult.stop` will no longer be called on that result object in
2139 response to a control-c.
2140
Michael Foord469b1f02010-04-26 23:41:26 +00002141
Michael Foordde4ceab2010-04-25 19:53:49 +00002142.. function:: removeHandler(function=None)
2143
2144 When called without arguments this function removes the control-c handler
2145 if it has been installed. This function can also be used as a test decorator
2146 to temporarily remove the handler whilst the test is being executed::
2147
2148 @unittest.removeHandler
2149 def test_signal_handling(self):
2150 ...