blob: 72d7dec4b22fe0cd604ce153fad770de5a77ad48 [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001:mod:`unittest` --- Unit testing framework
2==========================================
3
4.. module:: unittest
5 :synopsis: Unit testing framework for Python.
6.. moduleauthor:: Steve Purcell <stephen_purcell@yahoo.com>
7.. sectionauthor:: Steve Purcell <stephen_purcell@yahoo.com>
8.. sectionauthor:: Fred L. Drake, Jr. <fdrake@acm.org>
9.. sectionauthor:: Raymond Hettinger <python@rcn.com>
10
11
Georg Brandl116aa622007-08-15 14:28:22 +000012The Python unit testing framework, sometimes referred to as "PyUnit," is a
13Python language version of JUnit, by Kent Beck and Erich Gamma. JUnit is, in
14turn, a Java version of Kent's Smalltalk testing framework. Each is the de
15facto standard unit testing framework for its respective language.
16
17:mod:`unittest` supports test automation, sharing of setup and shutdown code for
18tests, aggregation of tests into collections, and independence of the tests from
19the reporting framework. The :mod:`unittest` module provides classes that make
20it easy to support these qualities for a set of tests.
21
22To achieve this, :mod:`unittest` supports some important concepts:
23
24test fixture
25 A :dfn:`test fixture` represents the preparation needed to perform one or more
26 tests, and any associate cleanup actions. This may involve, for example,
27 creating temporary or proxy databases, directories, or starting a server
28 process.
29
30test case
31 A :dfn:`test case` is the smallest unit of testing. It checks for a specific
32 response to a particular set of inputs. :mod:`unittest` provides a base class,
33 :class:`TestCase`, which may be used to create new test cases.
34
35test suite
36 A :dfn:`test suite` is a collection of test cases, test suites, or both. It is
37 used to aggregate tests that should be executed together.
38
39test runner
40 A :dfn:`test runner` is a component which orchestrates the execution of tests
41 and provides the outcome to the user. The runner may use a graphical interface,
42 a textual interface, or return a special value to indicate the results of
43 executing the tests.
44
45The test case and test fixture concepts are supported through the
46:class:`TestCase` and :class:`FunctionTestCase` classes; the former should be
47used when creating new tests, and the latter can be used when integrating
48existing test code with a :mod:`unittest`\ -driven framework. When building test
Benjamin Peterson52baa292009-03-24 00:56:30 +000049fixtures using :class:`TestCase`, the :meth:`~TestCase.setUp` and
50:meth:`~TestCase.tearDown` methods can be overridden to provide initialization
51and cleanup for the fixture. With :class:`FunctionTestCase`, existing functions
52can be passed to the constructor for these purposes. When the test is run, the
53fixture initialization is run first; if it succeeds, the cleanup method is run
54after the test has been executed, regardless of the outcome of the test. Each
55instance of the :class:`TestCase` will only be used to run a single test method,
56so a new fixture is created for each test.
Georg Brandl116aa622007-08-15 14:28:22 +000057
58Test suites are implemented by the :class:`TestSuite` class. This class allows
59individual tests and test suites to be aggregated; when the suite is executed,
Benjamin Peterson14a3dd72009-05-25 00:51:58 +000060all tests added directly to the suite and in "child" test suites are run.
Georg Brandl116aa622007-08-15 14:28:22 +000061
Benjamin Peterson52baa292009-03-24 00:56:30 +000062A test runner is an object that provides a single method,
63:meth:`~TestRunner.run`, which accepts a :class:`TestCase` or :class:`TestSuite`
64object as a parameter, and returns a result object. The class
65:class:`TestResult` is provided for use as the result object. :mod:`unittest`
66provides the :class:`TextTestRunner` as an example test runner which reports
67test results on the standard error stream by default. Alternate runners can be
68implemented for other environments (such as graphical environments) without any
69need to derive from a specific class.
Georg Brandl116aa622007-08-15 14:28:22 +000070
71
72.. seealso::
73
74 Module :mod:`doctest`
75 Another test-support module with a very different flavor.
76
77 `Simple Smalltalk Testing: With Patterns <http://www.XProgramming.com/testfram.htm>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000078 Kent Beck's original paper on testing frameworks using the pattern shared
79 by :mod:`unittest`.
Georg Brandl116aa622007-08-15 14:28:22 +000080
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000081 `Nose <http://code.google.com/p/python-nose/>`_ and `py.test <http://pytest.org>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000082 Third-party unittest frameworks with a lighter-weight syntax for writing
83 tests. For example, ``assert func(10) == 42``.
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000084
85 `python-mock <http://python-mock.sourceforge.net/>`_ and `minimock <http://blog.ianbicking.org/minimock.html>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000086 Tools for creating mock test objects (objects simulating external
87 resources).
88
89
90.. _unittest-command-line-interface:
91
92Command Line Interface
93----------------------
94
95The unittest module can be used from the command line to run tests from
96modules, classes or even individual test methods::
97
98 python -m unittest test_module1 test_module2
99 python -m unittest test_module.TestClass
100 python -m unittest test_module.TestClass.test_method
101
102You can pass in a list with any combination of module names, and fully
103qualified class or method names.
104
105You can run tests with more detail (higher verbosity) by passing in the -v flag::
106
Ezio Melotti176d6c42010-01-27 20:58:07 +0000107 python -m unittest -v test_module
Benjamin Petersond2397752009-06-27 23:45:02 +0000108
109For a list of all the command line options::
110
111 python -m unittest -h
112
Georg Brandl853947a2010-01-31 18:53:23 +0000113.. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +0000114 In earlier versions it was only possible to run individual test methods and
115 not modules or classes.
116
117The command line can also be used for test discovery, for running all of the
118tests in a project or just a subset.
119
120
121.. _unittest-test-discovery:
122
123Test Discovery
124--------------
125
Georg Brandl853947a2010-01-31 18:53:23 +0000126.. versionadded:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +0000127
128unittest supports simple test discovery. For a project's tests to be
129compatible with test discovery they must all be importable from the top level
130directory of the project; i.e. they must all be in Python packages.
131
132Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
133used from the command line. The basic command line usage is::
134
135 cd project_directory
136 python -m unittest discover
137
138The ``discover`` sub-command has the following options:
139
140 -v, --verbose Verbose output
141 -s directory Directory to start discovery ('.' default)
142 -p pattern Pattern to match test files ('test*.py' default)
143 -t directory Top level directory of project (default to
144 start directory)
145
146The -s, -p, & -t options can be passsed in as positional arguments. The
147following two command lines are equivalent::
148
Ezio Melotti176d6c42010-01-27 20:58:07 +0000149 python -m unittest discover -s project_directory -p '*_test.py'
150 python -m unittest discover project_directory '*_test.py'
Benjamin Petersond2397752009-06-27 23:45:02 +0000151
152Test modules and packages can customize test loading and discovery by through
153the `load_tests protocol`_.
Georg Brandl116aa622007-08-15 14:28:22 +0000154
155.. _unittest-minimal-example:
156
157Basic example
158-------------
159
160The :mod:`unittest` module provides a rich set of tools for constructing and
161running tests. This section demonstrates that a small subset of the tools
162suffice to meet the needs of most users.
163
164Here is a short script to test three functions from the :mod:`random` module::
165
166 import random
167 import unittest
168
169 class TestSequenceFunctions(unittest.TestCase):
170
171 def setUp(self):
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000172 self.seq = list(range(10))
Georg Brandl116aa622007-08-15 14:28:22 +0000173
Benjamin Peterson52baa292009-03-24 00:56:30 +0000174 def test_shuffle(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000175 # make sure the shuffled sequence does not lose any elements
176 random.shuffle(self.seq)
177 self.seq.sort()
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000178 self.assertEqual(self.seq, list(range(10)))
Georg Brandl116aa622007-08-15 14:28:22 +0000179
Benjamin Peterson52baa292009-03-24 00:56:30 +0000180 def test_choice(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000181 element = random.choice(self.seq)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000182 self.assertIn(element, self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000183
Benjamin Peterson52baa292009-03-24 00:56:30 +0000184 def test_sample(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000185 self.assertRaises(ValueError, random.sample, self.seq, 20)
186 for element in random.sample(self.seq, 5):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000187 self.assertIn(element, self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000188
189 if __name__ == '__main__':
190 unittest.main()
191
Benjamin Peterson52baa292009-03-24 00:56:30 +0000192A testcase is created by subclassing :class:`unittest.TestCase`. The three
Georg Brandl116aa622007-08-15 14:28:22 +0000193individual tests are defined with methods whose names start with the letters
194``test``. This naming convention informs the test runner about which methods
195represent tests.
196
Benjamin Peterson52baa292009-03-24 00:56:30 +0000197The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
198expected result; :meth:`~TestCase.assert_` to verify a condition; or
199:meth:`~TestCase.assertRaises` to verify that an expected exception gets raised.
200These methods are used instead of the :keyword:`assert` statement so the test
201runner can accumulate all test results and produce a report.
Georg Brandl116aa622007-08-15 14:28:22 +0000202
Benjamin Peterson52baa292009-03-24 00:56:30 +0000203When a :meth:`~TestCase.setUp` method is defined, the test runner will run that
204method prior to each test. Likewise, if a :meth:`~TestCase.tearDown` method is
205defined, the test runner will invoke that method after each test. In the
206example, :meth:`~TestCase.setUp` was used to create a fresh sequence for each
207test.
Georg Brandl116aa622007-08-15 14:28:22 +0000208
209The final block shows a simple way to run the tests. :func:`unittest.main`
210provides a command line interface to the test script. When run from the command
211line, the above script produces an output that looks like this::
212
213 ...
214 ----------------------------------------------------------------------
215 Ran 3 tests in 0.000s
216
217 OK
218
219Instead of :func:`unittest.main`, there are other ways to run the tests with a
220finer level of control, less terse output, and no requirement to be run from the
221command line. For example, the last two lines may be replaced with::
222
223 suite = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)
224 unittest.TextTestRunner(verbosity=2).run(suite)
225
226Running the revised script from the interpreter or another script produces the
227following output::
228
229 testchoice (__main__.TestSequenceFunctions) ... ok
230 testsample (__main__.TestSequenceFunctions) ... ok
231 testshuffle (__main__.TestSequenceFunctions) ... ok
232
233 ----------------------------------------------------------------------
234 Ran 3 tests in 0.110s
235
236 OK
237
238The above examples show the most commonly used :mod:`unittest` features which
239are sufficient to meet many everyday testing needs. The remainder of the
240documentation explores the full feature set from first principles.
241
Georg Brandl116aa622007-08-15 14:28:22 +0000242.. _organizing-tests:
243
244Organizing test code
245--------------------
246
247The basic building blocks of unit testing are :dfn:`test cases` --- single
248scenarios that must be set up and checked for correctness. In :mod:`unittest`,
249test cases are represented by instances of :mod:`unittest`'s :class:`TestCase`
250class. To make your own test cases you must write subclasses of
251:class:`TestCase`, or use :class:`FunctionTestCase`.
252
253An instance of a :class:`TestCase`\ -derived class is an object that can
254completely run a single test method, together with optional set-up and tidy-up
255code.
256
257The testing code of a :class:`TestCase` instance should be entirely self
258contained, such that it can be run either in isolation or in arbitrary
259combination with any number of other test cases.
260
Benjamin Peterson52baa292009-03-24 00:56:30 +0000261The simplest :class:`TestCase` subclass will simply override the
262:meth:`~TestCase.runTest` method in order to perform specific testing code::
Georg Brandl116aa622007-08-15 14:28:22 +0000263
264 import unittest
265
266 class DefaultWidgetSizeTestCase(unittest.TestCase):
267 def runTest(self):
268 widget = Widget('The widget')
269 self.assertEqual(widget.size(), (50, 50), 'incorrect default size')
270
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000271Note that in order to test something, we use the one of the :meth:`assert\*`
Benjamin Petersond2397752009-06-27 23:45:02 +0000272methods provided by the :class:`TestCase` base class. If the test fails, an
273exception will be raised, and :mod:`unittest` will identify the test case as a
274:dfn:`failure`. Any other exceptions will be treated as :dfn:`errors`. This
275helps you identify where the problem is: :dfn:`failures` are caused by incorrect
276results - a 5 where you expected a 6. :dfn:`Errors` are caused by incorrect
277code - e.g., a :exc:`TypeError` caused by an incorrect function call.
Georg Brandl116aa622007-08-15 14:28:22 +0000278
279The way to run a test case will be described later. For now, note that to
280construct an instance of such a test case, we call its constructor without
281arguments::
282
283 testCase = DefaultWidgetSizeTestCase()
284
285Now, such test cases can be numerous, and their set-up can be repetitive. In
286the above case, constructing a :class:`Widget` in each of 100 Widget test case
287subclasses would mean unsightly duplication.
288
289Luckily, we can factor out such set-up code by implementing a method called
Benjamin Peterson52baa292009-03-24 00:56:30 +0000290:meth:`~TestCase.setUp`, which the testing framework will automatically call for
291us when we run the test::
Georg Brandl116aa622007-08-15 14:28:22 +0000292
293 import unittest
294
295 class SimpleWidgetTestCase(unittest.TestCase):
296 def setUp(self):
297 self.widget = Widget('The widget')
298
299 class DefaultWidgetSizeTestCase(SimpleWidgetTestCase):
300 def runTest(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000301 self.assertEqual(self.widget.size(), (50,50),
302 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000303
304 class WidgetResizeTestCase(SimpleWidgetTestCase):
305 def runTest(self):
306 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000307 self.assertEqual(self.widget.size(), (100,150),
308 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000309
Benjamin Peterson52baa292009-03-24 00:56:30 +0000310If the :meth:`~TestCase.setUp` method raises an exception while the test is
311running, the framework will consider the test to have suffered an error, and the
312:meth:`~TestCase.runTest` method will not be executed.
Georg Brandl116aa622007-08-15 14:28:22 +0000313
Benjamin Peterson52baa292009-03-24 00:56:30 +0000314Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
315after the :meth:`~TestCase.runTest` method has been run::
Georg Brandl116aa622007-08-15 14:28:22 +0000316
317 import unittest
318
319 class SimpleWidgetTestCase(unittest.TestCase):
320 def setUp(self):
321 self.widget = Widget('The widget')
322
323 def tearDown(self):
324 self.widget.dispose()
325 self.widget = None
326
Benjamin Peterson52baa292009-03-24 00:56:30 +0000327If :meth:`~TestCase.setUp` succeeded, the :meth:`~TestCase.tearDown` method will
328be run whether :meth:`~TestCase.runTest` succeeded or not.
Georg Brandl116aa622007-08-15 14:28:22 +0000329
330Such a working environment for the testing code is called a :dfn:`fixture`.
331
332Often, many small test cases will use the same fixture. In this case, we would
333end up subclassing :class:`SimpleWidgetTestCase` into many small one-method
334classes such as :class:`DefaultWidgetSizeTestCase`. This is time-consuming and
Georg Brandl116aa622007-08-15 14:28:22 +0000335discouraging, so in the same vein as JUnit, :mod:`unittest` provides a simpler
336mechanism::
337
338 import unittest
339
340 class WidgetTestCase(unittest.TestCase):
341 def setUp(self):
342 self.widget = Widget('The widget')
343
344 def tearDown(self):
345 self.widget.dispose()
346 self.widget = None
347
348 def testDefaultSize(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000349 self.assertEqual(self.widget.size(), (50,50),
350 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000351
352 def testResize(self):
353 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000354 self.assertEqual(self.widget.size(), (100,150),
355 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000356
Benjamin Peterson52baa292009-03-24 00:56:30 +0000357Here we have not provided a :meth:`~TestCase.runTest` method, but have instead
358provided two different test methods. Class instances will now each run one of
359the :meth:`test\*` methods, with ``self.widget`` created and destroyed
360separately for each instance. When creating an instance we must specify the
361test method it is to run. We do this by passing the method name in the
362constructor::
Georg Brandl116aa622007-08-15 14:28:22 +0000363
364 defaultSizeTestCase = WidgetTestCase('testDefaultSize')
365 resizeTestCase = WidgetTestCase('testResize')
366
367Test case instances are grouped together according to the features they test.
368:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
369represented by :mod:`unittest`'s :class:`TestSuite` class::
370
371 widgetTestSuite = unittest.TestSuite()
372 widgetTestSuite.addTest(WidgetTestCase('testDefaultSize'))
373 widgetTestSuite.addTest(WidgetTestCase('testResize'))
374
375For the ease of running tests, as we will see later, it is a good idea to
376provide in each test module a callable object that returns a pre-built test
377suite::
378
379 def suite():
380 suite = unittest.TestSuite()
381 suite.addTest(WidgetTestCase('testDefaultSize'))
382 suite.addTest(WidgetTestCase('testResize'))
383 return suite
384
385or even::
386
387 def suite():
388 tests = ['testDefaultSize', 'testResize']
389
390 return unittest.TestSuite(map(WidgetTestCase, tests))
391
392Since it is a common pattern to create a :class:`TestCase` subclass with many
393similarly named test functions, :mod:`unittest` provides a :class:`TestLoader`
394class that can be used to automate the process of creating a test suite and
395populating it with individual tests. For example, ::
396
397 suite = unittest.TestLoader().loadTestsFromTestCase(WidgetTestCase)
398
399will create a test suite that will run ``WidgetTestCase.testDefaultSize()`` and
400``WidgetTestCase.testResize``. :class:`TestLoader` uses the ``'test'`` method
401name prefix to identify test methods automatically.
402
Mark Dickinsonc48d8342009-02-01 14:18:10 +0000403Note that the order in which the various test cases will be run is
404determined by sorting the test function names with respect to the
405built-in ordering for strings.
Georg Brandl116aa622007-08-15 14:28:22 +0000406
407Often it is desirable to group suites of test cases together, so as to run tests
408for the whole system at once. This is easy, since :class:`TestSuite` instances
409can be added to a :class:`TestSuite` just as :class:`TestCase` instances can be
410added to a :class:`TestSuite`::
411
412 suite1 = module1.TheTestSuite()
413 suite2 = module2.TheTestSuite()
414 alltests = unittest.TestSuite([suite1, suite2])
415
416You can place the definitions of test cases and test suites in the same modules
417as the code they are to test (such as :file:`widget.py`), but there are several
418advantages to placing the test code in a separate module, such as
419:file:`test_widget.py`:
420
421* The test module can be run standalone from the command line.
422
423* The test code can more easily be separated from shipped code.
424
425* There is less temptation to change test code to fit the code it tests without
426 a good reason.
427
428* Test code should be modified much less frequently than the code it tests.
429
430* Tested code can be refactored more easily.
431
432* Tests for modules written in C must be in separate modules anyway, so why not
433 be consistent?
434
435* If the testing strategy changes, there is no need to change the source code.
436
437
438.. _legacy-unit-tests:
439
440Re-using old test code
441----------------------
442
443Some users will find that they have existing test code that they would like to
444run from :mod:`unittest`, without converting every old test function to a
445:class:`TestCase` subclass.
446
447For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
448This subclass of :class:`TestCase` can be used to wrap an existing test
449function. Set-up and tear-down functions can also be provided.
450
451Given the following test function::
452
453 def testSomething():
454 something = makeSomething()
455 assert something.name is not None
456 # ...
457
458one can create an equivalent test case instance as follows::
459
460 testcase = unittest.FunctionTestCase(testSomething)
461
462If there are additional set-up and tear-down methods that should be called as
463part of the test case's operation, they can also be provided like so::
464
465 testcase = unittest.FunctionTestCase(testSomething,
466 setUp=makeSomethingDB,
467 tearDown=deleteSomethingDB)
468
469To make migrating existing test suites easier, :mod:`unittest` supports tests
470raising :exc:`AssertionError` to indicate test failure. However, it is
471recommended that you use the explicit :meth:`TestCase.fail\*` and
472:meth:`TestCase.assert\*` methods instead, as future versions of :mod:`unittest`
473may treat :exc:`AssertionError` differently.
474
475.. note::
476
Benjamin Petersond2397752009-06-27 23:45:02 +0000477 Even though :class:`FunctionTestCase` can be used to quickly convert an
478 existing test base over to a :mod:`unittest`\ -based system, this approach is
479 not recommended. Taking the time to set up proper :class:`TestCase`
480 subclasses will make future test refactorings infinitely easier.
Georg Brandl116aa622007-08-15 14:28:22 +0000481
Benjamin Peterson52baa292009-03-24 00:56:30 +0000482In some cases, the existing tests may have been written using the :mod:`doctest`
483module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
484automatically build :class:`unittest.TestSuite` instances from the existing
485:mod:`doctest`\ -based tests.
486
Georg Brandl116aa622007-08-15 14:28:22 +0000487
Benjamin Peterson5254c042009-03-23 22:25:03 +0000488.. _unittest-skipping:
489
490Skipping tests and expected failures
491------------------------------------
492
Michael Foordf5c851a2010-02-05 21:48:03 +0000493.. versionadded:: 3.1
494
Benjamin Peterson5254c042009-03-23 22:25:03 +0000495Unittest supports skipping individual test methods and even whole classes of
496tests. In addition, it supports marking a test as a "expected failure," a test
497that is broken and will fail, but shouldn't be counted as a failure on a
498:class:`TestResult`.
499
500Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
501or one of its conditional variants.
502
503Basic skipping looks like this: ::
504
505 class MyTestCase(unittest.TestCase):
506
507 @unittest.skip("demonstrating skipping")
508 def test_nothing(self):
509 self.fail("shouldn't happen")
510
Benjamin Petersond2397752009-06-27 23:45:02 +0000511 @unittest.skipIf(mylib.__version__ < (1, 3),
512 "not supported in this library version")
Benjamin Petersonded31c42009-03-30 15:04:16 +0000513 def test_format(self):
514 # Tests that work for only a certain version of the library.
515 pass
516
517 @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
518 def test_windows_support(self):
519 # windows specific testing code
520 pass
521
Benjamin Peterson5254c042009-03-23 22:25:03 +0000522This is the output of running the example above in verbose mode: ::
523
Benjamin Petersonded31c42009-03-30 15:04:16 +0000524 test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000525 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
Benjamin Petersonded31c42009-03-30 15:04:16 +0000526 test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000527
528 ----------------------------------------------------------------------
Benjamin Petersonded31c42009-03-30 15:04:16 +0000529 Ran 3 tests in 0.005s
530
531 OK (skipped=3)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000532
533Classes can be skipped just like methods: ::
534
535 @skip("showing class skipping")
536 class MySkippedTestCase(unittest.TestCase):
537 def test_not_run(self):
538 pass
539
Benjamin Peterson52baa292009-03-24 00:56:30 +0000540:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
541that needs to be set up is not available.
542
Benjamin Peterson5254c042009-03-23 22:25:03 +0000543Expected failures use the :func:`expectedFailure` decorator. ::
544
545 class ExpectedFailureTestCase(unittest.TestCase):
546 @unittest.expectedFailure
547 def test_fail(self):
548 self.assertEqual(1, 0, "broken")
549
550It's easy to roll your own skipping decorators by making a decorator that calls
551:func:`skip` on the test when it wants it to be skipped. This decorator skips
552the test unless the passed object has a certain attribute: ::
553
554 def skipUnlessHasattr(obj, attr):
555 if hasattr(obj, attr):
556 return lambda func: func
557 return unittest.skip("{0!r} doesn't have {1!r}".format(obj, attr))
558
559The following decorators implement test skipping and expected failures:
560
561.. function:: skip(reason)
562
563 Unconditionally skip the decorated test. *reason* should describe why the
564 test is being skipped.
565
566.. function:: skipIf(condition, reason)
567
568 Skip the decorated test if *condition* is true.
569
570.. function:: skipUnless(condition, reason)
571
572 Skip the decoratored test unless *condition* is true.
573
574.. function:: expectedFailure
575
576 Mark the test as an expected failure. If the test fails when run, the test
577 is not counted as a failure.
578
579
Georg Brandl116aa622007-08-15 14:28:22 +0000580.. _unittest-contents:
581
582Classes and functions
583---------------------
584
Benjamin Peterson52baa292009-03-24 00:56:30 +0000585This section describes in depth the API of :mod:`unittest`.
586
587
588.. _testcase-objects:
589
590Test cases
591~~~~~~~~~~
Georg Brandl116aa622007-08-15 14:28:22 +0000592
Georg Brandl7f01a132009-09-16 15:58:14 +0000593.. class:: TestCase(methodName='runTest')
Georg Brandl116aa622007-08-15 14:28:22 +0000594
595 Instances of the :class:`TestCase` class represent the smallest testable units
596 in the :mod:`unittest` universe. This class is intended to be used as a base
597 class, with specific tests being implemented by concrete subclasses. This class
598 implements the interface needed by the test runner to allow it to drive the
599 test, and methods that the test code can use to check for and report various
600 kinds of failure.
601
602 Each instance of :class:`TestCase` will run a single test method: the method
603 named *methodName*. If you remember, we had an earlier example that went
604 something like this::
605
606 def suite():
607 suite = unittest.TestSuite()
608 suite.addTest(WidgetTestCase('testDefaultSize'))
609 suite.addTest(WidgetTestCase('testResize'))
610 return suite
611
612 Here, we create two instances of :class:`WidgetTestCase`, each of which runs a
613 single test.
614
Benjamin Peterson52baa292009-03-24 00:56:30 +0000615 *methodName* defaults to :meth:`runTest`.
616
617 :class:`TestCase` instances provide three groups of methods: one group used
618 to run the test, another used by the test implementation to check conditions
619 and report failures, and some inquiry methods allowing information about the
620 test itself to be gathered.
621
622 Methods in the first group (running the test) are:
623
624
625 .. method:: setUp()
626
627 Method called to prepare the test fixture. This is called immediately
628 before calling the test method; any exception raised by this method will
629 be considered an error rather than a test failure. The default
630 implementation does nothing.
631
632
633 .. method:: tearDown()
634
635 Method called immediately after the test method has been called and the
636 result recorded. This is called even if the test method raised an
637 exception, so the implementation in subclasses may need to be particularly
638 careful about checking internal state. Any exception raised by this
639 method will be considered an error rather than a test failure. This
640 method will only be called if the :meth:`setUp` succeeds, regardless of
641 the outcome of the test method. The default implementation does nothing.
642
643
Georg Brandl7f01a132009-09-16 15:58:14 +0000644 .. method:: run(result=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000645
646 Run the test, collecting the result into the test result object passed as
647 *result*. If *result* is omitted or :const:`None`, a temporary result
Alexandre Vassalotti260484d2009-07-17 11:43:26 +0000648 object is created (by calling the :meth:`defaultTestResult` method) and
649 used. The result object is not returned to :meth:`run`'s caller.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000650
651 The same effect may be had by simply calling the :class:`TestCase`
652 instance.
653
654
Benjamin Petersone549ead2009-03-28 21:42:05 +0000655 .. method:: skipTest(reason)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000656
657 Calling this during the a test method or :meth:`setUp` skips the current
658 test. See :ref:`unittest-skipping` for more information.
659
660
661 .. method:: debug()
662
663 Run the test without collecting the result. This allows exceptions raised
664 by the test to be propagated to the caller, and can be used to support
665 running tests under a debugger.
666
667 The test code can use any of the following methods to check for and report
668 failures.
669
670
Georg Brandl7f01a132009-09-16 15:58:14 +0000671 .. method:: assertTrue(expr, msg=None)
672 assert_(expr, msg=None)
673 failUnless(expr, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000674
Georg Brandlff2ad0e2009-04-27 16:51:45 +0000675 Signal a test failure if *expr* is false; the explanation for the failure
Benjamin Peterson52baa292009-03-24 00:56:30 +0000676 will be *msg* if given, otherwise it will be :const:`None`.
677
Raymond Hettinger35a88362009-04-09 00:08:24 +0000678 .. deprecated:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000679 :meth:`failUnless`.
680
Benjamin Peterson52baa292009-03-24 00:56:30 +0000681
Georg Brandl7f01a132009-09-16 15:58:14 +0000682 .. method:: assertEqual(first, second, msg=None)
683 failUnlessEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000684
685 Test that *first* and *second* are equal. If the values do not compare
686 equal, the test will fail with the explanation given by *msg*, or
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000687 :const:`None`. Note that using :meth:`assertEqual` improves upon
688 doing the comparison as the first parameter to :meth:`assertTrue`: the
689 default value for *msg* include representations of both *first* and
690 *second*.
691
692 In addition, if *first* and *second* are the exact same type and one of
Michael Foord02834952010-02-08 23:10:39 +0000693 list, tuple, dict, set, frozenset or str or any type that a subclass
694 registers with :meth:`addTypeEqualityFunc` the type specific equality
695 function will be called in order to generate a more useful default
696 error message.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000697
Raymond Hettinger35a88362009-04-09 00:08:24 +0000698 .. versionchanged:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000699 Added the automatic calling of type specific equality function.
700
Michael Foord28a817e2010-02-09 00:03:57 +0000701 .. versionchanged:: 3.2
702 :meth:`assertMultiLineEqual` added as the default type equality
703 function for comparing strings.
Michael Foord02834952010-02-08 23:10:39 +0000704
Raymond Hettinger35a88362009-04-09 00:08:24 +0000705 .. deprecated:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000706 :meth:`failUnlessEqual`.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000707
708
Georg Brandl7f01a132009-09-16 15:58:14 +0000709 .. method:: assertNotEqual(first, second, msg=None)
710 failIfEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000711
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
Benjamin Peterson7fe73a12009-04-04 16:35:46 +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 Peterson52baa292009-03-24 00:56:30 +0000716 default value for *msg* can be computed to include representations of both
717 *first* and *second*.
718
Raymond Hettinger35a88362009-04-09 00:08:24 +0000719 .. deprecated:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000720 :meth:`failIfEqual`.
721
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000722
Georg Brandl7f01a132009-09-16 15:58:14 +0000723 .. method:: assertAlmostEqual(first, second, *, places=7, msg=None)
724 failUnlessAlmostEqual(first, second, *, places=7, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000725
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
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000735 .. versionchanged:: 3.2
736 Objects that compare equal are automatically almost equal.
737
Raymond Hettinger35a88362009-04-09 00:08:24 +0000738 .. deprecated:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000739 :meth:`failUnlessAlmostEqual`.
740
Benjamin Peterson52baa292009-03-24 00:56:30 +0000741
Georg Brandl7f01a132009-09-16 15:58:14 +0000742 .. method:: assertNotAlmostEqual(first, second, *, places=7, msg=None)
743 failIfAlmostEqual(first, second, *, places=7, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000744
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
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000754 .. versionchanged:: 3.2
755 Objects that compare equal automatically fail.
756
Raymond Hettinger35a88362009-04-09 00:08:24 +0000757 .. deprecated:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000758 :meth:`failIfAlmostEqual`.
759
760
761 .. 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
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000767 on the method name. If not, the test will fail with an explanation
Benjamin Peterson7fe73a12009-04-04 16:35:46 +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
Raymond Hettinger35a88362009-04-09 00:08:24 +0000773 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000774
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 Foord02834952010-02-08 23:10:39 +0000780 will be included in the error message. This method is used by default
781 when comparing strings with :meth:`assertEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000782
783 If specified *msg* will be used as the error message on failure.
784
Raymond Hettinger35a88362009-04-09 00:08:24 +0000785 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000786
787
Ezio Melotti732b6822010-01-16 19:40:06 +0000788 .. method:: assertRegexpMatches(text, regexp, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +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
Raymond Hettinger35a88362009-04-09 00:08:24 +0000795 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000796
797
798 .. method:: assertIn(first, second, msg=None)
799 assertNotIn(first, second, msg=None)
800
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000801 Tests that *first* is or is not in *second* with an explanatory error
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000802 message as appropriate.
803
804 If specified *msg* will be used as the error message on failure.
805
Raymond Hettinger35a88362009-04-09 00:08:24 +0000806 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000807
808
Michael Foorde9abbee2010-02-05 20:54:27 +0000809 .. method:: assertSameElements(actual, expected, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000810
Benjamin Peterson5e55b3e2010-02-03 02:35:45 +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.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000814
Michael Foorde9abbee2010-02-05 20:54:27 +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
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000819 If specified *msg* will be used as the error message on failure.
820
Raymond Hettinger35a88362009-04-09 00:08:24 +0000821 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000822
823
824 .. method:: assertSetEqual(set1, set2, msg=None)
825
826 Tests that two sets are equal. If not, an error message is constructed
Michael Foord02834952010-02-08 23:10:39 +0000827 that lists the differences between the sets. This method is used by
828 default when comparing sets or frozensets with :meth:`assertEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +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
Raymond Hettinger35a88362009-04-09 00:08:24 +0000835 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000836
837
838 .. method:: assertDictEqual(expected, actual, msg=None)
839
840 Test that two dictionaries are equal. If not, an error message is
Michael Foord02834952010-02-08 23:10:39 +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`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000844
845 If specified *msg* will be used as the error message on failure.
846
Raymond Hettinger35a88362009-04-09 00:08:24 +0000847 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000848
849
850 .. method:: assertDictContainsSubset(expected, actual, msg=None)
851
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000852 Tests whether the key/value pairs in dictionary *actual* are a
Benjamin Peterson7fe73a12009-04-04 16:35:46 +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
Raymond Hettinger35a88362009-04-09 00:08:24 +0000858 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000859
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 Foord02834952010-02-08 23:10:39 +0000867 These methods are used by default when comparing lists or tuples with
868 :meth:`assertEqual`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000869
870 If specified *msg* will be used as the error message on failure.
871
Raymond Hettinger35a88362009-04-09 00:08:24 +0000872 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000873
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
Raymond Hettinger35a88362009-04-09 00:08:24 +0000887 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000888
Benjamin Peterson52baa292009-03-24 00:56:30 +0000889
Georg Brandl7f01a132009-09-16 15:58:14 +0000890 .. method:: assertRaises(exception, callable, *args, **kwds)
891 failUnlessRaises(exception, callable, *args, **kwds)
892 assertRaises(exception)
893 failUnlessRaises(exception)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000894
895 Test that an exception is raised when *callable* is called with any
896 positional or keyword arguments that are also passed to
897 :meth:`assertRaises`. The test passes if *exception* is raised, is an
898 error if another exception is raised, or fails if no exception is raised.
899 To catch any of a group of exceptions, a tuple containing the exception
900 classes may be passed as *exception*.
901
Georg Brandl7f01a132009-09-16 15:58:14 +0000902 If only the *exception* argument is given, returns a context manager so
903 that the code under test can be written inline rather than as a function::
Benjamin Petersonded31c42009-03-30 15:04:16 +0000904
Michael Foord41531f22010-02-05 21:13:40 +0000905 with self.assertRaises(SomeException):
Benjamin Petersonded31c42009-03-30 15:04:16 +0000906 do_something()
907
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000908 The context manager will store the caught exception object in its
Ezio Melotti49008232010-02-08 21:57:48 +0000909 :attr:`exception` attribute. This can be useful if the intention
Michael Foord41531f22010-02-05 21:13:40 +0000910 is to perform additional checks on the exception raised::
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000911
Michael Foord41531f22010-02-05 21:13:40 +0000912 with self.assertRaises(SomeException) as cm:
913 do_something()
914
Ezio Melotti49008232010-02-08 21:57:48 +0000915 the_exception = cm.exception
Michael Foordb112a412010-02-05 23:32:33 +0000916 self.assertEqual(the_exception.error_code, 3)
Michael Foord41531f22010-02-05 21:13:40 +0000917
Ezio Melotti49008232010-02-08 21:57:48 +0000918 .. versionchanged:: 3.1
Benjamin Petersonded31c42009-03-30 15:04:16 +0000919 Added the ability to use :meth:`assertRaises` as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000920
Ezio Melotti49008232010-02-08 21:57:48 +0000921 .. versionchanged:: 3.2
922 Added the :attr:`exception` attribute.
923
Raymond Hettinger35a88362009-04-09 00:08:24 +0000924 .. deprecated:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000925 :meth:`failUnlessRaises`.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000926
Benjamin Peterson52baa292009-03-24 00:56:30 +0000927
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000928 .. method:: assertRaisesRegexp(exception, regexp[, callable, ...])
929
930 Like :meth:`assertRaises` but also tests that *regexp* matches
931 on the string representation of the raised exception. *regexp* may be
932 a regular expression object or a string containing a regular expression
933 suitable for use by :func:`re.search`. Examples::
934
935 self.assertRaisesRegexp(ValueError, 'invalid literal for.*XYZ$',
936 int, 'XYZ')
937
938 or::
939
940 with self.assertRaisesRegexp(ValueError, 'literal'):
941 int('XYZ')
942
Raymond Hettinger35a88362009-04-09 00:08:24 +0000943 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000944
945
Georg Brandl7f01a132009-09-16 15:58:14 +0000946 .. method:: assertIsNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000947
948 This signals a test failure if *expr* is not None.
949
Raymond Hettinger35a88362009-04-09 00:08:24 +0000950 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000951
952
Georg Brandl7f01a132009-09-16 15:58:14 +0000953 .. method:: assertIsNotNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000954
955 The inverse of the :meth:`assertIsNone` method.
956 This signals a test failure if *expr* is None.
957
Raymond Hettinger35a88362009-04-09 00:08:24 +0000958 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000959
960
Georg Brandl7f01a132009-09-16 15:58:14 +0000961 .. method:: assertIs(expr1, expr2, msg=None)
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000962
963 This signals a test failure if *expr1* and *expr2* don't evaluate to the same
964 object.
965
Georg Brandl705d9d52009-05-05 09:29:50 +0000966 .. versionadded:: 3.1
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000967
968
Georg Brandl7f01a132009-09-16 15:58:14 +0000969 .. method:: assertIsNot(expr1, expr2, msg=None)
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000970
971 The inverse of the :meth:`assertIs` method.
972 This signals a test failure if *expr1* and *expr2* evaluate to the same
973 object.
974
Georg Brandl705d9d52009-05-05 09:29:50 +0000975 .. versionadded:: 3.1
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +0000976
977
Benjamin Peterson6e8c7572009-10-04 20:19:21 +0000978 .. method:: assertIsInstance(obj, cls[, msg])
979
980 This signals a test failure if *obj* is not an instance of *cls* (which
981 can be a class or a tuple of classes, as supported by :func:`isinstance`).
982
983 .. versionadded:: 3.2
984
985
986 .. method:: assertNotIsInstance(obj, cls[, msg])
987
988 The inverse of the :meth:`assertIsInstance` method. This signals a test
989 failure if *obj* is an instance of *cls*.
990
991 .. versionadded:: 3.2
992
993
Georg Brandl7f01a132009-09-16 15:58:14 +0000994 .. method:: assertFalse(expr, msg=None)
995 failIf(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000996
997 The inverse of the :meth:`assertTrue` method is the :meth:`assertFalse` method.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000998 This signals a test failure if *expr* is true, with *msg* or :const:`None`
999 for the error message.
1000
Raymond Hettinger35a88362009-04-09 00:08:24 +00001001 .. deprecated:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001002 :meth:`failIf`.
1003
Benjamin Peterson52baa292009-03-24 00:56:30 +00001004
Georg Brandl7f01a132009-09-16 15:58:14 +00001005 .. method:: fail(msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001006
1007 Signals a test failure unconditionally, with *msg* or :const:`None` for
1008 the error message.
1009
1010
1011 .. attribute:: failureException
1012
1013 This class attribute gives the exception raised by the test method. If a
1014 test framework needs to use a specialized exception, possibly to carry
1015 additional information, it must subclass this exception in order to "play
1016 fair" with the framework. The initial value of this attribute is
1017 :exc:`AssertionError`.
1018
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001019
1020 .. attribute:: longMessage
1021
1022 If set to True then any explicit failure message you pass in to the
1023 assert methods will be appended to the end of the normal failure message.
1024 The normal messages contain useful information about the objects involved,
1025 for example the message from assertEqual shows you the repr of the two
1026 unequal objects. Setting this attribute to True allows you to have a
1027 custom error message in addition to the normal one.
1028
1029 This attribute defaults to False, meaning that a custom message passed
1030 to an assert method will silence the normal message.
1031
1032 The class setting can be overridden in individual tests by assigning an
1033 instance attribute to True or False before calling the assert methods.
1034
Raymond Hettinger35a88362009-04-09 00:08:24 +00001035 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001036
1037
Benjamin Peterson52baa292009-03-24 00:56:30 +00001038 Testing frameworks can use the following methods to collect information on
1039 the test:
1040
1041
1042 .. method:: countTestCases()
1043
1044 Return the number of tests represented by this test object. For
1045 :class:`TestCase` instances, this will always be ``1``.
1046
1047
1048 .. method:: defaultTestResult()
1049
1050 Return an instance of the test result class that should be used for this
1051 test case class (if no other result instance is provided to the
1052 :meth:`run` method).
1053
1054 For :class:`TestCase` instances, this will always be an instance of
1055 :class:`TestResult`; subclasses of :class:`TestCase` should override this
1056 as necessary.
1057
1058
1059 .. method:: id()
1060
1061 Return a string identifying the specific test case. This is usually the
1062 full name of the test method, including the module and class name.
1063
1064
1065 .. method:: shortDescription()
1066
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001067 Returns a description of the test, or :const:`None` if no description
1068 has been provided. The default implementation of this method
1069 returns the first line of the test method's docstring, if available,
1070 along with the method name.
1071
Raymond Hettinger35a88362009-04-09 00:08:24 +00001072 .. versionchanged:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001073 In earlier versions this only returned the first line of the test
1074 method's docstring, if available or the :const:`None`. That led to
1075 undesirable behavior of not printing the test name when someone was
1076 thoughtful enough to write a docstring.
1077
1078
1079 .. method:: addTypeEqualityFunc(typeobj, function)
1080
1081 Registers a type specific :meth:`assertEqual` equality checking
1082 function to be called by :meth:`assertEqual` when both objects it has
1083 been asked to compare are exactly *typeobj* (not subclasses).
1084 *function* must take two positional arguments and a third msg=None
1085 keyword argument just as :meth:`assertEqual` does. It must raise
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001086 ``self.failureException`` when inequality between the first two
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001087 parameters is detected.
1088
1089 One good use of custom equality checking functions for a type
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001090 is to raise ``self.failureException`` with an error message useful
1091 for debugging the problem by explaining the inequalities in detail.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001092
Raymond Hettinger35a88362009-04-09 00:08:24 +00001093 .. versionadded:: 3.1
Georg Brandl116aa622007-08-15 14:28:22 +00001094
1095
Georg Brandl7f01a132009-09-16 15:58:14 +00001096 .. method:: addCleanup(function, *args, **kwargs)
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001097
1098 Add a function to be called after :meth:`tearDown` to cleanup resources
1099 used during the test. Functions will be called in reverse order to the
1100 order they are added (LIFO). They are called with any arguments and
1101 keyword arguments passed into :meth:`addCleanup` when they are
1102 added.
1103
1104 If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
1105 then any cleanup functions added will still be called.
1106
Georg Brandl853947a2010-01-31 18:53:23 +00001107 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001108
1109
1110 .. method:: doCleanups()
1111
1112 This method is called uncoditionally after :meth:`tearDown`, or
1113 after :meth:`setUp` if :meth:`setUp` raises an exception.
1114
1115 It is responsible for calling all the cleanup functions added by
1116 :meth:`addCleanup`. If you need cleanup functions to be called
1117 *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
1118 yourself.
1119
1120 :meth:`doCleanups` pops methods off the stack of cleanup
1121 functions one at a time, so it can be called at any time.
1122
Georg Brandl853947a2010-01-31 18:53:23 +00001123 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001124
1125
Georg Brandl7f01a132009-09-16 15:58:14 +00001126.. class:: FunctionTestCase(testFunc, setUp=None, tearDown=None, description=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001127
1128 This class implements the portion of the :class:`TestCase` interface which
Benjamin Petersond2397752009-06-27 23:45:02 +00001129 allows the test runner to drive the test, but does not provide the methods
1130 which test code can use to check and report errors. This is used to create
1131 test cases using legacy test code, allowing it to be integrated into a
1132 :mod:`unittest`-based test framework.
Georg Brandl116aa622007-08-15 14:28:22 +00001133
1134
Benjamin Peterson52baa292009-03-24 00:56:30 +00001135.. _testsuite-objects:
1136
Benjamin Peterson52baa292009-03-24 00:56:30 +00001137Grouping tests
1138~~~~~~~~~~~~~~
1139
Georg Brandl7f01a132009-09-16 15:58:14 +00001140.. class:: TestSuite(tests=())
Georg Brandl116aa622007-08-15 14:28:22 +00001141
1142 This class represents an aggregation of individual tests cases and test suites.
1143 The class presents the interface needed by the test runner to allow it to be run
1144 as any other test case. Running a :class:`TestSuite` instance is the same as
1145 iterating over the suite, running each test individually.
1146
1147 If *tests* is given, it must be an iterable of individual test cases or other
1148 test suites that will be used to build the suite initially. Additional methods
1149 are provided to add test cases and suites to the collection later on.
1150
Benjamin Peterson14a3dd72009-05-25 00:51:58 +00001151 :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
1152 they do not actually implement a test. Instead, they are used to aggregate
1153 tests into groups of tests that should be run together. Some additional
1154 methods are available to add tests to :class:`TestSuite` instances:
Benjamin Peterson52baa292009-03-24 00:56:30 +00001155
1156
1157 .. method:: TestSuite.addTest(test)
1158
1159 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1160
1161
1162 .. method:: TestSuite.addTests(tests)
1163
1164 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
1165 instances to this test suite.
1166
Benjamin Petersond2397752009-06-27 23:45:02 +00001167 This is equivalent to iterating over *tests*, calling :meth:`addTest` for
1168 each element.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001169
1170 :class:`TestSuite` shares the following methods with :class:`TestCase`:
1171
1172
1173 .. method:: run(result)
1174
1175 Run the tests associated with this suite, collecting the result into the
1176 test result object passed as *result*. Note that unlike
1177 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
1178 be passed in.
1179
1180
1181 .. method:: debug()
1182
1183 Run the tests associated with this suite without collecting the
1184 result. This allows exceptions raised by the test to be propagated to the
1185 caller and can be used to support running tests under a debugger.
1186
1187
1188 .. method:: countTestCases()
1189
1190 Return the number of tests represented by this test object, including all
1191 individual tests and sub-suites.
1192
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001193
1194 .. method:: __iter__()
1195
1196 Tests grouped by a :class:`TestSuite` are always accessed by iteration.
1197 Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
1198 that this method maybe called several times on a single suite
1199 (for example when counting tests or comparing for equality)
1200 so the tests returned must be the same for repeated iterations.
1201
Georg Brandl853947a2010-01-31 18:53:23 +00001202 .. versionchanged:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001203 In earlier versions the :class:`TestSuite` accessed tests directly rather
1204 than through iteration, so overriding :meth:`__iter__` wasn't sufficient
1205 for providing tests.
1206
Benjamin Peterson52baa292009-03-24 00:56:30 +00001207 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
1208 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1209
1210
Benjamin Peterson52baa292009-03-24 00:56:30 +00001211Loading and running tests
1212~~~~~~~~~~~~~~~~~~~~~~~~~
1213
Georg Brandl116aa622007-08-15 14:28:22 +00001214.. class:: TestLoader()
1215
Benjamin Peterson52baa292009-03-24 00:56:30 +00001216 The :class:`TestLoader` class is used to create test suites from classes and
1217 modules. Normally, there is no need to create an instance of this class; the
1218 :mod:`unittest` module provides an instance that can be shared as
1219 ``unittest.defaultTestLoader``. Using a subclass or instance, however, allows
1220 customization of some configurable properties.
1221
1222 :class:`TestLoader` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001223
1224
Benjamin Peterson52baa292009-03-24 00:56:30 +00001225 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl116aa622007-08-15 14:28:22 +00001226
Benjamin Peterson52baa292009-03-24 00:56:30 +00001227 Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
1228 :class:`testCaseClass`.
1229
1230
1231 .. method:: loadTestsFromModule(module)
1232
1233 Return a suite of all tests cases contained in the given module. This
1234 method searches *module* for classes derived from :class:`TestCase` and
1235 creates an instance of the class for each test method defined for the
1236 class.
1237
Georg Brandle720c0a2009-04-27 16:20:50 +00001238 .. note::
Benjamin Peterson52baa292009-03-24 00:56:30 +00001239
1240 While using a hierarchy of :class:`TestCase`\ -derived classes can be
1241 convenient in sharing fixtures and helper functions, defining test
1242 methods on base classes that are not intended to be instantiated
1243 directly does not play well with this method. Doing so, however, can
1244 be useful when the fixtures are different and defined in subclasses.
1245
Benjamin Petersond2397752009-06-27 23:45:02 +00001246 If a module provides a ``load_tests`` function it will be called to
1247 load the tests. This allows modules to customize test loading.
1248 This is the `load_tests protocol`_.
1249
Georg Brandl853947a2010-01-31 18:53:23 +00001250 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001251 Support for ``load_tests`` added.
1252
Benjamin Peterson52baa292009-03-24 00:56:30 +00001253
Georg Brandl7f01a132009-09-16 15:58:14 +00001254 .. method:: loadTestsFromName(name, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001255
1256 Return a suite of all tests cases given a string specifier.
1257
1258 The specifier *name* is a "dotted name" that may resolve either to a
1259 module, a test case class, a test method within a test case class, a
1260 :class:`TestSuite` instance, or a callable object which returns a
1261 :class:`TestCase` or :class:`TestSuite` instance. These checks are
1262 applied in the order listed here; that is, a method on a possible test
1263 case class will be picked up as "a test method within a test case class",
1264 rather than "a callable object".
1265
1266 For example, if you have a module :mod:`SampleTests` containing a
1267 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
1268 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
Benjamin Petersond2397752009-06-27 23:45:02 +00001269 specifier ``'SampleTests.SampleTestCase'`` would cause this method to
1270 return a suite which will run all three test methods. Using the specifier
1271 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
1272 suite which will run only the :meth:`test_two` test method. The specifier
1273 can refer to modules and packages which have not been imported; they will
1274 be imported as a side-effect.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001275
1276 The method optionally resolves *name* relative to the given *module*.
1277
1278
Georg Brandl7f01a132009-09-16 15:58:14 +00001279 .. method:: loadTestsFromNames(names, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001280
1281 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
1282 than a single name. The return value is a test suite which supports all
1283 the tests defined for each name.
1284
1285
1286 .. method:: getTestCaseNames(testCaseClass)
1287
1288 Return a sorted sequence of method names found within *testCaseClass*;
1289 this should be a subclass of :class:`TestCase`.
1290
Benjamin Petersond2397752009-06-27 23:45:02 +00001291
1292 .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)
1293
1294 Find and return all test modules from the specified start directory,
1295 recursing into subdirectories to find them. Only test files that match
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001296 *pattern* will be loaded. (Using shell style pattern matching.) Only
1297 module names that are importable (i.e. are valid Python identifiers) will
1298 be loaded.
Benjamin Petersond2397752009-06-27 23:45:02 +00001299
1300 All test modules must be importable from the top level of the project. If
1301 the start directory is not the top level directory then the top level
1302 directory must be specified separately.
1303
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001304 If importing a module fails, for example due to a syntax error, then this
1305 will be recorded as a single error and discovery will continue.
1306
Benjamin Petersond2397752009-06-27 23:45:02 +00001307 If a test package name (directory with :file:`__init__.py`) matches the
1308 pattern then the package will be checked for a ``load_tests``
1309 function. If this exists then it will be called with *loader*, *tests*,
1310 *pattern*.
1311
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001312 If load_tests exists then discovery does *not* recurse into the package,
Benjamin Petersond2397752009-06-27 23:45:02 +00001313 ``load_tests`` is responsible for loading all tests in the package.
1314
1315 The pattern is deliberately not stored as a loader attribute so that
1316 packages can continue discovery themselves. *top_level_dir* is stored so
1317 ``load_tests`` does not need to pass this argument in to
1318 ``loader.discover()``.
1319
Georg Brandl853947a2010-01-31 18:53:23 +00001320 .. versionadded:: 3.2
1321
Benjamin Petersond2397752009-06-27 23:45:02 +00001322
Benjamin Peterson52baa292009-03-24 00:56:30 +00001323 The following attributes of a :class:`TestLoader` can be configured either by
1324 subclassing or assignment on an instance:
1325
1326
1327 .. attribute:: testMethodPrefix
1328
1329 String giving the prefix of method names which will be interpreted as test
1330 methods. The default value is ``'test'``.
1331
1332 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
1333 methods.
1334
1335
1336 .. attribute:: sortTestMethodsUsing
1337
1338 Function to be used to compare method names when sorting them in
1339 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
1340
1341
1342 .. attribute:: suiteClass
1343
1344 Callable object that constructs a test suite from a list of tests. No
1345 methods on the resulting object are needed. The default value is the
1346 :class:`TestSuite` class.
1347
1348 This affects all the :meth:`loadTestsFrom\*` methods.
1349
1350
Benjamin Peterson52baa292009-03-24 00:56:30 +00001351.. class:: TestResult
1352
1353 This class is used to compile information about which tests have succeeded
1354 and which have failed.
1355
1356 A :class:`TestResult` object stores the results of a set of tests. The
1357 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
1358 properly recorded; test authors do not need to worry about recording the
1359 outcome of tests.
1360
1361 Testing frameworks built on top of :mod:`unittest` may want access to the
1362 :class:`TestResult` object generated by running a set of tests for reporting
1363 purposes; a :class:`TestResult` instance is returned by the
1364 :meth:`TestRunner.run` method for this purpose.
1365
1366 :class:`TestResult` instances have the following attributes that will be of
1367 interest when inspecting the results of running a set of tests:
1368
1369
1370 .. attribute:: errors
1371
1372 A list containing 2-tuples of :class:`TestCase` instances and strings
1373 holding formatted tracebacks. Each tuple represents a test which raised an
1374 unexpected exception.
1375
Benjamin Peterson52baa292009-03-24 00:56:30 +00001376 .. 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
Benjamin Peterson52baa292009-03-24 00:56:30 +00001383 .. attribute:: skipped
1384
1385 A list containing 2-tuples of :class:`TestCase` instances and strings
1386 holding the reason for skipping the test.
1387
Benjamin Peterson70e32c82009-03-24 01:00:11 +00001388 .. versionadded:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +00001389
1390 .. attribute:: expectedFailures
1391
1392 A list contaning 2-tuples of :class:`TestCase` instances and strings
1393 holding formatted tracebacks. Each tuple represents a expected failures
1394 of the test case.
1395
1396 .. attribute:: unexpectedSuccesses
1397
1398 A list containing :class:`TestCase` instances that were marked as expected
1399 failures, but succeeded.
1400
1401 .. attribute:: shouldStop
1402
1403 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1404
1405
1406 .. attribute:: testsRun
1407
1408 The total number of tests run so far.
1409
1410
1411 .. method:: wasSuccessful()
1412
1413 Return :const:`True` if all tests run so far have passed, otherwise returns
1414 :const:`False`.
1415
1416
1417 .. method:: stop()
1418
1419 This method can be called to signal that the set of tests being run should
1420 be aborted by setting the :attr:`shouldStop` attribute to :const:`True`.
1421 :class:`TestRunner` objects should respect this flag and return without
1422 running any additional tests.
1423
1424 For example, this feature is used by the :class:`TextTestRunner` class to
1425 stop the test framework when the user signals an interrupt from the
1426 keyboard. Interactive tools which provide :class:`TestRunner`
1427 implementations can use this in a similar manner.
1428
1429 The following methods of the :class:`TestResult` class are used to maintain
1430 the internal data structures, and may be extended in subclasses to support
1431 additional reporting requirements. This is particularly useful in building
1432 tools which support interactive reporting while tests are being run.
1433
1434
1435 .. method:: startTest(test)
1436
1437 Called when the test case *test* is about to be run.
1438
1439 The default implementation simply increments the instance's :attr:`testsRun`
1440 counter.
1441
1442
1443 .. method:: stopTest(test)
1444
1445 Called after the test case *test* has been executed, regardless of the
1446 outcome.
1447
1448 The default implementation does nothing.
1449
1450
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001451 .. method:: startTestRun(test)
1452
1453 Called once before any tests are executed.
1454
Georg Brandl853947a2010-01-31 18:53:23 +00001455 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001456
1457
1458 .. method:: stopTestRun(test)
1459
Ezio Melotti176d6c42010-01-27 20:58:07 +00001460 Called once after all tests are executed.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001461
Georg Brandl853947a2010-01-31 18:53:23 +00001462 .. versionadded:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001463
1464
Benjamin Peterson52baa292009-03-24 00:56:30 +00001465 .. method:: addError(test, err)
1466
1467 Called when the test case *test* raises an unexpected exception *err* is a
1468 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1469 traceback)``.
1470
1471 The default implementation appends a tuple ``(test, formatted_err)`` to
1472 the instance's :attr:`errors` attribute, where *formatted_err* is a
1473 formatted traceback derived from *err*.
1474
1475
1476 .. method:: addFailure(test, err)
1477
Benjamin Petersond2397752009-06-27 23:45:02 +00001478 Called when the test case *test* signals a failure. *err* is a tuple of
1479 the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001480
1481 The default implementation appends a tuple ``(test, formatted_err)`` to
1482 the instance's :attr:`failures` attribute, where *formatted_err* is a
1483 formatted traceback derived from *err*.
1484
1485
1486 .. method:: addSuccess(test)
1487
1488 Called when the test case *test* succeeds.
1489
1490 The default implementation does nothing.
1491
1492
1493 .. method:: addSkip(test, reason)
1494
1495 Called when the test case *test* is skipped. *reason* is the reason the
1496 test gave for skipping.
1497
1498 The default implementation appends a tuple ``(test, reason)`` to the
1499 instance's :attr:`skipped` attribute.
1500
1501
1502 .. method:: addExpectedFailure(test, err)
1503
1504 Called when the test case *test* fails, but was marked with the
1505 :func:`expectedFailure` decorator.
1506
1507 The default implementation appends a tuple ``(test, formatted_err)`` to
1508 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1509 is a formatted traceback derived from *err*.
1510
1511
1512 .. method:: addUnexpectedSuccess(test)
1513
1514 Called when the test case *test* was marked with the
1515 :func:`expectedFailure` decorator, but succeeded.
1516
1517 The default implementation appends the test to the instance's
1518 :attr:`unexpectedSuccesses` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +00001519
1520
1521.. data:: defaultTestLoader
1522
1523 Instance of the :class:`TestLoader` class intended to be shared. If no
1524 customization of the :class:`TestLoader` is needed, this instance can be used
1525 instead of repeatedly creating new instances.
1526
1527
Georg Brandl7f01a132009-09-16 15:58:14 +00001528.. class:: TextTestRunner(stream=sys.stderr, descriptions=True, verbosity=1)
Georg Brandl116aa622007-08-15 14:28:22 +00001529
1530 A basic test runner implementation which prints results on standard error. It
1531 has a few configurable parameters, but is essentially very simple. Graphical
1532 applications which run test suites should provide alternate implementations.
1533
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001534 .. method:: _makeResult()
Georg Brandl116aa622007-08-15 14:28:22 +00001535
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001536 This method returns the instance of ``TestResult`` used by :meth:`run`.
1537 It is not intended to be called directly, but can be overridden in
1538 subclasses to provide a custom ``TestResult``.
1539
1540
Georg Brandl7f01a132009-09-16 15:58:14 +00001541.. function:: main(module='__main__', defaultTest=None, argv=None, testRunner=None, testLoader=unittest.loader.defaultTestLoader, exit=True, verbosity=1)
Georg Brandl116aa622007-08-15 14:28:22 +00001542
1543 A command-line program that runs a set of tests; this is primarily for making
1544 test modules conveniently executable. The simplest use for this function is to
1545 include the following line at the end of a test script::
1546
1547 if __name__ == '__main__':
1548 unittest.main()
1549
Benjamin Petersond2397752009-06-27 23:45:02 +00001550 You can run tests with more detailed information by passing in the verbosity
1551 argument::
1552
1553 if __name__ == '__main__':
1554 unittest.main(verbosity=2)
1555
Georg Brandl116aa622007-08-15 14:28:22 +00001556 The *testRunner* argument can either be a test runner class or an already
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001557 created instance of it. By default ``main`` calls :func:`sys.exit` with
1558 an exit code indicating success or failure of the tests run.
1559
1560 ``main`` supports being used from the interactive interpreter by passing in the
1561 argument ``exit=False``. This displays the result on standard output without
1562 calling :func:`sys.exit`::
1563
1564 >>> from unittest import main
1565 >>> main(module='test_module', exit=False)
1566
1567 Calling ``main`` actually returns an instance of the ``TestProgram`` class.
1568 This stores the result of the tests run as the ``result`` attribute.
1569
Georg Brandl853947a2010-01-31 18:53:23 +00001570 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001571 The ``exit`` and ``verbosity`` parameters were added.
1572
1573
1574load_tests Protocol
1575###################
1576
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001577
Georg Brandl853947a2010-01-31 18:53:23 +00001578.. versionadded:: 3.2
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001579
1580
Benjamin Petersond2397752009-06-27 23:45:02 +00001581Modules or packages can customize how tests are loaded from them during normal
1582test runs or test discovery by implementing a function called ``load_tests``.
1583
1584If a test module defines ``load_tests`` it will be called by
1585:meth:`TestLoader.loadTestsFromModule` with the following arguments::
1586
1587 load_tests(loader, standard_tests, None)
1588
1589It should return a :class:`TestSuite`.
1590
1591*loader* is the instance of :class:`TestLoader` doing the loading.
1592*standard_tests* are the tests that would be loaded by default from the
1593module. It is common for test modules to only want to add or remove tests
1594from the standard set of tests.
1595The third argument is used when loading packages as part of test discovery.
1596
1597A typical ``load_tests`` function that loads tests from a specific set of
1598:class:`TestCase` classes may look like::
1599
1600 test_cases = (TestCase1, TestCase2, TestCase3)
1601
1602 def load_tests(loader, tests, pattern):
1603 suite = TestSuite()
1604 for test_class in test_cases:
1605 tests = loader.loadTestsFromTestCase(test_class)
1606 suite.addTests(tests)
1607 return suite
1608
1609If discovery is started, either from the command line or by calling
1610:meth:`TestLoader.discover`, with a pattern that matches a package
1611name then the package :file:`__init__.py` will be checked for ``load_tests``.
1612
1613.. note::
1614
Ezio Melotti0639d5a2009-12-19 23:26:38 +00001615 The default pattern is 'test*.py'. This matches all Python files
Benjamin Petersond2397752009-06-27 23:45:02 +00001616 that start with 'test' but *won't* match any test directories.
1617
1618 A pattern like 'test*' will match test packages as well as
1619 modules.
1620
1621If the package :file:`__init__.py` defines ``load_tests`` then it will be
1622called and discovery not continued into the package. ``load_tests``
1623is called with the following arguments::
1624
1625 load_tests(loader, standard_tests, pattern)
1626
1627This should return a :class:`TestSuite` representing all the tests
1628from the package. (``standard_tests`` will only contain tests
1629collected from :file:`__init__.py`.)
1630
1631Because the pattern is passed into ``load_tests`` the package is free to
1632continue (and potentially modify) test discovery. A 'do nothing'
1633``load_tests`` function for a test package would look like::
1634
1635 def load_tests(loader, standard_tests, pattern):
1636 # top level directory cached on loader instance
1637 this_dir = os.path.dirname(__file__)
1638 package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
1639 standard_tests.addTests(package_tests)
1640 return standard_tests