blob: 9bd85d52da338674fd7198194f61b2726cf7986a [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
11
Georg Brandl116aa622007-08-15 14:28:22 +000012The Python unit testing framework, sometimes referred to as "PyUnit," is a
13Python language version of JUnit, by Kent Beck and Erich Gamma. JUnit is, in
14turn, a Java version of Kent's Smalltalk testing framework. Each is the de
15facto standard unit testing framework for its respective language.
16
17:mod:`unittest` supports test automation, sharing of setup and shutdown code for
18tests, aggregation of tests into collections, and independence of the tests from
19the reporting framework. The :mod:`unittest` module provides classes that make
20it easy to support these qualities for a set of tests.
21
22To achieve this, :mod:`unittest` supports some important concepts:
23
24test fixture
25 A :dfn:`test fixture` represents the preparation needed to perform one or more
26 tests, and any associate cleanup actions. This may involve, for example,
27 creating temporary or proxy databases, directories, or starting a server
28 process.
29
30test case
31 A :dfn:`test case` is the smallest unit of testing. It checks for a specific
32 response to a particular set of inputs. :mod:`unittest` provides a base class,
33 :class:`TestCase`, which may be used to create new test cases.
34
35test suite
36 A :dfn:`test suite` is a collection of test cases, test suites, or both. It is
37 used to aggregate tests that should be executed together.
38
39test runner
40 A :dfn:`test runner` is a component which orchestrates the execution of tests
41 and provides the outcome to the user. The runner may use a graphical interface,
42 a textual interface, or return a special value to indicate the results of
43 executing the tests.
44
45The test case and test fixture concepts are supported through the
46:class:`TestCase` and :class:`FunctionTestCase` classes; the former should be
47used when creating new tests, and the latter can be used when integrating
48existing test code with a :mod:`unittest`\ -driven framework. When building test
Benjamin Peterson52baa292009-03-24 00:56:30 +000049fixtures using :class:`TestCase`, the :meth:`~TestCase.setUp` and
50:meth:`~TestCase.tearDown` methods can be overridden to provide initialization
51and cleanup for the fixture. With :class:`FunctionTestCase`, existing functions
52can be passed to the constructor for these purposes. When the test is run, the
53fixture initialization is run first; if it succeeds, the cleanup method is run
54after the test has been executed, regardless of the outcome of the test. Each
55instance of the :class:`TestCase` will only be used to run a single test method,
56so a new fixture is created for each test.
Georg Brandl116aa622007-08-15 14:28:22 +000057
58Test suites are implemented by the :class:`TestSuite` class. This class allows
59individual tests and test suites to be aggregated; when the suite is executed,
Benjamin Peterson14a3dd72009-05-25 00:51:58 +000060all tests added directly to the suite and in "child" test suites are run.
Georg Brandl116aa622007-08-15 14:28:22 +000061
Benjamin Peterson52baa292009-03-24 00:56:30 +000062A test runner is an object that provides a single method,
63:meth:`~TestRunner.run`, which accepts a :class:`TestCase` or :class:`TestSuite`
64object as a parameter, and returns a result object. The class
65:class:`TestResult` is provided for use as the result object. :mod:`unittest`
66provides the :class:`TextTestRunner` as an example test runner which reports
67test results on the standard error stream by default. Alternate runners can be
68implemented for other environments (such as graphical environments) without any
69need to derive from a specific class.
Georg Brandl116aa622007-08-15 14:28:22 +000070
71
72.. seealso::
73
74 Module :mod:`doctest`
75 Another test-support module with a very different flavor.
76
77 `Simple Smalltalk Testing: With Patterns <http://www.XProgramming.com/testfram.htm>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000078 Kent Beck's original paper on testing frameworks using the pattern shared
79 by :mod:`unittest`.
Georg Brandl116aa622007-08-15 14:28:22 +000080
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000081 `Nose <http://code.google.com/p/python-nose/>`_ and `py.test <http://pytest.org>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000082 Third-party unittest frameworks with a lighter-weight syntax for writing
83 tests. For example, ``assert func(10) == 42``.
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000084
85 `python-mock <http://python-mock.sourceforge.net/>`_ and `minimock <http://blog.ianbicking.org/minimock.html>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000086 Tools for creating mock test objects (objects simulating external
87 resources).
88
89
90.. _unittest-command-line-interface:
91
92Command Line Interface
93----------------------
94
95The unittest module can be used from the command line to run tests from
96modules, classes or even individual test methods::
97
98 python -m unittest test_module1 test_module2
99 python -m unittest test_module.TestClass
100 python -m unittest test_module.TestClass.test_method
101
102You can pass in a list with any combination of module names, and fully
103qualified class or method names.
104
105You can run tests with more detail (higher verbosity) by passing in the -v flag::
106
Ezio Melotti176d6c42010-01-27 20:58:07 +0000107 python -m unittest -v test_module
Benjamin Petersond2397752009-06-27 23:45:02 +0000108
109For a list of all the command line options::
110
111 python -m unittest -h
112
Georg Brandl853947a2010-01-31 18:53:23 +0000113.. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +0000114 In earlier versions it was only possible to run individual test methods and
115 not modules or classes.
116
117The command line can also be used for test discovery, for running all of the
118tests in a project or just a subset.
119
120
121.. _unittest-test-discovery:
122
123Test Discovery
124--------------
125
Georg Brandl853947a2010-01-31 18:53:23 +0000126.. versionadded:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +0000127
128unittest supports simple test discovery. For a project's tests to be
129compatible with test discovery they must all be importable from the top level
130directory of the project; i.e. they must all be in Python packages.
131
132Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
133used from the command line. The basic command line usage is::
134
135 cd project_directory
136 python -m unittest discover
137
138The ``discover`` sub-command has the following options:
139
140 -v, --verbose Verbose output
141 -s directory Directory to start discovery ('.' default)
142 -p pattern Pattern to match test files ('test*.py' default)
143 -t directory Top level directory of project (default to
144 start directory)
145
146The -s, -p, & -t options can be passsed in as positional arguments. The
147following two command lines are equivalent::
148
Ezio Melotti176d6c42010-01-27 20:58:07 +0000149 python -m unittest discover -s project_directory -p '*_test.py'
150 python -m unittest discover project_directory '*_test.py'
Benjamin Petersond2397752009-06-27 23:45:02 +0000151
152Test modules and packages can customize test loading and discovery by through
153the `load_tests protocol`_.
Georg Brandl116aa622007-08-15 14:28:22 +0000154
155.. _unittest-minimal-example:
156
157Basic example
158-------------
159
160The :mod:`unittest` module provides a rich set of tools for constructing and
161running tests. This section demonstrates that a small subset of the tools
162suffice to meet the needs of most users.
163
164Here is a short script to test three functions from the :mod:`random` module::
165
166 import random
167 import unittest
168
169 class TestSequenceFunctions(unittest.TestCase):
170
171 def setUp(self):
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000172 self.seq = list(range(10))
Georg Brandl116aa622007-08-15 14:28:22 +0000173
Benjamin Peterson52baa292009-03-24 00:56:30 +0000174 def test_shuffle(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000175 # make sure the shuffled sequence does not lose any elements
176 random.shuffle(self.seq)
177 self.seq.sort()
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000178 self.assertEqual(self.seq, list(range(10)))
Georg Brandl116aa622007-08-15 14:28:22 +0000179
Benjamin Peterson847a4112010-03-14 15:04:17 +0000180 # should raise an exception for an immutable sequence
181 self.assertRaises(TypeError, random.shuffle, (1,2,3))
182
Benjamin Peterson52baa292009-03-24 00:56:30 +0000183 def test_choice(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000184 element = random.choice(self.seq)
Benjamin Peterson847a4112010-03-14 15:04:17 +0000185 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000186
Benjamin Peterson52baa292009-03-24 00:56:30 +0000187 def test_sample(self):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000188 with self.assertRaises(ValueError):
189 random.sample(self.seq, 20)
Georg Brandl116aa622007-08-15 14:28:22 +0000190 for element in random.sample(self.seq, 5):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000191 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000192
193 if __name__ == '__main__':
194 unittest.main()
195
Benjamin Peterson52baa292009-03-24 00:56:30 +0000196A testcase is created by subclassing :class:`unittest.TestCase`. The three
Georg Brandl116aa622007-08-15 14:28:22 +0000197individual tests are defined with methods whose names start with the letters
198``test``. This naming convention informs the test runner about which methods
199represent tests.
200
Benjamin Peterson52baa292009-03-24 00:56:30 +0000201The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
Michael Foord34c94622010-02-10 15:51:42 +0000202expected result; :meth:`~TestCase.assertTrue` to verify a condition; or
Benjamin Peterson52baa292009-03-24 00:56:30 +0000203:meth:`~TestCase.assertRaises` to verify that an expected exception gets raised.
204These methods are used instead of the :keyword:`assert` statement so the test
205runner can accumulate all test results and produce a report.
Georg Brandl116aa622007-08-15 14:28:22 +0000206
Benjamin Peterson52baa292009-03-24 00:56:30 +0000207When a :meth:`~TestCase.setUp` method is defined, the test runner will run that
208method prior to each test. Likewise, if a :meth:`~TestCase.tearDown` method is
209defined, the test runner will invoke that method after each test. In the
210example, :meth:`~TestCase.setUp` was used to create a fresh sequence for each
211test.
Georg Brandl116aa622007-08-15 14:28:22 +0000212
213The final block shows a simple way to run the tests. :func:`unittest.main`
214provides a command line interface to the test script. When run from the command
215line, the above script produces an output that looks like this::
216
217 ...
218 ----------------------------------------------------------------------
219 Ran 3 tests in 0.000s
220
221 OK
222
223Instead of :func:`unittest.main`, there are other ways to run the tests with a
224finer level of control, less terse output, and no requirement to be run from the
225command line. For example, the last two lines may be replaced with::
226
227 suite = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)
228 unittest.TextTestRunner(verbosity=2).run(suite)
229
230Running the revised script from the interpreter or another script produces the
231following output::
232
Ezio Melottid59e44a2010-02-28 03:46:13 +0000233 test_choice (__main__.TestSequenceFunctions) ... ok
234 test_sample (__main__.TestSequenceFunctions) ... ok
235 test_shuffle (__main__.TestSequenceFunctions) ... ok
Georg Brandl116aa622007-08-15 14:28:22 +0000236
237 ----------------------------------------------------------------------
238 Ran 3 tests in 0.110s
239
240 OK
241
242The above examples show the most commonly used :mod:`unittest` features which
243are sufficient to meet many everyday testing needs. The remainder of the
244documentation explores the full feature set from first principles.
245
Georg Brandl116aa622007-08-15 14:28:22 +0000246.. _organizing-tests:
247
248Organizing test code
249--------------------
250
251The basic building blocks of unit testing are :dfn:`test cases` --- single
252scenarios that must be set up and checked for correctness. In :mod:`unittest`,
253test cases are represented by instances of :mod:`unittest`'s :class:`TestCase`
254class. To make your own test cases you must write subclasses of
255:class:`TestCase`, or use :class:`FunctionTestCase`.
256
257An instance of a :class:`TestCase`\ -derived class is an object that can
258completely run a single test method, together with optional set-up and tidy-up
259code.
260
261The testing code of a :class:`TestCase` instance should be entirely self
262contained, such that it can be run either in isolation or in arbitrary
263combination with any number of other test cases.
264
Benjamin Peterson52baa292009-03-24 00:56:30 +0000265The simplest :class:`TestCase` subclass will simply override the
266:meth:`~TestCase.runTest` method in order to perform specific testing code::
Georg Brandl116aa622007-08-15 14:28:22 +0000267
268 import unittest
269
270 class DefaultWidgetSizeTestCase(unittest.TestCase):
271 def runTest(self):
272 widget = Widget('The widget')
273 self.assertEqual(widget.size(), (50, 50), 'incorrect default size')
274
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000275Note that in order to test something, we use the one of the :meth:`assert\*`
Benjamin Petersond2397752009-06-27 23:45:02 +0000276methods provided by the :class:`TestCase` base class. If the test fails, an
277exception will be raised, and :mod:`unittest` will identify the test case as a
278:dfn:`failure`. Any other exceptions will be treated as :dfn:`errors`. This
279helps you identify where the problem is: :dfn:`failures` are caused by incorrect
280results - a 5 where you expected a 6. :dfn:`Errors` are caused by incorrect
281code - e.g., a :exc:`TypeError` caused by an incorrect function call.
Georg Brandl116aa622007-08-15 14:28:22 +0000282
283The way to run a test case will be described later. For now, note that to
284construct an instance of such a test case, we call its constructor without
285arguments::
286
287 testCase = DefaultWidgetSizeTestCase()
288
289Now, such test cases can be numerous, and their set-up can be repetitive. In
290the above case, constructing a :class:`Widget` in each of 100 Widget test case
291subclasses would mean unsightly duplication.
292
293Luckily, we can factor out such set-up code by implementing a method called
Benjamin Peterson52baa292009-03-24 00:56:30 +0000294:meth:`~TestCase.setUp`, which the testing framework will automatically call for
295us when we run the test::
Georg Brandl116aa622007-08-15 14:28:22 +0000296
297 import unittest
298
299 class SimpleWidgetTestCase(unittest.TestCase):
300 def setUp(self):
301 self.widget = Widget('The widget')
302
303 class DefaultWidgetSizeTestCase(SimpleWidgetTestCase):
304 def runTest(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000305 self.assertEqual(self.widget.size(), (50,50),
306 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000307
308 class WidgetResizeTestCase(SimpleWidgetTestCase):
309 def runTest(self):
310 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000311 self.assertEqual(self.widget.size(), (100,150),
312 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000313
Benjamin Peterson52baa292009-03-24 00:56:30 +0000314If the :meth:`~TestCase.setUp` method raises an exception while the test is
315running, the framework will consider the test to have suffered an error, and the
316:meth:`~TestCase.runTest` method will not be executed.
Georg Brandl116aa622007-08-15 14:28:22 +0000317
Benjamin Peterson52baa292009-03-24 00:56:30 +0000318Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
319after the :meth:`~TestCase.runTest` method has been run::
Georg Brandl116aa622007-08-15 14:28:22 +0000320
321 import unittest
322
323 class SimpleWidgetTestCase(unittest.TestCase):
324 def setUp(self):
325 self.widget = Widget('The widget')
326
327 def tearDown(self):
328 self.widget.dispose()
329 self.widget = None
330
Benjamin Peterson52baa292009-03-24 00:56:30 +0000331If :meth:`~TestCase.setUp` succeeded, the :meth:`~TestCase.tearDown` method will
332be run whether :meth:`~TestCase.runTest` succeeded or not.
Georg Brandl116aa622007-08-15 14:28:22 +0000333
334Such a working environment for the testing code is called a :dfn:`fixture`.
335
336Often, many small test cases will use the same fixture. In this case, we would
337end up subclassing :class:`SimpleWidgetTestCase` into many small one-method
338classes such as :class:`DefaultWidgetSizeTestCase`. This is time-consuming and
Georg Brandl116aa622007-08-15 14:28:22 +0000339discouraging, so in the same vein as JUnit, :mod:`unittest` provides a simpler
340mechanism::
341
342 import unittest
343
344 class WidgetTestCase(unittest.TestCase):
345 def setUp(self):
346 self.widget = Widget('The widget')
347
348 def tearDown(self):
349 self.widget.dispose()
350 self.widget = None
351
Ezio Melottid59e44a2010-02-28 03:46:13 +0000352 def test_default_size(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000353 self.assertEqual(self.widget.size(), (50,50),
354 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000355
Ezio Melottid59e44a2010-02-28 03:46:13 +0000356 def test_resize(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000357 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000358 self.assertEqual(self.widget.size(), (100,150),
359 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000360
Benjamin Peterson52baa292009-03-24 00:56:30 +0000361Here we have not provided a :meth:`~TestCase.runTest` method, but have instead
362provided two different test methods. Class instances will now each run one of
Ezio Melottid59e44a2010-02-28 03:46:13 +0000363the :meth:`test_\*` methods, with ``self.widget`` created and destroyed
Benjamin Peterson52baa292009-03-24 00:56:30 +0000364separately for each instance. When creating an instance we must specify the
365test method it is to run. We do this by passing the method name in the
366constructor::
Georg Brandl116aa622007-08-15 14:28:22 +0000367
Ezio Melottid59e44a2010-02-28 03:46:13 +0000368 defaultSizeTestCase = WidgetTestCase('test_default_size')
369 resizeTestCase = WidgetTestCase('test_resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000370
371Test case instances are grouped together according to the features they test.
372:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
373represented by :mod:`unittest`'s :class:`TestSuite` class::
374
375 widgetTestSuite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000376 widgetTestSuite.addTest(WidgetTestCase('test_default_size'))
377 widgetTestSuite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000378
379For the ease of running tests, as we will see later, it is a good idea to
380provide in each test module a callable object that returns a pre-built test
381suite::
382
383 def suite():
384 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000385 suite.addTest(WidgetTestCase('test_default_size'))
386 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000387 return suite
388
389or even::
390
391 def suite():
Ezio Melottid59e44a2010-02-28 03:46:13 +0000392 tests = ['test_default_size', 'test_resize']
Georg Brandl116aa622007-08-15 14:28:22 +0000393
394 return unittest.TestSuite(map(WidgetTestCase, tests))
395
396Since it is a common pattern to create a :class:`TestCase` subclass with many
397similarly named test functions, :mod:`unittest` provides a :class:`TestLoader`
398class that can be used to automate the process of creating a test suite and
399populating it with individual tests. For example, ::
400
401 suite = unittest.TestLoader().loadTestsFromTestCase(WidgetTestCase)
402
Ezio Melottid59e44a2010-02-28 03:46:13 +0000403will create a test suite that will run ``WidgetTestCase.test_default_size()`` and
404``WidgetTestCase.test_resize``. :class:`TestLoader` uses the ``'test'`` method
Georg Brandl116aa622007-08-15 14:28:22 +0000405name prefix to identify test methods automatically.
406
Mark Dickinsonc48d8342009-02-01 14:18:10 +0000407Note that the order in which the various test cases will be run is
408determined by sorting the test function names with respect to the
409built-in ordering for strings.
Georg Brandl116aa622007-08-15 14:28:22 +0000410
411Often it is desirable to group suites of test cases together, so as to run tests
412for the whole system at once. This is easy, since :class:`TestSuite` instances
413can be added to a :class:`TestSuite` just as :class:`TestCase` instances can be
414added to a :class:`TestSuite`::
415
416 suite1 = module1.TheTestSuite()
417 suite2 = module2.TheTestSuite()
418 alltests = unittest.TestSuite([suite1, suite2])
419
420You can place the definitions of test cases and test suites in the same modules
421as the code they are to test (such as :file:`widget.py`), but there are several
422advantages to placing the test code in a separate module, such as
423:file:`test_widget.py`:
424
425* The test module can be run standalone from the command line.
426
427* The test code can more easily be separated from shipped code.
428
429* There is less temptation to change test code to fit the code it tests without
430 a good reason.
431
432* Test code should be modified much less frequently than the code it tests.
433
434* Tested code can be refactored more easily.
435
436* Tests for modules written in C must be in separate modules anyway, so why not
437 be consistent?
438
439* If the testing strategy changes, there is no need to change the source code.
440
441
442.. _legacy-unit-tests:
443
444Re-using old test code
445----------------------
446
447Some users will find that they have existing test code that they would like to
448run from :mod:`unittest`, without converting every old test function to a
449:class:`TestCase` subclass.
450
451For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
452This subclass of :class:`TestCase` can be used to wrap an existing test
453function. Set-up and tear-down functions can also be provided.
454
455Given the following test function::
456
457 def testSomething():
458 something = makeSomething()
459 assert something.name is not None
460 # ...
461
462one can create an equivalent test case instance as follows::
463
464 testcase = unittest.FunctionTestCase(testSomething)
465
466If there are additional set-up and tear-down methods that should be called as
467part of the test case's operation, they can also be provided like so::
468
469 testcase = unittest.FunctionTestCase(testSomething,
470 setUp=makeSomethingDB,
471 tearDown=deleteSomethingDB)
472
473To make migrating existing test suites easier, :mod:`unittest` supports tests
474raising :exc:`AssertionError` to indicate test failure. However, it is
475recommended that you use the explicit :meth:`TestCase.fail\*` and
476:meth:`TestCase.assert\*` methods instead, as future versions of :mod:`unittest`
477may treat :exc:`AssertionError` differently.
478
479.. note::
480
Benjamin Petersond2397752009-06-27 23:45:02 +0000481 Even though :class:`FunctionTestCase` can be used to quickly convert an
482 existing test base over to a :mod:`unittest`\ -based system, this approach is
483 not recommended. Taking the time to set up proper :class:`TestCase`
484 subclasses will make future test refactorings infinitely easier.
Georg Brandl116aa622007-08-15 14:28:22 +0000485
Benjamin Peterson52baa292009-03-24 00:56:30 +0000486In some cases, the existing tests may have been written using the :mod:`doctest`
487module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
488automatically build :class:`unittest.TestSuite` instances from the existing
489:mod:`doctest`\ -based tests.
490
Georg Brandl116aa622007-08-15 14:28:22 +0000491
Benjamin Peterson5254c042009-03-23 22:25:03 +0000492.. _unittest-skipping:
493
494Skipping tests and expected failures
495------------------------------------
496
Michael Foordf5c851a2010-02-05 21:48:03 +0000497.. versionadded:: 3.1
498
Benjamin Peterson5254c042009-03-23 22:25:03 +0000499Unittest supports skipping individual test methods and even whole classes of
500tests. In addition, it supports marking a test as a "expected failure," a test
501that is broken and will fail, but shouldn't be counted as a failure on a
502:class:`TestResult`.
503
504Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
505or one of its conditional variants.
506
507Basic skipping looks like this: ::
508
509 class MyTestCase(unittest.TestCase):
510
511 @unittest.skip("demonstrating skipping")
512 def test_nothing(self):
513 self.fail("shouldn't happen")
514
Benjamin Petersond2397752009-06-27 23:45:02 +0000515 @unittest.skipIf(mylib.__version__ < (1, 3),
516 "not supported in this library version")
Benjamin Petersonded31c42009-03-30 15:04:16 +0000517 def test_format(self):
518 # Tests that work for only a certain version of the library.
519 pass
520
521 @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
522 def test_windows_support(self):
523 # windows specific testing code
524 pass
525
Benjamin Peterson5254c042009-03-23 22:25:03 +0000526This is the output of running the example above in verbose mode: ::
527
Benjamin Petersonded31c42009-03-30 15:04:16 +0000528 test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000529 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
Benjamin Petersonded31c42009-03-30 15:04:16 +0000530 test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000531
532 ----------------------------------------------------------------------
Benjamin Petersonded31c42009-03-30 15:04:16 +0000533 Ran 3 tests in 0.005s
534
535 OK (skipped=3)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000536
537Classes can be skipped just like methods: ::
538
539 @skip("showing class skipping")
540 class MySkippedTestCase(unittest.TestCase):
541 def test_not_run(self):
542 pass
543
Benjamin Peterson52baa292009-03-24 00:56:30 +0000544:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
545that needs to be set up is not available.
546
Benjamin Peterson5254c042009-03-23 22:25:03 +0000547Expected failures use the :func:`expectedFailure` decorator. ::
548
549 class ExpectedFailureTestCase(unittest.TestCase):
550 @unittest.expectedFailure
551 def test_fail(self):
552 self.assertEqual(1, 0, "broken")
553
554It's easy to roll your own skipping decorators by making a decorator that calls
555:func:`skip` on the test when it wants it to be skipped. This decorator skips
556the test unless the passed object has a certain attribute: ::
557
558 def skipUnlessHasattr(obj, attr):
559 if hasattr(obj, attr):
560 return lambda func: func
561 return unittest.skip("{0!r} doesn't have {1!r}".format(obj, attr))
562
563The following decorators implement test skipping and expected failures:
564
565.. function:: skip(reason)
566
567 Unconditionally skip the decorated test. *reason* should describe why the
568 test is being skipped.
569
570.. function:: skipIf(condition, reason)
571
572 Skip the decorated test if *condition* is true.
573
574.. function:: skipUnless(condition, reason)
575
576 Skip the decoratored test unless *condition* is true.
577
578.. function:: expectedFailure
579
580 Mark the test as an expected failure. If the test fails when run, the test
581 is not counted as a failure.
582
583
Georg Brandl116aa622007-08-15 14:28:22 +0000584.. _unittest-contents:
585
586Classes and functions
587---------------------
588
Benjamin Peterson52baa292009-03-24 00:56:30 +0000589This section describes in depth the API of :mod:`unittest`.
590
591
592.. _testcase-objects:
593
594Test cases
595~~~~~~~~~~
Georg Brandl116aa622007-08-15 14:28:22 +0000596
Georg Brandl7f01a132009-09-16 15:58:14 +0000597.. class:: TestCase(methodName='runTest')
Georg Brandl116aa622007-08-15 14:28:22 +0000598
599 Instances of the :class:`TestCase` class represent the smallest testable units
600 in the :mod:`unittest` universe. This class is intended to be used as a base
601 class, with specific tests being implemented by concrete subclasses. This class
602 implements the interface needed by the test runner to allow it to drive the
603 test, and methods that the test code can use to check for and report various
604 kinds of failure.
605
606 Each instance of :class:`TestCase` will run a single test method: the method
607 named *methodName*. If you remember, we had an earlier example that went
608 something like this::
609
610 def suite():
611 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000612 suite.addTest(WidgetTestCase('test_default_size'))
613 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000614 return suite
615
616 Here, we create two instances of :class:`WidgetTestCase`, each of which runs a
617 single test.
618
Benjamin Peterson52baa292009-03-24 00:56:30 +0000619 *methodName* defaults to :meth:`runTest`.
620
621 :class:`TestCase` instances provide three groups of methods: one group used
622 to run the test, another used by the test implementation to check conditions
623 and report failures, and some inquiry methods allowing information about the
624 test itself to be gathered.
625
626 Methods in the first group (running the test) are:
627
628
629 .. method:: setUp()
630
631 Method called to prepare the test fixture. This is called immediately
632 before calling the test method; any exception raised by this method will
633 be considered an error rather than a test failure. The default
634 implementation does nothing.
635
636
637 .. method:: tearDown()
638
639 Method called immediately after the test method has been called and the
640 result recorded. This is called even if the test method raised an
641 exception, so the implementation in subclasses may need to be particularly
642 careful about checking internal state. Any exception raised by this
643 method will be considered an error rather than a test failure. This
644 method will only be called if the :meth:`setUp` succeeds, regardless of
645 the outcome of the test method. The default implementation does nothing.
646
647
Georg Brandl7f01a132009-09-16 15:58:14 +0000648 .. method:: run(result=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000649
650 Run the test, collecting the result into the test result object passed as
651 *result*. If *result* is omitted or :const:`None`, a temporary result
Alexandre Vassalotti260484d2009-07-17 11:43:26 +0000652 object is created (by calling the :meth:`defaultTestResult` method) and
653 used. The result object is not returned to :meth:`run`'s caller.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000654
655 The same effect may be had by simply calling the :class:`TestCase`
656 instance.
657
658
Benjamin Petersone549ead2009-03-28 21:42:05 +0000659 .. method:: skipTest(reason)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000660
661 Calling this during the a test method or :meth:`setUp` skips the current
662 test. See :ref:`unittest-skipping` for more information.
663
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000664 .. versionadded:: 2.7
665
Benjamin Peterson52baa292009-03-24 00:56:30 +0000666
667 .. method:: debug()
668
669 Run the test without collecting the result. This allows exceptions raised
670 by the test to be propagated to the caller, and can be used to support
671 running tests under a debugger.
672
673 The test code can use any of the following methods to check for and report
674 failures.
675
676
Georg Brandl7f01a132009-09-16 15:58:14 +0000677 .. method:: assertTrue(expr, msg=None)
678 assert_(expr, msg=None)
679 failUnless(expr, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000680
Georg Brandlff2ad0e2009-04-27 16:51:45 +0000681 Signal a test failure if *expr* is false; the explanation for the failure
Benjamin Peterson52baa292009-03-24 00:56:30 +0000682 will be *msg* if given, otherwise it will be :const:`None`.
683
Raymond Hettinger35a88362009-04-09 00:08:24 +0000684 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +0000685 :meth:`failUnless`; use one of the ``assert`` variants.
Michael Foord34c94622010-02-10 15:51:42 +0000686 :meth:`assert_`; use :meth:`assertTrue`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000687
Benjamin Peterson52baa292009-03-24 00:56:30 +0000688
Georg Brandl7f01a132009-09-16 15:58:14 +0000689 .. method:: assertEqual(first, second, msg=None)
690 failUnlessEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000691
692 Test that *first* and *second* are equal. If the values do not compare
693 equal, the test will fail with the explanation given by *msg*, or
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000694 :const:`None`. Note that using :meth:`assertEqual` improves upon
695 doing the comparison as the first parameter to :meth:`assertTrue`: the
696 default value for *msg* include representations of both *first* and
697 *second*.
698
699 In addition, if *first* and *second* are the exact same type and one of
Michael Foord02834952010-02-08 23:10:39 +0000700 list, tuple, dict, set, frozenset or str or any type that a subclass
701 registers with :meth:`addTypeEqualityFunc` the type specific equality
702 function will be called in order to generate a more useful default
703 error message.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000704
Raymond Hettinger35a88362009-04-09 00:08:24 +0000705 .. versionchanged:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000706 Added the automatic calling of type specific equality function.
707
Michael Foord28a817e2010-02-09 00:03:57 +0000708 .. versionchanged:: 3.2
709 :meth:`assertMultiLineEqual` added as the default type equality
710 function for comparing strings.
Michael Foord02834952010-02-08 23:10:39 +0000711
Raymond Hettinger35a88362009-04-09 00:08:24 +0000712 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +0000713 :meth:`failUnlessEqual`; use :meth:`assertEqual`.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000714
715
Georg Brandl7f01a132009-09-16 15:58:14 +0000716 .. method:: assertNotEqual(first, second, msg=None)
717 failIfEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000718
719 Test that *first* and *second* are not equal. If the values do compare
720 equal, the test will fail with the explanation given by *msg*, or
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000721 :const:`None`. Note that using :meth:`assertNotEqual` improves upon doing
722 the comparison as the first parameter to :meth:`assertTrue` is that the
Benjamin Peterson52baa292009-03-24 00:56:30 +0000723 default value for *msg* can be computed to include representations of both
724 *first* and *second*.
725
Raymond Hettinger35a88362009-04-09 00:08:24 +0000726 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +0000727 :meth:`failIfEqual`; use :meth:`assertNotEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000728
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000729
Georg Brandl7f01a132009-09-16 15:58:14 +0000730 .. method:: assertAlmostEqual(first, second, *, places=7, msg=None)
731 failUnlessAlmostEqual(first, second, *, places=7, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000732
733 Test that *first* and *second* are approximately equal by computing the
734 difference, rounding to the given number of decimal *places* (default 7),
735 and comparing to zero.
736
737 Note that comparing a given number of decimal places is not the same as
738 comparing a given number of significant digits. If the values do not
739 compare equal, the test will fail with the explanation given by *msg*, or
740 :const:`None`.
741
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000742 .. versionchanged:: 3.2
743 Objects that compare equal are automatically almost equal.
744
Raymond Hettinger35a88362009-04-09 00:08:24 +0000745 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +0000746 :meth:`failUnlessAlmostEqual`; use :meth:`assertAlmostEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000747
Benjamin Peterson52baa292009-03-24 00:56:30 +0000748
Georg Brandl7f01a132009-09-16 15:58:14 +0000749 .. method:: assertNotAlmostEqual(first, second, *, places=7, msg=None)
750 failIfAlmostEqual(first, second, *, places=7, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000751
752 Test that *first* and *second* are not approximately equal by computing
753 the difference, rounding to the given number of decimal *places* (default
754 7), and comparing to zero.
755
756 Note that comparing a given number of decimal places is not the same as
757 comparing a given number of significant digits. If the values do not
758 compare equal, the test will fail with the explanation given by *msg*, or
759 :const:`None`.
760
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000761 .. versionchanged:: 3.2
762 Objects that compare equal automatically fail.
763
Raymond Hettinger35a88362009-04-09 00:08:24 +0000764 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +0000765 :meth:`failIfAlmostEqual`; use :meth:`assertNotAlmostEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000766
767
768 .. method:: assertGreater(first, second, msg=None)
769 assertGreaterEqual(first, second, msg=None)
770 assertLess(first, second, msg=None)
771 assertLessEqual(first, second, msg=None)
772
773 Test that *first* is respectively >, >=, < or <= than *second* depending
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000774 on the method name. If not, the test will fail with an explanation
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000775 or with the explanation given by *msg*::
776
777 >>> self.assertGreaterEqual(3, 4)
778 AssertionError: "3" unexpectedly not greater than or equal to "4"
779
Raymond Hettinger35a88362009-04-09 00:08:24 +0000780 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000781
782
783 .. method:: assertMultiLineEqual(self, first, second, msg=None)
784
785 Test that the multiline string *first* is equal to the string *second*.
786 When not equal a diff of the two strings highlighting the differences
Michael Foord02834952010-02-08 23:10:39 +0000787 will be included in the error message. This method is used by default
788 when comparing strings with :meth:`assertEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000789
Michael Foordabd91d52010-03-20 18:09:14 +0000790 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000791
Raymond Hettinger35a88362009-04-09 00:08:24 +0000792 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000793
794
Ezio Melotti732b6822010-01-16 19:40:06 +0000795 .. method:: assertRegexpMatches(text, regexp, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000796
797 Verifies that a *regexp* search matches *text*. Fails with an error
798 message including the pattern and the *text*. *regexp* may be
799 a regular expression object or a string containing a regular expression
800 suitable for use by :func:`re.search`.
801
Raymond Hettinger35a88362009-04-09 00:08:24 +0000802 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000803
804
805 .. method:: assertIn(first, second, msg=None)
806 assertNotIn(first, second, msg=None)
807
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000808 Tests that *first* is or is not in *second* with an explanatory error
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000809 message as appropriate.
810
Michael Foordabd91d52010-03-20 18:09:14 +0000811 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000812
Raymond Hettinger35a88362009-04-09 00:08:24 +0000813 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000814
815
Michael Foorde9abbee2010-02-05 20:54:27 +0000816 .. method:: assertSameElements(actual, expected, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000817
Benjamin Peterson5e55b3e2010-02-03 02:35:45 +0000818 Test that sequence *expected* contains the same elements as *actual*,
819 regardless of their order. When they don't, an error message listing
820 the differences between the sequences will be generated.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000821
Michael Foorde9abbee2010-02-05 20:54:27 +0000822 Duplicate elements are ignored when comparing *actual* and *expected*.
823 It is the equivalent of ``assertEqual(set(expected), set(actual))``
Michael Foordabd91d52010-03-20 18:09:14 +0000824 but it works with sequences of unhashable objects as well. Because
825 duplicates are ignored, this method has been deprecated in favour of
826 :meth:`assertItemsEqual`.
Michael Foorde9abbee2010-02-05 20:54:27 +0000827
Michael Foordabd91d52010-03-20 18:09:14 +0000828 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000829
Raymond Hettinger35a88362009-04-09 00:08:24 +0000830 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000831
Michael Foordabd91d52010-03-20 18:09:14 +0000832 .. deprecated:: 3.2
833
834 .. method:: assertItemsEqual(actual, expected, msg=None)
835
836 Test that sequence *expected* contains the same elements as *actual*,
837 regardless of their order. When they don't, an error message listing the
838 differences between the sequences will be generated.
839
840 Duplicate elements are *not* ignored when comparing *actual* and
841 *expected*. It verifies if each element has the same count in both
842 sequences. It is the equivalent of ``assertEqual(sorted(expected),
843 sorted(actual))`` but it works with sequences of unhashable objects as
844 well.
845
846 If specified, *msg* will be used as the error message on failure.
847
848 .. versionadded:: 3.2
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000849
850 .. method:: assertSetEqual(set1, set2, msg=None)
851
852 Tests that two sets are equal. If not, an error message is constructed
Michael Foord02834952010-02-08 23:10:39 +0000853 that lists the differences between the sets. This method is used by
854 default when comparing sets or frozensets with :meth:`assertEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000855
856 Fails if either of *set1* or *set2* does not have a :meth:`set.difference`
857 method.
858
Michael Foordabd91d52010-03-20 18:09:14 +0000859 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000860
Raymond Hettinger35a88362009-04-09 00:08:24 +0000861 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000862
863
864 .. method:: assertDictEqual(expected, actual, msg=None)
865
866 Test that two dictionaries are equal. If not, an error message is
Michael Foord02834952010-02-08 23:10:39 +0000867 constructed that shows the differences in the dictionaries. This
868 method will be used by default to compare dictionaries in
869 calls to :meth:`assertEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000870
Michael Foordabd91d52010-03-20 18:09:14 +0000871 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000872
Raymond Hettinger35a88362009-04-09 00:08:24 +0000873 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000874
875
876 .. method:: assertDictContainsSubset(expected, actual, msg=None)
877
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000878 Tests whether the key/value pairs in dictionary *actual* are a
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000879 superset of those in *expected*. If not, an error message listing
880 the missing keys and mismatched values is generated.
881
Michael Foordabd91d52010-03-20 18:09:14 +0000882 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000883
Raymond Hettinger35a88362009-04-09 00:08:24 +0000884 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000885
886
887 .. method:: assertListEqual(list1, list2, msg=None)
888 assertTupleEqual(tuple1, tuple2, msg=None)
889
890 Tests that two lists or tuples are equal. If not an error message is
891 constructed that shows only the differences between the two. An error
892 is also raised if either of the parameters are of the wrong type.
Michael Foord02834952010-02-08 23:10:39 +0000893 These methods are used by default when comparing lists or tuples with
894 :meth:`assertEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000895
Michael Foordabd91d52010-03-20 18:09:14 +0000896 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000897
Raymond Hettinger35a88362009-04-09 00:08:24 +0000898 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000899
900
901 .. method:: assertSequenceEqual(seq1, seq2, msg=None, seq_type=None)
902
903 Tests that two sequences are equal. If a *seq_type* is supplied, both
904 *seq1* and *seq2* must be instances of *seq_type* or a failure will
905 be raised. If the sequences are different an error message is
906 constructed that shows the difference between the two.
907
Michael Foordabd91d52010-03-20 18:09:14 +0000908 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000909
910 This method is used to implement :meth:`assertListEqual` and
911 :meth:`assertTupleEqual`.
912
Raymond Hettinger35a88362009-04-09 00:08:24 +0000913 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000914
Benjamin Peterson52baa292009-03-24 00:56:30 +0000915
Georg Brandl7f01a132009-09-16 15:58:14 +0000916 .. method:: assertRaises(exception, callable, *args, **kwds)
917 failUnlessRaises(exception, callable, *args, **kwds)
918 assertRaises(exception)
919 failUnlessRaises(exception)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000920
921 Test that an exception is raised when *callable* is called with any
922 positional or keyword arguments that are also passed to
923 :meth:`assertRaises`. The test passes if *exception* is raised, is an
924 error if another exception is raised, or fails if no exception is raised.
925 To catch any of a group of exceptions, a tuple containing the exception
926 classes may be passed as *exception*.
927
Georg Brandl7f01a132009-09-16 15:58:14 +0000928 If only the *exception* argument is given, returns a context manager so
929 that the code under test can be written inline rather than as a function::
Benjamin Petersonded31c42009-03-30 15:04:16 +0000930
Michael Foord41531f22010-02-05 21:13:40 +0000931 with self.assertRaises(SomeException):
Benjamin Petersonded31c42009-03-30 15:04:16 +0000932 do_something()
933
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000934 The context manager will store the caught exception object in its
Ezio Melotti49008232010-02-08 21:57:48 +0000935 :attr:`exception` attribute. This can be useful if the intention
Michael Foord41531f22010-02-05 21:13:40 +0000936 is to perform additional checks on the exception raised::
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000937
Michael Foord41531f22010-02-05 21:13:40 +0000938 with self.assertRaises(SomeException) as cm:
939 do_something()
940
Ezio Melotti49008232010-02-08 21:57:48 +0000941 the_exception = cm.exception
Michael Foordb112a412010-02-05 23:32:33 +0000942 self.assertEqual(the_exception.error_code, 3)
Michael Foord41531f22010-02-05 21:13:40 +0000943
Ezio Melotti49008232010-02-08 21:57:48 +0000944 .. versionchanged:: 3.1
Benjamin Petersonded31c42009-03-30 15:04:16 +0000945 Added the ability to use :meth:`assertRaises` as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000946
Ezio Melotti49008232010-02-08 21:57:48 +0000947 .. versionchanged:: 3.2
948 Added the :attr:`exception` attribute.
949
Raymond Hettinger35a88362009-04-09 00:08:24 +0000950 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +0000951 :meth:`failUnlessRaises`; use :meth:`assertRaises`.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000952
Benjamin Peterson52baa292009-03-24 00:56:30 +0000953
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000954 .. method:: assertRaisesRegexp(exception, regexp[, callable, ...])
955
956 Like :meth:`assertRaises` but also tests that *regexp* matches
957 on the string representation of the raised exception. *regexp* may be
958 a regular expression object or a string containing a regular expression
959 suitable for use by :func:`re.search`. Examples::
960
961 self.assertRaisesRegexp(ValueError, 'invalid literal for.*XYZ$',
962 int, 'XYZ')
963
964 or::
965
966 with self.assertRaisesRegexp(ValueError, 'literal'):
967 int('XYZ')
968
Raymond Hettinger35a88362009-04-09 00:08:24 +0000969 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000970
971
Georg Brandl7f01a132009-09-16 15:58:14 +0000972 .. method:: assertIsNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000973
974 This signals a test failure if *expr* is not None.
975
Raymond Hettinger35a88362009-04-09 00:08:24 +0000976 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000977
978
Georg Brandl7f01a132009-09-16 15:58:14 +0000979 .. method:: assertIsNotNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000980
981 The inverse of the :meth:`assertIsNone` method.
982 This signals a test failure if *expr* is None.
983
Raymond Hettinger35a88362009-04-09 00:08:24 +0000984 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000985
986
Georg Brandl7f01a132009-09-16 15:58:14 +0000987 .. method:: assertIs(expr1, expr2, msg=None)
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000988
989 This signals a test failure if *expr1* and *expr2* don't evaluate to the same
990 object.
991
Georg Brandl705d9d52009-05-05 09:29:50 +0000992 .. versionadded:: 3.1
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000993
994
Georg Brandl7f01a132009-09-16 15:58:14 +0000995 .. method:: assertIsNot(expr1, expr2, msg=None)
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000996
997 The inverse of the :meth:`assertIs` method.
998 This signals a test failure if *expr1* and *expr2* evaluate to the same
999 object.
1000
Georg Brandl705d9d52009-05-05 09:29:50 +00001001 .. versionadded:: 3.1
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001002
1003
Benjamin Peterson6e8c7572009-10-04 20:19:21 +00001004 .. method:: assertIsInstance(obj, cls[, msg])
1005
1006 This signals a test failure if *obj* is not an instance of *cls* (which
1007 can be a class or a tuple of classes, as supported by :func:`isinstance`).
1008
1009 .. versionadded:: 3.2
1010
1011
1012 .. method:: assertNotIsInstance(obj, cls[, msg])
1013
1014 The inverse of the :meth:`assertIsInstance` method. This signals a test
1015 failure if *obj* is an instance of *cls*.
1016
1017 .. versionadded:: 3.2
1018
1019
Georg Brandl7f01a132009-09-16 15:58:14 +00001020 .. method:: assertFalse(expr, msg=None)
1021 failIf(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001022
1023 The inverse of the :meth:`assertTrue` method is the :meth:`assertFalse` method.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001024 This signals a test failure if *expr* is true, with *msg* or :const:`None`
1025 for the error message.
1026
Raymond Hettinger35a88362009-04-09 00:08:24 +00001027 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +00001028 :meth:`failIf`; use :meth:`assertFalse`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001029
Benjamin Peterson52baa292009-03-24 00:56:30 +00001030
Georg Brandl7f01a132009-09-16 15:58:14 +00001031 .. method:: fail(msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001032
1033 Signals a test failure unconditionally, with *msg* or :const:`None` for
1034 the error message.
1035
1036
1037 .. attribute:: failureException
1038
1039 This class attribute gives the exception raised by the test method. If a
1040 test framework needs to use a specialized exception, possibly to carry
1041 additional information, it must subclass this exception in order to "play
1042 fair" with the framework. The initial value of this attribute is
1043 :exc:`AssertionError`.
1044
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001045
1046 .. attribute:: longMessage
1047
1048 If set to True then any explicit failure message you pass in to the
1049 assert methods will be appended to the end of the normal failure message.
1050 The normal messages contain useful information about the objects involved,
1051 for example the message from assertEqual shows you the repr of the two
1052 unequal objects. Setting this attribute to True allows you to have a
1053 custom error message in addition to the normal one.
1054
1055 This attribute defaults to False, meaning that a custom message passed
1056 to an assert method will silence the normal message.
1057
1058 The class setting can be overridden in individual tests by assigning an
1059 instance attribute to True or False before calling the assert methods.
1060
Raymond Hettinger35a88362009-04-09 00:08:24 +00001061 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001062
1063
Benjamin Peterson52baa292009-03-24 00:56:30 +00001064 Testing frameworks can use the following methods to collect information on
1065 the test:
1066
1067
1068 .. method:: countTestCases()
1069
1070 Return the number of tests represented by this test object. For
1071 :class:`TestCase` instances, this will always be ``1``.
1072
1073
1074 .. method:: defaultTestResult()
1075
1076 Return an instance of the test result class that should be used for this
1077 test case class (if no other result instance is provided to the
1078 :meth:`run` method).
1079
1080 For :class:`TestCase` instances, this will always be an instance of
1081 :class:`TestResult`; subclasses of :class:`TestCase` should override this
1082 as necessary.
1083
1084
1085 .. method:: id()
1086
1087 Return a string identifying the specific test case. This is usually the
1088 full name of the test method, including the module and class name.
1089
1090
1091 .. method:: shortDescription()
1092
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001093 Returns a description of the test, or :const:`None` if no description
1094 has been provided. The default implementation of this method
1095 returns the first line of the test method's docstring, if available,
Michael Foord34c94622010-02-10 15:51:42 +00001096 or :const:`None`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001097
Michael Foord34c94622010-02-10 15:51:42 +00001098 .. versionchanged:: 3.1,3.2
1099 In 3.1 this was changed to add the test name to the short description
1100 even in the presence of a docstring. This caused compatibility issues
1101 with unittest extensions and adding the test name was moved to the
1102 :class:`TextTestResult`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001103
1104 .. method:: addTypeEqualityFunc(typeobj, function)
1105
1106 Registers a type specific :meth:`assertEqual` equality checking
1107 function to be called by :meth:`assertEqual` when both objects it has
1108 been asked to compare are exactly *typeobj* (not subclasses).
1109 *function* must take two positional arguments and a third msg=None
1110 keyword argument just as :meth:`assertEqual` does. It must raise
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001111 ``self.failureException`` when inequality between the first two
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001112 parameters is detected.
1113
1114 One good use of custom equality checking functions for a type
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001115 is to raise ``self.failureException`` with an error message useful
1116 for debugging the problem by explaining the inequalities in detail.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001117
Raymond Hettinger35a88362009-04-09 00:08:24 +00001118 .. versionadded:: 3.1
Georg Brandl116aa622007-08-15 14:28:22 +00001119
1120
Georg Brandl7f01a132009-09-16 15:58:14 +00001121 .. method:: addCleanup(function, *args, **kwargs)
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001122
1123 Add a function to be called after :meth:`tearDown` to cleanup resources
1124 used during the test. Functions will be called in reverse order to the
1125 order they are added (LIFO). They are called with any arguments and
1126 keyword arguments passed into :meth:`addCleanup` when they are
1127 added.
1128
1129 If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
1130 then any cleanup functions added will still be called.
1131
Georg Brandl853947a2010-01-31 18:53:23 +00001132 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001133
1134
1135 .. method:: doCleanups()
1136
1137 This method is called uncoditionally after :meth:`tearDown`, or
1138 after :meth:`setUp` if :meth:`setUp` raises an exception.
1139
1140 It is responsible for calling all the cleanup functions added by
1141 :meth:`addCleanup`. If you need cleanup functions to be called
1142 *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
1143 yourself.
1144
1145 :meth:`doCleanups` pops methods off the stack of cleanup
1146 functions one at a time, so it can be called at any time.
1147
Georg Brandl853947a2010-01-31 18:53:23 +00001148 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001149
1150
Georg Brandl7f01a132009-09-16 15:58:14 +00001151.. class:: FunctionTestCase(testFunc, setUp=None, tearDown=None, description=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001152
1153 This class implements the portion of the :class:`TestCase` interface which
Benjamin Petersond2397752009-06-27 23:45:02 +00001154 allows the test runner to drive the test, but does not provide the methods
1155 which test code can use to check and report errors. This is used to create
1156 test cases using legacy test code, allowing it to be integrated into a
1157 :mod:`unittest`-based test framework.
Georg Brandl116aa622007-08-15 14:28:22 +00001158
1159
Benjamin Peterson52baa292009-03-24 00:56:30 +00001160.. _testsuite-objects:
1161
Benjamin Peterson52baa292009-03-24 00:56:30 +00001162Grouping tests
1163~~~~~~~~~~~~~~
1164
Georg Brandl7f01a132009-09-16 15:58:14 +00001165.. class:: TestSuite(tests=())
Georg Brandl116aa622007-08-15 14:28:22 +00001166
1167 This class represents an aggregation of individual tests cases and test suites.
1168 The class presents the interface needed by the test runner to allow it to be run
1169 as any other test case. Running a :class:`TestSuite` instance is the same as
1170 iterating over the suite, running each test individually.
1171
1172 If *tests* is given, it must be an iterable of individual test cases or other
1173 test suites that will be used to build the suite initially. Additional methods
1174 are provided to add test cases and suites to the collection later on.
1175
Benjamin Peterson14a3dd72009-05-25 00:51:58 +00001176 :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
1177 they do not actually implement a test. Instead, they are used to aggregate
1178 tests into groups of tests that should be run together. Some additional
1179 methods are available to add tests to :class:`TestSuite` instances:
Benjamin Peterson52baa292009-03-24 00:56:30 +00001180
1181
1182 .. method:: TestSuite.addTest(test)
1183
1184 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1185
1186
1187 .. method:: TestSuite.addTests(tests)
1188
1189 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
1190 instances to this test suite.
1191
Benjamin Petersond2397752009-06-27 23:45:02 +00001192 This is equivalent to iterating over *tests*, calling :meth:`addTest` for
1193 each element.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001194
1195 :class:`TestSuite` shares the following methods with :class:`TestCase`:
1196
1197
1198 .. method:: run(result)
1199
1200 Run the tests associated with this suite, collecting the result into the
1201 test result object passed as *result*. Note that unlike
1202 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
1203 be passed in.
1204
1205
1206 .. method:: debug()
1207
1208 Run the tests associated with this suite without collecting the
1209 result. This allows exceptions raised by the test to be propagated to the
1210 caller and can be used to support running tests under a debugger.
1211
1212
1213 .. method:: countTestCases()
1214
1215 Return the number of tests represented by this test object, including all
1216 individual tests and sub-suites.
1217
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001218
1219 .. method:: __iter__()
1220
1221 Tests grouped by a :class:`TestSuite` are always accessed by iteration.
1222 Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
1223 that this method maybe called several times on a single suite
1224 (for example when counting tests or comparing for equality)
1225 so the tests returned must be the same for repeated iterations.
1226
Georg Brandl853947a2010-01-31 18:53:23 +00001227 .. versionchanged:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001228 In earlier versions the :class:`TestSuite` accessed tests directly rather
1229 than through iteration, so overriding :meth:`__iter__` wasn't sufficient
1230 for providing tests.
1231
Benjamin Peterson52baa292009-03-24 00:56:30 +00001232 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
1233 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1234
1235
Benjamin Peterson52baa292009-03-24 00:56:30 +00001236Loading and running tests
1237~~~~~~~~~~~~~~~~~~~~~~~~~
1238
Georg Brandl116aa622007-08-15 14:28:22 +00001239.. class:: TestLoader()
1240
Benjamin Peterson52baa292009-03-24 00:56:30 +00001241 The :class:`TestLoader` class is used to create test suites from classes and
1242 modules. Normally, there is no need to create an instance of this class; the
1243 :mod:`unittest` module provides an instance that can be shared as
1244 ``unittest.defaultTestLoader``. Using a subclass or instance, however, allows
1245 customization of some configurable properties.
1246
1247 :class:`TestLoader` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001248
Michael Foordabd91d52010-03-20 18:09:14 +00001249a
Benjamin Peterson52baa292009-03-24 00:56:30 +00001250 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl116aa622007-08-15 14:28:22 +00001251
Benjamin Peterson52baa292009-03-24 00:56:30 +00001252 Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
1253 :class:`testCaseClass`.
1254
1255
1256 .. method:: loadTestsFromModule(module)
1257
1258 Return a suite of all tests cases contained in the given module. This
1259 method searches *module* for classes derived from :class:`TestCase` and
1260 creates an instance of the class for each test method defined for the
1261 class.
1262
Georg Brandle720c0a2009-04-27 16:20:50 +00001263 .. note::
Benjamin Peterson52baa292009-03-24 00:56:30 +00001264
1265 While using a hierarchy of :class:`TestCase`\ -derived classes can be
1266 convenient in sharing fixtures and helper functions, defining test
1267 methods on base classes that are not intended to be instantiated
1268 directly does not play well with this method. Doing so, however, can
1269 be useful when the fixtures are different and defined in subclasses.
1270
Benjamin Petersond2397752009-06-27 23:45:02 +00001271 If a module provides a ``load_tests`` function it will be called to
1272 load the tests. This allows modules to customize test loading.
1273 This is the `load_tests protocol`_.
1274
Georg Brandl853947a2010-01-31 18:53:23 +00001275 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001276 Support for ``load_tests`` added.
1277
Benjamin Peterson52baa292009-03-24 00:56:30 +00001278
Georg Brandl7f01a132009-09-16 15:58:14 +00001279 .. method:: loadTestsFromName(name, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001280
1281 Return a suite of all tests cases given a string specifier.
1282
1283 The specifier *name* is a "dotted name" that may resolve either to a
1284 module, a test case class, a test method within a test case class, a
1285 :class:`TestSuite` instance, or a callable object which returns a
1286 :class:`TestCase` or :class:`TestSuite` instance. These checks are
1287 applied in the order listed here; that is, a method on a possible test
1288 case class will be picked up as "a test method within a test case class",
1289 rather than "a callable object".
1290
1291 For example, if you have a module :mod:`SampleTests` containing a
1292 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
1293 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
Benjamin Petersond2397752009-06-27 23:45:02 +00001294 specifier ``'SampleTests.SampleTestCase'`` would cause this method to
1295 return a suite which will run all three test methods. Using the specifier
1296 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
1297 suite which will run only the :meth:`test_two` test method. The specifier
1298 can refer to modules and packages which have not been imported; they will
1299 be imported as a side-effect.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001300
1301 The method optionally resolves *name* relative to the given *module*.
1302
1303
Georg Brandl7f01a132009-09-16 15:58:14 +00001304 .. method:: loadTestsFromNames(names, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001305
1306 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
1307 than a single name. The return value is a test suite which supports all
1308 the tests defined for each name.
1309
1310
1311 .. method:: getTestCaseNames(testCaseClass)
1312
1313 Return a sorted sequence of method names found within *testCaseClass*;
1314 this should be a subclass of :class:`TestCase`.
1315
Benjamin Petersond2397752009-06-27 23:45:02 +00001316
1317 .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)
1318
1319 Find and return all test modules from the specified start directory,
1320 recursing into subdirectories to find them. Only test files that match
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001321 *pattern* will be loaded. (Using shell style pattern matching.) Only
1322 module names that are importable (i.e. are valid Python identifiers) will
1323 be loaded.
Benjamin Petersond2397752009-06-27 23:45:02 +00001324
1325 All test modules must be importable from the top level of the project. If
1326 the start directory is not the top level directory then the top level
1327 directory must be specified separately.
1328
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001329 If importing a module fails, for example due to a syntax error, then this
1330 will be recorded as a single error and discovery will continue.
1331
Benjamin Petersond2397752009-06-27 23:45:02 +00001332 If a test package name (directory with :file:`__init__.py`) matches the
1333 pattern then the package will be checked for a ``load_tests``
1334 function. If this exists then it will be called with *loader*, *tests*,
1335 *pattern*.
1336
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001337 If load_tests exists then discovery does *not* recurse into the package,
Benjamin Petersond2397752009-06-27 23:45:02 +00001338 ``load_tests`` is responsible for loading all tests in the package.
1339
1340 The pattern is deliberately not stored as a loader attribute so that
1341 packages can continue discovery themselves. *top_level_dir* is stored so
1342 ``load_tests`` does not need to pass this argument in to
1343 ``loader.discover()``.
1344
Georg Brandl853947a2010-01-31 18:53:23 +00001345 .. versionadded:: 3.2
1346
Benjamin Petersond2397752009-06-27 23:45:02 +00001347
Benjamin Peterson52baa292009-03-24 00:56:30 +00001348 The following attributes of a :class:`TestLoader` can be configured either by
1349 subclassing or assignment on an instance:
1350
1351
1352 .. attribute:: testMethodPrefix
1353
1354 String giving the prefix of method names which will be interpreted as test
1355 methods. The default value is ``'test'``.
1356
1357 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
1358 methods.
1359
1360
1361 .. attribute:: sortTestMethodsUsing
1362
1363 Function to be used to compare method names when sorting them in
1364 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
1365
1366
1367 .. attribute:: suiteClass
1368
1369 Callable object that constructs a test suite from a list of tests. No
1370 methods on the resulting object are needed. The default value is the
1371 :class:`TestSuite` class.
1372
1373 This affects all the :meth:`loadTestsFrom\*` methods.
1374
1375
Benjamin Peterson52baa292009-03-24 00:56:30 +00001376.. class:: TestResult
1377
1378 This class is used to compile information about which tests have succeeded
1379 and which have failed.
1380
1381 A :class:`TestResult` object stores the results of a set of tests. The
1382 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
1383 properly recorded; test authors do not need to worry about recording the
1384 outcome of tests.
1385
1386 Testing frameworks built on top of :mod:`unittest` may want access to the
1387 :class:`TestResult` object generated by running a set of tests for reporting
1388 purposes; a :class:`TestResult` instance is returned by the
1389 :meth:`TestRunner.run` method for this purpose.
1390
1391 :class:`TestResult` instances have the following attributes that will be of
1392 interest when inspecting the results of running a set of tests:
1393
1394
1395 .. attribute:: errors
1396
1397 A list containing 2-tuples of :class:`TestCase` instances and strings
1398 holding formatted tracebacks. Each tuple represents a test which raised an
1399 unexpected exception.
1400
Benjamin Peterson52baa292009-03-24 00:56:30 +00001401 .. attribute:: failures
1402
1403 A list containing 2-tuples of :class:`TestCase` instances and strings
1404 holding formatted tracebacks. Each tuple represents a test where a failure
1405 was explicitly signalled using the :meth:`TestCase.fail\*` or
1406 :meth:`TestCase.assert\*` methods.
1407
Benjamin Peterson52baa292009-03-24 00:56:30 +00001408 .. attribute:: skipped
1409
1410 A list containing 2-tuples of :class:`TestCase` instances and strings
1411 holding the reason for skipping the test.
1412
Benjamin Peterson70e32c82009-03-24 01:00:11 +00001413 .. versionadded:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +00001414
1415 .. attribute:: expectedFailures
1416
1417 A list contaning 2-tuples of :class:`TestCase` instances and strings
1418 holding formatted tracebacks. Each tuple represents a expected failures
1419 of the test case.
1420
1421 .. attribute:: unexpectedSuccesses
1422
1423 A list containing :class:`TestCase` instances that were marked as expected
1424 failures, but succeeded.
1425
1426 .. attribute:: shouldStop
1427
1428 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1429
1430
1431 .. attribute:: testsRun
1432
1433 The total number of tests run so far.
1434
1435
1436 .. method:: wasSuccessful()
1437
1438 Return :const:`True` if all tests run so far have passed, otherwise returns
1439 :const:`False`.
1440
1441
1442 .. method:: stop()
1443
1444 This method can be called to signal that the set of tests being run should
1445 be aborted by setting the :attr:`shouldStop` attribute to :const:`True`.
1446 :class:`TestRunner` objects should respect this flag and return without
1447 running any additional tests.
1448
1449 For example, this feature is used by the :class:`TextTestRunner` class to
1450 stop the test framework when the user signals an interrupt from the
1451 keyboard. Interactive tools which provide :class:`TestRunner`
1452 implementations can use this in a similar manner.
1453
1454 The following methods of the :class:`TestResult` class are used to maintain
1455 the internal data structures, and may be extended in subclasses to support
1456 additional reporting requirements. This is particularly useful in building
1457 tools which support interactive reporting while tests are being run.
1458
1459
1460 .. method:: startTest(test)
1461
1462 Called when the test case *test* is about to be run.
1463
1464 The default implementation simply increments the instance's :attr:`testsRun`
1465 counter.
1466
1467
1468 .. method:: stopTest(test)
1469
1470 Called after the test case *test* has been executed, regardless of the
1471 outcome.
1472
1473 The default implementation does nothing.
1474
1475
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001476 .. method:: startTestRun(test)
1477
1478 Called once before any tests are executed.
1479
Georg Brandl853947a2010-01-31 18:53:23 +00001480 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001481
1482
1483 .. method:: stopTestRun(test)
1484
Ezio Melotti176d6c42010-01-27 20:58:07 +00001485 Called once after all tests are executed.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001486
Georg Brandl853947a2010-01-31 18:53:23 +00001487 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001488
1489
Benjamin Peterson52baa292009-03-24 00:56:30 +00001490 .. method:: addError(test, err)
1491
1492 Called when the test case *test* raises an unexpected exception *err* is a
1493 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1494 traceback)``.
1495
1496 The default implementation appends a tuple ``(test, formatted_err)`` to
1497 the instance's :attr:`errors` attribute, where *formatted_err* is a
1498 formatted traceback derived from *err*.
1499
1500
1501 .. method:: addFailure(test, err)
1502
Benjamin Petersond2397752009-06-27 23:45:02 +00001503 Called when the test case *test* signals a failure. *err* is a tuple of
1504 the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001505
1506 The default implementation appends a tuple ``(test, formatted_err)`` to
1507 the instance's :attr:`failures` attribute, where *formatted_err* is a
1508 formatted traceback derived from *err*.
1509
1510
1511 .. method:: addSuccess(test)
1512
1513 Called when the test case *test* succeeds.
1514
1515 The default implementation does nothing.
1516
1517
1518 .. method:: addSkip(test, reason)
1519
1520 Called when the test case *test* is skipped. *reason* is the reason the
1521 test gave for skipping.
1522
1523 The default implementation appends a tuple ``(test, reason)`` to the
1524 instance's :attr:`skipped` attribute.
1525
1526
1527 .. method:: addExpectedFailure(test, err)
1528
1529 Called when the test case *test* fails, but was marked with the
1530 :func:`expectedFailure` decorator.
1531
1532 The default implementation appends a tuple ``(test, formatted_err)`` to
1533 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1534 is a formatted traceback derived from *err*.
1535
1536
1537 .. method:: addUnexpectedSuccess(test)
1538
1539 Called when the test case *test* was marked with the
1540 :func:`expectedFailure` decorator, but succeeded.
1541
1542 The default implementation appends the test to the instance's
1543 :attr:`unexpectedSuccesses` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +00001544
Michael Foord34c94622010-02-10 15:51:42 +00001545.. class:: TextTestResult(stream, descriptions, verbosity)
1546
1547 A concrete implementation of :class:`TestResult` used by the
1548 :class:`TextTestRunner`.
1549
1550 .. versionadded:: 3.2
1551 This class was previously named ``_TextTestResult``. The old name still
1552 exists as an alias but is deprecated.
Georg Brandl116aa622007-08-15 14:28:22 +00001553
1554.. data:: defaultTestLoader
1555
1556 Instance of the :class:`TestLoader` class intended to be shared. If no
1557 customization of the :class:`TestLoader` is needed, this instance can be used
1558 instead of repeatedly creating new instances.
1559
1560
Michael Foord34c94622010-02-10 15:51:42 +00001561.. class:: TextTestRunner(stream=sys.stderr, descriptions=True, verbosity=1, runnerclass=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001562
1563 A basic test runner implementation which prints results on standard error. It
1564 has a few configurable parameters, but is essentially very simple. Graphical
1565 applications which run test suites should provide alternate implementations.
1566
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001567 .. method:: _makeResult()
Georg Brandl116aa622007-08-15 14:28:22 +00001568
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001569 This method returns the instance of ``TestResult`` used by :meth:`run`.
1570 It is not intended to be called directly, but can be overridden in
1571 subclasses to provide a custom ``TestResult``.
1572
Michael Foord34c94622010-02-10 15:51:42 +00001573 ``_makeResult()`` instantiates the class or callable passed in the
1574 ``TextTestRunner`` constructor as the ``resultclass`` argument. It
1575 defaults to :class::`TextTestResult` if no ``resultclass`` is provided.
1576 The result class is instantiated with the following arguments::
1577
1578 stream, descriptions, verbosity
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001579
Georg Brandl7f01a132009-09-16 15:58:14 +00001580.. function:: main(module='__main__', defaultTest=None, argv=None, testRunner=None, testLoader=unittest.loader.defaultTestLoader, exit=True, verbosity=1)
Georg Brandl116aa622007-08-15 14:28:22 +00001581
1582 A command-line program that runs a set of tests; this is primarily for making
1583 test modules conveniently executable. The simplest use for this function is to
1584 include the following line at the end of a test script::
1585
1586 if __name__ == '__main__':
1587 unittest.main()
1588
Benjamin Petersond2397752009-06-27 23:45:02 +00001589 You can run tests with more detailed information by passing in the verbosity
1590 argument::
1591
1592 if __name__ == '__main__':
1593 unittest.main(verbosity=2)
1594
Georg Brandl116aa622007-08-15 14:28:22 +00001595 The *testRunner* argument can either be a test runner class or an already
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001596 created instance of it. By default ``main`` calls :func:`sys.exit` with
1597 an exit code indicating success or failure of the tests run.
1598
1599 ``main`` supports being used from the interactive interpreter by passing in the
1600 argument ``exit=False``. This displays the result on standard output without
1601 calling :func:`sys.exit`::
1602
1603 >>> from unittest import main
1604 >>> main(module='test_module', exit=False)
1605
1606 Calling ``main`` actually returns an instance of the ``TestProgram`` class.
1607 This stores the result of the tests run as the ``result`` attribute.
1608
Georg Brandl853947a2010-01-31 18:53:23 +00001609 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001610 The ``exit`` and ``verbosity`` parameters were added.
1611
1612
1613load_tests Protocol
1614###################
1615
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001616
Georg Brandl853947a2010-01-31 18:53:23 +00001617.. versionadded:: 3.2
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001618
1619
Benjamin Petersond2397752009-06-27 23:45:02 +00001620Modules or packages can customize how tests are loaded from them during normal
1621test runs or test discovery by implementing a function called ``load_tests``.
1622
1623If a test module defines ``load_tests`` it will be called by
1624:meth:`TestLoader.loadTestsFromModule` with the following arguments::
1625
1626 load_tests(loader, standard_tests, None)
1627
1628It should return a :class:`TestSuite`.
1629
1630*loader* is the instance of :class:`TestLoader` doing the loading.
1631*standard_tests* are the tests that would be loaded by default from the
1632module. It is common for test modules to only want to add or remove tests
1633from the standard set of tests.
1634The third argument is used when loading packages as part of test discovery.
1635
1636A typical ``load_tests`` function that loads tests from a specific set of
1637:class:`TestCase` classes may look like::
1638
1639 test_cases = (TestCase1, TestCase2, TestCase3)
1640
1641 def load_tests(loader, tests, pattern):
1642 suite = TestSuite()
1643 for test_class in test_cases:
1644 tests = loader.loadTestsFromTestCase(test_class)
1645 suite.addTests(tests)
1646 return suite
1647
1648If discovery is started, either from the command line or by calling
1649:meth:`TestLoader.discover`, with a pattern that matches a package
1650name then the package :file:`__init__.py` will be checked for ``load_tests``.
1651
1652.. note::
1653
Ezio Melotti0639d5a2009-12-19 23:26:38 +00001654 The default pattern is 'test*.py'. This matches all Python files
Benjamin Petersond2397752009-06-27 23:45:02 +00001655 that start with 'test' but *won't* match any test directories.
1656
1657 A pattern like 'test*' will match test packages as well as
1658 modules.
1659
1660If the package :file:`__init__.py` defines ``load_tests`` then it will be
1661called and discovery not continued into the package. ``load_tests``
1662is called with the following arguments::
1663
1664 load_tests(loader, standard_tests, pattern)
1665
1666This should return a :class:`TestSuite` representing all the tests
1667from the package. (``standard_tests`` will only contain tests
1668collected from :file:`__init__.py`.)
1669
1670Because the pattern is passed into ``load_tests`` the package is free to
1671continue (and potentially modify) test discovery. A 'do nothing'
1672``load_tests`` function for a test package would look like::
1673
1674 def load_tests(loader, standard_tests, pattern):
1675 # top level directory cached on loader instance
1676 this_dir = os.path.dirname(__file__)
1677 package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
1678 standard_tests.addTests(package_tests)
1679 return standard_tests