blob: 22bf7f37518d7eda7daaf564c6c0e11a0985dbe0 [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001
2:mod:`unittest` --- Unit testing framework
3==========================================
4
5.. module:: unittest
6 :synopsis: Unit testing framework for Python.
7.. moduleauthor:: Steve Purcell <stephen_purcell@yahoo.com>
8.. sectionauthor:: Steve Purcell <stephen_purcell@yahoo.com>
9.. sectionauthor:: Fred L. Drake, Jr. <fdrake@acm.org>
10.. sectionauthor:: Raymond Hettinger <python@rcn.com>
11
12
Benjamin Peterson5254c042009-03-23 22:25:03 +000013.. versionchanged:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +000014 Added test :ref:`skipping and expected failures <unittest-skipping>`.
Benjamin Peterson5254c042009-03-23 22:25:03 +000015
Georg Brandl116aa622007-08-15 14:28:22 +000016The Python unit testing framework, sometimes referred to as "PyUnit," is a
17Python language version of JUnit, by Kent Beck and Erich Gamma. JUnit is, in
18turn, a Java version of Kent's Smalltalk testing framework. Each is the de
19facto standard unit testing framework for its respective language.
20
21:mod:`unittest` supports test automation, sharing of setup and shutdown code for
22tests, aggregation of tests into collections, and independence of the tests from
23the reporting framework. The :mod:`unittest` module provides classes that make
24it easy to support these qualities for a set of tests.
25
26To achieve this, :mod:`unittest` supports some important concepts:
27
28test fixture
29 A :dfn:`test fixture` represents the preparation needed to perform one or more
30 tests, and any associate cleanup actions. This may involve, for example,
31 creating temporary or proxy databases, directories, or starting a server
32 process.
33
34test case
35 A :dfn:`test case` is the smallest unit of testing. It checks for a specific
36 response to a particular set of inputs. :mod:`unittest` provides a base class,
37 :class:`TestCase`, which may be used to create new test cases.
38
39test suite
40 A :dfn:`test suite` is a collection of test cases, test suites, or both. It is
41 used to aggregate tests that should be executed together.
42
43test runner
44 A :dfn:`test runner` is a component which orchestrates the execution of tests
45 and provides the outcome to the user. The runner may use a graphical interface,
46 a textual interface, or return a special value to indicate the results of
47 executing the tests.
48
49The test case and test fixture concepts are supported through the
50:class:`TestCase` and :class:`FunctionTestCase` classes; the former should be
51used when creating new tests, and the latter can be used when integrating
52existing test code with a :mod:`unittest`\ -driven framework. When building test
Benjamin Peterson52baa292009-03-24 00:56:30 +000053fixtures using :class:`TestCase`, the :meth:`~TestCase.setUp` and
54:meth:`~TestCase.tearDown` methods can be overridden to provide initialization
55and cleanup for the fixture. With :class:`FunctionTestCase`, existing functions
56can be passed to the constructor for these purposes. When the test is run, the
57fixture initialization is run first; if it succeeds, the cleanup method is run
58after the test has been executed, regardless of the outcome of the test. Each
59instance of the :class:`TestCase` will only be used to run a single test method,
60so a new fixture is created for each test.
Georg Brandl116aa622007-08-15 14:28:22 +000061
62Test suites are implemented by the :class:`TestSuite` class. This class allows
63individual tests and test suites to be aggregated; when the suite is executed,
Benjamin Peterson5254c042009-03-23 22:25:03 +000064all tests added directly to the suite and in "child" test suites are run. A
65:class:`ClassTestSuite` contains the test cases of a class.
Georg Brandl116aa622007-08-15 14:28:22 +000066
Benjamin Peterson52baa292009-03-24 00:56:30 +000067A test runner is an object that provides a single method,
68:meth:`~TestRunner.run`, which accepts a :class:`TestCase` or :class:`TestSuite`
69object as a parameter, and returns a result object. The class
70:class:`TestResult` is provided for use as the result object. :mod:`unittest`
71provides the :class:`TextTestRunner` as an example test runner which reports
72test results on the standard error stream by default. Alternate runners can be
73implemented for other environments (such as graphical environments) without any
74need to derive from a specific class.
Georg Brandl116aa622007-08-15 14:28:22 +000075
76
77.. seealso::
78
79 Module :mod:`doctest`
80 Another test-support module with a very different flavor.
81
82 `Simple Smalltalk Testing: With Patterns <http://www.XProgramming.com/testfram.htm>`_
83 Kent Beck's original paper on testing frameworks using the pattern shared by
84 :mod:`unittest`.
85
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000086 `Nose <http://code.google.com/p/python-nose/>`_ and `py.test <http://pytest.org>`_
87 Third-party unittest frameworks with a lighter-weight syntax
88 for writing tests. For example, ``assert func(10) == 42``.
89
90 `python-mock <http://python-mock.sourceforge.net/>`_ and `minimock <http://blog.ianbicking.org/minimock.html>`_
91 Tools for creating mock test objects (objects simulating external resources).
Georg Brandl116aa622007-08-15 14:28:22 +000092
93.. _unittest-minimal-example:
94
95Basic example
96-------------
97
98The :mod:`unittest` module provides a rich set of tools for constructing and
99running tests. This section demonstrates that a small subset of the tools
100suffice to meet the needs of most users.
101
102Here is a short script to test three functions from the :mod:`random` module::
103
104 import random
105 import unittest
106
107 class TestSequenceFunctions(unittest.TestCase):
108
109 def setUp(self):
110 self.seq = range(10)
111
Benjamin Peterson52baa292009-03-24 00:56:30 +0000112 def test_shuffle(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000113 # make sure the shuffled sequence does not lose any elements
114 random.shuffle(self.seq)
115 self.seq.sort()
116 self.assertEqual(self.seq, range(10))
117
Benjamin Peterson52baa292009-03-24 00:56:30 +0000118 def test_choice(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000119 element = random.choice(self.seq)
120 self.assert_(element in self.seq)
121
Benjamin Peterson52baa292009-03-24 00:56:30 +0000122 def test_sample(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000123 self.assertRaises(ValueError, random.sample, self.seq, 20)
124 for element in random.sample(self.seq, 5):
125 self.assert_(element in self.seq)
126
127 if __name__ == '__main__':
128 unittest.main()
129
Benjamin Peterson52baa292009-03-24 00:56:30 +0000130A testcase is created by subclassing :class:`unittest.TestCase`. The three
Georg Brandl116aa622007-08-15 14:28:22 +0000131individual tests are defined with methods whose names start with the letters
132``test``. This naming convention informs the test runner about which methods
133represent tests.
134
Benjamin Peterson52baa292009-03-24 00:56:30 +0000135The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
136expected result; :meth:`~TestCase.assert_` to verify a condition; or
137:meth:`~TestCase.assertRaises` to verify that an expected exception gets raised.
138These methods are used instead of the :keyword:`assert` statement so the test
139runner can accumulate all test results and produce a report.
Georg Brandl116aa622007-08-15 14:28:22 +0000140
Benjamin Peterson52baa292009-03-24 00:56:30 +0000141When a :meth:`~TestCase.setUp` method is defined, the test runner will run that
142method prior to each test. Likewise, if a :meth:`~TestCase.tearDown` method is
143defined, the test runner will invoke that method after each test. In the
144example, :meth:`~TestCase.setUp` was used to create a fresh sequence for each
145test.
Georg Brandl116aa622007-08-15 14:28:22 +0000146
147The final block shows a simple way to run the tests. :func:`unittest.main`
148provides a command line interface to the test script. When run from the command
149line, the above script produces an output that looks like this::
150
151 ...
152 ----------------------------------------------------------------------
153 Ran 3 tests in 0.000s
154
155 OK
156
157Instead of :func:`unittest.main`, there are other ways to run the tests with a
158finer level of control, less terse output, and no requirement to be run from the
159command line. For example, the last two lines may be replaced with::
160
161 suite = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)
162 unittest.TextTestRunner(verbosity=2).run(suite)
163
164Running the revised script from the interpreter or another script produces the
165following output::
166
167 testchoice (__main__.TestSequenceFunctions) ... ok
168 testsample (__main__.TestSequenceFunctions) ... ok
169 testshuffle (__main__.TestSequenceFunctions) ... ok
170
171 ----------------------------------------------------------------------
172 Ran 3 tests in 0.110s
173
174 OK
175
176The above examples show the most commonly used :mod:`unittest` features which
177are sufficient to meet many everyday testing needs. The remainder of the
178documentation explores the full feature set from first principles.
179
180
181.. _organizing-tests:
182
183Organizing test code
184--------------------
185
186The basic building blocks of unit testing are :dfn:`test cases` --- single
187scenarios that must be set up and checked for correctness. In :mod:`unittest`,
188test cases are represented by instances of :mod:`unittest`'s :class:`TestCase`
189class. To make your own test cases you must write subclasses of
190:class:`TestCase`, or use :class:`FunctionTestCase`.
191
192An instance of a :class:`TestCase`\ -derived class is an object that can
193completely run a single test method, together with optional set-up and tidy-up
194code.
195
196The testing code of a :class:`TestCase` instance should be entirely self
197contained, such that it can be run either in isolation or in arbitrary
198combination with any number of other test cases.
199
Benjamin Peterson52baa292009-03-24 00:56:30 +0000200The simplest :class:`TestCase` subclass will simply override the
201:meth:`~TestCase.runTest` method in order to perform specific testing code::
Georg Brandl116aa622007-08-15 14:28:22 +0000202
203 import unittest
204
205 class DefaultWidgetSizeTestCase(unittest.TestCase):
206 def runTest(self):
207 widget = Widget('The widget')
208 self.assertEqual(widget.size(), (50, 50), 'incorrect default size')
209
210Note that in order to test something, we use the one of the :meth:`assert\*` or
211:meth:`fail\*` methods provided by the :class:`TestCase` base class. If the
212test fails, an exception will be raised, and :mod:`unittest` will identify the
213test case as a :dfn:`failure`. Any other exceptions will be treated as
214:dfn:`errors`. This helps you identify where the problem is: :dfn:`failures` are
215caused by incorrect results - a 5 where you expected a 6. :dfn:`Errors` are
216caused by incorrect code - e.g., a :exc:`TypeError` caused by an incorrect
217function call.
218
219The way to run a test case will be described later. For now, note that to
220construct an instance of such a test case, we call its constructor without
221arguments::
222
223 testCase = DefaultWidgetSizeTestCase()
224
225Now, such test cases can be numerous, and their set-up can be repetitive. In
226the above case, constructing a :class:`Widget` in each of 100 Widget test case
227subclasses would mean unsightly duplication.
228
229Luckily, we can factor out such set-up code by implementing a method called
Benjamin Peterson52baa292009-03-24 00:56:30 +0000230:meth:`~TestCase.setUp`, which the testing framework will automatically call for
231us when we run the test::
Georg Brandl116aa622007-08-15 14:28:22 +0000232
233 import unittest
234
235 class SimpleWidgetTestCase(unittest.TestCase):
236 def setUp(self):
237 self.widget = Widget('The widget')
238
239 class DefaultWidgetSizeTestCase(SimpleWidgetTestCase):
240 def runTest(self):
241 self.failUnless(self.widget.size() == (50,50),
242 'incorrect default size')
243
244 class WidgetResizeTestCase(SimpleWidgetTestCase):
245 def runTest(self):
246 self.widget.resize(100,150)
247 self.failUnless(self.widget.size() == (100,150),
248 'wrong size after resize')
249
Benjamin Peterson52baa292009-03-24 00:56:30 +0000250If the :meth:`~TestCase.setUp` method raises an exception while the test is
251running, the framework will consider the test to have suffered an error, and the
252:meth:`~TestCase.runTest` method will not be executed.
Georg Brandl116aa622007-08-15 14:28:22 +0000253
Benjamin Peterson52baa292009-03-24 00:56:30 +0000254Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
255after the :meth:`~TestCase.runTest` method has been run::
Georg Brandl116aa622007-08-15 14:28:22 +0000256
257 import unittest
258
259 class SimpleWidgetTestCase(unittest.TestCase):
260 def setUp(self):
261 self.widget = Widget('The widget')
262
263 def tearDown(self):
264 self.widget.dispose()
265 self.widget = None
266
Benjamin Peterson52baa292009-03-24 00:56:30 +0000267If :meth:`~TestCase.setUp` succeeded, the :meth:`~TestCase.tearDown` method will
268be run whether :meth:`~TestCase.runTest` succeeded or not.
Georg Brandl116aa622007-08-15 14:28:22 +0000269
270Such a working environment for the testing code is called a :dfn:`fixture`.
271
272Often, many small test cases will use the same fixture. In this case, we would
273end up subclassing :class:`SimpleWidgetTestCase` into many small one-method
274classes such as :class:`DefaultWidgetSizeTestCase`. This is time-consuming and
Georg Brandl116aa622007-08-15 14:28:22 +0000275discouraging, so in the same vein as JUnit, :mod:`unittest` provides a simpler
276mechanism::
277
278 import unittest
279
280 class WidgetTestCase(unittest.TestCase):
281 def setUp(self):
282 self.widget = Widget('The widget')
283
284 def tearDown(self):
285 self.widget.dispose()
286 self.widget = None
287
288 def testDefaultSize(self):
289 self.failUnless(self.widget.size() == (50,50),
290 'incorrect default size')
291
292 def testResize(self):
293 self.widget.resize(100,150)
294 self.failUnless(self.widget.size() == (100,150),
295 'wrong size after resize')
296
Benjamin Peterson52baa292009-03-24 00:56:30 +0000297Here we have not provided a :meth:`~TestCase.runTest` method, but have instead
298provided two different test methods. Class instances will now each run one of
299the :meth:`test\*` methods, with ``self.widget`` created and destroyed
300separately for each instance. When creating an instance we must specify the
301test method it is to run. We do this by passing the method name in the
302constructor::
Georg Brandl116aa622007-08-15 14:28:22 +0000303
304 defaultSizeTestCase = WidgetTestCase('testDefaultSize')
305 resizeTestCase = WidgetTestCase('testResize')
306
307Test case instances are grouped together according to the features they test.
308:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
309represented by :mod:`unittest`'s :class:`TestSuite` class::
310
311 widgetTestSuite = unittest.TestSuite()
312 widgetTestSuite.addTest(WidgetTestCase('testDefaultSize'))
313 widgetTestSuite.addTest(WidgetTestCase('testResize'))
314
315For the ease of running tests, as we will see later, it is a good idea to
316provide in each test module a callable object that returns a pre-built test
317suite::
318
319 def suite():
320 suite = unittest.TestSuite()
321 suite.addTest(WidgetTestCase('testDefaultSize'))
322 suite.addTest(WidgetTestCase('testResize'))
323 return suite
324
325or even::
326
327 def suite():
328 tests = ['testDefaultSize', 'testResize']
329
330 return unittest.TestSuite(map(WidgetTestCase, tests))
331
332Since it is a common pattern to create a :class:`TestCase` subclass with many
333similarly named test functions, :mod:`unittest` provides a :class:`TestLoader`
334class that can be used to automate the process of creating a test suite and
335populating it with individual tests. For example, ::
336
337 suite = unittest.TestLoader().loadTestsFromTestCase(WidgetTestCase)
338
339will create a test suite that will run ``WidgetTestCase.testDefaultSize()`` and
340``WidgetTestCase.testResize``. :class:`TestLoader` uses the ``'test'`` method
341name prefix to identify test methods automatically.
342
Mark Dickinsonc48d8342009-02-01 14:18:10 +0000343Note that the order in which the various test cases will be run is
344determined by sorting the test function names with respect to the
345built-in ordering for strings.
Georg Brandl116aa622007-08-15 14:28:22 +0000346
347Often it is desirable to group suites of test cases together, so as to run tests
348for the whole system at once. This is easy, since :class:`TestSuite` instances
349can be added to a :class:`TestSuite` just as :class:`TestCase` instances can be
350added to a :class:`TestSuite`::
351
352 suite1 = module1.TheTestSuite()
353 suite2 = module2.TheTestSuite()
354 alltests = unittest.TestSuite([suite1, suite2])
355
356You can place the definitions of test cases and test suites in the same modules
357as the code they are to test (such as :file:`widget.py`), but there are several
358advantages to placing the test code in a separate module, such as
359:file:`test_widget.py`:
360
361* The test module can be run standalone from the command line.
362
363* The test code can more easily be separated from shipped code.
364
365* There is less temptation to change test code to fit the code it tests without
366 a good reason.
367
368* Test code should be modified much less frequently than the code it tests.
369
370* Tested code can be refactored more easily.
371
372* Tests for modules written in C must be in separate modules anyway, so why not
373 be consistent?
374
375* If the testing strategy changes, there is no need to change the source code.
376
377
378.. _legacy-unit-tests:
379
380Re-using old test code
381----------------------
382
383Some users will find that they have existing test code that they would like to
384run from :mod:`unittest`, without converting every old test function to a
385:class:`TestCase` subclass.
386
387For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
388This subclass of :class:`TestCase` can be used to wrap an existing test
389function. Set-up and tear-down functions can also be provided.
390
391Given the following test function::
392
393 def testSomething():
394 something = makeSomething()
395 assert something.name is not None
396 # ...
397
398one can create an equivalent test case instance as follows::
399
400 testcase = unittest.FunctionTestCase(testSomething)
401
402If there are additional set-up and tear-down methods that should be called as
403part of the test case's operation, they can also be provided like so::
404
405 testcase = unittest.FunctionTestCase(testSomething,
406 setUp=makeSomethingDB,
407 tearDown=deleteSomethingDB)
408
409To make migrating existing test suites easier, :mod:`unittest` supports tests
410raising :exc:`AssertionError` to indicate test failure. However, it is
411recommended that you use the explicit :meth:`TestCase.fail\*` and
412:meth:`TestCase.assert\*` methods instead, as future versions of :mod:`unittest`
413may treat :exc:`AssertionError` differently.
414
415.. note::
416
417 Even though :class:`FunctionTestCase` can be used to quickly convert an existing
418 test base over to a :mod:`unittest`\ -based system, this approach is not
419 recommended. Taking the time to set up proper :class:`TestCase` subclasses will
420 make future test refactorings infinitely easier.
421
Benjamin Peterson52baa292009-03-24 00:56:30 +0000422In some cases, the existing tests may have been written using the :mod:`doctest`
423module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
424automatically build :class:`unittest.TestSuite` instances from the existing
425:mod:`doctest`\ -based tests.
426
Georg Brandl116aa622007-08-15 14:28:22 +0000427
Benjamin Peterson5254c042009-03-23 22:25:03 +0000428.. _unittest-skipping:
429
430Skipping tests and expected failures
431------------------------------------
432
433Unittest supports skipping individual test methods and even whole classes of
434tests. In addition, it supports marking a test as a "expected failure," a test
435that is broken and will fail, but shouldn't be counted as a failure on a
436:class:`TestResult`.
437
438Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
439or one of its conditional variants.
440
441Basic skipping looks like this: ::
442
443 class MyTestCase(unittest.TestCase):
444
445 @unittest.skip("demonstrating skipping")
446 def test_nothing(self):
447 self.fail("shouldn't happen")
448
449This is the output of running the example above in verbose mode: ::
450
451 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
452
453 ----------------------------------------------------------------------
454 Ran 1 test in 0.072s
455
456Classes can be skipped just like methods: ::
457
458 @skip("showing class skipping")
459 class MySkippedTestCase(unittest.TestCase):
460 def test_not_run(self):
461 pass
462
Benjamin Peterson52baa292009-03-24 00:56:30 +0000463:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
464that needs to be set up is not available.
465
Benjamin Peterson5254c042009-03-23 22:25:03 +0000466Expected failures use the :func:`expectedFailure` decorator. ::
467
468 class ExpectedFailureTestCase(unittest.TestCase):
469 @unittest.expectedFailure
470 def test_fail(self):
471 self.assertEqual(1, 0, "broken")
472
473It's easy to roll your own skipping decorators by making a decorator that calls
474:func:`skip` on the test when it wants it to be skipped. This decorator skips
475the test unless the passed object has a certain attribute: ::
476
477 def skipUnlessHasattr(obj, attr):
478 if hasattr(obj, attr):
479 return lambda func: func
480 return unittest.skip("{0!r} doesn't have {1!r}".format(obj, attr))
481
482The following decorators implement test skipping and expected failures:
483
484.. function:: skip(reason)
485
486 Unconditionally skip the decorated test. *reason* should describe why the
487 test is being skipped.
488
489.. function:: skipIf(condition, reason)
490
491 Skip the decorated test if *condition* is true.
492
493.. function:: skipUnless(condition, reason)
494
495 Skip the decoratored test unless *condition* is true.
496
497.. function:: expectedFailure
498
499 Mark the test as an expected failure. If the test fails when run, the test
500 is not counted as a failure.
501
502
Georg Brandl116aa622007-08-15 14:28:22 +0000503.. _unittest-contents:
504
505Classes and functions
506---------------------
507
Benjamin Peterson52baa292009-03-24 00:56:30 +0000508This section describes in depth the API of :mod:`unittest`.
509
510
511.. _testcase-objects:
512
513Test cases
514~~~~~~~~~~
Georg Brandl116aa622007-08-15 14:28:22 +0000515
516.. class:: TestCase([methodName])
517
518 Instances of the :class:`TestCase` class represent the smallest testable units
519 in the :mod:`unittest` universe. This class is intended to be used as a base
520 class, with specific tests being implemented by concrete subclasses. This class
521 implements the interface needed by the test runner to allow it to drive the
522 test, and methods that the test code can use to check for and report various
523 kinds of failure.
524
525 Each instance of :class:`TestCase` will run a single test method: the method
526 named *methodName*. If you remember, we had an earlier example that went
527 something like this::
528
529 def suite():
530 suite = unittest.TestSuite()
531 suite.addTest(WidgetTestCase('testDefaultSize'))
532 suite.addTest(WidgetTestCase('testResize'))
533 return suite
534
535 Here, we create two instances of :class:`WidgetTestCase`, each of which runs a
536 single test.
537
Benjamin Peterson52baa292009-03-24 00:56:30 +0000538 *methodName* defaults to :meth:`runTest`.
539
540 :class:`TestCase` instances provide three groups of methods: one group used
541 to run the test, another used by the test implementation to check conditions
542 and report failures, and some inquiry methods allowing information about the
543 test itself to be gathered.
544
545 Methods in the first group (running the test) are:
546
547
548 .. method:: setUp()
549
550 Method called to prepare the test fixture. This is called immediately
551 before calling the test method; any exception raised by this method will
552 be considered an error rather than a test failure. The default
553 implementation does nothing.
554
555
556 .. method:: tearDown()
557
558 Method called immediately after the test method has been called and the
559 result recorded. This is called even if the test method raised an
560 exception, so the implementation in subclasses may need to be particularly
561 careful about checking internal state. Any exception raised by this
562 method will be considered an error rather than a test failure. This
563 method will only be called if the :meth:`setUp` succeeds, regardless of
564 the outcome of the test method. The default implementation does nothing.
565
566
567 .. method:: run([result])
568
569 Run the test, collecting the result into the test result object passed as
570 *result*. If *result* is omitted or :const:`None`, a temporary result
571 object is created (by calling the :meth:`defaultTestCase` method) and
572 used; this result object is not returned to :meth:`run`'s caller.
573
574 The same effect may be had by simply calling the :class:`TestCase`
575 instance.
576
577
578 .. method:: skip(reason)
579
580 Calling this during the a test method or :meth:`setUp` skips the current
581 test. See :ref:`unittest-skipping` for more information.
582
583
584 .. method:: debug()
585
586 Run the test without collecting the result. This allows exceptions raised
587 by the test to be propagated to the caller, and can be used to support
588 running tests under a debugger.
589
590 The test code can use any of the following methods to check for and report
591 failures.
592
593
594 .. method:: assert_(expr[, msg])
595 failUnless(expr[, msg])
596 assertTrue(expr[, msg])
597
598 Signal a test failure if *expr* is false; the explanation for the error
599 will be *msg* if given, otherwise it will be :const:`None`.
600
601
602 .. method:: assertEqual(first, second[, msg])
603 failUnlessEqual(first, second[, msg])
604
605 Test that *first* and *second* are equal. If the values do not compare
606 equal, the test will fail with the explanation given by *msg*, or
607 :const:`None`. Note that using :meth:`failUnlessEqual` improves upon
608 doing the comparison as the first parameter to :meth:`failUnless`: the
609 default value for *msg* can be computed to include representations of both
610 *first* and *second*.
611
612
613 .. method:: assertNotEqual(first, second[, msg])
614 failIfEqual(first, second[, msg])
615
616 Test that *first* and *second* are not equal. If the values do compare
617 equal, the test will fail with the explanation given by *msg*, or
618 :const:`None`. Note that using :meth:`failIfEqual` improves upon doing
619 the comparison as the first parameter to :meth:`failUnless` is that the
620 default value for *msg* can be computed to include representations of both
621 *first* and *second*.
622
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000623
Benjamin Peterson52baa292009-03-24 00:56:30 +0000624 .. method:: assertAlmostEqual(first, second[, places[, msg]])
625 failUnlessAlmostEqual(first, second[, places[, msg]])
626
627 Test that *first* and *second* are approximately equal by computing the
628 difference, rounding to the given number of decimal *places* (default 7),
629 and comparing to zero.
630
631 Note that comparing a given number of decimal places is not the same as
632 comparing a given number of significant digits. If the values do not
633 compare equal, the test will fail with the explanation given by *msg*, or
634 :const:`None`.
635
636
637 .. method:: assertNotAlmostEqual(first, second[, places[, msg]])
638 failIfAlmostEqual(first, second[, places[, msg]])
639
640 Test that *first* and *second* are not approximately equal by computing
641 the difference, rounding to the given number of decimal *places* (default
642 7), and comparing to zero.
643
644 Note that comparing a given number of decimal places is not the same as
645 comparing a given number of significant digits. If the values do not
646 compare equal, the test will fail with the explanation given by *msg*, or
647 :const:`None`.
648
649
650 .. method:: assertRaises(exception[, callable, ...])
651 failUnlessRaises(exception[, callable, ...])
652
653 Test that an exception is raised when *callable* is called with any
654 positional or keyword arguments that are also passed to
655 :meth:`assertRaises`. The test passes if *exception* is raised, is an
656 error if another exception is raised, or fails if no exception is raised.
657 To catch any of a group of exceptions, a tuple containing the exception
658 classes may be passed as *exception*.
659
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000660 .. versionchanged:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +0000661
662 If *callable* is omitted or None, returns a context manager so that the
663 code under test can be written inline rather than as a function::
664
665 with self.failUnlessRaises(some_error_class):
666 do_something()
667
668 .. method:: failIf(expr[, msg])
669 assertFalse(expr[, msg])
670
671 The inverse of the :meth:`failUnless` method is the :meth:`failIf` method.
672 This signals a test failure if *expr* is true, with *msg* or :const:`None`
673 for the error message.
674
675
676 .. method:: fail([msg])
677
678 Signals a test failure unconditionally, with *msg* or :const:`None` for
679 the error message.
680
681
682 .. attribute:: failureException
683
684 This class attribute gives the exception raised by the test method. If a
685 test framework needs to use a specialized exception, possibly to carry
686 additional information, it must subclass this exception in order to "play
687 fair" with the framework. The initial value of this attribute is
688 :exc:`AssertionError`.
689
690 Testing frameworks can use the following methods to collect information on
691 the test:
692
693
694 .. method:: countTestCases()
695
696 Return the number of tests represented by this test object. For
697 :class:`TestCase` instances, this will always be ``1``.
698
699
700 .. method:: defaultTestResult()
701
702 Return an instance of the test result class that should be used for this
703 test case class (if no other result instance is provided to the
704 :meth:`run` method).
705
706 For :class:`TestCase` instances, this will always be an instance of
707 :class:`TestResult`; subclasses of :class:`TestCase` should override this
708 as necessary.
709
710
711 .. method:: id()
712
713 Return a string identifying the specific test case. This is usually the
714 full name of the test method, including the module and class name.
715
716
717 .. method:: shortDescription()
718
719 Returns a one-line description of the test, or :const:`None` if no
720 description has been provided. The default implementation of this method
721 returns the first line of the test method's docstring, if available, or
722 :const:`None`.
Georg Brandl116aa622007-08-15 14:28:22 +0000723
724
725.. class:: FunctionTestCase(testFunc[, setUp[, tearDown[, description]]])
726
727 This class implements the portion of the :class:`TestCase` interface which
728 allows the test runner to drive the test, but does not provide the methods which
729 test code can use to check and report errors. This is used to create test cases
730 using legacy test code, allowing it to be integrated into a :mod:`unittest`\
731 -based test framework.
732
733
Benjamin Peterson52baa292009-03-24 00:56:30 +0000734.. _testsuite-objects:
735
Benjamin Peterson52baa292009-03-24 00:56:30 +0000736Grouping tests
737~~~~~~~~~~~~~~
738
Georg Brandl116aa622007-08-15 14:28:22 +0000739.. class:: TestSuite([tests])
740
741 This class represents an aggregation of individual tests cases and test suites.
742 The class presents the interface needed by the test runner to allow it to be run
743 as any other test case. Running a :class:`TestSuite` instance is the same as
744 iterating over the suite, running each test individually.
745
746 If *tests* is given, it must be an iterable of individual test cases or other
747 test suites that will be used to build the suite initially. Additional methods
748 are provided to add test cases and suites to the collection later on.
749
Benjamin Peterson52baa292009-03-24 00:56:30 +0000750 :class:`TestSuite` (including :class:`ClassTestSuite`) objects behave much
751 like :class:`TestCase` objects, except they do not actually implement a test.
752 Instead, they are used to aggregate tests into groups of tests that should be
753 run together. Some additional methods are available to add tests to
754 :class:`TestSuite` instances:
755
756
757 .. method:: TestSuite.addTest(test)
758
759 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
760
761
762 .. method:: TestSuite.addTests(tests)
763
764 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
765 instances to this test suite.
766
767 This is equivalent to iterating over *tests*, calling :meth:`addTest` for each
768 element.
769
770 :class:`TestSuite` shares the following methods with :class:`TestCase`:
771
772
773 .. method:: run(result)
774
775 Run the tests associated with this suite, collecting the result into the
776 test result object passed as *result*. Note that unlike
777 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
778 be passed in.
779
780
781 .. method:: debug()
782
783 Run the tests associated with this suite without collecting the
784 result. This allows exceptions raised by the test to be propagated to the
785 caller and can be used to support running tests under a debugger.
786
787
788 .. method:: countTestCases()
789
790 Return the number of tests represented by this test object, including all
791 individual tests and sub-suites.
792
793 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
794 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
795
796
Benjamin Peterson5254c042009-03-23 22:25:03 +0000797.. class:: ClassTestSuite(tests, collected_from)
798
799 This subclass of :class:`TestSuite` repesents an aggregation of individuals
800 tests from one :class:`TestCase` class. *tests* is an iterable of
801 :class:`TestCase` instances created from the class. *collected_from* is the
802 class they came from.
803
Georg Brandl116aa622007-08-15 14:28:22 +0000804
Benjamin Peterson52baa292009-03-24 00:56:30 +0000805Loading and running tests
806~~~~~~~~~~~~~~~~~~~~~~~~~
807
Georg Brandl116aa622007-08-15 14:28:22 +0000808.. class:: TestLoader()
809
Benjamin Peterson52baa292009-03-24 00:56:30 +0000810 The :class:`TestLoader` class is used to create test suites from classes and
811 modules. Normally, there is no need to create an instance of this class; the
812 :mod:`unittest` module provides an instance that can be shared as
813 ``unittest.defaultTestLoader``. Using a subclass or instance, however, allows
814 customization of some configurable properties.
815
816 :class:`TestLoader` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +0000817
818
Benjamin Peterson52baa292009-03-24 00:56:30 +0000819 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl116aa622007-08-15 14:28:22 +0000820
Benjamin Peterson52baa292009-03-24 00:56:30 +0000821 Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
822 :class:`testCaseClass`.
823
824
825 .. method:: loadTestsFromModule(module)
826
827 Return a suite of all tests cases contained in the given module. This
828 method searches *module* for classes derived from :class:`TestCase` and
829 creates an instance of the class for each test method defined for the
830 class.
831
832 .. warning::
833
834 While using a hierarchy of :class:`TestCase`\ -derived classes can be
835 convenient in sharing fixtures and helper functions, defining test
836 methods on base classes that are not intended to be instantiated
837 directly does not play well with this method. Doing so, however, can
838 be useful when the fixtures are different and defined in subclasses.
839
840
841 .. method:: loadTestsFromName(name[, module])
842
843 Return a suite of all tests cases given a string specifier.
844
845 The specifier *name* is a "dotted name" that may resolve either to a
846 module, a test case class, a test method within a test case class, a
847 :class:`TestSuite` instance, or a callable object which returns a
848 :class:`TestCase` or :class:`TestSuite` instance. These checks are
849 applied in the order listed here; that is, a method on a possible test
850 case class will be picked up as "a test method within a test case class",
851 rather than "a callable object".
852
853 For example, if you have a module :mod:`SampleTests` containing a
854 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
855 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
856 specifier ``'SampleTests.SampleTestCase'`` would cause this method to return a
857 suite which will run all three test methods. Using the specifier
858 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test suite
859 which will run only the :meth:`test_two` test method. The specifier can refer
860 to modules and packages which have not been imported; they will be imported as a
861 side-effect.
862
863 The method optionally resolves *name* relative to the given *module*.
864
865
866 .. method:: loadTestsFromNames(names[, module])
867
868 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
869 than a single name. The return value is a test suite which supports all
870 the tests defined for each name.
871
872
873 .. method:: getTestCaseNames(testCaseClass)
874
875 Return a sorted sequence of method names found within *testCaseClass*;
876 this should be a subclass of :class:`TestCase`.
877
878 The following attributes of a :class:`TestLoader` can be configured either by
879 subclassing or assignment on an instance:
880
881
882 .. attribute:: testMethodPrefix
883
884 String giving the prefix of method names which will be interpreted as test
885 methods. The default value is ``'test'``.
886
887 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
888 methods.
889
890
891 .. attribute:: sortTestMethodsUsing
892
893 Function to be used to compare method names when sorting them in
894 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
895
896
897 .. attribute:: suiteClass
898
899 Callable object that constructs a test suite from a list of tests. No
900 methods on the resulting object are needed. The default value is the
901 :class:`TestSuite` class.
902
903 This affects all the :meth:`loadTestsFrom\*` methods.
904
905
906 .. attribute:: classSuiteClass
907
908 Callable object that constructs a test suite for the tests cases from one
909 class. The default value is :class:`ClassTestSuite`.
910
911
912.. class:: TestResult
913
914 This class is used to compile information about which tests have succeeded
915 and which have failed.
916
917 A :class:`TestResult` object stores the results of a set of tests. The
918 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
919 properly recorded; test authors do not need to worry about recording the
920 outcome of tests.
921
922 Testing frameworks built on top of :mod:`unittest` may want access to the
923 :class:`TestResult` object generated by running a set of tests for reporting
924 purposes; a :class:`TestResult` instance is returned by the
925 :meth:`TestRunner.run` method for this purpose.
926
927 :class:`TestResult` instances have the following attributes that will be of
928 interest when inspecting the results of running a set of tests:
929
930
931 .. attribute:: errors
932
933 A list containing 2-tuples of :class:`TestCase` instances and strings
934 holding formatted tracebacks. Each tuple represents a test which raised an
935 unexpected exception.
936
Benjamin Peterson52baa292009-03-24 00:56:30 +0000937
938 .. attribute:: failures
939
940 A list containing 2-tuples of :class:`TestCase` instances and strings
941 holding formatted tracebacks. Each tuple represents a test where a failure
942 was explicitly signalled using the :meth:`TestCase.fail\*` or
943 :meth:`TestCase.assert\*` methods.
944
Benjamin Peterson52baa292009-03-24 00:56:30 +0000945 .. attribute:: skipped
946
947 A list containing 2-tuples of :class:`TestCase` instances and strings
948 holding the reason for skipping the test.
949
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000950 .. versionadded:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +0000951
952 .. attribute:: expectedFailures
953
954 A list contaning 2-tuples of :class:`TestCase` instances and strings
955 holding formatted tracebacks. Each tuple represents a expected failures
956 of the test case.
957
958 .. attribute:: unexpectedSuccesses
959
960 A list containing :class:`TestCase` instances that were marked as expected
961 failures, but succeeded.
962
963 .. attribute:: shouldStop
964
965 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
966
967
968 .. attribute:: testsRun
969
970 The total number of tests run so far.
971
972
973 .. method:: wasSuccessful()
974
975 Return :const:`True` if all tests run so far have passed, otherwise returns
976 :const:`False`.
977
978
979 .. method:: stop()
980
981 This method can be called to signal that the set of tests being run should
982 be aborted by setting the :attr:`shouldStop` attribute to :const:`True`.
983 :class:`TestRunner` objects should respect this flag and return without
984 running any additional tests.
985
986 For example, this feature is used by the :class:`TextTestRunner` class to
987 stop the test framework when the user signals an interrupt from the
988 keyboard. Interactive tools which provide :class:`TestRunner`
989 implementations can use this in a similar manner.
990
991 The following methods of the :class:`TestResult` class are used to maintain
992 the internal data structures, and may be extended in subclasses to support
993 additional reporting requirements. This is particularly useful in building
994 tools which support interactive reporting while tests are being run.
995
996
997 .. method:: startTest(test)
998
999 Called when the test case *test* is about to be run.
1000
1001 The default implementation simply increments the instance's :attr:`testsRun`
1002 counter.
1003
1004
1005 .. method:: stopTest(test)
1006
1007 Called after the test case *test* has been executed, regardless of the
1008 outcome.
1009
1010 The default implementation does nothing.
1011
1012
1013 .. method:: addError(test, err)
1014
1015 Called when the test case *test* raises an unexpected exception *err* is a
1016 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1017 traceback)``.
1018
1019 The default implementation appends a tuple ``(test, formatted_err)`` to
1020 the instance's :attr:`errors` attribute, where *formatted_err* is a
1021 formatted traceback derived from *err*.
1022
1023
1024 .. method:: addFailure(test, err)
1025
1026 Called when the test case *test* signals a failure. *err* is a tuple of the form
1027 returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
1028
1029 The default implementation appends a tuple ``(test, formatted_err)`` to
1030 the instance's :attr:`failures` attribute, where *formatted_err* is a
1031 formatted traceback derived from *err*.
1032
1033
1034 .. method:: addSuccess(test)
1035
1036 Called when the test case *test* succeeds.
1037
1038 The default implementation does nothing.
1039
1040
1041 .. method:: addSkip(test, reason)
1042
1043 Called when the test case *test* is skipped. *reason* is the reason the
1044 test gave for skipping.
1045
1046 The default implementation appends a tuple ``(test, reason)`` to the
1047 instance's :attr:`skipped` attribute.
1048
1049
1050 .. method:: addExpectedFailure(test, err)
1051
1052 Called when the test case *test* fails, but was marked with the
1053 :func:`expectedFailure` decorator.
1054
1055 The default implementation appends a tuple ``(test, formatted_err)`` to
1056 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1057 is a formatted traceback derived from *err*.
1058
1059
1060 .. method:: addUnexpectedSuccess(test)
1061
1062 Called when the test case *test* was marked with the
1063 :func:`expectedFailure` decorator, but succeeded.
1064
1065 The default implementation appends the test to the instance's
1066 :attr:`unexpectedSuccesses` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +00001067
1068
1069.. data:: defaultTestLoader
1070
1071 Instance of the :class:`TestLoader` class intended to be shared. If no
1072 customization of the :class:`TestLoader` is needed, this instance can be used
1073 instead of repeatedly creating new instances.
1074
1075
1076.. class:: TextTestRunner([stream[, descriptions[, verbosity]]])
1077
1078 A basic test runner implementation which prints results on standard error. It
1079 has a few configurable parameters, but is essentially very simple. Graphical
1080 applications which run test suites should provide alternate implementations.
1081
1082
1083.. function:: main([module[, defaultTest[, argv[, testRunner[, testLoader]]]]])
1084
1085 A command-line program that runs a set of tests; this is primarily for making
1086 test modules conveniently executable. The simplest use for this function is to
1087 include the following line at the end of a test script::
1088
1089 if __name__ == '__main__':
1090 unittest.main()
1091
1092 The *testRunner* argument can either be a test runner class or an already
1093 created instance of it.