blob: b5b23c784f96600aa4b04baaab37aedc607ea5ab [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
Benjamin Petersonb48af542010-04-11 20:43:16 +000077 `unittest2: A backport of new unittest features for Python 2.4-2.6 <http://pypi.python.org/pypi/unittest2>`_
78 Many new features were added to unittest in Python 2.7, including test
79 discovery. unittest2 allows you to use these features with earlier
80 versions of Python.
81
Georg Brandl116aa622007-08-15 14:28:22 +000082 `Simple Smalltalk Testing: With Patterns <http://www.XProgramming.com/testfram.htm>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000083 Kent Beck's original paper on testing frameworks using the pattern shared
84 by :mod:`unittest`.
Georg Brandl116aa622007-08-15 14:28:22 +000085
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000086 `Nose <http://code.google.com/p/python-nose/>`_ and `py.test <http://pytest.org>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000087 Third-party unittest frameworks with a lighter-weight syntax for writing
88 tests. For example, ``assert func(10) == 42``.
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000089
Benjamin Petersonb48af542010-04-11 20:43:16 +000090 `The Python Testing Tools Taxonomy <http://pycheesecake.org/wiki/PythonTestingToolsTaxonomy>`_
91 An extensive list of Python testing tools including functional testing
92 frameworks and mock object libraries.
Benjamin Petersond2397752009-06-27 23:45:02 +000093
Benjamin Petersonb48af542010-04-11 20:43:16 +000094 `Testing in Python Mailing List <http://lists.idyll.org/listinfo/testing-in-python>`_
95 A special-interest-group for discussion of testing, and testing tools,
96 in Python.
Benjamin Petersond2397752009-06-27 23:45:02 +000097
Georg Brandl116aa622007-08-15 14:28:22 +000098.. _unittest-minimal-example:
99
100Basic example
101-------------
102
103The :mod:`unittest` module provides a rich set of tools for constructing and
104running tests. This section demonstrates that a small subset of the tools
105suffice to meet the needs of most users.
106
107Here is a short script to test three functions from the :mod:`random` module::
108
109 import random
110 import unittest
111
112 class TestSequenceFunctions(unittest.TestCase):
113
114 def setUp(self):
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000115 self.seq = list(range(10))
Georg Brandl116aa622007-08-15 14:28:22 +0000116
Benjamin Peterson52baa292009-03-24 00:56:30 +0000117 def test_shuffle(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000118 # make sure the shuffled sequence does not lose any elements
119 random.shuffle(self.seq)
120 self.seq.sort()
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000121 self.assertEqual(self.seq, list(range(10)))
Georg Brandl116aa622007-08-15 14:28:22 +0000122
Benjamin Peterson847a4112010-03-14 15:04:17 +0000123 # should raise an exception for an immutable sequence
124 self.assertRaises(TypeError, random.shuffle, (1,2,3))
125
Benjamin Peterson52baa292009-03-24 00:56:30 +0000126 def test_choice(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000127 element = random.choice(self.seq)
Benjamin Peterson847a4112010-03-14 15:04:17 +0000128 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000129
Benjamin Peterson52baa292009-03-24 00:56:30 +0000130 def test_sample(self):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000131 with self.assertRaises(ValueError):
132 random.sample(self.seq, 20)
Georg Brandl116aa622007-08-15 14:28:22 +0000133 for element in random.sample(self.seq, 5):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000134 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000135
136 if __name__ == '__main__':
137 unittest.main()
138
Benjamin Peterson52baa292009-03-24 00:56:30 +0000139A testcase is created by subclassing :class:`unittest.TestCase`. The three
Georg Brandl116aa622007-08-15 14:28:22 +0000140individual tests are defined with methods whose names start with the letters
141``test``. This naming convention informs the test runner about which methods
142represent tests.
143
Benjamin Peterson52baa292009-03-24 00:56:30 +0000144The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
Michael Foord34c94622010-02-10 15:51:42 +0000145expected result; :meth:`~TestCase.assertTrue` to verify a condition; or
Benjamin Peterson52baa292009-03-24 00:56:30 +0000146:meth:`~TestCase.assertRaises` to verify that an expected exception gets raised.
147These methods are used instead of the :keyword:`assert` statement so the test
148runner can accumulate all test results and produce a report.
Georg Brandl116aa622007-08-15 14:28:22 +0000149
Benjamin Peterson52baa292009-03-24 00:56:30 +0000150When a :meth:`~TestCase.setUp` method is defined, the test runner will run that
151method prior to each test. Likewise, if a :meth:`~TestCase.tearDown` method is
152defined, the test runner will invoke that method after each test. In the
153example, :meth:`~TestCase.setUp` was used to create a fresh sequence for each
154test.
Georg Brandl116aa622007-08-15 14:28:22 +0000155
156The final block shows a simple way to run the tests. :func:`unittest.main`
157provides a command line interface to the test script. When run from the command
158line, the above script produces an output that looks like this::
159
160 ...
161 ----------------------------------------------------------------------
162 Ran 3 tests in 0.000s
163
164 OK
165
166Instead of :func:`unittest.main`, there are other ways to run the tests with a
167finer level of control, less terse output, and no requirement to be run from the
168command line. For example, the last two lines may be replaced with::
169
170 suite = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)
171 unittest.TextTestRunner(verbosity=2).run(suite)
172
173Running the revised script from the interpreter or another script produces the
174following output::
175
Ezio Melottid59e44a2010-02-28 03:46:13 +0000176 test_choice (__main__.TestSequenceFunctions) ... ok
177 test_sample (__main__.TestSequenceFunctions) ... ok
178 test_shuffle (__main__.TestSequenceFunctions) ... ok
Georg Brandl116aa622007-08-15 14:28:22 +0000179
180 ----------------------------------------------------------------------
181 Ran 3 tests in 0.110s
182
183 OK
184
185The above examples show the most commonly used :mod:`unittest` features which
186are sufficient to meet many everyday testing needs. The remainder of the
187documentation explores the full feature set from first principles.
188
Benjamin Petersonb48af542010-04-11 20:43:16 +0000189
190.. _unittest-command-line-interface:
191
192Command Line Interface
193----------------------
194
195The unittest module can be used from the command line to run tests from
196modules, classes or even individual test methods::
197
198 python -m unittest test_module1 test_module2
199 python -m unittest test_module.TestClass
200 python -m unittest test_module.TestClass.test_method
201
202You can pass in a list with any combination of module names, and fully
203qualified class or method names.
204
205You can run tests with more detail (higher verbosity) by passing in the -v flag::
206
207 python -m unittest -v test_module
208
209For a list of all the command line options::
210
211 python -m unittest -h
212
213.. versionchanged:: 3.2
214 In earlier versions it was only possible to run individual test methods and
215 not modules or classes.
216
217
218failfast, catch and buffer command line options
219~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
220
221unittest supports three command options.
222
223* -f / --failfast
224
225 Stop the test run on the first error or failure.
226
227* -c / --catch
228
229 Control-c during the test run waits for the current test to end and then
230 reports all the results so far. A second control-c raises the normal
231 ``KeyboardInterrupt`` exception.
232
233 See `Signal Handling`_ for the functions that provide this functionality.
234
235* -b / --buffer
236
237 The standard out and standard error streams are buffered during the test
238 run. Output during a passing test is discarded. Output is echoed normally
239 on test fail or error and is added to the failure messages.
240
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000241.. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000242 The command line options ``-c``, ``-b`` and ``-f`` where added.
243
244The command line can also be used for test discovery, for running all of the
245tests in a project or just a subset.
246
247
248.. _unittest-test-discovery:
249
250Test Discovery
251--------------
252
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000253.. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000254
255Unittest supports simple test discovery. For a project's tests to be
256compatible with test discovery they must all be importable from the top level
257directory of the project (in other words, they must all be in Python packages).
258
259Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
260used from the command line. The basic command line usage is::
261
262 cd project_directory
263 python -m unittest discover
264
265The ``discover`` sub-command has the following options:
266
267 -v, --verbose Verbose output
268 -s directory Directory to start discovery ('.' default)
269 -p pattern Pattern to match test files ('test*.py' default)
270 -t directory Top level directory of project (default to
271 start directory)
272
273The -s, -p, & -t options can be passsed in as positional arguments. The
274following two command lines are equivalent::
275
276 python -m unittest discover -s project_directory -p '*_test.py'
277 python -m unittest discover project_directory '*_test.py'
278
Michael Foord16f3e902010-05-08 15:13:42 +0000279As well as being a path it is possible to pass a package name, for example
280``myproject.subpackage.test``, as the start directory. The package name you
281supply will then be imported and its location on the filesystem will be used
282as the start directory.
283
284.. caution::
285
286 Test discovery loads tests by importing them. Once test discovery has
287 found all the test files from the start directory you specify it turns the
288 paths into package names to import. For example `foo/bar/baz.py` will be
289 imported as ``foo.bar.baz``.
290
291 If you have a package installed globally and attempt test discovery on
292 a different copy of the package then the import *could* happen from the
293 wrong place. If this happens test discovery will warn you and exit.
294
295 If you supply the start directory as a package name rather than a
296 path to a directory then discover assumes that whichever location it
297 imports from is the location you intended, so you will not get the
298 warning.
299
Benjamin Petersonb48af542010-04-11 20:43:16 +0000300Test modules and packages can customize test loading and discovery by through
301the `load_tests protocol`_.
302
303
Georg Brandl116aa622007-08-15 14:28:22 +0000304.. _organizing-tests:
305
306Organizing test code
307--------------------
308
309The basic building blocks of unit testing are :dfn:`test cases` --- single
310scenarios that must be set up and checked for correctness. In :mod:`unittest`,
311test cases are represented by instances of :mod:`unittest`'s :class:`TestCase`
312class. To make your own test cases you must write subclasses of
313:class:`TestCase`, or use :class:`FunctionTestCase`.
314
315An instance of a :class:`TestCase`\ -derived class is an object that can
316completely run a single test method, together with optional set-up and tidy-up
317code.
318
319The testing code of a :class:`TestCase` instance should be entirely self
320contained, such that it can be run either in isolation or in arbitrary
321combination with any number of other test cases.
322
Benjamin Peterson52baa292009-03-24 00:56:30 +0000323The simplest :class:`TestCase` subclass will simply override the
324:meth:`~TestCase.runTest` method in order to perform specific testing code::
Georg Brandl116aa622007-08-15 14:28:22 +0000325
326 import unittest
327
328 class DefaultWidgetSizeTestCase(unittest.TestCase):
329 def runTest(self):
330 widget = Widget('The widget')
331 self.assertEqual(widget.size(), (50, 50), 'incorrect default size')
332
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000333Note that in order to test something, we use the one of the :meth:`assert\*`
Benjamin Petersond2397752009-06-27 23:45:02 +0000334methods provided by the :class:`TestCase` base class. If the test fails, an
335exception will be raised, and :mod:`unittest` will identify the test case as a
336:dfn:`failure`. Any other exceptions will be treated as :dfn:`errors`. This
337helps you identify where the problem is: :dfn:`failures` are caused by incorrect
338results - a 5 where you expected a 6. :dfn:`Errors` are caused by incorrect
339code - e.g., a :exc:`TypeError` caused by an incorrect function call.
Georg Brandl116aa622007-08-15 14:28:22 +0000340
341The way to run a test case will be described later. For now, note that to
342construct an instance of such a test case, we call its constructor without
343arguments::
344
345 testCase = DefaultWidgetSizeTestCase()
346
347Now, such test cases can be numerous, and their set-up can be repetitive. In
348the above case, constructing a :class:`Widget` in each of 100 Widget test case
349subclasses would mean unsightly duplication.
350
351Luckily, we can factor out such set-up code by implementing a method called
Benjamin Peterson52baa292009-03-24 00:56:30 +0000352:meth:`~TestCase.setUp`, which the testing framework will automatically call for
353us when we run the test::
Georg Brandl116aa622007-08-15 14:28:22 +0000354
355 import unittest
356
357 class SimpleWidgetTestCase(unittest.TestCase):
358 def setUp(self):
359 self.widget = Widget('The widget')
360
361 class DefaultWidgetSizeTestCase(SimpleWidgetTestCase):
362 def runTest(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000363 self.assertEqual(self.widget.size(), (50,50),
364 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000365
366 class WidgetResizeTestCase(SimpleWidgetTestCase):
367 def runTest(self):
368 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000369 self.assertEqual(self.widget.size(), (100,150),
370 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000371
Benjamin Peterson52baa292009-03-24 00:56:30 +0000372If the :meth:`~TestCase.setUp` method raises an exception while the test is
373running, the framework will consider the test to have suffered an error, and the
374:meth:`~TestCase.runTest` method will not be executed.
Georg Brandl116aa622007-08-15 14:28:22 +0000375
Benjamin Peterson52baa292009-03-24 00:56:30 +0000376Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
377after the :meth:`~TestCase.runTest` method has been run::
Georg Brandl116aa622007-08-15 14:28:22 +0000378
379 import unittest
380
381 class SimpleWidgetTestCase(unittest.TestCase):
382 def setUp(self):
383 self.widget = Widget('The widget')
384
385 def tearDown(self):
386 self.widget.dispose()
387 self.widget = None
388
Benjamin Peterson52baa292009-03-24 00:56:30 +0000389If :meth:`~TestCase.setUp` succeeded, the :meth:`~TestCase.tearDown` method will
390be run whether :meth:`~TestCase.runTest` succeeded or not.
Georg Brandl116aa622007-08-15 14:28:22 +0000391
392Such a working environment for the testing code is called a :dfn:`fixture`.
393
394Often, many small test cases will use the same fixture. In this case, we would
395end up subclassing :class:`SimpleWidgetTestCase` into many small one-method
396classes such as :class:`DefaultWidgetSizeTestCase`. This is time-consuming and
Georg Brandl116aa622007-08-15 14:28:22 +0000397discouraging, so in the same vein as JUnit, :mod:`unittest` provides a simpler
398mechanism::
399
400 import unittest
401
402 class WidgetTestCase(unittest.TestCase):
403 def setUp(self):
404 self.widget = Widget('The widget')
405
406 def tearDown(self):
407 self.widget.dispose()
408 self.widget = None
409
Ezio Melottid59e44a2010-02-28 03:46:13 +0000410 def test_default_size(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000411 self.assertEqual(self.widget.size(), (50,50),
412 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000413
Ezio Melottid59e44a2010-02-28 03:46:13 +0000414 def test_resize(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000415 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000416 self.assertEqual(self.widget.size(), (100,150),
417 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000418
Benjamin Peterson52baa292009-03-24 00:56:30 +0000419Here we have not provided a :meth:`~TestCase.runTest` method, but have instead
420provided two different test methods. Class instances will now each run one of
Ezio Melottid59e44a2010-02-28 03:46:13 +0000421the :meth:`test_\*` methods, with ``self.widget`` created and destroyed
Benjamin Peterson52baa292009-03-24 00:56:30 +0000422separately for each instance. When creating an instance we must specify the
423test method it is to run. We do this by passing the method name in the
424constructor::
Georg Brandl116aa622007-08-15 14:28:22 +0000425
Ezio Melottid59e44a2010-02-28 03:46:13 +0000426 defaultSizeTestCase = WidgetTestCase('test_default_size')
427 resizeTestCase = WidgetTestCase('test_resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000428
429Test case instances are grouped together according to the features they test.
430:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
431represented by :mod:`unittest`'s :class:`TestSuite` class::
432
433 widgetTestSuite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000434 widgetTestSuite.addTest(WidgetTestCase('test_default_size'))
435 widgetTestSuite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000436
437For the ease of running tests, as we will see later, it is a good idea to
438provide in each test module a callable object that returns a pre-built test
439suite::
440
441 def suite():
442 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000443 suite.addTest(WidgetTestCase('test_default_size'))
444 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000445 return suite
446
447or even::
448
449 def suite():
Ezio Melottid59e44a2010-02-28 03:46:13 +0000450 tests = ['test_default_size', 'test_resize']
Georg Brandl116aa622007-08-15 14:28:22 +0000451
452 return unittest.TestSuite(map(WidgetTestCase, tests))
453
454Since it is a common pattern to create a :class:`TestCase` subclass with many
455similarly named test functions, :mod:`unittest` provides a :class:`TestLoader`
456class that can be used to automate the process of creating a test suite and
457populating it with individual tests. For example, ::
458
459 suite = unittest.TestLoader().loadTestsFromTestCase(WidgetTestCase)
460
Ezio Melottid59e44a2010-02-28 03:46:13 +0000461will create a test suite that will run ``WidgetTestCase.test_default_size()`` and
462``WidgetTestCase.test_resize``. :class:`TestLoader` uses the ``'test'`` method
Georg Brandl116aa622007-08-15 14:28:22 +0000463name prefix to identify test methods automatically.
464
Mark Dickinsonc48d8342009-02-01 14:18:10 +0000465Note that the order in which the various test cases will be run is
466determined by sorting the test function names with respect to the
467built-in ordering for strings.
Georg Brandl116aa622007-08-15 14:28:22 +0000468
469Often it is desirable to group suites of test cases together, so as to run tests
470for the whole system at once. This is easy, since :class:`TestSuite` instances
471can be added to a :class:`TestSuite` just as :class:`TestCase` instances can be
472added to a :class:`TestSuite`::
473
474 suite1 = module1.TheTestSuite()
475 suite2 = module2.TheTestSuite()
476 alltests = unittest.TestSuite([suite1, suite2])
477
478You can place the definitions of test cases and test suites in the same modules
479as the code they are to test (such as :file:`widget.py`), but there are several
480advantages to placing the test code in a separate module, such as
481:file:`test_widget.py`:
482
483* The test module can be run standalone from the command line.
484
485* The test code can more easily be separated from shipped code.
486
487* There is less temptation to change test code to fit the code it tests without
488 a good reason.
489
490* Test code should be modified much less frequently than the code it tests.
491
492* Tested code can be refactored more easily.
493
494* Tests for modules written in C must be in separate modules anyway, so why not
495 be consistent?
496
497* If the testing strategy changes, there is no need to change the source code.
498
499
500.. _legacy-unit-tests:
501
502Re-using old test code
503----------------------
504
505Some users will find that they have existing test code that they would like to
506run from :mod:`unittest`, without converting every old test function to a
507:class:`TestCase` subclass.
508
509For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
510This subclass of :class:`TestCase` can be used to wrap an existing test
511function. Set-up and tear-down functions can also be provided.
512
513Given the following test function::
514
515 def testSomething():
516 something = makeSomething()
517 assert something.name is not None
518 # ...
519
520one can create an equivalent test case instance as follows::
521
522 testcase = unittest.FunctionTestCase(testSomething)
523
524If there are additional set-up and tear-down methods that should be called as
525part of the test case's operation, they can also be provided like so::
526
527 testcase = unittest.FunctionTestCase(testSomething,
528 setUp=makeSomethingDB,
529 tearDown=deleteSomethingDB)
530
531To make migrating existing test suites easier, :mod:`unittest` supports tests
532raising :exc:`AssertionError` to indicate test failure. However, it is
533recommended that you use the explicit :meth:`TestCase.fail\*` and
534:meth:`TestCase.assert\*` methods instead, as future versions of :mod:`unittest`
535may treat :exc:`AssertionError` differently.
536
537.. note::
538
Benjamin Petersond2397752009-06-27 23:45:02 +0000539 Even though :class:`FunctionTestCase` can be used to quickly convert an
540 existing test base over to a :mod:`unittest`\ -based system, this approach is
541 not recommended. Taking the time to set up proper :class:`TestCase`
542 subclasses will make future test refactorings infinitely easier.
Georg Brandl116aa622007-08-15 14:28:22 +0000543
Benjamin Peterson52baa292009-03-24 00:56:30 +0000544In some cases, the existing tests may have been written using the :mod:`doctest`
545module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
546automatically build :class:`unittest.TestSuite` instances from the existing
547:mod:`doctest`\ -based tests.
548
Georg Brandl116aa622007-08-15 14:28:22 +0000549
Benjamin Peterson5254c042009-03-23 22:25:03 +0000550.. _unittest-skipping:
551
552Skipping tests and expected failures
553------------------------------------
554
Michael Foordf5c851a2010-02-05 21:48:03 +0000555.. versionadded:: 3.1
556
Benjamin Peterson5254c042009-03-23 22:25:03 +0000557Unittest supports skipping individual test methods and even whole classes of
558tests. In addition, it supports marking a test as a "expected failure," a test
559that is broken and will fail, but shouldn't be counted as a failure on a
560:class:`TestResult`.
561
562Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
563or one of its conditional variants.
564
565Basic skipping looks like this: ::
566
567 class MyTestCase(unittest.TestCase):
568
569 @unittest.skip("demonstrating skipping")
570 def test_nothing(self):
571 self.fail("shouldn't happen")
572
Benjamin Petersond2397752009-06-27 23:45:02 +0000573 @unittest.skipIf(mylib.__version__ < (1, 3),
574 "not supported in this library version")
Benjamin Petersonded31c42009-03-30 15:04:16 +0000575 def test_format(self):
576 # Tests that work for only a certain version of the library.
577 pass
578
579 @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
580 def test_windows_support(self):
581 # windows specific testing code
582 pass
583
Benjamin Peterson5254c042009-03-23 22:25:03 +0000584This is the output of running the example above in verbose mode: ::
585
Benjamin Petersonded31c42009-03-30 15:04:16 +0000586 test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000587 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
Benjamin Petersonded31c42009-03-30 15:04:16 +0000588 test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000589
590 ----------------------------------------------------------------------
Benjamin Petersonded31c42009-03-30 15:04:16 +0000591 Ran 3 tests in 0.005s
592
593 OK (skipped=3)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000594
595Classes can be skipped just like methods: ::
596
597 @skip("showing class skipping")
598 class MySkippedTestCase(unittest.TestCase):
599 def test_not_run(self):
600 pass
601
Benjamin Peterson52baa292009-03-24 00:56:30 +0000602:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
603that needs to be set up is not available.
604
Benjamin Peterson5254c042009-03-23 22:25:03 +0000605Expected failures use the :func:`expectedFailure` decorator. ::
606
607 class ExpectedFailureTestCase(unittest.TestCase):
608 @unittest.expectedFailure
609 def test_fail(self):
610 self.assertEqual(1, 0, "broken")
611
612It's easy to roll your own skipping decorators by making a decorator that calls
613:func:`skip` on the test when it wants it to be skipped. This decorator skips
614the test unless the passed object has a certain attribute: ::
615
616 def skipUnlessHasattr(obj, attr):
617 if hasattr(obj, attr):
618 return lambda func: func
619 return unittest.skip("{0!r} doesn't have {1!r}".format(obj, attr))
620
621The following decorators implement test skipping and expected failures:
622
623.. function:: skip(reason)
624
625 Unconditionally skip the decorated test. *reason* should describe why the
626 test is being skipped.
627
628.. function:: skipIf(condition, reason)
629
630 Skip the decorated test if *condition* is true.
631
632.. function:: skipUnless(condition, reason)
633
634 Skip the decoratored test unless *condition* is true.
635
636.. function:: expectedFailure
637
638 Mark the test as an expected failure. If the test fails when run, the test
639 is not counted as a failure.
640
Benjamin Petersonb48af542010-04-11 20:43:16 +0000641Skipped tests will not have :meth:`setUp` or :meth:`tearDown` run around them.
642Skipped classes will not have :meth:`setUpClass` or :meth:`tearDownClass` run.
643
Benjamin Peterson5254c042009-03-23 22:25:03 +0000644
Georg Brandl116aa622007-08-15 14:28:22 +0000645.. _unittest-contents:
646
647Classes and functions
648---------------------
649
Benjamin Peterson52baa292009-03-24 00:56:30 +0000650This section describes in depth the API of :mod:`unittest`.
651
652
653.. _testcase-objects:
654
655Test cases
656~~~~~~~~~~
Georg Brandl116aa622007-08-15 14:28:22 +0000657
Georg Brandl7f01a132009-09-16 15:58:14 +0000658.. class:: TestCase(methodName='runTest')
Georg Brandl116aa622007-08-15 14:28:22 +0000659
660 Instances of the :class:`TestCase` class represent the smallest testable units
661 in the :mod:`unittest` universe. This class is intended to be used as a base
662 class, with specific tests being implemented by concrete subclasses. This class
663 implements the interface needed by the test runner to allow it to drive the
664 test, and methods that the test code can use to check for and report various
665 kinds of failure.
666
667 Each instance of :class:`TestCase` will run a single test method: the method
668 named *methodName*. If you remember, we had an earlier example that went
669 something like this::
670
671 def suite():
672 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000673 suite.addTest(WidgetTestCase('test_default_size'))
674 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000675 return suite
676
677 Here, we create two instances of :class:`WidgetTestCase`, each of which runs a
678 single test.
679
Benjamin Peterson52baa292009-03-24 00:56:30 +0000680 *methodName* defaults to :meth:`runTest`.
681
682 :class:`TestCase` instances provide three groups of methods: one group used
683 to run the test, another used by the test implementation to check conditions
684 and report failures, and some inquiry methods allowing information about the
685 test itself to be gathered.
686
687 Methods in the first group (running the test) are:
688
689
690 .. method:: setUp()
691
692 Method called to prepare the test fixture. This is called immediately
693 before calling the test method; any exception raised by this method will
694 be considered an error rather than a test failure. The default
695 implementation does nothing.
696
697
698 .. method:: tearDown()
699
700 Method called immediately after the test method has been called and the
701 result recorded. This is called even if the test method raised an
702 exception, so the implementation in subclasses may need to be particularly
703 careful about checking internal state. Any exception raised by this
704 method will be considered an error rather than a test failure. This
705 method will only be called if the :meth:`setUp` succeeds, regardless of
706 the outcome of the test method. The default implementation does nothing.
707
708
Benjamin Petersonb48af542010-04-11 20:43:16 +0000709 .. method:: setUpClass()
710
711 A class method called before tests in an individual class run.
712 ``setUpClass`` is called with the class as the only argument
713 and must be decorated as a :func:`classmethod`::
714
715 @classmethod
716 def setUpClass(cls):
717 ...
718
719 See `Class and Module Fixtures`_ for more details.
720
721 .. versionadded:: 3.2
722
723
724 .. method:: tearDownClass()
725
726 A class method called after tests in an individual class have run.
727 ``tearDownClass`` is called with the class as the only argument
728 and must be decorated as a :meth:`classmethod`::
729
730 @classmethod
731 def tearDownClass(cls):
732 ...
733
734 See `Class and Module Fixtures`_ for more details.
735
736 .. versionadded:: 3.2
737
738
Georg Brandl7f01a132009-09-16 15:58:14 +0000739 .. method:: run(result=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000740
741 Run the test, collecting the result into the test result object passed as
742 *result*. If *result* is omitted or :const:`None`, a temporary result
Alexandre Vassalotti260484d2009-07-17 11:43:26 +0000743 object is created (by calling the :meth:`defaultTestResult` method) and
744 used. The result object is not returned to :meth:`run`'s caller.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000745
746 The same effect may be had by simply calling the :class:`TestCase`
747 instance.
748
749
Benjamin Petersone549ead2009-03-28 21:42:05 +0000750 .. method:: skipTest(reason)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000751
752 Calling this during the a test method or :meth:`setUp` skips the current
753 test. See :ref:`unittest-skipping` for more information.
754
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000755 .. versionadded:: 3.1
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000756
Benjamin Peterson52baa292009-03-24 00:56:30 +0000757
758 .. method:: debug()
759
760 Run the test without collecting the result. This allows exceptions raised
761 by the test to be propagated to the caller, and can be used to support
762 running tests under a debugger.
763
764 The test code can use any of the following methods to check for and report
765 failures.
766
767
Georg Brandl7f01a132009-09-16 15:58:14 +0000768 .. method:: assertTrue(expr, msg=None)
769 assert_(expr, msg=None)
770 failUnless(expr, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000771
Georg Brandlff2ad0e2009-04-27 16:51:45 +0000772 Signal a test failure if *expr* is false; the explanation for the failure
Benjamin Peterson52baa292009-03-24 00:56:30 +0000773 will be *msg* if given, otherwise it will be :const:`None`.
774
Raymond Hettinger35a88362009-04-09 00:08:24 +0000775 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +0000776 :meth:`failUnless`; use one of the ``assert`` variants.
Michael Foord34c94622010-02-10 15:51:42 +0000777 :meth:`assert_`; use :meth:`assertTrue`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000778
Benjamin Peterson52baa292009-03-24 00:56:30 +0000779
Georg Brandl7f01a132009-09-16 15:58:14 +0000780 .. method:: assertEqual(first, second, msg=None)
781 failUnlessEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000782
783 Test that *first* and *second* are equal. If the values do not compare
784 equal, the test will fail with the explanation given by *msg*, or
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000785 :const:`None`. Note that using :meth:`assertEqual` improves upon
786 doing the comparison as the first parameter to :meth:`assertTrue`: the
787 default value for *msg* include representations of both *first* and
788 *second*.
789
790 In addition, if *first* and *second* are the exact same type and one of
Michael Foord02834952010-02-08 23:10:39 +0000791 list, tuple, dict, set, frozenset or str or any type that a subclass
792 registers with :meth:`addTypeEqualityFunc` the type specific equality
793 function will be called in order to generate a more useful default
794 error message.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000795
Raymond Hettinger35a88362009-04-09 00:08:24 +0000796 .. versionchanged:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000797 Added the automatic calling of type specific equality function.
798
Michael Foord28a817e2010-02-09 00:03:57 +0000799 .. versionchanged:: 3.2
800 :meth:`assertMultiLineEqual` added as the default type equality
801 function for comparing strings.
Michael Foord02834952010-02-08 23:10:39 +0000802
Raymond Hettinger35a88362009-04-09 00:08:24 +0000803 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +0000804 :meth:`failUnlessEqual`; use :meth:`assertEqual`.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000805
806
Georg Brandl7f01a132009-09-16 15:58:14 +0000807 .. method:: assertNotEqual(first, second, msg=None)
808 failIfEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000809
810 Test that *first* and *second* are not equal. If the values do compare
811 equal, the test will fail with the explanation given by *msg*, or
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000812 :const:`None`. Note that using :meth:`assertNotEqual` improves upon doing
813 the comparison as the first parameter to :meth:`assertTrue` is that the
Benjamin Peterson52baa292009-03-24 00:56:30 +0000814 default value for *msg* can be computed to include representations of both
815 *first* and *second*.
816
Raymond Hettinger35a88362009-04-09 00:08:24 +0000817 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +0000818 :meth:`failIfEqual`; use :meth:`assertNotEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000819
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000820
Benjamin Petersonb48af542010-04-11 20:43:16 +0000821 .. method:: assertAlmostEqual(first, second, *, places=7, msg=None, delta=None)
822 failUnlessAlmostEqual(first, second, *, places=7, msg=None, delta=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000823
824 Test that *first* and *second* are approximately equal by computing the
825 difference, rounding to the given number of decimal *places* (default 7),
826 and comparing to zero.
827
828 Note that comparing a given number of decimal places is not the same as
829 comparing a given number of significant digits. If the values do not
830 compare equal, the test will fail with the explanation given by *msg*, or
831 :const:`None`.
832
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000833 .. versionchanged:: 3.2
834 Objects that compare equal are automatically almost equal.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000835 Added the ``delta`` keyword argument.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000836
Raymond Hettinger35a88362009-04-09 00:08:24 +0000837 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +0000838 :meth:`failUnlessAlmostEqual`; use :meth:`assertAlmostEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000839
Benjamin Peterson52baa292009-03-24 00:56:30 +0000840
Benjamin Petersonb48af542010-04-11 20:43:16 +0000841 .. method:: assertNotAlmostEqual(first, second, *, places=7, msg=None, delta=None)
842 failIfAlmostEqual(first, second, *, places=7, msg=None, delta=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000843
844 Test that *first* and *second* are not approximately equal by computing
845 the difference, rounding to the given number of decimal *places* (default
846 7), and comparing to zero.
847
848 Note that comparing a given number of decimal places is not the same as
849 comparing a given number of significant digits. If the values do not
850 compare equal, the test will fail with the explanation given by *msg*, or
851 :const:`None`.
852
Benjamin Petersonb48af542010-04-11 20:43:16 +0000853 If *delta* is supplied instead of *places* then the the difference
854 between *first* and *second* must be more than *delta*.
855
856 Supplying both *delta* and *places* raises a ``TypeError``.
857
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000858 .. versionchanged:: 3.2
859 Objects that compare equal automatically fail.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000860 Added the ``delta`` keyword argument.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000861
Raymond Hettinger35a88362009-04-09 00:08:24 +0000862 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +0000863 :meth:`failIfAlmostEqual`; use :meth:`assertNotAlmostEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000864
865
866 .. method:: assertGreater(first, second, msg=None)
867 assertGreaterEqual(first, second, msg=None)
868 assertLess(first, second, msg=None)
869 assertLessEqual(first, second, msg=None)
870
871 Test that *first* is respectively >, >=, < or <= than *second* depending
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000872 on the method name. If not, the test will fail with an explanation
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000873 or with the explanation given by *msg*::
874
875 >>> self.assertGreaterEqual(3, 4)
876 AssertionError: "3" unexpectedly not greater than or equal to "4"
877
Raymond Hettinger35a88362009-04-09 00:08:24 +0000878 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000879
880
881 .. method:: assertMultiLineEqual(self, first, second, msg=None)
882
883 Test that the multiline string *first* is equal to the string *second*.
884 When not equal a diff of the two strings highlighting the differences
Michael Foord02834952010-02-08 23:10:39 +0000885 will be included in the error message. This method is used by default
886 when comparing strings with :meth:`assertEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000887
Michael Foordabd91d52010-03-20 18:09:14 +0000888 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000889
Raymond Hettinger35a88362009-04-09 00:08:24 +0000890 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000891
892
Ezio Melotti732b6822010-01-16 19:40:06 +0000893 .. method:: assertRegexpMatches(text, regexp, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000894
895 Verifies that a *regexp* search matches *text*. Fails with an error
896 message including the pattern and the *text*. *regexp* may be
897 a regular expression object or a string containing a regular expression
898 suitable for use by :func:`re.search`.
899
Raymond Hettinger35a88362009-04-09 00:08:24 +0000900 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000901
902
Benjamin Petersonb48af542010-04-11 20:43:16 +0000903 .. method:: assertNotRegexpMatches(text, regexp, msg=None)
904
905 Verifies that a *regexp* search does not match *text*. Fails with an error
906 message including the pattern and the *text*. *regexp* may be
907 a regular expression object or a string containing a regular expression
908 suitable for use by :func:`re.search`.
909
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000910 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000911
912
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000913 .. method:: assertIn(first, second, msg=None)
914 assertNotIn(first, second, msg=None)
915
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000916 Tests that *first* is or is not in *second* with an explanatory error
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000917 message as appropriate.
918
Michael Foordabd91d52010-03-20 18:09:14 +0000919 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000920
Raymond Hettinger35a88362009-04-09 00:08:24 +0000921 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000922
923
Michael Foorde9abbee2010-02-05 20:54:27 +0000924 .. method:: assertSameElements(actual, expected, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000925
Benjamin Peterson5e55b3e2010-02-03 02:35:45 +0000926 Test that sequence *expected* contains the same elements as *actual*,
927 regardless of their order. When they don't, an error message listing
928 the differences between the sequences will be generated.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000929
Michael Foorde9abbee2010-02-05 20:54:27 +0000930 Duplicate elements are ignored when comparing *actual* and *expected*.
931 It is the equivalent of ``assertEqual(set(expected), set(actual))``
Michael Foordabd91d52010-03-20 18:09:14 +0000932 but it works with sequences of unhashable objects as well. Because
933 duplicates are ignored, this method has been deprecated in favour of
934 :meth:`assertItemsEqual`.
Michael Foorde9abbee2010-02-05 20:54:27 +0000935
Michael Foordabd91d52010-03-20 18:09:14 +0000936 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000937
Raymond Hettinger35a88362009-04-09 00:08:24 +0000938 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000939
Michael Foordabd91d52010-03-20 18:09:14 +0000940 .. deprecated:: 3.2
941
942 .. method:: assertItemsEqual(actual, expected, msg=None)
943
944 Test that sequence *expected* contains the same elements as *actual*,
945 regardless of their order. When they don't, an error message listing the
946 differences between the sequences will be generated.
947
948 Duplicate elements are *not* ignored when comparing *actual* and
949 *expected*. It verifies if each element has the same count in both
950 sequences. It is the equivalent of ``assertEqual(sorted(expected),
951 sorted(actual))`` but it works with sequences of unhashable objects as
952 well.
953
954 If specified, *msg* will be used as the error message on failure.
955
956 .. versionadded:: 3.2
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000957
958 .. method:: assertSetEqual(set1, set2, msg=None)
959
960 Tests that two sets are equal. If not, an error message is constructed
Michael Foord02834952010-02-08 23:10:39 +0000961 that lists the differences between the sets. This method is used by
962 default when comparing sets or frozensets with :meth:`assertEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000963
964 Fails if either of *set1* or *set2* does not have a :meth:`set.difference`
965 method.
966
Michael Foordabd91d52010-03-20 18:09:14 +0000967 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000968
Raymond Hettinger35a88362009-04-09 00:08:24 +0000969 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000970
971
972 .. method:: assertDictEqual(expected, actual, msg=None)
973
974 Test that two dictionaries are equal. If not, an error message is
Michael Foord02834952010-02-08 23:10:39 +0000975 constructed that shows the differences in the dictionaries. This
976 method will be used by default to compare dictionaries in
977 calls to :meth:`assertEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000978
Michael Foordabd91d52010-03-20 18:09:14 +0000979 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000980
Raymond Hettinger35a88362009-04-09 00:08:24 +0000981 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000982
983
984 .. method:: assertDictContainsSubset(expected, actual, msg=None)
985
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000986 Tests whether the key/value pairs in dictionary *actual* are a
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000987 superset of those in *expected*. If not, an error message listing
988 the missing keys and mismatched values is generated.
989
Michael Foordabd91d52010-03-20 18:09:14 +0000990 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000991
Raymond Hettinger35a88362009-04-09 00:08:24 +0000992 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000993
994
995 .. method:: assertListEqual(list1, list2, msg=None)
996 assertTupleEqual(tuple1, tuple2, msg=None)
997
998 Tests that two lists or tuples are equal. If not an error message is
999 constructed that shows only the differences between the two. An error
1000 is also raised if either of the parameters are of the wrong type.
Michael Foord02834952010-02-08 23:10:39 +00001001 These methods are used by default when comparing lists or tuples with
1002 :meth:`assertEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001003
Michael Foordabd91d52010-03-20 18:09:14 +00001004 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001005
Raymond Hettinger35a88362009-04-09 00:08:24 +00001006 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001007
1008
1009 .. method:: assertSequenceEqual(seq1, seq2, msg=None, seq_type=None)
1010
1011 Tests that two sequences are equal. If a *seq_type* is supplied, both
1012 *seq1* and *seq2* must be instances of *seq_type* or a failure will
1013 be raised. If the sequences are different an error message is
1014 constructed that shows the difference between the two.
1015
Michael Foordabd91d52010-03-20 18:09:14 +00001016 If specified, *msg* will be used as the error message on failure.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001017
1018 This method is used to implement :meth:`assertListEqual` and
1019 :meth:`assertTupleEqual`.
1020
Raymond Hettinger35a88362009-04-09 00:08:24 +00001021 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001022
Benjamin Peterson52baa292009-03-24 00:56:30 +00001023
Georg Brandl7f01a132009-09-16 15:58:14 +00001024 .. method:: assertRaises(exception, callable, *args, **kwds)
1025 failUnlessRaises(exception, callable, *args, **kwds)
1026 assertRaises(exception)
1027 failUnlessRaises(exception)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001028
1029 Test that an exception is raised when *callable* is called with any
1030 positional or keyword arguments that are also passed to
1031 :meth:`assertRaises`. The test passes if *exception* is raised, is an
1032 error if another exception is raised, or fails if no exception is raised.
1033 To catch any of a group of exceptions, a tuple containing the exception
1034 classes may be passed as *exception*.
1035
Georg Brandl7f01a132009-09-16 15:58:14 +00001036 If only the *exception* argument is given, returns a context manager so
1037 that the code under test can be written inline rather than as a function::
Benjamin Petersonded31c42009-03-30 15:04:16 +00001038
Michael Foord41531f22010-02-05 21:13:40 +00001039 with self.assertRaises(SomeException):
Benjamin Petersonded31c42009-03-30 15:04:16 +00001040 do_something()
1041
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +00001042 The context manager will store the caught exception object in its
Ezio Melotti49008232010-02-08 21:57:48 +00001043 :attr:`exception` attribute. This can be useful if the intention
Michael Foord41531f22010-02-05 21:13:40 +00001044 is to perform additional checks on the exception raised::
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +00001045
Michael Foord41531f22010-02-05 21:13:40 +00001046 with self.assertRaises(SomeException) as cm:
1047 do_something()
1048
Ezio Melotti49008232010-02-08 21:57:48 +00001049 the_exception = cm.exception
Michael Foordb112a412010-02-05 23:32:33 +00001050 self.assertEqual(the_exception.error_code, 3)
Michael Foord41531f22010-02-05 21:13:40 +00001051
Ezio Melotti49008232010-02-08 21:57:48 +00001052 .. versionchanged:: 3.1
Benjamin Petersonded31c42009-03-30 15:04:16 +00001053 Added the ability to use :meth:`assertRaises` as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001054
Ezio Melotti49008232010-02-08 21:57:48 +00001055 .. versionchanged:: 3.2
1056 Added the :attr:`exception` attribute.
1057
Raymond Hettinger35a88362009-04-09 00:08:24 +00001058 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +00001059 :meth:`failUnlessRaises`; use :meth:`assertRaises`.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001060
Benjamin Peterson52baa292009-03-24 00:56:30 +00001061
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001062 .. method:: assertRaisesRegexp(exception, regexp[, callable, ...])
1063
1064 Like :meth:`assertRaises` but also tests that *regexp* matches
1065 on the string representation of the raised exception. *regexp* may be
1066 a regular expression object or a string containing a regular expression
1067 suitable for use by :func:`re.search`. Examples::
1068
1069 self.assertRaisesRegexp(ValueError, 'invalid literal for.*XYZ$',
1070 int, 'XYZ')
1071
1072 or::
1073
1074 with self.assertRaisesRegexp(ValueError, 'literal'):
1075 int('XYZ')
1076
Raymond Hettinger35a88362009-04-09 00:08:24 +00001077 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001078
1079
Georg Brandl7f01a132009-09-16 15:58:14 +00001080 .. method:: assertIsNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001081
1082 This signals a test failure if *expr* is not None.
1083
Raymond Hettinger35a88362009-04-09 00:08:24 +00001084 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001085
1086
Georg Brandl7f01a132009-09-16 15:58:14 +00001087 .. method:: assertIsNotNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001088
1089 The inverse of the :meth:`assertIsNone` method.
1090 This signals a test failure if *expr* is None.
1091
Raymond Hettinger35a88362009-04-09 00:08:24 +00001092 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001093
1094
Georg Brandl7f01a132009-09-16 15:58:14 +00001095 .. method:: assertIs(expr1, expr2, msg=None)
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001096
1097 This signals a test failure if *expr1* and *expr2* don't evaluate to the same
1098 object.
1099
Georg Brandl705d9d52009-05-05 09:29:50 +00001100 .. versionadded:: 3.1
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001101
1102
Georg Brandl7f01a132009-09-16 15:58:14 +00001103 .. method:: assertIsNot(expr1, expr2, msg=None)
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001104
1105 The inverse of the :meth:`assertIs` method.
1106 This signals a test failure if *expr1* and *expr2* evaluate to the same
1107 object.
1108
Georg Brandl705d9d52009-05-05 09:29:50 +00001109 .. versionadded:: 3.1
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001110
1111
Benjamin Peterson6e8c7572009-10-04 20:19:21 +00001112 .. method:: assertIsInstance(obj, cls[, msg])
1113
1114 This signals a test failure if *obj* is not an instance of *cls* (which
1115 can be a class or a tuple of classes, as supported by :func:`isinstance`).
1116
1117 .. versionadded:: 3.2
1118
1119
1120 .. method:: assertNotIsInstance(obj, cls[, msg])
1121
1122 The inverse of the :meth:`assertIsInstance` method. This signals a test
1123 failure if *obj* is an instance of *cls*.
1124
1125 .. versionadded:: 3.2
1126
1127
Georg Brandl7f01a132009-09-16 15:58:14 +00001128 .. method:: assertFalse(expr, msg=None)
1129 failIf(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001130
1131 The inverse of the :meth:`assertTrue` method is the :meth:`assertFalse` method.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001132 This signals a test failure if *expr* is true, with *msg* or :const:`None`
1133 for the error message.
1134
Raymond Hettinger35a88362009-04-09 00:08:24 +00001135 .. deprecated:: 3.1
Georg Brandl89fad142010-03-14 10:23:39 +00001136 :meth:`failIf`; use :meth:`assertFalse`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001137
Benjamin Peterson52baa292009-03-24 00:56:30 +00001138
Georg Brandl7f01a132009-09-16 15:58:14 +00001139 .. method:: fail(msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001140
1141 Signals a test failure unconditionally, with *msg* or :const:`None` for
1142 the error message.
1143
1144
1145 .. attribute:: failureException
1146
1147 This class attribute gives the exception raised by the test method. If a
1148 test framework needs to use a specialized exception, possibly to carry
1149 additional information, it must subclass this exception in order to "play
1150 fair" with the framework. The initial value of this attribute is
1151 :exc:`AssertionError`.
1152
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001153
1154 .. attribute:: longMessage
1155
1156 If set to True then any explicit failure message you pass in to the
1157 assert methods will be appended to the end of the normal failure message.
1158 The normal messages contain useful information about the objects involved,
1159 for example the message from assertEqual shows you the repr of the two
1160 unequal objects. Setting this attribute to True allows you to have a
1161 custom error message in addition to the normal one.
1162
1163 This attribute defaults to False, meaning that a custom message passed
1164 to an assert method will silence the normal message.
1165
1166 The class setting can be overridden in individual tests by assigning an
1167 instance attribute to True or False before calling the assert methods.
1168
Raymond Hettinger35a88362009-04-09 00:08:24 +00001169 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001170
1171
Benjamin Peterson52baa292009-03-24 00:56:30 +00001172 Testing frameworks can use the following methods to collect information on
1173 the test:
1174
1175
1176 .. method:: countTestCases()
1177
1178 Return the number of tests represented by this test object. For
1179 :class:`TestCase` instances, this will always be ``1``.
1180
1181
1182 .. method:: defaultTestResult()
1183
1184 Return an instance of the test result class that should be used for this
1185 test case class (if no other result instance is provided to the
1186 :meth:`run` method).
1187
1188 For :class:`TestCase` instances, this will always be an instance of
1189 :class:`TestResult`; subclasses of :class:`TestCase` should override this
1190 as necessary.
1191
1192
1193 .. method:: id()
1194
1195 Return a string identifying the specific test case. This is usually the
1196 full name of the test method, including the module and class name.
1197
1198
1199 .. method:: shortDescription()
1200
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001201 Returns a description of the test, or :const:`None` if no description
1202 has been provided. The default implementation of this method
1203 returns the first line of the test method's docstring, if available,
Michael Foord34c94622010-02-10 15:51:42 +00001204 or :const:`None`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001205
Michael Foord34c94622010-02-10 15:51:42 +00001206 .. versionchanged:: 3.1,3.2
1207 In 3.1 this was changed to add the test name to the short description
1208 even in the presence of a docstring. This caused compatibility issues
1209 with unittest extensions and adding the test name was moved to the
1210 :class:`TextTestResult`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001211
1212 .. method:: addTypeEqualityFunc(typeobj, function)
1213
1214 Registers a type specific :meth:`assertEqual` equality checking
1215 function to be called by :meth:`assertEqual` when both objects it has
1216 been asked to compare are exactly *typeobj* (not subclasses).
1217 *function* must take two positional arguments and a third msg=None
1218 keyword argument just as :meth:`assertEqual` does. It must raise
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001219 ``self.failureException`` when inequality between the first two
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001220 parameters is detected.
1221
1222 One good use of custom equality checking functions for a type
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001223 is to raise ``self.failureException`` with an error message useful
1224 for debugging the problem by explaining the inequalities in detail.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001225
Raymond Hettinger35a88362009-04-09 00:08:24 +00001226 .. versionadded:: 3.1
Georg Brandl116aa622007-08-15 14:28:22 +00001227
1228
Georg Brandl7f01a132009-09-16 15:58:14 +00001229 .. method:: addCleanup(function, *args, **kwargs)
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001230
1231 Add a function to be called after :meth:`tearDown` to cleanup resources
1232 used during the test. Functions will be called in reverse order to the
1233 order they are added (LIFO). They are called with any arguments and
1234 keyword arguments passed into :meth:`addCleanup` when they are
1235 added.
1236
1237 If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
1238 then any cleanup functions added will still be called.
1239
Georg Brandl853947a2010-01-31 18:53:23 +00001240 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001241
1242
1243 .. method:: doCleanups()
1244
Barry Warsaw0c9fd632010-04-12 14:50:57 +00001245 This method is called unconditionally after :meth:`tearDown`, or
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001246 after :meth:`setUp` if :meth:`setUp` raises an exception.
1247
1248 It is responsible for calling all the cleanup functions added by
1249 :meth:`addCleanup`. If you need cleanup functions to be called
1250 *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
1251 yourself.
1252
1253 :meth:`doCleanups` pops methods off the stack of cleanup
1254 functions one at a time, so it can be called at any time.
1255
Georg Brandl853947a2010-01-31 18:53:23 +00001256 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001257
1258
Georg Brandl7f01a132009-09-16 15:58:14 +00001259.. class:: FunctionTestCase(testFunc, setUp=None, tearDown=None, description=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001260
1261 This class implements the portion of the :class:`TestCase` interface which
Benjamin Petersond2397752009-06-27 23:45:02 +00001262 allows the test runner to drive the test, but does not provide the methods
1263 which test code can use to check and report errors. This is used to create
1264 test cases using legacy test code, allowing it to be integrated into a
1265 :mod:`unittest`-based test framework.
Georg Brandl116aa622007-08-15 14:28:22 +00001266
1267
Benjamin Peterson52baa292009-03-24 00:56:30 +00001268.. _testsuite-objects:
1269
Benjamin Peterson52baa292009-03-24 00:56:30 +00001270Grouping tests
1271~~~~~~~~~~~~~~
1272
Georg Brandl7f01a132009-09-16 15:58:14 +00001273.. class:: TestSuite(tests=())
Georg Brandl116aa622007-08-15 14:28:22 +00001274
1275 This class represents an aggregation of individual tests cases and test suites.
1276 The class presents the interface needed by the test runner to allow it to be run
1277 as any other test case. Running a :class:`TestSuite` instance is the same as
1278 iterating over the suite, running each test individually.
1279
1280 If *tests* is given, it must be an iterable of individual test cases or other
1281 test suites that will be used to build the suite initially. Additional methods
1282 are provided to add test cases and suites to the collection later on.
1283
Benjamin Peterson14a3dd72009-05-25 00:51:58 +00001284 :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
1285 they do not actually implement a test. Instead, they are used to aggregate
1286 tests into groups of tests that should be run together. Some additional
1287 methods are available to add tests to :class:`TestSuite` instances:
Benjamin Peterson52baa292009-03-24 00:56:30 +00001288
1289
1290 .. method:: TestSuite.addTest(test)
1291
1292 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1293
1294
1295 .. method:: TestSuite.addTests(tests)
1296
1297 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
1298 instances to this test suite.
1299
Benjamin Petersond2397752009-06-27 23:45:02 +00001300 This is equivalent to iterating over *tests*, calling :meth:`addTest` for
1301 each element.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001302
1303 :class:`TestSuite` shares the following methods with :class:`TestCase`:
1304
1305
1306 .. method:: run(result)
1307
1308 Run the tests associated with this suite, collecting the result into the
1309 test result object passed as *result*. Note that unlike
1310 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
1311 be passed in.
1312
1313
1314 .. method:: debug()
1315
1316 Run the tests associated with this suite without collecting the
1317 result. This allows exceptions raised by the test to be propagated to the
1318 caller and can be used to support running tests under a debugger.
1319
1320
1321 .. method:: countTestCases()
1322
1323 Return the number of tests represented by this test object, including all
1324 individual tests and sub-suites.
1325
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001326
1327 .. method:: __iter__()
1328
1329 Tests grouped by a :class:`TestSuite` are always accessed by iteration.
1330 Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
1331 that this method maybe called several times on a single suite
1332 (for example when counting tests or comparing for equality)
1333 so the tests returned must be the same for repeated iterations.
1334
Georg Brandl853947a2010-01-31 18:53:23 +00001335 .. versionchanged:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001336 In earlier versions the :class:`TestSuite` accessed tests directly rather
1337 than through iteration, so overriding :meth:`__iter__` wasn't sufficient
1338 for providing tests.
1339
Benjamin Peterson52baa292009-03-24 00:56:30 +00001340 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
1341 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1342
1343
Benjamin Peterson52baa292009-03-24 00:56:30 +00001344Loading and running tests
1345~~~~~~~~~~~~~~~~~~~~~~~~~
1346
Georg Brandl116aa622007-08-15 14:28:22 +00001347.. class:: TestLoader()
1348
Benjamin Peterson52baa292009-03-24 00:56:30 +00001349 The :class:`TestLoader` class is used to create test suites from classes and
1350 modules. Normally, there is no need to create an instance of this class; the
1351 :mod:`unittest` module provides an instance that can be shared as
1352 ``unittest.defaultTestLoader``. Using a subclass or instance, however, allows
1353 customization of some configurable properties.
1354
1355 :class:`TestLoader` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001356
Michael Foordabd91d52010-03-20 18:09:14 +00001357a
Benjamin Peterson52baa292009-03-24 00:56:30 +00001358 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl116aa622007-08-15 14:28:22 +00001359
Benjamin Peterson52baa292009-03-24 00:56:30 +00001360 Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
1361 :class:`testCaseClass`.
1362
1363
1364 .. method:: loadTestsFromModule(module)
1365
1366 Return a suite of all tests cases contained in the given module. This
1367 method searches *module* for classes derived from :class:`TestCase` and
1368 creates an instance of the class for each test method defined for the
1369 class.
1370
Georg Brandle720c0a2009-04-27 16:20:50 +00001371 .. note::
Benjamin Peterson52baa292009-03-24 00:56:30 +00001372
1373 While using a hierarchy of :class:`TestCase`\ -derived classes can be
1374 convenient in sharing fixtures and helper functions, defining test
1375 methods on base classes that are not intended to be instantiated
1376 directly does not play well with this method. Doing so, however, can
1377 be useful when the fixtures are different and defined in subclasses.
1378
Benjamin Petersond2397752009-06-27 23:45:02 +00001379 If a module provides a ``load_tests`` function it will be called to
1380 load the tests. This allows modules to customize test loading.
1381 This is the `load_tests protocol`_.
1382
Georg Brandl853947a2010-01-31 18:53:23 +00001383 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001384 Support for ``load_tests`` added.
1385
Benjamin Peterson52baa292009-03-24 00:56:30 +00001386
Georg Brandl7f01a132009-09-16 15:58:14 +00001387 .. method:: loadTestsFromName(name, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001388
1389 Return a suite of all tests cases given a string specifier.
1390
1391 The specifier *name* is a "dotted name" that may resolve either to a
1392 module, a test case class, a test method within a test case class, a
1393 :class:`TestSuite` instance, or a callable object which returns a
1394 :class:`TestCase` or :class:`TestSuite` instance. These checks are
1395 applied in the order listed here; that is, a method on a possible test
1396 case class will be picked up as "a test method within a test case class",
1397 rather than "a callable object".
1398
1399 For example, if you have a module :mod:`SampleTests` containing a
1400 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
1401 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
Benjamin Petersond2397752009-06-27 23:45:02 +00001402 specifier ``'SampleTests.SampleTestCase'`` would cause this method to
1403 return a suite which will run all three test methods. Using the specifier
1404 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
1405 suite which will run only the :meth:`test_two` test method. The specifier
1406 can refer to modules and packages which have not been imported; they will
1407 be imported as a side-effect.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001408
1409 The method optionally resolves *name* relative to the given *module*.
1410
1411
Georg Brandl7f01a132009-09-16 15:58:14 +00001412 .. method:: loadTestsFromNames(names, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001413
1414 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
1415 than a single name. The return value is a test suite which supports all
1416 the tests defined for each name.
1417
1418
1419 .. method:: getTestCaseNames(testCaseClass)
1420
1421 Return a sorted sequence of method names found within *testCaseClass*;
1422 this should be a subclass of :class:`TestCase`.
1423
Benjamin Petersond2397752009-06-27 23:45:02 +00001424
1425 .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)
1426
1427 Find and return all test modules from the specified start directory,
1428 recursing into subdirectories to find them. Only test files that match
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001429 *pattern* will be loaded. (Using shell style pattern matching.) Only
1430 module names that are importable (i.e. are valid Python identifiers) will
1431 be loaded.
Benjamin Petersond2397752009-06-27 23:45:02 +00001432
1433 All test modules must be importable from the top level of the project. If
1434 the start directory is not the top level directory then the top level
1435 directory must be specified separately.
1436
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001437 If importing a module fails, for example due to a syntax error, then this
1438 will be recorded as a single error and discovery will continue.
1439
Benjamin Petersond2397752009-06-27 23:45:02 +00001440 If a test package name (directory with :file:`__init__.py`) matches the
1441 pattern then the package will be checked for a ``load_tests``
1442 function. If this exists then it will be called with *loader*, *tests*,
1443 *pattern*.
1444
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001445 If load_tests exists then discovery does *not* recurse into the package,
Benjamin Petersond2397752009-06-27 23:45:02 +00001446 ``load_tests`` is responsible for loading all tests in the package.
1447
1448 The pattern is deliberately not stored as a loader attribute so that
1449 packages can continue discovery themselves. *top_level_dir* is stored so
1450 ``load_tests`` does not need to pass this argument in to
1451 ``loader.discover()``.
1452
Benjamin Petersonb48af542010-04-11 20:43:16 +00001453 *start_dir* can be a dotted module name as well as a directory.
1454
Georg Brandl853947a2010-01-31 18:53:23 +00001455 .. versionadded:: 3.2
1456
Benjamin Petersond2397752009-06-27 23:45:02 +00001457
Benjamin Peterson52baa292009-03-24 00:56:30 +00001458 The following attributes of a :class:`TestLoader` can be configured either by
1459 subclassing or assignment on an instance:
1460
1461
1462 .. attribute:: testMethodPrefix
1463
1464 String giving the prefix of method names which will be interpreted as test
1465 methods. The default value is ``'test'``.
1466
1467 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
1468 methods.
1469
1470
1471 .. attribute:: sortTestMethodsUsing
1472
1473 Function to be used to compare method names when sorting them in
1474 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
1475
1476
1477 .. attribute:: suiteClass
1478
1479 Callable object that constructs a test suite from a list of tests. No
1480 methods on the resulting object are needed. The default value is the
1481 :class:`TestSuite` class.
1482
1483 This affects all the :meth:`loadTestsFrom\*` methods.
1484
1485
Benjamin Peterson52baa292009-03-24 00:56:30 +00001486.. class:: TestResult
1487
1488 This class is used to compile information about which tests have succeeded
1489 and which have failed.
1490
1491 A :class:`TestResult` object stores the results of a set of tests. The
1492 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
1493 properly recorded; test authors do not need to worry about recording the
1494 outcome of tests.
1495
1496 Testing frameworks built on top of :mod:`unittest` may want access to the
1497 :class:`TestResult` object generated by running a set of tests for reporting
1498 purposes; a :class:`TestResult` instance is returned by the
1499 :meth:`TestRunner.run` method for this purpose.
1500
1501 :class:`TestResult` instances have the following attributes that will be of
1502 interest when inspecting the results of running a set of tests:
1503
1504
1505 .. attribute:: errors
1506
1507 A list containing 2-tuples of :class:`TestCase` instances and strings
1508 holding formatted tracebacks. Each tuple represents a test which raised an
1509 unexpected exception.
1510
Benjamin Peterson52baa292009-03-24 00:56:30 +00001511 .. attribute:: failures
1512
1513 A list containing 2-tuples of :class:`TestCase` instances and strings
1514 holding formatted tracebacks. Each tuple represents a test where a failure
1515 was explicitly signalled using the :meth:`TestCase.fail\*` or
1516 :meth:`TestCase.assert\*` methods.
1517
Benjamin Peterson52baa292009-03-24 00:56:30 +00001518 .. attribute:: skipped
1519
1520 A list containing 2-tuples of :class:`TestCase` instances and strings
1521 holding the reason for skipping the test.
1522
Benjamin Peterson70e32c82009-03-24 01:00:11 +00001523 .. versionadded:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +00001524
1525 .. attribute:: expectedFailures
1526
1527 A list contaning 2-tuples of :class:`TestCase` instances and strings
1528 holding formatted tracebacks. Each tuple represents a expected failures
1529 of the test case.
1530
1531 .. attribute:: unexpectedSuccesses
1532
1533 A list containing :class:`TestCase` instances that were marked as expected
1534 failures, but succeeded.
1535
1536 .. attribute:: shouldStop
1537
1538 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1539
1540
1541 .. attribute:: testsRun
1542
1543 The total number of tests run so far.
1544
1545
Benjamin Petersonb48af542010-04-11 20:43:16 +00001546 .. attribute:: buffer
1547
1548 If set to true, ``sys.stdout`` and ``sys.stderr`` will be buffered in between
1549 :meth:`startTest` and :meth:`stopTest` being called. Collected output will
1550 only be echoed onto the real ``sys.stdout`` and ``sys.stderr`` if the test
1551 fails or errors. Any output is also attached to the failure / error message.
1552
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001553 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001554
1555
1556 .. attribute:: failfast
1557
1558 If set to true :meth:`stop` will be called on the first failure or error,
1559 halting the test run.
1560
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001561 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001562
1563
Benjamin Peterson52baa292009-03-24 00:56:30 +00001564 .. method:: wasSuccessful()
1565
1566 Return :const:`True` if all tests run so far have passed, otherwise returns
1567 :const:`False`.
1568
1569
1570 .. method:: stop()
1571
1572 This method can be called to signal that the set of tests being run should
1573 be aborted by setting the :attr:`shouldStop` attribute to :const:`True`.
1574 :class:`TestRunner` objects should respect this flag and return without
1575 running any additional tests.
1576
1577 For example, this feature is used by the :class:`TextTestRunner` class to
1578 stop the test framework when the user signals an interrupt from the
1579 keyboard. Interactive tools which provide :class:`TestRunner`
1580 implementations can use this in a similar manner.
1581
1582 The following methods of the :class:`TestResult` class are used to maintain
1583 the internal data structures, and may be extended in subclasses to support
1584 additional reporting requirements. This is particularly useful in building
1585 tools which support interactive reporting while tests are being run.
1586
1587
1588 .. method:: startTest(test)
1589
1590 Called when the test case *test* is about to be run.
1591
Benjamin Peterson52baa292009-03-24 00:56:30 +00001592 .. method:: stopTest(test)
1593
1594 Called after the test case *test* has been executed, regardless of the
1595 outcome.
1596
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001597 .. method:: startTestRun(test)
1598
1599 Called once before any tests are executed.
1600
Georg Brandl853947a2010-01-31 18:53:23 +00001601 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001602
1603
1604 .. method:: stopTestRun(test)
1605
Ezio Melotti176d6c42010-01-27 20:58:07 +00001606 Called once after all tests are executed.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001607
Georg Brandl853947a2010-01-31 18:53:23 +00001608 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001609
1610
Benjamin Peterson52baa292009-03-24 00:56:30 +00001611 .. method:: addError(test, err)
1612
1613 Called when the test case *test* raises an unexpected exception *err* is a
1614 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1615 traceback)``.
1616
1617 The default implementation appends a tuple ``(test, formatted_err)`` to
1618 the instance's :attr:`errors` attribute, where *formatted_err* is a
1619 formatted traceback derived from *err*.
1620
1621
1622 .. method:: addFailure(test, err)
1623
Benjamin Petersond2397752009-06-27 23:45:02 +00001624 Called when the test case *test* signals a failure. *err* is a tuple of
1625 the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001626
1627 The default implementation appends a tuple ``(test, formatted_err)`` to
1628 the instance's :attr:`failures` attribute, where *formatted_err* is a
1629 formatted traceback derived from *err*.
1630
1631
1632 .. method:: addSuccess(test)
1633
1634 Called when the test case *test* succeeds.
1635
1636 The default implementation does nothing.
1637
1638
1639 .. method:: addSkip(test, reason)
1640
1641 Called when the test case *test* is skipped. *reason* is the reason the
1642 test gave for skipping.
1643
1644 The default implementation appends a tuple ``(test, reason)`` to the
1645 instance's :attr:`skipped` attribute.
1646
1647
1648 .. method:: addExpectedFailure(test, err)
1649
1650 Called when the test case *test* fails, but was marked with the
1651 :func:`expectedFailure` decorator.
1652
1653 The default implementation appends a tuple ``(test, formatted_err)`` to
1654 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1655 is a formatted traceback derived from *err*.
1656
1657
1658 .. method:: addUnexpectedSuccess(test)
1659
1660 Called when the test case *test* was marked with the
1661 :func:`expectedFailure` decorator, but succeeded.
1662
1663 The default implementation appends the test to the instance's
1664 :attr:`unexpectedSuccesses` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +00001665
Michael Foord34c94622010-02-10 15:51:42 +00001666.. class:: TextTestResult(stream, descriptions, verbosity)
1667
1668 A concrete implementation of :class:`TestResult` used by the
1669 :class:`TextTestRunner`.
1670
1671 .. versionadded:: 3.2
1672 This class was previously named ``_TextTestResult``. The old name still
1673 exists as an alias but is deprecated.
Georg Brandl116aa622007-08-15 14:28:22 +00001674
1675.. data:: defaultTestLoader
1676
1677 Instance of the :class:`TestLoader` class intended to be shared. If no
1678 customization of the :class:`TestLoader` is needed, this instance can be used
1679 instead of repeatedly creating new instances.
1680
1681
Michael Foord34c94622010-02-10 15:51:42 +00001682.. class:: TextTestRunner(stream=sys.stderr, descriptions=True, verbosity=1, runnerclass=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001683
1684 A basic test runner implementation which prints results on standard error. It
1685 has a few configurable parameters, but is essentially very simple. Graphical
1686 applications which run test suites should provide alternate implementations.
1687
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001688 .. method:: _makeResult()
Georg Brandl116aa622007-08-15 14:28:22 +00001689
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001690 This method returns the instance of ``TestResult`` used by :meth:`run`.
1691 It is not intended to be called directly, but can be overridden in
1692 subclasses to provide a custom ``TestResult``.
1693
Michael Foord34c94622010-02-10 15:51:42 +00001694 ``_makeResult()`` instantiates the class or callable passed in the
1695 ``TextTestRunner`` constructor as the ``resultclass`` argument. It
Benjamin Petersonb48af542010-04-11 20:43:16 +00001696 defaults to :class:`TextTestResult` if no ``resultclass`` is provided.
Michael Foord34c94622010-02-10 15:51:42 +00001697 The result class is instantiated with the following arguments::
1698
1699 stream, descriptions, verbosity
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001700
Benjamin Petersonb48af542010-04-11 20:43:16 +00001701.. 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 +00001702
1703 A command-line program that runs a set of tests; this is primarily for making
1704 test modules conveniently executable. The simplest use for this function is to
1705 include the following line at the end of a test script::
1706
1707 if __name__ == '__main__':
1708 unittest.main()
1709
Benjamin Petersond2397752009-06-27 23:45:02 +00001710 You can run tests with more detailed information by passing in the verbosity
1711 argument::
1712
1713 if __name__ == '__main__':
1714 unittest.main(verbosity=2)
1715
Georg Brandl116aa622007-08-15 14:28:22 +00001716 The *testRunner* argument can either be a test runner class or an already
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001717 created instance of it. By default ``main`` calls :func:`sys.exit` with
1718 an exit code indicating success or failure of the tests run.
1719
1720 ``main`` supports being used from the interactive interpreter by passing in the
1721 argument ``exit=False``. This displays the result on standard output without
1722 calling :func:`sys.exit`::
1723
1724 >>> from unittest import main
1725 >>> main(module='test_module', exit=False)
1726
Benjamin Petersonb48af542010-04-11 20:43:16 +00001727 The ``failfast``, ``catchbreak`` and ``buffer`` parameters have the same
1728 effect as the `failfast, catch and buffer command line options`_.
1729
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001730 Calling ``main`` actually returns an instance of the ``TestProgram`` class.
1731 This stores the result of the tests run as the ``result`` attribute.
1732
Georg Brandl853947a2010-01-31 18:53:23 +00001733 .. versionchanged:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001734 The ``exit``, ``verbosity``, ``failfast``, ``catchbreak`` and ``buffer``
1735 parameters were added.
Benjamin Petersond2397752009-06-27 23:45:02 +00001736
1737
1738load_tests Protocol
1739###################
1740
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001741
Georg Brandl853947a2010-01-31 18:53:23 +00001742.. versionadded:: 3.2
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001743
1744
Benjamin Petersond2397752009-06-27 23:45:02 +00001745Modules or packages can customize how tests are loaded from them during normal
1746test runs or test discovery by implementing a function called ``load_tests``.
1747
1748If a test module defines ``load_tests`` it will be called by
1749:meth:`TestLoader.loadTestsFromModule` with the following arguments::
1750
1751 load_tests(loader, standard_tests, None)
1752
1753It should return a :class:`TestSuite`.
1754
1755*loader* is the instance of :class:`TestLoader` doing the loading.
1756*standard_tests* are the tests that would be loaded by default from the
1757module. It is common for test modules to only want to add or remove tests
1758from the standard set of tests.
1759The third argument is used when loading packages as part of test discovery.
1760
1761A typical ``load_tests`` function that loads tests from a specific set of
1762:class:`TestCase` classes may look like::
1763
1764 test_cases = (TestCase1, TestCase2, TestCase3)
1765
1766 def load_tests(loader, tests, pattern):
1767 suite = TestSuite()
1768 for test_class in test_cases:
1769 tests = loader.loadTestsFromTestCase(test_class)
1770 suite.addTests(tests)
1771 return suite
1772
1773If discovery is started, either from the command line or by calling
1774:meth:`TestLoader.discover`, with a pattern that matches a package
1775name then the package :file:`__init__.py` will be checked for ``load_tests``.
1776
1777.. note::
1778
Ezio Melotti0639d5a2009-12-19 23:26:38 +00001779 The default pattern is 'test*.py'. This matches all Python files
Benjamin Petersond2397752009-06-27 23:45:02 +00001780 that start with 'test' but *won't* match any test directories.
1781
1782 A pattern like 'test*' will match test packages as well as
1783 modules.
1784
1785If the package :file:`__init__.py` defines ``load_tests`` then it will be
1786called and discovery not continued into the package. ``load_tests``
1787is called with the following arguments::
1788
1789 load_tests(loader, standard_tests, pattern)
1790
1791This should return a :class:`TestSuite` representing all the tests
1792from the package. (``standard_tests`` will only contain tests
1793collected from :file:`__init__.py`.)
1794
1795Because the pattern is passed into ``load_tests`` the package is free to
1796continue (and potentially modify) test discovery. A 'do nothing'
1797``load_tests`` function for a test package would look like::
1798
1799 def load_tests(loader, standard_tests, pattern):
1800 # top level directory cached on loader instance
1801 this_dir = os.path.dirname(__file__)
1802 package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
1803 standard_tests.addTests(package_tests)
1804 return standard_tests
Benjamin Petersonb48af542010-04-11 20:43:16 +00001805
1806
1807Class and Module Fixtures
1808-------------------------
1809
1810Class and module level fixtures are implemented in :class:`TestSuite`. When
1811the test suite encounters a test from a new class then :meth:`tearDownClass`
1812from the previous class (if there is one) is called, followed by
1813:meth:`setUpClass` from the new class.
1814
1815Similarly if a test is from a different module from the previous test then
1816``tearDownModule`` from the previous module is run, followed by
1817``setUpModule`` from the new module.
1818
1819After all the tests have run the final ``tearDownClass`` and
1820``tearDownModule`` are run.
1821
1822Note that shared fixtures do not play well with [potential] features like test
1823parallelization and they break test isolation. They should be used with care.
1824
1825The default ordering of tests created by the unittest test loaders is to group
1826all tests from the same modules and classes together. This will lead to
1827``setUpClass`` / ``setUpModule`` (etc) being called exactly once per class and
1828module. If you randomize the order, so that tests from different modules and
1829classes are adjacent to each other, then these shared fixture functions may be
1830called multiple times in a single test run.
1831
1832Shared fixtures are not intended to work with suites with non-standard
1833ordering. A ``BaseTestSuite`` still exists for frameworks that don't want to
1834support shared fixtures.
1835
1836If there are any exceptions raised during one of the shared fixture functions
1837the test is reported as an error. Because there is no corresponding test
1838instance an ``_ErrorHolder`` object (that has the same interface as a
1839:class:`TestCase`) is created to represent the error. If you are just using
1840the standard unittest test runner then this detail doesn't matter, but if you
1841are a framework author it may be relevant.
1842
1843
1844setUpClass and tearDownClass
1845~~~~~~~~~~~~~~~~~~~~~~~~~~~~
1846
1847These must be implemented as class methods::
1848
1849 import unittest
1850
1851 class Test(unittest.TestCase):
1852 @classmethod
1853 def setUpClass(cls):
1854 cls._connection = createExpensiveConnectionObject()
1855
1856 @classmethod
1857 def tearDownClass(cls):
1858 cls._connection.destroy()
1859
1860If you want the ``setUpClass`` and ``tearDownClass`` on base classes called
1861then you must call up to them yourself. The implementations in
1862:class:`TestCase` are empty.
1863
1864If an exception is raised during a ``setUpClass`` then the tests in the class
1865are not run and the ``tearDownClass`` is not run. Skipped classes will not
1866have ``setUpClass`` or ``tearDownClass`` run.
1867
1868
1869setUpModule and tearDownModule
1870~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
1871
1872These should be implemented as functions::
1873
1874 def setUpModule():
1875 createConnection()
1876
1877 def tearDownModule():
1878 closeConnection()
1879
1880If an exception is raised in a ``setUpModule`` then none of the tests in the
1881module will be run and the ``tearDownModule`` will not be run.
1882
1883
1884Signal Handling
1885---------------
1886
1887The -c / --catch command line option to unittest, along with the ``catchbreak``
1888parameter to :func:`unittest.main()`, provide more friendly handling of
1889control-c during a test run. With catch break behavior enabled control-c will
1890allow the currently running test to complete, and the test run will then end
1891and report all the results so far. A second control-c will raise a
1892``KeyboardInterrupt`` in the usual way.
1893
Michael Foordde4ceab2010-04-25 19:53:49 +00001894The control-c handling signal handler attempts to remain compatible with code or
1895tests that install their own :const:`signal.SIGINT` handler. If the ``unittest``
1896handler is called but *isn't* the installed :const:`signal.SIGINT` handler,
1897i.e. it has been replaced by the system under test and delegated to, then it
1898calls the default handler. This will normally be the expected behavior by code
1899that replaces an installed handler and delegates to it. For individual tests
1900that need ``unittest`` control-c handling disabled the :func:`removeHandler`
1901decorator can be used.
1902
1903There are a few utility functions for framework authors to enable control-c
1904handling functionality within test frameworks.
Benjamin Petersonb48af542010-04-11 20:43:16 +00001905
1906.. function:: installHandler()
1907
1908 Install the control-c handler. When a :const:`signal.SIGINT` is received
1909 (usually in response to the user pressing control-c) all registered results
1910 have :meth:`~TestResult.stop` called.
1911
Michael Foord469b1f02010-04-26 23:41:26 +00001912 .. versionadded:: 3.2
1913
Benjamin Petersonb48af542010-04-11 20:43:16 +00001914.. function:: registerResult(result)
1915
1916 Register a :class:`TestResult` object for control-c handling. Registering a
1917 result stores a weak reference to it, so it doesn't prevent the result from
1918 being garbage collected.
1919
Michael Foordde4ceab2010-04-25 19:53:49 +00001920 Registering a :class:`TestResult` object has no side-effects if control-c
1921 handling is not enabled, so test frameworks can unconditionally register
1922 all results they create independently of whether or not handling is enabled.
1923
Michael Foord469b1f02010-04-26 23:41:26 +00001924 .. versionadded:: 3.2
1925
Benjamin Petersonb48af542010-04-11 20:43:16 +00001926.. function:: removeResult(result)
1927
1928 Remove a registered result. Once a result has been removed then
1929 :meth:`~TestResult.stop` will no longer be called on that result object in
1930 response to a control-c.
1931
Michael Foord469b1f02010-04-26 23:41:26 +00001932 .. versionadded:: 3.2
1933
Michael Foordde4ceab2010-04-25 19:53:49 +00001934.. function:: removeHandler(function=None)
1935
1936 When called without arguments this function removes the control-c handler
1937 if it has been installed. This function can also be used as a test decorator
1938 to temporarily remove the handler whilst the test is being executed::
1939
1940 @unittest.removeHandler
1941 def test_signal_handling(self):
1942 ...
1943
Michael Foord469b1f02010-04-26 23:41:26 +00001944 .. versionadded:: 3.2
1945