blob: 06f3d831fde3d71d02b2c9754e38edddb3687b88 [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
Michael Foord90efac72011-01-03 15:39:49 +0000100 The script :file:`Tools/unittestgui/unittestgui.py` in the Python source distribution is
101 a GUI tool for test discovery and execution. This is intended largely for ease of use
Senthil Kumaran847c33c2012-10-27 11:04:55 -0700102 for those new to unit testing. For production environments it is
103 recommended that tests be driven by a continuous integration system such as
104 `Buildbot <http://buildbot.net/trac>`_, `Jenkins <http://jenkins-ci.org>`_
105 or `Hudson <http://hudson-ci.org/>`_.
Michael Foord90efac72011-01-03 15:39:49 +0000106
107
Georg Brandl116aa622007-08-15 14:28:22 +0000108.. _unittest-minimal-example:
109
110Basic example
111-------------
112
113The :mod:`unittest` module provides a rich set of tools for constructing and
114running tests. This section demonstrates that a small subset of the tools
115suffice to meet the needs of most users.
116
117Here is a short script to test three functions from the :mod:`random` module::
118
119 import random
120 import unittest
121
122 class TestSequenceFunctions(unittest.TestCase):
123
124 def setUp(self):
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000125 self.seq = list(range(10))
Georg Brandl116aa622007-08-15 14:28:22 +0000126
Benjamin Peterson52baa292009-03-24 00:56:30 +0000127 def test_shuffle(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000128 # make sure the shuffled sequence does not lose any elements
129 random.shuffle(self.seq)
130 self.seq.sort()
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000131 self.assertEqual(self.seq, list(range(10)))
Georg Brandl116aa622007-08-15 14:28:22 +0000132
Benjamin Peterson847a4112010-03-14 15:04:17 +0000133 # should raise an exception for an immutable sequence
134 self.assertRaises(TypeError, random.shuffle, (1,2,3))
135
Benjamin Peterson52baa292009-03-24 00:56:30 +0000136 def test_choice(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000137 element = random.choice(self.seq)
Benjamin Peterson847a4112010-03-14 15:04:17 +0000138 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000139
Benjamin Peterson52baa292009-03-24 00:56:30 +0000140 def test_sample(self):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000141 with self.assertRaises(ValueError):
142 random.sample(self.seq, 20)
Georg Brandl116aa622007-08-15 14:28:22 +0000143 for element in random.sample(self.seq, 5):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000144 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000145
146 if __name__ == '__main__':
147 unittest.main()
148
Benjamin Peterson52baa292009-03-24 00:56:30 +0000149A testcase is created by subclassing :class:`unittest.TestCase`. The three
Georg Brandl116aa622007-08-15 14:28:22 +0000150individual tests are defined with methods whose names start with the letters
151``test``. This naming convention informs the test runner about which methods
152represent tests.
153
Benjamin Peterson52baa292009-03-24 00:56:30 +0000154The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
Michael Foord34c94622010-02-10 15:51:42 +0000155expected result; :meth:`~TestCase.assertTrue` to verify a condition; or
Benjamin Peterson52baa292009-03-24 00:56:30 +0000156:meth:`~TestCase.assertRaises` to verify that an expected exception gets raised.
157These methods are used instead of the :keyword:`assert` statement so the test
158runner can accumulate all test results and produce a report.
Georg Brandl116aa622007-08-15 14:28:22 +0000159
Benjamin Peterson52baa292009-03-24 00:56:30 +0000160When a :meth:`~TestCase.setUp` method is defined, the test runner will run that
161method prior to each test. Likewise, if a :meth:`~TestCase.tearDown` method is
162defined, the test runner will invoke that method after each test. In the
163example, :meth:`~TestCase.setUp` was used to create a fresh sequence for each
164test.
Georg Brandl116aa622007-08-15 14:28:22 +0000165
166The final block shows a simple way to run the tests. :func:`unittest.main`
Éric Araujo713d3032010-11-18 16:38:46 +0000167provides a command-line interface to the test script. When run from the command
Georg Brandl116aa622007-08-15 14:28:22 +0000168line, the above script produces an output that looks like this::
169
170 ...
171 ----------------------------------------------------------------------
172 Ran 3 tests in 0.000s
173
174 OK
175
176Instead of :func:`unittest.main`, there are other ways to run the tests with a
177finer level of control, less terse output, and no requirement to be run from the
178command line. For example, the last two lines may be replaced with::
179
180 suite = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)
181 unittest.TextTestRunner(verbosity=2).run(suite)
182
183Running the revised script from the interpreter or another script produces the
184following output::
185
Ezio Melottid59e44a2010-02-28 03:46:13 +0000186 test_choice (__main__.TestSequenceFunctions) ... ok
187 test_sample (__main__.TestSequenceFunctions) ... ok
188 test_shuffle (__main__.TestSequenceFunctions) ... ok
Georg Brandl116aa622007-08-15 14:28:22 +0000189
190 ----------------------------------------------------------------------
191 Ran 3 tests in 0.110s
192
193 OK
194
195The above examples show the most commonly used :mod:`unittest` features which
196are sufficient to meet many everyday testing needs. The remainder of the
197documentation explores the full feature set from first principles.
198
Benjamin Petersonb48af542010-04-11 20:43:16 +0000199
200.. _unittest-command-line-interface:
201
Éric Araujo76338ec2010-11-26 23:46:18 +0000202Command-Line Interface
Benjamin Petersonb48af542010-04-11 20:43:16 +0000203----------------------
204
205The unittest module can be used from the command line to run tests from
206modules, classes or even individual test methods::
207
208 python -m unittest test_module1 test_module2
209 python -m unittest test_module.TestClass
210 python -m unittest test_module.TestClass.test_method
211
212You can pass in a list with any combination of module names, and fully
213qualified class or method names.
214
Michael Foord37d120a2010-12-04 01:11:21 +0000215Test modules can be specified by file path as well::
216
217 python -m unittest tests/test_something.py
218
219This allows you to use the shell filename completion to specify the test module.
220The file specified must still be importable as a module. The path is converted
221to a module name by removing the '.py' and converting path separators into '.'.
222If you want to execute a test file that isn't importable as a module you should
223execute the file directly instead.
224
Benjamin Petersonb48af542010-04-11 20:43:16 +0000225You can run tests with more detail (higher verbosity) by passing in the -v flag::
226
227 python -m unittest -v test_module
228
Michael Foord086f3082010-11-21 21:28:01 +0000229When executed without arguments :ref:`unittest-test-discovery` is started::
230
231 python -m unittest
232
Éric Araujo713d3032010-11-18 16:38:46 +0000233For a list of all the command-line options::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000234
235 python -m unittest -h
236
Georg Brandl67b21b72010-08-17 15:07:14 +0000237.. versionchanged:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000238 In earlier versions it was only possible to run individual test methods and
239 not modules or classes.
240
241
Éric Araujo76338ec2010-11-26 23:46:18 +0000242Command-line options
243~~~~~~~~~~~~~~~~~~~~
Benjamin Petersonb48af542010-04-11 20:43:16 +0000244
Éric Araujod3309df2010-11-21 03:09:17 +0000245:program:`unittest` supports these command-line options:
Benjamin Petersonb48af542010-04-11 20:43:16 +0000246
Éric Araujo713d3032010-11-18 16:38:46 +0000247.. program:: unittest
Benjamin Petersonb48af542010-04-11 20:43:16 +0000248
Éric Araujo713d3032010-11-18 16:38:46 +0000249.. cmdoption:: -b, --buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +0000250
Éric Araujo713d3032010-11-18 16:38:46 +0000251 The standard output and standard error streams are buffered during the test
252 run. Output during a passing test is discarded. Output is echoed normally
253 on test fail or error and is added to the failure messages.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000254
Éric Araujo713d3032010-11-18 16:38:46 +0000255.. cmdoption:: -c, --catch
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000256
Éric Araujo713d3032010-11-18 16:38:46 +0000257 Control-C during the test run waits for the current test to end and then
258 reports all the results so far. A second control-C raises the normal
259 :exc:`KeyboardInterrupt` exception.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000260
Éric Araujo713d3032010-11-18 16:38:46 +0000261 See `Signal Handling`_ for the functions that provide this functionality.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000262
Éric Araujo713d3032010-11-18 16:38:46 +0000263.. cmdoption:: -f, --failfast
264
265 Stop the test run on the first error or failure.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000266
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000267.. versionadded:: 3.2
Éric Araujod6c5f742010-12-16 00:07:01 +0000268 The command-line options ``-b``, ``-c`` and ``-f`` were added.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000269
270The command line can also be used for test discovery, for running all of the
271tests in a project or just a subset.
272
273
274.. _unittest-test-discovery:
275
276Test Discovery
277--------------
278
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000279.. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000280
Ezio Melotti3d995842011-03-08 16:17:35 +0200281Unittest supports simple test discovery. In order to be compatible with test
282discovery, all of the test files must be :ref:`modules <tut-modules>` or
283:ref:`packages <tut-packages>` importable from the top-level directory of
284the project (this means that their filenames must be valid
285:ref:`identifiers <identifiers>`).
Benjamin Petersonb48af542010-04-11 20:43:16 +0000286
287Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
Éric Araujo713d3032010-11-18 16:38:46 +0000288used from the command line. The basic command-line usage is::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000289
290 cd project_directory
291 python -m unittest discover
292
Michael Foord086f3082010-11-21 21:28:01 +0000293.. note::
294
295 As a shortcut, ``python -m unittest`` is the equivalent of
296 ``python -m unittest discover``. If you want to pass arguments to test
Éric Araujo941afed2011-09-01 02:47:34 +0200297 discovery the ``discover`` sub-command must be used explicitly.
Michael Foord086f3082010-11-21 21:28:01 +0000298
Benjamin Petersonb48af542010-04-11 20:43:16 +0000299The ``discover`` sub-command has the following options:
300
Éric Araujo713d3032010-11-18 16:38:46 +0000301.. program:: unittest discover
302
303.. cmdoption:: -v, --verbose
304
305 Verbose output
306
307.. cmdoption:: -s directory
308
Éric Araujo941afed2011-09-01 02:47:34 +0200309 Directory to start discovery (``.`` default)
Éric Araujo713d3032010-11-18 16:38:46 +0000310
311.. cmdoption:: -p pattern
312
Éric Araujo941afed2011-09-01 02:47:34 +0200313 Pattern to match test files (``test*.py`` default)
Éric Araujo713d3032010-11-18 16:38:46 +0000314
315.. cmdoption:: -t directory
316
317 Top level directory of project (defaults to start directory)
Benjamin Petersonb48af542010-04-11 20:43:16 +0000318
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000319The :option:`-s`, :option:`-p`, and :option:`-t` options can be passed in
320as positional arguments in that order. The following two command lines
321are equivalent::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000322
323 python -m unittest discover -s project_directory -p '*_test.py'
324 python -m unittest discover project_directory '*_test.py'
325
Michael Foord16f3e902010-05-08 15:13:42 +0000326As well as being a path it is possible to pass a package name, for example
327``myproject.subpackage.test``, as the start directory. The package name you
328supply will then be imported and its location on the filesystem will be used
329as the start directory.
330
331.. caution::
332
Senthil Kumaran916bd382010-10-15 12:55:19 +0000333 Test discovery loads tests by importing them. Once test discovery has found
334 all the test files from the start directory you specify it turns the paths
335 into package names to import. For example :file:`foo/bar/baz.py` will be
Michael Foord16f3e902010-05-08 15:13:42 +0000336 imported as ``foo.bar.baz``.
337
338 If you have a package installed globally and attempt test discovery on
339 a different copy of the package then the import *could* happen from the
340 wrong place. If this happens test discovery will warn you and exit.
341
342 If you supply the start directory as a package name rather than a
343 path to a directory then discover assumes that whichever location it
344 imports from is the location you intended, so you will not get the
345 warning.
346
Benjamin Petersonb48af542010-04-11 20:43:16 +0000347Test modules and packages can customize test loading and discovery by through
348the `load_tests protocol`_.
349
350
Georg Brandl116aa622007-08-15 14:28:22 +0000351.. _organizing-tests:
352
353Organizing test code
354--------------------
355
356The basic building blocks of unit testing are :dfn:`test cases` --- single
357scenarios that must be set up and checked for correctness. In :mod:`unittest`,
Raymond Hettinger833ad0e2011-02-06 21:00:38 +0000358test cases are represented by :class:`unittest.TestCase` instances.
359To make your own test cases you must write subclasses of
360:class:`TestCase` or use :class:`FunctionTestCase`.
Georg Brandl116aa622007-08-15 14:28:22 +0000361
362An instance of a :class:`TestCase`\ -derived class is an object that can
363completely run a single test method, together with optional set-up and tidy-up
364code.
365
366The testing code of a :class:`TestCase` instance should be entirely self
367contained, such that it can be run either in isolation or in arbitrary
368combination with any number of other test cases.
369
Benjamin Peterson52baa292009-03-24 00:56:30 +0000370The simplest :class:`TestCase` subclass will simply override the
371:meth:`~TestCase.runTest` method in order to perform specific testing code::
Georg Brandl116aa622007-08-15 14:28:22 +0000372
373 import unittest
374
375 class DefaultWidgetSizeTestCase(unittest.TestCase):
376 def runTest(self):
377 widget = Widget('The widget')
378 self.assertEqual(widget.size(), (50, 50), 'incorrect default size')
379
Sandro Tosi41b24042012-01-21 10:59:37 +0100380Note that in order to test something, we use one of the :meth:`assert\*`
Benjamin Petersond2397752009-06-27 23:45:02 +0000381methods provided by the :class:`TestCase` base class. If the test fails, an
382exception will be raised, and :mod:`unittest` will identify the test case as a
383:dfn:`failure`. Any other exceptions will be treated as :dfn:`errors`. This
384helps you identify where the problem is: :dfn:`failures` are caused by incorrect
385results - a 5 where you expected a 6. :dfn:`Errors` are caused by incorrect
386code - e.g., a :exc:`TypeError` caused by an incorrect function call.
Georg Brandl116aa622007-08-15 14:28:22 +0000387
388The way to run a test case will be described later. For now, note that to
389construct an instance of such a test case, we call its constructor without
390arguments::
391
392 testCase = DefaultWidgetSizeTestCase()
393
394Now, such test cases can be numerous, and their set-up can be repetitive. In
395the above case, constructing a :class:`Widget` in each of 100 Widget test case
396subclasses would mean unsightly duplication.
397
398Luckily, we can factor out such set-up code by implementing a method called
Benjamin Peterson52baa292009-03-24 00:56:30 +0000399:meth:`~TestCase.setUp`, which the testing framework will automatically call for
400us when we run the test::
Georg Brandl116aa622007-08-15 14:28:22 +0000401
402 import unittest
403
404 class SimpleWidgetTestCase(unittest.TestCase):
405 def setUp(self):
406 self.widget = Widget('The widget')
407
408 class DefaultWidgetSizeTestCase(SimpleWidgetTestCase):
409 def runTest(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000410 self.assertEqual(self.widget.size(), (50,50),
411 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000412
413 class WidgetResizeTestCase(SimpleWidgetTestCase):
414 def runTest(self):
415 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000416 self.assertEqual(self.widget.size(), (100,150),
417 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000418
Benjamin Peterson52baa292009-03-24 00:56:30 +0000419If the :meth:`~TestCase.setUp` method raises an exception while the test is
420running, the framework will consider the test to have suffered an error, and the
421:meth:`~TestCase.runTest` method will not be executed.
Georg Brandl116aa622007-08-15 14:28:22 +0000422
Benjamin Peterson52baa292009-03-24 00:56:30 +0000423Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
424after the :meth:`~TestCase.runTest` method has been run::
Georg Brandl116aa622007-08-15 14:28:22 +0000425
426 import unittest
427
428 class SimpleWidgetTestCase(unittest.TestCase):
429 def setUp(self):
430 self.widget = Widget('The widget')
431
432 def tearDown(self):
433 self.widget.dispose()
434 self.widget = None
435
Benjamin Peterson52baa292009-03-24 00:56:30 +0000436If :meth:`~TestCase.setUp` succeeded, the :meth:`~TestCase.tearDown` method will
437be run whether :meth:`~TestCase.runTest` succeeded or not.
Georg Brandl116aa622007-08-15 14:28:22 +0000438
439Such a working environment for the testing code is called a :dfn:`fixture`.
440
441Often, many small test cases will use the same fixture. In this case, we would
442end up subclassing :class:`SimpleWidgetTestCase` into many small one-method
443classes such as :class:`DefaultWidgetSizeTestCase`. This is time-consuming and
Georg Brandl116aa622007-08-15 14:28:22 +0000444discouraging, so in the same vein as JUnit, :mod:`unittest` provides a simpler
445mechanism::
446
447 import unittest
448
449 class WidgetTestCase(unittest.TestCase):
450 def setUp(self):
451 self.widget = Widget('The widget')
452
453 def tearDown(self):
454 self.widget.dispose()
455 self.widget = None
456
Ezio Melottid59e44a2010-02-28 03:46:13 +0000457 def test_default_size(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000458 self.assertEqual(self.widget.size(), (50,50),
459 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000460
Ezio Melottid59e44a2010-02-28 03:46:13 +0000461 def test_resize(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000462 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000463 self.assertEqual(self.widget.size(), (100,150),
464 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000465
Benjamin Peterson52baa292009-03-24 00:56:30 +0000466Here we have not provided a :meth:`~TestCase.runTest` method, but have instead
467provided two different test methods. Class instances will now each run one of
Ezio Melottid59e44a2010-02-28 03:46:13 +0000468the :meth:`test_\*` methods, with ``self.widget`` created and destroyed
Benjamin Peterson52baa292009-03-24 00:56:30 +0000469separately for each instance. When creating an instance we must specify the
470test method it is to run. We do this by passing the method name in the
471constructor::
Georg Brandl116aa622007-08-15 14:28:22 +0000472
Ezio Melottid59e44a2010-02-28 03:46:13 +0000473 defaultSizeTestCase = WidgetTestCase('test_default_size')
474 resizeTestCase = WidgetTestCase('test_resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000475
476Test case instances are grouped together according to the features they test.
477:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
478represented by :mod:`unittest`'s :class:`TestSuite` class::
479
480 widgetTestSuite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000481 widgetTestSuite.addTest(WidgetTestCase('test_default_size'))
482 widgetTestSuite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000483
484For the ease of running tests, as we will see later, it is a good idea to
485provide in each test module a callable object that returns a pre-built test
486suite::
487
488 def suite():
489 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000490 suite.addTest(WidgetTestCase('test_default_size'))
491 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000492 return suite
493
494or even::
495
496 def suite():
Ezio Melottid59e44a2010-02-28 03:46:13 +0000497 tests = ['test_default_size', 'test_resize']
Georg Brandl116aa622007-08-15 14:28:22 +0000498
499 return unittest.TestSuite(map(WidgetTestCase, tests))
500
501Since it is a common pattern to create a :class:`TestCase` subclass with many
502similarly named test functions, :mod:`unittest` provides a :class:`TestLoader`
503class that can be used to automate the process of creating a test suite and
504populating it with individual tests. For example, ::
505
506 suite = unittest.TestLoader().loadTestsFromTestCase(WidgetTestCase)
507
Ezio Melottid59e44a2010-02-28 03:46:13 +0000508will create a test suite that will run ``WidgetTestCase.test_default_size()`` and
509``WidgetTestCase.test_resize``. :class:`TestLoader` uses the ``'test'`` method
Georg Brandl116aa622007-08-15 14:28:22 +0000510name prefix to identify test methods automatically.
511
Mark Dickinsonc48d8342009-02-01 14:18:10 +0000512Note that the order in which the various test cases will be run is
513determined by sorting the test function names with respect to the
514built-in ordering for strings.
Georg Brandl116aa622007-08-15 14:28:22 +0000515
516Often it is desirable to group suites of test cases together, so as to run tests
517for the whole system at once. This is easy, since :class:`TestSuite` instances
518can be added to a :class:`TestSuite` just as :class:`TestCase` instances can be
519added to a :class:`TestSuite`::
520
521 suite1 = module1.TheTestSuite()
522 suite2 = module2.TheTestSuite()
523 alltests = unittest.TestSuite([suite1, suite2])
524
525You can place the definitions of test cases and test suites in the same modules
526as the code they are to test (such as :file:`widget.py`), but there are several
527advantages to placing the test code in a separate module, such as
528:file:`test_widget.py`:
529
530* The test module can be run standalone from the command line.
531
532* The test code can more easily be separated from shipped code.
533
534* There is less temptation to change test code to fit the code it tests without
535 a good reason.
536
537* Test code should be modified much less frequently than the code it tests.
538
539* Tested code can be refactored more easily.
540
541* Tests for modules written in C must be in separate modules anyway, so why not
542 be consistent?
543
544* If the testing strategy changes, there is no need to change the source code.
545
546
547.. _legacy-unit-tests:
548
549Re-using old test code
550----------------------
551
552Some users will find that they have existing test code that they would like to
553run from :mod:`unittest`, without converting every old test function to a
554:class:`TestCase` subclass.
555
556For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
557This subclass of :class:`TestCase` can be used to wrap an existing test
558function. Set-up and tear-down functions can also be provided.
559
560Given the following test function::
561
562 def testSomething():
563 something = makeSomething()
564 assert something.name is not None
565 # ...
566
567one can create an equivalent test case instance as follows::
568
569 testcase = unittest.FunctionTestCase(testSomething)
570
571If there are additional set-up and tear-down methods that should be called as
572part of the test case's operation, they can also be provided like so::
573
574 testcase = unittest.FunctionTestCase(testSomething,
575 setUp=makeSomethingDB,
576 tearDown=deleteSomethingDB)
577
578To make migrating existing test suites easier, :mod:`unittest` supports tests
579raising :exc:`AssertionError` to indicate test failure. However, it is
580recommended that you use the explicit :meth:`TestCase.fail\*` and
581:meth:`TestCase.assert\*` methods instead, as future versions of :mod:`unittest`
582may treat :exc:`AssertionError` differently.
583
584.. note::
585
Benjamin Petersond2397752009-06-27 23:45:02 +0000586 Even though :class:`FunctionTestCase` can be used to quickly convert an
587 existing test base over to a :mod:`unittest`\ -based system, this approach is
588 not recommended. Taking the time to set up proper :class:`TestCase`
589 subclasses will make future test refactorings infinitely easier.
Georg Brandl116aa622007-08-15 14:28:22 +0000590
Benjamin Peterson52baa292009-03-24 00:56:30 +0000591In some cases, the existing tests may have been written using the :mod:`doctest`
592module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
593automatically build :class:`unittest.TestSuite` instances from the existing
594:mod:`doctest`\ -based tests.
595
Georg Brandl116aa622007-08-15 14:28:22 +0000596
Benjamin Peterson5254c042009-03-23 22:25:03 +0000597.. _unittest-skipping:
598
599Skipping tests and expected failures
600------------------------------------
601
Michael Foordf5c851a2010-02-05 21:48:03 +0000602.. versionadded:: 3.1
603
Benjamin Peterson5254c042009-03-23 22:25:03 +0000604Unittest supports skipping individual test methods and even whole classes of
605tests. In addition, it supports marking a test as a "expected failure," a test
606that is broken and will fail, but shouldn't be counted as a failure on a
607:class:`TestResult`.
608
609Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
610or one of its conditional variants.
611
612Basic skipping looks like this: ::
613
614 class MyTestCase(unittest.TestCase):
615
616 @unittest.skip("demonstrating skipping")
617 def test_nothing(self):
618 self.fail("shouldn't happen")
619
Benjamin Petersond2397752009-06-27 23:45:02 +0000620 @unittest.skipIf(mylib.__version__ < (1, 3),
621 "not supported in this library version")
Benjamin Petersonded31c42009-03-30 15:04:16 +0000622 def test_format(self):
623 # Tests that work for only a certain version of the library.
624 pass
625
626 @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
627 def test_windows_support(self):
628 # windows specific testing code
629 pass
630
Benjamin Peterson5254c042009-03-23 22:25:03 +0000631This is the output of running the example above in verbose mode: ::
632
Benjamin Petersonded31c42009-03-30 15:04:16 +0000633 test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000634 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
Benjamin Petersonded31c42009-03-30 15:04:16 +0000635 test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000636
637 ----------------------------------------------------------------------
Benjamin Petersonded31c42009-03-30 15:04:16 +0000638 Ran 3 tests in 0.005s
639
640 OK (skipped=3)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000641
642Classes can be skipped just like methods: ::
643
Sandro Tosi317075d2012-03-31 18:34:59 +0200644 @unittest.skip("showing class skipping")
Benjamin Peterson5254c042009-03-23 22:25:03 +0000645 class MySkippedTestCase(unittest.TestCase):
646 def test_not_run(self):
647 pass
648
Benjamin Peterson52baa292009-03-24 00:56:30 +0000649:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
650that needs to be set up is not available.
651
Benjamin Peterson5254c042009-03-23 22:25:03 +0000652Expected failures use the :func:`expectedFailure` decorator. ::
653
654 class ExpectedFailureTestCase(unittest.TestCase):
655 @unittest.expectedFailure
656 def test_fail(self):
657 self.assertEqual(1, 0, "broken")
658
659It's easy to roll your own skipping decorators by making a decorator that calls
660:func:`skip` on the test when it wants it to be skipped. This decorator skips
661the test unless the passed object has a certain attribute: ::
662
663 def skipUnlessHasattr(obj, attr):
664 if hasattr(obj, attr):
665 return lambda func: func
666 return unittest.skip("{0!r} doesn't have {1!r}".format(obj, attr))
667
668The following decorators implement test skipping and expected failures:
669
Georg Brandl8a1caa22010-07-29 16:01:11 +0000670.. decorator:: skip(reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000671
672 Unconditionally skip the decorated test. *reason* should describe why the
673 test is being skipped.
674
Georg Brandl8a1caa22010-07-29 16:01:11 +0000675.. decorator:: skipIf(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000676
677 Skip the decorated test if *condition* is true.
678
Georg Brandl8a1caa22010-07-29 16:01:11 +0000679.. decorator:: skipUnless(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000680
Georg Brandl6faee4e2010-09-21 14:48:28 +0000681 Skip the decorated test unless *condition* is true.
Benjamin Peterson5254c042009-03-23 22:25:03 +0000682
Georg Brandl8a1caa22010-07-29 16:01:11 +0000683.. decorator:: expectedFailure
Benjamin Peterson5254c042009-03-23 22:25:03 +0000684
685 Mark the test as an expected failure. If the test fails when run, the test
686 is not counted as a failure.
687
Benjamin Petersonb48af542010-04-11 20:43:16 +0000688Skipped tests will not have :meth:`setUp` or :meth:`tearDown` run around them.
689Skipped classes will not have :meth:`setUpClass` or :meth:`tearDownClass` run.
690
Benjamin Peterson5254c042009-03-23 22:25:03 +0000691
Georg Brandl116aa622007-08-15 14:28:22 +0000692.. _unittest-contents:
693
694Classes and functions
695---------------------
696
Benjamin Peterson52baa292009-03-24 00:56:30 +0000697This section describes in depth the API of :mod:`unittest`.
698
699
700.. _testcase-objects:
701
702Test cases
703~~~~~~~~~~
Georg Brandl116aa622007-08-15 14:28:22 +0000704
Georg Brandl7f01a132009-09-16 15:58:14 +0000705.. class:: TestCase(methodName='runTest')
Georg Brandl116aa622007-08-15 14:28:22 +0000706
707 Instances of the :class:`TestCase` class represent the smallest testable units
708 in the :mod:`unittest` universe. This class is intended to be used as a base
709 class, with specific tests being implemented by concrete subclasses. This class
710 implements the interface needed by the test runner to allow it to drive the
711 test, and methods that the test code can use to check for and report various
712 kinds of failure.
713
714 Each instance of :class:`TestCase` will run a single test method: the method
715 named *methodName*. If you remember, we had an earlier example that went
716 something like this::
717
718 def suite():
719 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000720 suite.addTest(WidgetTestCase('test_default_size'))
721 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000722 return suite
723
724 Here, we create two instances of :class:`WidgetTestCase`, each of which runs a
725 single test.
726
Michael Foord1341bb02011-03-14 19:01:46 -0400727 .. versionchanged:: 3.2
Éric Araujo941afed2011-09-01 02:47:34 +0200728 :class:`TestCase` can be instantiated successfully without providing a method
729 name. This makes it easier to experiment with :class:`TestCase` from the
Michael Foord32e1d832011-01-03 17:00:11 +0000730 interactive interpreter.
731
Benjamin Peterson52baa292009-03-24 00:56:30 +0000732 *methodName* defaults to :meth:`runTest`.
733
734 :class:`TestCase` instances provide three groups of methods: one group used
735 to run the test, another used by the test implementation to check conditions
736 and report failures, and some inquiry methods allowing information about the
737 test itself to be gathered.
738
739 Methods in the first group (running the test) are:
740
741
742 .. method:: setUp()
743
744 Method called to prepare the test fixture. This is called immediately
745 before calling the test method; any exception raised by this method will
746 be considered an error rather than a test failure. The default
747 implementation does nothing.
748
749
750 .. method:: tearDown()
751
752 Method called immediately after the test method has been called and the
753 result recorded. This is called even if the test method raised an
754 exception, so the implementation in subclasses may need to be particularly
755 careful about checking internal state. Any exception raised by this
756 method will be considered an error rather than a test failure. This
757 method will only be called if the :meth:`setUp` succeeds, regardless of
758 the outcome of the test method. The default implementation does nothing.
759
760
Benjamin Petersonb48af542010-04-11 20:43:16 +0000761 .. method:: setUpClass()
762
763 A class method called before tests in an individual class run.
764 ``setUpClass`` is called with the class as the only argument
765 and must be decorated as a :func:`classmethod`::
766
767 @classmethod
768 def setUpClass(cls):
769 ...
770
771 See `Class and Module Fixtures`_ for more details.
772
773 .. versionadded:: 3.2
774
775
776 .. method:: tearDownClass()
777
778 A class method called after tests in an individual class have run.
779 ``tearDownClass`` is called with the class as the only argument
780 and must be decorated as a :meth:`classmethod`::
781
782 @classmethod
783 def tearDownClass(cls):
784 ...
785
786 See `Class and Module Fixtures`_ for more details.
787
788 .. versionadded:: 3.2
789
790
Georg Brandl7f01a132009-09-16 15:58:14 +0000791 .. method:: run(result=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000792
793 Run the test, collecting the result into the test result object passed as
Ezio Melotti75b2a5e2010-11-20 10:13:45 +0000794 *result*. If *result* is omitted or ``None``, a temporary result
Alexandre Vassalotti260484d2009-07-17 11:43:26 +0000795 object is created (by calling the :meth:`defaultTestResult` method) and
Michael Foord1341bb02011-03-14 19:01:46 -0400796 used. The result object is returned to :meth:`run`'s caller.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000797
798 The same effect may be had by simply calling the :class:`TestCase`
799 instance.
800
Michael Foord1341bb02011-03-14 19:01:46 -0400801 .. versionchanged:: 3.3
802 Previous versions of ``run`` did not return the result. Neither did
803 calling an instance.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000804
Benjamin Petersone549ead2009-03-28 21:42:05 +0000805 .. method:: skipTest(reason)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000806
Stefan Kraha5bf3f52010-05-19 16:09:41 +0000807 Calling this during a test method or :meth:`setUp` skips the current
Benjamin Peterson52baa292009-03-24 00:56:30 +0000808 test. See :ref:`unittest-skipping` for more information.
809
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000810 .. versionadded:: 3.1
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000811
Benjamin Peterson52baa292009-03-24 00:56:30 +0000812
813 .. method:: debug()
814
815 Run the test without collecting the result. This allows exceptions raised
816 by the test to be propagated to the caller, and can be used to support
817 running tests under a debugger.
818
Ezio Melotti22170ed2010-11-20 09:57:27 +0000819 .. _assert-methods:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000820
Ezio Melotti4370b302010-11-03 20:39:14 +0000821 The :class:`TestCase` class provides a number of methods to check for and
822 report failures, such as:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000823
Ezio Melotti4370b302010-11-03 20:39:14 +0000824 +-----------------------------------------+-----------------------------+---------------+
825 | Method | Checks that | New in |
826 +=========================================+=============================+===============+
827 | :meth:`assertEqual(a, b) | ``a == b`` | |
828 | <TestCase.assertEqual>` | | |
829 +-----------------------------------------+-----------------------------+---------------+
830 | :meth:`assertNotEqual(a, b) | ``a != b`` | |
831 | <TestCase.assertNotEqual>` | | |
832 +-----------------------------------------+-----------------------------+---------------+
833 | :meth:`assertTrue(x) | ``bool(x) is True`` | |
834 | <TestCase.assertTrue>` | | |
835 +-----------------------------------------+-----------------------------+---------------+
836 | :meth:`assertFalse(x) | ``bool(x) is False`` | |
837 | <TestCase.assertFalse>` | | |
838 +-----------------------------------------+-----------------------------+---------------+
839 | :meth:`assertIs(a, b) | ``a is b`` | 3.1 |
840 | <TestCase.assertIs>` | | |
841 +-----------------------------------------+-----------------------------+---------------+
842 | :meth:`assertIsNot(a, b) | ``a is not b`` | 3.1 |
843 | <TestCase.assertIsNot>` | | |
844 +-----------------------------------------+-----------------------------+---------------+
845 | :meth:`assertIsNone(x) | ``x is None`` | 3.1 |
846 | <TestCase.assertIsNone>` | | |
847 +-----------------------------------------+-----------------------------+---------------+
848 | :meth:`assertIsNotNone(x) | ``x is not None`` | 3.1 |
849 | <TestCase.assertIsNotNone>` | | |
850 +-----------------------------------------+-----------------------------+---------------+
851 | :meth:`assertIn(a, b) | ``a in b`` | 3.1 |
852 | <TestCase.assertIn>` | | |
853 +-----------------------------------------+-----------------------------+---------------+
854 | :meth:`assertNotIn(a, b) | ``a not in b`` | 3.1 |
855 | <TestCase.assertNotIn>` | | |
856 +-----------------------------------------+-----------------------------+---------------+
857 | :meth:`assertIsInstance(a, b) | ``isinstance(a, b)`` | 3.2 |
858 | <TestCase.assertIsInstance>` | | |
859 +-----------------------------------------+-----------------------------+---------------+
860 | :meth:`assertNotIsInstance(a, b) | ``not isinstance(a, b)`` | 3.2 |
861 | <TestCase.assertNotIsInstance>` | | |
862 +-----------------------------------------+-----------------------------+---------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000863
Ezio Melottib4dc2502011-05-06 15:01:41 +0300864 All the assert methods accept a *msg* argument that, if specified, is used
865 as the error message on failure (see also :data:`longMessage`).
866 Note that the *msg* keyword argument can be passed to :meth:`assertRaises`,
867 :meth:`assertRaisesRegex`, :meth:`assertWarns`, :meth:`assertWarnsRegex`
868 only when they are used as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000869
Michael Foorde180d392011-01-28 19:51:48 +0000870 .. method:: assertEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000871
Michael Foorde180d392011-01-28 19:51:48 +0000872 Test that *first* and *second* are equal. If the values do not
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000873 compare equal, the test will fail.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000874
Michael Foorde180d392011-01-28 19:51:48 +0000875 In addition, if *first* and *second* are the exact same type and one of
Michael Foord02834952010-02-08 23:10:39 +0000876 list, tuple, dict, set, frozenset or str or any type that a subclass
Ezio Melotti9ecb6be2012-01-16 08:28:54 +0200877 registers with :meth:`addTypeEqualityFunc` the type-specific equality
Michael Foord02834952010-02-08 23:10:39 +0000878 function will be called in order to generate a more useful default
Ezio Melotti22170ed2010-11-20 09:57:27 +0000879 error message (see also the :ref:`list of type-specific methods
880 <type-specific-methods>`).
Ezio Melotti4841fd62010-11-05 15:43:40 +0000881
Raymond Hettinger35a88362009-04-09 00:08:24 +0000882 .. versionchanged:: 3.1
Ezio Melotti9ecb6be2012-01-16 08:28:54 +0200883 Added the automatic calling of type-specific equality function.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000884
Michael Foord28a817e2010-02-09 00:03:57 +0000885 .. versionchanged:: 3.2
886 :meth:`assertMultiLineEqual` added as the default type equality
887 function for comparing strings.
Michael Foord02834952010-02-08 23:10:39 +0000888
Benjamin Peterson52baa292009-03-24 00:56:30 +0000889
Michael Foorde180d392011-01-28 19:51:48 +0000890 .. method:: assertNotEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000891
Michael Foorde180d392011-01-28 19:51:48 +0000892 Test that *first* and *second* are not equal. If the values do
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000893 compare equal, the test will fail.
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000894
Ezio Melotti4370b302010-11-03 20:39:14 +0000895 .. method:: assertTrue(expr, msg=None)
Ezio Melotti4841fd62010-11-05 15:43:40 +0000896 assertFalse(expr, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000897
Ezio Melotti4841fd62010-11-05 15:43:40 +0000898 Test that *expr* is true (or false).
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000899
Ezio Melotti4841fd62010-11-05 15:43:40 +0000900 Note that this is equivalent to ``bool(expr) is True`` and not to ``expr
901 is True`` (use ``assertIs(expr, True)`` for the latter). This method
902 should also be avoided when more specific methods are available (e.g.
903 ``assertEqual(a, b)`` instead of ``assertTrue(a == b)``), because they
904 provide a better error message in case of failure.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000905
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000906
Michael Foorde180d392011-01-28 19:51:48 +0000907 .. method:: assertIs(first, second, msg=None)
908 assertIsNot(first, second, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000909
Michael Foorde180d392011-01-28 19:51:48 +0000910 Test that *first* and *second* evaluate (or don't evaluate) to the
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000911 same object.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000912
Raymond Hettinger35a88362009-04-09 00:08:24 +0000913 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000914
915
Ezio Melotti4370b302010-11-03 20:39:14 +0000916 .. method:: assertIsNone(expr, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000917 assertIsNotNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000918
Ezio Melotti9794a262010-11-04 14:52:13 +0000919 Test that *expr* is (or is not) None.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000920
Ezio Melotti4370b302010-11-03 20:39:14 +0000921 .. versionadded:: 3.1
Benjamin Petersonb48af542010-04-11 20:43:16 +0000922
923
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000924 .. method:: assertIn(first, second, msg=None)
925 assertNotIn(first, second, msg=None)
926
Ezio Melotti4841fd62010-11-05 15:43:40 +0000927 Test that *first* is (or is not) in *second*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000928
Raymond Hettinger35a88362009-04-09 00:08:24 +0000929 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000930
931
Ezio Melotti9c02c2f2010-11-03 20:45:31 +0000932 .. method:: assertIsInstance(obj, cls, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000933 assertNotIsInstance(obj, cls, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000934
Ezio Melotti9794a262010-11-04 14:52:13 +0000935 Test that *obj* is (or is not) an instance of *cls* (which can be a
936 class or a tuple of classes, as supported by :func:`isinstance`).
Ezio Melotti80a61e82011-12-19 07:04:48 +0200937 To check for the exact type, use :func:`assertIs(type(obj), cls) <assertIs>`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000938
Ezio Melotti4370b302010-11-03 20:39:14 +0000939 .. versionadded:: 3.2
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000940
941
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000942
Ezio Melotti4370b302010-11-03 20:39:14 +0000943 It is also possible to check that exceptions and warnings are raised using
944 the following methods:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000945
Ezio Melotti4370b302010-11-03 20:39:14 +0000946 +---------------------------------------------------------+--------------------------------------+------------+
947 | Method | Checks that | New in |
948 +=========================================================+======================================+============+
Éric Araujo941afed2011-09-01 02:47:34 +0200949 | :meth:`assertRaises(exc, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *exc* | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000950 | <TestCase.assertRaises>` | | |
951 +---------------------------------------------------------+--------------------------------------+------------+
Éric Araujo941afed2011-09-01 02:47:34 +0200952 | :meth:`assertRaisesRegex(exc, re, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *exc* | 3.1 |
953 | <TestCase.assertRaisesRegex>` | and the message matches *re* | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000954 +---------------------------------------------------------+--------------------------------------+------------+
Éric Araujo941afed2011-09-01 02:47:34 +0200955 | :meth:`assertWarns(warn, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *warn* | 3.2 |
Ezio Melotti4370b302010-11-03 20:39:14 +0000956 | <TestCase.assertWarns>` | | |
957 +---------------------------------------------------------+--------------------------------------+------------+
Éric Araujo941afed2011-09-01 02:47:34 +0200958 | :meth:`assertWarnsRegex(warn, re, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises *warn* | 3.2 |
959 | <TestCase.assertWarnsRegex>` | and the message matches *re* | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000960 +---------------------------------------------------------+--------------------------------------+------------+
Benjamin Peterson52baa292009-03-24 00:56:30 +0000961
Georg Brandl7f01a132009-09-16 15:58:14 +0000962 .. method:: assertRaises(exception, callable, *args, **kwds)
Ezio Melottib4dc2502011-05-06 15:01:41 +0300963 assertRaises(exception, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000964
965 Test that an exception is raised when *callable* is called with any
966 positional or keyword arguments that are also passed to
967 :meth:`assertRaises`. The test passes if *exception* is raised, is an
968 error if another exception is raised, or fails if no exception is raised.
969 To catch any of a group of exceptions, a tuple containing the exception
970 classes may be passed as *exception*.
971
Ezio Melottib4dc2502011-05-06 15:01:41 +0300972 If only the *exception* and possibly the *msg* arguments are given,
973 return a context manager so that the code under test can be written
974 inline rather than as a function::
Benjamin Petersonded31c42009-03-30 15:04:16 +0000975
Michael Foord41531f22010-02-05 21:13:40 +0000976 with self.assertRaises(SomeException):
Benjamin Petersonded31c42009-03-30 15:04:16 +0000977 do_something()
978
Ezio Melottib4dc2502011-05-06 15:01:41 +0300979 When used as a context manager, :meth:`assertRaises` accepts the
980 additional keyword argument *msg*.
981
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000982 The context manager will store the caught exception object in its
Ezio Melotti49008232010-02-08 21:57:48 +0000983 :attr:`exception` attribute. This can be useful if the intention
Michael Foord41531f22010-02-05 21:13:40 +0000984 is to perform additional checks on the exception raised::
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000985
Georg Brandl8a1caa22010-07-29 16:01:11 +0000986 with self.assertRaises(SomeException) as cm:
987 do_something()
Michael Foord41531f22010-02-05 21:13:40 +0000988
Georg Brandl8a1caa22010-07-29 16:01:11 +0000989 the_exception = cm.exception
990 self.assertEqual(the_exception.error_code, 3)
Michael Foord41531f22010-02-05 21:13:40 +0000991
Ezio Melotti49008232010-02-08 21:57:48 +0000992 .. versionchanged:: 3.1
Benjamin Petersonded31c42009-03-30 15:04:16 +0000993 Added the ability to use :meth:`assertRaises` as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000994
Ezio Melotti49008232010-02-08 21:57:48 +0000995 .. versionchanged:: 3.2
996 Added the :attr:`exception` attribute.
997
Ezio Melottib4dc2502011-05-06 15:01:41 +0300998 .. versionchanged:: 3.3
999 Added the *msg* keyword argument when used as a context manager.
1000
Benjamin Peterson52baa292009-03-24 00:56:30 +00001001
Ezio Melottied3a7d22010-12-01 02:32:32 +00001002 .. method:: assertRaisesRegex(exception, regex, callable, *args, **kwds)
Ezio Melottib4dc2502011-05-06 15:01:41 +03001003 assertRaisesRegex(exception, regex, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001004
Ezio Melottied3a7d22010-12-01 02:32:32 +00001005 Like :meth:`assertRaises` but also tests that *regex* matches
1006 on the string representation of the raised exception. *regex* may be
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001007 a regular expression object or a string containing a regular expression
1008 suitable for use by :func:`re.search`. Examples::
1009
Ezio Melottied3a7d22010-12-01 02:32:32 +00001010 self.assertRaisesRegex(ValueError, 'invalid literal for.*XYZ$',
1011 int, 'XYZ')
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001012
1013 or::
1014
Ezio Melottied3a7d22010-12-01 02:32:32 +00001015 with self.assertRaisesRegex(ValueError, 'literal'):
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001016 int('XYZ')
1017
Georg Brandl419e3de2010-12-01 15:44:25 +00001018 .. versionadded:: 3.1
1019 under the name ``assertRaisesRegexp``.
Ezio Melottib4dc2502011-05-06 15:01:41 +03001020
Ezio Melottied3a7d22010-12-01 02:32:32 +00001021 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001022 Renamed to :meth:`assertRaisesRegex`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001023
Ezio Melottib4dc2502011-05-06 15:01:41 +03001024 .. versionchanged:: 3.3
1025 Added the *msg* keyword argument when used as a context manager.
1026
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001027
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001028 .. method:: assertWarns(warning, callable, *args, **kwds)
Ezio Melottib4dc2502011-05-06 15:01:41 +03001029 assertWarns(warning, msg=None)
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001030
1031 Test that a warning is triggered when *callable* is called with any
1032 positional or keyword arguments that are also passed to
1033 :meth:`assertWarns`. The test passes if *warning* is triggered and
1034 fails if it isn't. Also, any unexpected exception is an error.
1035 To catch any of a group of warnings, a tuple containing the warning
1036 classes may be passed as *warnings*.
1037
Ezio Melottib4dc2502011-05-06 15:01:41 +03001038 If only the *warning* and possibly the *msg* arguments are given,
1039 returns a context manager so that the code under test can be written
1040 inline rather than as a function::
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001041
1042 with self.assertWarns(SomeWarning):
1043 do_something()
1044
Ezio Melottib4dc2502011-05-06 15:01:41 +03001045 When used as a context manager, :meth:`assertRaises` accepts the
1046 additional keyword argument *msg*.
1047
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001048 The context manager will store the caught warning object in its
1049 :attr:`warning` attribute, and the source line which triggered the
1050 warnings in the :attr:`filename` and :attr:`lineno` attributes.
1051 This can be useful if the intention is to perform additional checks
1052 on the exception raised::
1053
1054 with self.assertWarns(SomeWarning) as cm:
1055 do_something()
1056
1057 self.assertIn('myfile.py', cm.filename)
1058 self.assertEqual(320, cm.lineno)
1059
1060 This method works regardless of the warning filters in place when it
1061 is called.
1062
1063 .. versionadded:: 3.2
1064
Ezio Melottib4dc2502011-05-06 15:01:41 +03001065 .. versionchanged:: 3.3
1066 Added the *msg* keyword argument when used as a context manager.
1067
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001068
Ezio Melottied3a7d22010-12-01 02:32:32 +00001069 .. method:: assertWarnsRegex(warning, regex, callable, *args, **kwds)
Ezio Melottib4dc2502011-05-06 15:01:41 +03001070 assertWarnsRegex(warning, regex, msg=None)
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001071
Ezio Melottied3a7d22010-12-01 02:32:32 +00001072 Like :meth:`assertWarns` but also tests that *regex* matches on the
1073 message of the triggered warning. *regex* may be a regular expression
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001074 object or a string containing a regular expression suitable for use
1075 by :func:`re.search`. Example::
1076
Ezio Melottied3a7d22010-12-01 02:32:32 +00001077 self.assertWarnsRegex(DeprecationWarning,
1078 r'legacy_function\(\) is deprecated',
1079 legacy_function, 'XYZ')
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001080
1081 or::
1082
Ezio Melottied3a7d22010-12-01 02:32:32 +00001083 with self.assertWarnsRegex(RuntimeWarning, 'unsafe frobnicating'):
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001084 frobnicate('/etc/passwd')
1085
1086 .. versionadded:: 3.2
1087
Ezio Melottib4dc2502011-05-06 15:01:41 +03001088 .. versionchanged:: 3.3
1089 Added the *msg* keyword argument when used as a context manager.
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001090
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001091
Ezio Melotti4370b302010-11-03 20:39:14 +00001092 There are also other methods used to perform more specific checks, such as:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001093
Ezio Melotti4370b302010-11-03 20:39:14 +00001094 +---------------------------------------+--------------------------------+--------------+
1095 | Method | Checks that | New in |
1096 +=======================================+================================+==============+
1097 | :meth:`assertAlmostEqual(a, b) | ``round(a-b, 7) == 0`` | |
1098 | <TestCase.assertAlmostEqual>` | | |
1099 +---------------------------------------+--------------------------------+--------------+
1100 | :meth:`assertNotAlmostEqual(a, b) | ``round(a-b, 7) != 0`` | |
1101 | <TestCase.assertNotAlmostEqual>` | | |
1102 +---------------------------------------+--------------------------------+--------------+
1103 | :meth:`assertGreater(a, b) | ``a > b`` | 3.1 |
1104 | <TestCase.assertGreater>` | | |
1105 +---------------------------------------+--------------------------------+--------------+
1106 | :meth:`assertGreaterEqual(a, b) | ``a >= b`` | 3.1 |
1107 | <TestCase.assertGreaterEqual>` | | |
1108 +---------------------------------------+--------------------------------+--------------+
1109 | :meth:`assertLess(a, b) | ``a < b`` | 3.1 |
1110 | <TestCase.assertLess>` | | |
1111 +---------------------------------------+--------------------------------+--------------+
1112 | :meth:`assertLessEqual(a, b) | ``a <= b`` | 3.1 |
1113 | <TestCase.assertLessEqual>` | | |
1114 +---------------------------------------+--------------------------------+--------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +00001115 | :meth:`assertRegex(s, re) | ``regex.search(s)`` | 3.1 |
1116 | <TestCase.assertRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001117 +---------------------------------------+--------------------------------+--------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +00001118 | :meth:`assertNotRegex(s, re) | ``not regex.search(s)`` | 3.2 |
1119 | <TestCase.assertNotRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001120 +---------------------------------------+--------------------------------+--------------+
Éric Araujo941afed2011-09-01 02:47:34 +02001121 | :meth:`assertCountEqual(a, b) | *a* and *b* have the same | 3.2 |
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001122 | <TestCase.assertCountEqual>` | elements in the same number, | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001123 | | regardless of their order | |
1124 +---------------------------------------+--------------------------------+--------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001125
1126
Michael Foorde180d392011-01-28 19:51:48 +00001127 .. method:: assertAlmostEqual(first, second, places=7, msg=None, delta=None)
1128 assertNotAlmostEqual(first, second, places=7, msg=None, delta=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001129
Michael Foorde180d392011-01-28 19:51:48 +00001130 Test that *first* and *second* are approximately (or not approximately)
Ezio Melotti4841fd62010-11-05 15:43:40 +00001131 equal by computing the difference, rounding to the given number of
1132 decimal *places* (default 7), and comparing to zero. Note that these
1133 methods round the values to the given number of *decimal places* (i.e.
1134 like the :func:`round` function) and not *significant digits*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001135
Ezio Melotti4370b302010-11-03 20:39:14 +00001136 If *delta* is supplied instead of *places* then the difference
Michael Foorde180d392011-01-28 19:51:48 +00001137 between *first* and *second* must be less (or more) than *delta*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001138
Ezio Melotti4370b302010-11-03 20:39:14 +00001139 Supplying both *delta* and *places* raises a ``TypeError``.
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001140
Ezio Melotti4370b302010-11-03 20:39:14 +00001141 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001142 :meth:`assertAlmostEqual` automatically considers almost equal objects
1143 that compare equal. :meth:`assertNotAlmostEqual` automatically fails
1144 if the objects compare equal. Added the *delta* keyword argument.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001145
Ezio Melotti4370b302010-11-03 20:39:14 +00001146
Michael Foorde180d392011-01-28 19:51:48 +00001147 .. method:: assertGreater(first, second, msg=None)
1148 assertGreaterEqual(first, second, msg=None)
1149 assertLess(first, second, msg=None)
1150 assertLessEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001151
Michael Foorde180d392011-01-28 19:51:48 +00001152 Test that *first* is respectively >, >=, < or <= than *second* depending
Ezio Melotti4841fd62010-11-05 15:43:40 +00001153 on the method name. If not, the test will fail::
Ezio Melotti4370b302010-11-03 20:39:14 +00001154
1155 >>> self.assertGreaterEqual(3, 4)
1156 AssertionError: "3" unexpectedly not greater than or equal to "4"
1157
1158 .. versionadded:: 3.1
1159
1160
Ezio Melottied3a7d22010-12-01 02:32:32 +00001161 .. method:: assertRegex(text, regex, msg=None)
1162 assertNotRegex(text, regex, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001163
Ezio Melottied3a7d22010-12-01 02:32:32 +00001164 Test that a *regex* search matches (or does not match) *text*. In case
Ezio Melotti4841fd62010-11-05 15:43:40 +00001165 of failure, the error message will include the pattern and the *text* (or
Ezio Melottied3a7d22010-12-01 02:32:32 +00001166 the pattern and the part of *text* that unexpectedly matched). *regex*
Ezio Melotti4370b302010-11-03 20:39:14 +00001167 may be a regular expression object or a string containing a regular
1168 expression suitable for use by :func:`re.search`.
1169
Georg Brandl419e3de2010-12-01 15:44:25 +00001170 .. versionadded:: 3.1
1171 under the name ``assertRegexpMatches``.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001172 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001173 The method ``assertRegexpMatches()`` has been renamed to
1174 :meth:`.assertRegex`.
1175 .. versionadded:: 3.2
1176 :meth:`.assertNotRegex`.
Ezio Melotti4370b302010-11-03 20:39:14 +00001177
1178
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001179 .. method:: assertCountEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001180
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001181 Test that sequence *first* contains the same elements as *second*,
Ezio Melotti4370b302010-11-03 20:39:14 +00001182 regardless of their order. When they don't, an error message listing the
1183 differences between the sequences will be generated.
1184
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001185 Duplicate elements are *not* ignored when comparing *first* and
1186 *second*. It verifies whether each element has the same count in both
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001187 sequences. Equivalent to:
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001188 ``assertEqual(Counter(list(first)), Counter(list(second)))``
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001189 but works with sequences of unhashable objects as well.
Ezio Melotti4370b302010-11-03 20:39:14 +00001190
Ezio Melotti4370b302010-11-03 20:39:14 +00001191 .. versionadded:: 3.2
1192
Ezio Melotti4370b302010-11-03 20:39:14 +00001193
Ezio Melotti22170ed2010-11-20 09:57:27 +00001194 .. _type-specific-methods:
Ezio Melotti4370b302010-11-03 20:39:14 +00001195
Ezio Melotti22170ed2010-11-20 09:57:27 +00001196 The :meth:`assertEqual` method dispatches the equality check for objects of
1197 the same type to different type-specific methods. These methods are already
1198 implemented for most of the built-in types, but it's also possible to
1199 register new methods using :meth:`addTypeEqualityFunc`:
1200
1201 .. method:: addTypeEqualityFunc(typeobj, function)
1202
1203 Registers a type-specific method called by :meth:`assertEqual` to check
1204 if two objects of exactly the same *typeobj* (not subclasses) compare
1205 equal. *function* must take two positional arguments and a third msg=None
1206 keyword argument just as :meth:`assertEqual` does. It must raise
1207 :data:`self.failureException(msg) <failureException>` when inequality
1208 between the first two parameters is detected -- possibly providing useful
1209 information and explaining the inequalities in details in the error
1210 message.
1211
1212 .. versionadded:: 3.1
1213
1214 The list of type-specific methods automatically used by
1215 :meth:`~TestCase.assertEqual` are summarized in the following table. Note
1216 that it's usually not necessary to invoke these methods directly.
Ezio Melotti4370b302010-11-03 20:39:14 +00001217
1218 +-----------------------------------------+-----------------------------+--------------+
1219 | Method | Used to compare | New in |
1220 +=========================================+=============================+==============+
1221 | :meth:`assertMultiLineEqual(a, b) | strings | 3.1 |
1222 | <TestCase.assertMultiLineEqual>` | | |
1223 +-----------------------------------------+-----------------------------+--------------+
1224 | :meth:`assertSequenceEqual(a, b) | sequences | 3.1 |
1225 | <TestCase.assertSequenceEqual>` | | |
1226 +-----------------------------------------+-----------------------------+--------------+
1227 | :meth:`assertListEqual(a, b) | lists | 3.1 |
1228 | <TestCase.assertListEqual>` | | |
1229 +-----------------------------------------+-----------------------------+--------------+
1230 | :meth:`assertTupleEqual(a, b) | tuples | 3.1 |
1231 | <TestCase.assertTupleEqual>` | | |
1232 +-----------------------------------------+-----------------------------+--------------+
1233 | :meth:`assertSetEqual(a, b) | sets or frozensets | 3.1 |
1234 | <TestCase.assertSetEqual>` | | |
1235 +-----------------------------------------+-----------------------------+--------------+
1236 | :meth:`assertDictEqual(a, b) | dicts | 3.1 |
1237 | <TestCase.assertDictEqual>` | | |
1238 +-----------------------------------------+-----------------------------+--------------+
1239
1240
1241
Michael Foorde180d392011-01-28 19:51:48 +00001242 .. method:: assertMultiLineEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001243
Michael Foorde180d392011-01-28 19:51:48 +00001244 Test that the multiline string *first* is equal to the string *second*.
Ezio Melotti4370b302010-11-03 20:39:14 +00001245 When not equal a diff of the two strings highlighting the differences
1246 will be included in the error message. This method is used by default
1247 when comparing strings with :meth:`assertEqual`.
1248
Ezio Melotti4370b302010-11-03 20:39:14 +00001249 .. versionadded:: 3.1
1250
1251
Michael Foorde180d392011-01-28 19:51:48 +00001252 .. method:: assertSequenceEqual(first, second, msg=None, seq_type=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001253
1254 Tests that two sequences are equal. If a *seq_type* is supplied, both
Michael Foorde180d392011-01-28 19:51:48 +00001255 *first* and *second* must be instances of *seq_type* or a failure will
Ezio Melotti4370b302010-11-03 20:39:14 +00001256 be raised. If the sequences are different an error message is
1257 constructed that shows the difference between the two.
1258
Ezio Melotti22170ed2010-11-20 09:57:27 +00001259 This method is not called directly by :meth:`assertEqual`, but
1260 it's used to implement :meth:`assertListEqual` and
Ezio Melotti4370b302010-11-03 20:39:14 +00001261 :meth:`assertTupleEqual`.
1262
1263 .. versionadded:: 3.1
1264
1265
Michael Foorde180d392011-01-28 19:51:48 +00001266 .. method:: assertListEqual(first, second, msg=None)
1267 assertTupleEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001268
Ezio Melotti49ccd512012-08-29 17:50:42 +03001269 Tests that two lists or tuples are equal. If not, an error message is
Ezio Melotti4370b302010-11-03 20:39:14 +00001270 constructed that shows only the differences between the two. An error
1271 is also raised if either of the parameters are of the wrong type.
1272 These methods are used by default when comparing lists or tuples with
1273 :meth:`assertEqual`.
1274
Ezio Melotti4370b302010-11-03 20:39:14 +00001275 .. versionadded:: 3.1
1276
1277
Michael Foorde180d392011-01-28 19:51:48 +00001278 .. method:: assertSetEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001279
1280 Tests that two sets are equal. If not, an error message is constructed
1281 that lists the differences between the sets. This method is used by
1282 default when comparing sets or frozensets with :meth:`assertEqual`.
1283
Michael Foorde180d392011-01-28 19:51:48 +00001284 Fails if either of *first* or *second* does not have a :meth:`set.difference`
Ezio Melotti4370b302010-11-03 20:39:14 +00001285 method.
1286
Ezio Melotti4370b302010-11-03 20:39:14 +00001287 .. versionadded:: 3.1
1288
1289
Michael Foorde180d392011-01-28 19:51:48 +00001290 .. method:: assertDictEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001291
1292 Test that two dictionaries are equal. If not, an error message is
1293 constructed that shows the differences in the dictionaries. This
1294 method will be used by default to compare dictionaries in
1295 calls to :meth:`assertEqual`.
1296
Ezio Melotti4370b302010-11-03 20:39:14 +00001297 .. versionadded:: 3.1
1298
1299
1300
Ezio Melotti22170ed2010-11-20 09:57:27 +00001301 .. _other-methods-and-attrs:
1302
Ezio Melotti4370b302010-11-03 20:39:14 +00001303 Finally the :class:`TestCase` provides the following methods and attributes:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001304
Benjamin Peterson52baa292009-03-24 00:56:30 +00001305
Georg Brandl7f01a132009-09-16 15:58:14 +00001306 .. method:: fail(msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001307
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001308 Signals a test failure unconditionally, with *msg* or ``None`` for
Benjamin Peterson52baa292009-03-24 00:56:30 +00001309 the error message.
1310
1311
1312 .. attribute:: failureException
1313
1314 This class attribute gives the exception raised by the test method. If a
1315 test framework needs to use a specialized exception, possibly to carry
1316 additional information, it must subclass this exception in order to "play
1317 fair" with the framework. The initial value of this attribute is
1318 :exc:`AssertionError`.
1319
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001320
1321 .. attribute:: longMessage
1322
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001323 If set to ``True`` then any explicit failure message you pass in to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001324 :ref:`assert methods <assert-methods>` will be appended to the end of the
1325 normal failure message. The normal messages contain useful information
1326 about the objects involved, for example the message from assertEqual
1327 shows you the repr of the two unequal objects. Setting this attribute
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001328 to ``True`` allows you to have a custom error message in addition to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001329 normal one.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001330
Michael Foord5074df62010-12-03 00:53:09 +00001331 This attribute defaults to ``True``. If set to False then a custom message
1332 passed to an assert method will silence the normal message.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001333
1334 The class setting can be overridden in individual tests by assigning an
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001335 instance attribute to ``True`` or ``False`` before calling the assert methods.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001336
Raymond Hettinger35a88362009-04-09 00:08:24 +00001337 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001338
1339
Michael Foord98b3e762010-06-05 21:59:55 +00001340 .. attribute:: maxDiff
1341
1342 This attribute controls the maximum length of diffs output by assert
1343 methods that report diffs on failure. It defaults to 80*8 characters.
1344 Assert methods affected by this attribute are
1345 :meth:`assertSequenceEqual` (including all the sequence comparison
1346 methods that delegate to it), :meth:`assertDictEqual` and
1347 :meth:`assertMultiLineEqual`.
1348
1349 Setting ``maxDiff`` to None means that there is no maximum length of
1350 diffs.
1351
1352 .. versionadded:: 3.2
1353
1354
Benjamin Peterson52baa292009-03-24 00:56:30 +00001355 Testing frameworks can use the following methods to collect information on
1356 the test:
1357
1358
1359 .. method:: countTestCases()
1360
1361 Return the number of tests represented by this test object. For
1362 :class:`TestCase` instances, this will always be ``1``.
1363
1364
1365 .. method:: defaultTestResult()
1366
1367 Return an instance of the test result class that should be used for this
1368 test case class (if no other result instance is provided to the
1369 :meth:`run` method).
1370
1371 For :class:`TestCase` instances, this will always be an instance of
1372 :class:`TestResult`; subclasses of :class:`TestCase` should override this
1373 as necessary.
1374
1375
1376 .. method:: id()
1377
1378 Return a string identifying the specific test case. This is usually the
1379 full name of the test method, including the module and class name.
1380
1381
1382 .. method:: shortDescription()
1383
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001384 Returns a description of the test, or ``None`` if no description
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001385 has been provided. The default implementation of this method
1386 returns the first line of the test method's docstring, if available,
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001387 or ``None``.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001388
Georg Brandl419e3de2010-12-01 15:44:25 +00001389 .. versionchanged:: 3.1
Michael Foord34c94622010-02-10 15:51:42 +00001390 In 3.1 this was changed to add the test name to the short description
Georg Brandl419e3de2010-12-01 15:44:25 +00001391 even in the presence of a docstring. This caused compatibility issues
Michael Foord34c94622010-02-10 15:51:42 +00001392 with unittest extensions and adding the test name was moved to the
Georg Brandl419e3de2010-12-01 15:44:25 +00001393 :class:`TextTestResult` in Python 3.2.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001394
Georg Brandl116aa622007-08-15 14:28:22 +00001395
Georg Brandl7f01a132009-09-16 15:58:14 +00001396 .. method:: addCleanup(function, *args, **kwargs)
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001397
1398 Add a function to be called after :meth:`tearDown` to cleanup resources
1399 used during the test. Functions will be called in reverse order to the
1400 order they are added (LIFO). They are called with any arguments and
1401 keyword arguments passed into :meth:`addCleanup` when they are
1402 added.
1403
1404 If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
1405 then any cleanup functions added will still be called.
1406
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001407 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001408
1409
1410 .. method:: doCleanups()
1411
Barry Warsaw0c9fd632010-04-12 14:50:57 +00001412 This method is called unconditionally after :meth:`tearDown`, or
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001413 after :meth:`setUp` if :meth:`setUp` raises an exception.
1414
1415 It is responsible for calling all the cleanup functions added by
1416 :meth:`addCleanup`. If you need cleanup functions to be called
1417 *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
1418 yourself.
1419
1420 :meth:`doCleanups` pops methods off the stack of cleanup
1421 functions one at a time, so it can be called at any time.
1422
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001423 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001424
1425
Georg Brandl7f01a132009-09-16 15:58:14 +00001426.. class:: FunctionTestCase(testFunc, setUp=None, tearDown=None, description=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001427
1428 This class implements the portion of the :class:`TestCase` interface which
Benjamin Petersond2397752009-06-27 23:45:02 +00001429 allows the test runner to drive the test, but does not provide the methods
1430 which test code can use to check and report errors. This is used to create
1431 test cases using legacy test code, allowing it to be integrated into a
1432 :mod:`unittest`-based test framework.
Georg Brandl116aa622007-08-15 14:28:22 +00001433
1434
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001435.. _deprecated-aliases:
1436
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001437Deprecated aliases
1438##################
1439
1440For historical reasons, some of the :class:`TestCase` methods had one or more
1441aliases that are now deprecated. The following table lists the correct names
1442along with their deprecated aliases:
1443
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001444 ============================== ====================== ======================
1445 Method Name Deprecated alias Deprecated alias
1446 ============================== ====================== ======================
1447 :meth:`.assertEqual` failUnlessEqual assertEquals
1448 :meth:`.assertNotEqual` failIfEqual assertNotEquals
1449 :meth:`.assertTrue` failUnless assert\_
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001450 :meth:`.assertFalse` failIf
1451 :meth:`.assertRaises` failUnlessRaises
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001452 :meth:`.assertAlmostEqual` failUnlessAlmostEqual assertAlmostEquals
1453 :meth:`.assertNotAlmostEqual` failIfAlmostEqual assertNotAlmostEquals
Ezio Melottied3a7d22010-12-01 02:32:32 +00001454 :meth:`.assertRegex` assertRegexpMatches
1455 :meth:`.assertRaisesRegex` assertRaisesRegexp
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001456 ============================== ====================== ======================
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001457
Ezio Melotti361467e2011-04-03 17:37:58 +03001458 .. deprecated:: 3.1
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001459 the fail* aliases listed in the second column.
1460 .. deprecated:: 3.2
1461 the assert* aliases listed in the third column.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001462 .. deprecated:: 3.2
1463 ``assertRegexpMatches`` and ``assertRaisesRegexp`` have been renamed to
1464 :meth:`.assertRegex` and :meth:`.assertRaisesRegex`
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001465
1466
Benjamin Peterson52baa292009-03-24 00:56:30 +00001467.. _testsuite-objects:
1468
Benjamin Peterson52baa292009-03-24 00:56:30 +00001469Grouping tests
1470~~~~~~~~~~~~~~
1471
Georg Brandl7f01a132009-09-16 15:58:14 +00001472.. class:: TestSuite(tests=())
Georg Brandl116aa622007-08-15 14:28:22 +00001473
1474 This class represents an aggregation of individual tests cases and test suites.
1475 The class presents the interface needed by the test runner to allow it to be run
1476 as any other test case. Running a :class:`TestSuite` instance is the same as
1477 iterating over the suite, running each test individually.
1478
1479 If *tests* is given, it must be an iterable of individual test cases or other
1480 test suites that will be used to build the suite initially. Additional methods
1481 are provided to add test cases and suites to the collection later on.
1482
Benjamin Peterson14a3dd72009-05-25 00:51:58 +00001483 :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
1484 they do not actually implement a test. Instead, they are used to aggregate
1485 tests into groups of tests that should be run together. Some additional
1486 methods are available to add tests to :class:`TestSuite` instances:
Benjamin Peterson52baa292009-03-24 00:56:30 +00001487
1488
1489 .. method:: TestSuite.addTest(test)
1490
1491 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1492
1493
1494 .. method:: TestSuite.addTests(tests)
1495
1496 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
1497 instances to this test suite.
1498
Benjamin Petersond2397752009-06-27 23:45:02 +00001499 This is equivalent to iterating over *tests*, calling :meth:`addTest` for
1500 each element.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001501
1502 :class:`TestSuite` shares the following methods with :class:`TestCase`:
1503
1504
1505 .. method:: run(result)
1506
1507 Run the tests associated with this suite, collecting the result into the
1508 test result object passed as *result*. Note that unlike
1509 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
1510 be passed in.
1511
1512
1513 .. method:: debug()
1514
1515 Run the tests associated with this suite without collecting the
1516 result. This allows exceptions raised by the test to be propagated to the
1517 caller and can be used to support running tests under a debugger.
1518
1519
1520 .. method:: countTestCases()
1521
1522 Return the number of tests represented by this test object, including all
1523 individual tests and sub-suites.
1524
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001525
1526 .. method:: __iter__()
1527
1528 Tests grouped by a :class:`TestSuite` are always accessed by iteration.
1529 Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
1530 that this method maybe called several times on a single suite
1531 (for example when counting tests or comparing for equality)
1532 so the tests returned must be the same for repeated iterations.
1533
Georg Brandl853947a2010-01-31 18:53:23 +00001534 .. versionchanged:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001535 In earlier versions the :class:`TestSuite` accessed tests directly rather
1536 than through iteration, so overriding :meth:`__iter__` wasn't sufficient
1537 for providing tests.
1538
Benjamin Peterson52baa292009-03-24 00:56:30 +00001539 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
1540 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1541
1542
Benjamin Peterson52baa292009-03-24 00:56:30 +00001543Loading and running tests
1544~~~~~~~~~~~~~~~~~~~~~~~~~
1545
Georg Brandl116aa622007-08-15 14:28:22 +00001546.. class:: TestLoader()
1547
Benjamin Peterson52baa292009-03-24 00:56:30 +00001548 The :class:`TestLoader` class is used to create test suites from classes and
1549 modules. Normally, there is no need to create an instance of this class; the
1550 :mod:`unittest` module provides an instance that can be shared as
Ezio Melottib8e336b2012-04-29 10:52:18 +03001551 :data:`unittest.defaultTestLoader`. Using a subclass or instance, however,
1552 allows customization of some configurable properties.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001553
1554 :class:`TestLoader` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001555
Ezio Melotti9c02c2f2010-11-03 20:45:31 +00001556
Benjamin Peterson52baa292009-03-24 00:56:30 +00001557 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl116aa622007-08-15 14:28:22 +00001558
Benjamin Peterson52baa292009-03-24 00:56:30 +00001559 Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
1560 :class:`testCaseClass`.
1561
1562
1563 .. method:: loadTestsFromModule(module)
1564
1565 Return a suite of all tests cases contained in the given module. This
1566 method searches *module* for classes derived from :class:`TestCase` and
1567 creates an instance of the class for each test method defined for the
1568 class.
1569
Georg Brandle720c0a2009-04-27 16:20:50 +00001570 .. note::
Benjamin Peterson52baa292009-03-24 00:56:30 +00001571
1572 While using a hierarchy of :class:`TestCase`\ -derived classes can be
1573 convenient in sharing fixtures and helper functions, defining test
1574 methods on base classes that are not intended to be instantiated
1575 directly does not play well with this method. Doing so, however, can
1576 be useful when the fixtures are different and defined in subclasses.
1577
Benjamin Petersond2397752009-06-27 23:45:02 +00001578 If a module provides a ``load_tests`` function it will be called to
1579 load the tests. This allows modules to customize test loading.
1580 This is the `load_tests protocol`_.
1581
Georg Brandl853947a2010-01-31 18:53:23 +00001582 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001583 Support for ``load_tests`` added.
1584
Benjamin Peterson52baa292009-03-24 00:56:30 +00001585
Georg Brandl7f01a132009-09-16 15:58:14 +00001586 .. method:: loadTestsFromName(name, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001587
1588 Return a suite of all tests cases given a string specifier.
1589
1590 The specifier *name* is a "dotted name" that may resolve either to a
1591 module, a test case class, a test method within a test case class, a
1592 :class:`TestSuite` instance, or a callable object which returns a
1593 :class:`TestCase` or :class:`TestSuite` instance. These checks are
1594 applied in the order listed here; that is, a method on a possible test
1595 case class will be picked up as "a test method within a test case class",
1596 rather than "a callable object".
1597
1598 For example, if you have a module :mod:`SampleTests` containing a
1599 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
1600 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
Benjamin Petersond2397752009-06-27 23:45:02 +00001601 specifier ``'SampleTests.SampleTestCase'`` would cause this method to
1602 return a suite which will run all three test methods. Using the specifier
1603 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
1604 suite which will run only the :meth:`test_two` test method. The specifier
1605 can refer to modules and packages which have not been imported; they will
1606 be imported as a side-effect.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001607
1608 The method optionally resolves *name* relative to the given *module*.
1609
1610
Georg Brandl7f01a132009-09-16 15:58:14 +00001611 .. method:: loadTestsFromNames(names, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001612
1613 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
1614 than a single name. The return value is a test suite which supports all
1615 the tests defined for each name.
1616
1617
1618 .. method:: getTestCaseNames(testCaseClass)
1619
1620 Return a sorted sequence of method names found within *testCaseClass*;
1621 this should be a subclass of :class:`TestCase`.
1622
Benjamin Petersond2397752009-06-27 23:45:02 +00001623
1624 .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)
1625
1626 Find and return all test modules from the specified start directory,
1627 recursing into subdirectories to find them. Only test files that match
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001628 *pattern* will be loaded. (Using shell style pattern matching.) Only
1629 module names that are importable (i.e. are valid Python identifiers) will
1630 be loaded.
Benjamin Petersond2397752009-06-27 23:45:02 +00001631
1632 All test modules must be importable from the top level of the project. If
1633 the start directory is not the top level directory then the top level
1634 directory must be specified separately.
1635
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001636 If importing a module fails, for example due to a syntax error, then this
1637 will be recorded as a single error and discovery will continue.
1638
Benjamin Petersond2397752009-06-27 23:45:02 +00001639 If a test package name (directory with :file:`__init__.py`) matches the
1640 pattern then the package will be checked for a ``load_tests``
1641 function. If this exists then it will be called with *loader*, *tests*,
1642 *pattern*.
1643
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001644 If load_tests exists then discovery does *not* recurse into the package,
Benjamin Petersond2397752009-06-27 23:45:02 +00001645 ``load_tests`` is responsible for loading all tests in the package.
1646
1647 The pattern is deliberately not stored as a loader attribute so that
1648 packages can continue discovery themselves. *top_level_dir* is stored so
1649 ``load_tests`` does not need to pass this argument in to
1650 ``loader.discover()``.
1651
Benjamin Petersonb48af542010-04-11 20:43:16 +00001652 *start_dir* can be a dotted module name as well as a directory.
1653
Georg Brandl853947a2010-01-31 18:53:23 +00001654 .. versionadded:: 3.2
1655
Benjamin Petersond2397752009-06-27 23:45:02 +00001656
Benjamin Peterson52baa292009-03-24 00:56:30 +00001657 The following attributes of a :class:`TestLoader` can be configured either by
1658 subclassing or assignment on an instance:
1659
1660
1661 .. attribute:: testMethodPrefix
1662
1663 String giving the prefix of method names which will be interpreted as test
1664 methods. The default value is ``'test'``.
1665
1666 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
1667 methods.
1668
1669
1670 .. attribute:: sortTestMethodsUsing
1671
1672 Function to be used to compare method names when sorting them in
1673 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
1674
1675
1676 .. attribute:: suiteClass
1677
1678 Callable object that constructs a test suite from a list of tests. No
1679 methods on the resulting object are needed. The default value is the
1680 :class:`TestSuite` class.
1681
1682 This affects all the :meth:`loadTestsFrom\*` methods.
1683
1684
Benjamin Peterson52baa292009-03-24 00:56:30 +00001685.. class:: TestResult
1686
1687 This class is used to compile information about which tests have succeeded
1688 and which have failed.
1689
1690 A :class:`TestResult` object stores the results of a set of tests. The
1691 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
1692 properly recorded; test authors do not need to worry about recording the
1693 outcome of tests.
1694
1695 Testing frameworks built on top of :mod:`unittest` may want access to the
1696 :class:`TestResult` object generated by running a set of tests for reporting
1697 purposes; a :class:`TestResult` instance is returned by the
1698 :meth:`TestRunner.run` method for this purpose.
1699
1700 :class:`TestResult` instances have the following attributes that will be of
1701 interest when inspecting the results of running a set of tests:
1702
1703
1704 .. attribute:: errors
1705
1706 A list containing 2-tuples of :class:`TestCase` instances and strings
1707 holding formatted tracebacks. Each tuple represents a test which raised an
1708 unexpected exception.
1709
Benjamin Peterson52baa292009-03-24 00:56:30 +00001710 .. attribute:: failures
1711
1712 A list containing 2-tuples of :class:`TestCase` instances and strings
1713 holding formatted tracebacks. Each tuple represents a test where a failure
1714 was explicitly signalled using the :meth:`TestCase.fail\*` or
1715 :meth:`TestCase.assert\*` methods.
1716
Benjamin Peterson52baa292009-03-24 00:56:30 +00001717 .. attribute:: skipped
1718
1719 A list containing 2-tuples of :class:`TestCase` instances and strings
1720 holding the reason for skipping the test.
1721
Benjamin Peterson70e32c82009-03-24 01:00:11 +00001722 .. versionadded:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +00001723
1724 .. attribute:: expectedFailures
1725
Georg Brandl6faee4e2010-09-21 14:48:28 +00001726 A list containing 2-tuples of :class:`TestCase` instances and strings
1727 holding formatted tracebacks. Each tuple represents an expected failure
Benjamin Peterson52baa292009-03-24 00:56:30 +00001728 of the test case.
1729
1730 .. attribute:: unexpectedSuccesses
1731
1732 A list containing :class:`TestCase` instances that were marked as expected
1733 failures, but succeeded.
1734
1735 .. attribute:: shouldStop
1736
1737 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1738
1739
1740 .. attribute:: testsRun
1741
1742 The total number of tests run so far.
1743
1744
Georg Brandl12037202010-12-02 22:35:25 +00001745 .. attribute:: buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +00001746
1747 If set to true, ``sys.stdout`` and ``sys.stderr`` will be buffered in between
1748 :meth:`startTest` and :meth:`stopTest` being called. Collected output will
1749 only be echoed onto the real ``sys.stdout`` and ``sys.stderr`` if the test
1750 fails or errors. Any output is also attached to the failure / error message.
1751
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001752 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001753
1754
1755 .. attribute:: failfast
1756
1757 If set to true :meth:`stop` will be called on the first failure or error,
1758 halting the test run.
1759
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001760 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001761
1762
Benjamin Peterson52baa292009-03-24 00:56:30 +00001763 .. method:: wasSuccessful()
1764
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001765 Return ``True`` if all tests run so far have passed, otherwise returns
1766 ``False``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001767
1768
1769 .. method:: stop()
1770
1771 This method can be called to signal that the set of tests being run should
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001772 be aborted by setting the :attr:`shouldStop` attribute to ``True``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001773 :class:`TestRunner` objects should respect this flag and return without
1774 running any additional tests.
1775
1776 For example, this feature is used by the :class:`TextTestRunner` class to
1777 stop the test framework when the user signals an interrupt from the
1778 keyboard. Interactive tools which provide :class:`TestRunner`
1779 implementations can use this in a similar manner.
1780
1781 The following methods of the :class:`TestResult` class are used to maintain
1782 the internal data structures, and may be extended in subclasses to support
1783 additional reporting requirements. This is particularly useful in building
1784 tools which support interactive reporting while tests are being run.
1785
1786
1787 .. method:: startTest(test)
1788
1789 Called when the test case *test* is about to be run.
1790
Benjamin Peterson52baa292009-03-24 00:56:30 +00001791 .. method:: stopTest(test)
1792
1793 Called after the test case *test* has been executed, regardless of the
1794 outcome.
1795
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001796 .. method:: startTestRun(test)
1797
1798 Called once before any tests are executed.
1799
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001800 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001801
1802
1803 .. method:: stopTestRun(test)
1804
Ezio Melotti176d6c42010-01-27 20:58:07 +00001805 Called once after all tests are executed.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001806
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001807 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001808
1809
Benjamin Peterson52baa292009-03-24 00:56:30 +00001810 .. method:: addError(test, err)
1811
1812 Called when the test case *test* raises an unexpected exception *err* is a
1813 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1814 traceback)``.
1815
1816 The default implementation appends a tuple ``(test, formatted_err)`` to
1817 the instance's :attr:`errors` attribute, where *formatted_err* is a
1818 formatted traceback derived from *err*.
1819
1820
1821 .. method:: addFailure(test, err)
1822
Benjamin Petersond2397752009-06-27 23:45:02 +00001823 Called when the test case *test* signals a failure. *err* is a tuple of
1824 the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001825
1826 The default implementation appends a tuple ``(test, formatted_err)`` to
1827 the instance's :attr:`failures` attribute, where *formatted_err* is a
1828 formatted traceback derived from *err*.
1829
1830
1831 .. method:: addSuccess(test)
1832
1833 Called when the test case *test* succeeds.
1834
1835 The default implementation does nothing.
1836
1837
1838 .. method:: addSkip(test, reason)
1839
1840 Called when the test case *test* is skipped. *reason* is the reason the
1841 test gave for skipping.
1842
1843 The default implementation appends a tuple ``(test, reason)`` to the
1844 instance's :attr:`skipped` attribute.
1845
1846
1847 .. method:: addExpectedFailure(test, err)
1848
1849 Called when the test case *test* fails, but was marked with the
1850 :func:`expectedFailure` decorator.
1851
1852 The default implementation appends a tuple ``(test, formatted_err)`` to
1853 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1854 is a formatted traceback derived from *err*.
1855
1856
1857 .. method:: addUnexpectedSuccess(test)
1858
1859 Called when the test case *test* was marked with the
1860 :func:`expectedFailure` decorator, but succeeded.
1861
1862 The default implementation appends the test to the instance's
1863 :attr:`unexpectedSuccesses` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +00001864
Georg Brandl67b21b72010-08-17 15:07:14 +00001865
Michael Foord34c94622010-02-10 15:51:42 +00001866.. class:: TextTestResult(stream, descriptions, verbosity)
1867
Georg Brandl67b21b72010-08-17 15:07:14 +00001868 A concrete implementation of :class:`TestResult` used by the
1869 :class:`TextTestRunner`.
Michael Foord34c94622010-02-10 15:51:42 +00001870
Georg Brandl67b21b72010-08-17 15:07:14 +00001871 .. versionadded:: 3.2
1872 This class was previously named ``_TextTestResult``. The old name still
1873 exists as an alias but is deprecated.
1874
Georg Brandl116aa622007-08-15 14:28:22 +00001875
1876.. data:: defaultTestLoader
1877
1878 Instance of the :class:`TestLoader` class intended to be shared. If no
1879 customization of the :class:`TestLoader` is needed, this instance can be used
1880 instead of repeatedly creating new instances.
1881
1882
Michael Foordd218e952011-01-03 12:55:11 +00001883.. class:: TextTestRunner(stream=None, descriptions=True, verbosity=1, runnerclass=None, warnings=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001884
Michael Foordd218e952011-01-03 12:55:11 +00001885 A basic test runner implementation that outputs results to a stream. If *stream*
Éric Araujo941afed2011-09-01 02:47:34 +02001886 is ``None``, the default, :data:`sys.stderr` is used as the output stream. This class
Georg Brandl116aa622007-08-15 14:28:22 +00001887 has a few configurable parameters, but is essentially very simple. Graphical
1888 applications which run test suites should provide alternate implementations.
1889
Ezio Melotti60901872010-12-01 00:56:10 +00001890 By default this runner shows :exc:`DeprecationWarning`,
1891 :exc:`PendingDeprecationWarning`, and :exc:`ImportWarning` even if they are
1892 :ref:`ignored by default <warning-ignored>`. Deprecation warnings caused by
1893 :ref:`deprecated unittest methods <deprecated-aliases>` are also
1894 special-cased and, when the warning filters are ``'default'`` or ``'always'``,
1895 they will appear only once per-module, in order to avoid too many warning
Georg Brandl46402372010-12-04 19:06:18 +00001896 messages. This behavior can be overridden using the :option:`-Wd` or
Ezio Melotti60901872010-12-01 00:56:10 +00001897 :option:`-Wa` options and leaving *warnings* to ``None``.
1898
Michael Foordd218e952011-01-03 12:55:11 +00001899 .. versionchanged:: 3.2
1900 Added the ``warnings`` argument.
1901
1902 .. versionchanged:: 3.2
Éric Araujo941afed2011-09-01 02:47:34 +02001903 The default stream is set to :data:`sys.stderr` at instantiation time rather
Michael Foordd218e952011-01-03 12:55:11 +00001904 than import time.
1905
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001906 .. method:: _makeResult()
Georg Brandl116aa622007-08-15 14:28:22 +00001907
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001908 This method returns the instance of ``TestResult`` used by :meth:`run`.
1909 It is not intended to be called directly, but can be overridden in
1910 subclasses to provide a custom ``TestResult``.
1911
Michael Foord34c94622010-02-10 15:51:42 +00001912 ``_makeResult()`` instantiates the class or callable passed in the
1913 ``TextTestRunner`` constructor as the ``resultclass`` argument. It
Benjamin Petersonb48af542010-04-11 20:43:16 +00001914 defaults to :class:`TextTestResult` if no ``resultclass`` is provided.
Michael Foord34c94622010-02-10 15:51:42 +00001915 The result class is instantiated with the following arguments::
1916
1917 stream, descriptions, verbosity
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001918
Ezio Melotti60901872010-12-01 00:56:10 +00001919
Georg Brandl419e3de2010-12-01 15:44:25 +00001920.. function:: main(module='__main__', defaultTest=None, argv=None, testRunner=None, \
Ezio Melotti40dcb1d2011-03-10 13:46:50 +02001921 testLoader=unittest.defaultTestLoader, exit=True, verbosity=1, \
Georg Brandl419e3de2010-12-01 15:44:25 +00001922 failfast=None, catchbreak=None, buffer=None, warnings=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001923
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03001924 A command-line program that loads a set of tests from *module* and runs them;
1925 this is primarily for making test modules conveniently executable.
1926 The simplest use for this function is to include the following line at the
1927 end of a test script::
Georg Brandl116aa622007-08-15 14:28:22 +00001928
1929 if __name__ == '__main__':
1930 unittest.main()
1931
Benjamin Petersond2397752009-06-27 23:45:02 +00001932 You can run tests with more detailed information by passing in the verbosity
1933 argument::
1934
1935 if __name__ == '__main__':
1936 unittest.main(verbosity=2)
1937
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03001938 The *argv* argument can be a list of options passed to the program, with the
1939 first element being the program name. If not specified or ``None``,
1940 the values of :data:`sys.argv` are used.
1941
Georg Brandl116aa622007-08-15 14:28:22 +00001942 The *testRunner* argument can either be a test runner class or an already
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001943 created instance of it. By default ``main`` calls :func:`sys.exit` with
1944 an exit code indicating success or failure of the tests run.
1945
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03001946 The *testLoader* argument has to be a :class:`TestLoader` instance,
1947 and defaults to :data:`defaultTestLoader`.
1948
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001949 ``main`` supports being used from the interactive interpreter by passing in the
1950 argument ``exit=False``. This displays the result on standard output without
1951 calling :func:`sys.exit`::
1952
1953 >>> from unittest import main
1954 >>> main(module='test_module', exit=False)
1955
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03001956 The *failfast*, *catchbreak* and *buffer* parameters have the same
Éric Araujo76338ec2010-11-26 23:46:18 +00001957 effect as the same-name `command-line options`_.
Benjamin Petersonb48af542010-04-11 20:43:16 +00001958
Ezio Melotti60901872010-12-01 00:56:10 +00001959 The *warning* argument specifies the :ref:`warning filter <warning-filter>`
1960 that should be used while running the tests. If it's not specified, it will
1961 remain ``None`` if a :option:`-W` option is passed to :program:`python`,
1962 otherwise it will be set to ``'default'``.
1963
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001964 Calling ``main`` actually returns an instance of the ``TestProgram`` class.
1965 This stores the result of the tests run as the ``result`` attribute.
1966
Éric Araujo971dc012010-12-16 03:13:05 +00001967 .. versionchanged:: 3.1
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03001968 The *exit* parameter was added.
Éric Araujo971dc012010-12-16 03:13:05 +00001969
Georg Brandl853947a2010-01-31 18:53:23 +00001970 .. versionchanged:: 3.2
Ezio Melotti3d6d7a52012-04-30 19:10:28 +03001971 The *verbosity*, *failfast*, *catchbreak*, *buffer*
1972 and *warnings* parameters were added.
Benjamin Petersond2397752009-06-27 23:45:02 +00001973
1974
1975load_tests Protocol
1976###################
1977
Georg Brandl853947a2010-01-31 18:53:23 +00001978.. versionadded:: 3.2
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001979
Benjamin Petersond2397752009-06-27 23:45:02 +00001980Modules or packages can customize how tests are loaded from them during normal
1981test runs or test discovery by implementing a function called ``load_tests``.
1982
1983If a test module defines ``load_tests`` it will be called by
1984:meth:`TestLoader.loadTestsFromModule` with the following arguments::
1985
1986 load_tests(loader, standard_tests, None)
1987
1988It should return a :class:`TestSuite`.
1989
1990*loader* is the instance of :class:`TestLoader` doing the loading.
1991*standard_tests* are the tests that would be loaded by default from the
1992module. It is common for test modules to only want to add or remove tests
1993from the standard set of tests.
1994The third argument is used when loading packages as part of test discovery.
1995
1996A typical ``load_tests`` function that loads tests from a specific set of
1997:class:`TestCase` classes may look like::
1998
1999 test_cases = (TestCase1, TestCase2, TestCase3)
2000
2001 def load_tests(loader, tests, pattern):
2002 suite = TestSuite()
2003 for test_class in test_cases:
2004 tests = loader.loadTestsFromTestCase(test_class)
2005 suite.addTests(tests)
2006 return suite
2007
2008If discovery is started, either from the command line or by calling
2009:meth:`TestLoader.discover`, with a pattern that matches a package
2010name then the package :file:`__init__.py` will be checked for ``load_tests``.
2011
2012.. note::
2013
Ezio Melotti0639d5a2009-12-19 23:26:38 +00002014 The default pattern is 'test*.py'. This matches all Python files
Benjamin Petersond2397752009-06-27 23:45:02 +00002015 that start with 'test' but *won't* match any test directories.
2016
2017 A pattern like 'test*' will match test packages as well as
2018 modules.
2019
2020If the package :file:`__init__.py` defines ``load_tests`` then it will be
2021called and discovery not continued into the package. ``load_tests``
2022is called with the following arguments::
2023
2024 load_tests(loader, standard_tests, pattern)
2025
2026This should return a :class:`TestSuite` representing all the tests
2027from the package. (``standard_tests`` will only contain tests
2028collected from :file:`__init__.py`.)
2029
2030Because the pattern is passed into ``load_tests`` the package is free to
2031continue (and potentially modify) test discovery. A 'do nothing'
2032``load_tests`` function for a test package would look like::
2033
2034 def load_tests(loader, standard_tests, pattern):
2035 # top level directory cached on loader instance
2036 this_dir = os.path.dirname(__file__)
2037 package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
2038 standard_tests.addTests(package_tests)
2039 return standard_tests
Benjamin Petersonb48af542010-04-11 20:43:16 +00002040
2041
2042Class and Module Fixtures
2043-------------------------
2044
2045Class and module level fixtures are implemented in :class:`TestSuite`. When
2046the test suite encounters a test from a new class then :meth:`tearDownClass`
2047from the previous class (if there is one) is called, followed by
2048:meth:`setUpClass` from the new class.
2049
2050Similarly if a test is from a different module from the previous test then
2051``tearDownModule`` from the previous module is run, followed by
2052``setUpModule`` from the new module.
2053
2054After all the tests have run the final ``tearDownClass`` and
2055``tearDownModule`` are run.
2056
2057Note that shared fixtures do not play well with [potential] features like test
2058parallelization and they break test isolation. They should be used with care.
2059
2060The default ordering of tests created by the unittest test loaders is to group
2061all tests from the same modules and classes together. This will lead to
2062``setUpClass`` / ``setUpModule`` (etc) being called exactly once per class and
2063module. If you randomize the order, so that tests from different modules and
2064classes are adjacent to each other, then these shared fixture functions may be
2065called multiple times in a single test run.
2066
2067Shared fixtures are not intended to work with suites with non-standard
2068ordering. A ``BaseTestSuite`` still exists for frameworks that don't want to
2069support shared fixtures.
2070
2071If there are any exceptions raised during one of the shared fixture functions
2072the test is reported as an error. Because there is no corresponding test
2073instance an ``_ErrorHolder`` object (that has the same interface as a
2074:class:`TestCase`) is created to represent the error. If you are just using
2075the standard unittest test runner then this detail doesn't matter, but if you
2076are a framework author it may be relevant.
2077
2078
2079setUpClass and tearDownClass
2080~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2081
2082These must be implemented as class methods::
2083
2084 import unittest
2085
2086 class Test(unittest.TestCase):
2087 @classmethod
2088 def setUpClass(cls):
2089 cls._connection = createExpensiveConnectionObject()
2090
2091 @classmethod
2092 def tearDownClass(cls):
2093 cls._connection.destroy()
2094
2095If you want the ``setUpClass`` and ``tearDownClass`` on base classes called
2096then you must call up to them yourself. The implementations in
2097:class:`TestCase` are empty.
2098
2099If an exception is raised during a ``setUpClass`` then the tests in the class
2100are not run and the ``tearDownClass`` is not run. Skipped classes will not
Michael Foord98b3e762010-06-05 21:59:55 +00002101have ``setUpClass`` or ``tearDownClass`` run. If the exception is a
2102``SkipTest`` exception then the class will be reported as having been skipped
2103instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002104
2105
2106setUpModule and tearDownModule
2107~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2108
2109These should be implemented as functions::
2110
2111 def setUpModule():
2112 createConnection()
2113
2114 def tearDownModule():
2115 closeConnection()
2116
2117If an exception is raised in a ``setUpModule`` then none of the tests in the
Michael Foord98b3e762010-06-05 21:59:55 +00002118module will be run and the ``tearDownModule`` will not be run. If the exception is a
2119``SkipTest`` exception then the module will be reported as having been skipped
2120instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002121
2122
2123Signal Handling
2124---------------
2125
Georg Brandl419e3de2010-12-01 15:44:25 +00002126.. versionadded:: 3.2
2127
Éric Araujo8acb67c2010-11-26 23:31:07 +00002128The :option:`-c/--catch <unittest -c>` command-line option to unittest,
Éric Araujo76338ec2010-11-26 23:46:18 +00002129along with the ``catchbreak`` parameter to :func:`unittest.main()`, provide
2130more friendly handling of control-C during a test run. With catch break
2131behavior enabled control-C will allow the currently running test to complete,
2132and the test run will then end and report all the results so far. A second
2133control-c will raise a :exc:`KeyboardInterrupt` in the usual way.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002134
Michael Foordde4ceab2010-04-25 19:53:49 +00002135The control-c handling signal handler attempts to remain compatible with code or
2136tests that install their own :const:`signal.SIGINT` handler. If the ``unittest``
2137handler is called but *isn't* the installed :const:`signal.SIGINT` handler,
2138i.e. it has been replaced by the system under test and delegated to, then it
2139calls the default handler. This will normally be the expected behavior by code
2140that replaces an installed handler and delegates to it. For individual tests
2141that need ``unittest`` control-c handling disabled the :func:`removeHandler`
2142decorator can be used.
2143
2144There are a few utility functions for framework authors to enable control-c
2145handling functionality within test frameworks.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002146
2147.. function:: installHandler()
2148
2149 Install the control-c handler. When a :const:`signal.SIGINT` is received
2150 (usually in response to the user pressing control-c) all registered results
2151 have :meth:`~TestResult.stop` called.
2152
Michael Foord469b1f02010-04-26 23:41:26 +00002153
Benjamin Petersonb48af542010-04-11 20:43:16 +00002154.. function:: registerResult(result)
2155
2156 Register a :class:`TestResult` object for control-c handling. Registering a
2157 result stores a weak reference to it, so it doesn't prevent the result from
2158 being garbage collected.
2159
Michael Foordde4ceab2010-04-25 19:53:49 +00002160 Registering a :class:`TestResult` object has no side-effects if control-c
2161 handling is not enabled, so test frameworks can unconditionally register
2162 all results they create independently of whether or not handling is enabled.
2163
Michael Foord469b1f02010-04-26 23:41:26 +00002164
Benjamin Petersonb48af542010-04-11 20:43:16 +00002165.. function:: removeResult(result)
2166
2167 Remove a registered result. Once a result has been removed then
2168 :meth:`~TestResult.stop` will no longer be called on that result object in
2169 response to a control-c.
2170
Michael Foord469b1f02010-04-26 23:41:26 +00002171
Michael Foordde4ceab2010-04-25 19:53:49 +00002172.. function:: removeHandler(function=None)
2173
2174 When called without arguments this function removes the control-c handler
2175 if it has been installed. This function can also be used as a test decorator
2176 to temporarily remove the handler whilst the test is being executed::
2177
2178 @unittest.removeHandler
2179 def test_signal_handling(self):
2180 ...