blob: a4e2cab1388f5a1557a62993088a6f8d9ceb59b8 [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001:mod:`unittest` --- Unit testing framework
2==========================================
3
4.. module:: unittest
5 :synopsis: Unit testing framework for Python.
6.. moduleauthor:: Steve Purcell <stephen_purcell@yahoo.com>
7.. sectionauthor:: Steve Purcell <stephen_purcell@yahoo.com>
8.. sectionauthor:: Fred L. Drake, Jr. <fdrake@acm.org>
9.. sectionauthor:: Raymond Hettinger <python@rcn.com>
10
Ezio Melotti22170ed2010-11-20 09:57:27 +000011(If you are already familiar with the basic concepts of testing, you might want
12to skip to :ref:`the list of assert methods <assert-methods>`.)
Georg Brandl116aa622007-08-15 14:28:22 +000013
Georg Brandl116aa622007-08-15 14:28:22 +000014The Python unit testing framework, sometimes referred to as "PyUnit," is a
15Python language version of JUnit, by Kent Beck and Erich Gamma. JUnit is, in
16turn, a Java version of Kent's Smalltalk testing framework. Each is the de
17facto standard unit testing framework for its respective language.
18
19:mod:`unittest` supports test automation, sharing of setup and shutdown code for
20tests, aggregation of tests into collections, and independence of the tests from
21the reporting framework. The :mod:`unittest` module provides classes that make
22it easy to support these qualities for a set of tests.
23
24To achieve this, :mod:`unittest` supports some important concepts:
25
26test fixture
27 A :dfn:`test fixture` represents the preparation needed to perform one or more
28 tests, and any associate cleanup actions. This may involve, for example,
29 creating temporary or proxy databases, directories, or starting a server
30 process.
31
32test case
33 A :dfn:`test case` is the smallest unit of testing. It checks for a specific
34 response to a particular set of inputs. :mod:`unittest` provides a base class,
35 :class:`TestCase`, which may be used to create new test cases.
36
37test suite
38 A :dfn:`test suite` is a collection of test cases, test suites, or both. It is
39 used to aggregate tests that should be executed together.
40
41test runner
42 A :dfn:`test runner` is a component which orchestrates the execution of tests
43 and provides the outcome to the user. The runner may use a graphical interface,
44 a textual interface, or return a special value to indicate the results of
45 executing the tests.
46
47The test case and test fixture concepts are supported through the
48:class:`TestCase` and :class:`FunctionTestCase` classes; the former should be
49used when creating new tests, and the latter can be used when integrating
50existing test code with a :mod:`unittest`\ -driven framework. When building test
Benjamin Peterson52baa292009-03-24 00:56:30 +000051fixtures using :class:`TestCase`, the :meth:`~TestCase.setUp` and
52:meth:`~TestCase.tearDown` methods can be overridden to provide initialization
53and cleanup for the fixture. With :class:`FunctionTestCase`, existing functions
54can be passed to the constructor for these purposes. When the test is run, the
55fixture initialization is run first; if it succeeds, the cleanup method is run
56after the test has been executed, regardless of the outcome of the test. Each
57instance of the :class:`TestCase` will only be used to run a single test method,
58so a new fixture is created for each test.
Georg Brandl116aa622007-08-15 14:28:22 +000059
60Test suites are implemented by the :class:`TestSuite` class. This class allows
61individual tests and test suites to be aggregated; when the suite is executed,
Benjamin Peterson14a3dd72009-05-25 00:51:58 +000062all tests added directly to the suite and in "child" test suites are run.
Georg Brandl116aa622007-08-15 14:28:22 +000063
Benjamin Peterson52baa292009-03-24 00:56:30 +000064A test runner is an object that provides a single method,
65:meth:`~TestRunner.run`, which accepts a :class:`TestCase` or :class:`TestSuite`
66object as a parameter, and returns a result object. The class
67:class:`TestResult` is provided for use as the result object. :mod:`unittest`
68provides the :class:`TextTestRunner` as an example test runner which reports
69test results on the standard error stream by default. Alternate runners can be
70implemented for other environments (such as graphical environments) without any
71need to derive from a specific class.
Georg Brandl116aa622007-08-15 14:28:22 +000072
73
74.. seealso::
75
76 Module :mod:`doctest`
77 Another test-support module with a very different flavor.
78
Benjamin Petersonb48af542010-04-11 20:43:16 +000079 `unittest2: A backport of new unittest features for Python 2.4-2.6 <http://pypi.python.org/pypi/unittest2>`_
80 Many new features were added to unittest in Python 2.7, including test
81 discovery. unittest2 allows you to use these features with earlier
82 versions of Python.
83
Georg Brandl116aa622007-08-15 14:28:22 +000084 `Simple Smalltalk Testing: With Patterns <http://www.XProgramming.com/testfram.htm>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000085 Kent Beck's original paper on testing frameworks using the pattern shared
86 by :mod:`unittest`.
Georg Brandl116aa622007-08-15 14:28:22 +000087
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000088 `Nose <http://code.google.com/p/python-nose/>`_ and `py.test <http://pytest.org>`_
Benjamin Petersond2397752009-06-27 23:45:02 +000089 Third-party unittest frameworks with a lighter-weight syntax for writing
90 tests. For example, ``assert func(10) == 42``.
Raymond Hettinger6b232cd2009-03-24 00:22:53 +000091
Benjamin Petersonb48af542010-04-11 20:43:16 +000092 `The Python Testing Tools Taxonomy <http://pycheesecake.org/wiki/PythonTestingToolsTaxonomy>`_
93 An extensive list of Python testing tools including functional testing
94 frameworks and mock object libraries.
Benjamin Petersond2397752009-06-27 23:45:02 +000095
Benjamin Petersonb48af542010-04-11 20:43:16 +000096 `Testing in Python Mailing List <http://lists.idyll.org/listinfo/testing-in-python>`_
97 A special-interest-group for discussion of testing, and testing tools,
98 in Python.
Benjamin Petersond2397752009-06-27 23:45:02 +000099
Michael Foord90efac72011-01-03 15:39:49 +0000100 The script :file:`Tools/unittestgui/unittestgui.py` in the Python source distribution is
101 a GUI tool for test discovery and execution. This is intended largely for ease of use
102 for those new to unit testing. For production environments it is recommended that
103 tests be driven by a continuous integration system such as `Hudson <http://hudson-ci.org/>`_
104 or `Buildbot <http://buildbot.net/trac>`_.
105
106
Georg Brandl116aa622007-08-15 14:28:22 +0000107.. _unittest-minimal-example:
108
109Basic example
110-------------
111
112The :mod:`unittest` module provides a rich set of tools for constructing and
113running tests. This section demonstrates that a small subset of the tools
114suffice to meet the needs of most users.
115
116Here is a short script to test three functions from the :mod:`random` module::
117
118 import random
119 import unittest
120
121 class TestSequenceFunctions(unittest.TestCase):
122
123 def setUp(self):
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000124 self.seq = list(range(10))
Georg Brandl116aa622007-08-15 14:28:22 +0000125
Benjamin Peterson52baa292009-03-24 00:56:30 +0000126 def test_shuffle(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000127 # make sure the shuffled sequence does not lose any elements
128 random.shuffle(self.seq)
129 self.seq.sort()
Benjamin Petersonbe0e1772009-07-25 01:02:01 +0000130 self.assertEqual(self.seq, list(range(10)))
Georg Brandl116aa622007-08-15 14:28:22 +0000131
Benjamin Peterson847a4112010-03-14 15:04:17 +0000132 # should raise an exception for an immutable sequence
133 self.assertRaises(TypeError, random.shuffle, (1,2,3))
134
Benjamin Peterson52baa292009-03-24 00:56:30 +0000135 def test_choice(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000136 element = random.choice(self.seq)
Benjamin Peterson847a4112010-03-14 15:04:17 +0000137 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000138
Benjamin Peterson52baa292009-03-24 00:56:30 +0000139 def test_sample(self):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000140 with self.assertRaises(ValueError):
141 random.sample(self.seq, 20)
Georg Brandl116aa622007-08-15 14:28:22 +0000142 for element in random.sample(self.seq, 5):
Benjamin Peterson847a4112010-03-14 15:04:17 +0000143 self.assertTrue(element in self.seq)
Georg Brandl116aa622007-08-15 14:28:22 +0000144
145 if __name__ == '__main__':
146 unittest.main()
147
Benjamin Peterson52baa292009-03-24 00:56:30 +0000148A testcase is created by subclassing :class:`unittest.TestCase`. The three
Georg Brandl116aa622007-08-15 14:28:22 +0000149individual tests are defined with methods whose names start with the letters
150``test``. This naming convention informs the test runner about which methods
151represent tests.
152
Benjamin Peterson52baa292009-03-24 00:56:30 +0000153The crux of each test is a call to :meth:`~TestCase.assertEqual` to check for an
Michael Foord34c94622010-02-10 15:51:42 +0000154expected result; :meth:`~TestCase.assertTrue` to verify a condition; or
Benjamin Peterson52baa292009-03-24 00:56:30 +0000155:meth:`~TestCase.assertRaises` to verify that an expected exception gets raised.
156These methods are used instead of the :keyword:`assert` statement so the test
157runner can accumulate all test results and produce a report.
Georg Brandl116aa622007-08-15 14:28:22 +0000158
Benjamin Peterson52baa292009-03-24 00:56:30 +0000159When a :meth:`~TestCase.setUp` method is defined, the test runner will run that
160method prior to each test. Likewise, if a :meth:`~TestCase.tearDown` method is
161defined, the test runner will invoke that method after each test. In the
162example, :meth:`~TestCase.setUp` was used to create a fresh sequence for each
163test.
Georg Brandl116aa622007-08-15 14:28:22 +0000164
165The final block shows a simple way to run the tests. :func:`unittest.main`
Éric Araujo713d3032010-11-18 16:38:46 +0000166provides a command-line interface to the test script. When run from the command
Georg Brandl116aa622007-08-15 14:28:22 +0000167line, the above script produces an output that looks like this::
168
169 ...
170 ----------------------------------------------------------------------
171 Ran 3 tests in 0.000s
172
173 OK
174
175Instead of :func:`unittest.main`, there are other ways to run the tests with a
176finer level of control, less terse output, and no requirement to be run from the
177command line. For example, the last two lines may be replaced with::
178
179 suite = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)
180 unittest.TextTestRunner(verbosity=2).run(suite)
181
182Running the revised script from the interpreter or another script produces the
183following output::
184
Ezio Melottid59e44a2010-02-28 03:46:13 +0000185 test_choice (__main__.TestSequenceFunctions) ... ok
186 test_sample (__main__.TestSequenceFunctions) ... ok
187 test_shuffle (__main__.TestSequenceFunctions) ... ok
Georg Brandl116aa622007-08-15 14:28:22 +0000188
189 ----------------------------------------------------------------------
190 Ran 3 tests in 0.110s
191
192 OK
193
194The above examples show the most commonly used :mod:`unittest` features which
195are sufficient to meet many everyday testing needs. The remainder of the
196documentation explores the full feature set from first principles.
197
Benjamin Petersonb48af542010-04-11 20:43:16 +0000198
199.. _unittest-command-line-interface:
200
Éric Araujo76338ec2010-11-26 23:46:18 +0000201Command-Line Interface
Benjamin Petersonb48af542010-04-11 20:43:16 +0000202----------------------
203
204The unittest module can be used from the command line to run tests from
205modules, classes or even individual test methods::
206
207 python -m unittest test_module1 test_module2
208 python -m unittest test_module.TestClass
209 python -m unittest test_module.TestClass.test_method
210
211You can pass in a list with any combination of module names, and fully
212qualified class or method names.
213
Michael Foord37d120a2010-12-04 01:11:21 +0000214Test modules can be specified by file path as well::
215
216 python -m unittest tests/test_something.py
217
218This allows you to use the shell filename completion to specify the test module.
219The file specified must still be importable as a module. The path is converted
220to a module name by removing the '.py' and converting path separators into '.'.
221If you want to execute a test file that isn't importable as a module you should
222execute the file directly instead.
223
Benjamin Petersonb48af542010-04-11 20:43:16 +0000224You can run tests with more detail (higher verbosity) by passing in the -v flag::
225
226 python -m unittest -v test_module
227
Michael Foord086f3082010-11-21 21:28:01 +0000228When executed without arguments :ref:`unittest-test-discovery` is started::
229
230 python -m unittest
231
Éric Araujo713d3032010-11-18 16:38:46 +0000232For a list of all the command-line options::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000233
234 python -m unittest -h
235
Georg Brandl67b21b72010-08-17 15:07:14 +0000236.. versionchanged:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000237 In earlier versions it was only possible to run individual test methods and
238 not modules or classes.
239
240
Éric Araujo76338ec2010-11-26 23:46:18 +0000241Command-line options
242~~~~~~~~~~~~~~~~~~~~
Benjamin Petersonb48af542010-04-11 20:43:16 +0000243
Éric Araujod3309df2010-11-21 03:09:17 +0000244:program:`unittest` supports these command-line options:
Benjamin Petersonb48af542010-04-11 20:43:16 +0000245
Éric Araujo713d3032010-11-18 16:38:46 +0000246.. program:: unittest
Benjamin Petersonb48af542010-04-11 20:43:16 +0000247
Éric Araujo713d3032010-11-18 16:38:46 +0000248.. cmdoption:: -b, --buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +0000249
Éric Araujo713d3032010-11-18 16:38:46 +0000250 The standard output and standard error streams are buffered during the test
251 run. Output during a passing test is discarded. Output is echoed normally
252 on test fail or error and is added to the failure messages.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000253
Éric Araujo713d3032010-11-18 16:38:46 +0000254.. cmdoption:: -c, --catch
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000255
Éric Araujo713d3032010-11-18 16:38:46 +0000256 Control-C during the test run waits for the current test to end and then
257 reports all the results so far. A second control-C raises the normal
258 :exc:`KeyboardInterrupt` exception.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000259
Éric Araujo713d3032010-11-18 16:38:46 +0000260 See `Signal Handling`_ for the functions that provide this functionality.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000261
Éric Araujo713d3032010-11-18 16:38:46 +0000262.. cmdoption:: -f, --failfast
263
264 Stop the test run on the first error or failure.
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000265
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000266.. versionadded:: 3.2
Éric Araujod6c5f742010-12-16 00:07:01 +0000267 The command-line options ``-b``, ``-c`` and ``-f`` were added.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000268
269The command line can also be used for test discovery, for running all of the
270tests in a project or just a subset.
271
272
273.. _unittest-test-discovery:
274
275Test Discovery
276--------------
277
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000278.. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +0000279
Ezio Melotti3d995842011-03-08 16:17:35 +0200280Unittest supports simple test discovery. In order to be compatible with test
281discovery, all of the test files must be :ref:`modules <tut-modules>` or
282:ref:`packages <tut-packages>` importable from the top-level directory of
283the project (this means that their filenames must be valid
284:ref:`identifiers <identifiers>`).
Benjamin Petersonb48af542010-04-11 20:43:16 +0000285
286Test discovery is implemented in :meth:`TestLoader.discover`, but can also be
Éric Araujo713d3032010-11-18 16:38:46 +0000287used from the command line. The basic command-line usage is::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000288
289 cd project_directory
290 python -m unittest discover
291
Michael Foord086f3082010-11-21 21:28:01 +0000292.. note::
293
294 As a shortcut, ``python -m unittest`` is the equivalent of
295 ``python -m unittest discover``. If you want to pass arguments to test
296 discovery the `discover` sub-command must be used explicitly.
297
Benjamin Petersonb48af542010-04-11 20:43:16 +0000298The ``discover`` sub-command has the following options:
299
Éric Araujo713d3032010-11-18 16:38:46 +0000300.. program:: unittest discover
301
302.. cmdoption:: -v, --verbose
303
304 Verbose output
305
306.. cmdoption:: -s directory
307
308 Directory to start discovery ('.' default)
309
310.. cmdoption:: -p pattern
311
312 Pattern to match test files ('test*.py' default)
313
314.. cmdoption:: -t directory
315
316 Top level directory of project (defaults to start directory)
Benjamin Petersonb48af542010-04-11 20:43:16 +0000317
Benjamin Petersond7c3ed52010-06-27 22:32:30 +0000318The :option:`-s`, :option:`-p`, and :option:`-t` options can be passed in
319as positional arguments in that order. The following two command lines
320are equivalent::
Benjamin Petersonb48af542010-04-11 20:43:16 +0000321
322 python -m unittest discover -s project_directory -p '*_test.py'
323 python -m unittest discover project_directory '*_test.py'
324
Michael Foord16f3e902010-05-08 15:13:42 +0000325As well as being a path it is possible to pass a package name, for example
326``myproject.subpackage.test``, as the start directory. The package name you
327supply will then be imported and its location on the filesystem will be used
328as the start directory.
329
330.. caution::
331
Senthil Kumaran916bd382010-10-15 12:55:19 +0000332 Test discovery loads tests by importing them. Once test discovery has found
333 all the test files from the start directory you specify it turns the paths
334 into package names to import. For example :file:`foo/bar/baz.py` will be
Michael Foord16f3e902010-05-08 15:13:42 +0000335 imported as ``foo.bar.baz``.
336
337 If you have a package installed globally and attempt test discovery on
338 a different copy of the package then the import *could* happen from the
339 wrong place. If this happens test discovery will warn you and exit.
340
341 If you supply the start directory as a package name rather than a
342 path to a directory then discover assumes that whichever location it
343 imports from is the location you intended, so you will not get the
344 warning.
345
Benjamin Petersonb48af542010-04-11 20:43:16 +0000346Test modules and packages can customize test loading and discovery by through
347the `load_tests protocol`_.
348
349
Georg Brandl116aa622007-08-15 14:28:22 +0000350.. _organizing-tests:
351
352Organizing test code
353--------------------
354
355The basic building blocks of unit testing are :dfn:`test cases` --- single
356scenarios that must be set up and checked for correctness. In :mod:`unittest`,
Raymond Hettinger833ad0e2011-02-06 21:00:38 +0000357test cases are represented by :class:`unittest.TestCase` instances.
358To make your own test cases you must write subclasses of
359:class:`TestCase` or use :class:`FunctionTestCase`.
Georg Brandl116aa622007-08-15 14:28:22 +0000360
361An instance of a :class:`TestCase`\ -derived class is an object that can
362completely run a single test method, together with optional set-up and tidy-up
363code.
364
365The testing code of a :class:`TestCase` instance should be entirely self
366contained, such that it can be run either in isolation or in arbitrary
367combination with any number of other test cases.
368
Benjamin Peterson52baa292009-03-24 00:56:30 +0000369The simplest :class:`TestCase` subclass will simply override the
370:meth:`~TestCase.runTest` method in order to perform specific testing code::
Georg Brandl116aa622007-08-15 14:28:22 +0000371
372 import unittest
373
374 class DefaultWidgetSizeTestCase(unittest.TestCase):
375 def runTest(self):
376 widget = Widget('The widget')
377 self.assertEqual(widget.size(), (50, 50), 'incorrect default size')
378
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000379Note that in order to test something, we use the one of the :meth:`assert\*`
Benjamin Petersond2397752009-06-27 23:45:02 +0000380methods provided by the :class:`TestCase` base class. If the test fails, an
381exception will be raised, and :mod:`unittest` will identify the test case as a
382:dfn:`failure`. Any other exceptions will be treated as :dfn:`errors`. This
383helps you identify where the problem is: :dfn:`failures` are caused by incorrect
384results - a 5 where you expected a 6. :dfn:`Errors` are caused by incorrect
385code - e.g., a :exc:`TypeError` caused by an incorrect function call.
Georg Brandl116aa622007-08-15 14:28:22 +0000386
387The way to run a test case will be described later. For now, note that to
388construct an instance of such a test case, we call its constructor without
389arguments::
390
391 testCase = DefaultWidgetSizeTestCase()
392
393Now, such test cases can be numerous, and their set-up can be repetitive. In
394the above case, constructing a :class:`Widget` in each of 100 Widget test case
395subclasses would mean unsightly duplication.
396
397Luckily, we can factor out such set-up code by implementing a method called
Benjamin Peterson52baa292009-03-24 00:56:30 +0000398:meth:`~TestCase.setUp`, which the testing framework will automatically call for
399us when we run the test::
Georg Brandl116aa622007-08-15 14:28:22 +0000400
401 import unittest
402
403 class SimpleWidgetTestCase(unittest.TestCase):
404 def setUp(self):
405 self.widget = Widget('The widget')
406
407 class DefaultWidgetSizeTestCase(SimpleWidgetTestCase):
408 def runTest(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000409 self.assertEqual(self.widget.size(), (50,50),
410 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000411
412 class WidgetResizeTestCase(SimpleWidgetTestCase):
413 def runTest(self):
414 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000415 self.assertEqual(self.widget.size(), (100,150),
416 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000417
Benjamin Peterson52baa292009-03-24 00:56:30 +0000418If the :meth:`~TestCase.setUp` method raises an exception while the test is
419running, the framework will consider the test to have suffered an error, and the
420:meth:`~TestCase.runTest` method will not be executed.
Georg Brandl116aa622007-08-15 14:28:22 +0000421
Benjamin Peterson52baa292009-03-24 00:56:30 +0000422Similarly, we can provide a :meth:`~TestCase.tearDown` method that tidies up
423after the :meth:`~TestCase.runTest` method has been run::
Georg Brandl116aa622007-08-15 14:28:22 +0000424
425 import unittest
426
427 class SimpleWidgetTestCase(unittest.TestCase):
428 def setUp(self):
429 self.widget = Widget('The widget')
430
431 def tearDown(self):
432 self.widget.dispose()
433 self.widget = None
434
Benjamin Peterson52baa292009-03-24 00:56:30 +0000435If :meth:`~TestCase.setUp` succeeded, the :meth:`~TestCase.tearDown` method will
436be run whether :meth:`~TestCase.runTest` succeeded or not.
Georg Brandl116aa622007-08-15 14:28:22 +0000437
438Such a working environment for the testing code is called a :dfn:`fixture`.
439
440Often, many small test cases will use the same fixture. In this case, we would
441end up subclassing :class:`SimpleWidgetTestCase` into many small one-method
442classes such as :class:`DefaultWidgetSizeTestCase`. This is time-consuming and
Georg Brandl116aa622007-08-15 14:28:22 +0000443discouraging, so in the same vein as JUnit, :mod:`unittest` provides a simpler
444mechanism::
445
446 import unittest
447
448 class WidgetTestCase(unittest.TestCase):
449 def setUp(self):
450 self.widget = Widget('The widget')
451
452 def tearDown(self):
453 self.widget.dispose()
454 self.widget = None
455
Ezio Melottid59e44a2010-02-28 03:46:13 +0000456 def test_default_size(self):
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000457 self.assertEqual(self.widget.size(), (50,50),
458 'incorrect default size')
Georg Brandl116aa622007-08-15 14:28:22 +0000459
Ezio Melottid59e44a2010-02-28 03:46:13 +0000460 def test_resize(self):
Georg Brandl116aa622007-08-15 14:28:22 +0000461 self.widget.resize(100,150)
Ezio Melotti2d6c39b2010-02-04 20:27:41 +0000462 self.assertEqual(self.widget.size(), (100,150),
463 'wrong size after resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000464
Benjamin Peterson52baa292009-03-24 00:56:30 +0000465Here we have not provided a :meth:`~TestCase.runTest` method, but have instead
466provided two different test methods. Class instances will now each run one of
Ezio Melottid59e44a2010-02-28 03:46:13 +0000467the :meth:`test_\*` methods, with ``self.widget`` created and destroyed
Benjamin Peterson52baa292009-03-24 00:56:30 +0000468separately for each instance. When creating an instance we must specify the
469test method it is to run. We do this by passing the method name in the
470constructor::
Georg Brandl116aa622007-08-15 14:28:22 +0000471
Ezio Melottid59e44a2010-02-28 03:46:13 +0000472 defaultSizeTestCase = WidgetTestCase('test_default_size')
473 resizeTestCase = WidgetTestCase('test_resize')
Georg Brandl116aa622007-08-15 14:28:22 +0000474
475Test case instances are grouped together according to the features they test.
476:mod:`unittest` provides a mechanism for this: the :dfn:`test suite`,
477represented by :mod:`unittest`'s :class:`TestSuite` class::
478
479 widgetTestSuite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000480 widgetTestSuite.addTest(WidgetTestCase('test_default_size'))
481 widgetTestSuite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000482
483For the ease of running tests, as we will see later, it is a good idea to
484provide in each test module a callable object that returns a pre-built test
485suite::
486
487 def suite():
488 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000489 suite.addTest(WidgetTestCase('test_default_size'))
490 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000491 return suite
492
493or even::
494
495 def suite():
Ezio Melottid59e44a2010-02-28 03:46:13 +0000496 tests = ['test_default_size', 'test_resize']
Georg Brandl116aa622007-08-15 14:28:22 +0000497
498 return unittest.TestSuite(map(WidgetTestCase, tests))
499
500Since it is a common pattern to create a :class:`TestCase` subclass with many
501similarly named test functions, :mod:`unittest` provides a :class:`TestLoader`
502class that can be used to automate the process of creating a test suite and
503populating it with individual tests. For example, ::
504
505 suite = unittest.TestLoader().loadTestsFromTestCase(WidgetTestCase)
506
Ezio Melottid59e44a2010-02-28 03:46:13 +0000507will create a test suite that will run ``WidgetTestCase.test_default_size()`` and
508``WidgetTestCase.test_resize``. :class:`TestLoader` uses the ``'test'`` method
Georg Brandl116aa622007-08-15 14:28:22 +0000509name prefix to identify test methods automatically.
510
Mark Dickinsonc48d8342009-02-01 14:18:10 +0000511Note that the order in which the various test cases will be run is
512determined by sorting the test function names with respect to the
513built-in ordering for strings.
Georg Brandl116aa622007-08-15 14:28:22 +0000514
515Often it is desirable to group suites of test cases together, so as to run tests
516for the whole system at once. This is easy, since :class:`TestSuite` instances
517can be added to a :class:`TestSuite` just as :class:`TestCase` instances can be
518added to a :class:`TestSuite`::
519
520 suite1 = module1.TheTestSuite()
521 suite2 = module2.TheTestSuite()
522 alltests = unittest.TestSuite([suite1, suite2])
523
524You can place the definitions of test cases and test suites in the same modules
525as the code they are to test (such as :file:`widget.py`), but there are several
526advantages to placing the test code in a separate module, such as
527:file:`test_widget.py`:
528
529* The test module can be run standalone from the command line.
530
531* The test code can more easily be separated from shipped code.
532
533* There is less temptation to change test code to fit the code it tests without
534 a good reason.
535
536* Test code should be modified much less frequently than the code it tests.
537
538* Tested code can be refactored more easily.
539
540* Tests for modules written in C must be in separate modules anyway, so why not
541 be consistent?
542
543* If the testing strategy changes, there is no need to change the source code.
544
545
546.. _legacy-unit-tests:
547
548Re-using old test code
549----------------------
550
551Some users will find that they have existing test code that they would like to
552run from :mod:`unittest`, without converting every old test function to a
553:class:`TestCase` subclass.
554
555For this reason, :mod:`unittest` provides a :class:`FunctionTestCase` class.
556This subclass of :class:`TestCase` can be used to wrap an existing test
557function. Set-up and tear-down functions can also be provided.
558
559Given the following test function::
560
561 def testSomething():
562 something = makeSomething()
563 assert something.name is not None
564 # ...
565
566one can create an equivalent test case instance as follows::
567
568 testcase = unittest.FunctionTestCase(testSomething)
569
570If there are additional set-up and tear-down methods that should be called as
571part of the test case's operation, they can also be provided like so::
572
573 testcase = unittest.FunctionTestCase(testSomething,
574 setUp=makeSomethingDB,
575 tearDown=deleteSomethingDB)
576
577To make migrating existing test suites easier, :mod:`unittest` supports tests
578raising :exc:`AssertionError` to indicate test failure. However, it is
579recommended that you use the explicit :meth:`TestCase.fail\*` and
580:meth:`TestCase.assert\*` methods instead, as future versions of :mod:`unittest`
581may treat :exc:`AssertionError` differently.
582
583.. note::
584
Benjamin Petersond2397752009-06-27 23:45:02 +0000585 Even though :class:`FunctionTestCase` can be used to quickly convert an
586 existing test base over to a :mod:`unittest`\ -based system, this approach is
587 not recommended. Taking the time to set up proper :class:`TestCase`
588 subclasses will make future test refactorings infinitely easier.
Georg Brandl116aa622007-08-15 14:28:22 +0000589
Benjamin Peterson52baa292009-03-24 00:56:30 +0000590In some cases, the existing tests may have been written using the :mod:`doctest`
591module. If so, :mod:`doctest` provides a :class:`DocTestSuite` class that can
592automatically build :class:`unittest.TestSuite` instances from the existing
593:mod:`doctest`\ -based tests.
594
Georg Brandl116aa622007-08-15 14:28:22 +0000595
Benjamin Peterson5254c042009-03-23 22:25:03 +0000596.. _unittest-skipping:
597
598Skipping tests and expected failures
599------------------------------------
600
Michael Foordf5c851a2010-02-05 21:48:03 +0000601.. versionadded:: 3.1
602
Benjamin Peterson5254c042009-03-23 22:25:03 +0000603Unittest supports skipping individual test methods and even whole classes of
604tests. In addition, it supports marking a test as a "expected failure," a test
605that is broken and will fail, but shouldn't be counted as a failure on a
606:class:`TestResult`.
607
608Skipping a test is simply a matter of using the :func:`skip` :term:`decorator`
609or one of its conditional variants.
610
611Basic skipping looks like this: ::
612
613 class MyTestCase(unittest.TestCase):
614
615 @unittest.skip("demonstrating skipping")
616 def test_nothing(self):
617 self.fail("shouldn't happen")
618
Benjamin Petersond2397752009-06-27 23:45:02 +0000619 @unittest.skipIf(mylib.__version__ < (1, 3),
620 "not supported in this library version")
Benjamin Petersonded31c42009-03-30 15:04:16 +0000621 def test_format(self):
622 # Tests that work for only a certain version of the library.
623 pass
624
625 @unittest.skipUnless(sys.platform.startswith("win"), "requires Windows")
626 def test_windows_support(self):
627 # windows specific testing code
628 pass
629
Benjamin Peterson5254c042009-03-23 22:25:03 +0000630This is the output of running the example above in verbose mode: ::
631
Benjamin Petersonded31c42009-03-30 15:04:16 +0000632 test_format (__main__.MyTestCase) ... skipped 'not supported in this library version'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000633 test_nothing (__main__.MyTestCase) ... skipped 'demonstrating skipping'
Benjamin Petersonded31c42009-03-30 15:04:16 +0000634 test_windows_support (__main__.MyTestCase) ... skipped 'requires Windows'
Benjamin Peterson5254c042009-03-23 22:25:03 +0000635
636 ----------------------------------------------------------------------
Benjamin Petersonded31c42009-03-30 15:04:16 +0000637 Ran 3 tests in 0.005s
638
639 OK (skipped=3)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000640
641Classes can be skipped just like methods: ::
642
643 @skip("showing class skipping")
644 class MySkippedTestCase(unittest.TestCase):
645 def test_not_run(self):
646 pass
647
Benjamin Peterson52baa292009-03-24 00:56:30 +0000648:meth:`TestCase.setUp` can also skip the test. This is useful when a resource
649that needs to be set up is not available.
650
Benjamin Peterson5254c042009-03-23 22:25:03 +0000651Expected failures use the :func:`expectedFailure` decorator. ::
652
653 class ExpectedFailureTestCase(unittest.TestCase):
654 @unittest.expectedFailure
655 def test_fail(self):
656 self.assertEqual(1, 0, "broken")
657
658It's easy to roll your own skipping decorators by making a decorator that calls
659:func:`skip` on the test when it wants it to be skipped. This decorator skips
660the test unless the passed object has a certain attribute: ::
661
662 def skipUnlessHasattr(obj, attr):
663 if hasattr(obj, attr):
664 return lambda func: func
665 return unittest.skip("{0!r} doesn't have {1!r}".format(obj, attr))
666
667The following decorators implement test skipping and expected failures:
668
Georg Brandl8a1caa22010-07-29 16:01:11 +0000669.. decorator:: skip(reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000670
671 Unconditionally skip the decorated test. *reason* should describe why the
672 test is being skipped.
673
Georg Brandl8a1caa22010-07-29 16:01:11 +0000674.. decorator:: skipIf(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000675
676 Skip the decorated test if *condition* is true.
677
Georg Brandl8a1caa22010-07-29 16:01:11 +0000678.. decorator:: skipUnless(condition, reason)
Benjamin Peterson5254c042009-03-23 22:25:03 +0000679
Georg Brandl6faee4e2010-09-21 14:48:28 +0000680 Skip the decorated test unless *condition* is true.
Benjamin Peterson5254c042009-03-23 22:25:03 +0000681
Georg Brandl8a1caa22010-07-29 16:01:11 +0000682.. decorator:: expectedFailure
Benjamin Peterson5254c042009-03-23 22:25:03 +0000683
684 Mark the test as an expected failure. If the test fails when run, the test
685 is not counted as a failure.
686
Benjamin Petersonb48af542010-04-11 20:43:16 +0000687Skipped tests will not have :meth:`setUp` or :meth:`tearDown` run around them.
688Skipped classes will not have :meth:`setUpClass` or :meth:`tearDownClass` run.
689
Benjamin Peterson5254c042009-03-23 22:25:03 +0000690
Georg Brandl116aa622007-08-15 14:28:22 +0000691.. _unittest-contents:
692
693Classes and functions
694---------------------
695
Benjamin Peterson52baa292009-03-24 00:56:30 +0000696This section describes in depth the API of :mod:`unittest`.
697
698
699.. _testcase-objects:
700
701Test cases
702~~~~~~~~~~
Georg Brandl116aa622007-08-15 14:28:22 +0000703
Georg Brandl7f01a132009-09-16 15:58:14 +0000704.. class:: TestCase(methodName='runTest')
Georg Brandl116aa622007-08-15 14:28:22 +0000705
706 Instances of the :class:`TestCase` class represent the smallest testable units
707 in the :mod:`unittest` universe. This class is intended to be used as a base
708 class, with specific tests being implemented by concrete subclasses. This class
709 implements the interface needed by the test runner to allow it to drive the
710 test, and methods that the test code can use to check for and report various
711 kinds of failure.
712
713 Each instance of :class:`TestCase` will run a single test method: the method
714 named *methodName*. If you remember, we had an earlier example that went
715 something like this::
716
717 def suite():
718 suite = unittest.TestSuite()
Ezio Melottid59e44a2010-02-28 03:46:13 +0000719 suite.addTest(WidgetTestCase('test_default_size'))
720 suite.addTest(WidgetTestCase('test_resize'))
Georg Brandl116aa622007-08-15 14:28:22 +0000721 return suite
722
723 Here, we create two instances of :class:`WidgetTestCase`, each of which runs a
724 single test.
725
Michael Foord1341bb02011-03-14 19:01:46 -0400726 .. versionchanged:: 3.2
Michael Foord32e1d832011-01-03 17:00:11 +0000727 `TestCase` can be instantiated successfully without providing a method
728 name. This makes it easier to experiment with `TestCase` from the
729 interactive interpreter.
730
Benjamin Peterson52baa292009-03-24 00:56:30 +0000731 *methodName* defaults to :meth:`runTest`.
732
733 :class:`TestCase` instances provide three groups of methods: one group used
734 to run the test, another used by the test implementation to check conditions
735 and report failures, and some inquiry methods allowing information about the
736 test itself to be gathered.
737
738 Methods in the first group (running the test) are:
739
740
741 .. method:: setUp()
742
743 Method called to prepare the test fixture. This is called immediately
744 before calling the test method; any exception raised by this method will
745 be considered an error rather than a test failure. The default
746 implementation does nothing.
747
748
749 .. method:: tearDown()
750
751 Method called immediately after the test method has been called and the
752 result recorded. This is called even if the test method raised an
753 exception, so the implementation in subclasses may need to be particularly
754 careful about checking internal state. Any exception raised by this
755 method will be considered an error rather than a test failure. This
756 method will only be called if the :meth:`setUp` succeeds, regardless of
757 the outcome of the test method. The default implementation does nothing.
758
759
Benjamin Petersonb48af542010-04-11 20:43:16 +0000760 .. method:: setUpClass()
761
762 A class method called before tests in an individual class run.
763 ``setUpClass`` is called with the class as the only argument
764 and must be decorated as a :func:`classmethod`::
765
766 @classmethod
767 def setUpClass(cls):
768 ...
769
770 See `Class and Module Fixtures`_ for more details.
771
772 .. versionadded:: 3.2
773
774
775 .. method:: tearDownClass()
776
777 A class method called after tests in an individual class have run.
778 ``tearDownClass`` is called with the class as the only argument
779 and must be decorated as a :meth:`classmethod`::
780
781 @classmethod
782 def tearDownClass(cls):
783 ...
784
785 See `Class and Module Fixtures`_ for more details.
786
787 .. versionadded:: 3.2
788
789
Georg Brandl7f01a132009-09-16 15:58:14 +0000790 .. method:: run(result=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000791
792 Run the test, collecting the result into the test result object passed as
Ezio Melotti75b2a5e2010-11-20 10:13:45 +0000793 *result*. If *result* is omitted or ``None``, a temporary result
Alexandre Vassalotti260484d2009-07-17 11:43:26 +0000794 object is created (by calling the :meth:`defaultTestResult` method) and
Michael Foord1341bb02011-03-14 19:01:46 -0400795 used. The result object is returned to :meth:`run`'s caller.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000796
797 The same effect may be had by simply calling the :class:`TestCase`
798 instance.
799
Michael Foord1341bb02011-03-14 19:01:46 -0400800 .. versionchanged:: 3.3
801 Previous versions of ``run`` did not return the result. Neither did
802 calling an instance.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000803
Benjamin Petersone549ead2009-03-28 21:42:05 +0000804 .. method:: skipTest(reason)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000805
Stefan Kraha5bf3f52010-05-19 16:09:41 +0000806 Calling this during a test method or :meth:`setUp` skips the current
Benjamin Peterson52baa292009-03-24 00:56:30 +0000807 test. See :ref:`unittest-skipping` for more information.
808
Ezio Melotti7afd3f52010-04-20 09:32:54 +0000809 .. versionadded:: 3.1
Benjamin Peterson08bf91c2010-04-11 16:12:57 +0000810
Benjamin Peterson52baa292009-03-24 00:56:30 +0000811
812 .. method:: debug()
813
814 Run the test without collecting the result. This allows exceptions raised
815 by the test to be propagated to the caller, and can be used to support
816 running tests under a debugger.
817
Ezio Melotti22170ed2010-11-20 09:57:27 +0000818 .. _assert-methods:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000819
Ezio Melotti4370b302010-11-03 20:39:14 +0000820 The :class:`TestCase` class provides a number of methods to check for and
821 report failures, such as:
Benjamin Peterson52baa292009-03-24 00:56:30 +0000822
Ezio Melotti4370b302010-11-03 20:39:14 +0000823 +-----------------------------------------+-----------------------------+---------------+
824 | Method | Checks that | New in |
825 +=========================================+=============================+===============+
826 | :meth:`assertEqual(a, b) | ``a == b`` | |
827 | <TestCase.assertEqual>` | | |
828 +-----------------------------------------+-----------------------------+---------------+
829 | :meth:`assertNotEqual(a, b) | ``a != b`` | |
830 | <TestCase.assertNotEqual>` | | |
831 +-----------------------------------------+-----------------------------+---------------+
832 | :meth:`assertTrue(x) | ``bool(x) is True`` | |
833 | <TestCase.assertTrue>` | | |
834 +-----------------------------------------+-----------------------------+---------------+
835 | :meth:`assertFalse(x) | ``bool(x) is False`` | |
836 | <TestCase.assertFalse>` | | |
837 +-----------------------------------------+-----------------------------+---------------+
838 | :meth:`assertIs(a, b) | ``a is b`` | 3.1 |
839 | <TestCase.assertIs>` | | |
840 +-----------------------------------------+-----------------------------+---------------+
841 | :meth:`assertIsNot(a, b) | ``a is not b`` | 3.1 |
842 | <TestCase.assertIsNot>` | | |
843 +-----------------------------------------+-----------------------------+---------------+
844 | :meth:`assertIsNone(x) | ``x is None`` | 3.1 |
845 | <TestCase.assertIsNone>` | | |
846 +-----------------------------------------+-----------------------------+---------------+
847 | :meth:`assertIsNotNone(x) | ``x is not None`` | 3.1 |
848 | <TestCase.assertIsNotNone>` | | |
849 +-----------------------------------------+-----------------------------+---------------+
850 | :meth:`assertIn(a, b) | ``a in b`` | 3.1 |
851 | <TestCase.assertIn>` | | |
852 +-----------------------------------------+-----------------------------+---------------+
853 | :meth:`assertNotIn(a, b) | ``a not in b`` | 3.1 |
854 | <TestCase.assertNotIn>` | | |
855 +-----------------------------------------+-----------------------------+---------------+
856 | :meth:`assertIsInstance(a, b) | ``isinstance(a, b)`` | 3.2 |
857 | <TestCase.assertIsInstance>` | | |
858 +-----------------------------------------+-----------------------------+---------------+
859 | :meth:`assertNotIsInstance(a, b) | ``not isinstance(a, b)`` | 3.2 |
860 | <TestCase.assertNotIsInstance>` | | |
861 +-----------------------------------------+-----------------------------+---------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000862
Ezio Melotti22170ed2010-11-20 09:57:27 +0000863 All the assert methods (except :meth:`assertRaises`,
Ezio Melottied3a7d22010-12-01 02:32:32 +0000864 :meth:`assertRaisesRegex`, :meth:`assertWarns`, :meth:`assertWarnsRegex`)
Ezio Melotti22170ed2010-11-20 09:57:27 +0000865 accept a *msg* argument that, if specified, is used as the error message on
866 failure (see also :data:`longMessage`).
Benjamin Peterson52baa292009-03-24 00:56:30 +0000867
Michael Foorde180d392011-01-28 19:51:48 +0000868 .. method:: assertEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000869
Michael Foorde180d392011-01-28 19:51:48 +0000870 Test that *first* and *second* are equal. If the values do not
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000871 compare equal, the test will fail.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000872
Michael Foorde180d392011-01-28 19:51:48 +0000873 In addition, if *first* and *second* are the exact same type and one of
Michael Foord02834952010-02-08 23:10:39 +0000874 list, tuple, dict, set, frozenset or str or any type that a subclass
875 registers with :meth:`addTypeEqualityFunc` the type specific equality
876 function will be called in order to generate a more useful default
Ezio Melotti22170ed2010-11-20 09:57:27 +0000877 error message (see also the :ref:`list of type-specific methods
878 <type-specific-methods>`).
Ezio Melotti4841fd62010-11-05 15:43:40 +0000879
Raymond Hettinger35a88362009-04-09 00:08:24 +0000880 .. versionchanged:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000881 Added the automatic calling of type specific equality function.
882
Michael Foord28a817e2010-02-09 00:03:57 +0000883 .. versionchanged:: 3.2
884 :meth:`assertMultiLineEqual` added as the default type equality
885 function for comparing strings.
Michael Foord02834952010-02-08 23:10:39 +0000886
Benjamin Peterson52baa292009-03-24 00:56:30 +0000887
Michael Foorde180d392011-01-28 19:51:48 +0000888 .. method:: assertNotEqual(first, second, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000889
Michael Foorde180d392011-01-28 19:51:48 +0000890 Test that *first* and *second* are not equal. If the values do
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000891 compare equal, the test will fail.
Benjamin Peterson70e32c82009-03-24 01:00:11 +0000892
Ezio Melotti4370b302010-11-03 20:39:14 +0000893 .. method:: assertTrue(expr, msg=None)
Ezio Melotti4841fd62010-11-05 15:43:40 +0000894 assertFalse(expr, msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000895
Ezio Melotti4841fd62010-11-05 15:43:40 +0000896 Test that *expr* is true (or false).
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000897
Ezio Melotti4841fd62010-11-05 15:43:40 +0000898 Note that this is equivalent to ``bool(expr) is True`` and not to ``expr
899 is True`` (use ``assertIs(expr, True)`` for the latter). This method
900 should also be avoided when more specific methods are available (e.g.
901 ``assertEqual(a, b)`` instead of ``assertTrue(a == b)``), because they
902 provide a better error message in case of failure.
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +0000903
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000904
Michael Foorde180d392011-01-28 19:51:48 +0000905 .. method:: assertIs(first, second, msg=None)
906 assertIsNot(first, second, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000907
Michael Foorde180d392011-01-28 19:51:48 +0000908 Test that *first* and *second* evaluate (or don't evaluate) to the
Ezio Melottiaddc6f52010-12-18 20:00:04 +0000909 same object.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000910
Raymond Hettinger35a88362009-04-09 00:08:24 +0000911 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000912
913
Ezio Melotti4370b302010-11-03 20:39:14 +0000914 .. method:: assertIsNone(expr, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000915 assertIsNotNone(expr, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000916
Ezio Melotti9794a262010-11-04 14:52:13 +0000917 Test that *expr* is (or is not) None.
Benjamin Petersonb48af542010-04-11 20:43:16 +0000918
Ezio Melotti4370b302010-11-03 20:39:14 +0000919 .. versionadded:: 3.1
Benjamin Petersonb48af542010-04-11 20:43:16 +0000920
921
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000922 .. method:: assertIn(first, second, msg=None)
923 assertNotIn(first, second, msg=None)
924
Ezio Melotti4841fd62010-11-05 15:43:40 +0000925 Test that *first* is (or is not) in *second*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000926
Raymond Hettinger35a88362009-04-09 00:08:24 +0000927 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000928
929
Ezio Melotti9c02c2f2010-11-03 20:45:31 +0000930 .. method:: assertIsInstance(obj, cls, msg=None)
Ezio Melotti9794a262010-11-04 14:52:13 +0000931 assertNotIsInstance(obj, cls, msg=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000932
Ezio Melotti9794a262010-11-04 14:52:13 +0000933 Test that *obj* is (or is not) an instance of *cls* (which can be a
934 class or a tuple of classes, as supported by :func:`isinstance`).
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000935
Ezio Melotti4370b302010-11-03 20:39:14 +0000936 .. versionadded:: 3.2
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000937
938
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000939
Ezio Melotti4370b302010-11-03 20:39:14 +0000940 It is also possible to check that exceptions and warnings are raised using
941 the following methods:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000942
Ezio Melotti4370b302010-11-03 20:39:14 +0000943 +---------------------------------------------------------+--------------------------------------+------------+
944 | Method | Checks that | New in |
945 +=========================================================+======================================+============+
946 | :meth:`assertRaises(exc, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `exc` | |
947 | <TestCase.assertRaises>` | | |
948 +---------------------------------------------------------+--------------------------------------+------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +0000949 | :meth:`assertRaisesRegex(exc, re, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `exc` | 3.1 |
950 | <TestCase.assertRaisesRegex>` | and the message matches `re` | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000951 +---------------------------------------------------------+--------------------------------------+------------+
952 | :meth:`assertWarns(warn, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `warn` | 3.2 |
953 | <TestCase.assertWarns>` | | |
954 +---------------------------------------------------------+--------------------------------------+------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +0000955 | :meth:`assertWarnsRegex(warn, re, fun, *args, **kwds) | ``fun(*args, **kwds)`` raises `warn` | 3.2 |
956 | <TestCase.assertWarnsRegex>` | and the message matches `re` | |
Ezio Melotti4370b302010-11-03 20:39:14 +0000957 +---------------------------------------------------------+--------------------------------------+------------+
Benjamin Peterson52baa292009-03-24 00:56:30 +0000958
Georg Brandl7f01a132009-09-16 15:58:14 +0000959 .. method:: assertRaises(exception, callable, *args, **kwds)
Georg Brandl7f01a132009-09-16 15:58:14 +0000960 assertRaises(exception)
Benjamin Peterson52baa292009-03-24 00:56:30 +0000961
962 Test that an exception is raised when *callable* is called with any
963 positional or keyword arguments that are also passed to
964 :meth:`assertRaises`. The test passes if *exception* is raised, is an
965 error if another exception is raised, or fails if no exception is raised.
966 To catch any of a group of exceptions, a tuple containing the exception
967 classes may be passed as *exception*.
968
Georg Brandl7f01a132009-09-16 15:58:14 +0000969 If only the *exception* argument is given, returns a context manager so
970 that the code under test can be written inline rather than as a function::
Benjamin Petersonded31c42009-03-30 15:04:16 +0000971
Michael Foord41531f22010-02-05 21:13:40 +0000972 with self.assertRaises(SomeException):
Benjamin Petersonded31c42009-03-30 15:04:16 +0000973 do_something()
974
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000975 The context manager will store the caught exception object in its
Ezio Melotti49008232010-02-08 21:57:48 +0000976 :attr:`exception` attribute. This can be useful if the intention
Michael Foord41531f22010-02-05 21:13:40 +0000977 is to perform additional checks on the exception raised::
Kristján Valur Jónsson92a653a2009-11-13 16:10:13 +0000978
Georg Brandl8a1caa22010-07-29 16:01:11 +0000979 with self.assertRaises(SomeException) as cm:
980 do_something()
Michael Foord41531f22010-02-05 21:13:40 +0000981
Georg Brandl8a1caa22010-07-29 16:01:11 +0000982 the_exception = cm.exception
983 self.assertEqual(the_exception.error_code, 3)
Michael Foord41531f22010-02-05 21:13:40 +0000984
Ezio Melotti49008232010-02-08 21:57:48 +0000985 .. versionchanged:: 3.1
Benjamin Petersonded31c42009-03-30 15:04:16 +0000986 Added the ability to use :meth:`assertRaises` as a context manager.
Benjamin Peterson52baa292009-03-24 00:56:30 +0000987
Ezio Melotti49008232010-02-08 21:57:48 +0000988 .. versionchanged:: 3.2
989 Added the :attr:`exception` attribute.
990
Benjamin Peterson52baa292009-03-24 00:56:30 +0000991
Ezio Melottied3a7d22010-12-01 02:32:32 +0000992 .. method:: assertRaisesRegex(exception, regex, callable, *args, **kwds)
993 assertRaisesRegex(exception, regex)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000994
Ezio Melottied3a7d22010-12-01 02:32:32 +0000995 Like :meth:`assertRaises` but also tests that *regex* matches
996 on the string representation of the raised exception. *regex* may be
Benjamin Peterson7fe73a12009-04-04 16:35:46 +0000997 a regular expression object or a string containing a regular expression
998 suitable for use by :func:`re.search`. Examples::
999
Ezio Melottied3a7d22010-12-01 02:32:32 +00001000 self.assertRaisesRegex(ValueError, 'invalid literal for.*XYZ$',
1001 int, 'XYZ')
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001002
1003 or::
1004
Ezio Melottied3a7d22010-12-01 02:32:32 +00001005 with self.assertRaisesRegex(ValueError, 'literal'):
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001006 int('XYZ')
1007
Georg Brandl419e3de2010-12-01 15:44:25 +00001008 .. versionadded:: 3.1
1009 under the name ``assertRaisesRegexp``.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001010 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001011 Renamed to :meth:`assertRaisesRegex`.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001012
1013
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001014 .. method:: assertWarns(warning, callable, *args, **kwds)
1015 assertWarns(warning)
1016
1017 Test that a warning is triggered when *callable* is called with any
1018 positional or keyword arguments that are also passed to
1019 :meth:`assertWarns`. The test passes if *warning* is triggered and
1020 fails if it isn't. Also, any unexpected exception is an error.
1021 To catch any of a group of warnings, a tuple containing the warning
1022 classes may be passed as *warnings*.
1023
1024 If only the *warning* argument is given, returns a context manager so
1025 that the code under test can be written inline rather than as a function::
1026
1027 with self.assertWarns(SomeWarning):
1028 do_something()
1029
1030 The context manager will store the caught warning object in its
1031 :attr:`warning` attribute, and the source line which triggered the
1032 warnings in the :attr:`filename` and :attr:`lineno` attributes.
1033 This can be useful if the intention is to perform additional checks
1034 on the exception raised::
1035
1036 with self.assertWarns(SomeWarning) as cm:
1037 do_something()
1038
1039 self.assertIn('myfile.py', cm.filename)
1040 self.assertEqual(320, cm.lineno)
1041
1042 This method works regardless of the warning filters in place when it
1043 is called.
1044
1045 .. versionadded:: 3.2
1046
1047
Ezio Melottied3a7d22010-12-01 02:32:32 +00001048 .. method:: assertWarnsRegex(warning, regex, callable, *args, **kwds)
1049 assertWarnsRegex(warning, regex)
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001050
Ezio Melottied3a7d22010-12-01 02:32:32 +00001051 Like :meth:`assertWarns` but also tests that *regex* matches on the
1052 message of the triggered warning. *regex* may be a regular expression
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001053 object or a string containing a regular expression suitable for use
1054 by :func:`re.search`. Example::
1055
Ezio Melottied3a7d22010-12-01 02:32:32 +00001056 self.assertWarnsRegex(DeprecationWarning,
1057 r'legacy_function\(\) is deprecated',
1058 legacy_function, 'XYZ')
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001059
1060 or::
1061
Ezio Melottied3a7d22010-12-01 02:32:32 +00001062 with self.assertWarnsRegex(RuntimeWarning, 'unsafe frobnicating'):
Antoine Pitrou4bc12ef2010-09-06 19:25:46 +00001063 frobnicate('/etc/passwd')
1064
1065 .. versionadded:: 3.2
1066
1067
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001068
Ezio Melotti4370b302010-11-03 20:39:14 +00001069 There are also other methods used to perform more specific checks, such as:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001070
Ezio Melotti4370b302010-11-03 20:39:14 +00001071 +---------------------------------------+--------------------------------+--------------+
1072 | Method | Checks that | New in |
1073 +=======================================+================================+==============+
1074 | :meth:`assertAlmostEqual(a, b) | ``round(a-b, 7) == 0`` | |
1075 | <TestCase.assertAlmostEqual>` | | |
1076 +---------------------------------------+--------------------------------+--------------+
1077 | :meth:`assertNotAlmostEqual(a, b) | ``round(a-b, 7) != 0`` | |
1078 | <TestCase.assertNotAlmostEqual>` | | |
1079 +---------------------------------------+--------------------------------+--------------+
1080 | :meth:`assertGreater(a, b) | ``a > b`` | 3.1 |
1081 | <TestCase.assertGreater>` | | |
1082 +---------------------------------------+--------------------------------+--------------+
1083 | :meth:`assertGreaterEqual(a, b) | ``a >= b`` | 3.1 |
1084 | <TestCase.assertGreaterEqual>` | | |
1085 +---------------------------------------+--------------------------------+--------------+
1086 | :meth:`assertLess(a, b) | ``a < b`` | 3.1 |
1087 | <TestCase.assertLess>` | | |
1088 +---------------------------------------+--------------------------------+--------------+
1089 | :meth:`assertLessEqual(a, b) | ``a <= b`` | 3.1 |
1090 | <TestCase.assertLessEqual>` | | |
1091 +---------------------------------------+--------------------------------+--------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +00001092 | :meth:`assertRegex(s, re) | ``regex.search(s)`` | 3.1 |
1093 | <TestCase.assertRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001094 +---------------------------------------+--------------------------------+--------------+
Ezio Melottied3a7d22010-12-01 02:32:32 +00001095 | :meth:`assertNotRegex(s, re) | ``not regex.search(s)`` | 3.2 |
1096 | <TestCase.assertNotRegex>` | | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001097 +---------------------------------------+--------------------------------+--------------+
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001098 | :meth:`assertCountEqual(a, b) | `a` and `b` have the same | 3.2 |
1099 | <TestCase.assertCountEqual>` | elements in the same number, | |
Ezio Melotti4370b302010-11-03 20:39:14 +00001100 | | regardless of their order | |
1101 +---------------------------------------+--------------------------------+--------------+
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001102
1103
Michael Foorde180d392011-01-28 19:51:48 +00001104 .. method:: assertAlmostEqual(first, second, places=7, msg=None, delta=None)
1105 assertNotAlmostEqual(first, second, places=7, msg=None, delta=None)
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001106
Michael Foorde180d392011-01-28 19:51:48 +00001107 Test that *first* and *second* are approximately (or not approximately)
Ezio Melotti4841fd62010-11-05 15:43:40 +00001108 equal by computing the difference, rounding to the given number of
1109 decimal *places* (default 7), and comparing to zero. Note that these
1110 methods round the values to the given number of *decimal places* (i.e.
1111 like the :func:`round` function) and not *significant digits*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001112
Ezio Melotti4370b302010-11-03 20:39:14 +00001113 If *delta* is supplied instead of *places* then the difference
Michael Foorde180d392011-01-28 19:51:48 +00001114 between *first* and *second* must be less (or more) than *delta*.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001115
Ezio Melotti4370b302010-11-03 20:39:14 +00001116 Supplying both *delta* and *places* raises a ``TypeError``.
Benjamin Petersonf47ed4a2009-04-11 20:45:40 +00001117
Ezio Melotti4370b302010-11-03 20:39:14 +00001118 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001119 :meth:`assertAlmostEqual` automatically considers almost equal objects
1120 that compare equal. :meth:`assertNotAlmostEqual` automatically fails
1121 if the objects compare equal. Added the *delta* keyword argument.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001122
Ezio Melotti4370b302010-11-03 20:39:14 +00001123
Michael Foorde180d392011-01-28 19:51:48 +00001124 .. method:: assertGreater(first, second, msg=None)
1125 assertGreaterEqual(first, second, msg=None)
1126 assertLess(first, second, msg=None)
1127 assertLessEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001128
Michael Foorde180d392011-01-28 19:51:48 +00001129 Test that *first* is respectively >, >=, < or <= than *second* depending
Ezio Melotti4841fd62010-11-05 15:43:40 +00001130 on the method name. If not, the test will fail::
Ezio Melotti4370b302010-11-03 20:39:14 +00001131
1132 >>> self.assertGreaterEqual(3, 4)
1133 AssertionError: "3" unexpectedly not greater than or equal to "4"
1134
1135 .. versionadded:: 3.1
1136
1137
Ezio Melottied3a7d22010-12-01 02:32:32 +00001138 .. method:: assertRegex(text, regex, msg=None)
1139 assertNotRegex(text, regex, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001140
Ezio Melottied3a7d22010-12-01 02:32:32 +00001141 Test that a *regex* search matches (or does not match) *text*. In case
Ezio Melotti4841fd62010-11-05 15:43:40 +00001142 of failure, the error message will include the pattern and the *text* (or
Ezio Melottied3a7d22010-12-01 02:32:32 +00001143 the pattern and the part of *text* that unexpectedly matched). *regex*
Ezio Melotti4370b302010-11-03 20:39:14 +00001144 may be a regular expression object or a string containing a regular
1145 expression suitable for use by :func:`re.search`.
1146
Georg Brandl419e3de2010-12-01 15:44:25 +00001147 .. versionadded:: 3.1
1148 under the name ``assertRegexpMatches``.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001149 .. versionchanged:: 3.2
Georg Brandl419e3de2010-12-01 15:44:25 +00001150 The method ``assertRegexpMatches()`` has been renamed to
1151 :meth:`.assertRegex`.
1152 .. versionadded:: 3.2
1153 :meth:`.assertNotRegex`.
Ezio Melotti4370b302010-11-03 20:39:14 +00001154
1155
Ezio Melottiaddc6f52010-12-18 20:00:04 +00001156 .. method:: assertDictContainsSubset(subset, dictionary, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001157
Ezio Melottiaddc6f52010-12-18 20:00:04 +00001158 Tests whether the key/value pairs in *dictionary* are a superset of
1159 those in *subset*. If not, an error message listing the missing keys
1160 and mismatched values is generated.
Ezio Melotti4370b302010-11-03 20:39:14 +00001161
Raymond Hettinger8ebe27f2010-12-21 19:24:26 +00001162 Note, the arguments are in the opposite order of what the method name
1163 dictates. Instead, consider using the set-methods on :ref:`dictionary
1164 views <dict-views>`, for example: ``d.keys() <= e.keys()`` or
1165 ``d.items() <= d.items()``.
1166
Ezio Melotti4370b302010-11-03 20:39:14 +00001167 .. versionadded:: 3.1
Raymond Hettinger8ebe27f2010-12-21 19:24:26 +00001168 .. deprecated:: 3.2
Ezio Melotti4370b302010-11-03 20:39:14 +00001169
1170
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001171 .. method:: assertCountEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001172
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001173 Test that sequence *first* contains the same elements as *second*,
Ezio Melotti4370b302010-11-03 20:39:14 +00001174 regardless of their order. When they don't, an error message listing the
1175 differences between the sequences will be generated.
1176
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001177 Duplicate elements are *not* ignored when comparing *first* and
1178 *second*. It verifies whether each element has the same count in both
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001179 sequences. Equivalent to:
Raymond Hettinger57bd00a2010-12-24 21:51:48 +00001180 ``assertEqual(Counter(list(first)), Counter(list(second)))``
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001181 but works with sequences of unhashable objects as well.
Ezio Melotti4370b302010-11-03 20:39:14 +00001182
Ezio Melotti4370b302010-11-03 20:39:14 +00001183 .. versionadded:: 3.2
1184
Michael Foorde180d392011-01-28 19:51:48 +00001185 .. method:: assertSameElements(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001186
Michael Foorde180d392011-01-28 19:51:48 +00001187 Test that sequence *first* contains the same elements as *second*,
Ezio Melotti4370b302010-11-03 20:39:14 +00001188 regardless of their order. When they don't, an error message listing
1189 the differences between the sequences will be generated.
1190
Michael Foorde180d392011-01-28 19:51:48 +00001191 Duplicate elements are ignored when comparing *first* and *second*.
1192 It is the equivalent of ``assertEqual(set(first), set(second))``
Ezio Melotti4370b302010-11-03 20:39:14 +00001193 but it works with sequences of unhashable objects as well. Because
1194 duplicates are ignored, this method has been deprecated in favour of
Raymond Hettinger6e165b32010-11-27 09:31:37 +00001195 :meth:`assertCountEqual`.
Ezio Melotti4370b302010-11-03 20:39:14 +00001196
Ezio Melotti4370b302010-11-03 20:39:14 +00001197 .. versionadded:: 3.1
1198 .. deprecated:: 3.2
1199
1200
Ezio Melotti22170ed2010-11-20 09:57:27 +00001201 .. _type-specific-methods:
Ezio Melotti4370b302010-11-03 20:39:14 +00001202
Ezio Melotti22170ed2010-11-20 09:57:27 +00001203 The :meth:`assertEqual` method dispatches the equality check for objects of
1204 the same type to different type-specific methods. These methods are already
1205 implemented for most of the built-in types, but it's also possible to
1206 register new methods using :meth:`addTypeEqualityFunc`:
1207
1208 .. method:: addTypeEqualityFunc(typeobj, function)
1209
1210 Registers a type-specific method called by :meth:`assertEqual` to check
1211 if two objects of exactly the same *typeobj* (not subclasses) compare
1212 equal. *function* must take two positional arguments and a third msg=None
1213 keyword argument just as :meth:`assertEqual` does. It must raise
1214 :data:`self.failureException(msg) <failureException>` when inequality
1215 between the first two parameters is detected -- possibly providing useful
1216 information and explaining the inequalities in details in the error
1217 message.
1218
1219 .. versionadded:: 3.1
1220
1221 The list of type-specific methods automatically used by
1222 :meth:`~TestCase.assertEqual` are summarized in the following table. Note
1223 that it's usually not necessary to invoke these methods directly.
Ezio Melotti4370b302010-11-03 20:39:14 +00001224
1225 +-----------------------------------------+-----------------------------+--------------+
1226 | Method | Used to compare | New in |
1227 +=========================================+=============================+==============+
1228 | :meth:`assertMultiLineEqual(a, b) | strings | 3.1 |
1229 | <TestCase.assertMultiLineEqual>` | | |
1230 +-----------------------------------------+-----------------------------+--------------+
1231 | :meth:`assertSequenceEqual(a, b) | sequences | 3.1 |
1232 | <TestCase.assertSequenceEqual>` | | |
1233 +-----------------------------------------+-----------------------------+--------------+
1234 | :meth:`assertListEqual(a, b) | lists | 3.1 |
1235 | <TestCase.assertListEqual>` | | |
1236 +-----------------------------------------+-----------------------------+--------------+
1237 | :meth:`assertTupleEqual(a, b) | tuples | 3.1 |
1238 | <TestCase.assertTupleEqual>` | | |
1239 +-----------------------------------------+-----------------------------+--------------+
1240 | :meth:`assertSetEqual(a, b) | sets or frozensets | 3.1 |
1241 | <TestCase.assertSetEqual>` | | |
1242 +-----------------------------------------+-----------------------------+--------------+
1243 | :meth:`assertDictEqual(a, b) | dicts | 3.1 |
1244 | <TestCase.assertDictEqual>` | | |
1245 +-----------------------------------------+-----------------------------+--------------+
1246
1247
1248
Michael Foorde180d392011-01-28 19:51:48 +00001249 .. method:: assertMultiLineEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001250
Michael Foorde180d392011-01-28 19:51:48 +00001251 Test that the multiline string *first* is equal to the string *second*.
Ezio Melotti4370b302010-11-03 20:39:14 +00001252 When not equal a diff of the two strings highlighting the differences
1253 will be included in the error message. This method is used by default
1254 when comparing strings with :meth:`assertEqual`.
1255
Ezio Melotti4370b302010-11-03 20:39:14 +00001256 .. versionadded:: 3.1
1257
1258
Michael Foorde180d392011-01-28 19:51:48 +00001259 .. method:: assertSequenceEqual(first, second, msg=None, seq_type=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001260
1261 Tests that two sequences are equal. If a *seq_type* is supplied, both
Michael Foorde180d392011-01-28 19:51:48 +00001262 *first* and *second* must be instances of *seq_type* or a failure will
Ezio Melotti4370b302010-11-03 20:39:14 +00001263 be raised. If the sequences are different an error message is
1264 constructed that shows the difference between the two.
1265
Ezio Melotti22170ed2010-11-20 09:57:27 +00001266 This method is not called directly by :meth:`assertEqual`, but
1267 it's used to implement :meth:`assertListEqual` and
Ezio Melotti4370b302010-11-03 20:39:14 +00001268 :meth:`assertTupleEqual`.
1269
1270 .. versionadded:: 3.1
1271
1272
Michael Foorde180d392011-01-28 19:51:48 +00001273 .. method:: assertListEqual(first, second, msg=None)
1274 assertTupleEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001275
1276 Tests that two lists or tuples are equal. If not an error message is
1277 constructed that shows only the differences between the two. An error
1278 is also raised if either of the parameters are of the wrong type.
1279 These methods are used by default when comparing lists or tuples with
1280 :meth:`assertEqual`.
1281
Ezio Melotti4370b302010-11-03 20:39:14 +00001282 .. versionadded:: 3.1
1283
1284
Michael Foorde180d392011-01-28 19:51:48 +00001285 .. method:: assertSetEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001286
1287 Tests that two sets are equal. If not, an error message is constructed
1288 that lists the differences between the sets. This method is used by
1289 default when comparing sets or frozensets with :meth:`assertEqual`.
1290
Michael Foorde180d392011-01-28 19:51:48 +00001291 Fails if either of *first* or *second* does not have a :meth:`set.difference`
Ezio Melotti4370b302010-11-03 20:39:14 +00001292 method.
1293
Ezio Melotti4370b302010-11-03 20:39:14 +00001294 .. versionadded:: 3.1
1295
1296
Michael Foorde180d392011-01-28 19:51:48 +00001297 .. method:: assertDictEqual(first, second, msg=None)
Ezio Melotti4370b302010-11-03 20:39:14 +00001298
1299 Test that two dictionaries are equal. If not, an error message is
1300 constructed that shows the differences in the dictionaries. This
1301 method will be used by default to compare dictionaries in
1302 calls to :meth:`assertEqual`.
1303
Ezio Melotti4370b302010-11-03 20:39:14 +00001304 .. versionadded:: 3.1
1305
1306
1307
Ezio Melotti22170ed2010-11-20 09:57:27 +00001308 .. _other-methods-and-attrs:
1309
Ezio Melotti4370b302010-11-03 20:39:14 +00001310 Finally the :class:`TestCase` provides the following methods and attributes:
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001311
Benjamin Peterson52baa292009-03-24 00:56:30 +00001312
Georg Brandl7f01a132009-09-16 15:58:14 +00001313 .. method:: fail(msg=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001314
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001315 Signals a test failure unconditionally, with *msg* or ``None`` for
Benjamin Peterson52baa292009-03-24 00:56:30 +00001316 the error message.
1317
1318
1319 .. attribute:: failureException
1320
1321 This class attribute gives the exception raised by the test method. If a
1322 test framework needs to use a specialized exception, possibly to carry
1323 additional information, it must subclass this exception in order to "play
1324 fair" with the framework. The initial value of this attribute is
1325 :exc:`AssertionError`.
1326
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001327
1328 .. attribute:: longMessage
1329
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001330 If set to ``True`` then any explicit failure message you pass in to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001331 :ref:`assert methods <assert-methods>` will be appended to the end of the
1332 normal failure message. The normal messages contain useful information
1333 about the objects involved, for example the message from assertEqual
1334 shows you the repr of the two unequal objects. Setting this attribute
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001335 to ``True`` allows you to have a custom error message in addition to the
Ezio Melotti22170ed2010-11-20 09:57:27 +00001336 normal one.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001337
Michael Foord5074df62010-12-03 00:53:09 +00001338 This attribute defaults to ``True``. If set to False then a custom message
1339 passed to an assert method will silence the normal message.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001340
1341 The class setting can be overridden in individual tests by assigning an
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001342 instance attribute to ``True`` or ``False`` before calling the assert methods.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001343
Raymond Hettinger35a88362009-04-09 00:08:24 +00001344 .. versionadded:: 3.1
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001345
1346
Michael Foord98b3e762010-06-05 21:59:55 +00001347 .. attribute:: maxDiff
1348
1349 This attribute controls the maximum length of diffs output by assert
1350 methods that report diffs on failure. It defaults to 80*8 characters.
1351 Assert methods affected by this attribute are
1352 :meth:`assertSequenceEqual` (including all the sequence comparison
1353 methods that delegate to it), :meth:`assertDictEqual` and
1354 :meth:`assertMultiLineEqual`.
1355
1356 Setting ``maxDiff`` to None means that there is no maximum length of
1357 diffs.
1358
1359 .. versionadded:: 3.2
1360
1361
Benjamin Peterson52baa292009-03-24 00:56:30 +00001362 Testing frameworks can use the following methods to collect information on
1363 the test:
1364
1365
1366 .. method:: countTestCases()
1367
1368 Return the number of tests represented by this test object. For
1369 :class:`TestCase` instances, this will always be ``1``.
1370
1371
1372 .. method:: defaultTestResult()
1373
1374 Return an instance of the test result class that should be used for this
1375 test case class (if no other result instance is provided to the
1376 :meth:`run` method).
1377
1378 For :class:`TestCase` instances, this will always be an instance of
1379 :class:`TestResult`; subclasses of :class:`TestCase` should override this
1380 as necessary.
1381
1382
1383 .. method:: id()
1384
1385 Return a string identifying the specific test case. This is usually the
1386 full name of the test method, including the module and class name.
1387
1388
1389 .. method:: shortDescription()
1390
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001391 Returns a description of the test, or ``None`` if no description
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001392 has been provided. The default implementation of this method
1393 returns the first line of the test method's docstring, if available,
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001394 or ``None``.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001395
Georg Brandl419e3de2010-12-01 15:44:25 +00001396 .. versionchanged:: 3.1
Michael Foord34c94622010-02-10 15:51:42 +00001397 In 3.1 this was changed to add the test name to the short description
Georg Brandl419e3de2010-12-01 15:44:25 +00001398 even in the presence of a docstring. This caused compatibility issues
Michael Foord34c94622010-02-10 15:51:42 +00001399 with unittest extensions and adding the test name was moved to the
Georg Brandl419e3de2010-12-01 15:44:25 +00001400 :class:`TextTestResult` in Python 3.2.
Benjamin Peterson7fe73a12009-04-04 16:35:46 +00001401
Georg Brandl116aa622007-08-15 14:28:22 +00001402
Georg Brandl7f01a132009-09-16 15:58:14 +00001403 .. method:: addCleanup(function, *args, **kwargs)
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001404
1405 Add a function to be called after :meth:`tearDown` to cleanup resources
1406 used during the test. Functions will be called in reverse order to the
1407 order they are added (LIFO). They are called with any arguments and
1408 keyword arguments passed into :meth:`addCleanup` when they are
1409 added.
1410
1411 If :meth:`setUp` fails, meaning that :meth:`tearDown` is not called,
1412 then any cleanup functions added will still be called.
1413
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001414 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001415
1416
1417 .. method:: doCleanups()
1418
Barry Warsaw0c9fd632010-04-12 14:50:57 +00001419 This method is called unconditionally after :meth:`tearDown`, or
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001420 after :meth:`setUp` if :meth:`setUp` raises an exception.
1421
1422 It is responsible for calling all the cleanup functions added by
1423 :meth:`addCleanup`. If you need cleanup functions to be called
1424 *prior* to :meth:`tearDown` then you can call :meth:`doCleanups`
1425 yourself.
1426
1427 :meth:`doCleanups` pops methods off the stack of cleanup
1428 functions one at a time, so it can be called at any time.
1429
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001430 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001431
1432
Georg Brandl7f01a132009-09-16 15:58:14 +00001433.. class:: FunctionTestCase(testFunc, setUp=None, tearDown=None, description=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001434
1435 This class implements the portion of the :class:`TestCase` interface which
Benjamin Petersond2397752009-06-27 23:45:02 +00001436 allows the test runner to drive the test, but does not provide the methods
1437 which test code can use to check and report errors. This is used to create
1438 test cases using legacy test code, allowing it to be integrated into a
1439 :mod:`unittest`-based test framework.
Georg Brandl116aa622007-08-15 14:28:22 +00001440
1441
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001442.. _deprecated-aliases:
1443
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001444Deprecated aliases
1445##################
1446
1447For historical reasons, some of the :class:`TestCase` methods had one or more
1448aliases that are now deprecated. The following table lists the correct names
1449along with their deprecated aliases:
1450
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001451 ============================== ====================== ======================
1452 Method Name Deprecated alias Deprecated alias
1453 ============================== ====================== ======================
1454 :meth:`.assertEqual` failUnlessEqual assertEquals
1455 :meth:`.assertNotEqual` failIfEqual assertNotEquals
1456 :meth:`.assertTrue` failUnless assert\_
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001457 :meth:`.assertFalse` failIf
1458 :meth:`.assertRaises` failUnlessRaises
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001459 :meth:`.assertAlmostEqual` failUnlessAlmostEqual assertAlmostEquals
1460 :meth:`.assertNotAlmostEqual` failIfAlmostEqual assertNotAlmostEquals
Ezio Melottied3a7d22010-12-01 02:32:32 +00001461 :meth:`.assertRegex` assertRegexpMatches
1462 :meth:`.assertRaisesRegex` assertRaisesRegexp
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001463 ============================== ====================== ======================
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001464
Ezio Melotti2baf1a62010-11-22 12:56:58 +00001465 .. deprecated-removed:: 3.1 3.3
1466 the fail* aliases listed in the second column.
1467 .. deprecated:: 3.2
1468 the assert* aliases listed in the third column.
Ezio Melottied3a7d22010-12-01 02:32:32 +00001469 .. deprecated:: 3.2
1470 ``assertRegexpMatches`` and ``assertRaisesRegexp`` have been renamed to
1471 :meth:`.assertRegex` and :meth:`.assertRaisesRegex`
Ezio Melotti8f2e07b2010-11-04 19:09:28 +00001472
1473
Benjamin Peterson52baa292009-03-24 00:56:30 +00001474.. _testsuite-objects:
1475
Benjamin Peterson52baa292009-03-24 00:56:30 +00001476Grouping tests
1477~~~~~~~~~~~~~~
1478
Georg Brandl7f01a132009-09-16 15:58:14 +00001479.. class:: TestSuite(tests=())
Georg Brandl116aa622007-08-15 14:28:22 +00001480
1481 This class represents an aggregation of individual tests cases and test suites.
1482 The class presents the interface needed by the test runner to allow it to be run
1483 as any other test case. Running a :class:`TestSuite` instance is the same as
1484 iterating over the suite, running each test individually.
1485
1486 If *tests* is given, it must be an iterable of individual test cases or other
1487 test suites that will be used to build the suite initially. Additional methods
1488 are provided to add test cases and suites to the collection later on.
1489
Benjamin Peterson14a3dd72009-05-25 00:51:58 +00001490 :class:`TestSuite` objects behave much like :class:`TestCase` objects, except
1491 they do not actually implement a test. Instead, they are used to aggregate
1492 tests into groups of tests that should be run together. Some additional
1493 methods are available to add tests to :class:`TestSuite` instances:
Benjamin Peterson52baa292009-03-24 00:56:30 +00001494
1495
1496 .. method:: TestSuite.addTest(test)
1497
1498 Add a :class:`TestCase` or :class:`TestSuite` to the suite.
1499
1500
1501 .. method:: TestSuite.addTests(tests)
1502
1503 Add all the tests from an iterable of :class:`TestCase` and :class:`TestSuite`
1504 instances to this test suite.
1505
Benjamin Petersond2397752009-06-27 23:45:02 +00001506 This is equivalent to iterating over *tests*, calling :meth:`addTest` for
1507 each element.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001508
1509 :class:`TestSuite` shares the following methods with :class:`TestCase`:
1510
1511
1512 .. method:: run(result)
1513
1514 Run the tests associated with this suite, collecting the result into the
1515 test result object passed as *result*. Note that unlike
1516 :meth:`TestCase.run`, :meth:`TestSuite.run` requires the result object to
1517 be passed in.
1518
1519
1520 .. method:: debug()
1521
1522 Run the tests associated with this suite without collecting the
1523 result. This allows exceptions raised by the test to be propagated to the
1524 caller and can be used to support running tests under a debugger.
1525
1526
1527 .. method:: countTestCases()
1528
1529 Return the number of tests represented by this test object, including all
1530 individual tests and sub-suites.
1531
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001532
1533 .. method:: __iter__()
1534
1535 Tests grouped by a :class:`TestSuite` are always accessed by iteration.
1536 Subclasses can lazily provide tests by overriding :meth:`__iter__`. Note
1537 that this method maybe called several times on a single suite
1538 (for example when counting tests or comparing for equality)
1539 so the tests returned must be the same for repeated iterations.
1540
Georg Brandl853947a2010-01-31 18:53:23 +00001541 .. versionchanged:: 3.2
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001542 In earlier versions the :class:`TestSuite` accessed tests directly rather
1543 than through iteration, so overriding :meth:`__iter__` wasn't sufficient
1544 for providing tests.
1545
Benjamin Peterson52baa292009-03-24 00:56:30 +00001546 In the typical usage of a :class:`TestSuite` object, the :meth:`run` method
1547 is invoked by a :class:`TestRunner` rather than by the end-user test harness.
1548
1549
Benjamin Peterson52baa292009-03-24 00:56:30 +00001550Loading and running tests
1551~~~~~~~~~~~~~~~~~~~~~~~~~
1552
Georg Brandl116aa622007-08-15 14:28:22 +00001553.. class:: TestLoader()
1554
Benjamin Peterson52baa292009-03-24 00:56:30 +00001555 The :class:`TestLoader` class is used to create test suites from classes and
1556 modules. Normally, there is no need to create an instance of this class; the
1557 :mod:`unittest` module provides an instance that can be shared as
1558 ``unittest.defaultTestLoader``. Using a subclass or instance, however, allows
1559 customization of some configurable properties.
1560
1561 :class:`TestLoader` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001562
Ezio Melotti9c02c2f2010-11-03 20:45:31 +00001563
Benjamin Peterson52baa292009-03-24 00:56:30 +00001564 .. method:: loadTestsFromTestCase(testCaseClass)
Georg Brandl116aa622007-08-15 14:28:22 +00001565
Benjamin Peterson52baa292009-03-24 00:56:30 +00001566 Return a suite of all tests cases contained in the :class:`TestCase`\ -derived
1567 :class:`testCaseClass`.
1568
1569
1570 .. method:: loadTestsFromModule(module)
1571
1572 Return a suite of all tests cases contained in the given module. This
1573 method searches *module* for classes derived from :class:`TestCase` and
1574 creates an instance of the class for each test method defined for the
1575 class.
1576
Georg Brandle720c0a2009-04-27 16:20:50 +00001577 .. note::
Benjamin Peterson52baa292009-03-24 00:56:30 +00001578
1579 While using a hierarchy of :class:`TestCase`\ -derived classes can be
1580 convenient in sharing fixtures and helper functions, defining test
1581 methods on base classes that are not intended to be instantiated
1582 directly does not play well with this method. Doing so, however, can
1583 be useful when the fixtures are different and defined in subclasses.
1584
Benjamin Petersond2397752009-06-27 23:45:02 +00001585 If a module provides a ``load_tests`` function it will be called to
1586 load the tests. This allows modules to customize test loading.
1587 This is the `load_tests protocol`_.
1588
Georg Brandl853947a2010-01-31 18:53:23 +00001589 .. versionchanged:: 3.2
Benjamin Petersond2397752009-06-27 23:45:02 +00001590 Support for ``load_tests`` added.
1591
Benjamin Peterson52baa292009-03-24 00:56:30 +00001592
Georg Brandl7f01a132009-09-16 15:58:14 +00001593 .. method:: loadTestsFromName(name, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001594
1595 Return a suite of all tests cases given a string specifier.
1596
1597 The specifier *name* is a "dotted name" that may resolve either to a
1598 module, a test case class, a test method within a test case class, a
1599 :class:`TestSuite` instance, or a callable object which returns a
1600 :class:`TestCase` or :class:`TestSuite` instance. These checks are
1601 applied in the order listed here; that is, a method on a possible test
1602 case class will be picked up as "a test method within a test case class",
1603 rather than "a callable object".
1604
1605 For example, if you have a module :mod:`SampleTests` containing a
1606 :class:`TestCase`\ -derived class :class:`SampleTestCase` with three test
1607 methods (:meth:`test_one`, :meth:`test_two`, and :meth:`test_three`), the
Benjamin Petersond2397752009-06-27 23:45:02 +00001608 specifier ``'SampleTests.SampleTestCase'`` would cause this method to
1609 return a suite which will run all three test methods. Using the specifier
1610 ``'SampleTests.SampleTestCase.test_two'`` would cause it to return a test
1611 suite which will run only the :meth:`test_two` test method. The specifier
1612 can refer to modules and packages which have not been imported; they will
1613 be imported as a side-effect.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001614
1615 The method optionally resolves *name* relative to the given *module*.
1616
1617
Georg Brandl7f01a132009-09-16 15:58:14 +00001618 .. method:: loadTestsFromNames(names, module=None)
Benjamin Peterson52baa292009-03-24 00:56:30 +00001619
1620 Similar to :meth:`loadTestsFromName`, but takes a sequence of names rather
1621 than a single name. The return value is a test suite which supports all
1622 the tests defined for each name.
1623
1624
1625 .. method:: getTestCaseNames(testCaseClass)
1626
1627 Return a sorted sequence of method names found within *testCaseClass*;
1628 this should be a subclass of :class:`TestCase`.
1629
Benjamin Petersond2397752009-06-27 23:45:02 +00001630
1631 .. method:: discover(start_dir, pattern='test*.py', top_level_dir=None)
1632
1633 Find and return all test modules from the specified start directory,
1634 recursing into subdirectories to find them. Only test files that match
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001635 *pattern* will be loaded. (Using shell style pattern matching.) Only
1636 module names that are importable (i.e. are valid Python identifiers) will
1637 be loaded.
Benjamin Petersond2397752009-06-27 23:45:02 +00001638
1639 All test modules must be importable from the top level of the project. If
1640 the start directory is not the top level directory then the top level
1641 directory must be specified separately.
1642
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001643 If importing a module fails, for example due to a syntax error, then this
1644 will be recorded as a single error and discovery will continue.
1645
Benjamin Petersond2397752009-06-27 23:45:02 +00001646 If a test package name (directory with :file:`__init__.py`) matches the
1647 pattern then the package will be checked for a ``load_tests``
1648 function. If this exists then it will be called with *loader*, *tests*,
1649 *pattern*.
1650
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001651 If load_tests exists then discovery does *not* recurse into the package,
Benjamin Petersond2397752009-06-27 23:45:02 +00001652 ``load_tests`` is responsible for loading all tests in the package.
1653
1654 The pattern is deliberately not stored as a loader attribute so that
1655 packages can continue discovery themselves. *top_level_dir* is stored so
1656 ``load_tests`` does not need to pass this argument in to
1657 ``loader.discover()``.
1658
Benjamin Petersonb48af542010-04-11 20:43:16 +00001659 *start_dir* can be a dotted module name as well as a directory.
1660
Georg Brandl853947a2010-01-31 18:53:23 +00001661 .. versionadded:: 3.2
1662
Benjamin Petersond2397752009-06-27 23:45:02 +00001663
Benjamin Peterson52baa292009-03-24 00:56:30 +00001664 The following attributes of a :class:`TestLoader` can be configured either by
1665 subclassing or assignment on an instance:
1666
1667
1668 .. attribute:: testMethodPrefix
1669
1670 String giving the prefix of method names which will be interpreted as test
1671 methods. The default value is ``'test'``.
1672
1673 This affects :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*`
1674 methods.
1675
1676
1677 .. attribute:: sortTestMethodsUsing
1678
1679 Function to be used to compare method names when sorting them in
1680 :meth:`getTestCaseNames` and all the :meth:`loadTestsFrom\*` methods.
1681
1682
1683 .. attribute:: suiteClass
1684
1685 Callable object that constructs a test suite from a list of tests. No
1686 methods on the resulting object are needed. The default value is the
1687 :class:`TestSuite` class.
1688
1689 This affects all the :meth:`loadTestsFrom\*` methods.
1690
1691
Benjamin Peterson52baa292009-03-24 00:56:30 +00001692.. class:: TestResult
1693
1694 This class is used to compile information about which tests have succeeded
1695 and which have failed.
1696
1697 A :class:`TestResult` object stores the results of a set of tests. The
1698 :class:`TestCase` and :class:`TestSuite` classes ensure that results are
1699 properly recorded; test authors do not need to worry about recording the
1700 outcome of tests.
1701
1702 Testing frameworks built on top of :mod:`unittest` may want access to the
1703 :class:`TestResult` object generated by running a set of tests for reporting
1704 purposes; a :class:`TestResult` instance is returned by the
1705 :meth:`TestRunner.run` method for this purpose.
1706
1707 :class:`TestResult` instances have the following attributes that will be of
1708 interest when inspecting the results of running a set of tests:
1709
1710
1711 .. attribute:: errors
1712
1713 A list containing 2-tuples of :class:`TestCase` instances and strings
1714 holding formatted tracebacks. Each tuple represents a test which raised an
1715 unexpected exception.
1716
Benjamin Peterson52baa292009-03-24 00:56:30 +00001717 .. attribute:: failures
1718
1719 A list containing 2-tuples of :class:`TestCase` instances and strings
1720 holding formatted tracebacks. Each tuple represents a test where a failure
1721 was explicitly signalled using the :meth:`TestCase.fail\*` or
1722 :meth:`TestCase.assert\*` methods.
1723
Benjamin Peterson52baa292009-03-24 00:56:30 +00001724 .. attribute:: skipped
1725
1726 A list containing 2-tuples of :class:`TestCase` instances and strings
1727 holding the reason for skipping the test.
1728
Benjamin Peterson70e32c82009-03-24 01:00:11 +00001729 .. versionadded:: 3.1
Benjamin Peterson52baa292009-03-24 00:56:30 +00001730
1731 .. attribute:: expectedFailures
1732
Georg Brandl6faee4e2010-09-21 14:48:28 +00001733 A list containing 2-tuples of :class:`TestCase` instances and strings
1734 holding formatted tracebacks. Each tuple represents an expected failure
Benjamin Peterson52baa292009-03-24 00:56:30 +00001735 of the test case.
1736
1737 .. attribute:: unexpectedSuccesses
1738
1739 A list containing :class:`TestCase` instances that were marked as expected
1740 failures, but succeeded.
1741
1742 .. attribute:: shouldStop
1743
1744 Set to ``True`` when the execution of tests should stop by :meth:`stop`.
1745
1746
1747 .. attribute:: testsRun
1748
1749 The total number of tests run so far.
1750
1751
Georg Brandl12037202010-12-02 22:35:25 +00001752 .. attribute:: buffer
Benjamin Petersonb48af542010-04-11 20:43:16 +00001753
1754 If set to true, ``sys.stdout`` and ``sys.stderr`` will be buffered in between
1755 :meth:`startTest` and :meth:`stopTest` being called. Collected output will
1756 only be echoed onto the real ``sys.stdout`` and ``sys.stderr`` if the test
1757 fails or errors. Any output is also attached to the failure / error message.
1758
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001759 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001760
1761
1762 .. attribute:: failfast
1763
1764 If set to true :meth:`stop` will be called on the first failure or error,
1765 halting the test run.
1766
Ezio Melotti7afd3f52010-04-20 09:32:54 +00001767 .. versionadded:: 3.2
Benjamin Petersonb48af542010-04-11 20:43:16 +00001768
1769
Benjamin Peterson52baa292009-03-24 00:56:30 +00001770 .. method:: wasSuccessful()
1771
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001772 Return ``True`` if all tests run so far have passed, otherwise returns
1773 ``False``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001774
1775
1776 .. method:: stop()
1777
1778 This method can be called to signal that the set of tests being run should
Ezio Melotti75b2a5e2010-11-20 10:13:45 +00001779 be aborted by setting the :attr:`shouldStop` attribute to ``True``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001780 :class:`TestRunner` objects should respect this flag and return without
1781 running any additional tests.
1782
1783 For example, this feature is used by the :class:`TextTestRunner` class to
1784 stop the test framework when the user signals an interrupt from the
1785 keyboard. Interactive tools which provide :class:`TestRunner`
1786 implementations can use this in a similar manner.
1787
1788 The following methods of the :class:`TestResult` class are used to maintain
1789 the internal data structures, and may be extended in subclasses to support
1790 additional reporting requirements. This is particularly useful in building
1791 tools which support interactive reporting while tests are being run.
1792
1793
1794 .. method:: startTest(test)
1795
1796 Called when the test case *test* is about to be run.
1797
Benjamin Peterson52baa292009-03-24 00:56:30 +00001798 .. method:: stopTest(test)
1799
1800 Called after the test case *test* has been executed, regardless of the
1801 outcome.
1802
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001803 .. method:: startTestRun(test)
1804
1805 Called once before any tests are executed.
1806
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001807 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001808
1809
1810 .. method:: stopTestRun(test)
1811
Ezio Melotti176d6c42010-01-27 20:58:07 +00001812 Called once after all tests are executed.
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001813
Ezio Melotti2d1e88a2011-03-10 12:16:35 +02001814 .. versionadded:: 3.1
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001815
1816
Benjamin Peterson52baa292009-03-24 00:56:30 +00001817 .. method:: addError(test, err)
1818
1819 Called when the test case *test* raises an unexpected exception *err* is a
1820 tuple of the form returned by :func:`sys.exc_info`: ``(type, value,
1821 traceback)``.
1822
1823 The default implementation appends a tuple ``(test, formatted_err)`` to
1824 the instance's :attr:`errors` attribute, where *formatted_err* is a
1825 formatted traceback derived from *err*.
1826
1827
1828 .. method:: addFailure(test, err)
1829
Benjamin Petersond2397752009-06-27 23:45:02 +00001830 Called when the test case *test* signals a failure. *err* is a tuple of
1831 the form returned by :func:`sys.exc_info`: ``(type, value, traceback)``.
Benjamin Peterson52baa292009-03-24 00:56:30 +00001832
1833 The default implementation appends a tuple ``(test, formatted_err)`` to
1834 the instance's :attr:`failures` attribute, where *formatted_err* is a
1835 formatted traceback derived from *err*.
1836
1837
1838 .. method:: addSuccess(test)
1839
1840 Called when the test case *test* succeeds.
1841
1842 The default implementation does nothing.
1843
1844
1845 .. method:: addSkip(test, reason)
1846
1847 Called when the test case *test* is skipped. *reason* is the reason the
1848 test gave for skipping.
1849
1850 The default implementation appends a tuple ``(test, reason)`` to the
1851 instance's :attr:`skipped` attribute.
1852
1853
1854 .. method:: addExpectedFailure(test, err)
1855
1856 Called when the test case *test* fails, but was marked with the
1857 :func:`expectedFailure` decorator.
1858
1859 The default implementation appends a tuple ``(test, formatted_err)`` to
1860 the instance's :attr:`expectedFailures` attribute, where *formatted_err*
1861 is a formatted traceback derived from *err*.
1862
1863
1864 .. method:: addUnexpectedSuccess(test)
1865
1866 Called when the test case *test* was marked with the
1867 :func:`expectedFailure` decorator, but succeeded.
1868
1869 The default implementation appends the test to the instance's
1870 :attr:`unexpectedSuccesses` attribute.
Georg Brandl116aa622007-08-15 14:28:22 +00001871
Georg Brandl67b21b72010-08-17 15:07:14 +00001872
Michael Foord34c94622010-02-10 15:51:42 +00001873.. class:: TextTestResult(stream, descriptions, verbosity)
1874
Georg Brandl67b21b72010-08-17 15:07:14 +00001875 A concrete implementation of :class:`TestResult` used by the
1876 :class:`TextTestRunner`.
Michael Foord34c94622010-02-10 15:51:42 +00001877
Georg Brandl67b21b72010-08-17 15:07:14 +00001878 .. versionadded:: 3.2
1879 This class was previously named ``_TextTestResult``. The old name still
1880 exists as an alias but is deprecated.
1881
Georg Brandl116aa622007-08-15 14:28:22 +00001882
1883.. data:: defaultTestLoader
1884
1885 Instance of the :class:`TestLoader` class intended to be shared. If no
1886 customization of the :class:`TestLoader` is needed, this instance can be used
1887 instead of repeatedly creating new instances.
1888
1889
Michael Foordd218e952011-01-03 12:55:11 +00001890.. class:: TextTestRunner(stream=None, descriptions=True, verbosity=1, runnerclass=None, warnings=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001891
Michael Foordd218e952011-01-03 12:55:11 +00001892 A basic test runner implementation that outputs results to a stream. If *stream*
1893 is `None`, the default, `sys.stderr` is used as the output stream. This class
Georg Brandl116aa622007-08-15 14:28:22 +00001894 has a few configurable parameters, but is essentially very simple. Graphical
1895 applications which run test suites should provide alternate implementations.
1896
Ezio Melotti60901872010-12-01 00:56:10 +00001897 By default this runner shows :exc:`DeprecationWarning`,
1898 :exc:`PendingDeprecationWarning`, and :exc:`ImportWarning` even if they are
1899 :ref:`ignored by default <warning-ignored>`. Deprecation warnings caused by
1900 :ref:`deprecated unittest methods <deprecated-aliases>` are also
1901 special-cased and, when the warning filters are ``'default'`` or ``'always'``,
1902 they will appear only once per-module, in order to avoid too many warning
Georg Brandl46402372010-12-04 19:06:18 +00001903 messages. This behavior can be overridden using the :option:`-Wd` or
Ezio Melotti60901872010-12-01 00:56:10 +00001904 :option:`-Wa` options and leaving *warnings* to ``None``.
1905
Michael Foordd218e952011-01-03 12:55:11 +00001906 .. versionchanged:: 3.2
1907 Added the ``warnings`` argument.
1908
1909 .. versionchanged:: 3.2
1910 The default stream is set to `sys.stderr` at instantiation time rather
1911 than import time.
1912
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001913 .. method:: _makeResult()
Georg Brandl116aa622007-08-15 14:28:22 +00001914
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001915 This method returns the instance of ``TestResult`` used by :meth:`run`.
1916 It is not intended to be called directly, but can be overridden in
1917 subclasses to provide a custom ``TestResult``.
1918
Michael Foord34c94622010-02-10 15:51:42 +00001919 ``_makeResult()`` instantiates the class or callable passed in the
1920 ``TextTestRunner`` constructor as the ``resultclass`` argument. It
Benjamin Petersonb48af542010-04-11 20:43:16 +00001921 defaults to :class:`TextTestResult` if no ``resultclass`` is provided.
Michael Foord34c94622010-02-10 15:51:42 +00001922 The result class is instantiated with the following arguments::
1923
1924 stream, descriptions, verbosity
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001925
Ezio Melotti60901872010-12-01 00:56:10 +00001926
Georg Brandl419e3de2010-12-01 15:44:25 +00001927.. function:: main(module='__main__', defaultTest=None, argv=None, testRunner=None, \
Ezio Melotti40dcb1d2011-03-10 13:46:50 +02001928 testLoader=unittest.defaultTestLoader, exit=True, verbosity=1, \
Georg Brandl419e3de2010-12-01 15:44:25 +00001929 failfast=None, catchbreak=None, buffer=None, warnings=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001930
1931 A command-line program that runs a set of tests; this is primarily for making
1932 test modules conveniently executable. The simplest use for this function is to
1933 include the following line at the end of a test script::
1934
1935 if __name__ == '__main__':
1936 unittest.main()
1937
Benjamin Petersond2397752009-06-27 23:45:02 +00001938 You can run tests with more detailed information by passing in the verbosity
1939 argument::
1940
1941 if __name__ == '__main__':
1942 unittest.main(verbosity=2)
1943
Georg Brandl116aa622007-08-15 14:28:22 +00001944 The *testRunner* argument can either be a test runner class or an already
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001945 created instance of it. By default ``main`` calls :func:`sys.exit` with
1946 an exit code indicating success or failure of the tests run.
1947
1948 ``main`` supports being used from the interactive interpreter by passing in the
1949 argument ``exit=False``. This displays the result on standard output without
1950 calling :func:`sys.exit`::
1951
1952 >>> from unittest import main
1953 >>> main(module='test_module', exit=False)
1954
Benjamin Petersonb48af542010-04-11 20:43:16 +00001955 The ``failfast``, ``catchbreak`` and ``buffer`` parameters have the same
Éric Araujo76338ec2010-11-26 23:46:18 +00001956 effect as the same-name `command-line options`_.
Benjamin Petersonb48af542010-04-11 20:43:16 +00001957
Ezio Melotti60901872010-12-01 00:56:10 +00001958 The *warning* argument specifies the :ref:`warning filter <warning-filter>`
1959 that should be used while running the tests. If it's not specified, it will
1960 remain ``None`` if a :option:`-W` option is passed to :program:`python`,
1961 otherwise it will be set to ``'default'``.
1962
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001963 Calling ``main`` actually returns an instance of the ``TestProgram`` class.
1964 This stores the result of the tests run as the ``result`` attribute.
1965
Éric Araujo971dc012010-12-16 03:13:05 +00001966 .. versionchanged:: 3.1
1967 The ``exit`` parameter was added.
1968
Georg Brandl853947a2010-01-31 18:53:23 +00001969 .. versionchanged:: 3.2
Éric Araujo971dc012010-12-16 03:13:05 +00001970 The ``verbosity``, ``failfast``, ``catchbreak``, ``buffer``
Ezio Melotti60901872010-12-01 00:56:10 +00001971 and ``warnings`` parameters were added.
Benjamin Petersond2397752009-06-27 23:45:02 +00001972
1973
1974load_tests Protocol
1975###################
1976
Georg Brandl853947a2010-01-31 18:53:23 +00001977.. versionadded:: 3.2
Benjamin Peterson4ac9ce42009-10-04 14:49:41 +00001978
Benjamin Petersond2397752009-06-27 23:45:02 +00001979Modules or packages can customize how tests are loaded from them during normal
1980test runs or test discovery by implementing a function called ``load_tests``.
1981
1982If a test module defines ``load_tests`` it will be called by
1983:meth:`TestLoader.loadTestsFromModule` with the following arguments::
1984
1985 load_tests(loader, standard_tests, None)
1986
1987It should return a :class:`TestSuite`.
1988
1989*loader* is the instance of :class:`TestLoader` doing the loading.
1990*standard_tests* are the tests that would be loaded by default from the
1991module. It is common for test modules to only want to add or remove tests
1992from the standard set of tests.
1993The third argument is used when loading packages as part of test discovery.
1994
1995A typical ``load_tests`` function that loads tests from a specific set of
1996:class:`TestCase` classes may look like::
1997
1998 test_cases = (TestCase1, TestCase2, TestCase3)
1999
2000 def load_tests(loader, tests, pattern):
2001 suite = TestSuite()
2002 for test_class in test_cases:
2003 tests = loader.loadTestsFromTestCase(test_class)
2004 suite.addTests(tests)
2005 return suite
2006
2007If discovery is started, either from the command line or by calling
2008:meth:`TestLoader.discover`, with a pattern that matches a package
2009name then the package :file:`__init__.py` will be checked for ``load_tests``.
2010
2011.. note::
2012
Ezio Melotti0639d5a2009-12-19 23:26:38 +00002013 The default pattern is 'test*.py'. This matches all Python files
Benjamin Petersond2397752009-06-27 23:45:02 +00002014 that start with 'test' but *won't* match any test directories.
2015
2016 A pattern like 'test*' will match test packages as well as
2017 modules.
2018
2019If the package :file:`__init__.py` defines ``load_tests`` then it will be
2020called and discovery not continued into the package. ``load_tests``
2021is called with the following arguments::
2022
2023 load_tests(loader, standard_tests, pattern)
2024
2025This should return a :class:`TestSuite` representing all the tests
2026from the package. (``standard_tests`` will only contain tests
2027collected from :file:`__init__.py`.)
2028
2029Because the pattern is passed into ``load_tests`` the package is free to
2030continue (and potentially modify) test discovery. A 'do nothing'
2031``load_tests`` function for a test package would look like::
2032
2033 def load_tests(loader, standard_tests, pattern):
2034 # top level directory cached on loader instance
2035 this_dir = os.path.dirname(__file__)
2036 package_tests = loader.discover(start_dir=this_dir, pattern=pattern)
2037 standard_tests.addTests(package_tests)
2038 return standard_tests
Benjamin Petersonb48af542010-04-11 20:43:16 +00002039
2040
2041Class and Module Fixtures
2042-------------------------
2043
2044Class and module level fixtures are implemented in :class:`TestSuite`. When
2045the test suite encounters a test from a new class then :meth:`tearDownClass`
2046from the previous class (if there is one) is called, followed by
2047:meth:`setUpClass` from the new class.
2048
2049Similarly if a test is from a different module from the previous test then
2050``tearDownModule`` from the previous module is run, followed by
2051``setUpModule`` from the new module.
2052
2053After all the tests have run the final ``tearDownClass`` and
2054``tearDownModule`` are run.
2055
2056Note that shared fixtures do not play well with [potential] features like test
2057parallelization and they break test isolation. They should be used with care.
2058
2059The default ordering of tests created by the unittest test loaders is to group
2060all tests from the same modules and classes together. This will lead to
2061``setUpClass`` / ``setUpModule`` (etc) being called exactly once per class and
2062module. If you randomize the order, so that tests from different modules and
2063classes are adjacent to each other, then these shared fixture functions may be
2064called multiple times in a single test run.
2065
2066Shared fixtures are not intended to work with suites with non-standard
2067ordering. A ``BaseTestSuite`` still exists for frameworks that don't want to
2068support shared fixtures.
2069
2070If there are any exceptions raised during one of the shared fixture functions
2071the test is reported as an error. Because there is no corresponding test
2072instance an ``_ErrorHolder`` object (that has the same interface as a
2073:class:`TestCase`) is created to represent the error. If you are just using
2074the standard unittest test runner then this detail doesn't matter, but if you
2075are a framework author it may be relevant.
2076
2077
2078setUpClass and tearDownClass
2079~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2080
2081These must be implemented as class methods::
2082
2083 import unittest
2084
2085 class Test(unittest.TestCase):
2086 @classmethod
2087 def setUpClass(cls):
2088 cls._connection = createExpensiveConnectionObject()
2089
2090 @classmethod
2091 def tearDownClass(cls):
2092 cls._connection.destroy()
2093
2094If you want the ``setUpClass`` and ``tearDownClass`` on base classes called
2095then you must call up to them yourself. The implementations in
2096:class:`TestCase` are empty.
2097
2098If an exception is raised during a ``setUpClass`` then the tests in the class
2099are not run and the ``tearDownClass`` is not run. Skipped classes will not
Michael Foord98b3e762010-06-05 21:59:55 +00002100have ``setUpClass`` or ``tearDownClass`` run. If the exception is a
2101``SkipTest`` exception then the class will be reported as having been skipped
2102instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002103
2104
2105setUpModule and tearDownModule
2106~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
2107
2108These should be implemented as functions::
2109
2110 def setUpModule():
2111 createConnection()
2112
2113 def tearDownModule():
2114 closeConnection()
2115
2116If an exception is raised in a ``setUpModule`` then none of the tests in the
Michael Foord98b3e762010-06-05 21:59:55 +00002117module will be run and the ``tearDownModule`` will not be run. If the exception is a
2118``SkipTest`` exception then the module will be reported as having been skipped
2119instead of as an error.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002120
2121
2122Signal Handling
2123---------------
2124
Georg Brandl419e3de2010-12-01 15:44:25 +00002125.. versionadded:: 3.2
2126
Éric Araujo8acb67c2010-11-26 23:31:07 +00002127The :option:`-c/--catch <unittest -c>` command-line option to unittest,
Éric Araujo76338ec2010-11-26 23:46:18 +00002128along with the ``catchbreak`` parameter to :func:`unittest.main()`, provide
2129more friendly handling of control-C during a test run. With catch break
2130behavior enabled control-C will allow the currently running test to complete,
2131and the test run will then end and report all the results so far. A second
2132control-c will raise a :exc:`KeyboardInterrupt` in the usual way.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002133
Michael Foordde4ceab2010-04-25 19:53:49 +00002134The control-c handling signal handler attempts to remain compatible with code or
2135tests that install their own :const:`signal.SIGINT` handler. If the ``unittest``
2136handler is called but *isn't* the installed :const:`signal.SIGINT` handler,
2137i.e. it has been replaced by the system under test and delegated to, then it
2138calls the default handler. This will normally be the expected behavior by code
2139that replaces an installed handler and delegates to it. For individual tests
2140that need ``unittest`` control-c handling disabled the :func:`removeHandler`
2141decorator can be used.
2142
2143There are a few utility functions for framework authors to enable control-c
2144handling functionality within test frameworks.
Benjamin Petersonb48af542010-04-11 20:43:16 +00002145
2146.. function:: installHandler()
2147
2148 Install the control-c handler. When a :const:`signal.SIGINT` is received
2149 (usually in response to the user pressing control-c) all registered results
2150 have :meth:`~TestResult.stop` called.
2151
Michael Foord469b1f02010-04-26 23:41:26 +00002152
Benjamin Petersonb48af542010-04-11 20:43:16 +00002153.. function:: registerResult(result)
2154
2155 Register a :class:`TestResult` object for control-c handling. Registering a
2156 result stores a weak reference to it, so it doesn't prevent the result from
2157 being garbage collected.
2158
Michael Foordde4ceab2010-04-25 19:53:49 +00002159 Registering a :class:`TestResult` object has no side-effects if control-c
2160 handling is not enabled, so test frameworks can unconditionally register
2161 all results they create independently of whether or not handling is enabled.
2162
Michael Foord469b1f02010-04-26 23:41:26 +00002163
Benjamin Petersonb48af542010-04-11 20:43:16 +00002164.. function:: removeResult(result)
2165
2166 Remove a registered result. Once a result has been removed then
2167 :meth:`~TestResult.stop` will no longer be called on that result object in
2168 response to a control-c.
2169
Michael Foord469b1f02010-04-26 23:41:26 +00002170
Michael Foordde4ceab2010-04-25 19:53:49 +00002171.. function:: removeHandler(function=None)
2172
2173 When called without arguments this function removes the control-c handler
2174 if it has been installed. This function can also be used as a test decorator
2175 to temporarily remove the handler whilst the test is being executed::
2176
2177 @unittest.removeHandler
2178 def test_signal_handling(self):
2179 ...