blob: 90f5bec0fc8072ce6117c41e5f0fc89e73b2f54b [file] [log] [blame]
Georg Brandl8ec7f652007-08-15 14:28:01 +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
12.. versionadded:: 2.1
Benjamin Peterson692428e2009-03-23 21:50:21 +000013
Georg Brandl8ec7f652007-08-15 14:28:01 +000014The Python unit testing framework, sometimes referred to as "PyUnit," is a
15Python language version of JUnit, by Kent Beck and Erich Gamma. JUnit is, in
16turn, a Java version of Kent's Smalltalk testing framework. Each is the de
17facto standard unit testing framework for its respective language.
18
19:mod:`unittest` supports test automation, sharing of setup and shutdown code for
20tests, aggregation of tests into collections, and independence of the tests from
21the reporting framework. The :mod:`unittest` module provides classes that make
22it easy to support these qualities for a set of tests.
23
24To achieve this, :mod:`unittest` supports some important concepts:
25
26test fixture
27 A :dfn:`test fixture` represents the preparation needed to perform one or more
28 tests, and any associate cleanup actions. This may involve, for example,
29 creating temporary or proxy databases, directories, or starting a server
30 process.
31
32test case
33 A :dfn:`test case` is the smallest unit of testing. It checks for a specific
34 response to a particular set of inputs. :mod:`unittest` provides a base class,
35 :class:`TestCase`, which may be used to create new test cases.
36
37test suite
38 A :dfn:`test suite` is a collection of test cases, test suites, or both. It is
39 used to aggregate tests that should be executed together.
40
41test runner
42 A :dfn:`test runner` is a component which orchestrates the execution of tests
43 and provides the outcome to the user. The runner may use a graphical interface,
44 a textual interface, or return a special value to indicate the results of
45 executing the tests.
46
47The test case and test fixture concepts are supported through the
48:class:`TestCase` and :class:`FunctionTestCase` classes; the former should be
49used when creating new tests, and the latter can be used when integrating
50existing test code with a :mod:`unittest`\ -driven framework. When building test
Benjamin Peterson99721e02009-03-23 23:10:14 +000051fixtures using :class:`TestCase`, the :meth:`~TestCase.setUp` and
52:meth:`~TestCase.tearDown` methods can be overridden to provide initialization
53and cleanup for the fixture. With :class:`FunctionTestCase`, existing functions
54can be passed to the constructor for these purposes. When the test is run, the
55fixture initialization is run first; if it succeeds, the cleanup method is run
56after the test has been executed, regardless of the outcome of the test. Each
57instance of the :class:`TestCase` will only be used to run a single test method,
58so a new fixture is created for each test.
Georg Brandl8ec7f652007-08-15 14:28:01 +000059
60Test suites are implemented by the :class:`TestSuite` class. This class allows
61individual tests and test suites to be aggregated; when the suite is executed,
Benjamin Peterson176a56c2009-05-25 00:48:58 +000062all tests added directly to the suite and in "child" test suites are run.
Georg Brandl8ec7f652007-08-15 14:28:01 +000063
Benjamin Peterson99721e02009-03-23 23:10:14 +000064A test runner is an object that provides a single method,
65:meth:`~TestRunner.run`, which accepts a :class:`TestCase` or :class:`TestSuite`
66object as a parameter, and returns a result object. The class
67:class:`TestResult` is provided for use as the result object. :mod:`unittest`
68provides the :class:`TextTestRunner` as an example test runner which reports
69test results on the standard error stream by default. Alternate runners can be
70implemented for other environments (such as graphical environments) without any
71need to derive from a specific class.
Georg Brandl8ec7f652007-08-15 14:28:01 +000072
73
74.. seealso::
75
76 Module :mod:`doctest`
77 Another test-support module with a very different flavor.
78
Georg Brandld198b762009-05-31 14:15:25 +000079 `Simple Smalltalk Testing: With Patterns <http://www.XProgramming.com/testfram.htm>`_
Georg Brandl2fcd1732009-05-30 10:45:40 +000080 Kent Beck's original paper on testing frameworks using the pattern shared
81 by :mod:`unittest`.
Georg Brandl8ec7f652007-08-15 14:28:01 +000082
Georg Brandld198b762009-05-31 14:15:25 +000083 `Nose <http://code.google.com/p/python-nose/>`_ and `py.test <http://pytest.org>`_
Georg Brandl2fcd1732009-05-30 10:45:40 +000084 Third-party unittest frameworks with a lighter-weight syntax for writing
85 tests. For example, ``assert func(10) == 42``.
Raymond Hettinger21b617b2009-03-24 00:17:11 +000086
Georg Brandld198b762009-05-31 14:15:25 +000087 `python-mock <http://python-mock.sourceforge.net/>`_ and `minimock <http://blog.ianbicking.org/minimock.html>`_
Georg Brandl2fcd1732009-05-30 10:45:40 +000088 Tools for creating mock test objects (objects simulating external
89 resources).
Georg Brandl8ec7f652007-08-15 14:28:01 +000090
Michael Foordb4a81c82009-05-29 20:33:46 +000091
Michael Foordb4a81c82009-05-29 20:33:46 +000092
Michael Foord5d31e052009-05-11 17:59:43 +000093
Georg Brandl8ec7f652007-08-15 14:28:01 +000094.. _unittest-minimal-example:
95
96Basic example
97-------------
98
99The :mod:`unittest` module provides a rich set of tools for constructing and
100running tests. This section demonstrates that a small subset of the tools
101suffice to meet the needs of most users.
102
103Here is a short script to test three functions from the :mod:`random` module::
104
105 import random
106 import unittest
107
108 class TestSequenceFunctions(unittest.TestCase):
109
110 def setUp(self):
111 self.seq = range(10)
112
Benjamin Peterson99721e02009-03-23 23:10:14 +0000113 def test_shuffle(self):
Georg Brandl8ec7f652007-08-15 14:28:01 +0000114 # make sure the shuffled sequence does not lose any elements
115 random.shuffle(self.seq)
116 self.seq.sort()
117 self.assertEqual(self.seq, range(10))
118
Benjamin Peterson99721e02009-03-23 23:10:14 +0000119 def test_choice(self):
Georg Brandl8ec7f652007-08-15 14:28:01 +0000120 element = random.choice(self.seq)
Ezio Melotti85ee3e12010-02-04 20:06:38 +0000121 self.assertIn(element, self.seq)
Georg Brandl8ec7f652007-08-15 14:28:01 +0000122
Benjamin Peterson99721e02009-03-23 23:10:14 +0000123 def test_sample(self):
Georg Brandl8ec7f652007-08-15 14:28:01 +0000124 self.assertRaises(ValueError, random.sample, self.seq, 20)
125 for element in random.sample(self.seq, 5):
Ezio Melotti85ee3e12010-02-04 20:06:38 +0000126 self.assertIn(element, self.seq)
Georg Brandl8ec7f652007-08-15 14:28:01 +0000127
128 if __name__ == '__main__':
129 unittest.main()
130
Benjamin Peterson99721e02009-03-23 23:10:14 +0000131A testcase is created by subclassing :class:`unittest.TestCase`. The three
Georg Brandl8ec7f652007-08-15 14:28:01 +0000132individual tests are defined with methods whose names start with the letters
133``test``. This naming convention informs the test runner about which methods
134represent tests.
135
Benjamin Peterson99721e02009-03-23 23:10:14 +0000136The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
137expected result; :meth:`~TestCase.assert_` to verify a condition; or
138:meth:`~TestCase.assertRaises` to verify that an expected exception gets raised.
139These methods are used instead of the :keyword:`assert` statement so the test
140runner can accumulate all test results and produce a report.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000141
Benjamin Peterson99721e02009-03-23 23:10:14 +0000142When a :meth:`~TestCase.setUp` method is defined, the test runner will run that
143method prior to each test. Likewise, if a :meth:`~TestCase.tearDown` method is
144defined, the test runner will invoke that method after each test. In the
145example, :meth:`~TestCase.setUp` was used to create a fresh sequence for each
146test.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000147
148The final block shows a simple way to run the tests. :func:`unittest.main`
149provides a command line interface to the test script. When run from the command
150line, the above script produces an output that looks like this::
151
152 ...
153 ----------------------------------------------------------------------
154 Ran 3 tests in 0.000s
155
156 OK
157
158Instead of :func:`unittest.main`, there are other ways to run the tests with a
159finer level of control, less terse output, and no requirement to be run from the
160command line. For example, the last two lines may be replaced with::
161
162 suite = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)
163 unittest.TextTestRunner(verbosity=2).run(suite)
164
165Running the revised script from the interpreter or another script produces the
166following output::
167
168 testchoice (__main__.TestSequenceFunctions) ... ok
169 testsample (__main__.TestSequenceFunctions) ... ok
170 testshuffle (__main__.TestSequenceFunctions) ... ok
171
172 ----------------------------------------------------------------------
173 Ran 3 tests in 0.110s
174
175 OK
176
177The above examples show the most commonly used :mod:`unittest` features which
178are sufficient to meet many everyday testing needs. The remainder of the
179documentation explores the full feature set from first principles.
180
Raymond Hettingerb09f1982009-05-29 21:20:41 +0000181
182.. _unittest-command-line-interface:
183
184Command Line Interface
185----------------------
186
187The unittest module can be used from the command line to run tests from
188modules, classes or even individual test methods::
189
190 python -m unittest test_module1 test_module2
191 python -m unittest test_module.TestClass
192 python -m unittest test_module.TestClass.test_method
193
194You can pass in a list with any combination of module names, and fully
195qualified class or method names.
196
197You can run tests with more detail (higher verbosity) by passing in the -v flag::
198
Ezio Melotti062d2b52009-12-19 22:41:49 +0000199 python -m unittest -v test_module
Raymond Hettingerb09f1982009-05-29 21:20:41 +0000200
201For a list of all the command line options::
202
203 python -m unittest -h
204
205.. versionchanged:: 2.7
206 In earlier versions it was only possible to run individual test methods and
207 not modules or classes.
208
209The command line can also be used for test discovery, for running all of the
210tests in a project or just a subset.
211
212
213.. _unittest-test-discovery:
214
215Test Discovery
216--------------
217
218.. versionadded:: 2.7
219
220Unittest supports simple test discovery. For a project's tests to be
221compatible with test discovery they must all be importable from the top level
222directory of the project (in other words, they must all be in Python packages).
223
224Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
225used from the command line. The basic command line usage is::
226
227 cd project_directory
228 python -m unittest discover
229
230The ``discover`` sub-command has the following options:
231
232 -v, --verbose Verbose output
233 -s directory Directory to start discovery ('.' default)
234 -p pattern Pattern to match test files ('test*.py' default)
235 -t directory Top level directory of project (default to
236 start directory)
237
238The -s, -p, & -t options can be passsed in as positional arguments. The
239following two command lines are equivalent::
240
Ezio Melotti7b4e02c2010-01-27 20:25:11 +0000241 python -m unittest discover -s project_directory -p '*_test.py'
242 python -m unittest discover project_directory '*_test.py'
Raymond Hettingerb09f1982009-05-29 21:20:41 +0000243
244Test modules and packages can customize test loading and discovery by through
245the `load_tests protocol`_.
246
247
Georg Brandl8ec7f652007-08-15 14:28:01 +0000248.. _organizing-tests:
249
250Organizing test code
251--------------------
252
253The basic building blocks of unit testing are :dfn:`test cases` --- single
254scenarios that must be set up and checked for correctness. In :mod:`unittest`,
255test cases are represented by instances of :mod:`unittest`'s :class:`TestCase`
256class. To make your own test cases you must write subclasses of
257:class:`TestCase`, or use :class:`FunctionTestCase`.
258
259An instance of a :class:`TestCase`\ -derived class is an object that can
260completely run a single test method, together with optional set-up and tidy-up
261code.
262
263The testing code of a :class:`TestCase` instance should be entirely self
264contained, such that it can be run either in isolation or in arbitrary
265combination with any number of other test cases.
266
Benjamin Peterson99721e02009-03-23 23:10:14 +0000267The simplest :class:`TestCase` subclass will simply override the
268:meth:`~TestCase.runTest` method in order to perform specific testing code::
Georg Brandl8ec7f652007-08-15 14:28:01 +0000269
270 import unittest
271
272 class DefaultWidgetSizeTestCase(unittest.TestCase):
273 def runTest(self):
274 widget = Widget('The widget')
275 self.assertEqual(widget.size(), (50, 50), 'incorrect default size')
276
Gregory P. Smith28399852009-03-31 16:54:10 +0000277Note that in order to test something, we use the one of the :meth:`assert\*`
Georg Brandl2fcd1732009-05-30 10:45:40 +0000278methods provided by the :class:`TestCase` base class. If the test fails, an
279exception will be raised, and :mod:`unittest` will identify the test case as a
280:dfn:`failure`. Any other exceptions will be treated as :dfn:`errors`. This
281helps you identify where the problem is: :dfn:`failures` are caused by incorrect
282results - a 5 where you expected a 6. :dfn:`Errors` are caused by incorrect
283code - e.g., a :exc:`TypeError` caused by an incorrect function call.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000284
285The way to run a test case will be described later. For now, note that to
286construct an instance of such a test case, we call its constructor without
287arguments::
288
289 testCase = DefaultWidgetSizeTestCase()
290
291Now, such test cases can be numerous, and their set-up can be repetitive. In
292the above case, constructing a :class:`Widget` in each of 100 Widget test case
293subclasses would mean unsightly duplication.
294
295Luckily, we can factor out such set-up code by implementing a method called
Benjamin Peterson99721e02009-03-23 23:10:14 +0000296:meth:`~TestCase.setUp`, which the testing framework will automatically call for
297us when we run the test::
Georg Brandl8ec7f652007-08-15 14:28:01 +0000298
299 import unittest
300
301 class SimpleWidgetTestCase(unittest.TestCase):
302 def setUp(self):
303 self.widget = Widget('The widget')
304
305 class DefaultWidgetSizeTestCase(SimpleWidgetTestCase):
306 def runTest(self):
Ezio Melotti85ee3e12010-02-04 20:06:38 +0000307 self.assertEqual(self.widget.size(), (50,50),
308 'incorrect default size')
Georg Brandl8ec7f652007-08-15 14:28:01 +0000309
310 class WidgetResizeTestCase(SimpleWidgetTestCase):
311 def runTest(self):
312 self.widget.resize(100,150)
Ezio Melotti85ee3e12010-02-04 20:06:38 +0000313 self.assertEqual(self.widget.size(), (100,150),
314 'wrong size after resize')
Georg Brandl8ec7f652007-08-15 14:28:01 +0000315
Benjamin Peterson99721e02009-03-23 23:10:14 +0000316If the :meth:`~TestCase.setUp` method raises an exception while the test is
317running, the framework will consider the test to have suffered an error, and the
318:meth:`~TestCase.runTest` method will not be executed.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000319
Benjamin Peterson99721e02009-03-23 23:10:14 +0000320Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
321after the :meth:`~TestCase.runTest` method has been run::
Georg Brandl8ec7f652007-08-15 14:28:01 +0000322
323 import unittest
324
325 class SimpleWidgetTestCase(unittest.TestCase):
326 def setUp(self):
327 self.widget = Widget('The widget')
328
329 def tearDown(self):
330 self.widget.dispose()
331 self.widget = None
332
Benjamin Peterson99721e02009-03-23 23:10:14 +0000333If :meth:`~TestCase.setUp` succeeded, the :meth:`~TestCase.tearDown` method will
334be run whether :meth:`~TestCase.runTest` succeeded or not.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000335
336Such a working environment for the testing code is called a :dfn:`fixture`.
337
338Often, many small test cases will use the same fixture. In this case, we would
339end up subclassing :class:`SimpleWidgetTestCase` into many small one-method
340classes such as :class:`DefaultWidgetSizeTestCase`. This is time-consuming and
Georg Brandl8ec7f652007-08-15 14:28:01 +0000341discouraging, so in the same vein as JUnit, :mod:`unittest` provides a simpler
342mechanism::
343
344 import unittest
345
346 class WidgetTestCase(unittest.TestCase):
347 def setUp(self):
348 self.widget = Widget('The widget')
349
350 def tearDown(self):
351 self.widget.dispose()
352 self.widget = None
353
354 def testDefaultSize(self):
Ezio Melotti85ee3e12010-02-04 20:06:38 +0000355 self.assertEqual(self.widget.size(), (50,50),
356 'incorrect default size')
Georg Brandl8ec7f652007-08-15 14:28:01 +0000357
358 def testResize(self):
359 self.widget.resize(100,150)
Ezio Melotti85ee3e12010-02-04 20:06:38 +0000360 self.assertEqual(self.widget.size(), (100,150),
361 'wrong size after resize')
Georg Brandl8ec7f652007-08-15 14:28:01 +0000362
Benjamin Peterson99721e02009-03-23 23:10:14 +0000363Here we have not provided a :meth:`~TestCase.runTest` method, but have instead
364provided two different test methods. Class instances will now each run one of
365the :meth:`test\*` methods, with ``self.widget`` created and destroyed
366separately for each instance. When creating an instance we must specify the
367test method it is to run. We do this by passing the method name in the
368constructor::
Georg Brandl8ec7f652007-08-15 14:28:01 +0000369
370 defaultSizeTestCase = WidgetTestCase('testDefaultSize')
371 resizeTestCase = WidgetTestCase('testResize')
372
373Test case instances are grouped together according to the features they test.
374:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
375represented by :mod:`unittest`'s :class:`TestSuite` class::
376
377 widgetTestSuite = unittest.TestSuite()
378 widgetTestSuite.addTest(WidgetTestCase('testDefaultSize'))
379 widgetTestSuite.addTest(WidgetTestCase('testResize'))
380
381For the ease of running tests, as we will see later, it is a good idea to
382provide in each test module a callable object that returns a pre-built test
383suite::
384
385 def suite():
386 suite = unittest.TestSuite()
387 suite.addTest(WidgetTestCase('testDefaultSize'))
388 suite.addTest(WidgetTestCase('testResize'))
389 return suite
390
391or even::
392
393 def suite():
394 tests = ['testDefaultSize', 'testResize']
395
396 return unittest.TestSuite(map(WidgetTestCase, tests))
397
398Since it is a common pattern to create a :class:`TestCase` subclass with many
399similarly named test functions, :mod:`unittest` provides a :class:`TestLoader`
400class that can be used to automate the process of creating a test suite and
401populating it with individual tests. For example, ::
402
403 suite = unittest.TestLoader().loadTestsFromTestCase(WidgetTestCase)
404
405will create a test suite that will run ``WidgetTestCase.testDefaultSize()`` and
406``WidgetTestCase.testResize``. :class:`TestLoader` uses the ``'test'`` method
407name prefix to identify test methods automatically.
408
409Note that the order in which the various test cases will be run is determined by
410sorting the test function names with the built-in :func:`cmp` function.
411
412Often it is desirable to group suites of test cases together, so as to run tests
413for the whole system at once. This is easy, since :class:`TestSuite` instances
414can be added to a :class:`TestSuite` just as :class:`TestCase` instances can be
415added to a :class:`TestSuite`::
416
417 suite1 = module1.TheTestSuite()
418 suite2 = module2.TheTestSuite()
419 alltests = unittest.TestSuite([suite1, suite2])
420
421You can place the definitions of test cases and test suites in the same modules
422as the code they are to test (such as :file:`widget.py`), but there are several
423advantages to placing the test code in a separate module, such as
424:file:`test_widget.py`:
425
426* The test module can be run standalone from the command line.
427
428* The test code can more easily be separated from shipped code.
429
430* There is less temptation to change test code to fit the code it tests without
431 a good reason.
432
433* Test code should be modified much less frequently than the code it tests.
434
435* Tested code can be refactored more easily.
436
437* Tests for modules written in C must be in separate modules anyway, so why not
438 be consistent?
439
440* If the testing strategy changes, there is no need to change the source code.
441
442
443.. _legacy-unit-tests:
444
445Re-using old test code
446----------------------
447
448Some users will find that they have existing test code that they would like to
449run from :mod:`unittest`, without converting every old test function to a
450:class:`TestCase` subclass.
451
452For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
453This subclass of :class:`TestCase` can be used to wrap an existing test
454function. Set-up and tear-down functions can also be provided.
455
456Given the following test function::
457
458 def testSomething():
459 something = makeSomething()
460 assert something.name is not None
461 # ...
462
463one can create an equivalent test case instance as follows::
464
465 testcase = unittest.FunctionTestCase(testSomething)
466
467If there are additional set-up and tear-down methods that should be called as
468part of the test case's operation, they can also be provided like so::
469
470 testcase = unittest.FunctionTestCase(testSomething,
471 setUp=makeSomethingDB,
472 tearDown=deleteSomethingDB)
473
474To make migrating existing test suites easier, :mod:`unittest` supports tests
475raising :exc:`AssertionError` to indicate test failure. However, it is
476recommended that you use the explicit :meth:`TestCase.fail\*` and
477:meth:`TestCase.assert\*` methods instead, as future versions of :mod:`unittest`
478may treat :exc:`AssertionError` differently.
479
480.. note::
481
Georg Brandl2fcd1732009-05-30 10:45:40 +0000482 Even though :class:`FunctionTestCase` can be used to quickly convert an
483 existing test base over to a :mod:`unittest`\ -based system, this approach is
484 not recommended. Taking the time to set up proper :class:`TestCase`
485 subclasses will make future test refactorings infinitely easier.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000486
Benjamin Peterson99721e02009-03-23 23:10:14 +0000487In some cases, the existing tests may have been written using the :mod:`doctest`
488module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
489automatically build :class:`unittest.TestSuite` instances from the existing
490:mod:`doctest`\ -based tests.
491
Georg Brandl8ec7f652007-08-15 14:28:01 +0000492
Benjamin Peterson692428e2009-03-23 21:50:21 +0000493.. _unittest-skipping:
494
495Skipping tests and expected failures
496------------------------------------
497
Michael Foordfb0844b2010-02-05 21:45:12 +0000498.. versionadded:: 2.7
499
Benjamin Peterson692428e2009-03-23 21:50:21 +0000500Unittest supports skipping individual test methods and even whole classes of
501tests. In addition, it supports marking a test as a "expected failure," a test
502that is broken and will fail, but shouldn't be counted as a failure on a
503:class:`TestResult`.
504
505Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
506or one of its conditional variants.
507
508Basic skipping looks like this: ::
509
510 class MyTestCase(unittest.TestCase):
511
512 @unittest.skip("demonstrating skipping")
513 def test_nothing(self):
514 self.fail("shouldn't happen")
515
Georg Brandl2fcd1732009-05-30 10:45:40 +0000516 @unittest.skipIf(mylib.__version__ < (1, 3),
517 "not supported in this library version")
Benjamin Petersonbe76d4c2009-03-29 03:16:57 +0000518 def test_format(self):
Benjamin Peterson097aafd2009-03-29 03:39:58 +0000519 # Tests that work for only a certain version of the library.
Benjamin Petersonbe76d4c2009-03-29 03:16:57 +0000520 pass
521
522 @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
523 def test_windows_support(self):
524 # windows specific testing code
525 pass
526
Benjamin Peterson692428e2009-03-23 21:50:21 +0000527This is the output of running the example above in verbose mode: ::
528
Benjamin Peterson097aafd2009-03-29 03:39:58 +0000529 test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
Benjamin Peterson692428e2009-03-23 21:50:21 +0000530 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
Benjamin Petersonbe76d4c2009-03-29 03:16:57 +0000531 test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
Benjamin Peterson692428e2009-03-23 21:50:21 +0000532
533 ----------------------------------------------------------------------
Benjamin Petersonbe76d4c2009-03-29 03:16:57 +0000534 Ran 3 tests in 0.005s
535
536 OK (skipped=3)
Benjamin Peterson692428e2009-03-23 21:50:21 +0000537
538Classes can be skipped just like methods: ::
539
540 @skip("showing class skipping")
541 class MySkippedTestCase(unittest.TestCase):
542 def test_not_run(self):
543 pass
544
Benjamin Peterson31b78062009-03-23 23:13:36 +0000545:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
546that needs to be set up is not available.
547
Benjamin Peterson692428e2009-03-23 21:50:21 +0000548Expected failures use the :func:`expectedFailure` decorator. ::
549
550 class ExpectedFailureTestCase(unittest.TestCase):
551 @unittest.expectedFailure
552 def test_fail(self):
553 self.assertEqual(1, 0, "broken")
554
555It's easy to roll your own skipping decorators by making a decorator that calls
556:func:`skip` on the test when it wants it to be skipped. This decorator skips
557the test unless the passed object has a certain attribute: ::
558
559 def skipUnlessHasattr(obj, attr):
560 if hasattr(obj, attr):
561 return lambda func: func
562 return unittest.skip("{0!r} doesn't have {1!r}".format(obj, attr))
563
564The following decorators implement test skipping and expected failures:
565
566.. function:: skip(reason)
567
568 Unconditionally skip the decorated test. *reason* should describe why the
569 test is being skipped.
570
571.. function:: skipIf(condition, reason)
572
573 Skip the decorated test if *condition* is true.
574
575.. function:: skipUnless(condition, reason)
576
577 Skip the decoratored test unless *condition* is true.
578
579.. function:: expectedFailure
580
581 Mark the test as an expected failure. If the test fails when run, the test
582 is not counted as a failure.
583
584
Georg Brandl8ec7f652007-08-15 14:28:01 +0000585.. _unittest-contents:
586
587Classes and functions
588---------------------
589
Benjamin Peterson99721e02009-03-23 23:10:14 +0000590This section describes in depth the API of :mod:`unittest`.
591
592
593.. _testcase-objects:
594
595Test cases
596~~~~~~~~~~
Georg Brandl8ec7f652007-08-15 14:28:01 +0000597
598.. class:: TestCase([methodName])
599
600 Instances of the :class:`TestCase` class represent the smallest testable units
601 in the :mod:`unittest` universe. This class is intended to be used as a base
602 class, with specific tests being implemented by concrete subclasses. This class
603 implements the interface needed by the test runner to allow it to drive the
604 test, and methods that the test code can use to check for and report various
605 kinds of failure.
606
607 Each instance of :class:`TestCase` will run a single test method: the method
608 named *methodName*. If you remember, we had an earlier example that went
609 something like this::
610
611 def suite():
612 suite = unittest.TestSuite()
613 suite.addTest(WidgetTestCase('testDefaultSize'))
614 suite.addTest(WidgetTestCase('testResize'))
615 return suite
616
617 Here, we create two instances of :class:`WidgetTestCase`, each of which runs a
618 single test.
619
Benjamin Peterson99721e02009-03-23 23:10:14 +0000620 *methodName* defaults to :meth:`runTest`.
621
622 :class:`TestCase` instances provide three groups of methods: one group used
623 to run the test, another used by the test implementation to check conditions
624 and report failures, and some inquiry methods allowing information about the
625 test itself to be gathered.
626
627 Methods in the first group (running the test) are:
628
629
630 .. method:: setUp()
631
632 Method called to prepare the test fixture. This is called immediately
633 before calling the test method; any exception raised by this method will
634 be considered an error rather than a test failure. The default
635 implementation does nothing.
636
637
638 .. method:: tearDown()
639
640 Method called immediately after the test method has been called and the
641 result recorded. This is called even if the test method raised an
642 exception, so the implementation in subclasses may need to be particularly
643 careful about checking internal state. Any exception raised by this
644 method will be considered an error rather than a test failure. This
645 method will only be called if the :meth:`setUp` succeeds, regardless of
646 the outcome of the test method. The default implementation does nothing.
647
648
649 .. method:: run([result])
650
651 Run the test, collecting the result into the test result object passed as
652 *result*. If *result* is omitted or :const:`None`, a temporary result
Ezio Melottic2f5a592009-06-30 22:51:06 +0000653 object is created (by calling the :meth:`defaultTestResult` method) and
654 used. The result object is not returned to :meth:`run`'s caller.
Benjamin Peterson99721e02009-03-23 23:10:14 +0000655
656 The same effect may be had by simply calling the :class:`TestCase`
657 instance.
658
659
Benjamin Peterson47d97382009-03-26 20:05:50 +0000660 .. method:: skipTest(reason)
Benjamin Peterson99721e02009-03-23 23:10:14 +0000661
Benjamin Peterson31b78062009-03-23 23:13:36 +0000662 Calling this during the a test method or :meth:`setUp` skips the current
663 test. See :ref:`unittest-skipping` for more information.
Benjamin Peterson99721e02009-03-23 23:10:14 +0000664
665
666 .. method:: debug()
667
668 Run the test without collecting the result. This allows exceptions raised
669 by the test to be propagated to the caller, and can be used to support
670 running tests under a debugger.
671
672 The test code can use any of the following methods to check for and report
673 failures.
674
675
Gregory P. Smith28399852009-03-31 16:54:10 +0000676 .. method:: assertTrue(expr[, msg])
677 assert_(expr[, msg])
Benjamin Peterson99721e02009-03-23 23:10:14 +0000678 failUnless(expr[, msg])
Benjamin Peterson99721e02009-03-23 23:10:14 +0000679
Georg Brandl64034bb2009-04-25 14:51:31 +0000680 Signal a test failure if *expr* is false; the explanation for the failure
Benjamin Peterson99721e02009-03-23 23:10:14 +0000681 will be *msg* if given, otherwise it will be :const:`None`.
682
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000683 .. deprecated:: 2.7
Georg Brandl1c7c7302010-02-06 10:08:21 +0000684 :meth:`failUnless`; use one of the ``assert`` variants.
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000685
Benjamin Peterson99721e02009-03-23 23:10:14 +0000686
687 .. method:: assertEqual(first, second[, msg])
688 failUnlessEqual(first, second[, msg])
689
690 Test that *first* and *second* are equal. If the values do not compare
691 equal, the test will fail with the explanation given by *msg*, or
Gregory P. Smith28399852009-03-31 16:54:10 +0000692 :const:`None`. Note that using :meth:`assertEqual` improves upon
693 doing the comparison as the first parameter to :meth:`assertTrue`: the
694 default value for *msg* include representations of both *first* and
695 *second*.
696
697 In addition, if *first* and *second* are the exact same type and one of
Michael Foordfe6349c2010-02-08 22:41:16 +0000698 list, tuple, dict, set, frozenset or unicode or any type that a subclass
Gregory P. Smith28399852009-03-31 16:54:10 +0000699 registers :meth:`addTypeEqualityFunc` the type specific equality function
700 will be called in order to generate a more useful default error message.
701
702 .. versionchanged:: 2.7
703 Added the automatic calling of type specific equality function.
Benjamin Peterson99721e02009-03-23 23:10:14 +0000704
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000705 .. deprecated:: 2.7
Georg Brandl1c7c7302010-02-06 10:08:21 +0000706 :meth:`failUnlessEqual`; use :meth:`assertEqual`.
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000707
Benjamin Peterson99721e02009-03-23 23:10:14 +0000708
709 .. method:: assertNotEqual(first, second[, msg])
710 failIfEqual(first, second[, msg])
711
712 Test that *first* and *second* are not equal. If the values do compare
713 equal, the test will fail with the explanation given by *msg*, or
Gregory P. Smith28399852009-03-31 16:54:10 +0000714 :const:`None`. Note that using :meth:`assertNotEqual` improves upon doing
715 the comparison as the first parameter to :meth:`assertTrue` is that the
Benjamin Peterson99721e02009-03-23 23:10:14 +0000716 default value for *msg* can be computed to include representations of both
717 *first* and *second*.
718
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000719 .. deprecated:: 2.7
Georg Brandl1c7c7302010-02-06 10:08:21 +0000720 :meth:`failIfEqual`; use :meth:`assertNotEqual`.
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000721
Benjamin Peterson99721e02009-03-23 23:10:14 +0000722
723 .. method:: assertAlmostEqual(first, second[, places[, msg]])
724 failUnlessAlmostEqual(first, second[, places[, msg]])
725
726 Test that *first* and *second* are approximately equal by computing the
727 difference, rounding to the given number of decimal *places* (default 7),
728 and comparing to zero.
729
730 Note that comparing a given number of decimal places is not the same as
731 comparing a given number of significant digits. If the values do not
732 compare equal, the test will fail with the explanation given by *msg*, or
733 :const:`None`.
734
Michael Foordc3f79372009-09-13 16:40:02 +0000735 .. versionchanged:: 2.7
736 Objects that compare equal are automatically almost equal.
737
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000738 .. deprecated:: 2.7
Georg Brandl1c7c7302010-02-06 10:08:21 +0000739 :meth:`failUnlessAlmostEqual`; use :meth:`assertAlmostEqual`.
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000740
Benjamin Peterson99721e02009-03-23 23:10:14 +0000741
742 .. method:: assertNotAlmostEqual(first, second[, places[, msg]])
743 failIfAlmostEqual(first, second[, places[, msg]])
744
745 Test that *first* and *second* are not approximately equal by computing
746 the difference, rounding to the given number of decimal *places* (default
747 7), and comparing to zero.
748
749 Note that comparing a given number of decimal places is not the same as
750 comparing a given number of significant digits. If the values do not
751 compare equal, the test will fail with the explanation given by *msg*, or
752 :const:`None`.
753
Michael Foordc3f79372009-09-13 16:40:02 +0000754 .. versionchanged:: 2.7
755 Objects that compare equal automatically fail.
756
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000757 .. deprecated:: 2.7
Georg Brandl1c7c7302010-02-06 10:08:21 +0000758 :meth:`failIfAlmostEqual`; use :meth:`assertNotAlmostEqual`.
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000759
Benjamin Peterson99721e02009-03-23 23:10:14 +0000760
Gregory P. Smith28399852009-03-31 16:54:10 +0000761 .. method:: assertGreater(first, second, msg=None)
762 assertGreaterEqual(first, second, msg=None)
763 assertLess(first, second, msg=None)
764 assertLessEqual(first, second, msg=None)
765
766 Test that *first* is respectively >, >=, < or <= than *second* depending
Andrew M. Kuchling59631852009-04-09 11:23:36 +0000767 on the method name. If not, the test will fail with an explanation
Gregory P. Smith28399852009-03-31 16:54:10 +0000768 or with the explanation given by *msg*::
769
770 >>> self.assertGreaterEqual(3, 4)
771 AssertionError: "3" unexpectedly not greater than or equal to "4"
772
773 .. versionadded:: 2.7
774
775
776 .. method:: assertMultiLineEqual(self, first, second, msg=None)
777
778 Test that the multiline string *first* is equal to the string *second*.
779 When not equal a diff of the two strings highlighting the differences
Michael Foordfe6349c2010-02-08 22:41:16 +0000780 will be included in the error message. This method is used by default
781 when comparing Unicode strings with :meth:`assertEqual`.
Gregory P. Smith28399852009-03-31 16:54:10 +0000782
783 If specified *msg* will be used as the error message on failure.
784
785 .. versionadded:: 2.7
786
787
Ezio Melotti5afe42b2010-01-16 19:36:42 +0000788 .. method:: assertRegexpMatches(text, regexp, msg=None)
Gregory P. Smith28399852009-03-31 16:54:10 +0000789
790 Verifies that a *regexp* search matches *text*. Fails with an error
791 message including the pattern and the *text*. *regexp* may be
792 a regular expression object or a string containing a regular expression
793 suitable for use by :func:`re.search`.
794
795 .. versionadded:: 2.7
796
797
798 .. method:: assertIn(first, second, msg=None)
799 assertNotIn(first, second, msg=None)
800
Andrew M. Kuchling59631852009-04-09 11:23:36 +0000801 Tests that *first* is or is not in *second* with an explanatory error
Gregory P. Smith28399852009-03-31 16:54:10 +0000802 message as appropriate.
803
804 If specified *msg* will be used as the error message on failure.
805
806 .. versionadded:: 2.7
807
808
Michael Foord1c430012010-02-05 20:52:14 +0000809 .. method:: assertSameElements(actual, expected, msg=None)
Gregory P. Smith28399852009-03-31 16:54:10 +0000810
Michael Foorde70c72c2010-01-31 19:59:26 +0000811 Test that sequence *expected* contains the same elements as *actual*,
812 regardless of their order. When they don't, an error message listing
813 the differences between the sequences will be generated.
Gregory P. Smith28399852009-03-31 16:54:10 +0000814
Michael Foord1c430012010-02-05 20:52:14 +0000815 Duplicate elements are ignored when comparing *actual* and *expected*.
816 It is the equivalent of ``assertEqual(set(expected), set(actual))``
817 but it works with sequences of unhashable objects as well.
818
Gregory P. Smith28399852009-03-31 16:54:10 +0000819 If specified *msg* will be used as the error message on failure.
820
821 .. versionadded:: 2.7
822
823
824 .. method:: assertSetEqual(set1, set2, msg=None)
825
826 Tests that two sets are equal. If not, an error message is constructed
Michael Foordfe6349c2010-02-08 22:41:16 +0000827 that lists the differences between the sets. This method is used by
828 default when comparing sets or frozensets with :meth:`assertEqual`.
Gregory P. Smith28399852009-03-31 16:54:10 +0000829
830 Fails if either of *set1* or *set2* does not have a :meth:`set.difference`
831 method.
832
833 If specified *msg* will be used as the error message on failure.
834
835 .. versionadded:: 2.7
836
837
838 .. method:: assertDictEqual(expected, actual, msg=None)
839
840 Test that two dictionaries are equal. If not, an error message is
Michael Foordfe6349c2010-02-08 22:41:16 +0000841 constructed that shows the differences in the dictionaries. This
842 method will be used by default to compare dictionaries in
843 calls to :meth:`assertEqual`.
Gregory P. Smith28399852009-03-31 16:54:10 +0000844
845 If specified *msg* will be used as the error message on failure.
846
847 .. versionadded:: 2.7
848
849
850 .. method:: assertDictContainsSubset(expected, actual, msg=None)
851
Andrew M. Kuchling59631852009-04-09 11:23:36 +0000852 Tests whether the key/value pairs in dictionary *actual* are a
Gregory P. Smith28399852009-03-31 16:54:10 +0000853 superset of those in *expected*. If not, an error message listing
854 the missing keys and mismatched values is generated.
855
856 If specified *msg* will be used as the error message on failure.
857
858 .. versionadded:: 2.7
859
860
861 .. method:: assertListEqual(list1, list2, msg=None)
862 assertTupleEqual(tuple1, tuple2, msg=None)
863
864 Tests that two lists or tuples are equal. If not an error message is
865 constructed that shows only the differences between the two. An error
866 is also raised if either of the parameters are of the wrong type.
Michael Foordfe6349c2010-02-08 22:41:16 +0000867 These methods are used by default when comparing lists or tuples with
868 :meth:`assertEqual`.
Gregory P. Smith28399852009-03-31 16:54:10 +0000869
870 If specified *msg* will be used as the error message on failure.
871
872 .. versionadded:: 2.7
873
874
875 .. method:: assertSequenceEqual(seq1, seq2, msg=None, seq_type=None)
876
877 Tests that two sequences are equal. If a *seq_type* is supplied, both
878 *seq1* and *seq2* must be instances of *seq_type* or a failure will
879 be raised. If the sequences are different an error message is
880 constructed that shows the difference between the two.
881
882 If specified *msg* will be used as the error message on failure.
883
884 This method is used to implement :meth:`assertListEqual` and
885 :meth:`assertTupleEqual`.
886
887 .. versionadded:: 2.7
888
889
Benjamin Peterson99721e02009-03-23 23:10:14 +0000890 .. method:: assertRaises(exception[, callable, ...])
891 failUnlessRaises(exception[, callable, ...])
892
893 Test that an exception is raised when *callable* is called with any
894 positional or keyword arguments that are also passed to
895 :meth:`assertRaises`. The test passes if *exception* is raised, is an
896 error if another exception is raised, or fails if no exception is raised.
897 To catch any of a group of exceptions, a tuple containing the exception
898 classes may be passed as *exception*.
899
Benjamin Peterson7233acc2009-03-29 03:31:40 +0000900 If *callable* is omitted or None, returns a context manager so that the
901 code under test can be written inline rather than as a function::
902
Michael Foord1f3fa8a2010-02-05 21:07:38 +0000903 with self.assertRaises(SomeException):
Benjamin Peterson7233acc2009-03-29 03:31:40 +0000904 do_something()
905
Kristján Valur Jónssone2a77982009-08-27 22:20:21 +0000906 The context manager will store the caught exception object in its
Georg Brandldc3694b2010-02-07 17:02:22 +0000907 :attr:`exception` attribute. This can be useful if the intention
Michael Foord1f3fa8a2010-02-05 21:07:38 +0000908 is to perform additional checks on the exception raised::
909
910 with self.assertRaises(SomeException) as cm:
911 do_something()
912
Georg Brandldc3694b2010-02-07 17:02:22 +0000913 the_exception = cm.exception
Michael Foordba7732e2010-02-05 23:28:12 +0000914 self.assertEqual(the_exception.error_code, 3)
Kristján Valur Jónssone2a77982009-08-27 22:20:21 +0000915
Benjamin Peterson99721e02009-03-23 23:10:14 +0000916 .. versionchanged:: 2.7
Benjamin Peterson7233acc2009-03-29 03:31:40 +0000917 Added the ability to use :meth:`assertRaises` as a context manager.
Benjamin Peterson99721e02009-03-23 23:10:14 +0000918
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000919 .. deprecated:: 2.7
Georg Brandl1c7c7302010-02-06 10:08:21 +0000920 :meth:`failUnlessRaises`; use :meth:`assertRaises`.
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000921
Benjamin Peterson99721e02009-03-23 23:10:14 +0000922
Gregory P. Smith28399852009-03-31 16:54:10 +0000923 .. method:: assertRaisesRegexp(exception, regexp[, callable, ...])
Benjamin Peterson99721e02009-03-23 23:10:14 +0000924
Gregory P. Smith28399852009-03-31 16:54:10 +0000925 Like :meth:`assertRaises` but also tests that *regexp* matches
926 on the string representation of the raised exception. *regexp* may be
927 a regular expression object or a string containing a regular expression
928 suitable for use by :func:`re.search`. Examples::
929
930 self.assertRaisesRegexp(ValueError, 'invalid literal for.*XYZ$',
931 int, 'XYZ')
932
933 or::
934
935 with self.assertRaisesRegexp(ValueError, 'literal'):
936 int('XYZ')
937
938 .. versionadded:: 2.7
939
940
941 .. method:: assertIsNone(expr[, msg])
942
943 This signals a test failure if *expr* is not None.
944
945 .. versionadded:: 2.7
946
947
948 .. method:: assertIsNotNone(expr[, msg])
949
950 The inverse of the :meth:`assertIsNone` method.
951 This signals a test failure if *expr* is None.
952
953 .. versionadded:: 2.7
954
955
Michael Foordf2dfef12009-04-05 19:19:28 +0000956 .. method:: assertIs(expr1, expr2[, msg])
957
958 This signals a test failure if *expr1* and *expr2* don't evaluate to the same
959 object.
960
961 .. versionadded:: 2.7
962
963
964 .. method:: assertIsNot(expr1, expr2[, msg])
965
966 The inverse of the :meth:`assertIs` method.
967 This signals a test failure if *expr1* and *expr2* evaluate to the same
968 object.
969
970 .. versionadded:: 2.7
971
972
Georg Brandlf895cf52009-10-01 20:59:31 +0000973 .. method:: assertIsInstance(obj, cls[, msg])
974
975 This signals a test failure if *obj* is not an instance of *cls* (which
976 can be a class or a tuple of classes, as supported by :func:`isinstance`).
977
978 .. versionadded:: 2.7
979
980
981 .. method:: assertNotIsInstance(obj, cls[, msg])
982
983 The inverse of the :meth:`assertIsInstance` method. This signals a test
984 failure if *obj* is an instance of *cls*.
985
986 .. versionadded:: 2.7
987
988
Gregory P. Smith28399852009-03-31 16:54:10 +0000989 .. method:: assertFalse(expr[, msg])
990 failIf(expr[, msg])
991
992 The inverse of the :meth:`assertTrue` method is the :meth:`assertFalse` method.
Benjamin Peterson99721e02009-03-23 23:10:14 +0000993 This signals a test failure if *expr* is true, with *msg* or :const:`None`
994 for the error message.
995
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000996 .. deprecated:: 2.7
Georg Brandl1c7c7302010-02-06 10:08:21 +0000997 :meth:`failIf`; use :meth:`assertFalse`.
Gregory P. Smith65ff0052009-03-31 19:59:14 +0000998
Benjamin Peterson99721e02009-03-23 23:10:14 +0000999
1000 .. method:: fail([msg])
1001
1002 Signals a test failure unconditionally, with *msg* or :const:`None` for
1003 the error message.
1004
1005
1006 .. attribute:: failureException
1007
1008 This class attribute gives the exception raised by the test method. If a
1009 test framework needs to use a specialized exception, possibly to carry
1010 additional information, it must subclass this exception in order to "play
1011 fair" with the framework. The initial value of this attribute is
1012 :exc:`AssertionError`.
1013
Michael Foord345b2fe2009-04-02 03:20:38 +00001014
1015 .. attribute:: longMessage
1016
1017 If set to True then any explicit failure message you pass in to the
1018 assert methods will be appended to the end of the normal failure message.
1019 The normal messages contain useful information about the objects involved,
1020 for example the message from assertEqual shows you the repr of the two
1021 unequal objects. Setting this attribute to True allows you to have a
1022 custom error message in addition to the normal one.
1023
1024 This attribute defaults to False, meaning that a custom message passed
1025 to an assert method will silence the normal message.
1026
1027 The class setting can be overridden in individual tests by assigning an
1028 instance attribute to True or False before calling the assert methods.
1029
1030 .. versionadded:: 2.7
1031
1032
Benjamin Peterson99721e02009-03-23 23:10:14 +00001033 Testing frameworks can use the following methods to collect information on
1034 the test:
1035
1036
1037 .. method:: countTestCases()
1038
1039 Return the number of tests represented by this test object. For
1040 :class:`TestCase` instances, this will always be ``1``.
1041
1042
1043 .. method:: defaultTestResult()
1044
1045 Return an instance of the test result class that should be used for this
1046 test case class (if no other result instance is provided to the
1047 :meth:`run` method).
1048
1049 For :class:`TestCase` instances, this will always be an instance of
1050 :class:`TestResult`; subclasses of :class:`TestCase` should override this
1051 as necessary.
1052
1053
1054 .. method:: id()
1055
1056 Return a string identifying the specific test case. This is usually the
1057 full name of the test method, including the module and class name.
1058
1059
1060 .. method:: shortDescription()
1061
Gregory P. Smith28399852009-03-31 16:54:10 +00001062 Returns a description of the test, or :const:`None` if no description
1063 has been provided. The default implementation of this method
1064 returns the first line of the test method's docstring, if available,
1065 along with the method name.
1066
1067 .. versionchanged:: 2.7
Gregory P. Smith28399852009-03-31 16:54:10 +00001068 In earlier versions this only returned the first line of the test
1069 method's docstring, if available or the :const:`None`. That led to
1070 undesirable behavior of not printing the test name when someone was
1071 thoughtful enough to write a docstring.
1072
1073
1074 .. method:: addTypeEqualityFunc(typeobj, function)
1075
1076 Registers a type specific :meth:`assertEqual` equality checking
1077 function to be called by :meth:`assertEqual` when both objects it has
1078 been asked to compare are exactly *typeobj* (not subclasses).
1079 *function* must take two positional arguments and a third msg=None
1080 keyword argument just as :meth:`assertEqual` does. It must raise
Andrew M. Kuchling59631852009-04-09 11:23:36 +00001081 ``self.failureException`` when inequality between the first two
Gregory P. Smith28399852009-03-31 16:54:10 +00001082 parameters is detected.
1083
1084 One good use of custom equality checking functions for a type
Andrew M. Kuchling59631852009-04-09 11:23:36 +00001085 is to raise ``self.failureException`` with an error message useful
1086 for debugging the problem by explaining the inequalities in detail.
Gregory P. Smith28399852009-03-31 16:54:10 +00001087
1088 .. versionadded:: 2.7
Georg Brandl8ec7f652007-08-15 14:28:01 +00001089
1090
Michael Foorde2fb98f2009-05-02 20:15:05 +00001091 .. method:: addCleanup(function[, *args[, **kwargs]])
1092
1093 Add a function to be called after :meth:`tearDown` to cleanup resources
1094 used during the test. Functions will be called in reverse order to the
1095 order they are added (LIFO). They are called with any arguments and
1096 keyword arguments passed into :meth:`addCleanup` when they are
1097 added.
1098
1099 If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
1100 then any cleanup functions added will still be called.
1101
1102 .. versionadded:: 2.7
1103
1104
1105 .. method:: doCleanups()
1106
1107 This method is called uncoditionally after :meth:`tearDown`, or
1108 after :meth:`setUp` if :meth:`setUp` raises an exception.
1109
1110 It is responsible for calling all the cleanup functions added by
1111 :meth:`addCleanup`. If you need cleanup functions to be called
1112 *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
1113 yourself.
1114
1115 :meth:`doCleanups` pops methods off the stack of cleanup
1116 functions one at a time, so it can be called at any time.
1117
1118 .. versionadded:: 2.7
1119
1120
Georg Brandl8ec7f652007-08-15 14:28:01 +00001121.. class:: FunctionTestCase(testFunc[, setUp[, tearDown[, description]]])
1122
1123 This class implements the portion of the :class:`TestCase` interface which
Georg Brandl2fcd1732009-05-30 10:45:40 +00001124 allows the test runner to drive the test, but does not provide the methods
1125 which test code can use to check and report errors. This is used to create
1126 test cases using legacy test code, allowing it to be integrated into a
1127 :mod:`unittest`-based test framework.
Georg Brandl8ec7f652007-08-15 14:28:01 +00001128
1129
Benjamin Peterson99721e02009-03-23 23:10:14 +00001130.. _testsuite-objects:
1131
1132Grouping tests
1133~~~~~~~~~~~~~~
1134
Georg Brandl8ec7f652007-08-15 14:28:01 +00001135.. class:: TestSuite([tests])
1136
1137 This class represents an aggregation of individual tests cases and test suites.
1138 The class presents the interface needed by the test runner to allow it to be run
1139 as any other test case. Running a :class:`TestSuite` instance is the same as
1140 iterating over the suite, running each test individually.
1141
1142 If *tests* is given, it must be an iterable of individual test cases or other
1143 test suites that will be used to build the suite initially. Additional methods
1144 are provided to add test cases and suites to the collection later on.
1145
Benjamin Peterson176a56c2009-05-25 00:48:58 +00001146 :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
1147 they do not actually implement a test. Instead, they are used to aggregate
1148 tests into groups of tests that should be run together. Some additional
1149 methods are available to add tests to :class:`TestSuite` instances:
Benjamin Peterson99721e02009-03-23 23:10:14 +00001150
1151
1152 .. method:: TestSuite.addTest(test)
1153
1154 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1155
1156
1157 .. method:: TestSuite.addTests(tests)
1158
1159 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
1160 instances to this test suite.
1161
Georg Brandl2fcd1732009-05-30 10:45:40 +00001162 This is equivalent to iterating over *tests*, calling :meth:`addTest` for
1163 each element.
Benjamin Peterson99721e02009-03-23 23:10:14 +00001164
1165 :class:`TestSuite` shares the following methods with :class:`TestCase`:
1166
1167
1168 .. method:: run(result)
1169
1170 Run the tests associated with this suite, collecting the result into the
1171 test result object passed as *result*. Note that unlike
1172 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
1173 be passed in.
1174
1175
1176 .. method:: debug()
1177
1178 Run the tests associated with this suite without collecting the
1179 result. This allows exceptions raised by the test to be propagated to the
1180 caller and can be used to support running tests under a debugger.
1181
1182
1183 .. method:: countTestCases()
1184
1185 Return the number of tests represented by this test object, including all
1186 individual tests and sub-suites.
1187
Georg Brandl9bc66822009-04-27 17:04:23 +00001188
1189 .. method:: __iter__()
1190
1191 Tests grouped by a :class:`TestSuite` are always accessed by iteration.
1192 Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
1193 that this method maybe called several times on a single suite
1194 (for example when counting tests or comparing for equality)
1195 so the tests returned must be the same for repeated iterations.
1196
1197 .. versionchanged:: 2.7
1198 In earlier versions the :class:`TestSuite` accessed tests directly rather
1199 than through iteration, so overriding :meth:`__iter__` wasn't sufficient
1200 for providing tests.
1201
Benjamin Peterson99721e02009-03-23 23:10:14 +00001202 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
1203 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1204
1205
Benjamin Peterson99721e02009-03-23 23:10:14 +00001206Loading and running tests
1207~~~~~~~~~~~~~~~~~~~~~~~~~
1208
Georg Brandl8ec7f652007-08-15 14:28:01 +00001209.. class:: TestLoader()
1210
Benjamin Peterson99721e02009-03-23 23:10:14 +00001211 The :class:`TestLoader` class is used to create test suites from classes and
1212 modules. Normally, there is no need to create an instance of this class; the
1213 :mod:`unittest` module provides an instance that can be shared as
1214 ``unittest.defaultTestLoader``. Using a subclass or instance, however, allows
1215 customization of some configurable properties.
1216
1217 :class:`TestLoader` objects have the following methods:
Georg Brandl8ec7f652007-08-15 14:28:01 +00001218
1219
Benjamin Peterson99721e02009-03-23 23:10:14 +00001220 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl8ec7f652007-08-15 14:28:01 +00001221
Benjamin Peterson99721e02009-03-23 23:10:14 +00001222 Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
1223 :class:`testCaseClass`.
1224
1225
1226 .. method:: loadTestsFromModule(module)
1227
1228 Return a suite of all tests cases contained in the given module. This
1229 method searches *module* for classes derived from :class:`TestCase` and
1230 creates an instance of the class for each test method defined for the
1231 class.
1232
Georg Brandl16a57f62009-04-27 15:29:09 +00001233 .. note::
Benjamin Peterson99721e02009-03-23 23:10:14 +00001234
1235 While using a hierarchy of :class:`TestCase`\ -derived classes can be
1236 convenient in sharing fixtures and helper functions, defining test
1237 methods on base classes that are not intended to be instantiated
1238 directly does not play well with this method. Doing so, however, can
1239 be useful when the fixtures are different and defined in subclasses.
1240
Michael Foordb4a81c82009-05-29 20:33:46 +00001241 If a module provides a ``load_tests`` function it will be called to
1242 load the tests. This allows modules to customize test loading.
1243 This is the `load_tests protocol`_.
1244
1245 .. versionchanged:: 2.7
1246 Support for ``load_tests`` added.
1247
Benjamin Peterson99721e02009-03-23 23:10:14 +00001248
1249 .. method:: loadTestsFromName(name[, module])
1250
1251 Return a suite of all tests cases given a string specifier.
1252
1253 The specifier *name* is a "dotted name" that may resolve either to a
1254 module, a test case class, a test method within a test case class, a
1255 :class:`TestSuite` instance, or a callable object which returns a
1256 :class:`TestCase` or :class:`TestSuite` instance. These checks are
1257 applied in the order listed here; that is, a method on a possible test
1258 case class will be picked up as "a test method within a test case class",
1259 rather than "a callable object".
1260
1261 For example, if you have a module :mod:`SampleTests` containing a
Georg Brandl2fcd1732009-05-30 10:45:40 +00001262 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
1263 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
1264 specifier ``'SampleTests.SampleTestCase'`` would cause this method to
1265 return a suite which will run all three test methods. Using the specifier
1266 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
1267 suite which will run only the :meth:`test_two` test method. The specifier
1268 can refer to modules and packages which have not been imported; they will
1269 be imported as a side-effect.
Benjamin Peterson99721e02009-03-23 23:10:14 +00001270
1271 The method optionally resolves *name* relative to the given *module*.
1272
1273
1274 .. method:: loadTestsFromNames(names[, module])
1275
1276 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
1277 than a single name. The return value is a test suite which supports all
1278 the tests defined for each name.
1279
1280
1281 .. method:: getTestCaseNames(testCaseClass)
1282
1283 Return a sorted sequence of method names found within *testCaseClass*;
1284 this should be a subclass of :class:`TestCase`.
1285
Michael Foordb4a81c82009-05-29 20:33:46 +00001286
1287 .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)
1288
1289 Find and return all test modules from the specified start directory,
1290 recursing into subdirectories to find them. Only test files that match
Michael Foorde91ea562009-09-13 19:07:03 +00001291 *pattern* will be loaded. (Using shell style pattern matching.) Only
1292 module names that are importable (i.e. are valid Python identifiers) will
1293 be loaded.
Michael Foordb4a81c82009-05-29 20:33:46 +00001294
1295 All test modules must be importable from the top level of the project. If
1296 the start directory is not the top level directory then the top level
1297 directory must be specified separately.
1298
Michael Foorde91ea562009-09-13 19:07:03 +00001299 If importing a module fails, for example due to a syntax error, then this
1300 will be recorded as a single error and discovery will continue.
1301
Michael Foordb4a81c82009-05-29 20:33:46 +00001302 If a test package name (directory with :file:`__init__.py`) matches the
1303 pattern then the package will be checked for a ``load_tests``
1304 function. If this exists then it will be called with *loader*, *tests*,
1305 *pattern*.
1306
Michael Foorddc0460a2009-09-13 19:08:18 +00001307 If load_tests exists then discovery does *not* recurse into the package,
Michael Foordb4a81c82009-05-29 20:33:46 +00001308 ``load_tests`` is responsible for loading all tests in the package.
1309
1310 The pattern is deliberately not stored as a loader attribute so that
1311 packages can continue discovery themselves. *top_level_dir* is stored so
1312 ``load_tests`` does not need to pass this argument in to
1313 ``loader.discover()``.
1314
Michael Foord17565e52009-09-27 20:08:23 +00001315 .. versionadded:: 2.7
Michael Foordb4a81c82009-05-29 20:33:46 +00001316
Benjamin Peterson99721e02009-03-23 23:10:14 +00001317 The following attributes of a :class:`TestLoader` can be configured either by
1318 subclassing or assignment on an instance:
1319
1320
1321 .. attribute:: testMethodPrefix
1322
1323 String giving the prefix of method names which will be interpreted as test
1324 methods. The default value is ``'test'``.
1325
1326 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
1327 methods.
1328
1329
1330 .. attribute:: sortTestMethodsUsing
1331
1332 Function to be used to compare method names when sorting them in
1333 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods. The
1334 default value is the built-in :func:`cmp` function; the attribute can also
1335 be set to :const:`None` to disable the sort.
1336
1337
1338 .. attribute:: suiteClass
1339
1340 Callable object that constructs a test suite from a list of tests. No
1341 methods on the resulting object are needed. The default value is the
1342 :class:`TestSuite` class.
1343
1344 This affects all the :meth:`loadTestsFrom\*` methods.
1345
1346
Benjamin Peterson99721e02009-03-23 23:10:14 +00001347.. class:: TestResult
1348
1349 This class is used to compile information about which tests have succeeded
1350 and which have failed.
1351
1352 A :class:`TestResult` object stores the results of a set of tests. The
1353 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
1354 properly recorded; test authors do not need to worry about recording the
1355 outcome of tests.
1356
1357 Testing frameworks built on top of :mod:`unittest` may want access to the
1358 :class:`TestResult` object generated by running a set of tests for reporting
1359 purposes; a :class:`TestResult` instance is returned by the
1360 :meth:`TestRunner.run` method for this purpose.
1361
1362 :class:`TestResult` instances have the following attributes that will be of
1363 interest when inspecting the results of running a set of tests:
1364
1365
1366 .. attribute:: errors
1367
1368 A list containing 2-tuples of :class:`TestCase` instances and strings
1369 holding formatted tracebacks. Each tuple represents a test which raised an
1370 unexpected exception.
1371
1372 .. versionchanged:: 2.2
Benjamin Peterson99721e02009-03-23 23:10:14 +00001373 Contains formatted tracebacks instead of :func:`sys.exc_info` results.
1374
1375
1376 .. attribute:: failures
1377
1378 A list containing 2-tuples of :class:`TestCase` instances and strings
1379 holding formatted tracebacks. Each tuple represents a test where a failure
1380 was explicitly signalled using the :meth:`TestCase.fail\*` or
1381 :meth:`TestCase.assert\*` methods.
1382
1383 .. versionchanged:: 2.2
Benjamin Peterson99721e02009-03-23 23:10:14 +00001384 Contains formatted tracebacks instead of :func:`sys.exc_info` results.
1385
1386 .. attribute:: skipped
1387
1388 A list containing 2-tuples of :class:`TestCase` instances and strings
1389 holding the reason for skipping the test.
1390
1391 .. versionadded:: 2.7
1392
1393 .. attribute:: expectedFailures
1394
1395 A list contaning 2-tuples of :class:`TestCase` instances and strings
1396 holding formatted tracebacks. Each tuple represents a expected failures
1397 of the test case.
1398
1399 .. attribute:: unexpectedSuccesses
1400
1401 A list containing :class:`TestCase` instances that were marked as expected
1402 failures, but succeeded.
1403
1404 .. attribute:: shouldStop
1405
1406 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1407
1408
1409 .. attribute:: testsRun
1410
1411 The total number of tests run so far.
1412
1413
1414 .. method:: wasSuccessful()
1415
1416 Return :const:`True` if all tests run so far have passed, otherwise returns
1417 :const:`False`.
1418
1419
1420 .. method:: stop()
1421
1422 This method can be called to signal that the set of tests being run should
1423 be aborted by setting the :attr:`shouldStop` attribute to :const:`True`.
1424 :class:`TestRunner` objects should respect this flag and return without
1425 running any additional tests.
1426
1427 For example, this feature is used by the :class:`TextTestRunner` class to
1428 stop the test framework when the user signals an interrupt from the
1429 keyboard. Interactive tools which provide :class:`TestRunner`
1430 implementations can use this in a similar manner.
1431
1432 The following methods of the :class:`TestResult` class are used to maintain
1433 the internal data structures, and may be extended in subclasses to support
1434 additional reporting requirements. This is particularly useful in building
1435 tools which support interactive reporting while tests are being run.
1436
1437
1438 .. method:: startTest(test)
1439
1440 Called when the test case *test* is about to be run.
1441
1442 The default implementation simply increments the instance's :attr:`testsRun`
1443 counter.
1444
1445
1446 .. method:: stopTest(test)
1447
1448 Called after the test case *test* has been executed, regardless of the
1449 outcome.
1450
1451 The default implementation does nothing.
1452
1453
Michael Foord07ef4872009-05-02 22:43:34 +00001454 .. method:: startTestRun(test)
1455
1456 Called once before any tests are executed.
1457
1458 .. versionadded:: 2.7
1459
1460
1461 .. method:: stopTestRun(test)
1462
Ezio Melotti7b4e02c2010-01-27 20:25:11 +00001463 Called once after all tests are executed.
Michael Foord07ef4872009-05-02 22:43:34 +00001464
1465 .. versionadded:: 2.7
1466
1467
Benjamin Peterson99721e02009-03-23 23:10:14 +00001468 .. method:: addError(test, err)
1469
1470 Called when the test case *test* raises an unexpected exception *err* is a
1471 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1472 traceback)``.
1473
1474 The default implementation appends a tuple ``(test, formatted_err)`` to
1475 the instance's :attr:`errors` attribute, where *formatted_err* is a
1476 formatted traceback derived from *err*.
1477
1478
1479 .. method:: addFailure(test, err)
1480
Michael Foordb4a81c82009-05-29 20:33:46 +00001481 Called when the test case *test* signals a failure. *err* is a tuple of
1482 the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
Benjamin Peterson99721e02009-03-23 23:10:14 +00001483
1484 The default implementation appends a tuple ``(test, formatted_err)`` to
1485 the instance's :attr:`failures` attribute, where *formatted_err* is a
1486 formatted traceback derived from *err*.
1487
1488
1489 .. method:: addSuccess(test)
1490
1491 Called when the test case *test* succeeds.
1492
1493 The default implementation does nothing.
1494
1495
1496 .. method:: addSkip(test, reason)
1497
1498 Called when the test case *test* is skipped. *reason* is the reason the
1499 test gave for skipping.
1500
1501 The default implementation appends a tuple ``(test, reason)`` to the
1502 instance's :attr:`skipped` attribute.
1503
1504
1505 .. method:: addExpectedFailure(test, err)
1506
1507 Called when the test case *test* fails, but was marked with the
1508 :func:`expectedFailure` decorator.
1509
1510 The default implementation appends a tuple ``(test, formatted_err)`` to
1511 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1512 is a formatted traceback derived from *err*.
1513
1514
1515 .. method:: addUnexpectedSuccess(test)
1516
1517 Called when the test case *test* was marked with the
1518 :func:`expectedFailure` decorator, but succeeded.
1519
1520 The default implementation appends the test to the instance's
1521 :attr:`unexpectedSuccesses` attribute.
Georg Brandl8ec7f652007-08-15 14:28:01 +00001522
1523
1524.. data:: defaultTestLoader
1525
1526 Instance of the :class:`TestLoader` class intended to be shared. If no
1527 customization of the :class:`TestLoader` is needed, this instance can be used
1528 instead of repeatedly creating new instances.
1529
1530
1531.. class:: TextTestRunner([stream[, descriptions[, verbosity]]])
1532
1533 A basic test runner implementation which prints results on standard error. It
1534 has a few configurable parameters, but is essentially very simple. Graphical
1535 applications which run test suites should provide alternate implementations.
1536
Georg Brandl9bc66822009-04-27 17:04:23 +00001537 .. method:: _makeResult()
1538
1539 This method returns the instance of ``TestResult`` used by :meth:`run`.
1540 It is not intended to be called directly, but can be overridden in
1541 subclasses to provide a custom ``TestResult``.
1542
Georg Brandl8ec7f652007-08-15 14:28:01 +00001543
Michael Foord5d31e052009-05-11 17:59:43 +00001544.. function:: main([module[, defaultTest[, argv[, testRunner[, testLoader[, exit, [verbosity]]]]]]])
Georg Brandl8ec7f652007-08-15 14:28:01 +00001545
1546 A command-line program that runs a set of tests; this is primarily for making
1547 test modules conveniently executable. The simplest use for this function is to
1548 include the following line at the end of a test script::
1549
1550 if __name__ == '__main__':
1551 unittest.main()
1552
Michael Foord5d31e052009-05-11 17:59:43 +00001553 You can run tests with more detailed information by passing in the verbosity
1554 argument::
1555
1556 if __name__ == '__main__':
1557 unittest.main(verbosity=2)
1558
Georg Brandl8ec7f652007-08-15 14:28:01 +00001559 The *testRunner* argument can either be a test runner class or an already
Michael Foord829f6b82009-05-02 11:43:06 +00001560 created instance of it. By default ``main`` calls :func:`sys.exit` with
1561 an exit code indicating success or failure of the tests run.
1562
1563 ``main`` supports being used from the interactive interpreter by passing in the
1564 argument ``exit=False``. This displays the result on standard output without
1565 calling :func:`sys.exit`::
1566
1567 >>> from unittest import main
1568 >>> main(module='test_module', exit=False)
1569
1570 Calling ``main`` actually returns an instance of the ``TestProgram`` class.
1571 This stores the result of the tests run as the ``result`` attribute.
1572
1573 .. versionchanged:: 2.7
Michael Foord5d31e052009-05-11 17:59:43 +00001574 The ``exit`` and ``verbosity`` parameters were added.
Michael Foordb4a81c82009-05-29 20:33:46 +00001575
1576
1577load_tests Protocol
1578###################
1579
Michael Foord17565e52009-09-27 20:08:23 +00001580
1581.. versionadded:: 2.7
1582
1583
Michael Foordb4a81c82009-05-29 20:33:46 +00001584Modules or packages can customize how tests are loaded from them during normal
1585test runs or test discovery by implementing a function called ``load_tests``.
1586
1587If a test module defines ``load_tests`` it will be called by
1588:meth:`TestLoader.loadTestsFromModule` with the following arguments::
1589
1590 load_tests(loader, standard_tests, None)
1591
1592It should return a :class:`TestSuite`.
1593
1594*loader* is the instance of :class:`TestLoader` doing the loading.
1595*standard_tests* are the tests that would be loaded by default from the
1596module. It is common for test modules to only want to add or remove tests
1597from the standard set of tests.
1598The third argument is used when loading packages as part of test discovery.
1599
1600A typical ``load_tests`` function that loads tests from a specific set of
1601:class:`TestCase` classes may look like::
1602
1603 test_cases = (TestCase1, TestCase2, TestCase3)
1604
1605 def load_tests(loader, tests, pattern):
1606 suite = TestSuite()
1607 for test_class in test_cases:
1608 tests = loader.loadTestsFromTestCase(test_class)
1609 suite.addTests(tests)
1610 return suite
1611
1612If discovery is started, either from the command line or by calling
1613:meth:`TestLoader.discover`, with a pattern that matches a package
1614name then the package :file:`__init__.py` will be checked for ``load_tests``.
1615
1616.. note::
1617
Ezio Melotti062d2b52009-12-19 22:41:49 +00001618 The default pattern is 'test*.py'. This matches all Python files
Michael Foordb4a81c82009-05-29 20:33:46 +00001619 that start with 'test' but *won't* match any test directories.
1620
1621 A pattern like 'test*' will match test packages as well as
1622 modules.
1623
1624If the package :file:`__init__.py` defines ``load_tests`` then it will be
1625called and discovery not continued into the package. ``load_tests``
1626is called with the following arguments::
1627
1628 load_tests(loader, standard_tests, pattern)
1629
1630This should return a :class:`TestSuite` representing all the tests
1631from the package. (``standard_tests`` will only contain tests
1632collected from :file:`__init__.py`.)
1633
1634Because the pattern is passed into ``load_tests`` the package is free to
1635continue (and potentially modify) test discovery. A 'do nothing'
1636``load_tests`` function for a test package would look like::
1637
1638 def load_tests(loader, standard_tests, pattern):
1639 # top level directory cached on loader instance
1640 this_dir = os.path.dirname(__file__)
1641 package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
1642 standard_tests.addTests(package_tests)
1643 return standard_tests