blob: 91fede07f166d54a8c3687451726f64c27b78982 [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001:mod:`unittest` --- Unit testing framework
2==========================================
3
4.. module:: unittest
5 :synopsis: Unit testing framework for Python.
6.. moduleauthor:: Steve Purcell <stephen_purcell@yahoo.com>
7.. sectionauthor:: Steve Purcell <stephen_purcell@yahoo.com>
8.. sectionauthor:: Fred L. Drake, Jr. <fdrake@acm.org>
9.. sectionauthor:: Raymond Hettinger <python@rcn.com>
10
Ezio Melotti22170ed2010-11-20 09:57:27 +000011(If you are already familiar with the basic concepts of testing, you might want
12to skip to :ref:`the list of assert methods <assert-methods>`.)
Georg Brandl116aa622007-08-15 14:28:22 +000013
Georg Brandl116aa622007-08-15 14:28:22 +000014The Python unit testing framework, sometimes referred to as "PyUnit," is a
15Python language version of JUnit, by Kent Beck and Erich Gamma. JUnit is, in
16turn, a Java version of Kent's Smalltalk testing framework. Each is the de
17facto standard unit testing framework for its respective language.
18
19:mod:`unittest` supports test automation, sharing of setup and shutdown code for
20tests, aggregation of tests into collections, and independence of the tests from
21the reporting framework. The :mod:`unittest` module provides classes that make
22it easy to support these qualities for a set of tests.
23
24To achieve this, :mod:`unittest` supports some important concepts:
25
26test fixture
27 A :dfn:`test fixture` represents the preparation needed to perform one or more
28 tests, and any associate cleanup actions. This may involve, for example,
29 creating temporary or proxy databases, directories, or starting a server
30 process.
31
32test case
33 A :dfn:`test case` is the smallest unit of testing. It checks for a specific
34 response to a particular set of inputs. :mod:`unittest` provides a base class,
35 :class:`TestCase`, which may be used to create new test cases.
36
37test suite
38 A :dfn:`test suite` is a collection of test cases, test suites, or both. It is
39 used to aggregate tests that should be executed together.
40
41test runner
42 A :dfn:`test runner` is a component which orchestrates the execution of tests
43 and provides the outcome to the user. The runner may use a graphical interface,
44 a textual interface, or return a special value to indicate the results of
45 executing the tests.
46
47The test case and test fixture concepts are supported through the
48:class:`TestCase` and :class:`FunctionTestCase` classes; the former should be
49used when creating new tests, and the latter can be used when integrating
50existing test code with a :mod:`unittest`\ -driven framework. When building test
Benjamin Peterson52baa292009-03-24 00:56:30 +000051fixtures using :class:`TestCase`, the :meth:`~TestCase.setUp` and
52:meth:`~TestCase.tearDown` methods can be overridden to provide initialization
53and cleanup for the fixture. With :class:`FunctionTestCase`, existing functions
54can be passed to the constructor for these purposes. When the test is run, the
55fixture initialization is run first; if it succeeds, the cleanup method is run
56after the test has been executed, regardless of the outcome of the test. Each
57instance of the :class:`TestCase` will only be used to run a single test method,
58so a new fixture is created for each test.
Georg Brandl116aa622007-08-15 14:28:22 +000059
60Test suites are implemented by the :class:`TestSuite` class. This class allows
61individual tests and test suites to be aggregated; when the suite is executed,
Benjamin Peterson14a3dd72009-05-25 00:51:58 +000062all tests added directly to the suite and in "child" test suites are run.
Georg Brandl116aa622007-08-15 14:28:22 +000063
Benjamin Peterson52baa292009-03-24 00:56:30 +000064A test runner is an object that provides a single method,
65:meth:`~TestRunner.run`, which accepts a :class:`TestCase` or :class:`TestSuite`
66object as a parameter, and returns a result object. The class
67:class:`TestResult` is provided for use as the result object. :mod:`unittest`
68provides the :class:`TextTestRunner` as an example test runner which reports
69test results on the standard error stream by default. Alternate runners can be
70implemented for other environments (such as graphical environments) without any
71need to derive from a specific class.
Georg Brandl116aa622007-08-15 14:28:22 +000072
73
74.. seealso::
75
76 Module :mod:`doctest`
77 Another test-support module with a very different flavor.
78
Benjamin Petersonb48af542010-04-11 20:43:16 +000079 `unittest2: A backport of new unittest features for Python 2.4-2.6 <http://pypi.python.org/pypi/unittest2>`_
80 Many new features were added to unittest in Python 2.7, including test
81 discovery. unittest2 allows you to use these features with earlier
82 versions of Python.
83
Georg Brandl116aa622007-08-15 14:28:22 +000084 `Simple Smalltalk Testing: With Patterns <http://www.XProgramming.com/testfram.htm>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000085 Kent Beck's original paper on testing frameworks using the pattern shared
86 by :mod:`unittest`.
Georg Brandl116aa622007-08-15 14:28:22 +000087
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000088 `Nose <http://code.google.com/p/python-nose/>`_ and `py.test <http://pytest.org>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000089 Third-party unittest frameworks with a lighter-weight syntax for writing
90 tests. For example, ``assert func(10) == 42``.
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000091
Benjamin Petersonb48af542010-04-11 20:43:16 +000092 `The Python Testing Tools Taxonomy <http://pycheesecake.org/wiki/PythonTestingToolsTaxonomy>`_
93 An extensive list of Python testing tools including functional testing
94 frameworks and mock object libraries.
Benjamin Petersond2397752009-06-27 23:45:02 +000095
Benjamin Petersonb48af542010-04-11 20:43:16 +000096 `Testing in Python Mailing List <http://lists.idyll.org/listinfo/testing-in-python>`_
97 A special-interest-group for discussion of testing, and testing tools,
98 in Python.
Benjamin Petersond2397752009-06-27 23:45:02 +000099
Georg Brandl116aa622007-08-15 14:28:22 +0000100.. _unittest-minimal-example:
101
102Basic example
103-------------
104
105The :mod:`unittest` module provides a rich set of tools for constructing and
106running tests. This section demonstrates that a small subset of the tools
107suffice to meet the needs of most users.
108
109Here is a short script to test three functions from the :mod:`random` module::
110
111 import random
112 import unittest
113
114 class TestSequenceFunctions(unittest.TestCase):
115
116 def setUp(self):
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000117 self.seq = list(range(10))
Georg Brandl116aa622007-08-15 14:28:22 +0000118
Benjamin Peterson52baa292009-03-24 00:56:30 +0000119 def test_shuffle(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000120 # make sure the shuffled sequence does not lose any elements
121 random.shuffle(self.seq)
122 self.seq.sort()
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000123 self.assertEqual(self.seq, list(range(10)))
Georg Brandl116aa622007-08-15 14:28:22 +0000124
Benjamin Peterson847a4112010-03-14 15:04:17 +0000125 # should raise an exception for an immutable sequence
126 self.assertRaises(TypeError, random.shuffle, (1,2,3))
127
Benjamin Peterson52baa292009-03-24 00:56:30 +0000128 def test_choice(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000129 element = random.choice(self.seq)
Benjamin Peterson847a4112010-03-14 15:04:17 +0000130 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000131
Benjamin Peterson52baa292009-03-24 00:56:30 +0000132 def test_sample(self):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000133 with self.assertRaises(ValueError):
134 random.sample(self.seq, 20)
Georg Brandl116aa622007-08-15 14:28:22 +0000135 for element in random.sample(self.seq, 5):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000136 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000137
138 if __name__ == '__main__':
139 unittest.main()
140
Benjamin Peterson52baa292009-03-24 00:56:30 +0000141A testcase is created by subclassing :class:`unittest.TestCase`. The three
Georg Brandl116aa622007-08-15 14:28:22 +0000142individual tests are defined with methods whose names start with the letters
143``test``. This naming convention informs the test runner about which methods
144represent tests.
145
Benjamin Peterson52baa292009-03-24 00:56:30 +0000146The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
Michael Foord34c94622010-02-10 15:51:42 +0000147expected result; :meth:`~TestCase.assertTrue` to verify a condition; or
Benjamin Peterson52baa292009-03-24 00:56:30 +0000148:meth:`~TestCase.assertRaises` to verify that an expected exception gets raised.
149These methods are used instead of the :keyword:`assert` statement so the test
150runner can accumulate all test results and produce a report.
Georg Brandl116aa622007-08-15 14:28:22 +0000151
Benjamin Peterson52baa292009-03-24 00:56:30 +0000152When a :meth:`~TestCase.setUp` method is defined, the test runner will run that
153method prior to each test. Likewise, if a :meth:`~TestCase.tearDown` method is
154defined, the test runner will invoke that method after each test. In the
155example, :meth:`~TestCase.setUp` was used to create a fresh sequence for each
156test.
Georg Brandl116aa622007-08-15 14:28:22 +0000157
158The final block shows a simple way to run the tests. :func:`unittest.main`
Éric Araujo713d3032010-11-18 16:38:46 +0000159provides a command-line interface to the test script. When run from the command
Georg Brandl116aa622007-08-15 14:28:22 +0000160line, the above script produces an output that looks like this::
161
162 ...
163 ----------------------------------------------------------------------
164 Ran 3 tests in 0.000s
165
166 OK
167
168Instead of :func:`unittest.main`, there are other ways to run the tests with a
169finer level of control, less terse output, and no requirement to be run from the
170command line. For example, the last two lines may be replaced with::
171
172 suite = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)
173 unittest.TextTestRunner(verbosity=2).run(suite)
174
175Running the revised script from the interpreter or another script produces the
176following output::
177
Ezio Melottid59e44a2010-02-28 03:46:13 +0000178 test_choice (__main__.TestSequenceFunctions) ... ok
179 test_sample (__main__.TestSequenceFunctions) ... ok
180 test_shuffle (__main__.TestSequenceFunctions) ... ok
Georg Brandl116aa622007-08-15 14:28:22 +0000181
182 ----------------------------------------------------------------------
183 Ran 3 tests in 0.110s
184
185 OK
186
187The above examples show the most commonly used :mod:`unittest` features which
188are sufficient to meet many everyday testing needs. The remainder of the
189documentation explores the full feature set from first principles.
190
Benjamin Petersonb48af542010-04-11 20:43:16 +0000191
192.. _unittest-command-line-interface:
193
194Command Line Interface
195----------------------
196
197The unittest module can be used from the command line to run tests from
198modules, classes or even individual test methods::
199
200 python -m unittest test_module1 test_module2
201 python -m unittest test_module.TestClass
202 python -m unittest test_module.TestClass.test_method
203
204You can pass in a list with any combination of module names, and fully
205qualified class or method names.
206
207You can run tests with more detail (higher verbosity) by passing in the -v flag::
208
209 python -m unittest -v test_module
210
Michael Foord086f3082010-11-21 21:28:01 +0000211When executed without arguments :ref:`unittest-test-discovery` is started::
212
213 python -m unittest
214
Éric Araujo713d3032010-11-18 16:38:46 +0000215For a list of all the command-line options::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000216
217 python -m unittest -h
218
Georg Brandl67b21b72010-08-17 15:07:14 +0000219.. versionchanged:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000220 In earlier versions it was only possible to run individual test methods and
221 not modules or classes.
222
223
Éric Araujo713d3032010-11-18 16:38:46 +0000224failfast, catch and buffer command-line options
Benjamin Petersonb48af542010-04-11 20:43:16 +0000225~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
226
Éric Araujod3309df2010-11-21 03:09:17 +0000227:program:`unittest` supports these command-line options:
Benjamin Petersonb48af542010-04-11 20:43:16 +0000228
Éric Araujo713d3032010-11-18 16:38:46 +0000229.. program:: unittest
Benjamin Petersonb48af542010-04-11 20:43:16 +0000230
Éric Araujo713d3032010-11-18 16:38:46 +0000231.. cmdoption:: -b, --buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +0000232
Éric Araujo713d3032010-11-18 16:38:46 +0000233 The standard output and standard error streams are buffered during the test
234 run. Output during a passing test is discarded. Output is echoed normally
235 on test fail or error and is added to the failure messages.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000236
Éric Araujo713d3032010-11-18 16:38:46 +0000237.. cmdoption:: -c, --catch
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000238
Éric Araujo713d3032010-11-18 16:38:46 +0000239 Control-C during the test run waits for the current test to end and then
240 reports all the results so far. A second control-C raises the normal
241 :exc:`KeyboardInterrupt` exception.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000242
Éric Araujo713d3032010-11-18 16:38:46 +0000243 See `Signal Handling`_ for the functions that provide this functionality.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000244
Éric Araujo713d3032010-11-18 16:38:46 +0000245.. cmdoption:: -f, --failfast
246
247 Stop the test run on the first error or failure.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000248
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000249.. versionadded:: 3.2
Éric Araujo713d3032010-11-18 16:38:46 +0000250 The command-line options :option:`-c`, :option:`-b` and :option:`-f` were added.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000251
252The command line can also be used for test discovery, for running all of the
253tests in a project or just a subset.
254
255
256.. _unittest-test-discovery:
257
258Test Discovery
259--------------
260
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000261.. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000262
263Unittest supports simple test discovery. For a project's tests to be
264compatible with test discovery they must all be importable from the top level
265directory of the project (in other words, they must all be in Python packages).
266
267Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
Éric Araujo713d3032010-11-18 16:38:46 +0000268used from the command line. The basic command-line usage is::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000269
270 cd project_directory
271 python -m unittest discover
272
Michael Foord086f3082010-11-21 21:28:01 +0000273.. note::
274
275 As a shortcut, ``python -m unittest`` is the equivalent of
276 ``python -m unittest discover``. If you want to pass arguments to test
277 discovery the `discover` sub-command must be used explicitly.
278
Benjamin Petersonb48af542010-04-11 20:43:16 +0000279The ``discover`` sub-command has the following options:
280
Éric Araujo713d3032010-11-18 16:38:46 +0000281.. program:: unittest discover
282
283.. cmdoption:: -v, --verbose
284
285 Verbose output
286
287.. cmdoption:: -s directory
288
289 Directory to start discovery ('.' default)
290
291.. cmdoption:: -p pattern
292
293 Pattern to match test files ('test*.py' default)
294
295.. cmdoption:: -t directory
296
297 Top level directory of project (defaults to start directory)
Benjamin Petersonb48af542010-04-11 20:43:16 +0000298
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000299The :option:`-s`, :option:`-p`, and :option:`-t` options can be passed in
300as positional arguments in that order. The following two command lines
301are equivalent::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000302
303 python -m unittest discover -s project_directory -p '*_test.py'
304 python -m unittest discover project_directory '*_test.py'
305
Michael Foord16f3e902010-05-08 15:13:42 +0000306As well as being a path it is possible to pass a package name, for example
307``myproject.subpackage.test``, as the start directory. The package name you
308supply will then be imported and its location on the filesystem will be used
309as the start directory.
310
311.. caution::
312
Senthil Kumaran916bd382010-10-15 12:55:19 +0000313 Test discovery loads tests by importing them. Once test discovery has found
314 all the test files from the start directory you specify it turns the paths
315 into package names to import. For example :file:`foo/bar/baz.py` will be
Michael Foord16f3e902010-05-08 15:13:42 +0000316 imported as ``foo.bar.baz``.
317
318 If you have a package installed globally and attempt test discovery on
319 a different copy of the package then the import *could* happen from the
320 wrong place. If this happens test discovery will warn you and exit.
321
322 If you supply the start directory as a package name rather than a
323 path to a directory then discover assumes that whichever location it
324 imports from is the location you intended, so you will not get the
325 warning.
326
Benjamin Petersonb48af542010-04-11 20:43:16 +0000327Test modules and packages can customize test loading and discovery by through
328the `load_tests protocol`_.
329
330
Georg Brandl116aa622007-08-15 14:28:22 +0000331.. _organizing-tests:
332
333Organizing test code
334--------------------
335
336The basic building blocks of unit testing are :dfn:`test cases` --- single
337scenarios that must be set up and checked for correctness. In :mod:`unittest`,
338test cases are represented by instances of :mod:`unittest`'s :class:`TestCase`
339class. To make your own test cases you must write subclasses of
340:class:`TestCase`, or use :class:`FunctionTestCase`.
341
342An instance of a :class:`TestCase`\ -derived class is an object that can
343completely run a single test method, together with optional set-up and tidy-up
344code.
345
346The testing code of a :class:`TestCase` instance should be entirely self
347contained, such that it can be run either in isolation or in arbitrary
348combination with any number of other test cases.
349
Benjamin Peterson52baa292009-03-24 00:56:30 +0000350The simplest :class:`TestCase` subclass will simply override the
351:meth:`~TestCase.runTest` method in order to perform specific testing code::
Georg Brandl116aa622007-08-15 14:28:22 +0000352
353 import unittest
354
355 class DefaultWidgetSizeTestCase(unittest.TestCase):
356 def runTest(self):
357 widget = Widget('The widget')
358 self.assertEqual(widget.size(), (50, 50), 'incorrect default size')
359
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000360Note that in order to test something, we use the one of the :meth:`assert\*`
Benjamin Petersond2397752009-06-27 23:45:02 +0000361methods provided by the :class:`TestCase` base class. If the test fails, an
362exception will be raised, and :mod:`unittest` will identify the test case as a
363:dfn:`failure`. Any other exceptions will be treated as :dfn:`errors`. This
364helps you identify where the problem is: :dfn:`failures` are caused by incorrect
365results - a 5 where you expected a 6. :dfn:`Errors` are caused by incorrect
366code - e.g., a :exc:`TypeError` caused by an incorrect function call.
Georg Brandl116aa622007-08-15 14:28:22 +0000367
368The way to run a test case will be described later. For now, note that to
369construct an instance of such a test case, we call its constructor without
370arguments::
371
372 testCase = DefaultWidgetSizeTestCase()
373
374Now, such test cases can be numerous, and their set-up can be repetitive. In
375the above case, constructing a :class:`Widget` in each of 100 Widget test case
376subclasses would mean unsightly duplication.
377
378Luckily, we can factor out such set-up code by implementing a method called
Benjamin Peterson52baa292009-03-24 00:56:30 +0000379:meth:`~TestCase.setUp`, which the testing framework will automatically call for
380us when we run the test::
Georg Brandl116aa622007-08-15 14:28:22 +0000381
382 import unittest
383
384 class SimpleWidgetTestCase(unittest.TestCase):
385 def setUp(self):
386 self.widget = Widget('The widget')
387
388 class DefaultWidgetSizeTestCase(SimpleWidgetTestCase):
389 def runTest(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000390 self.assertEqual(self.widget.size(), (50,50),
391 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000392
393 class WidgetResizeTestCase(SimpleWidgetTestCase):
394 def runTest(self):
395 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000396 self.assertEqual(self.widget.size(), (100,150),
397 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000398
Benjamin Peterson52baa292009-03-24 00:56:30 +0000399If the :meth:`~TestCase.setUp` method raises an exception while the test is
400running, the framework will consider the test to have suffered an error, and the
401:meth:`~TestCase.runTest` method will not be executed.
Georg Brandl116aa622007-08-15 14:28:22 +0000402
Benjamin Peterson52baa292009-03-24 00:56:30 +0000403Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
404after the :meth:`~TestCase.runTest` method has been run::
Georg Brandl116aa622007-08-15 14:28:22 +0000405
406 import unittest
407
408 class SimpleWidgetTestCase(unittest.TestCase):
409 def setUp(self):
410 self.widget = Widget('The widget')
411
412 def tearDown(self):
413 self.widget.dispose()
414 self.widget = None
415
Benjamin Peterson52baa292009-03-24 00:56:30 +0000416If :meth:`~TestCase.setUp` succeeded, the :meth:`~TestCase.tearDown` method will
417be run whether :meth:`~TestCase.runTest` succeeded or not.
Georg Brandl116aa622007-08-15 14:28:22 +0000418
419Such a working environment for the testing code is called a :dfn:`fixture`.
420
421Often, many small test cases will use the same fixture. In this case, we would
422end up subclassing :class:`SimpleWidgetTestCase` into many small one-method
423classes such as :class:`DefaultWidgetSizeTestCase`. This is time-consuming and
Georg Brandl116aa622007-08-15 14:28:22 +0000424discouraging, so in the same vein as JUnit, :mod:`unittest` provides a simpler
425mechanism::
426
427 import unittest
428
429 class WidgetTestCase(unittest.TestCase):
430 def setUp(self):
431 self.widget = Widget('The widget')
432
433 def tearDown(self):
434 self.widget.dispose()
435 self.widget = None
436
Ezio Melottid59e44a2010-02-28 03:46:13 +0000437 def test_default_size(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000438 self.assertEqual(self.widget.size(), (50,50),
439 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000440
Ezio Melottid59e44a2010-02-28 03:46:13 +0000441 def test_resize(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000442 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000443 self.assertEqual(self.widget.size(), (100,150),
444 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000445
Benjamin Peterson52baa292009-03-24 00:56:30 +0000446Here we have not provided a :meth:`~TestCase.runTest` method, but have instead
447provided two different test methods. Class instances will now each run one of
Ezio Melottid59e44a2010-02-28 03:46:13 +0000448the :meth:`test_\*` methods, with ``self.widget`` created and destroyed
Benjamin Peterson52baa292009-03-24 00:56:30 +0000449separately for each instance. When creating an instance we must specify the
450test method it is to run. We do this by passing the method name in the
451constructor::
Georg Brandl116aa622007-08-15 14:28:22 +0000452
Ezio Melottid59e44a2010-02-28 03:46:13 +0000453 defaultSizeTestCase = WidgetTestCase('test_default_size')
454 resizeTestCase = WidgetTestCase('test_resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000455
456Test case instances are grouped together according to the features they test.
457:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
458represented by :mod:`unittest`'s :class:`TestSuite` class::
459
460 widgetTestSuite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000461 widgetTestSuite.addTest(WidgetTestCase('test_default_size'))
462 widgetTestSuite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000463
464For the ease of running tests, as we will see later, it is a good idea to
465provide in each test module a callable object that returns a pre-built test
466suite::
467
468 def suite():
469 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000470 suite.addTest(WidgetTestCase('test_default_size'))
471 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000472 return suite
473
474or even::
475
476 def suite():
Ezio Melottid59e44a2010-02-28 03:46:13 +0000477 tests = ['test_default_size', 'test_resize']
Georg Brandl116aa622007-08-15 14:28:22 +0000478
479 return unittest.TestSuite(map(WidgetTestCase, tests))
480
481Since it is a common pattern to create a :class:`TestCase` subclass with many
482similarly named test functions, :mod:`unittest` provides a :class:`TestLoader`
483class that can be used to automate the process of creating a test suite and
484populating it with individual tests. For example, ::
485
486 suite = unittest.TestLoader().loadTestsFromTestCase(WidgetTestCase)
487
Ezio Melottid59e44a2010-02-28 03:46:13 +0000488will create a test suite that will run ``WidgetTestCase.test_default_size()`` and
489``WidgetTestCase.test_resize``. :class:`TestLoader` uses the ``'test'`` method
Georg Brandl116aa622007-08-15 14:28:22 +0000490name prefix to identify test methods automatically.
491
Mark Dickinsonc48d8342009-02-01 14:18:10 +0000492Note that the order in which the various test cases will be run is
493determined by sorting the test function names with respect to the
494built-in ordering for strings.
Georg Brandl116aa622007-08-15 14:28:22 +0000495
496Often it is desirable to group suites of test cases together, so as to run tests
497for the whole system at once. This is easy, since :class:`TestSuite` instances
498can be added to a :class:`TestSuite` just as :class:`TestCase` instances can be
499added to a :class:`TestSuite`::
500
501 suite1 = module1.TheTestSuite()
502 suite2 = module2.TheTestSuite()
503 alltests = unittest.TestSuite([suite1, suite2])
504
505You can place the definitions of test cases and test suites in the same modules
506as the code they are to test (such as :file:`widget.py`), but there are several
507advantages to placing the test code in a separate module, such as
508:file:`test_widget.py`:
509
510* The test module can be run standalone from the command line.
511
512* The test code can more easily be separated from shipped code.
513
514* There is less temptation to change test code to fit the code it tests without
515 a good reason.
516
517* Test code should be modified much less frequently than the code it tests.
518
519* Tested code can be refactored more easily.
520
521* Tests for modules written in C must be in separate modules anyway, so why not
522 be consistent?
523
524* If the testing strategy changes, there is no need to change the source code.
525
526
527.. _legacy-unit-tests:
528
529Re-using old test code
530----------------------
531
532Some users will find that they have existing test code that they would like to
533run from :mod:`unittest`, without converting every old test function to a
534:class:`TestCase` subclass.
535
536For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
537This subclass of :class:`TestCase` can be used to wrap an existing test
538function. Set-up and tear-down functions can also be provided.
539
540Given the following test function::
541
542 def testSomething():
543 something = makeSomething()
544 assert something.name is not None
545 # ...
546
547one can create an equivalent test case instance as follows::
548
549 testcase = unittest.FunctionTestCase(testSomething)
550
551If there are additional set-up and tear-down methods that should be called as
552part of the test case's operation, they can also be provided like so::
553
554 testcase = unittest.FunctionTestCase(testSomething,
555 setUp=makeSomethingDB,
556 tearDown=deleteSomethingDB)
557
558To make migrating existing test suites easier, :mod:`unittest` supports tests
559raising :exc:`AssertionError` to indicate test failure. However, it is
560recommended that you use the explicit :meth:`TestCase.fail\*` and
561:meth:`TestCase.assert\*` methods instead, as future versions of :mod:`unittest`
562may treat :exc:`AssertionError` differently.
563
564.. note::
565
Benjamin Petersond2397752009-06-27 23:45:02 +0000566 Even though :class:`FunctionTestCase` can be used to quickly convert an
567 existing test base over to a :mod:`unittest`\ -based system, this approach is
568 not recommended. Taking the time to set up proper :class:`TestCase`
569 subclasses will make future test refactorings infinitely easier.
Georg Brandl116aa622007-08-15 14:28:22 +0000570
Benjamin Peterson52baa292009-03-24 00:56:30 +0000571In some cases, the existing tests may have been written using the :mod:`doctest`
572module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
573automatically build :class:`unittest.TestSuite` instances from the existing
574:mod:`doctest`\ -based tests.
575
Georg Brandl116aa622007-08-15 14:28:22 +0000576
Benjamin Peterson5254c042009-03-23 22:25:03 +0000577.. _unittest-skipping:
578
579Skipping tests and expected failures
580------------------------------------
581
Michael Foordf5c851a2010-02-05 21:48:03 +0000582.. versionadded:: 3.1
583
Benjamin Peterson5254c042009-03-23 22:25:03 +0000584Unittest supports skipping individual test methods and even whole classes of
585tests. In addition, it supports marking a test as a "expected failure," a test
586that is broken and will fail, but shouldn't be counted as a failure on a
587:class:`TestResult`.
588
589Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
590or one of its conditional variants.
591
592Basic skipping looks like this: ::
593
594 class MyTestCase(unittest.TestCase):
595
596 @unittest.skip("demonstrating skipping")
597 def test_nothing(self):
598 self.fail("shouldn't happen")
599
Benjamin Petersond2397752009-06-27 23:45:02 +0000600 @unittest.skipIf(mylib.__version__ < (1, 3),
601 "not supported in this library version")
Benjamin Petersonded31c42009-03-30 15:04:16 +0000602 def test_format(self):
603 # Tests that work for only a certain version of the library.
604 pass
605
606 @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
607 def test_windows_support(self):
608 # windows specific testing code
609 pass
610
Benjamin Peterson5254c042009-03-23 22:25:03 +0000611This is the output of running the example above in verbose mode: ::
612
Benjamin Petersonded31c42009-03-30 15:04:16 +0000613 test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000614 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
Benjamin Petersonded31c42009-03-30 15:04:16 +0000615 test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000616
617 ----------------------------------------------------------------------
Benjamin Petersonded31c42009-03-30 15:04:16 +0000618 Ran 3 tests in 0.005s
619
620 OK (skipped=3)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000621
622Classes can be skipped just like methods: ::
623
624 @skip("showing class skipping")
625 class MySkippedTestCase(unittest.TestCase):
626 def test_not_run(self):
627 pass
628
Benjamin Peterson52baa292009-03-24 00:56:30 +0000629:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
630that needs to be set up is not available.
631
Benjamin Peterson5254c042009-03-23 22:25:03 +0000632Expected failures use the :func:`expectedFailure` decorator. ::
633
634 class ExpectedFailureTestCase(unittest.TestCase):
635 @unittest.expectedFailure
636 def test_fail(self):
637 self.assertEqual(1, 0, "broken")
638
639It's easy to roll your own skipping decorators by making a decorator that calls
640:func:`skip` on the test when it wants it to be skipped. This decorator skips
641the test unless the passed object has a certain attribute: ::
642
643 def skipUnlessHasattr(obj, attr):
644 if hasattr(obj, attr):
645 return lambda func: func
646 return unittest.skip("{0!r} doesn't have {1!r}".format(obj, attr))
647
648The following decorators implement test skipping and expected failures:
649
Georg Brandl8a1caa22010-07-29 16:01:11 +0000650.. decorator:: skip(reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000651
652 Unconditionally skip the decorated test. *reason* should describe why the
653 test is being skipped.
654
Georg Brandl8a1caa22010-07-29 16:01:11 +0000655.. decorator:: skipIf(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000656
657 Skip the decorated test if *condition* is true.
658
Georg Brandl8a1caa22010-07-29 16:01:11 +0000659.. decorator:: skipUnless(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000660
Georg Brandl6faee4e2010-09-21 14:48:28 +0000661 Skip the decorated test unless *condition* is true.
Benjamin Peterson5254c042009-03-23 22:25:03 +0000662
Georg Brandl8a1caa22010-07-29 16:01:11 +0000663.. decorator:: expectedFailure
Benjamin Peterson5254c042009-03-23 22:25:03 +0000664
665 Mark the test as an expected failure. If the test fails when run, the test
666 is not counted as a failure.
667
Benjamin Petersonb48af542010-04-11 20:43:16 +0000668Skipped tests will not have :meth:`setUp` or :meth:`tearDown` run around them.
669Skipped classes will not have :meth:`setUpClass` or :meth:`tearDownClass` run.
670
Benjamin Peterson5254c042009-03-23 22:25:03 +0000671
Georg Brandl116aa622007-08-15 14:28:22 +0000672.. _unittest-contents:
673
674Classes and functions
675---------------------
676
Benjamin Peterson52baa292009-03-24 00:56:30 +0000677This section describes in depth the API of :mod:`unittest`.
678
679
680.. _testcase-objects:
681
682Test cases
683~~~~~~~~~~
Georg Brandl116aa622007-08-15 14:28:22 +0000684
Georg Brandl7f01a132009-09-16 15:58:14 +0000685.. class:: TestCase(methodName='runTest')
Georg Brandl116aa622007-08-15 14:28:22 +0000686
687 Instances of the :class:`TestCase` class represent the smallest testable units
688 in the :mod:`unittest` universe. This class is intended to be used as a base
689 class, with specific tests being implemented by concrete subclasses. This class
690 implements the interface needed by the test runner to allow it to drive the
691 test, and methods that the test code can use to check for and report various
692 kinds of failure.
693
694 Each instance of :class:`TestCase` will run a single test method: the method
695 named *methodName*. If you remember, we had an earlier example that went
696 something like this::
697
698 def suite():
699 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000700 suite.addTest(WidgetTestCase('test_default_size'))
701 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000702 return suite
703
704 Here, we create two instances of :class:`WidgetTestCase`, each of which runs a
705 single test.
706
Benjamin Peterson52baa292009-03-24 00:56:30 +0000707 *methodName* defaults to :meth:`runTest`.
708
709 :class:`TestCase` instances provide three groups of methods: one group used
710 to run the test, another used by the test implementation to check conditions
711 and report failures, and some inquiry methods allowing information about the
712 test itself to be gathered.
713
714 Methods in the first group (running the test) are:
715
716
717 .. method:: setUp()
718
719 Method called to prepare the test fixture. This is called immediately
720 before calling the test method; any exception raised by this method will
721 be considered an error rather than a test failure. The default
722 implementation does nothing.
723
724
725 .. method:: tearDown()
726
727 Method called immediately after the test method has been called and the
728 result recorded. This is called even if the test method raised an
729 exception, so the implementation in subclasses may need to be particularly
730 careful about checking internal state. Any exception raised by this
731 method will be considered an error rather than a test failure. This
732 method will only be called if the :meth:`setUp` succeeds, regardless of
733 the outcome of the test method. The default implementation does nothing.
734
735
Benjamin Petersonb48af542010-04-11 20:43:16 +0000736 .. method:: setUpClass()
737
738 A class method called before tests in an individual class run.
739 ``setUpClass`` is called with the class as the only argument
740 and must be decorated as a :func:`classmethod`::
741
742 @classmethod
743 def setUpClass(cls):
744 ...
745
746 See `Class and Module Fixtures`_ for more details.
747
748 .. versionadded:: 3.2
749
750
751 .. method:: tearDownClass()
752
753 A class method called after tests in an individual class have run.
754 ``tearDownClass`` is called with the class as the only argument
755 and must be decorated as a :meth:`classmethod`::
756
757 @classmethod
758 def tearDownClass(cls):
759 ...
760
761 See `Class and Module Fixtures`_ for more details.
762
763 .. versionadded:: 3.2
764
765
Georg Brandl7f01a132009-09-16 15:58:14 +0000766 .. method:: run(result=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000767
768 Run the test, collecting the result into the test result object passed as
Ezio Melotti75b2a5e2010-11-20 10:13:45 +0000769 *result*. If *result* is omitted or ``None``, a temporary result
Alexandre Vassalotti260484d2009-07-17 11:43:26 +0000770 object is created (by calling the :meth:`defaultTestResult` method) and
771 used. The result object is not returned to :meth:`run`'s caller.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000772
773 The same effect may be had by simply calling the :class:`TestCase`
774 instance.
775
776
Benjamin Petersone549ead2009-03-28 21:42:05 +0000777 .. method:: skipTest(reason)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000778
Stefan Kraha5bf3f52010-05-19 16:09:41 +0000779 Calling this during a test method or :meth:`setUp` skips the current
Benjamin Peterson52baa292009-03-24 00:56:30 +0000780 test. See :ref:`unittest-skipping` for more information.
781
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000782 .. versionadded:: 3.1
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000783
Benjamin Peterson52baa292009-03-24 00:56:30 +0000784
785 .. method:: debug()
786
787 Run the test without collecting the result. This allows exceptions raised
788 by the test to be propagated to the caller, and can be used to support
789 running tests under a debugger.
790
Ezio Melotti22170ed2010-11-20 09:57:27 +0000791 .. _assert-methods:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000792
Ezio Melotti4370b302010-11-03 20:39:14 +0000793 The :class:`TestCase` class provides a number of methods to check for and
794 report failures, such as:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000795
Ezio Melotti4370b302010-11-03 20:39:14 +0000796 +-----------------------------------------+-----------------------------+---------------+
797 | Method | Checks that | New in |
798 +=========================================+=============================+===============+
799 | :meth:`assertEqual(a, b) | ``a == b`` | |
800 | <TestCase.assertEqual>` | | |
801 +-----------------------------------------+-----------------------------+---------------+
802 | :meth:`assertNotEqual(a, b) | ``a != b`` | |
803 | <TestCase.assertNotEqual>` | | |
804 +-----------------------------------------+-----------------------------+---------------+
805 | :meth:`assertTrue(x) | ``bool(x) is True`` | |
806 | <TestCase.assertTrue>` | | |
807 +-----------------------------------------+-----------------------------+---------------+
808 | :meth:`assertFalse(x) | ``bool(x) is False`` | |
809 | <TestCase.assertFalse>` | | |
810 +-----------------------------------------+-----------------------------+---------------+
811 | :meth:`assertIs(a, b) | ``a is b`` | 3.1 |
812 | <TestCase.assertIs>` | | |
813 +-----------------------------------------+-----------------------------+---------------+
814 | :meth:`assertIsNot(a, b) | ``a is not b`` | 3.1 |
815 | <TestCase.assertIsNot>` | | |
816 +-----------------------------------------+-----------------------------+---------------+
817 | :meth:`assertIsNone(x) | ``x is None`` | 3.1 |
818 | <TestCase.assertIsNone>` | | |
819 +-----------------------------------------+-----------------------------+---------------+
820 | :meth:`assertIsNotNone(x) | ``x is not None`` | 3.1 |
821 | <TestCase.assertIsNotNone>` | | |
822 +-----------------------------------------+-----------------------------+---------------+
823 | :meth:`assertIn(a, b) | ``a in b`` | 3.1 |
824 | <TestCase.assertIn>` | | |
825 +-----------------------------------------+-----------------------------+---------------+
826 | :meth:`assertNotIn(a, b) | ``a not in b`` | 3.1 |
827 | <TestCase.assertNotIn>` | | |
828 +-----------------------------------------+-----------------------------+---------------+
829 | :meth:`assertIsInstance(a, b) | ``isinstance(a, b)`` | 3.2 |
830 | <TestCase.assertIsInstance>` | | |
831 +-----------------------------------------+-----------------------------+---------------+
832 | :meth:`assertNotIsInstance(a, b) | ``not isinstance(a, b)`` | 3.2 |
833 | <TestCase.assertNotIsInstance>` | | |
834 +-----------------------------------------+-----------------------------+---------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000835
Ezio Melotti22170ed2010-11-20 09:57:27 +0000836 All the assert methods (except :meth:`assertRaises`,
837 :meth:`assertRaisesRegexp`, :meth:`assertWarns`, :meth:`assertWarnsRegexp`)
838 accept a *msg* argument that, if specified, is used as the error message on
839 failure (see also :data:`longMessage`).
Benjamin Peterson52baa292009-03-24 00:56:30 +0000840
Georg Brandl7f01a132009-09-16 15:58:14 +0000841 .. method:: assertEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000842
843 Test that *first* and *second* are equal. If the values do not compare
Ezio Melotti4841fd62010-11-05 15:43:40 +0000844 equal, the test will fail.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000845
846 In addition, if *first* and *second* are the exact same type and one of
Michael Foord02834952010-02-08 23:10:39 +0000847 list, tuple, dict, set, frozenset or str or any type that a subclass
848 registers with :meth:`addTypeEqualityFunc` the type specific equality
849 function will be called in order to generate a more useful default
Ezio Melotti22170ed2010-11-20 09:57:27 +0000850 error message (see also the :ref:`list of type-specific methods
851 <type-specific-methods>`).
Ezio Melotti4841fd62010-11-05 15:43:40 +0000852
Raymond Hettinger35a88362009-04-09 00:08:24 +0000853 .. versionchanged:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000854 Added the automatic calling of type specific equality function.
855
Michael Foord28a817e2010-02-09 00:03:57 +0000856 .. versionchanged:: 3.2
857 :meth:`assertMultiLineEqual` added as the default type equality
858 function for comparing strings.
Michael Foord02834952010-02-08 23:10:39 +0000859
Benjamin Peterson52baa292009-03-24 00:56:30 +0000860
Georg Brandl7f01a132009-09-16 15:58:14 +0000861 .. method:: assertNotEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000862
863 Test that *first* and *second* are not equal. If the values do compare
Ezio Melotti4841fd62010-11-05 15:43:40 +0000864 equal, the test will fail.
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000865
Ezio Melotti4370b302010-11-03 20:39:14 +0000866 .. method:: assertTrue(expr, msg=None)
Ezio Melotti4841fd62010-11-05 15:43:40 +0000867 assertFalse(expr, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000868
Ezio Melotti4841fd62010-11-05 15:43:40 +0000869 Test that *expr* is true (or false).
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000870
Ezio Melotti4841fd62010-11-05 15:43:40 +0000871 Note that this is equivalent to ``bool(expr) is True`` and not to ``expr
872 is True`` (use ``assertIs(expr, True)`` for the latter). This method
873 should also be avoided when more specific methods are available (e.g.
874 ``assertEqual(a, b)`` instead of ``assertTrue(a == b)``), because they
875 provide a better error message in case of failure.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000876
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000877
Ezio Melotti9794a262010-11-04 14:52:13 +0000878 .. method:: assertIs(first, second, msg=None)
879 assertIsNot(first, second, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000880
Ezio Melotti9794a262010-11-04 14:52:13 +0000881 Test that *first* and *second* evaluate (or don't evaluate) to the same object.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000882
Raymond Hettinger35a88362009-04-09 00:08:24 +0000883 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000884
885
Ezio Melotti4370b302010-11-03 20:39:14 +0000886 .. method:: assertIsNone(expr, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000887 assertIsNotNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000888
Ezio Melotti9794a262010-11-04 14:52:13 +0000889 Test that *expr* is (or is not) None.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000890
Ezio Melotti4370b302010-11-03 20:39:14 +0000891 .. versionadded:: 3.1
Benjamin Petersonb48af542010-04-11 20:43:16 +0000892
893
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000894 .. method:: assertIn(first, second, msg=None)
895 assertNotIn(first, second, msg=None)
896
Ezio Melotti4841fd62010-11-05 15:43:40 +0000897 Test that *first* is (or is not) in *second*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000898
Raymond Hettinger35a88362009-04-09 00:08:24 +0000899 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000900
901
Ezio Melotti9c02c2f2010-11-03 20:45:31 +0000902 .. method:: assertIsInstance(obj, cls, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000903 assertNotIsInstance(obj, cls, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000904
Ezio Melotti9794a262010-11-04 14:52:13 +0000905 Test that *obj* is (or is not) an instance of *cls* (which can be a
906 class or a tuple of classes, as supported by :func:`isinstance`).
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000907
Ezio Melotti4370b302010-11-03 20:39:14 +0000908 .. versionadded:: 3.2
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000909
910
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000911
Ezio Melotti4370b302010-11-03 20:39:14 +0000912 It is also possible to check that exceptions and warnings are raised using
913 the following methods:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000914
Ezio Melotti4370b302010-11-03 20:39:14 +0000915 +---------------------------------------------------------+--------------------------------------+------------+
916 | Method | Checks that | New in |
917 +=========================================================+======================================+============+
918 | :meth:`assertRaises(exc, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `exc` | |
919 | <TestCase.assertRaises>` | | |
920 +---------------------------------------------------------+--------------------------------------+------------+
921 | :meth:`assertRaisesRegexp(exc, re, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `exc` | 3.1 |
922 | <TestCase.assertRaisesRegexp>` | and the message matches `re` | |
923 +---------------------------------------------------------+--------------------------------------+------------+
924 | :meth:`assertWarns(warn, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `warn` | 3.2 |
925 | <TestCase.assertWarns>` | | |
926 +---------------------------------------------------------+--------------------------------------+------------+
927 | :meth:`assertWarnsRegexp(warn, re, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `warn` | 3.2 |
928 | <TestCase.assertWarnsRegexp>` | and the message matches `re` | |
929 +---------------------------------------------------------+--------------------------------------+------------+
Benjamin Peterson52baa292009-03-24 00:56:30 +0000930
Georg Brandl7f01a132009-09-16 15:58:14 +0000931 .. method:: assertRaises(exception, callable, *args, **kwds)
Georg Brandl7f01a132009-09-16 15:58:14 +0000932 assertRaises(exception)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000933
934 Test that an exception is raised when *callable* is called with any
935 positional or keyword arguments that are also passed to
936 :meth:`assertRaises`. The test passes if *exception* is raised, is an
937 error if another exception is raised, or fails if no exception is raised.
938 To catch any of a group of exceptions, a tuple containing the exception
939 classes may be passed as *exception*.
940
Georg Brandl7f01a132009-09-16 15:58:14 +0000941 If only the *exception* argument is given, returns a context manager so
942 that the code under test can be written inline rather than as a function::
Benjamin Petersonded31c42009-03-30 15:04:16 +0000943
Michael Foord41531f22010-02-05 21:13:40 +0000944 with self.assertRaises(SomeException):
Benjamin Petersonded31c42009-03-30 15:04:16 +0000945 do_something()
946
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000947 The context manager will store the caught exception object in its
Ezio Melotti49008232010-02-08 21:57:48 +0000948 :attr:`exception` attribute. This can be useful if the intention
Michael Foord41531f22010-02-05 21:13:40 +0000949 is to perform additional checks on the exception raised::
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000950
Georg Brandl8a1caa22010-07-29 16:01:11 +0000951 with self.assertRaises(SomeException) as cm:
952 do_something()
Michael Foord41531f22010-02-05 21:13:40 +0000953
Georg Brandl8a1caa22010-07-29 16:01:11 +0000954 the_exception = cm.exception
955 self.assertEqual(the_exception.error_code, 3)
Michael Foord41531f22010-02-05 21:13:40 +0000956
Ezio Melotti49008232010-02-08 21:57:48 +0000957 .. versionchanged:: 3.1
Benjamin Petersonded31c42009-03-30 15:04:16 +0000958 Added the ability to use :meth:`assertRaises` as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000959
Ezio Melotti49008232010-02-08 21:57:48 +0000960 .. versionchanged:: 3.2
961 Added the :attr:`exception` attribute.
962
Benjamin Peterson52baa292009-03-24 00:56:30 +0000963
Ezio Melotti327433f2010-11-03 20:51:17 +0000964 .. method:: assertRaisesRegexp(exception, regexp, callable, *args, **kwds)
965 assertRaisesRegexp(exception, regexp)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000966
967 Like :meth:`assertRaises` but also tests that *regexp* matches
968 on the string representation of the raised exception. *regexp* may be
969 a regular expression object or a string containing a regular expression
970 suitable for use by :func:`re.search`. Examples::
971
972 self.assertRaisesRegexp(ValueError, 'invalid literal for.*XYZ$',
973 int, 'XYZ')
974
975 or::
976
977 with self.assertRaisesRegexp(ValueError, 'literal'):
978 int('XYZ')
979
Raymond Hettinger35a88362009-04-09 00:08:24 +0000980 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000981
982
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +0000983 .. method:: assertWarns(warning, callable, *args, **kwds)
984 assertWarns(warning)
985
986 Test that a warning is triggered when *callable* is called with any
987 positional or keyword arguments that are also passed to
988 :meth:`assertWarns`. The test passes if *warning* is triggered and
989 fails if it isn't. Also, any unexpected exception is an error.
990 To catch any of a group of warnings, a tuple containing the warning
991 classes may be passed as *warnings*.
992
993 If only the *warning* argument is given, returns a context manager so
994 that the code under test can be written inline rather than as a function::
995
996 with self.assertWarns(SomeWarning):
997 do_something()
998
999 The context manager will store the caught warning object in its
1000 :attr:`warning` attribute, and the source line which triggered the
1001 warnings in the :attr:`filename` and :attr:`lineno` attributes.
1002 This can be useful if the intention is to perform additional checks
1003 on the exception raised::
1004
1005 with self.assertWarns(SomeWarning) as cm:
1006 do_something()
1007
1008 self.assertIn('myfile.py', cm.filename)
1009 self.assertEqual(320, cm.lineno)
1010
1011 This method works regardless of the warning filters in place when it
1012 is called.
1013
1014 .. versionadded:: 3.2
1015
1016
Ezio Melotti327433f2010-11-03 20:51:17 +00001017 .. method:: assertWarnsRegexp(warning, regexp, callable, *args, **kwds)
1018 assertWarnsRegexp(warning, regexp)
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001019
1020 Like :meth:`assertWarns` but also tests that *regexp* matches on the
1021 message of the triggered warning. *regexp* may be a regular expression
1022 object or a string containing a regular expression suitable for use
1023 by :func:`re.search`. Example::
1024
1025 self.assertWarnsRegexp(DeprecationWarning,
1026 r'legacy_function\(\) is deprecated',
1027 legacy_function, 'XYZ')
1028
1029 or::
1030
1031 with self.assertWarnsRegexp(RuntimeWarning, 'unsafe frobnicating'):
1032 frobnicate('/etc/passwd')
1033
1034 .. versionadded:: 3.2
1035
1036
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001037
Ezio Melotti4370b302010-11-03 20:39:14 +00001038 There are also other methods used to perform more specific checks, such as:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001039
Ezio Melotti4370b302010-11-03 20:39:14 +00001040 +---------------------------------------+--------------------------------+--------------+
1041 | Method | Checks that | New in |
1042 +=======================================+================================+==============+
1043 | :meth:`assertAlmostEqual(a, b) | ``round(a-b, 7) == 0`` | |
1044 | <TestCase.assertAlmostEqual>` | | |
1045 +---------------------------------------+--------------------------------+--------------+
1046 | :meth:`assertNotAlmostEqual(a, b) | ``round(a-b, 7) != 0`` | |
1047 | <TestCase.assertNotAlmostEqual>` | | |
1048 +---------------------------------------+--------------------------------+--------------+
1049 | :meth:`assertGreater(a, b) | ``a > b`` | 3.1 |
1050 | <TestCase.assertGreater>` | | |
1051 +---------------------------------------+--------------------------------+--------------+
1052 | :meth:`assertGreaterEqual(a, b) | ``a >= b`` | 3.1 |
1053 | <TestCase.assertGreaterEqual>` | | |
1054 +---------------------------------------+--------------------------------+--------------+
1055 | :meth:`assertLess(a, b) | ``a < b`` | 3.1 |
1056 | <TestCase.assertLess>` | | |
1057 +---------------------------------------+--------------------------------+--------------+
1058 | :meth:`assertLessEqual(a, b) | ``a <= b`` | 3.1 |
1059 | <TestCase.assertLessEqual>` | | |
1060 +---------------------------------------+--------------------------------+--------------+
1061 | :meth:`assertRegexpMatches(s, re) | ``regex.search(s)`` | 3.1 |
1062 | <TestCase.assertRegexpMatches>` | | |
1063 +---------------------------------------+--------------------------------+--------------+
1064 | :meth:`assertNotRegexpMatches(s, re) | ``not regex.search(s)`` | 3.2 |
1065 | <TestCase.assertNotRegexpMatches>` | | |
1066 +---------------------------------------+--------------------------------+--------------+
1067 | :meth:`assertDictContainsSubset(a, b) | all the key/value pairs | 3.1 |
1068 | <TestCase.assertDictContainsSubset>` | in `a` exist in `b` | |
1069 +---------------------------------------+--------------------------------+--------------+
1070 | :meth:`assertItemsEqual(a, b) | `a` and `b` have the same | 3.2 |
1071 | <TestCase.assertItemsEqual>` | elements in the same number, | |
1072 | | regardless of their order | |
1073 +---------------------------------------+--------------------------------+--------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001074
1075
Ezio Melotti4370b302010-11-03 20:39:14 +00001076 .. method:: assertAlmostEqual(first, second, places=7, msg=None, delta=None)
Ezio Melotti4841fd62010-11-05 15:43:40 +00001077 assertNotAlmostEqual(first, second, places=7, msg=None, delta=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001078
Ezio Melotti4841fd62010-11-05 15:43:40 +00001079 Test that *first* and *second* are approximately (or not approximately)
1080 equal by computing the difference, rounding to the given number of
1081 decimal *places* (default 7), and comparing to zero. Note that these
1082 methods round the values to the given number of *decimal places* (i.e.
1083 like the :func:`round` function) and not *significant digits*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001084
Ezio Melotti4370b302010-11-03 20:39:14 +00001085 If *delta* is supplied instead of *places* then the difference
Ezio Melotti4841fd62010-11-05 15:43:40 +00001086 between *first* and *second* must be less (or more) than *delta*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001087
Ezio Melotti4370b302010-11-03 20:39:14 +00001088 Supplying both *delta* and *places* raises a ``TypeError``.
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001089
Ezio Melotti4370b302010-11-03 20:39:14 +00001090 .. versionchanged:: 3.2
Ezio Melotti4841fd62010-11-05 15:43:40 +00001091 assertAlmostEqual automatically considers almost equal objects that compare equal.
1092 assertNotAlmostEqual automatically fails if the objects compare equal.
Ezio Melotti4370b302010-11-03 20:39:14 +00001093 Added the ``delta`` keyword argument.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001094
Ezio Melotti4370b302010-11-03 20:39:14 +00001095
Ezio Melotti4370b302010-11-03 20:39:14 +00001096 .. method:: assertGreater(first, second, msg=None)
1097 assertGreaterEqual(first, second, msg=None)
1098 assertLess(first, second, msg=None)
1099 assertLessEqual(first, second, msg=None)
1100
1101 Test that *first* is respectively >, >=, < or <= than *second* depending
Ezio Melotti4841fd62010-11-05 15:43:40 +00001102 on the method name. If not, the test will fail::
Ezio Melotti4370b302010-11-03 20:39:14 +00001103
1104 >>> self.assertGreaterEqual(3, 4)
1105 AssertionError: "3" unexpectedly not greater than or equal to "4"
1106
1107 .. versionadded:: 3.1
1108
1109
1110 .. method:: assertRegexpMatches(text, regexp, msg=None)
Ezio Melotti4841fd62010-11-05 15:43:40 +00001111 assertNotRegexpMatches(text, regexp, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001112
Ezio Melotti4841fd62010-11-05 15:43:40 +00001113 Test that a *regexp* search matches (or does not match) *text*. In case
1114 of failure, the error message will include the pattern and the *text* (or
1115 the pattern and the part of *text* that unexpectedly matched). *regexp*
Ezio Melotti4370b302010-11-03 20:39:14 +00001116 may be a regular expression object or a string containing a regular
1117 expression suitable for use by :func:`re.search`.
1118
Ezio Melotti4841fd62010-11-05 15:43:40 +00001119 .. versionadded:: 3.1 :meth:`~TestCase.assertRegexpMatches`
1120 .. versionadded:: 3.2 :meth:`~TestCase.assertNotRegexpMatches`
Ezio Melotti4370b302010-11-03 20:39:14 +00001121
1122
1123 .. method:: assertDictContainsSubset(expected, actual, msg=None)
1124
1125 Tests whether the key/value pairs in dictionary *actual* are a
1126 superset of those in *expected*. If not, an error message listing
1127 the missing keys and mismatched values is generated.
1128
Ezio Melotti4370b302010-11-03 20:39:14 +00001129 .. versionadded:: 3.1
1130
1131
1132 .. method:: assertItemsEqual(actual, expected, msg=None)
1133
1134 Test that sequence *expected* contains the same elements as *actual*,
1135 regardless of their order. When they don't, an error message listing the
1136 differences between the sequences will be generated.
1137
1138 Duplicate elements are *not* ignored when comparing *actual* and
1139 *expected*. It verifies if each element has the same count in both
1140 sequences. It is the equivalent of ``assertEqual(sorted(expected),
1141 sorted(actual))`` but it works with sequences of unhashable objects as
1142 well.
1143
Ezio Melotti4370b302010-11-03 20:39:14 +00001144 .. versionadded:: 3.2
1145
1146
1147 .. method:: assertSameElements(actual, expected, msg=None)
1148
1149 Test that sequence *expected* contains the same elements as *actual*,
1150 regardless of their order. When they don't, an error message listing
1151 the differences between the sequences will be generated.
1152
1153 Duplicate elements are ignored when comparing *actual* and *expected*.
1154 It is the equivalent of ``assertEqual(set(expected), set(actual))``
1155 but it works with sequences of unhashable objects as well. Because
1156 duplicates are ignored, this method has been deprecated in favour of
1157 :meth:`assertItemsEqual`.
1158
Ezio Melotti4370b302010-11-03 20:39:14 +00001159 .. versionadded:: 3.1
1160 .. deprecated:: 3.2
1161
1162
Ezio Melotti22170ed2010-11-20 09:57:27 +00001163 .. _type-specific-methods:
Ezio Melotti4370b302010-11-03 20:39:14 +00001164
Ezio Melotti22170ed2010-11-20 09:57:27 +00001165 The :meth:`assertEqual` method dispatches the equality check for objects of
1166 the same type to different type-specific methods. These methods are already
1167 implemented for most of the built-in types, but it's also possible to
1168 register new methods using :meth:`addTypeEqualityFunc`:
1169
1170 .. method:: addTypeEqualityFunc(typeobj, function)
1171
1172 Registers a type-specific method called by :meth:`assertEqual` to check
1173 if two objects of exactly the same *typeobj* (not subclasses) compare
1174 equal. *function* must take two positional arguments and a third msg=None
1175 keyword argument just as :meth:`assertEqual` does. It must raise
1176 :data:`self.failureException(msg) <failureException>` when inequality
1177 between the first two parameters is detected -- possibly providing useful
1178 information and explaining the inequalities in details in the error
1179 message.
1180
1181 .. versionadded:: 3.1
1182
1183 The list of type-specific methods automatically used by
1184 :meth:`~TestCase.assertEqual` are summarized in the following table. Note
1185 that it's usually not necessary to invoke these methods directly.
Ezio Melotti4370b302010-11-03 20:39:14 +00001186
1187 +-----------------------------------------+-----------------------------+--------------+
1188 | Method | Used to compare | New in |
1189 +=========================================+=============================+==============+
1190 | :meth:`assertMultiLineEqual(a, b) | strings | 3.1 |
1191 | <TestCase.assertMultiLineEqual>` | | |
1192 +-----------------------------------------+-----------------------------+--------------+
1193 | :meth:`assertSequenceEqual(a, b) | sequences | 3.1 |
1194 | <TestCase.assertSequenceEqual>` | | |
1195 +-----------------------------------------+-----------------------------+--------------+
1196 | :meth:`assertListEqual(a, b) | lists | 3.1 |
1197 | <TestCase.assertListEqual>` | | |
1198 +-----------------------------------------+-----------------------------+--------------+
1199 | :meth:`assertTupleEqual(a, b) | tuples | 3.1 |
1200 | <TestCase.assertTupleEqual>` | | |
1201 +-----------------------------------------+-----------------------------+--------------+
1202 | :meth:`assertSetEqual(a, b) | sets or frozensets | 3.1 |
1203 | <TestCase.assertSetEqual>` | | |
1204 +-----------------------------------------+-----------------------------+--------------+
1205 | :meth:`assertDictEqual(a, b) | dicts | 3.1 |
1206 | <TestCase.assertDictEqual>` | | |
1207 +-----------------------------------------+-----------------------------+--------------+
1208
1209
1210
Ezio Melotti9c02c2f2010-11-03 20:45:31 +00001211 .. method:: assertMultiLineEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001212
1213 Test that the multiline string *first* is equal to the string *second*.
1214 When not equal a diff of the two strings highlighting the differences
1215 will be included in the error message. This method is used by default
1216 when comparing strings with :meth:`assertEqual`.
1217
Ezio Melotti4370b302010-11-03 20:39:14 +00001218 .. versionadded:: 3.1
1219
1220
1221 .. method:: assertSequenceEqual(seq1, seq2, msg=None, seq_type=None)
1222
1223 Tests that two sequences are equal. If a *seq_type* is supplied, both
1224 *seq1* and *seq2* must be instances of *seq_type* or a failure will
1225 be raised. If the sequences are different an error message is
1226 constructed that shows the difference between the two.
1227
Ezio Melotti22170ed2010-11-20 09:57:27 +00001228 This method is not called directly by :meth:`assertEqual`, but
1229 it's used to implement :meth:`assertListEqual` and
Ezio Melotti4370b302010-11-03 20:39:14 +00001230 :meth:`assertTupleEqual`.
1231
1232 .. versionadded:: 3.1
1233
1234
1235 .. method:: assertListEqual(list1, list2, msg=None)
1236 assertTupleEqual(tuple1, tuple2, msg=None)
1237
1238 Tests that two lists or tuples are equal. If not an error message is
1239 constructed that shows only the differences between the two. An error
1240 is also raised if either of the parameters are of the wrong type.
1241 These methods are used by default when comparing lists or tuples with
1242 :meth:`assertEqual`.
1243
Ezio Melotti4370b302010-11-03 20:39:14 +00001244 .. versionadded:: 3.1
1245
1246
1247 .. method:: assertSetEqual(set1, set2, msg=None)
1248
1249 Tests that two sets are equal. If not, an error message is constructed
1250 that lists the differences between the sets. This method is used by
1251 default when comparing sets or frozensets with :meth:`assertEqual`.
1252
1253 Fails if either of *set1* or *set2* does not have a :meth:`set.difference`
1254 method.
1255
Ezio Melotti4370b302010-11-03 20:39:14 +00001256 .. versionadded:: 3.1
1257
1258
1259 .. method:: assertDictEqual(expected, actual, msg=None)
1260
1261 Test that two dictionaries are equal. If not, an error message is
1262 constructed that shows the differences in the dictionaries. This
1263 method will be used by default to compare dictionaries in
1264 calls to :meth:`assertEqual`.
1265
Ezio Melotti4370b302010-11-03 20:39:14 +00001266 .. versionadded:: 3.1
1267
1268
1269
Ezio Melotti22170ed2010-11-20 09:57:27 +00001270 .. _other-methods-and-attrs:
1271
Ezio Melotti4370b302010-11-03 20:39:14 +00001272 Finally the :class:`TestCase` provides the following methods and attributes:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001273
Benjamin Peterson52baa292009-03-24 00:56:30 +00001274
Georg Brandl7f01a132009-09-16 15:58:14 +00001275 .. method:: fail(msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001276
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001277 Signals a test failure unconditionally, with *msg* or ``None`` for
Benjamin Peterson52baa292009-03-24 00:56:30 +00001278 the error message.
1279
1280
1281 .. attribute:: failureException
1282
1283 This class attribute gives the exception raised by the test method. If a
1284 test framework needs to use a specialized exception, possibly to carry
1285 additional information, it must subclass this exception in order to "play
1286 fair" with the framework. The initial value of this attribute is
1287 :exc:`AssertionError`.
1288
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001289
1290 .. attribute:: longMessage
1291
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001292 If set to ``True`` then any explicit failure message you pass in to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001293 :ref:`assert methods <assert-methods>` will be appended to the end of the
1294 normal failure message. The normal messages contain useful information
1295 about the objects involved, for example the message from assertEqual
1296 shows you the repr of the two unequal objects. Setting this attribute
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001297 to ``True`` allows you to have a custom error message in addition to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001298 normal one.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001299
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001300 This attribute defaults to ``False``, meaning that a custom message passed
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001301 to an assert method will silence the normal message.
1302
1303 The class setting can be overridden in individual tests by assigning an
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001304 instance attribute to ``True`` or ``False`` before calling the assert methods.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001305
Raymond Hettinger35a88362009-04-09 00:08:24 +00001306 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001307
1308
Michael Foord98b3e762010-06-05 21:59:55 +00001309 .. attribute:: maxDiff
1310
1311 This attribute controls the maximum length of diffs output by assert
1312 methods that report diffs on failure. It defaults to 80*8 characters.
1313 Assert methods affected by this attribute are
1314 :meth:`assertSequenceEqual` (including all the sequence comparison
1315 methods that delegate to it), :meth:`assertDictEqual` and
1316 :meth:`assertMultiLineEqual`.
1317
1318 Setting ``maxDiff`` to None means that there is no maximum length of
1319 diffs.
1320
1321 .. versionadded:: 3.2
1322
1323
Benjamin Peterson52baa292009-03-24 00:56:30 +00001324 Testing frameworks can use the following methods to collect information on
1325 the test:
1326
1327
1328 .. method:: countTestCases()
1329
1330 Return the number of tests represented by this test object. For
1331 :class:`TestCase` instances, this will always be ``1``.
1332
1333
1334 .. method:: defaultTestResult()
1335
1336 Return an instance of the test result class that should be used for this
1337 test case class (if no other result instance is provided to the
1338 :meth:`run` method).
1339
1340 For :class:`TestCase` instances, this will always be an instance of
1341 :class:`TestResult`; subclasses of :class:`TestCase` should override this
1342 as necessary.
1343
1344
1345 .. method:: id()
1346
1347 Return a string identifying the specific test case. This is usually the
1348 full name of the test method, including the module and class name.
1349
1350
1351 .. method:: shortDescription()
1352
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001353 Returns a description of the test, or ``None`` if no description
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001354 has been provided. The default implementation of this method
1355 returns the first line of the test method's docstring, if available,
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001356 or ``None``.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001357
Michael Foord34c94622010-02-10 15:51:42 +00001358 .. versionchanged:: 3.1,3.2
1359 In 3.1 this was changed to add the test name to the short description
1360 even in the presence of a docstring. This caused compatibility issues
1361 with unittest extensions and adding the test name was moved to the
1362 :class:`TextTestResult`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001363
Georg Brandl116aa622007-08-15 14:28:22 +00001364
Georg Brandl7f01a132009-09-16 15:58:14 +00001365 .. method:: addCleanup(function, *args, **kwargs)
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001366
1367 Add a function to be called after :meth:`tearDown` to cleanup resources
1368 used during the test. Functions will be called in reverse order to the
1369 order they are added (LIFO). They are called with any arguments and
1370 keyword arguments passed into :meth:`addCleanup` when they are
1371 added.
1372
1373 If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
1374 then any cleanup functions added will still be called.
1375
Georg Brandl853947a2010-01-31 18:53:23 +00001376 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001377
1378
1379 .. method:: doCleanups()
1380
Barry Warsaw0c9fd632010-04-12 14:50:57 +00001381 This method is called unconditionally after :meth:`tearDown`, or
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001382 after :meth:`setUp` if :meth:`setUp` raises an exception.
1383
1384 It is responsible for calling all the cleanup functions added by
1385 :meth:`addCleanup`. If you need cleanup functions to be called
1386 *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
1387 yourself.
1388
1389 :meth:`doCleanups` pops methods off the stack of cleanup
1390 functions one at a time, so it can be called at any time.
1391
Georg Brandl853947a2010-01-31 18:53:23 +00001392 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001393
1394
Georg Brandl7f01a132009-09-16 15:58:14 +00001395.. class:: FunctionTestCase(testFunc, setUp=None, tearDown=None, description=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001396
1397 This class implements the portion of the :class:`TestCase` interface which
Benjamin Petersond2397752009-06-27 23:45:02 +00001398 allows the test runner to drive the test, but does not provide the methods
1399 which test code can use to check and report errors. This is used to create
1400 test cases using legacy test code, allowing it to be integrated into a
1401 :mod:`unittest`-based test framework.
Georg Brandl116aa622007-08-15 14:28:22 +00001402
1403
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001404.. _deprecated-aliases:
1405
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001406Deprecated aliases
1407##################
1408
1409For historical reasons, some of the :class:`TestCase` methods had one or more
1410aliases that are now deprecated. The following table lists the correct names
1411along with their deprecated aliases:
1412
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001413 ============================== ====================== ======================
1414 Method Name Deprecated alias Deprecated alias
1415 ============================== ====================== ======================
1416 :meth:`.assertEqual` failUnlessEqual assertEquals
1417 :meth:`.assertNotEqual` failIfEqual assertNotEquals
1418 :meth:`.assertTrue` failUnless assert\_
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001419 :meth:`.assertFalse` failIf
1420 :meth:`.assertRaises` failUnlessRaises
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001421 :meth:`.assertAlmostEqual` failUnlessAlmostEqual assertAlmostEquals
1422 :meth:`.assertNotAlmostEqual` failIfAlmostEqual assertNotAlmostEquals
1423 ============================== ====================== ======================
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001424
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001425 .. deprecated-removed:: 3.1 3.3
1426 the fail* aliases listed in the second column.
1427 .. deprecated:: 3.2
1428 the assert* aliases listed in the third column.
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001429
1430
1431
Benjamin Peterson52baa292009-03-24 00:56:30 +00001432.. _testsuite-objects:
1433
Benjamin Peterson52baa292009-03-24 00:56:30 +00001434Grouping tests
1435~~~~~~~~~~~~~~
1436
Georg Brandl7f01a132009-09-16 15:58:14 +00001437.. class:: TestSuite(tests=())
Georg Brandl116aa622007-08-15 14:28:22 +00001438
1439 This class represents an aggregation of individual tests cases and test suites.
1440 The class presents the interface needed by the test runner to allow it to be run
1441 as any other test case. Running a :class:`TestSuite` instance is the same as
1442 iterating over the suite, running each test individually.
1443
1444 If *tests* is given, it must be an iterable of individual test cases or other
1445 test suites that will be used to build the suite initially. Additional methods
1446 are provided to add test cases and suites to the collection later on.
1447
Benjamin Peterson14a3dd72009-05-25 00:51:58 +00001448 :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
1449 they do not actually implement a test. Instead, they are used to aggregate
1450 tests into groups of tests that should be run together. Some additional
1451 methods are available to add tests to :class:`TestSuite` instances:
Benjamin Peterson52baa292009-03-24 00:56:30 +00001452
1453
1454 .. method:: TestSuite.addTest(test)
1455
1456 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1457
1458
1459 .. method:: TestSuite.addTests(tests)
1460
1461 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
1462 instances to this test suite.
1463
Benjamin Petersond2397752009-06-27 23:45:02 +00001464 This is equivalent to iterating over *tests*, calling :meth:`addTest` for
1465 each element.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001466
1467 :class:`TestSuite` shares the following methods with :class:`TestCase`:
1468
1469
1470 .. method:: run(result)
1471
1472 Run the tests associated with this suite, collecting the result into the
1473 test result object passed as *result*. Note that unlike
1474 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
1475 be passed in.
1476
1477
1478 .. method:: debug()
1479
1480 Run the tests associated with this suite without collecting the
1481 result. This allows exceptions raised by the test to be propagated to the
1482 caller and can be used to support running tests under a debugger.
1483
1484
1485 .. method:: countTestCases()
1486
1487 Return the number of tests represented by this test object, including all
1488 individual tests and sub-suites.
1489
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001490
1491 .. method:: __iter__()
1492
1493 Tests grouped by a :class:`TestSuite` are always accessed by iteration.
1494 Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
1495 that this method maybe called several times on a single suite
1496 (for example when counting tests or comparing for equality)
1497 so the tests returned must be the same for repeated iterations.
1498
Georg Brandl853947a2010-01-31 18:53:23 +00001499 .. versionchanged:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001500 In earlier versions the :class:`TestSuite` accessed tests directly rather
1501 than through iteration, so overriding :meth:`__iter__` wasn't sufficient
1502 for providing tests.
1503
Benjamin Peterson52baa292009-03-24 00:56:30 +00001504 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
1505 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1506
1507
Benjamin Peterson52baa292009-03-24 00:56:30 +00001508Loading and running tests
1509~~~~~~~~~~~~~~~~~~~~~~~~~
1510
Georg Brandl116aa622007-08-15 14:28:22 +00001511.. class:: TestLoader()
1512
Benjamin Peterson52baa292009-03-24 00:56:30 +00001513 The :class:`TestLoader` class is used to create test suites from classes and
1514 modules. Normally, there is no need to create an instance of this class; the
1515 :mod:`unittest` module provides an instance that can be shared as
1516 ``unittest.defaultTestLoader``. Using a subclass or instance, however, allows
1517 customization of some configurable properties.
1518
1519 :class:`TestLoader` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001520
Ezio Melotti9c02c2f2010-11-03 20:45:31 +00001521
Benjamin Peterson52baa292009-03-24 00:56:30 +00001522 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl116aa622007-08-15 14:28:22 +00001523
Benjamin Peterson52baa292009-03-24 00:56:30 +00001524 Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
1525 :class:`testCaseClass`.
1526
1527
1528 .. method:: loadTestsFromModule(module)
1529
1530 Return a suite of all tests cases contained in the given module. This
1531 method searches *module* for classes derived from :class:`TestCase` and
1532 creates an instance of the class for each test method defined for the
1533 class.
1534
Georg Brandle720c0a2009-04-27 16:20:50 +00001535 .. note::
Benjamin Peterson52baa292009-03-24 00:56:30 +00001536
1537 While using a hierarchy of :class:`TestCase`\ -derived classes can be
1538 convenient in sharing fixtures and helper functions, defining test
1539 methods on base classes that are not intended to be instantiated
1540 directly does not play well with this method. Doing so, however, can
1541 be useful when the fixtures are different and defined in subclasses.
1542
Benjamin Petersond2397752009-06-27 23:45:02 +00001543 If a module provides a ``load_tests`` function it will be called to
1544 load the tests. This allows modules to customize test loading.
1545 This is the `load_tests protocol`_.
1546
Georg Brandl853947a2010-01-31 18:53:23 +00001547 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001548 Support for ``load_tests`` added.
1549
Benjamin Peterson52baa292009-03-24 00:56:30 +00001550
Georg Brandl7f01a132009-09-16 15:58:14 +00001551 .. method:: loadTestsFromName(name, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001552
1553 Return a suite of all tests cases given a string specifier.
1554
1555 The specifier *name* is a "dotted name" that may resolve either to a
1556 module, a test case class, a test method within a test case class, a
1557 :class:`TestSuite` instance, or a callable object which returns a
1558 :class:`TestCase` or :class:`TestSuite` instance. These checks are
1559 applied in the order listed here; that is, a method on a possible test
1560 case class will be picked up as "a test method within a test case class",
1561 rather than "a callable object".
1562
1563 For example, if you have a module :mod:`SampleTests` containing a
1564 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
1565 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
Benjamin Petersond2397752009-06-27 23:45:02 +00001566 specifier ``'SampleTests.SampleTestCase'`` would cause this method to
1567 return a suite which will run all three test methods. Using the specifier
1568 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
1569 suite which will run only the :meth:`test_two` test method. The specifier
1570 can refer to modules and packages which have not been imported; they will
1571 be imported as a side-effect.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001572
1573 The method optionally resolves *name* relative to the given *module*.
1574
1575
Georg Brandl7f01a132009-09-16 15:58:14 +00001576 .. method:: loadTestsFromNames(names, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001577
1578 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
1579 than a single name. The return value is a test suite which supports all
1580 the tests defined for each name.
1581
1582
1583 .. method:: getTestCaseNames(testCaseClass)
1584
1585 Return a sorted sequence of method names found within *testCaseClass*;
1586 this should be a subclass of :class:`TestCase`.
1587
Benjamin Petersond2397752009-06-27 23:45:02 +00001588
1589 .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)
1590
1591 Find and return all test modules from the specified start directory,
1592 recursing into subdirectories to find them. Only test files that match
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001593 *pattern* will be loaded. (Using shell style pattern matching.) Only
1594 module names that are importable (i.e. are valid Python identifiers) will
1595 be loaded.
Benjamin Petersond2397752009-06-27 23:45:02 +00001596
1597 All test modules must be importable from the top level of the project. If
1598 the start directory is not the top level directory then the top level
1599 directory must be specified separately.
1600
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001601 If importing a module fails, for example due to a syntax error, then this
1602 will be recorded as a single error and discovery will continue.
1603
Benjamin Petersond2397752009-06-27 23:45:02 +00001604 If a test package name (directory with :file:`__init__.py`) matches the
1605 pattern then the package will be checked for a ``load_tests``
1606 function. If this exists then it will be called with *loader*, *tests*,
1607 *pattern*.
1608
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001609 If load_tests exists then discovery does *not* recurse into the package,
Benjamin Petersond2397752009-06-27 23:45:02 +00001610 ``load_tests`` is responsible for loading all tests in the package.
1611
1612 The pattern is deliberately not stored as a loader attribute so that
1613 packages can continue discovery themselves. *top_level_dir* is stored so
1614 ``load_tests`` does not need to pass this argument in to
1615 ``loader.discover()``.
1616
Benjamin Petersonb48af542010-04-11 20:43:16 +00001617 *start_dir* can be a dotted module name as well as a directory.
1618
Georg Brandl853947a2010-01-31 18:53:23 +00001619 .. versionadded:: 3.2
1620
Benjamin Petersond2397752009-06-27 23:45:02 +00001621
Benjamin Peterson52baa292009-03-24 00:56:30 +00001622 The following attributes of a :class:`TestLoader` can be configured either by
1623 subclassing or assignment on an instance:
1624
1625
1626 .. attribute:: testMethodPrefix
1627
1628 String giving the prefix of method names which will be interpreted as test
1629 methods. The default value is ``'test'``.
1630
1631 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
1632 methods.
1633
1634
1635 .. attribute:: sortTestMethodsUsing
1636
1637 Function to be used to compare method names when sorting them in
1638 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
1639
1640
1641 .. attribute:: suiteClass
1642
1643 Callable object that constructs a test suite from a list of tests. No
1644 methods on the resulting object are needed. The default value is the
1645 :class:`TestSuite` class.
1646
1647 This affects all the :meth:`loadTestsFrom\*` methods.
1648
1649
Benjamin Peterson52baa292009-03-24 00:56:30 +00001650.. class:: TestResult
1651
1652 This class is used to compile information about which tests have succeeded
1653 and which have failed.
1654
1655 A :class:`TestResult` object stores the results of a set of tests. The
1656 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
1657 properly recorded; test authors do not need to worry about recording the
1658 outcome of tests.
1659
1660 Testing frameworks built on top of :mod:`unittest` may want access to the
1661 :class:`TestResult` object generated by running a set of tests for reporting
1662 purposes; a :class:`TestResult` instance is returned by the
1663 :meth:`TestRunner.run` method for this purpose.
1664
1665 :class:`TestResult` instances have the following attributes that will be of
1666 interest when inspecting the results of running a set of tests:
1667
1668
1669 .. attribute:: errors
1670
1671 A list containing 2-tuples of :class:`TestCase` instances and strings
1672 holding formatted tracebacks. Each tuple represents a test which raised an
1673 unexpected exception.
1674
Benjamin Peterson52baa292009-03-24 00:56:30 +00001675 .. attribute:: failures
1676
1677 A list containing 2-tuples of :class:`TestCase` instances and strings
1678 holding formatted tracebacks. Each tuple represents a test where a failure
1679 was explicitly signalled using the :meth:`TestCase.fail\*` or
1680 :meth:`TestCase.assert\*` methods.
1681
Benjamin Peterson52baa292009-03-24 00:56:30 +00001682 .. attribute:: skipped
1683
1684 A list containing 2-tuples of :class:`TestCase` instances and strings
1685 holding the reason for skipping the test.
1686
Benjamin Peterson70e32c82009-03-24 01:00:11 +00001687 .. versionadded:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +00001688
1689 .. attribute:: expectedFailures
1690
Georg Brandl6faee4e2010-09-21 14:48:28 +00001691 A list containing 2-tuples of :class:`TestCase` instances and strings
1692 holding formatted tracebacks. Each tuple represents an expected failure
Benjamin Peterson52baa292009-03-24 00:56:30 +00001693 of the test case.
1694
1695 .. attribute:: unexpectedSuccesses
1696
1697 A list containing :class:`TestCase` instances that were marked as expected
1698 failures, but succeeded.
1699
1700 .. attribute:: shouldStop
1701
1702 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1703
1704
1705 .. attribute:: testsRun
1706
1707 The total number of tests run so far.
1708
1709
Benjamin Petersonb48af542010-04-11 20:43:16 +00001710 .. attribute:: buffer
1711
1712 If set to true, ``sys.stdout`` and ``sys.stderr`` will be buffered in between
1713 :meth:`startTest` and :meth:`stopTest` being called. Collected output will
1714 only be echoed onto the real ``sys.stdout`` and ``sys.stderr`` if the test
1715 fails or errors. Any output is also attached to the failure / error message.
1716
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001717 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001718
1719
1720 .. attribute:: failfast
1721
1722 If set to true :meth:`stop` will be called on the first failure or error,
1723 halting the test run.
1724
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001725 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001726
1727
Benjamin Peterson52baa292009-03-24 00:56:30 +00001728 .. method:: wasSuccessful()
1729
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001730 Return ``True`` if all tests run so far have passed, otherwise returns
1731 ``False``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001732
1733
1734 .. method:: stop()
1735
1736 This method can be called to signal that the set of tests being run should
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001737 be aborted by setting the :attr:`shouldStop` attribute to ``True``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001738 :class:`TestRunner` objects should respect this flag and return without
1739 running any additional tests.
1740
1741 For example, this feature is used by the :class:`TextTestRunner` class to
1742 stop the test framework when the user signals an interrupt from the
1743 keyboard. Interactive tools which provide :class:`TestRunner`
1744 implementations can use this in a similar manner.
1745
1746 The following methods of the :class:`TestResult` class are used to maintain
1747 the internal data structures, and may be extended in subclasses to support
1748 additional reporting requirements. This is particularly useful in building
1749 tools which support interactive reporting while tests are being run.
1750
1751
1752 .. method:: startTest(test)
1753
1754 Called when the test case *test* is about to be run.
1755
Benjamin Peterson52baa292009-03-24 00:56:30 +00001756 .. method:: stopTest(test)
1757
1758 Called after the test case *test* has been executed, regardless of the
1759 outcome.
1760
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001761 .. method:: startTestRun(test)
1762
1763 Called once before any tests are executed.
1764
Georg Brandl853947a2010-01-31 18:53:23 +00001765 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001766
1767
1768 .. method:: stopTestRun(test)
1769
Ezio Melotti176d6c42010-01-27 20:58:07 +00001770 Called once after all tests are executed.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001771
Georg Brandl853947a2010-01-31 18:53:23 +00001772 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001773
1774
Benjamin Peterson52baa292009-03-24 00:56:30 +00001775 .. method:: addError(test, err)
1776
1777 Called when the test case *test* raises an unexpected exception *err* is a
1778 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1779 traceback)``.
1780
1781 The default implementation appends a tuple ``(test, formatted_err)`` to
1782 the instance's :attr:`errors` attribute, where *formatted_err* is a
1783 formatted traceback derived from *err*.
1784
1785
1786 .. method:: addFailure(test, err)
1787
Benjamin Petersond2397752009-06-27 23:45:02 +00001788 Called when the test case *test* signals a failure. *err* is a tuple of
1789 the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001790
1791 The default implementation appends a tuple ``(test, formatted_err)`` to
1792 the instance's :attr:`failures` attribute, where *formatted_err* is a
1793 formatted traceback derived from *err*.
1794
1795
1796 .. method:: addSuccess(test)
1797
1798 Called when the test case *test* succeeds.
1799
1800 The default implementation does nothing.
1801
1802
1803 .. method:: addSkip(test, reason)
1804
1805 Called when the test case *test* is skipped. *reason* is the reason the
1806 test gave for skipping.
1807
1808 The default implementation appends a tuple ``(test, reason)`` to the
1809 instance's :attr:`skipped` attribute.
1810
1811
1812 .. method:: addExpectedFailure(test, err)
1813
1814 Called when the test case *test* fails, but was marked with the
1815 :func:`expectedFailure` decorator.
1816
1817 The default implementation appends a tuple ``(test, formatted_err)`` to
1818 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1819 is a formatted traceback derived from *err*.
1820
1821
1822 .. method:: addUnexpectedSuccess(test)
1823
1824 Called when the test case *test* was marked with the
1825 :func:`expectedFailure` decorator, but succeeded.
1826
1827 The default implementation appends the test to the instance's
1828 :attr:`unexpectedSuccesses` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +00001829
Georg Brandl67b21b72010-08-17 15:07:14 +00001830
Michael Foord34c94622010-02-10 15:51:42 +00001831.. class:: TextTestResult(stream, descriptions, verbosity)
1832
Georg Brandl67b21b72010-08-17 15:07:14 +00001833 A concrete implementation of :class:`TestResult` used by the
1834 :class:`TextTestRunner`.
Michael Foord34c94622010-02-10 15:51:42 +00001835
Georg Brandl67b21b72010-08-17 15:07:14 +00001836 .. versionadded:: 3.2
1837 This class was previously named ``_TextTestResult``. The old name still
1838 exists as an alias but is deprecated.
1839
Georg Brandl116aa622007-08-15 14:28:22 +00001840
1841.. data:: defaultTestLoader
1842
1843 Instance of the :class:`TestLoader` class intended to be shared. If no
1844 customization of the :class:`TestLoader` is needed, this instance can be used
1845 instead of repeatedly creating new instances.
1846
1847
Michael Foord34c94622010-02-10 15:51:42 +00001848.. class:: TextTestRunner(stream=sys.stderr, descriptions=True, verbosity=1, runnerclass=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001849
1850 A basic test runner implementation which prints results on standard error. It
1851 has a few configurable parameters, but is essentially very simple. Graphical
1852 applications which run test suites should provide alternate implementations.
1853
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001854 .. method:: _makeResult()
Georg Brandl116aa622007-08-15 14:28:22 +00001855
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001856 This method returns the instance of ``TestResult`` used by :meth:`run`.
1857 It is not intended to be called directly, but can be overridden in
1858 subclasses to provide a custom ``TestResult``.
1859
Michael Foord34c94622010-02-10 15:51:42 +00001860 ``_makeResult()`` instantiates the class or callable passed in the
1861 ``TextTestRunner`` constructor as the ``resultclass`` argument. It
Benjamin Petersonb48af542010-04-11 20:43:16 +00001862 defaults to :class:`TextTestResult` if no ``resultclass`` is provided.
Michael Foord34c94622010-02-10 15:51:42 +00001863 The result class is instantiated with the following arguments::
1864
1865 stream, descriptions, verbosity
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001866
Benjamin Petersonb48af542010-04-11 20:43:16 +00001867.. function:: main(module='__main__', defaultTest=None, argv=None, testRunner=None, testLoader=unittest.loader.defaultTestLoader, exit=True, verbosity=1, failfast=None, catchbreak=None, buffer=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001868
1869 A command-line program that runs a set of tests; this is primarily for making
1870 test modules conveniently executable. The simplest use for this function is to
1871 include the following line at the end of a test script::
1872
1873 if __name__ == '__main__':
1874 unittest.main()
1875
Benjamin Petersond2397752009-06-27 23:45:02 +00001876 You can run tests with more detailed information by passing in the verbosity
1877 argument::
1878
1879 if __name__ == '__main__':
1880 unittest.main(verbosity=2)
1881
Georg Brandl116aa622007-08-15 14:28:22 +00001882 The *testRunner* argument can either be a test runner class or an already
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001883 created instance of it. By default ``main`` calls :func:`sys.exit` with
1884 an exit code indicating success or failure of the tests run.
1885
1886 ``main`` supports being used from the interactive interpreter by passing in the
1887 argument ``exit=False``. This displays the result on standard output without
1888 calling :func:`sys.exit`::
1889
1890 >>> from unittest import main
1891 >>> main(module='test_module', exit=False)
1892
Benjamin Petersonb48af542010-04-11 20:43:16 +00001893 The ``failfast``, ``catchbreak`` and ``buffer`` parameters have the same
Éric Araujo713d3032010-11-18 16:38:46 +00001894 effect as the `failfast, catch and buffer command-line options`_.
Benjamin Petersonb48af542010-04-11 20:43:16 +00001895
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001896 Calling ``main`` actually returns an instance of the ``TestProgram`` class.
1897 This stores the result of the tests run as the ``result`` attribute.
1898
Georg Brandl853947a2010-01-31 18:53:23 +00001899 .. versionchanged:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001900 The ``exit``, ``verbosity``, ``failfast``, ``catchbreak`` and ``buffer``
1901 parameters were added.
Benjamin Petersond2397752009-06-27 23:45:02 +00001902
1903
1904load_tests Protocol
1905###################
1906
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001907
Georg Brandl853947a2010-01-31 18:53:23 +00001908.. versionadded:: 3.2
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001909
1910
Benjamin Petersond2397752009-06-27 23:45:02 +00001911Modules or packages can customize how tests are loaded from them during normal
1912test runs or test discovery by implementing a function called ``load_tests``.
1913
1914If a test module defines ``load_tests`` it will be called by
1915:meth:`TestLoader.loadTestsFromModule` with the following arguments::
1916
1917 load_tests(loader, standard_tests, None)
1918
1919It should return a :class:`TestSuite`.
1920
1921*loader* is the instance of :class:`TestLoader` doing the loading.
1922*standard_tests* are the tests that would be loaded by default from the
1923module. It is common for test modules to only want to add or remove tests
1924from the standard set of tests.
1925The third argument is used when loading packages as part of test discovery.
1926
1927A typical ``load_tests`` function that loads tests from a specific set of
1928:class:`TestCase` classes may look like::
1929
1930 test_cases = (TestCase1, TestCase2, TestCase3)
1931
1932 def load_tests(loader, tests, pattern):
1933 suite = TestSuite()
1934 for test_class in test_cases:
1935 tests = loader.loadTestsFromTestCase(test_class)
1936 suite.addTests(tests)
1937 return suite
1938
1939If discovery is started, either from the command line or by calling
1940:meth:`TestLoader.discover`, with a pattern that matches a package
1941name then the package :file:`__init__.py` will be checked for ``load_tests``.
1942
1943.. note::
1944
Ezio Melotti0639d5a2009-12-19 23:26:38 +00001945 The default pattern is 'test*.py'. This matches all Python files
Benjamin Petersond2397752009-06-27 23:45:02 +00001946 that start with 'test' but *won't* match any test directories.
1947
1948 A pattern like 'test*' will match test packages as well as
1949 modules.
1950
1951If the package :file:`__init__.py` defines ``load_tests`` then it will be
1952called and discovery not continued into the package. ``load_tests``
1953is called with the following arguments::
1954
1955 load_tests(loader, standard_tests, pattern)
1956
1957This should return a :class:`TestSuite` representing all the tests
1958from the package. (``standard_tests`` will only contain tests
1959collected from :file:`__init__.py`.)
1960
1961Because the pattern is passed into ``load_tests`` the package is free to
1962continue (and potentially modify) test discovery. A 'do nothing'
1963``load_tests`` function for a test package would look like::
1964
1965 def load_tests(loader, standard_tests, pattern):
1966 # top level directory cached on loader instance
1967 this_dir = os.path.dirname(__file__)
1968 package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
1969 standard_tests.addTests(package_tests)
1970 return standard_tests
Benjamin Petersonb48af542010-04-11 20:43:16 +00001971
1972
1973Class and Module Fixtures
1974-------------------------
1975
1976Class and module level fixtures are implemented in :class:`TestSuite`. When
1977the test suite encounters a test from a new class then :meth:`tearDownClass`
1978from the previous class (if there is one) is called, followed by
1979:meth:`setUpClass` from the new class.
1980
1981Similarly if a test is from a different module from the previous test then
1982``tearDownModule`` from the previous module is run, followed by
1983``setUpModule`` from the new module.
1984
1985After all the tests have run the final ``tearDownClass`` and
1986``tearDownModule`` are run.
1987
1988Note that shared fixtures do not play well with [potential] features like test
1989parallelization and they break test isolation. They should be used with care.
1990
1991The default ordering of tests created by the unittest test loaders is to group
1992all tests from the same modules and classes together. This will lead to
1993``setUpClass`` / ``setUpModule`` (etc) being called exactly once per class and
1994module. If you randomize the order, so that tests from different modules and
1995classes are adjacent to each other, then these shared fixture functions may be
1996called multiple times in a single test run.
1997
1998Shared fixtures are not intended to work with suites with non-standard
1999ordering. A ``BaseTestSuite`` still exists for frameworks that don't want to
2000support shared fixtures.
2001
2002If there are any exceptions raised during one of the shared fixture functions
2003the test is reported as an error. Because there is no corresponding test
2004instance an ``_ErrorHolder`` object (that has the same interface as a
2005:class:`TestCase`) is created to represent the error. If you are just using
2006the standard unittest test runner then this detail doesn't matter, but if you
2007are a framework author it may be relevant.
2008
2009
2010setUpClass and tearDownClass
2011~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2012
2013These must be implemented as class methods::
2014
2015 import unittest
2016
2017 class Test(unittest.TestCase):
2018 @classmethod
2019 def setUpClass(cls):
2020 cls._connection = createExpensiveConnectionObject()
2021
2022 @classmethod
2023 def tearDownClass(cls):
2024 cls._connection.destroy()
2025
2026If you want the ``setUpClass`` and ``tearDownClass`` on base classes called
2027then you must call up to them yourself. The implementations in
2028:class:`TestCase` are empty.
2029
2030If an exception is raised during a ``setUpClass`` then the tests in the class
2031are not run and the ``tearDownClass`` is not run. Skipped classes will not
Michael Foord98b3e762010-06-05 21:59:55 +00002032have ``setUpClass`` or ``tearDownClass`` run. If the exception is a
2033``SkipTest`` exception then the class will be reported as having been skipped
2034instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002035
2036
2037setUpModule and tearDownModule
2038~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2039
2040These should be implemented as functions::
2041
2042 def setUpModule():
2043 createConnection()
2044
2045 def tearDownModule():
2046 closeConnection()
2047
2048If an exception is raised in a ``setUpModule`` then none of the tests in the
Michael Foord98b3e762010-06-05 21:59:55 +00002049module will be run and the ``tearDownModule`` will not be run. If the exception is a
2050``SkipTest`` exception then the module will be reported as having been skipped
2051instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002052
2053
2054Signal Handling
2055---------------
2056
Éric Araujod3309df2010-11-21 03:09:17 +00002057The ``-c``/``--catch`` command-line option to unittest, along with the ``catchbreak``
Benjamin Petersonb48af542010-04-11 20:43:16 +00002058parameter to :func:`unittest.main()`, provide more friendly handling of
Benjamin Petersond7c3ed52010-06-27 22:32:30 +00002059control-C during a test run. With catch break behavior enabled control-C will
Benjamin Petersonb48af542010-04-11 20:43:16 +00002060allow the currently running test to complete, and the test run will then end
2061and report all the results so far. A second control-c will raise a
Benjamin Petersond7c3ed52010-06-27 22:32:30 +00002062:exc:`KeyboardInterrupt` in the usual way.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002063
Michael Foordde4ceab2010-04-25 19:53:49 +00002064The control-c handling signal handler attempts to remain compatible with code or
2065tests that install their own :const:`signal.SIGINT` handler. If the ``unittest``
2066handler is called but *isn't* the installed :const:`signal.SIGINT` handler,
2067i.e. it has been replaced by the system under test and delegated to, then it
2068calls the default handler. This will normally be the expected behavior by code
2069that replaces an installed handler and delegates to it. For individual tests
2070that need ``unittest`` control-c handling disabled the :func:`removeHandler`
2071decorator can be used.
2072
2073There are a few utility functions for framework authors to enable control-c
2074handling functionality within test frameworks.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002075
2076.. function:: installHandler()
2077
2078 Install the control-c handler. When a :const:`signal.SIGINT` is received
2079 (usually in response to the user pressing control-c) all registered results
2080 have :meth:`~TestResult.stop` called.
2081
Michael Foord469b1f02010-04-26 23:41:26 +00002082 .. versionadded:: 3.2
2083
Benjamin Petersonb48af542010-04-11 20:43:16 +00002084.. function:: registerResult(result)
2085
2086 Register a :class:`TestResult` object for control-c handling. Registering a
2087 result stores a weak reference to it, so it doesn't prevent the result from
2088 being garbage collected.
2089
Michael Foordde4ceab2010-04-25 19:53:49 +00002090 Registering a :class:`TestResult` object has no side-effects if control-c
2091 handling is not enabled, so test frameworks can unconditionally register
2092 all results they create independently of whether or not handling is enabled.
2093
Michael Foord469b1f02010-04-26 23:41:26 +00002094 .. versionadded:: 3.2
2095
Benjamin Petersonb48af542010-04-11 20:43:16 +00002096.. function:: removeResult(result)
2097
2098 Remove a registered result. Once a result has been removed then
2099 :meth:`~TestResult.stop` will no longer be called on that result object in
2100 response to a control-c.
2101
Michael Foord469b1f02010-04-26 23:41:26 +00002102 .. versionadded:: 3.2
2103
Michael Foordde4ceab2010-04-25 19:53:49 +00002104.. function:: removeHandler(function=None)
2105
2106 When called without arguments this function removes the control-c handler
2107 if it has been installed. This function can also be used as a test decorator
2108 to temporarily remove the handler whilst the test is being executed::
2109
2110 @unittest.removeHandler
2111 def test_signal_handling(self):
2112 ...
2113
Michael Foord469b1f02010-04-26 23:41:26 +00002114 .. versionadded:: 3.2
2115