unittest.rst 88.3 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
:mod:`unittest` --- Unit testing framework
==========================================

.. module:: unittest
   :synopsis: Unit testing framework for Python.
.. moduleauthor:: Steve Purcell <stephen_purcell@yahoo.com>
.. sectionauthor:: Steve Purcell <stephen_purcell@yahoo.com>
.. sectionauthor:: Fred L. Drake, Jr. <fdrake@acm.org>
.. sectionauthor:: Raymond Hettinger <python@rcn.com>

11 12
(If you are already familiar with the basic concepts of testing, you might want
to skip to :ref:`the list of assert methods <assert-methods>`.)
13

14 15 16 17 18
The :mod:`unittest` unit testing framework was originally inspired by JUnit
and has a similar flavor as major unit testing frameworks in other
languages.  It supports test automation, sharing of setup and shutdown code
for tests, aggregation of tests into collections, and independence of the
tests from the reporting framework.
19

20 21
To achieve this, :mod:`unittest` supports some important concepts in an
object-oriented way:
22 23 24 25 26 27 28 29

test fixture
   A :dfn:`test fixture` represents the preparation needed to perform one or more
   tests, and any associate cleanup actions.  This may involve, for example,
   creating temporary or proxy databases, directories, or starting a server
   process.

test case
30
   A :dfn:`test case` is the individual unit of testing.  It checks for a specific
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
   response to a particular set of inputs.  :mod:`unittest` provides a base class,
   :class:`TestCase`, which may be used to create new test cases.

test suite
   A :dfn:`test suite` is a collection of test cases, test suites, or both.  It is
   used to aggregate tests that should be executed together.

test runner
   A :dfn:`test runner` is a component which orchestrates the execution of tests
   and provides the outcome to the user.  The runner may use a graphical interface,
   a textual interface, or return a special value to indicate the results of
   executing the tests.


.. seealso::

   Module :mod:`doctest`
      Another test-support module with a very different flavor.

   `Simple Smalltalk Testing: With Patterns <http://www.XProgramming.com/testfram.htm>`_
51 52
      Kent Beck's original paper on testing frameworks using the pattern shared
      by :mod:`unittest`.
53

54
   `Nose <http://code.google.com/p/python-nose/>`_ and `py.test <http://pytest.org>`_
55 56
      Third-party unittest frameworks with a lighter-weight syntax for writing
      tests.  For example, ``assert func(10) == 42``.
57

58
   `The Python Testing Tools Taxonomy <http://wiki.python.org/moin/PythonTestingToolsTaxonomy>`_
Benjamin Peterson's avatar
Benjamin Peterson committed
59 60
      An extensive list of Python testing tools including functional testing
      frameworks and mock object libraries.
61

Benjamin Peterson's avatar
Benjamin Peterson committed
62 63 64
   `Testing in Python Mailing List <http://lists.idyll.org/listinfo/testing-in-python>`_
      A special-interest-group for discussion of testing, and testing tools,
      in Python.
65

66 67
   The script :file:`Tools/unittestgui/unittestgui.py` in the Python source distribution is
   a GUI tool for test discovery and execution.  This is intended largely for ease of use
68 69 70 71
   for those new to unit testing.  For production environments it is
   recommended that tests be driven by a continuous integration system such as
   `Buildbot <http://buildbot.net/trac>`_, `Jenkins <http://jenkins-ci.org>`_
   or  `Hudson <http://hudson-ci.org/>`_.
72 73


74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90
.. _unittest-minimal-example:

Basic example
-------------

The :mod:`unittest` module provides a rich set of tools for constructing and
running tests.  This section demonstrates that a small subset of the tools
suffice to meet the needs of most users.

Here is a short script to test three functions from the :mod:`random` module::

   import random
   import unittest

   class TestSequenceFunctions(unittest.TestCase):

       def setUp(self):
91
           self.seq = list(range(10))
92

93
       def test_shuffle(self):
94 95 96
           # make sure the shuffled sequence does not lose any elements
           random.shuffle(self.seq)
           self.seq.sort()
97
           self.assertEqual(self.seq, list(range(10)))
98

99 100 101
           # should raise an exception for an immutable sequence
           self.assertRaises(TypeError, random.shuffle, (1,2,3))

102
       def test_choice(self):
103
           element = random.choice(self.seq)
104
           self.assertTrue(element in self.seq)
105

106
       def test_sample(self):
107 108
           with self.assertRaises(ValueError):
               random.sample(self.seq, 20)
109
           for element in random.sample(self.seq, 5):
110
               self.assertTrue(element in self.seq)
111 112 113 114

   if __name__ == '__main__':
       unittest.main()

115
A testcase is created by subclassing :class:`unittest.TestCase`.  The three
116 117 118 119
individual tests are defined with methods whose names start with the letters
``test``.  This naming convention informs the test runner about which methods
represent tests.

120
The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
121
expected result; :meth:`~TestCase.assertTrue` to verify a condition; or
122 123 124
:meth:`~TestCase.assertRaises` to verify that an expected exception gets raised.
These methods are used instead of the :keyword:`assert` statement so the test
runner can accumulate all test results and produce a report.
125

126 127 128 129 130
When a :meth:`~TestCase.setUp` method is defined, the test runner will run that
method prior to each test.  Likewise, if a :meth:`~TestCase.tearDown` method is
defined, the test runner will invoke that method after each test.  In the
example, :meth:`~TestCase.setUp` was used to create a fresh sequence for each
test.
131 132

The final block shows a simple way to run the tests. :func:`unittest.main`
133
provides a command-line interface to the test script.  When run from the command
134 135 136 137 138 139 140 141
line, the above script produces an output that looks like this::

   ...
   ----------------------------------------------------------------------
   Ran 3 tests in 0.000s

   OK

142 143
Passing the ``-v`` option to your test script will instruct :func:`unittest.main`
to enable a higher level of verbosity, and produce the following output::
144

145 146 147
   test_choice (__main__.TestSequenceFunctions) ... ok
   test_sample (__main__.TestSequenceFunctions) ... ok
   test_shuffle (__main__.TestSequenceFunctions) ... ok
148 149 150 151 152 153 154 155 156 157

   ----------------------------------------------------------------------
   Ran 3 tests in 0.110s

   OK

The above examples show the most commonly used :mod:`unittest` features which
are sufficient to meet many everyday testing needs.  The remainder of the
documentation explores the full feature set from first principles.

Benjamin Peterson's avatar
Benjamin Peterson committed
158 159 160

.. _unittest-command-line-interface:

161
Command-Line Interface
Benjamin Peterson's avatar
Benjamin Peterson committed
162 163 164 165 166 167 168 169 170 171 172 173
----------------------

The unittest module can be used from the command line to run tests from
modules, classes or even individual test methods::

   python -m unittest test_module1 test_module2
   python -m unittest test_module.TestClass
   python -m unittest test_module.TestClass.test_method

You can pass in a list with any combination of module names, and fully
qualified class or method names.

174 175 176 177 178 179 180 181 182 183
Test modules can be specified by file path as well::

   python -m unittest tests/test_something.py

This allows you to use the shell filename completion to specify the test module.
The file specified must still be importable as a module. The path is converted
to a module name by removing the '.py' and converting path separators into '.'.
If you want to execute a test file that isn't importable as a module you should
execute the file directly instead.

Benjamin Peterson's avatar
Benjamin Peterson committed
184 185 186 187
You can run tests with more detail (higher verbosity) by passing in the -v flag::

   python -m unittest -v test_module

188 189 190 191
When executed without arguments :ref:`unittest-test-discovery` is started::

   python -m unittest

192
For a list of all the command-line options::
Benjamin Peterson's avatar
Benjamin Peterson committed
193 194 195

   python -m unittest -h

196
.. versionchanged:: 3.2
Benjamin Peterson's avatar
Benjamin Peterson committed
197 198 199 200
   In earlier versions it was only possible to run individual test methods and
   not modules or classes.


201 202
Command-line options
~~~~~~~~~~~~~~~~~~~~
Benjamin Peterson's avatar
Benjamin Peterson committed
203

204
:program:`unittest` supports these command-line options:
Benjamin Peterson's avatar
Benjamin Peterson committed
205

206
.. program:: unittest
Benjamin Peterson's avatar
Benjamin Peterson committed
207

208
.. cmdoption:: -b, --buffer
Benjamin Peterson's avatar
Benjamin Peterson committed
209

210 211 212
   The standard output and standard error streams are buffered during the test
   run. Output during a passing test is discarded. Output is echoed normally
   on test fail or error and is added to the failure messages.
Benjamin Peterson's avatar
Benjamin Peterson committed
213

214
.. cmdoption:: -c, --catch
Benjamin Peterson's avatar
Benjamin Peterson committed
215

216 217 218
   Control-C during the test run waits for the current test to end and then
   reports all the results so far. A second control-C raises the normal
   :exc:`KeyboardInterrupt` exception.
Benjamin Peterson's avatar
Benjamin Peterson committed
219

220
   See `Signal Handling`_ for the functions that provide this functionality.
Benjamin Peterson's avatar
Benjamin Peterson committed
221

222 223 224
.. cmdoption:: -f, --failfast

   Stop the test run on the first error or failure.
Benjamin Peterson's avatar
Benjamin Peterson committed
225

226
.. versionadded:: 3.2
227
   The command-line options ``-b``, ``-c`` and ``-f`` were added.
Benjamin Peterson's avatar
Benjamin Peterson committed
228 229 230 231 232 233 234 235 236 237

The command line can also be used for test discovery, for running all of the
tests in a project or just a subset.


.. _unittest-test-discovery:

Test Discovery
--------------

238
.. versionadded:: 3.2
Benjamin Peterson's avatar
Benjamin Peterson committed
239

240 241
Unittest supports simple test discovery. In order to be compatible with test
discovery, all of the test files must be :ref:`modules <tut-modules>` or
242 243 244 245
:ref:`packages <tut-packages>` (including :term:`namespace packages
<namespace package>`) importable from the top-level directory of
the project (this means that their filenames must be valid :ref:`identifiers
<identifiers>`).
Benjamin Peterson's avatar
Benjamin Peterson committed
246 247

Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
248
used from the command line. The basic command-line usage is::
Benjamin Peterson's avatar
Benjamin Peterson committed
249 250 251 252

   cd project_directory
   python -m unittest discover

253 254 255 256
.. note::

   As a shortcut, ``python -m unittest`` is the equivalent of
   ``python -m unittest discover``. If you want to pass arguments to test
257
   discovery the ``discover`` sub-command must be used explicitly.
258

Benjamin Peterson's avatar
Benjamin Peterson committed
259 260
The ``discover`` sub-command has the following options:

261 262 263 264 265 266
.. program:: unittest discover

.. cmdoption:: -v, --verbose

   Verbose output

267
.. cmdoption:: -s, --start-directory directory
268

269
   Directory to start discovery (``.`` default)
270

271
.. cmdoption:: -p, --pattern pattern
272

273
   Pattern to match test files (``test*.py`` default)
274

275
.. cmdoption:: -t, --top-level-directory directory
276 277

   Top level directory of project (defaults to start directory)
Benjamin Peterson's avatar
Benjamin Peterson committed
278

Benjamin Peterson's avatar
Benjamin Peterson committed
279 280 281
The :option:`-s`, :option:`-p`, and :option:`-t` options can be passed in
as positional arguments in that order. The following two command lines
are equivalent::
Benjamin Peterson's avatar
Benjamin Peterson committed
282 283 284 285

   python -m unittest discover -s project_directory -p '*_test.py'
   python -m unittest discover project_directory '*_test.py'

286 287 288 289 290 291 292
As well as being a path it is possible to pass a package name, for example
``myproject.subpackage.test``, as the start directory. The package name you
supply will then be imported and its location on the filesystem will be used
as the start directory.

.. caution::

293 294 295
    Test discovery loads tests by importing them. Once test discovery has found
    all the test files from the start directory you specify it turns the paths
    into package names to import. For example :file:`foo/bar/baz.py` will be
296 297 298 299 300 301 302 303 304 305 306
    imported as ``foo.bar.baz``.

    If you have a package installed globally and attempt test discovery on
    a different copy of the package then the import *could* happen from the
    wrong place. If this happens test discovery will warn you and exit.

    If you supply the start directory as a package name rather than a
    path to a directory then discover assumes that whichever location it
    imports from is the location you intended, so you will not get the
    warning.

Benjamin Peterson's avatar
Benjamin Peterson committed
307 308 309
Test modules and packages can customize test loading and discovery by through
the `load_tests protocol`_.

310 311 312
.. versionchanged:: 3.4
   Test discovery supports :term:`namespace packages <namespace package>`.

Benjamin Peterson's avatar
Benjamin Peterson committed
313

314 315 316 317 318 319 320
.. _organizing-tests:

Organizing test code
--------------------

The basic building blocks of unit testing are :dfn:`test cases` --- single
scenarios that must be set up and checked for correctness.  In :mod:`unittest`,
321 322 323
test cases are represented by :class:`unittest.TestCase` instances.
To make your own test cases you must write subclasses of
:class:`TestCase` or use :class:`FunctionTestCase`.
324 325 326 327 328

The testing code of a :class:`TestCase` instance should be entirely self
contained, such that it can be run either in isolation or in arbitrary
combination with any number of other test cases.

329 330 331
The simplest :class:`TestCase` subclass will simply implement a test method
(i.e. a method whose name starts with ``test``) in order to perform specific
testing code::
332 333 334 335

   import unittest

   class DefaultWidgetSizeTestCase(unittest.TestCase):
336
       def test_default_widget_size(self):
337
           widget = Widget('The widget')
338
           self.assertEqual(widget.size(), (50, 50))
339

340
Note that in order to test something, we use one of the :meth:`assert\*`
341 342
methods provided by the :class:`TestCase` base class.  If the test fails, an
exception will be raised, and :mod:`unittest` will identify the test case as a
343
:dfn:`failure`.  Any other exceptions will be treated as :dfn:`errors`.
344

345 346 347 348
Tests can be numerous, and their set-up can be repetitive.  Luckily, we
can factor out set-up code by implementing a method called
:meth:`~TestCase.setUp`, which the testing framework will automatically
call for every single test we run::
349 350 351 352 353 354 355

   import unittest

   class SimpleWidgetTestCase(unittest.TestCase):
       def setUp(self):
           self.widget = Widget('The widget')

356
       def test_default_widget_size(self):
357 358
           self.assertEqual(self.widget.size(), (50,50),
                            'incorrect default size')
359

360
       def test_widget_resize(self):
361
           self.widget.resize(100,150)
362 363
           self.assertEqual(self.widget.size(), (100,150),
                            'wrong size after resize')
364

365 366 367 368 369
.. note::
   The order in which the various tests will be run is determined
   by sorting the test method names with respect to the built-in
   ordering for strings.

370
If the :meth:`~TestCase.setUp` method raises an exception while the test is
371 372
running, the framework will consider the test to have suffered an error, and
the test method will not be executed.
373

374
Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
375
after the test method has been run::
376 377 378 379 380 381 382 383 384 385

   import unittest

   class SimpleWidgetTestCase(unittest.TestCase):
       def setUp(self):
           self.widget = Widget('The widget')

       def tearDown(self):
           self.widget.dispose()

386 387
If :meth:`~TestCase.setUp` succeeded, :meth:`~TestCase.tearDown` will be
run whether the test method succeeded or not.
388 389 390 391 392

Such a working environment for the testing code is called a :dfn:`fixture`.

Test case instances are grouped together according to the features they test.
:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
393 394 395
represented by :mod:`unittest`'s :class:`TestSuite` class.  In most cases,
calling :func:`unittest.main` will do the right thing and collect all the
module's test cases for you, and then execute them.
396

397 398
However, should you want to customize the building of your test suite,
you can do it yourself::
399 400 401

   def suite():
       suite = unittest.TestSuite()
402 403
       suite.addTest(WidgetTestCase('test_default_size'))
       suite.addTest(WidgetTestCase('test_resize'))
404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447
       return suite

You can place the definitions of test cases and test suites in the same modules
as the code they are to test (such as :file:`widget.py`), but there are several
advantages to placing the test code in a separate module, such as
:file:`test_widget.py`:

* The test module can be run standalone from the command line.

* The test code can more easily be separated from shipped code.

* There is less temptation to change test code to fit the code it tests without
  a good reason.

* Test code should be modified much less frequently than the code it tests.

* Tested code can be refactored more easily.

* Tests for modules written in C must be in separate modules anyway, so why not
  be consistent?

* If the testing strategy changes, there is no need to change the source code.


.. _legacy-unit-tests:

Re-using old test code
----------------------

Some users will find that they have existing test code that they would like to
run from :mod:`unittest`, without converting every old test function to a
:class:`TestCase` subclass.

For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
This subclass of :class:`TestCase` can be used to wrap an existing test
function.  Set-up and tear-down functions can also be provided.

Given the following test function::

   def testSomething():
       something = makeSomething()
       assert something.name is not None
       # ...

448 449
one can create an equivalent test case instance as follows, with optional
set-up and tear-down methods::
450 451 452 453 454 455 456

   testcase = unittest.FunctionTestCase(testSomething,
                                        setUp=makeSomethingDB,
                                        tearDown=deleteSomethingDB)

.. note::

457 458 459 460
   Even though :class:`FunctionTestCase` can be used to quickly convert an
   existing test base over to a :mod:`unittest`\ -based system, this approach is
   not recommended.  Taking the time to set up proper :class:`TestCase`
   subclasses will make future test refactorings infinitely easier.
461

462 463 464 465 466
In some cases, the existing tests may have been written using the :mod:`doctest`
module.  If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
automatically build :class:`unittest.TestSuite` instances from the existing
:mod:`doctest`\ -based tests.

467

468 469 470 471 472
.. _unittest-skipping:

Skipping tests and expected failures
------------------------------------

473 474
.. versionadded:: 3.1

475 476 477 478 479 480 481 482
Unittest supports skipping individual test methods and even whole classes of
tests.  In addition, it supports marking a test as a "expected failure," a test
that is broken and will fail, but shouldn't be counted as a failure on a
:class:`TestResult`.

Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
or one of its conditional variants.

483
Basic skipping looks like this::
484 485 486 487 488 489 490

   class MyTestCase(unittest.TestCase):

       @unittest.skip("demonstrating skipping")
       def test_nothing(self):
           self.fail("shouldn't happen")

491 492
       @unittest.skipIf(mylib.__version__ < (1, 3),
                        "not supported in this library version")
493 494 495 496 497 498 499 500 501
       def test_format(self):
           # Tests that work for only a certain version of the library.
           pass

       @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
       def test_windows_support(self):
           # windows specific testing code
           pass

502
This is the output of running the example above in verbose mode::
503

504
   test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
505
   test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
506
   test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
507 508

   ----------------------------------------------------------------------
509 510 511
   Ran 3 tests in 0.005s

   OK (skipped=3)
512

513
Classes can be skipped just like methods::
514

515
   @unittest.skip("showing class skipping")
516 517 518 519
   class MySkippedTestCase(unittest.TestCase):
       def test_not_run(self):
           pass

520 521 522
:meth:`TestCase.setUp` can also skip the test.  This is useful when a resource
that needs to be set up is not available.

523 524 525 526 527 528 529 530 531
Expected failures use the :func:`expectedFailure` decorator. ::

   class ExpectedFailureTestCase(unittest.TestCase):
       @unittest.expectedFailure
       def test_fail(self):
           self.assertEqual(1, 0, "broken")

It's easy to roll your own skipping decorators by making a decorator that calls
:func:`skip` on the test when it wants it to be skipped.  This decorator skips
532
the test unless the passed object has a certain attribute::
533 534 535 536

   def skipUnlessHasattr(obj, attr):
       if hasattr(obj, attr):
           return lambda func: func
537
       return unittest.skip("{!r} doesn't have {!r}".format(obj, attr))
538 539 540

The following decorators implement test skipping and expected failures:

541
.. decorator:: skip(reason)
542 543 544 545

   Unconditionally skip the decorated test.  *reason* should describe why the
   test is being skipped.

546
.. decorator:: skipIf(condition, reason)
547 548 549

   Skip the decorated test if *condition* is true.

550
.. decorator:: skipUnless(condition, reason)
551

Georg Brandl's avatar
Georg Brandl committed
552
   Skip the decorated test unless *condition* is true.
553

554
.. decorator:: expectedFailure
555 556 557 558

   Mark the test as an expected failure.  If the test fails when run, the test
   is not counted as a failure.

559 560 561 562 563 564 565
.. exception:: SkipTest(reason)

   This exception is raised to skip a test.

   Usually you can use :meth:`TestCase.skipTest` or one of the skipping
   decorators instead of raising this directly.

566 567 568
Skipped tests will not have :meth:`~TestCase.setUp` or :meth:`~TestCase.tearDown` run around them.
Skipped classes will not have :meth:`~TestCase.setUpClass` or :meth:`~TestCase.tearDownClass` run.
Skipped modules will not have :func:`setUpModule` or :func:`tearDownModule` run.
Benjamin Peterson's avatar
Benjamin Peterson committed
569

570

571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632
.. _subtests:

Distinguishing test iterations using subtests
---------------------------------------------

.. versionadded:: 3.4

When some of your tests differ only by a some very small differences, for
instance some parameters, unittest allows you to distinguish them inside
the body of a test method using the :meth:`~TestCase.subTest` context manager.

For example, the following test::

   class NumbersTest(unittest.TestCase):

       def test_even(self):
           """
           Test that numbers between 0 and 5 are all even.
           """
           for i in range(0, 6):
               with self.subTest(i=i):
                   self.assertEqual(i % 2, 0)

will produce the following output::

   ======================================================================
   FAIL: test_even (__main__.NumbersTest) (i=1)
   ----------------------------------------------------------------------
   Traceback (most recent call last):
     File "subtests.py", line 32, in test_even
       self.assertEqual(i % 2, 0)
   AssertionError: 1 != 0

   ======================================================================
   FAIL: test_even (__main__.NumbersTest) (i=3)
   ----------------------------------------------------------------------
   Traceback (most recent call last):
     File "subtests.py", line 32, in test_even
       self.assertEqual(i % 2, 0)
   AssertionError: 1 != 0

   ======================================================================
   FAIL: test_even (__main__.NumbersTest) (i=5)
   ----------------------------------------------------------------------
   Traceback (most recent call last):
     File "subtests.py", line 32, in test_even
       self.assertEqual(i % 2, 0)
   AssertionError: 1 != 0

Without using a subtest, execution would stop after the first failure,
and the error would be less easy to diagnose because the value of ``i``
wouldn't be displayed::

   ======================================================================
   FAIL: test_even (__main__.NumbersTest)
   ----------------------------------------------------------------------
   Traceback (most recent call last):
     File "subtests.py", line 32, in test_even
       self.assertEqual(i % 2, 0)
   AssertionError: 1 != 0


633 634 635 636 637
.. _unittest-contents:

Classes and functions
---------------------

638 639 640 641 642 643 644
This section describes in depth the API of :mod:`unittest`.


.. _testcase-objects:

Test cases
~~~~~~~~~~
645

646
.. class:: TestCase(methodName='runTest')
647

648
   Instances of the :class:`TestCase` class represent the logical test units
649 650 651
   in the :mod:`unittest` universe.  This class is intended to be used as a base
   class, with specific tests being implemented by concrete subclasses.  This class
   implements the interface needed by the test runner to allow it to drive the
652
   tests, and methods that the test code can use to check for and report various
653 654
   kinds of failure.

655 656 657 658 659 660
   Each instance of :class:`TestCase` will run a single base method: the method
   named *methodName*.  However, the standard implementation of the default
   *methodName*, ``runTest()``, will run every method starting with ``test``
   as an individual test, and count successes and failures accordingly.
   Therefore, in most uses of :class:`TestCase`, you will neither change
   the *methodName* nor reimplement the default ``runTest()`` method.
661

662
   .. versionchanged:: 3.2
663 664 665
      :class:`TestCase` can be instantiated successfully without providing a
      *methodName*. This makes it easier to experiment with :class:`TestCase`
      from the interactive interpreter.
666

667 668 669 670
   :class:`TestCase` instances provide three groups of methods: one group used
   to run the test, another used by the test implementation to check conditions
   and report failures, and some inquiry methods allowing information about the
   test itself to be gathered.
671

672
   Methods in the first group (running the test) are:
673

674
   .. method:: setUp()
675

676
      Method called to prepare the test fixture.  This is called immediately
Terry Jan Reedy's avatar
Terry Jan Reedy committed
677 678
      before calling the test method; other than :exc:`AssertionError` or :exc:`SkipTest`,
      any exception raised by this method will be considered an error rather than
679
      a test failure. The default implementation does nothing.
680 681


682
   .. method:: tearDown()
683

684 685 686
      Method called immediately after the test method has been called and the
      result recorded.  This is called even if the test method raised an
      exception, so the implementation in subclasses may need to be particularly
Terry Jan Reedy's avatar
Terry Jan Reedy committed
687 688 689
      careful about checking internal state.  Any exception, other than :exc:`AssertionError`
      or :exc:`SkipTest`, raised by this method will be considered an error rather than a
      test failure.  This method will only be called if the :meth:`setUp` succeeds,
690
      regardless of the outcome of the test method. The default implementation does nothing.
691 692


Benjamin Peterson's avatar
Benjamin Peterson committed
693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722
   .. method:: setUpClass()

      A class method called before tests in an individual class run.
      ``setUpClass`` is called with the class as the only argument
      and must be decorated as a :func:`classmethod`::

        @classmethod
        def setUpClass(cls):
            ...

      See `Class and Module Fixtures`_ for more details.

      .. versionadded:: 3.2


   .. method:: tearDownClass()

      A class method called after tests in an individual class have run.
      ``tearDownClass`` is called with the class as the only argument
      and must be decorated as a :meth:`classmethod`::

        @classmethod
        def tearDownClass(cls):
            ...

      See `Class and Module Fixtures`_ for more details.

      .. versionadded:: 3.2


723
   .. method:: run(result=None)
724

725 726 727 728 729
      Run the test, collecting the result into the :class:`TestResult` object
      passed as *result*.  If *result* is omitted or ``None``, a temporary
      result object is created (by calling the :meth:`defaultTestResult`
      method) and used. The result object is returned to :meth:`run`'s
      caller.
730

731 732
      The same effect may be had by simply calling the :class:`TestCase`
      instance.
733

734 735 736
      .. versionchanged:: 3.3
         Previous versions of ``run`` did not return the result. Neither did
         calling an instance.
737

Benjamin Peterson's avatar
Benjamin Peterson committed
738
   .. method:: skipTest(reason)
739

740
      Calling this during a test method or :meth:`setUp` skips the current
741
      test.  See :ref:`unittest-skipping` for more information.
742

743
      .. versionadded:: 3.1
Benjamin Peterson's avatar
Benjamin Peterson committed
744

745

746 747 748 749 750 751 752 753 754 755 756 757 758 759 760
   .. method:: subTest(msg=None, **params)

      Return a context manager which executes the enclosed code block as a
      subtest.  *msg* and *params* are optional, arbitrary values which are
      displayed whenever a subtest fails, allowing you to identify them
      clearly.

      A test case can contain any number of subtest declarations, and
      they can be arbitrarily nested.

      See :ref:`subtests` for more information.

      .. versionadded:: 3.4


761
   .. method:: debug()
762

763 764 765
      Run the test without collecting the result.  This allows exceptions raised
      by the test to be propagated to the caller, and can be used to support
      running tests under a debugger.
766

767
   .. _assert-methods:
768

769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810
   The :class:`TestCase` class provides a number of methods to check for and
   report failures, such as:

   +-----------------------------------------+-----------------------------+---------------+
   | Method                                  | Checks that                 | New in        |
   +=========================================+=============================+===============+
   | :meth:`assertEqual(a, b)                | ``a == b``                  |               |
   | <TestCase.assertEqual>`                 |                             |               |
   +-----------------------------------------+-----------------------------+---------------+
   | :meth:`assertNotEqual(a, b)             | ``a != b``                  |               |
   | <TestCase.assertNotEqual>`              |                             |               |
   +-----------------------------------------+-----------------------------+---------------+
   | :meth:`assertTrue(x)                    | ``bool(x) is True``         |               |
   | <TestCase.assertTrue>`                  |                             |               |
   +-----------------------------------------+-----------------------------+---------------+
   | :meth:`assertFalse(x)                   | ``bool(x) is False``        |               |
   | <TestCase.assertFalse>`                 |                             |               |
   +-----------------------------------------+-----------------------------+---------------+
   | :meth:`assertIs(a, b)                   | ``a is b``                  | 3.1           |
   | <TestCase.assertIs>`                    |                             |               |
   +-----------------------------------------+-----------------------------+---------------+
   | :meth:`assertIsNot(a, b)                | ``a is not b``              | 3.1           |
   | <TestCase.assertIsNot>`                 |                             |               |
   +-----------------------------------------+-----------------------------+---------------+
   | :meth:`assertIsNone(x)                  | ``x is None``               | 3.1           |
   | <TestCase.assertIsNone>`                |                             |               |
   +-----------------------------------------+-----------------------------+---------------+
   | :meth:`assertIsNotNone(x)               | ``x is not None``           | 3.1           |
   | <TestCase.assertIsNotNone>`             |                             |               |
   +-----------------------------------------+-----------------------------+---------------+
   | :meth:`assertIn(a, b)                   | ``a in b``                  | 3.1           |
   | <TestCase.assertIn>`                    |                             |               |
   +-----------------------------------------+-----------------------------+---------------+
   | :meth:`assertNotIn(a, b)                | ``a not in b``              | 3.1           |
   | <TestCase.assertNotIn>`                 |                             |               |
   +-----------------------------------------+-----------------------------+---------------+
   | :meth:`assertIsInstance(a, b)           | ``isinstance(a, b)``        | 3.2           |
   | <TestCase.assertIsInstance>`            |                             |               |
   +-----------------------------------------+-----------------------------+---------------+
   | :meth:`assertNotIsInstance(a, b)        | ``not isinstance(a, b)``    | 3.2           |
   | <TestCase.assertNotIsInstance>`         |                             |               |
   +-----------------------------------------+-----------------------------+---------------+
811

812 813 814 815 816
   All the assert methods accept a *msg* argument that, if specified, is used
   as the error message on failure (see also :data:`longMessage`).
   Note that the *msg* keyword argument can be passed to :meth:`assertRaises`,
   :meth:`assertRaisesRegex`, :meth:`assertWarns`, :meth:`assertWarnsRegex`
   only when they are used as a context manager.
817

818
   .. method:: assertEqual(first, second, msg=None)
819

820
      Test that *first* and *second* are equal.  If the values do not
821
      compare equal, the test will fail.
822

823
      In addition, if *first* and *second* are the exact same type and one of
824
      list, tuple, dict, set, frozenset or str or any type that a subclass
825
      registers with :meth:`addTypeEqualityFunc` the type-specific equality
826
      function will be called in order to generate a more useful default
827 828
      error message (see also the :ref:`list of type-specific methods
      <type-specific-methods>`).
829

830
      .. versionchanged:: 3.1
831
         Added the automatic calling of type-specific equality function.
832

833 834 835
      .. versionchanged:: 3.2
         :meth:`assertMultiLineEqual` added as the default type equality
         function for comparing strings.
836

837

838
   .. method:: assertNotEqual(first, second, msg=None)
839

840
      Test that *first* and *second* are not equal.  If the values do
841
      compare equal, the test will fail.
842

843
   .. method:: assertTrue(expr, msg=None)
844
               assertFalse(expr, msg=None)
Benjamin Peterson's avatar
Benjamin Peterson committed
845

846
      Test that *expr* is true (or false).
Benjamin Peterson's avatar
Benjamin Peterson committed
847

848 849 850 851 852
      Note that this is equivalent to ``bool(expr) is True`` and not to ``expr
      is True`` (use ``assertIs(expr, True)`` for the latter).  This method
      should also be avoided when more specific methods are available (e.g.
      ``assertEqual(a, b)`` instead of ``assertTrue(a == b)``), because they
      provide a better error message in case of failure.
Benjamin Peterson's avatar
Benjamin Peterson committed
853

854

855 856
   .. method:: assertIs(first, second, msg=None)
               assertIsNot(first, second, msg=None)
857

858
      Test that *first* and *second* evaluate (or don't evaluate) to the
859
      same object.
860

861
      .. versionadded:: 3.1
862 863


864
   .. method:: assertIsNone(expr, msg=None)
865
               assertIsNotNone(expr, msg=None)
866

867
      Test that *expr* is (or is not) None.
Benjamin Peterson's avatar
Benjamin Peterson committed
868

869
      .. versionadded:: 3.1
Benjamin Peterson's avatar
Benjamin Peterson committed
870 871


872 873 874
   .. method:: assertIn(first, second, msg=None)
               assertNotIn(first, second, msg=None)

875
      Test that *first* is (or is not) in *second*.
876

877
      .. versionadded:: 3.1
878 879


880
   .. method:: assertIsInstance(obj, cls, msg=None)
881
               assertNotIsInstance(obj, cls, msg=None)
882

883 884
      Test that *obj* is (or is not) an instance of *cls* (which can be a
      class or a tuple of classes, as supported by :func:`isinstance`).
885
      To check for the exact type, use :func:`assertIs(type(obj), cls) <assertIs>`.
886

887
      .. versionadded:: 3.2
888 889 890



891 892
   It is also possible to check the production of exceptions, warnings and
   log messages using the following methods:
893

894 895 896
   +---------------------------------------------------------+--------------------------------------+------------+
   | Method                                                  | Checks that                          | New in     |
   +=========================================================+======================================+============+
897
   | :meth:`assertRaises(exc, fun, *args, **kwds)            | ``fun(*args, **kwds)`` raises *exc*  |            |
898 899
   | <TestCase.assertRaises>`                                |                                      |            |
   +---------------------------------------------------------+--------------------------------------+------------+
900 901
   | :meth:`assertRaisesRegex(exc, r, fun, *args, **kwds)    | ``fun(*args, **kwds)`` raises *exc*  | 3.1        |
   | <TestCase.assertRaisesRegex>`                           | and the message matches regex *r*    |            |
902
   +---------------------------------------------------------+--------------------------------------+------------+
903
   | :meth:`assertWarns(warn, fun, *args, **kwds)            | ``fun(*args, **kwds)`` raises *warn* | 3.2        |
904 905
   | <TestCase.assertWarns>`                                 |                                      |            |
   +---------------------------------------------------------+--------------------------------------+------------+
906 907
   | :meth:`assertWarnsRegex(warn, r, fun, *args, **kwds)    | ``fun(*args, **kwds)`` raises *warn* | 3.2        |
   | <TestCase.assertWarnsRegex>`                            | and the message matches regex *r*    |            |
908
   +---------------------------------------------------------+--------------------------------------+------------+
909 910
   | :meth:`assertLogs(logger, level)                        | The ``with`` block logs on *logger*  | 3.4        |
   | <TestCase.assertLogs>`                                  | with minimum *level*                 |            |
911
   +---------------------------------------------------------+--------------------------------------+------------+
912

913
   .. method:: assertRaises(exception, callable, *args, **kwds)
914
               assertRaises(exception, msg=None)
915

916 917 918 919 920 921
      Test that an exception is raised when *callable* is called with any
      positional or keyword arguments that are also passed to
      :meth:`assertRaises`.  The test passes if *exception* is raised, is an
      error if another exception is raised, or fails if no exception is raised.
      To catch any of a group of exceptions, a tuple containing the exception
      classes may be passed as *exception*.
922

923 924 925
      If only the *exception* and possibly the *msg* arguments are given,
      return a context manager so that the code under test can be written
      inline rather than as a function::
926

927
         with self.assertRaises(SomeException):
928 929
             do_something()

930 931 932
      When used as a context manager, :meth:`assertRaises` accepts the
      additional keyword argument *msg*.

933
      The context manager will store the caught exception object in its
934
      :attr:`exception` attribute.  This can be useful if the intention
935
      is to perform additional checks on the exception raised::
936

937 938
         with self.assertRaises(SomeException) as cm:
             do_something()
939

940 941
         the_exception = cm.exception
         self.assertEqual(the_exception.error_code, 3)
942

943
      .. versionchanged:: 3.1
944
         Added the ability to use :meth:`assertRaises` as a context manager.
945

946 947 948
      .. versionchanged:: 3.2
         Added the :attr:`exception` attribute.

949 950 951
      .. versionchanged:: 3.3
         Added the *msg* keyword argument when used as a context manager.

952

953
   .. method:: assertRaisesRegex(exception, regex, callable, *args, **kwds)
954
               assertRaisesRegex(exception, regex, msg=None)
955

956 957
      Like :meth:`assertRaises` but also tests that *regex* matches
      on the string representation of the raised exception.  *regex* may be
958 959 960
      a regular expression object or a string containing a regular expression
      suitable for use by :func:`re.search`.  Examples::

961
         self.assertRaisesRegex(ValueError, "invalid literal for.*XYZ'$",
962
                                int, 'XYZ')
963 964 965

      or::

966
         with self.assertRaisesRegex(ValueError, 'literal'):
967 968
            int('XYZ')

969 970
      .. versionadded:: 3.1
         under the name ``assertRaisesRegexp``.
971

972
      .. versionchanged:: 3.2
973
         Renamed to :meth:`assertRaisesRegex`.
974

975 976 977
      .. versionchanged:: 3.3
         Added the *msg* keyword argument when used as a context manager.

978

979
   .. method:: assertWarns(warning, callable, *args, **kwds)
980
               assertWarns(warning, msg=None)
981 982 983 984

      Test that a warning is triggered when *callable* is called with any
      positional or keyword arguments that are also passed to
      :meth:`assertWarns`.  The test passes if *warning* is triggered and
985
      fails if it isn't.  Any exception is an error.
986 987 988
      To catch any of a group of warnings, a tuple containing the warning
      classes may be passed as *warnings*.

989
      If only the *warning* and possibly the *msg* arguments are given,
990
      return a context manager so that the code under test can be written
991
      inline rather than as a function::
992 993 994 995

         with self.assertWarns(SomeWarning):
             do_something()

996
      When used as a context manager, :meth:`assertWarns` accepts the
997 998
      additional keyword argument *msg*.

999 1000 1001 1002
      The context manager will store the caught warning object in its
      :attr:`warning` attribute, and the source line which triggered the
      warnings in the :attr:`filename` and :attr:`lineno` attributes.
      This can be useful if the intention is to perform additional checks
1003
      on the warning caught::
1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015

         with self.assertWarns(SomeWarning) as cm:
             do_something()

         self.assertIn('myfile.py', cm.filename)
         self.assertEqual(320, cm.lineno)

      This method works regardless of the warning filters in place when it
      is called.

      .. versionadded:: 3.2

1016 1017 1018
      .. versionchanged:: 3.3
         Added the *msg* keyword argument when used as a context manager.

1019

1020
   .. method:: assertWarnsRegex(warning, regex, callable, *args, **kwds)
1021
               assertWarnsRegex(warning, regex, msg=None)
1022

1023 1024
      Like :meth:`assertWarns` but also tests that *regex* matches on the
      message of the triggered warning.  *regex* may be a regular expression
1025 1026 1027
      object or a string containing a regular expression suitable for use
      by :func:`re.search`.  Example::

1028 1029 1030
         self.assertWarnsRegex(DeprecationWarning,
                               r'legacy_function\(\) is deprecated',
                               legacy_function, 'XYZ')
1031 1032 1033

      or::

1034
         with self.assertWarnsRegex(RuntimeWarning, 'unsafe frobnicating'):
1035 1036 1037 1038
             frobnicate('/etc/passwd')

      .. versionadded:: 3.2

1039 1040
      .. versionchanged:: 3.3
         Added the *msg* keyword argument when used as a context manager.
1041

1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082
   .. method:: assertLogs(logger=None, level=None)

      A context manager to test that at least one message is logged on
      the *logger* or one of its children, with at least the given
      *level*.

      If given, *logger* should be a :class:`logging.Logger` object or a
      :class:`str` giving the name of a logger.  The default is the root
      logger, which will catch all messages.

      If given, *level* should be either a numeric logging level or
      its string equivalent (for example either ``"ERROR"`` or
      :attr:`logging.ERROR`).  The default is :attr:`logging.INFO`.

      The test passes if at least one message emitted inside the ``with``
      block matches the *logger* and *level* conditions, otherwise it fails.

      The object returned by the context manager is a recording helper
      which keeps tracks of the matching log messages.  It has two
      attributes:

      .. attribute:: records

         A list of :class:`logging.LogRecord` objects of the matching
         log messages.

      .. attribute:: output

         A list of :class:`str` objects with the formatted output of
         matching messages.

      Example::

         with self.assertLogs('foo', level='INFO') as cm:
            logging.getLogger('foo').info('first message')
            logging.getLogger('foo.bar').error('second message')
         self.assertEqual(cm.output, ['INFO:foo:first message',
                                      'ERROR:foo.bar:second message'])

      .. versionadded:: 3.4

1083

1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106
   There are also other methods used to perform more specific checks, such as:

   +---------------------------------------+--------------------------------+--------------+
   | Method                                | Checks that                    | New in       |
   +=======================================+================================+==============+
   | :meth:`assertAlmostEqual(a, b)        | ``round(a-b, 7) == 0``         |              |
   | <TestCase.assertAlmostEqual>`         |                                |              |
   +---------------------------------------+--------------------------------+--------------+
   | :meth:`assertNotAlmostEqual(a, b)     | ``round(a-b, 7) != 0``         |              |
   | <TestCase.assertNotAlmostEqual>`      |                                |              |
   +---------------------------------------+--------------------------------+--------------+
   | :meth:`assertGreater(a, b)            | ``a > b``                      | 3.1          |
   | <TestCase.assertGreater>`             |                                |              |
   +---------------------------------------+--------------------------------+--------------+
   | :meth:`assertGreaterEqual(a, b)       | ``a >= b``                     | 3.1          |
   | <TestCase.assertGreaterEqual>`        |                                |              |
   +---------------------------------------+--------------------------------+--------------+
   | :meth:`assertLess(a, b)               | ``a < b``                      | 3.1          |
   | <TestCase.assertLess>`                |                                |              |
   +---------------------------------------+--------------------------------+--------------+
   | :meth:`assertLessEqual(a, b)          | ``a <= b``                     | 3.1          |
   | <TestCase.assertLessEqual>`           |                                |              |
   +---------------------------------------+--------------------------------+--------------+
1107
   | :meth:`assertRegex(s, r)              | ``r.search(s)``                | 3.1          |
1108
   | <TestCase.assertRegex>`               |                                |              |
1109
   +---------------------------------------+--------------------------------+--------------+
1110
   | :meth:`assertNotRegex(s, r)           | ``not r.search(s)``            | 3.2          |
1111
   | <TestCase.assertNotRegex>`            |                                |              |
1112
   +---------------------------------------+--------------------------------+--------------+
1113
   | :meth:`assertCountEqual(a, b)         | *a* and *b* have the same      | 3.2          |
1114
   | <TestCase.assertCountEqual>`          | elements in the same number,   |              |
1115 1116
   |                                       | regardless of their order      |              |
   +---------------------------------------+--------------------------------+--------------+
1117

1118

1119 1120
   .. method:: assertAlmostEqual(first, second, places=7, msg=None, delta=None)
               assertNotAlmostEqual(first, second, places=7, msg=None, delta=None)
1121

1122
      Test that *first* and *second* are approximately (or not approximately)
1123 1124 1125 1126
      equal by computing the difference, rounding to the given number of
      decimal *places* (default 7), and comparing to zero.  Note that these
      methods round the values to the given number of *decimal places* (i.e.
      like the :func:`round` function) and not *significant digits*.
1127

1128
      If *delta* is supplied instead of *places* then the difference
1129
      between *first* and *second* must be less or equal to (or greater than) *delta*.
1130

1131
      Supplying both *delta* and *places* raises a ``TypeError``.
1132

1133
      .. versionchanged:: 3.2
1134 1135 1136
         :meth:`assertAlmostEqual` automatically considers almost equal objects
         that compare equal.  :meth:`assertNotAlmostEqual` automatically fails
         if the objects compare equal.  Added the *delta* keyword argument.
Benjamin Peterson's avatar
Benjamin Peterson committed
1137

1138

1139 1140 1141 1142
   .. method:: assertGreater(first, second, msg=None)
               assertGreaterEqual(first, second, msg=None)
               assertLess(first, second, msg=None)
               assertLessEqual(first, second, msg=None)
1143

1144
      Test that *first* is respectively >, >=, < or <= than *second* depending
1145
      on the method name.  If not, the test will fail::
1146 1147 1148

         >>> self.assertGreaterEqual(3, 4)
         AssertionError: "3" unexpectedly not greater than or equal to "4"
Benjamin Peterson's avatar
Benjamin Peterson committed
1149

1150
      .. versionadded:: 3.1
Benjamin Peterson's avatar
Benjamin Peterson committed
1151 1152


1153 1154
   .. method:: assertRegex(text, regex, msg=None)
               assertNotRegex(text, regex, msg=None)
Benjamin Peterson's avatar
Benjamin Peterson committed
1155

1156
      Test that a *regex* search matches (or does not match) *text*.  In case
1157
      of failure, the error message will include the pattern and the *text* (or
1158
      the pattern and the part of *text* that unexpectedly matched).  *regex*
1159 1160
      may be a regular expression object or a string containing a regular
      expression suitable for use by :func:`re.search`.
1161

1162 1163
      .. versionadded:: 3.1
         under the name ``assertRegexpMatches``.
1164
      .. versionchanged:: 3.2
1165 1166 1167 1168
         The method ``assertRegexpMatches()`` has been renamed to
         :meth:`.assertRegex`.
      .. versionadded:: 3.2
         :meth:`.assertNotRegex`.
1169 1170


1171
   .. method:: assertCountEqual(first, second, msg=None)
1172

1173
      Test that sequence *first* contains the same elements as *second*,
1174 1175 1176
      regardless of their order. When they don't, an error message listing the
      differences between the sequences will be generated.

1177 1178
      Duplicate elements are *not* ignored when comparing *first* and
      *second*. It verifies whether each element has the same count in both
1179
      sequences. Equivalent to:
1180
      ``assertEqual(Counter(list(first)), Counter(list(second)))``
1181
      but works with sequences of unhashable objects as well.
1182

1183 1184
      .. versionadded:: 3.2

1185

1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204
   .. _type-specific-methods:

   The :meth:`assertEqual` method dispatches the equality check for objects of
   the same type to different type-specific methods.  These methods are already
   implemented for most of the built-in types, but it's also possible to
   register new methods using :meth:`addTypeEqualityFunc`:

   .. method:: addTypeEqualityFunc(typeobj, function)

      Registers a type-specific method called by :meth:`assertEqual` to check
      if two objects of exactly the same *typeobj* (not subclasses) compare
      equal.  *function* must take two positional arguments and a third msg=None
      keyword argument just as :meth:`assertEqual` does.  It must raise
      :data:`self.failureException(msg) <failureException>` when inequality
      between the first two parameters is detected -- possibly providing useful
      information and explaining the inequalities in details in the error
      message.

      .. versionadded:: 3.1
1205

1206 1207 1208
   The list of type-specific methods automatically used by
   :meth:`~TestCase.assertEqual` are summarized in the following table.  Note
   that it's usually not necessary to invoke these methods directly.
1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233

   +-----------------------------------------+-----------------------------+--------------+
   | Method                                  | Used to compare             | New in       |
   +=========================================+=============================+==============+
   | :meth:`assertMultiLineEqual(a, b)       | strings                     | 3.1          |
   | <TestCase.assertMultiLineEqual>`        |                             |              |
   +-----------------------------------------+-----------------------------+--------------+
   | :meth:`assertSequenceEqual(a, b)        | sequences                   | 3.1          |
   | <TestCase.assertSequenceEqual>`         |                             |              |
   +-----------------------------------------+-----------------------------+--------------+
   | :meth:`assertListEqual(a, b)            | lists                       | 3.1          |
   | <TestCase.assertListEqual>`             |                             |              |
   +-----------------------------------------+-----------------------------+--------------+
   | :meth:`assertTupleEqual(a, b)           | tuples                      | 3.1          |
   | <TestCase.assertTupleEqual>`            |                             |              |
   +-----------------------------------------+-----------------------------+--------------+
   | :meth:`assertSetEqual(a, b)             | sets or frozensets          | 3.1          |
   | <TestCase.assertSetEqual>`              |                             |              |
   +-----------------------------------------+-----------------------------+--------------+
   | :meth:`assertDictEqual(a, b)            | dicts                       | 3.1          |
   | <TestCase.assertDictEqual>`             |                             |              |
   +-----------------------------------------+-----------------------------+--------------+



1234
   .. method:: assertMultiLineEqual(first, second, msg=None)
1235

1236
      Test that the multiline string *first* is equal to the string *second*.
1237 1238 1239 1240 1241 1242 1243
      When not equal a diff of the two strings highlighting the differences
      will be included in the error message. This method is used by default
      when comparing strings with :meth:`assertEqual`.

      .. versionadded:: 3.1


1244
   .. method:: assertSequenceEqual(first, second, msg=None, seq_type=None)
1245 1246

      Tests that two sequences are equal.  If a *seq_type* is supplied, both
1247
      *first* and *second* must be instances of *seq_type* or a failure will
1248 1249 1250
      be raised.  If the sequences are different an error message is
      constructed that shows the difference between the two.

1251 1252
      This method is not called directly by :meth:`assertEqual`, but
      it's used to implement :meth:`assertListEqual` and
1253 1254 1255 1256 1257
      :meth:`assertTupleEqual`.

      .. versionadded:: 3.1


1258 1259
   .. method:: assertListEqual(first, second, msg=None)
               assertTupleEqual(first, second, msg=None)
1260

Ezio Melotti's avatar
Ezio Melotti committed
1261
      Tests that two lists or tuples are equal.  If not, an error message is
1262 1263 1264 1265 1266 1267 1268 1269
      constructed that shows only the differences between the two.  An error
      is also raised if either of the parameters are of the wrong type.
      These methods are used by default when comparing lists or tuples with
      :meth:`assertEqual`.

      .. versionadded:: 3.1


1270
   .. method:: assertSetEqual(first, second, msg=None)
1271 1272 1273 1274 1275

      Tests that two sets are equal.  If not, an error message is constructed
      that lists the differences between the sets.  This method is used by
      default when comparing sets or frozensets with :meth:`assertEqual`.

1276
      Fails if either of *first* or *second* does not have a :meth:`set.difference`
1277 1278 1279 1280 1281
      method.

      .. versionadded:: 3.1


1282
   .. method:: assertDictEqual(first, second, msg=None)
1283 1284 1285 1286 1287 1288 1289 1290 1291 1292

      Test that two dictionaries are equal.  If not, an error message is
      constructed that shows the differences in the dictionaries. This
      method will be used by default to compare dictionaries in
      calls to :meth:`assertEqual`.

      .. versionadded:: 3.1



1293 1294
   .. _other-methods-and-attrs:

1295
   Finally the :class:`TestCase` provides the following methods and attributes:
1296

1297

1298
   .. method:: fail(msg=None)
1299

1300
      Signals a test failure unconditionally, with *msg* or ``None`` for
1301
      the error message.
1302 1303


1304
   .. attribute:: failureException
1305

1306 1307 1308 1309 1310
      This class attribute gives the exception raised by the test method.  If a
      test framework needs to use a specialized exception, possibly to carry
      additional information, it must subclass this exception in order to "play
      fair" with the framework.  The initial value of this attribute is
      :exc:`AssertionError`.
1311

1312 1313 1314

   .. attribute:: longMessage

1315
      If set to ``True`` then any explicit failure message you pass in to the
1316 1317 1318 1319
      :ref:`assert methods <assert-methods>` will be appended to the end of the
      normal failure message.  The normal messages contain useful information
      about the objects involved, for example the message from assertEqual
      shows you the repr of the two unequal objects. Setting this attribute
1320
      to ``True`` allows you to have a custom error message in addition to the
1321
      normal one.
1322

1323 1324
      This attribute defaults to ``True``. If set to False then a custom message
      passed to an assert method will silence the normal message.
1325 1326

      The class setting can be overridden in individual tests by assigning an
1327
      instance attribute to ``True`` or ``False`` before calling the assert methods.
1328

1329
      .. versionadded:: 3.1
1330 1331


1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346
   .. attribute:: maxDiff

      This attribute controls the maximum length of diffs output by assert
      methods that report diffs on failure. It defaults to 80*8 characters.
      Assert methods affected by this attribute are
      :meth:`assertSequenceEqual` (including all the sequence comparison
      methods that delegate to it), :meth:`assertDictEqual` and
      :meth:`assertMultiLineEqual`.

      Setting ``maxDiff`` to None means that there is no maximum length of
      diffs.

      .. versionadded:: 3.2


1347 1348
   Testing frameworks can use the following methods to collect information on
   the test:
1349 1350


1351
   .. method:: countTestCases()
1352

1353 1354
      Return the number of tests represented by this test object.  For
      :class:`TestCase` instances, this will always be ``1``.
1355 1356


1357
   .. method:: defaultTestResult()
1358

1359 1360 1361
      Return an instance of the test result class that should be used for this
      test case class (if no other result instance is provided to the
      :meth:`run` method).
1362

1363 1364 1365
      For :class:`TestCase` instances, this will always be an instance of
      :class:`TestResult`; subclasses of :class:`TestCase` should override this
      as necessary.
1366 1367


1368
   .. method:: id()
1369

1370 1371
      Return a string identifying the specific test case.  This is usually the
      full name of the test method, including the module and class name.
1372 1373


1374
   .. method:: shortDescription()
1375

1376
      Returns a description of the test, or ``None`` if no description
1377 1378
      has been provided.  The default implementation of this method
      returns the first line of the test method's docstring, if available,
1379
      or ``None``.
1380

1381
      .. versionchanged:: 3.1
1382
         In 3.1 this was changed to add the test name to the short description
1383
         even in the presence of a docstring.  This caused compatibility issues
1384
         with unittest extensions and adding the test name was moved to the
1385
         :class:`TextTestResult` in Python 3.2.
1386

1387

1388
   .. method:: addCleanup(function, *args, **kwargs)
Benjamin Peterson's avatar
Benjamin Peterson committed
1389 1390 1391 1392 1393 1394 1395 1396 1397 1398

      Add a function to be called after :meth:`tearDown` to cleanup resources
      used during the test. Functions will be called in reverse order to the
      order they are added (LIFO). They are called with any arguments and
      keyword arguments passed into :meth:`addCleanup` when they are
      added.

      If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
      then any cleanup functions added will still be called.

1399
      .. versionadded:: 3.1
Benjamin Peterson's avatar
Benjamin Peterson committed
1400 1401 1402 1403


   .. method:: doCleanups()

Barry Warsaw's avatar
Barry Warsaw committed
1404
      This method is called unconditionally after :meth:`tearDown`, or
Benjamin Peterson's avatar
Benjamin Peterson committed
1405 1406 1407 1408 1409 1410 1411 1412 1413 1414
      after :meth:`setUp` if :meth:`setUp` raises an exception.

      It is responsible for calling all the cleanup functions added by
      :meth:`addCleanup`. If you need cleanup functions to be called
      *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
      yourself.

      :meth:`doCleanups` pops methods off the stack of cleanup
      functions one at a time, so it can be called at any time.

1415
      .. versionadded:: 3.1
Benjamin Peterson's avatar
Benjamin Peterson committed
1416 1417


1418
.. class:: FunctionTestCase(testFunc, setUp=None, tearDown=None, description=None)
1419

1420
   This class implements the portion of the :class:`TestCase` interface which
1421 1422 1423 1424
   allows the test runner to drive the test, but does not provide the methods
   which test code can use to check and report errors.  This is used to create
   test cases using legacy test code, allowing it to be integrated into a
   :mod:`unittest`-based test framework.
1425 1426


1427 1428
.. _deprecated-aliases:

1429 1430 1431 1432 1433 1434 1435
Deprecated aliases
##################

For historical reasons, some of the :class:`TestCase` methods had one or more
aliases that are now deprecated.  The following table lists the correct names
along with their deprecated aliases:

1436 1437 1438 1439 1440 1441
   ==============================  ====================== ======================
    Method Name                     Deprecated alias       Deprecated alias
   ==============================  ====================== ======================
    :meth:`.assertEqual`            failUnlessEqual        assertEquals
    :meth:`.assertNotEqual`         failIfEqual            assertNotEquals
    :meth:`.assertTrue`             failUnless             assert\_
1442 1443
    :meth:`.assertFalse`            failIf
    :meth:`.assertRaises`           failUnlessRaises
1444 1445
    :meth:`.assertAlmostEqual`      failUnlessAlmostEqual  assertAlmostEquals
    :meth:`.assertNotAlmostEqual`   failIfAlmostEqual      assertNotAlmostEquals
1446 1447
    :meth:`.assertRegex`                                   assertRegexpMatches
    :meth:`.assertRaisesRegex`                             assertRaisesRegexp
1448 1449
   ==============================  ====================== ======================

1450
   .. deprecated:: 3.1
1451 1452 1453
         the fail* aliases listed in the second column.
   .. deprecated:: 3.2
         the assert* aliases listed in the third column.
1454 1455 1456
   .. deprecated:: 3.2
         ``assertRegexpMatches`` and ``assertRaisesRegexp`` have been renamed to
         :meth:`.assertRegex` and :meth:`.assertRaisesRegex`
1457 1458


1459
.. _testsuite-objects:
1460

1461 1462
Grouping tests
~~~~~~~~~~~~~~
1463

1464
.. class:: TestSuite(tests=())
1465

1466 1467 1468 1469
   This class represents an aggregation of individual tests cases and test suites.
   The class presents the interface needed by the test runner to allow it to be run
   as any other test case.  Running a :class:`TestSuite` instance is the same as
   iterating over the suite, running each test individually.
1470

1471 1472 1473
   If *tests* is given, it must be an iterable of individual test cases or other
   test suites that will be used to build the suite initially. Additional methods
   are provided to add test cases and suites to the collection later on.
1474

1475 1476 1477 1478
   :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
   they do not actually implement a test.  Instead, they are used to aggregate
   tests into groups of tests that should be run together. Some additional
   methods are available to add tests to :class:`TestSuite` instances:
1479 1480


1481
   .. method:: TestSuite.addTest(test)
1482

1483
      Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1484 1485


1486
   .. method:: TestSuite.addTests(tests)
1487

1488 1489
      Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
      instances to this test suite.
1490

1491 1492
      This is equivalent to iterating over *tests*, calling :meth:`addTest` for
      each element.
1493

1494
   :class:`TestSuite` shares the following methods with :class:`TestCase`:
1495 1496


1497
   .. method:: run(result)
1498

1499 1500 1501 1502
      Run the tests associated with this suite, collecting the result into the
      test result object passed as *result*.  Note that unlike
      :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
      be passed in.
1503 1504


1505
   .. method:: debug()
1506

1507 1508 1509
      Run the tests associated with this suite without collecting the
      result. This allows exceptions raised by the test to be propagated to the
      caller and can be used to support running tests under a debugger.
1510 1511


1512
   .. method:: countTestCases()
1513

1514 1515
      Return the number of tests represented by this test object, including all
      individual tests and sub-suites.
1516

Benjamin Peterson's avatar
Benjamin Peterson committed
1517 1518 1519 1520 1521

   .. method:: __iter__()

      Tests grouped by a :class:`TestSuite` are always accessed by iteration.
      Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
1522 1523 1524 1525 1526 1527 1528
      that this method may be called several times on a single suite (for
      example when counting tests or comparing for equality) so the tests
      returned by repeated iterations before :meth:`TestSuite.run` must be the
      same for each call iteration. After :meth:`TestSuite.run`, callers should
      not rely on the tests returned by this method unless the caller uses a
      subclass that overrides :meth:`TestSuite._removeTestAtIndex` to preserve
      test references.
Benjamin Peterson's avatar
Benjamin Peterson committed
1529

1530
      .. versionchanged:: 3.2
Benjamin Peterson's avatar
Benjamin Peterson committed
1531 1532 1533 1534
         In earlier versions the :class:`TestSuite` accessed tests directly rather
         than through iteration, so overriding :meth:`__iter__` wasn't sufficient
         for providing tests.

1535 1536 1537 1538 1539
      .. versionchanged:: 3.4
         In earlier versions the :class:`TestSuite` held references to each
         :class:`TestCase` after :meth:`TestSuite.run`. Subclasses can restore
         that behavior by overriding :meth:`TestSuite._removeTestAtIndex`.

1540 1541
   In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
   is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1542 1543


1544 1545
Loading and running tests
~~~~~~~~~~~~~~~~~~~~~~~~~
1546

1547
.. class:: TestLoader()
1548

1549 1550 1551
   The :class:`TestLoader` class is used to create test suites from classes and
   modules.  Normally, there is no need to create an instance of this class; the
   :mod:`unittest` module provides an instance that can be shared as
1552 1553
   :data:`unittest.defaultTestLoader`.  Using a subclass or instance, however,
   allows customization of some configurable properties.
1554

1555
   :class:`TestLoader` objects have the following methods:
1556

1557

1558
   .. method:: loadTestsFromTestCase(testCaseClass)
1559

1560 1561
      Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
      :class:`testCaseClass`.
1562 1563


1564
   .. method:: loadTestsFromModule(module)
1565

1566 1567 1568 1569
      Return a suite of all tests cases contained in the given module. This
      method searches *module* for classes derived from :class:`TestCase` and
      creates an instance of the class for each test method defined for the
      class.
1570

1571
      .. note::
1572

1573 1574 1575 1576 1577
         While using a hierarchy of :class:`TestCase`\ -derived classes can be
         convenient in sharing fixtures and helper functions, defining test
         methods on base classes that are not intended to be instantiated
         directly does not play well with this method.  Doing so, however, can
         be useful when the fixtures are different and defined in subclasses.
1578

1579 1580 1581 1582
      If a module provides a ``load_tests`` function it will be called to
      load the tests. This allows modules to customize test loading.
      This is the `load_tests protocol`_.

1583
      .. versionchanged:: 3.2
1584 1585
         Support for ``load_tests`` added.

1586

1587
   .. method:: loadTestsFromName(name, module=None)
1588

1589
      Return a suite of all tests cases given a string specifier.
1590

1591 1592 1593 1594 1595 1596 1597
      The specifier *name* is a "dotted name" that may resolve either to a
      module, a test case class, a test method within a test case class, a
      :class:`TestSuite` instance, or a callable object which returns a
      :class:`TestCase` or :class:`TestSuite` instance.  These checks are
      applied in the order listed here; that is, a method on a possible test
      case class will be picked up as "a test method within a test case class",
      rather than "a callable object".
1598

1599 1600 1601
      For example, if you have a module :mod:`SampleTests` containing a
      :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
      methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
1602 1603 1604 1605 1606 1607
      specifier ``'SampleTests.SampleTestCase'`` would cause this method to
      return a suite which will run all three test methods. Using the specifier
      ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
      suite which will run only the :meth:`test_two` test method. The specifier
      can refer to modules and packages which have not been imported; they will
      be imported as a side-effect.
1608

1609
      The method optionally resolves *name* relative to the given *module*.
1610 1611


1612
   .. method:: loadTestsFromNames(names, module=None)
1613

1614 1615 1616
      Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
      than a single name.  The return value is a test suite which supports all
      the tests defined for each name.
1617 1618


1619
   .. method:: getTestCaseNames(testCaseClass)
1620

1621 1622
      Return a sorted sequence of method names found within *testCaseClass*;
      this should be a subclass of :class:`TestCase`.
1623

1624 1625 1626

   .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)

1627 1628 1629 1630 1631
      Find all the test modules by recursing into subdirectories from the
      specified start directory, and return a TestSuite object containing them.
      Only test files that match *pattern* will be loaded. (Using shell style
      pattern matching.) Only module names that are importable (i.e. are valid
      Python identifiers) will be loaded.
1632 1633 1634 1635 1636

      All test modules must be importable from the top level of the project. If
      the start directory is not the top level directory then the top level
      directory must be specified separately.

Benjamin Peterson's avatar
Benjamin Peterson committed
1637
      If importing a module fails, for example due to a syntax error, then this
1638
      will be recorded as a single error and discovery will continue.  If the
Ezio Melotti's avatar
Ezio Melotti committed
1639
      import failure is due to :exc:`SkipTest` being raised, it will be recorded
1640
      as a skip instead of an error.
Benjamin Peterson's avatar
Benjamin Peterson committed
1641

1642 1643 1644 1645 1646
      If a test package name (directory with :file:`__init__.py`) matches the
      pattern then the package will be checked for a ``load_tests``
      function. If this exists then it will be called with *loader*, *tests*,
      *pattern*.

1647
      If ``load_tests`` exists then discovery does *not* recurse into the package,
1648 1649 1650 1651 1652 1653 1654
      ``load_tests`` is responsible for loading all tests in the package.

      The pattern is deliberately not stored as a loader attribute so that
      packages can continue discovery themselves. *top_level_dir* is stored so
      ``load_tests`` does not need to pass this argument in to
      ``loader.discover()``.

Benjamin Peterson's avatar
Benjamin Peterson committed
1655 1656
      *start_dir* can be a dotted module name as well as a directory.

1657 1658
      .. versionadded:: 3.2

1659
      .. versionchanged:: 3.4
Ezio Melotti's avatar
Ezio Melotti committed
1660
         Modules that raise :exc:`SkipTest` on import are recorded as skips,
1661 1662 1663 1664 1665
           not errors.
         Discovery works for :term:`namespace packages <namespace package>`.
         Paths are sorted before being imported so that execution order is
           the same even if the underlying file system's ordering is not
           dependent on file name.
1666

1667

1668 1669
   The following attributes of a :class:`TestLoader` can be configured either by
   subclassing or assignment on an instance:
1670 1671


1672
   .. attribute:: testMethodPrefix
1673

1674 1675
      String giving the prefix of method names which will be interpreted as test
      methods.  The default value is ``'test'``.
1676

1677 1678
      This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
      methods.
1679 1680


1681
   .. attribute:: sortTestMethodsUsing
1682

1683 1684
      Function to be used to compare method names when sorting them in
      :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
1685 1686


1687
   .. attribute:: suiteClass
1688

1689 1690 1691
      Callable object that constructs a test suite from a list of tests. No
      methods on the resulting object are needed.  The default value is the
      :class:`TestSuite` class.
1692

1693
      This affects all the :meth:`loadTestsFrom\*` methods.
1694 1695


1696
.. class:: TestResult
1697

1698 1699
   This class is used to compile information about which tests have succeeded
   and which have failed.
1700

1701 1702 1703 1704
   A :class:`TestResult` object stores the results of a set of tests.  The
   :class:`TestCase` and :class:`TestSuite` classes ensure that results are
   properly recorded; test authors do not need to worry about recording the
   outcome of tests.
1705

1706 1707 1708 1709
   Testing frameworks built on top of :mod:`unittest` may want access to the
   :class:`TestResult` object generated by running a set of tests for reporting
   purposes; a :class:`TestResult` instance is returned by the
   :meth:`TestRunner.run` method for this purpose.
1710

1711 1712
   :class:`TestResult` instances have the following attributes that will be of
   interest when inspecting the results of running a set of tests:
1713 1714


1715
   .. attribute:: errors
1716

1717 1718 1719
      A list containing 2-tuples of :class:`TestCase` instances and strings
      holding formatted tracebacks. Each tuple represents a test which raised an
      unexpected exception.
1720

1721
   .. attribute:: failures
1722

1723 1724
      A list containing 2-tuples of :class:`TestCase` instances and strings
      holding formatted tracebacks. Each tuple represents a test where a failure
1725
      was explicitly signalled using the :meth:`TestCase.assert\*` methods.
1726

1727
   .. attribute:: skipped
1728

1729 1730
      A list containing 2-tuples of :class:`TestCase` instances and strings
      holding the reason for skipping the test.
1731

1732
      .. versionadded:: 3.1
1733

1734
   .. attribute:: expectedFailures
1735

Georg Brandl's avatar
Georg Brandl committed
1736 1737
      A list containing 2-tuples of :class:`TestCase` instances and strings
      holding formatted tracebacks.  Each tuple represents an expected failure
1738
      of the test case.
1739

1740
   .. attribute:: unexpectedSuccesses
1741

1742 1743
      A list containing :class:`TestCase` instances that were marked as expected
      failures, but succeeded.
1744

1745
   .. attribute:: shouldStop
1746

1747
      Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1748 1749


1750
   .. attribute:: testsRun
1751

1752
      The total number of tests run so far.
1753 1754


1755
   .. attribute:: buffer
Benjamin Peterson's avatar
Benjamin Peterson committed
1756 1757 1758 1759 1760 1761

      If set to true, ``sys.stdout`` and ``sys.stderr`` will be buffered in between
      :meth:`startTest` and :meth:`stopTest` being called. Collected output will
      only be echoed onto the real ``sys.stdout`` and ``sys.stderr`` if the test
      fails or errors. Any output is also attached to the failure / error message.

1762
      .. versionadded:: 3.2
Benjamin Peterson's avatar
Benjamin Peterson committed
1763 1764 1765 1766 1767 1768 1769


   .. attribute:: failfast

      If set to true :meth:`stop` will be called on the first failure or error,
      halting the test run.

1770
      .. versionadded:: 3.2
Benjamin Peterson's avatar
Benjamin Peterson committed
1771 1772


1773
   .. method:: wasSuccessful()
1774

1775 1776
      Return ``True`` if all tests run so far have passed, otherwise returns
      ``False``.
1777

1778 1779 1780 1781
      .. versionchanged:: 3.4
         Returns ``False`` if there were any :attr:`unexpectedSuccesses`
         from tests marked with the :func:`expectedFailure` decorator.

1782

1783
   .. method:: stop()
1784

1785
      This method can be called to signal that the set of tests being run should
1786
      be aborted by setting the :attr:`shouldStop` attribute to ``True``.
1787 1788
      :class:`TestRunner` objects should respect this flag and return without
      running any additional tests.
1789

1790 1791 1792 1793
      For example, this feature is used by the :class:`TextTestRunner` class to
      stop the test framework when the user signals an interrupt from the
      keyboard.  Interactive tools which provide :class:`TestRunner`
      implementations can use this in a similar manner.
1794

1795 1796 1797 1798
   The following methods of the :class:`TestResult` class are used to maintain
   the internal data structures, and may be extended in subclasses to support
   additional reporting requirements.  This is particularly useful in building
   tools which support interactive reporting while tests are being run.
1799

1800

1801
   .. method:: startTest(test)
1802

1803
      Called when the test case *test* is about to be run.
1804

1805
   .. method:: stopTest(test)
1806

1807 1808
      Called after the test case *test* has been executed, regardless of the
      outcome.
1809

1810
   .. method:: startTestRun()
Benjamin Peterson's avatar
Benjamin Peterson committed
1811 1812 1813

      Called once before any tests are executed.

1814
      .. versionadded:: 3.1
Benjamin Peterson's avatar
Benjamin Peterson committed
1815 1816


1817
   .. method:: stopTestRun()
Benjamin Peterson's avatar
Benjamin Peterson committed
1818

1819
      Called once after all tests are executed.
Benjamin Peterson's avatar
Benjamin Peterson committed
1820

1821
      .. versionadded:: 3.1
Benjamin Peterson's avatar
Benjamin Peterson committed
1822 1823


1824
   .. method:: addError(test, err)
1825

1826
      Called when the test case *test* raises an unexpected exception. *err* is a
1827 1828
      tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
      traceback)``.
1829

1830 1831 1832
      The default implementation appends a tuple ``(test, formatted_err)`` to
      the instance's :attr:`errors` attribute, where *formatted_err* is a
      formatted traceback derived from *err*.
1833 1834


1835
   .. method:: addFailure(test, err)
1836

1837 1838
      Called when the test case *test* signals a failure. *err* is a tuple of
      the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
1839

1840 1841 1842
      The default implementation appends a tuple ``(test, formatted_err)`` to
      the instance's :attr:`failures` attribute, where *formatted_err* is a
      formatted traceback derived from *err*.
1843 1844


1845
   .. method:: addSuccess(test)
1846

1847
      Called when the test case *test* succeeds.
1848

1849
      The default implementation does nothing.
1850 1851


1852
   .. method:: addSkip(test, reason)
1853

1854 1855
      Called when the test case *test* is skipped.  *reason* is the reason the
      test gave for skipping.
1856

1857 1858
      The default implementation appends a tuple ``(test, reason)`` to the
      instance's :attr:`skipped` attribute.
1859 1860


1861
   .. method:: addExpectedFailure(test, err)
1862

1863 1864
      Called when the test case *test* fails, but was marked with the
      :func:`expectedFailure` decorator.
1865

1866 1867 1868
      The default implementation appends a tuple ``(test, formatted_err)`` to
      the instance's :attr:`expectedFailures` attribute, where *formatted_err*
      is a formatted traceback derived from *err*.
1869 1870


1871
   .. method:: addUnexpectedSuccess(test)
1872

1873 1874
      Called when the test case *test* was marked with the
      :func:`expectedFailure` decorator, but succeeded.
1875

1876 1877
      The default implementation appends the test to the instance's
      :attr:`unexpectedSuccesses` attribute.
1878

1879

1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895
   .. method:: addSubTest(test, subtest, outcome)

      Called when a subtest finishes.  *test* is the test case
      corresponding to the test method.  *subtest* is a custom
      :class:`TestCase` instance describing the subtest.

      If *outcome* is :const:`None`, the subtest succeeded.  Otherwise,
      it failed with an exception where *outcome* is a tuple of the form
      returned by :func:`sys.exc_info`: ``(type, value, traceback)``.

      The default implementation does nothing when the outcome is a
      success, and records subtest failures as normal failures.

      .. versionadded:: 3.4


1896 1897
.. class:: TextTestResult(stream, descriptions, verbosity)

1898 1899 1900 1901 1902 1903
   A concrete implementation of :class:`TestResult` used by the
   :class:`TextTestRunner`.

   .. versionadded:: 3.2
      This class was previously named ``_TextTestResult``. The old name still
      exists as an alias but is deprecated.
1904

1905

1906
.. data:: defaultTestLoader
1907

1908 1909 1910
   Instance of the :class:`TestLoader` class intended to be shared.  If no
   customization of the :class:`TestLoader` is needed, this instance can be used
   instead of repeatedly creating new instances.
1911 1912


1913 1914
.. class:: TextTestRunner(stream=None, descriptions=True, verbosity=1, failfast=False, \
                          buffer=False, resultclass=None, warnings=None)
1915

1916
   A basic test runner implementation that outputs results to a stream. If *stream*
1917
   is ``None``, the default, :data:`sys.stderr` is used as the output stream. This class
1918 1919
   has a few configurable parameters, but is essentially very simple.  Graphical
   applications which run test suites should provide alternate implementations.
1920

1921
   By default this runner shows :exc:`DeprecationWarning`,
1922
   :exc:`PendingDeprecationWarning`, :exc:`ResourceWarning` and
1923 1924 1925 1926 1927 1928 1929
   :exc:`ImportWarning` even if they are :ref:`ignored by default
   <warning-ignored>`. Deprecation warnings caused by :ref:`deprecated unittest
   methods <deprecated-aliases>` are also special-cased and, when the warning
   filters are ``'default'`` or ``'always'``, they will appear only once
   per-module, in order to avoid too many warning messages.  This behavior can
   be overridden using the :option:`-Wd` or :option:`-Wa` options and leaving
   *warnings* to ``None``.
1930

1931 1932 1933 1934
   .. versionchanged:: 3.2
      Added the ``warnings`` argument.

   .. versionchanged:: 3.2
1935
      The default stream is set to :data:`sys.stderr` at instantiation time rather
1936 1937
      than import time.

Benjamin Peterson's avatar
Benjamin Peterson committed
1938 1939 1940 1941 1942
   .. method:: _makeResult()

      This method returns the instance of ``TestResult`` used by :meth:`run`.
      It is not intended to be called directly, but can be overridden in
      subclasses to provide a custom ``TestResult``.
1943

1944 1945
      ``_makeResult()`` instantiates the class or callable passed in the
      ``TextTestRunner`` constructor as the ``resultclass`` argument. It
Benjamin Peterson's avatar
Benjamin Peterson committed
1946
      defaults to :class:`TextTestResult` if no ``resultclass`` is provided.
1947 1948 1949
      The result class is instantiated with the following arguments::

        stream, descriptions, verbosity
Benjamin Peterson's avatar
Benjamin Peterson committed
1950

1951 1952 1953 1954 1955 1956 1957 1958
   .. method:: run(test)

      This method is the main public interface to the `TextTestRunner`. This
      method takes a :class:`TestSuite` or :class:`TestCase` instance. A
      :class:`TestResult` is created by calling
      :func:`_makeResult` and the test(s) are run and the
      results printed to stdout.

1959

1960
.. function:: main(module='__main__', defaultTest=None, argv=None, testRunner=None, \
1961
                   testLoader=unittest.defaultTestLoader, exit=True, verbosity=1, \
1962
                   failfast=None, catchbreak=None, buffer=None, warnings=None)
1963

1964 1965 1966 1967
   A command-line program that loads a set of tests from *module* and runs them;
   this is primarily for making test modules conveniently executable.
   The simplest use for this function is to include the following line at the
   end of a test script::
1968 1969 1970 1971

      if __name__ == '__main__':
          unittest.main()

1972 1973 1974 1975 1976 1977
   You can run tests with more detailed information by passing in the verbosity
   argument::

      if __name__ == '__main__':
          unittest.main(verbosity=2)

1978 1979 1980 1981
   The *defaultTest* argument is either the name of a single test or an
   iterable of test names to run if no test names are specified via *argv*.  If
   not specified or ``None`` and no test names are provided via *argv*, all
   tests found in *module* are run.
1982

1983 1984 1985 1986
   The *argv* argument can be a list of options passed to the program, with the
   first element being the program name.  If not specified or ``None``,
   the values of :data:`sys.argv` are used.

1987
   The *testRunner* argument can either be a test runner class or an already
Benjamin Peterson's avatar
Benjamin Peterson committed
1988 1989 1990
   created instance of it. By default ``main`` calls :func:`sys.exit` with
   an exit code indicating success or failure of the tests run.

1991 1992 1993
   The *testLoader* argument has to be a :class:`TestLoader` instance,
   and defaults to :data:`defaultTestLoader`.

Benjamin Peterson's avatar
Benjamin Peterson committed
1994 1995 1996 1997 1998 1999 2000
   ``main`` supports being used from the interactive interpreter by passing in the
   argument ``exit=False``. This displays the result on standard output without
   calling :func:`sys.exit`::

      >>> from unittest import main
      >>> main(module='test_module', exit=False)

2001
   The *failfast*, *catchbreak* and *buffer* parameters have the same
2002
   effect as the same-name `command-line options`_.
Benjamin Peterson's avatar
Benjamin Peterson committed
2003

2004 2005 2006 2007 2008
   The *warning* argument specifies the :ref:`warning filter <warning-filter>`
   that should be used while running the tests.  If it's not specified, it will
   remain ``None`` if a :option:`-W` option is passed to :program:`python`,
   otherwise it will be set to ``'default'``.

Benjamin Peterson's avatar
Benjamin Peterson committed
2009 2010 2011
   Calling ``main`` actually returns an instance of the ``TestProgram`` class.
   This stores the result of the tests run as the ``result`` attribute.

Éric Araujo's avatar
Éric Araujo committed
2012
   .. versionchanged:: 3.1
2013
      The *exit* parameter was added.
Éric Araujo's avatar
Éric Araujo committed
2014

2015
   .. versionchanged:: 3.2
2016 2017
      The *verbosity*, *failfast*, *catchbreak*, *buffer*
      and *warnings* parameters were added.
2018

2019 2020 2021 2022
   .. versionchanged:: 3.4
      The *defaultTest* parameter was changed to also accept an iterable of
      test names.

2023 2024 2025 2026

load_tests Protocol
###################

2027
.. versionadded:: 3.2
Benjamin Peterson's avatar
Benjamin Peterson committed
2028

2029 2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062
Modules or packages can customize how tests are loaded from them during normal
test runs or test discovery by implementing a function called ``load_tests``.

If a test module defines ``load_tests`` it will be called by
:meth:`TestLoader.loadTestsFromModule` with the following arguments::

    load_tests(loader, standard_tests, None)

It should return a :class:`TestSuite`.

*loader* is the instance of :class:`TestLoader` doing the loading.
*standard_tests* are the tests that would be loaded by default from the
module. It is common for test modules to only want to add or remove tests
from the standard set of tests.
The third argument is used when loading packages as part of test discovery.

A typical ``load_tests`` function that loads tests from a specific set of
:class:`TestCase` classes may look like::

    test_cases = (TestCase1, TestCase2, TestCase3)

    def load_tests(loader, tests, pattern):
        suite = TestSuite()
        for test_class in test_cases:
            tests = loader.loadTestsFromTestCase(test_class)
            suite.addTests(tests)
        return suite

If discovery is started, either from the command line or by calling
:meth:`TestLoader.discover`, with a pattern that matches a package
name then the package :file:`__init__.py` will be checked for ``load_tests``.

.. note::

2063 2064
   The default pattern is ``'test*.py'``. This matches all Python files
   that start with ``'test'`` but *won't* match any test directories.
2065

2066
   A pattern like ``'test*'`` will match test packages as well as
2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088
   modules.

If the package :file:`__init__.py` defines ``load_tests`` then it will be
called and discovery not continued into the package. ``load_tests``
is called with the following arguments::

    load_tests(loader, standard_tests, pattern)

This should return a :class:`TestSuite` representing all the tests
from the package. (``standard_tests`` will only contain tests
collected from :file:`__init__.py`.)

Because the pattern is passed into ``load_tests`` the package is free to
continue (and potentially modify) test discovery. A 'do nothing'
``load_tests`` function for a test package would look like::

    def load_tests(loader, standard_tests, pattern):
        # top level directory cached on loader instance
        this_dir = os.path.dirname(__file__)
        package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
        standard_tests.addTests(package_tests)
        return standard_tests
Benjamin Peterson's avatar
Benjamin Peterson committed
2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149


Class and Module Fixtures
-------------------------

Class and module level fixtures are implemented in :class:`TestSuite`. When
the test suite encounters a test from a new class then :meth:`tearDownClass`
from the previous class (if there is one) is called, followed by
:meth:`setUpClass` from the new class.

Similarly if a test is from a different module from the previous test then
``tearDownModule`` from the previous module is run, followed by
``setUpModule`` from the new module.

After all the tests have run the final ``tearDownClass`` and
``tearDownModule`` are run.

Note that shared fixtures do not play well with [potential] features like test
parallelization and they break test isolation. They should be used with care.

The default ordering of tests created by the unittest test loaders is to group
all tests from the same modules and classes together. This will lead to
``setUpClass`` / ``setUpModule`` (etc) being called exactly once per class and
module. If you randomize the order, so that tests from different modules and
classes are adjacent to each other, then these shared fixture functions may be
called multiple times in a single test run.

Shared fixtures are not intended to work with suites with non-standard
ordering. A ``BaseTestSuite`` still exists for frameworks that don't want to
support shared fixtures.

If there are any exceptions raised during one of the shared fixture functions
the test is reported as an error. Because there is no corresponding test
instance an ``_ErrorHolder`` object (that has the same interface as a
:class:`TestCase`) is created to represent the error. If you are just using
the standard unittest test runner then this detail doesn't matter, but if you
are a framework author it may be relevant.


setUpClass and tearDownClass
~~~~~~~~~~~~~~~~~~~~~~~~~~~~

These must be implemented as class methods::

    import unittest

    class Test(unittest.TestCase):
        @classmethod
        def setUpClass(cls):
            cls._connection = createExpensiveConnectionObject()

        @classmethod
        def tearDownClass(cls):
            cls._connection.destroy()

If you want the ``setUpClass`` and ``tearDownClass`` on base classes called
then you must call up to them yourself. The implementations in
:class:`TestCase` are empty.

If an exception is raised during a ``setUpClass`` then the tests in the class
are not run and the ``tearDownClass`` is not run. Skipped classes will not
2150
have ``setUpClass`` or ``tearDownClass`` run. If the exception is a
2151
:exc:`SkipTest` exception then the class will be reported as having been skipped
2152
instead of as an error.
Benjamin Peterson's avatar
Benjamin Peterson committed
2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166


setUpModule and tearDownModule
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

These should be implemented as functions::

    def setUpModule():
        createConnection()

    def tearDownModule():
        closeConnection()

If an exception is raised in a ``setUpModule`` then none of the tests in the
2167
module will be run and the ``tearDownModule`` will not be run. If the exception is a
2168
:exc:`SkipTest` exception then the module will be reported as having been skipped
2169
instead of as an error.
Benjamin Peterson's avatar
Benjamin Peterson committed
2170 2171 2172 2173 2174


Signal Handling
---------------

2175 2176
.. versionadded:: 3.2

2177
The :option:`-c/--catch <unittest -c>` command-line option to unittest,
2178 2179 2180 2181 2182
along with the ``catchbreak`` parameter to :func:`unittest.main()`, provide
more friendly handling of control-C during a test run. With catch break
behavior enabled control-C will allow the currently running test to complete,
and the test run will then end and report all the results so far. A second
control-c will raise a :exc:`KeyboardInterrupt` in the usual way.
Benjamin Peterson's avatar
Benjamin Peterson committed
2183

2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194
The control-c handling signal handler attempts to remain compatible with code or
tests that install their own :const:`signal.SIGINT` handler. If the ``unittest``
handler is called but *isn't* the installed :const:`signal.SIGINT` handler,
i.e. it has been replaced by the system under test and delegated to, then it
calls the default handler. This will normally be the expected behavior by code
that replaces an installed handler and delegates to it. For individual tests
that need ``unittest`` control-c handling disabled the :func:`removeHandler`
decorator can be used.

There are a few utility functions for framework authors to enable control-c
handling functionality within test frameworks.
Benjamin Peterson's avatar
Benjamin Peterson committed
2195 2196 2197 2198 2199 2200 2201

.. function:: installHandler()

   Install the control-c handler. When a :const:`signal.SIGINT` is received
   (usually in response to the user pressing control-c) all registered results
   have :meth:`~TestResult.stop` called.

2202

Benjamin Peterson's avatar
Benjamin Peterson committed
2203 2204 2205 2206 2207 2208
.. function:: registerResult(result)

   Register a :class:`TestResult` object for control-c handling. Registering a
   result stores a weak reference to it, so it doesn't prevent the result from
   being garbage collected.

2209 2210 2211 2212
   Registering a :class:`TestResult` object has no side-effects if control-c
   handling is not enabled, so test frameworks can unconditionally register
   all results they create independently of whether or not handling is enabled.

2213

Benjamin Peterson's avatar
Benjamin Peterson committed
2214 2215 2216 2217 2218 2219
.. function:: removeResult(result)

   Remove a registered result. Once a result has been removed then
   :meth:`~TestResult.stop` will no longer be called on that result object in
   response to a control-c.

2220

2221 2222 2223 2224 2225 2226 2227 2228 2229
.. function:: removeHandler(function=None)

   When called without arguments this function removes the control-c handler
   if it has been installed. This function can also be used as a test decorator
   to temporarily remove the handler whilst the test is being executed::

      @unittest.removeHandler
      def test_signal_handling(self):
          ...