blob: 6138e300746d276bee52b06b531e96d803a97e5c [file] [log] [blame]
Georg Brandl83e51f42012-10-10 16:45:11 +02001:keepdoctest:
2
Georg Brandl116aa622007-08-15 14:28:22 +00003:mod:`doctest` --- Test interactive Python examples
4===================================================
5
6.. module:: doctest
7 :synopsis: Test pieces of code within docstrings.
8.. moduleauthor:: Tim Peters <tim@python.org>
9.. sectionauthor:: Tim Peters <tim@python.org>
10.. sectionauthor:: Moshe Zadka <moshez@debian.org>
11.. sectionauthor:: Edward Loper <edloper@users.sourceforge.net>
12
13
14The :mod:`doctest` module searches for pieces of text that look like interactive
15Python sessions, and then executes those sessions to verify that they work
16exactly as shown. There are several common ways to use doctest:
17
18* To check that a module's docstrings are up-to-date by verifying that all
19 interactive examples still work as documented.
20
21* To perform regression testing by verifying that interactive examples from a
22 test file or a test object work as expected.
23
24* To write tutorial documentation for a package, liberally illustrated with
25 input-output examples. Depending on whether the examples or the expository text
26 are emphasized, this has the flavor of "literate testing" or "executable
27 documentation".
28
29Here's a complete but small example module::
30
31 """
32 This is the "example" module.
33
34 The example module supplies one function, factorial(). For example,
35
36 >>> factorial(5)
37 120
38 """
39
40 def factorial(n):
41 """Return the factorial of n, an exact integer >= 0.
42
Georg Brandl116aa622007-08-15 14:28:22 +000043 >>> [factorial(n) for n in range(6)]
44 [1, 1, 2, 6, 24, 120]
Georg Brandl116aa622007-08-15 14:28:22 +000045 >>> factorial(30)
Georg Brandl5c106642007-11-29 17:41:05 +000046 265252859812191058636308480000000
Georg Brandl116aa622007-08-15 14:28:22 +000047 >>> factorial(-1)
48 Traceback (most recent call last):
49 ...
50 ValueError: n must be >= 0
51
52 Factorials of floats are OK, but the float must be an exact integer:
53 >>> factorial(30.1)
54 Traceback (most recent call last):
55 ...
56 ValueError: n must be exact integer
57 >>> factorial(30.0)
Georg Brandl5c106642007-11-29 17:41:05 +000058 265252859812191058636308480000000
Georg Brandl116aa622007-08-15 14:28:22 +000059
60 It must also not be ridiculously large:
61 >>> factorial(1e100)
62 Traceback (most recent call last):
63 ...
64 OverflowError: n too large
65 """
66
Georg Brandl116aa622007-08-15 14:28:22 +000067 import math
68 if not n >= 0:
69 raise ValueError("n must be >= 0")
70 if math.floor(n) != n:
71 raise ValueError("n must be exact integer")
72 if n+1 == n: # catch a value like 1e300
73 raise OverflowError("n too large")
74 result = 1
75 factor = 2
76 while factor <= n:
77 result *= factor
78 factor += 1
79 return result
80
Georg Brandl116aa622007-08-15 14:28:22 +000081
82 if __name__ == "__main__":
Guido van Rossum04110fb2007-08-24 16:32:05 +000083 import doctest
84 doctest.testmod()
Georg Brandl116aa622007-08-15 14:28:22 +000085
86If you run :file:`example.py` directly from the command line, :mod:`doctest`
87works its magic::
88
89 $ python example.py
90 $
91
92There's no output! That's normal, and it means all the examples worked. Pass
Éric Araujo713d3032010-11-18 16:38:46 +000093``-v`` to the script, and :mod:`doctest` prints a detailed log of what
Georg Brandl116aa622007-08-15 14:28:22 +000094it's trying, and prints a summary at the end::
95
96 $ python example.py -v
97 Trying:
98 factorial(5)
99 Expecting:
100 120
101 ok
102 Trying:
103 [factorial(n) for n in range(6)]
104 Expecting:
105 [1, 1, 2, 6, 24, 120]
106 ok
Georg Brandl116aa622007-08-15 14:28:22 +0000107
108And so on, eventually ending with::
109
110 Trying:
111 factorial(1e100)
112 Expecting:
113 Traceback (most recent call last):
114 ...
115 OverflowError: n too large
116 ok
Georg Brandl116aa622007-08-15 14:28:22 +0000117 2 items passed all tests:
118 1 tests in __main__
119 8 tests in __main__.factorial
Guido van Rossum04110fb2007-08-24 16:32:05 +0000120 9 tests in 2 items.
Georg Brandl116aa622007-08-15 14:28:22 +0000121 9 passed and 0 failed.
122 Test passed.
123 $
124
125That's all you need to know to start making productive use of :mod:`doctest`!
126Jump in. The following sections provide full details. Note that there are many
127examples of doctests in the standard Python test suite and libraries.
128Especially useful examples can be found in the standard test file
129:file:`Lib/test/test_doctest.py`.
130
131
132.. _doctest-simple-testmod:
133
134Simple Usage: Checking Examples in Docstrings
135---------------------------------------------
136
137The simplest way to start using doctest (but not necessarily the way you'll
138continue to do it) is to end each module :mod:`M` with::
139
Guido van Rossum04110fb2007-08-24 16:32:05 +0000140 if __name__ == "__main__":
Georg Brandl116aa622007-08-15 14:28:22 +0000141 import doctest
142 doctest.testmod()
143
Georg Brandl116aa622007-08-15 14:28:22 +0000144:mod:`doctest` then examines docstrings in module :mod:`M`.
145
146Running the module as a script causes the examples in the docstrings to get
147executed and verified::
148
149 python M.py
150
151This won't display anything unless an example fails, in which case the failing
152example(s) and the cause(s) of the failure(s) are printed to stdout, and the
153final line of output is ``***Test Failed*** N failures.``, where *N* is the
154number of examples that failed.
155
Éric Araujo713d3032010-11-18 16:38:46 +0000156Run it with the ``-v`` switch instead::
Georg Brandl116aa622007-08-15 14:28:22 +0000157
158 python M.py -v
159
160and a detailed report of all examples tried is printed to standard output, along
161with assorted summaries at the end.
162
163You can force verbose mode by passing ``verbose=True`` to :func:`testmod`, or
164prohibit it by passing ``verbose=False``. In either of those cases,
Éric Araujo713d3032010-11-18 16:38:46 +0000165``sys.argv`` is not examined by :func:`testmod` (so passing ``-v`` or not
Georg Brandl116aa622007-08-15 14:28:22 +0000166has no effect).
167
Georg Brandl31835852008-05-12 17:38:56 +0000168There is also a command line shortcut for running :func:`testmod`. You can
169instruct the Python interpreter to run the doctest module directly from the
170standard library and pass the module name(s) on the command line::
Georg Brandl116aa622007-08-15 14:28:22 +0000171
172 python -m doctest -v example.py
173
174This will import :file:`example.py` as a standalone module and run
175:func:`testmod` on it. Note that this may not work correctly if the file is
176part of a package and imports other submodules from that package.
177
178For more information on :func:`testmod`, see section :ref:`doctest-basic-api`.
179
180
181.. _doctest-simple-testfile:
182
183Simple Usage: Checking Examples in a Text File
184----------------------------------------------
185
186Another simple application of doctest is testing interactive examples in a text
187file. This can be done with the :func:`testfile` function::
188
189 import doctest
190 doctest.testfile("example.txt")
191
192That short script executes and verifies any interactive Python examples
193contained in the file :file:`example.txt`. The file content is treated as if it
194were a single giant docstring; the file doesn't need to contain a Python
195program! For example, perhaps :file:`example.txt` contains this::
196
197 The ``example`` module
198 ======================
199
200 Using ``factorial``
201 -------------------
202
203 This is an example text file in reStructuredText format. First import
204 ``factorial`` from the ``example`` module:
205
206 >>> from example import factorial
207
208 Now use it:
209
210 >>> factorial(6)
211 120
212
213Running ``doctest.testfile("example.txt")`` then finds the error in this
214documentation::
215
216 File "./example.txt", line 14, in example.txt
217 Failed example:
218 factorial(6)
219 Expected:
220 120
221 Got:
222 720
223
224As with :func:`testmod`, :func:`testfile` won't display anything unless an
225example fails. If an example does fail, then the failing example(s) and the
226cause(s) of the failure(s) are printed to stdout, using the same format as
227:func:`testmod`.
228
229By default, :func:`testfile` looks for files in the calling module's directory.
230See section :ref:`doctest-basic-api` for a description of the optional arguments
231that can be used to tell it to look for files in other locations.
232
233Like :func:`testmod`, :func:`testfile`'s verbosity can be set with the
Éric Araujo713d3032010-11-18 16:38:46 +0000234``-v`` command-line switch or with the optional keyword argument
Georg Brandl116aa622007-08-15 14:28:22 +0000235*verbose*.
236
Georg Brandl31835852008-05-12 17:38:56 +0000237There is also a command line shortcut for running :func:`testfile`. You can
238instruct the Python interpreter to run the doctest module directly from the
239standard library and pass the file name(s) on the command line::
Georg Brandl116aa622007-08-15 14:28:22 +0000240
241 python -m doctest -v example.txt
242
243Because the file name does not end with :file:`.py`, :mod:`doctest` infers that
244it must be run with :func:`testfile`, not :func:`testmod`.
245
246For more information on :func:`testfile`, see section :ref:`doctest-basic-api`.
247
248
249.. _doctest-how-it-works:
250
251How It Works
252------------
253
254This section examines in detail how doctest works: which docstrings it looks at,
255how it finds interactive examples, what execution context it uses, how it
256handles exceptions, and how option flags can be used to control its behavior.
257This is the information that you need to know to write doctest examples; for
258information about actually running doctest on these examples, see the following
259sections.
260
261
262.. _doctest-which-docstrings:
263
264Which Docstrings Are Examined?
265^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
266
267The module docstring, and all function, class and method docstrings are
268searched. Objects imported into the module are not searched.
269
270In addition, if ``M.__test__`` exists and "is true", it must be a dict, and each
271entry maps a (string) name to a function object, class object, or string.
272Function and class object docstrings found from ``M.__test__`` are searched, and
273strings are treated as if they were docstrings. In output, a key ``K`` in
274``M.__test__`` appears with name ::
275
276 <name of M>.__test__.K
277
278Any classes found are recursively searched similarly, to test docstrings in
279their contained methods and nested classes.
280
Georg Brandl116aa622007-08-15 14:28:22 +0000281
282.. _doctest-finding-examples:
283
284How are Docstring Examples Recognized?
285^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
286
R. David Murray9691e592010-06-15 23:46:40 +0000287In most cases a copy-and-paste of an interactive console session works fine,
288but doctest isn't trying to do an exact emulation of any specific Python shell.
Georg Brandl116aa622007-08-15 14:28:22 +0000289
Georg Brandl116aa622007-08-15 14:28:22 +0000290::
291
292 >>> # comments are ignored
293 >>> x = 12
294 >>> x
295 12
296 >>> if x == 13:
Georg Brandl6911e3c2007-09-04 07:15:32 +0000297 ... print("yes")
Georg Brandl116aa622007-08-15 14:28:22 +0000298 ... else:
Georg Brandl6911e3c2007-09-04 07:15:32 +0000299 ... print("no")
300 ... print("NO")
301 ... print("NO!!!")
Georg Brandl116aa622007-08-15 14:28:22 +0000302 ...
303 no
304 NO
305 NO!!!
306 >>>
307
308Any expected output must immediately follow the final ``'>>> '`` or ``'... '``
309line containing the code, and the expected output (if any) extends to the next
310``'>>> '`` or all-whitespace line.
311
312The fine print:
313
314* Expected output cannot contain an all-whitespace line, since such a line is
315 taken to signal the end of expected output. If expected output does contain a
316 blank line, put ``<BLANKLINE>`` in your doctest example each place a blank line
317 is expected.
318
R. David Murray9691e592010-06-15 23:46:40 +0000319* All hard tab characters are expanded to spaces, using 8-column tab stops.
320 Tabs in output generated by the tested code are not modified. Because any
321 hard tabs in the sample output *are* expanded, this means that if the code
322 output includes hard tabs, the only way the doctest can pass is if the
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700323 :const:`NORMALIZE_WHITESPACE` option or :ref:`directive <doctest-directives>`
324 is in effect.
R. David Murray9691e592010-06-15 23:46:40 +0000325 Alternatively, the test can be rewritten to capture the output and compare it
326 to an expected value as part of the test. This handling of tabs in the
327 source was arrived at through trial and error, and has proven to be the least
328 error prone way of handling them. It is possible to use a different
329 algorithm for handling tabs by writing a custom :class:`DocTestParser` class.
330
Georg Brandl116aa622007-08-15 14:28:22 +0000331* Output to stdout is captured, but not output to stderr (exception tracebacks
332 are captured via a different means).
333
334* If you continue a line via backslashing in an interactive session, or for any
335 other reason use a backslash, you should use a raw docstring, which will
336 preserve your backslashes exactly as you type them::
337
338 >>> def f(x):
339 ... r'''Backslashes in a raw docstring: m\n'''
Georg Brandl6911e3c2007-09-04 07:15:32 +0000340 >>> print(f.__doc__)
Georg Brandl116aa622007-08-15 14:28:22 +0000341 Backslashes in a raw docstring: m\n
342
343 Otherwise, the backslash will be interpreted as part of the string. For example,
Ezio Melotti694f2332012-09-20 09:47:03 +0300344 the ``\n`` above would be interpreted as a newline character. Alternatively, you
Georg Brandl116aa622007-08-15 14:28:22 +0000345 can double each backslash in the doctest version (and not use a raw string)::
346
347 >>> def f(x):
348 ... '''Backslashes in a raw docstring: m\\n'''
Georg Brandl6911e3c2007-09-04 07:15:32 +0000349 >>> print(f.__doc__)
Georg Brandl116aa622007-08-15 14:28:22 +0000350 Backslashes in a raw docstring: m\n
351
352* The starting column doesn't matter::
353
354 >>> assert "Easy!"
355 >>> import math
356 >>> math.floor(1.9)
R. David Murray7c5714f2009-11-23 03:13:23 +0000357 1
Georg Brandl116aa622007-08-15 14:28:22 +0000358
359 and as many leading whitespace characters are stripped from the expected output
360 as appeared in the initial ``'>>> '`` line that started the example.
361
362
363.. _doctest-execution-context:
364
365What's the Execution Context?
366^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
367
368By default, each time :mod:`doctest` finds a docstring to test, it uses a
369*shallow copy* of :mod:`M`'s globals, so that running tests doesn't change the
370module's real globals, and so that one test in :mod:`M` can't leave behind
371crumbs that accidentally allow another test to work. This means examples can
372freely use any names defined at top-level in :mod:`M`, and names defined earlier
373in the docstring being run. Examples cannot see names defined in other
374docstrings.
375
376You can force use of your own dict as the execution context by passing
377``globs=your_dict`` to :func:`testmod` or :func:`testfile` instead.
378
379
380.. _doctest-exceptions:
381
382What About Exceptions?
383^^^^^^^^^^^^^^^^^^^^^^
384
385No problem, provided that the traceback is the only output produced by the
386example: just paste in the traceback. [#]_ Since tracebacks contain details
387that are likely to change rapidly (for example, exact file paths and line
388numbers), this is one case where doctest works hard to be flexible in what it
389accepts.
390
391Simple example::
392
393 >>> [1, 2, 3].remove(42)
394 Traceback (most recent call last):
395 File "<stdin>", line 1, in ?
396 ValueError: list.remove(x): x not in list
397
398That doctest succeeds if :exc:`ValueError` is raised, with the ``list.remove(x):
399x not in list`` detail as shown.
400
401The expected output for an exception must start with a traceback header, which
402may be either of the following two lines, indented the same as the first line of
403the example::
404
405 Traceback (most recent call last):
406 Traceback (innermost last):
407
408The traceback header is followed by an optional traceback stack, whose contents
409are ignored by doctest. The traceback stack is typically omitted, or copied
410verbatim from an interactive session.
411
412The traceback stack is followed by the most interesting part: the line(s)
413containing the exception type and detail. This is usually the last line of a
414traceback, but can extend across multiple lines if the exception has a
415multi-line detail::
416
417 >>> raise ValueError('multi\n line\ndetail')
418 Traceback (most recent call last):
419 File "<stdin>", line 1, in ?
420 ValueError: multi
421 line
422 detail
423
424The last three lines (starting with :exc:`ValueError`) are compared against the
425exception's type and detail, and the rest are ignored.
426
427Best practice is to omit the traceback stack, unless it adds significant
428documentation value to the example. So the last example is probably better as::
429
430 >>> raise ValueError('multi\n line\ndetail')
431 Traceback (most recent call last):
432 ...
433 ValueError: multi
434 line
435 detail
436
437Note that tracebacks are treated very specially. In particular, in the
438rewritten example, the use of ``...`` is independent of doctest's
439:const:`ELLIPSIS` option. The ellipsis in that example could be left out, or
440could just as well be three (or three hundred) commas or digits, or an indented
441transcript of a Monty Python skit.
442
443Some details you should read once, but won't need to remember:
444
445* Doctest can't guess whether your expected output came from an exception
446 traceback or from ordinary printing. So, e.g., an example that expects
447 ``ValueError: 42 is prime`` will pass whether :exc:`ValueError` is actually
448 raised or if the example merely prints that traceback text. In practice,
449 ordinary output rarely begins with a traceback header line, so this doesn't
450 create real problems.
451
452* Each line of the traceback stack (if present) must be indented further than
453 the first line of the example, *or* start with a non-alphanumeric character.
454 The first line following the traceback header indented the same and starting
455 with an alphanumeric is taken to be the start of the exception detail. Of
456 course this does the right thing for genuine tracebacks.
457
Nick Coghlan5e76e942010-06-12 13:42:46 +0000458* When the :const:`IGNORE_EXCEPTION_DETAIL` doctest option is specified,
459 everything following the leftmost colon and any module information in the
460 exception name is ignored.
Georg Brandl116aa622007-08-15 14:28:22 +0000461
462* The interactive shell omits the traceback header line for some
463 :exc:`SyntaxError`\ s. But doctest uses the traceback header line to
464 distinguish exceptions from non-exceptions. So in the rare case where you need
465 to test a :exc:`SyntaxError` that omits the traceback header, you will need to
466 manually add the traceback header line to your test example.
467
468* For some :exc:`SyntaxError`\ s, Python displays the character position of the
469 syntax error, using a ``^`` marker::
470
471 >>> 1 1
472 File "<stdin>", line 1
473 1 1
474 ^
475 SyntaxError: invalid syntax
476
477 Since the lines showing the position of the error come before the exception type
478 and detail, they are not checked by doctest. For example, the following test
479 would pass, even though it puts the ``^`` marker in the wrong location::
480
481 >>> 1 1
482 Traceback (most recent call last):
483 File "<stdin>", line 1
484 1 1
485 ^
486 SyntaxError: invalid syntax
487
Georg Brandl116aa622007-08-15 14:28:22 +0000488
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700489.. _option-flags-and-directives:
Georg Brandl116aa622007-08-15 14:28:22 +0000490.. _doctest-options:
491
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700492Option Flags
493^^^^^^^^^^^^
Georg Brandl116aa622007-08-15 14:28:22 +0000494
495A number of option flags control various aspects of doctest's behavior.
496Symbolic names for the flags are supplied as module constants, which can be
497or'ed together and passed to various functions. The names can also be used in
R David Murray5707d502013-06-23 14:24:13 -0400498:ref:`doctest directives <doctest-directives>`, and may be passed to the
499doctest command line interface via the ``-o`` option.
500
501.. versionadded:: 3.4 the ``-o`` command line option
Georg Brandl116aa622007-08-15 14:28:22 +0000502
503The first group of options define test semantics, controlling aspects of how
504doctest decides whether actual output matches an example's expected output:
505
506
507.. data:: DONT_ACCEPT_TRUE_FOR_1
508
509 By default, if an expected output block contains just ``1``, an actual output
510 block containing just ``1`` or just ``True`` is considered to be a match, and
511 similarly for ``0`` versus ``False``. When :const:`DONT_ACCEPT_TRUE_FOR_1` is
512 specified, neither substitution is allowed. The default behavior caters to that
513 Python changed the return type of many functions from integer to boolean;
514 doctests expecting "little integer" output still work in these cases. This
515 option will probably go away, but not for several years.
516
517
518.. data:: DONT_ACCEPT_BLANKLINE
519
520 By default, if an expected output block contains a line containing only the
521 string ``<BLANKLINE>``, then that line will match a blank line in the actual
522 output. Because a genuinely blank line delimits the expected output, this is
523 the only way to communicate that a blank line is expected. When
524 :const:`DONT_ACCEPT_BLANKLINE` is specified, this substitution is not allowed.
525
526
527.. data:: NORMALIZE_WHITESPACE
528
529 When specified, all sequences of whitespace (blanks and newlines) are treated as
530 equal. Any sequence of whitespace within the expected output will match any
531 sequence of whitespace within the actual output. By default, whitespace must
532 match exactly. :const:`NORMALIZE_WHITESPACE` is especially useful when a line of
533 expected output is very long, and you want to wrap it across multiple lines in
534 your source.
535
536
537.. data:: ELLIPSIS
538
539 When specified, an ellipsis marker (``...``) in the expected output can match
540 any substring in the actual output. This includes substrings that span line
541 boundaries, and empty substrings, so it's best to keep usage of this simple.
542 Complicated uses can lead to the same kinds of "oops, it matched too much!"
543 surprises that ``.*`` is prone to in regular expressions.
544
545
546.. data:: IGNORE_EXCEPTION_DETAIL
547
548 When specified, an example that expects an exception passes if an exception of
549 the expected type is raised, even if the exception detail does not match. For
550 example, an example expecting ``ValueError: 42`` will pass if the actual
551 exception raised is ``ValueError: 3*14``, but will fail, e.g., if
552 :exc:`TypeError` is raised.
553
Nick Coghlan5e76e942010-06-12 13:42:46 +0000554 It will also ignore the module name used in Python 3 doctest reports. Hence
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700555 both of these variations will work with the flag specified, regardless of
556 whether the test is run under Python 2.7 or Python 3.2 (or later versions)::
Nick Coghlan5e76e942010-06-12 13:42:46 +0000557
Chris Jerdonek3fa8c592012-10-10 08:34:38 -0700558 >>> raise CustomError('message')
Nick Coghlan5e76e942010-06-12 13:42:46 +0000559 Traceback (most recent call last):
560 CustomError: message
561
Chris Jerdonek3fa8c592012-10-10 08:34:38 -0700562 >>> raise CustomError('message')
Nick Coghlan5e76e942010-06-12 13:42:46 +0000563 Traceback (most recent call last):
564 my_module.CustomError: message
565
566 Note that :const:`ELLIPSIS` can also be used to ignore the
567 details of the exception message, but such a test may still fail based
568 on whether or not the module details are printed as part of the
569 exception name. Using :const:`IGNORE_EXCEPTION_DETAIL` and the details
570 from Python 2.3 is also the only clear way to write a doctest that doesn't
571 care about the exception detail yet continues to pass under Python 2.3 or
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700572 earlier (those releases do not support :ref:`doctest directives
573 <doctest-directives>` and ignore them as irrelevant comments). For example::
Georg Brandl116aa622007-08-15 14:28:22 +0000574
Chris Jerdonek3fa8c592012-10-10 08:34:38 -0700575 >>> (1, 2)[3] = 'moo'
Georg Brandl116aa622007-08-15 14:28:22 +0000576 Traceback (most recent call last):
577 File "<stdin>", line 1, in ?
578 TypeError: object doesn't support item assignment
579
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700580 passes under Python 2.3 and later Python versions with the flag specified,
581 even though the detail
Nick Coghlan5e76e942010-06-12 13:42:46 +0000582 changed in Python 2.4 to say "does not" instead of "doesn't".
583
584 .. versionchanged:: 3.2
Georg Brandl67b21b72010-08-17 15:07:14 +0000585 :const:`IGNORE_EXCEPTION_DETAIL` now also ignores any information relating
586 to the module containing the exception under test.
Georg Brandl116aa622007-08-15 14:28:22 +0000587
588
589.. data:: SKIP
590
591 When specified, do not run the example at all. This can be useful in contexts
592 where doctest examples serve as both documentation and test cases, and an
593 example should be included for documentation purposes, but should not be
594 checked. E.g., the example's output might be random; or the example might
595 depend on resources which would be unavailable to the test driver.
596
597 The SKIP flag can also be used for temporarily "commenting out" examples.
598
599
600.. data:: COMPARISON_FLAGS
601
602 A bitmask or'ing together all the comparison flags above.
603
604The second group of options controls how test failures are reported:
605
606
607.. data:: REPORT_UDIFF
608
609 When specified, failures that involve multi-line expected and actual outputs are
610 displayed using a unified diff.
611
612
613.. data:: REPORT_CDIFF
614
615 When specified, failures that involve multi-line expected and actual outputs
616 will be displayed using a context diff.
617
618
619.. data:: REPORT_NDIFF
620
621 When specified, differences are computed by ``difflib.Differ``, using the same
622 algorithm as the popular :file:`ndiff.py` utility. This is the only method that
623 marks differences within lines as well as across lines. For example, if a line
624 of expected output contains digit ``1`` where actual output contains letter
625 ``l``, a line is inserted with a caret marking the mismatching column positions.
626
627
628.. data:: REPORT_ONLY_FIRST_FAILURE
629
630 When specified, display the first failing example in each doctest, but suppress
631 output for all remaining examples. This will prevent doctest from reporting
632 correct examples that break because of earlier failures; but it might also hide
633 incorrect examples that fail independently of the first failure. When
634 :const:`REPORT_ONLY_FIRST_FAILURE` is specified, the remaining examples are
635 still run, and still count towards the total number of failures reported; only
636 the output is suppressed.
637
638
R David Murray5a9d7062012-11-21 15:09:21 -0500639.. data:: FAIL_FAST
640
641 When specified, exit after the first failing example and don't attempt to run
R David Murray60dd6e52012-11-22 06:22:41 -0500642 the remaining examples. Thus, the number of failures reported will be at most
643 1. This flag may be useful during debugging, since examples after the first
R David Murray5a9d7062012-11-21 15:09:21 -0500644 failure won't even produce debugging output.
645
R David Murray5707d502013-06-23 14:24:13 -0400646 The doctest command line accepts the option ``-f`` as a shorthand for ``-o
647 FAIL_FAST``.
648
R David Murray5a9d7062012-11-21 15:09:21 -0500649 .. versionadded:: 3.4
650
651
Georg Brandl116aa622007-08-15 14:28:22 +0000652.. data:: REPORTING_FLAGS
653
654 A bitmask or'ing together all the reporting flags above.
655
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700656
657There is also a way to register new option flag names, though this isn't
658useful unless you intend to extend :mod:`doctest` internals via subclassing:
659
660
661.. function:: register_optionflag(name)
662
663 Create a new option flag with a given name, and return the new flag's integer
664 value. :func:`register_optionflag` can be used when subclassing
665 :class:`OutputChecker` or :class:`DocTestRunner` to create new options that are
666 supported by your subclasses. :func:`register_optionflag` should always be
667 called using the following idiom::
668
669 MY_FLAG = register_optionflag('MY_FLAG')
670
671
672.. _doctest-directives:
673
674Directives
675^^^^^^^^^^
676
677Doctest directives may be used to modify the :ref:`option flags
678<doctest-options>` for an individual example. Doctest directives are
679special Python comments following an example's source code:
Georg Brandl116aa622007-08-15 14:28:22 +0000680
681.. productionlist:: doctest
682 directive: "#" "doctest:" `directive_options`
683 directive_options: `directive_option` ("," `directive_option`)\*
684 directive_option: `on_or_off` `directive_option_name`
685 on_or_off: "+" \| "-"
686 directive_option_name: "DONT_ACCEPT_BLANKLINE" \| "NORMALIZE_WHITESPACE" \| ...
687
688Whitespace is not allowed between the ``+`` or ``-`` and the directive option
689name. The directive option name can be any of the option flag names explained
690above.
691
692An example's doctest directives modify doctest's behavior for that single
693example. Use ``+`` to enable the named behavior, or ``-`` to disable it.
694
Georg Brandl83e51f42012-10-10 16:45:11 +0200695For example, this test passes::
Nick Coghlan8f80e0a2012-10-03 12:21:44 +0530696
Georg Brandl83e51f42012-10-10 16:45:11 +0200697 >>> print(list(range(20))) # doctest: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000698 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9,
699 10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
700
701Without the directive it would fail, both because the actual output doesn't have
702two blanks before the single-digit list elements, and because the actual output
703is on a single line. This test also passes, and also requires a directive to do
Georg Brandl83e51f42012-10-10 16:45:11 +0200704so::
Georg Brandl116aa622007-08-15 14:28:22 +0000705
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000706 >>> print(list(range(20))) # doctest: +ELLIPSIS
Georg Brandl116aa622007-08-15 14:28:22 +0000707 [0, 1, ..., 18, 19]
708
Nick Coghlan0b26ccf2012-10-03 13:52:48 +0530709Multiple directives can be used on a single physical line, separated by
Georg Brandl83e51f42012-10-10 16:45:11 +0200710commas::
Georg Brandl116aa622007-08-15 14:28:22 +0000711
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000712 >>> print(list(range(20))) # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000713 [0, 1, ..., 18, 19]
714
715If multiple directive comments are used for a single example, then they are
Georg Brandl83e51f42012-10-10 16:45:11 +0200716combined::
Georg Brandl116aa622007-08-15 14:28:22 +0000717
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000718 >>> print(list(range(20))) # doctest: +ELLIPSIS
719 ... # doctest: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000720 [0, 1, ..., 18, 19]
721
722As the previous example shows, you can add ``...`` lines to your example
723containing only directives. This can be useful when an example is too long for
Georg Brandl83e51f42012-10-10 16:45:11 +0200724a directive to comfortably fit on the same line::
Georg Brandl116aa622007-08-15 14:28:22 +0000725
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000726 >>> print(list(range(5)) + list(range(10, 20)) + list(range(30, 40)))
Georg Brandl116aa622007-08-15 14:28:22 +0000727 ... # doctest: +ELLIPSIS
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000728 [0, ..., 4, 10, ..., 19, 30, ..., 39]
Georg Brandl116aa622007-08-15 14:28:22 +0000729
730Note that since all options are disabled by default, and directives apply only
731to the example they appear in, enabling options (via ``+`` in a directive) is
732usually the only meaningful choice. However, option flags can also be passed to
733functions that run doctests, establishing different defaults. In such cases,
734disabling an option via ``-`` in a directive can be useful.
735
Georg Brandl116aa622007-08-15 14:28:22 +0000736
737.. _doctest-warnings:
738
739Warnings
740^^^^^^^^
741
742:mod:`doctest` is serious about requiring exact matches in expected output. If
743even a single character doesn't match, the test fails. This will probably
744surprise you a few times, as you learn exactly what Python does and doesn't
745guarantee about output. For example, when printing a dict, Python doesn't
746guarantee that the key-value pairs will be printed in any particular order, so a
Christian Heimes5b5e81c2007-12-31 16:14:33 +0000747test like ::
Georg Brandl116aa622007-08-15 14:28:22 +0000748
749 >>> foo()
750 {"Hermione": "hippogryph", "Harry": "broomstick"}
751
752is vulnerable! One workaround is to do ::
753
754 >>> foo() == {"Hermione": "hippogryph", "Harry": "broomstick"}
755 True
756
757instead. Another is to do ::
758
Ezio Melotti8f7649e2009-09-13 04:48:45 +0000759 >>> d = sorted(foo().items())
Georg Brandl116aa622007-08-15 14:28:22 +0000760 >>> d
761 [('Harry', 'broomstick'), ('Hermione', 'hippogryph')]
762
763There are others, but you get the idea.
764
765Another bad idea is to print things that embed an object address, like ::
766
767 >>> id(1.0) # certain to fail some of the time
768 7948648
769 >>> class C: pass
770 >>> C() # the default repr() for instances embeds an address
771 <__main__.C instance at 0x00AC18F0>
772
Georg Brandl23a87de2012-10-10 16:56:15 +0200773The :const:`ELLIPSIS` directive gives a nice approach for the last example::
Georg Brandl116aa622007-08-15 14:28:22 +0000774
775 >>> C() #doctest: +ELLIPSIS
776 <__main__.C instance at 0x...>
777
778Floating-point numbers are also subject to small output variations across
779platforms, because Python defers to the platform C library for float formatting,
780and C libraries vary widely in quality here. ::
781
782 >>> 1./7 # risky
783 0.14285714285714285
Georg Brandl6911e3c2007-09-04 07:15:32 +0000784 >>> print(1./7) # safer
Georg Brandl116aa622007-08-15 14:28:22 +0000785 0.142857142857
Georg Brandl6911e3c2007-09-04 07:15:32 +0000786 >>> print(round(1./7, 6)) # much safer
Georg Brandl116aa622007-08-15 14:28:22 +0000787 0.142857
788
789Numbers of the form ``I/2.**J`` are safe across all platforms, and I often
790contrive doctest examples to produce numbers of that form::
791
792 >>> 3./4 # utterly safe
793 0.75
794
795Simple fractions are also easier for people to understand, and that makes for
796better documentation.
797
798
799.. _doctest-basic-api:
800
801Basic API
802---------
803
804The functions :func:`testmod` and :func:`testfile` provide a simple interface to
805doctest that should be sufficient for most basic uses. For a less formal
806introduction to these two functions, see sections :ref:`doctest-simple-testmod`
807and :ref:`doctest-simple-testfile`.
808
809
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000810.. function:: testfile(filename, module_relative=True, name=None, package=None, globs=None, verbose=None, report=True, optionflags=0, extraglobs=None, raise_on_error=False, parser=DocTestParser(), encoding=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000811
812 All arguments except *filename* are optional, and should be specified in keyword
813 form.
814
815 Test examples in the file named *filename*. Return ``(failure_count,
816 test_count)``.
817
818 Optional argument *module_relative* specifies how the filename should be
819 interpreted:
820
821 * If *module_relative* is ``True`` (the default), then *filename* specifies an
822 OS-independent module-relative path. By default, this path is relative to the
823 calling module's directory; but if the *package* argument is specified, then it
824 is relative to that package. To ensure OS-independence, *filename* should use
825 ``/`` characters to separate path segments, and may not be an absolute path
826 (i.e., it may not begin with ``/``).
827
828 * If *module_relative* is ``False``, then *filename* specifies an OS-specific
829 path. The path may be absolute or relative; relative paths are resolved with
830 respect to the current working directory.
831
832 Optional argument *name* gives the name of the test; by default, or if ``None``,
833 ``os.path.basename(filename)`` is used.
834
835 Optional argument *package* is a Python package or the name of a Python package
836 whose directory should be used as the base directory for a module-relative
837 filename. If no package is specified, then the calling module's directory is
838 used as the base directory for module-relative filenames. It is an error to
839 specify *package* if *module_relative* is ``False``.
840
841 Optional argument *globs* gives a dict to be used as the globals when executing
842 examples. A new shallow copy of this dict is created for the doctest, so its
843 examples start with a clean slate. By default, or if ``None``, a new empty dict
844 is used.
845
846 Optional argument *extraglobs* gives a dict merged into the globals used to
847 execute examples. This works like :meth:`dict.update`: if *globs* and
848 *extraglobs* have a common key, the associated value in *extraglobs* appears in
849 the combined dict. By default, or if ``None``, no extra globals are used. This
850 is an advanced feature that allows parameterization of doctests. For example, a
851 doctest can be written for a base class, using a generic name for the class,
852 then reused to test any number of subclasses by passing an *extraglobs* dict
853 mapping the generic name to the subclass to be tested.
854
855 Optional argument *verbose* prints lots of stuff if true, and prints only
856 failures if false; by default, or if ``None``, it's true if and only if ``'-v'``
857 is in ``sys.argv``.
858
859 Optional argument *report* prints a summary at the end when true, else prints
860 nothing at the end. In verbose mode, the summary is detailed, else the summary
861 is very brief (in fact, empty if all tests passed).
862
863 Optional argument *optionflags* or's together option flags. See section
864 :ref:`doctest-options`.
865
866 Optional argument *raise_on_error* defaults to false. If true, an exception is
867 raised upon the first failure or unexpected exception in an example. This
868 allows failures to be post-mortem debugged. Default behavior is to continue
869 running examples.
870
871 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass) that
872 should be used to extract tests from the files. It defaults to a normal parser
873 (i.e., ``DocTestParser()``).
874
875 Optional argument *encoding* specifies an encoding that should be used to
876 convert the file to unicode.
877
Georg Brandl116aa622007-08-15 14:28:22 +0000878
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000879.. function:: testmod(m=None, name=None, globs=None, verbose=None, report=True, optionflags=0, extraglobs=None, raise_on_error=False, exclude_empty=False)
Georg Brandl116aa622007-08-15 14:28:22 +0000880
881 All arguments are optional, and all except for *m* should be specified in
882 keyword form.
883
884 Test examples in docstrings in functions and classes reachable from module *m*
885 (or module :mod:`__main__` if *m* is not supplied or is ``None``), starting with
886 ``m.__doc__``.
887
888 Also test examples reachable from dict ``m.__test__``, if it exists and is not
889 ``None``. ``m.__test__`` maps names (strings) to functions, classes and
890 strings; function and class docstrings are searched for examples; strings are
891 searched directly, as if they were docstrings.
892
893 Only docstrings attached to objects belonging to module *m* are searched.
894
895 Return ``(failure_count, test_count)``.
896
897 Optional argument *name* gives the name of the module; by default, or if
898 ``None``, ``m.__name__`` is used.
899
900 Optional argument *exclude_empty* defaults to false. If true, objects for which
901 no doctests are found are excluded from consideration. The default is a backward
902 compatibility hack, so that code still using :meth:`doctest.master.summarize` in
903 conjunction with :func:`testmod` continues to get output for objects with no
904 tests. The *exclude_empty* argument to the newer :class:`DocTestFinder`
905 constructor defaults to true.
906
907 Optional arguments *extraglobs*, *verbose*, *report*, *optionflags*,
908 *raise_on_error*, and *globs* are the same as for function :func:`testfile`
909 above, except that *globs* defaults to ``m.__dict__``.
910
Georg Brandl116aa622007-08-15 14:28:22 +0000911
912There's also a function to run the doctests associated with a single object.
913This function is provided for backward compatibility. There are no plans to
914deprecate it, but it's rarely useful:
915
916
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000917.. function:: run_docstring_examples(f, globs, verbose=False, name="NoName", compileflags=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +0000918
919 Test examples associated with object *f*; for example, *f* may be a module,
920 function, or class object.
921
922 A shallow copy of dictionary argument *globs* is used for the execution context.
923
924 Optional argument *name* is used in failure messages, and defaults to
925 ``"NoName"``.
926
927 If optional argument *verbose* is true, output is generated even if there are no
928 failures. By default, output is generated only in case of an example failure.
929
930 Optional argument *compileflags* gives the set of flags that should be used by
931 the Python compiler when running the examples. By default, or if ``None``,
932 flags are deduced corresponding to the set of future features found in *globs*.
933
934 Optional argument *optionflags* works as for function :func:`testfile` above.
935
936
937.. _doctest-unittest-api:
938
939Unittest API
940------------
941
942As your collection of doctest'ed modules grows, you'll want a way to run all
Georg Brandl31835852008-05-12 17:38:56 +0000943their doctests systematically. :mod:`doctest` provides two functions that can
944be used to create :mod:`unittest` test suites from modules and text files
Georg Brandla8514832010-07-10 12:20:38 +0000945containing doctests. To integrate with :mod:`unittest` test discovery, include
946a :func:`load_tests` function in your test module::
Georg Brandl116aa622007-08-15 14:28:22 +0000947
948 import unittest
949 import doctest
Georg Brandla8514832010-07-10 12:20:38 +0000950 import my_module_with_doctests
Georg Brandl116aa622007-08-15 14:28:22 +0000951
Georg Brandla8514832010-07-10 12:20:38 +0000952 def load_tests(loader, tests, ignore):
953 tests.addTests(doctest.DocTestSuite(my_module_with_doctests))
R. David Murray796343b2010-12-13 22:50:30 +0000954 return tests
Georg Brandl116aa622007-08-15 14:28:22 +0000955
956There are two main functions for creating :class:`unittest.TestSuite` instances
957from text files and modules with doctests:
958
959
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000960.. function:: DocFileSuite(*paths, module_relative=True, package=None, setUp=None, tearDown=None, globs=None, optionflags=0, parser=DocTestParser(), encoding=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000961
962 Convert doctest tests from one or more text files to a
963 :class:`unittest.TestSuite`.
964
965 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
966 and runs the interactive examples in each file. If an example in any file
967 fails, then the synthesized unit test fails, and a :exc:`failureException`
968 exception is raised showing the name of the file containing the test and a
969 (sometimes approximate) line number.
970
971 Pass one or more paths (as strings) to text files to be examined.
972
973 Options may be provided as keyword arguments:
974
975 Optional argument *module_relative* specifies how the filenames in *paths*
976 should be interpreted:
977
Benjamin Petersond23f8222009-04-05 19:13:16 +0000978 * If *module_relative* is ``True`` (the default), then each filename in
979 *paths* specifies an OS-independent module-relative path. By default, this
980 path is relative to the calling module's directory; but if the *package*
981 argument is specified, then it is relative to that package. To ensure
982 OS-independence, each filename should use ``/`` characters to separate path
983 segments, and may not be an absolute path (i.e., it may not begin with
984 ``/``).
Georg Brandl116aa622007-08-15 14:28:22 +0000985
Benjamin Petersond23f8222009-04-05 19:13:16 +0000986 * If *module_relative* is ``False``, then each filename in *paths* specifies
987 an OS-specific path. The path may be absolute or relative; relative paths
988 are resolved with respect to the current working directory.
Georg Brandl116aa622007-08-15 14:28:22 +0000989
Benjamin Petersond23f8222009-04-05 19:13:16 +0000990 Optional argument *package* is a Python package or the name of a Python
991 package whose directory should be used as the base directory for
992 module-relative filenames in *paths*. If no package is specified, then the
993 calling module's directory is used as the base directory for module-relative
994 filenames. It is an error to specify *package* if *module_relative* is
995 ``False``.
Georg Brandl116aa622007-08-15 14:28:22 +0000996
Benjamin Petersond23f8222009-04-05 19:13:16 +0000997 Optional argument *setUp* specifies a set-up function for the test suite.
998 This is called before running the tests in each file. The *setUp* function
Georg Brandl116aa622007-08-15 14:28:22 +0000999 will be passed a :class:`DocTest` object. The setUp function can access the
1000 test globals as the *globs* attribute of the test passed.
1001
Benjamin Petersond23f8222009-04-05 19:13:16 +00001002 Optional argument *tearDown* specifies a tear-down function for the test
1003 suite. This is called after running the tests in each file. The *tearDown*
1004 function will be passed a :class:`DocTest` object. The setUp function can
1005 access the test globals as the *globs* attribute of the test passed.
1006
Georg Brandl116aa622007-08-15 14:28:22 +00001007 Optional argument *globs* is a dictionary containing the initial global
1008 variables for the tests. A new copy of this dictionary is created for each
1009 test. By default, *globs* is a new empty dictionary.
1010
1011 Optional argument *optionflags* specifies the default doctest options for the
1012 tests, created by or-ing together individual option flags. See section
Benjamin Petersond23f8222009-04-05 19:13:16 +00001013 :ref:`doctest-options`. See function :func:`set_unittest_reportflags` below
1014 for a better way to set reporting options.
Georg Brandl116aa622007-08-15 14:28:22 +00001015
Benjamin Petersond23f8222009-04-05 19:13:16 +00001016 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass)
1017 that should be used to extract tests from the files. It defaults to a normal
1018 parser (i.e., ``DocTestParser()``).
Georg Brandl116aa622007-08-15 14:28:22 +00001019
1020 Optional argument *encoding* specifies an encoding that should be used to
1021 convert the file to unicode.
1022
Georg Brandl55ac8f02007-09-01 13:51:09 +00001023 The global ``__file__`` is added to the globals provided to doctests loaded
1024 from a text file using :func:`DocFileSuite`.
Georg Brandl116aa622007-08-15 14:28:22 +00001025
1026
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001027.. function:: DocTestSuite(module=None, globs=None, extraglobs=None, test_finder=None, setUp=None, tearDown=None, checker=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001028
1029 Convert doctest tests for a module to a :class:`unittest.TestSuite`.
1030
1031 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
1032 and runs each doctest in the module. If any of the doctests fail, then the
1033 synthesized unit test fails, and a :exc:`failureException` exception is raised
1034 showing the name of the file containing the test and a (sometimes approximate)
1035 line number.
1036
1037 Optional argument *module* provides the module to be tested. It can be a module
1038 object or a (possibly dotted) module name. If not specified, the module calling
1039 this function is used.
1040
1041 Optional argument *globs* is a dictionary containing the initial global
1042 variables for the tests. A new copy of this dictionary is created for each
1043 test. By default, *globs* is a new empty dictionary.
1044
1045 Optional argument *extraglobs* specifies an extra set of global variables, which
1046 is merged into *globs*. By default, no extra globals are used.
1047
1048 Optional argument *test_finder* is the :class:`DocTestFinder` object (or a
1049 drop-in replacement) that is used to extract doctests from the module.
1050
1051 Optional arguments *setUp*, *tearDown*, and *optionflags* are the same as for
1052 function :func:`DocFileSuite` above.
1053
Georg Brandl55ac8f02007-09-01 13:51:09 +00001054 This function uses the same search technique as :func:`testmod`.
Georg Brandl116aa622007-08-15 14:28:22 +00001055
R David Murray5abd76a2012-09-10 10:15:58 -04001056 .. note::
1057 Unlike :func:`testmod` and :class:`DocTestFinder`, this function raises
1058 a :exc:`ValueError` if *module* contains no docstrings. You can prevent
1059 this error by passing a :class:`DocTestFinder` instance as the
1060 *test_finder* argument with its *exclude_empty* keyword argument set
1061 to ``False``::
1062
1063 >>> finder = doctest.DocTestFinder(exclude_empty=False)
1064 >>> suite = doctest.DocTestSuite(test_finder=finder)
1065
Georg Brandl116aa622007-08-15 14:28:22 +00001066
1067Under the covers, :func:`DocTestSuite` creates a :class:`unittest.TestSuite` out
1068of :class:`doctest.DocTestCase` instances, and :class:`DocTestCase` is a
1069subclass of :class:`unittest.TestCase`. :class:`DocTestCase` isn't documented
1070here (it's an internal detail), but studying its code can answer questions about
1071the exact details of :mod:`unittest` integration.
1072
1073Similarly, :func:`DocFileSuite` creates a :class:`unittest.TestSuite` out of
1074:class:`doctest.DocFileCase` instances, and :class:`DocFileCase` is a subclass
1075of :class:`DocTestCase`.
1076
1077So both ways of creating a :class:`unittest.TestSuite` run instances of
1078:class:`DocTestCase`. This is important for a subtle reason: when you run
1079:mod:`doctest` functions yourself, you can control the :mod:`doctest` options in
1080use directly, by passing option flags to :mod:`doctest` functions. However, if
1081you're writing a :mod:`unittest` framework, :mod:`unittest` ultimately controls
1082when and how tests get run. The framework author typically wants to control
1083:mod:`doctest` reporting options (perhaps, e.g., specified by command line
1084options), but there's no way to pass options through :mod:`unittest` to
1085:mod:`doctest` test runners.
1086
1087For this reason, :mod:`doctest` also supports a notion of :mod:`doctest`
1088reporting flags specific to :mod:`unittest` support, via this function:
1089
1090
1091.. function:: set_unittest_reportflags(flags)
1092
1093 Set the :mod:`doctest` reporting flags to use.
1094
1095 Argument *flags* or's together option flags. See section
1096 :ref:`doctest-options`. Only "reporting flags" can be used.
1097
1098 This is a module-global setting, and affects all future doctests run by module
1099 :mod:`unittest`: the :meth:`runTest` method of :class:`DocTestCase` looks at
1100 the option flags specified for the test case when the :class:`DocTestCase`
1101 instance was constructed. If no reporting flags were specified (which is the
1102 typical and expected case), :mod:`doctest`'s :mod:`unittest` reporting flags are
1103 or'ed into the option flags, and the option flags so augmented are passed to the
1104 :class:`DocTestRunner` instance created to run the doctest. If any reporting
1105 flags were specified when the :class:`DocTestCase` instance was constructed,
1106 :mod:`doctest`'s :mod:`unittest` reporting flags are ignored.
1107
1108 The value of the :mod:`unittest` reporting flags in effect before the function
1109 was called is returned by the function.
1110
Georg Brandl116aa622007-08-15 14:28:22 +00001111
1112.. _doctest-advanced-api:
1113
1114Advanced API
1115------------
1116
1117The basic API is a simple wrapper that's intended to make doctest easy to use.
1118It is fairly flexible, and should meet most users' needs; however, if you
1119require more fine-grained control over testing, or wish to extend doctest's
1120capabilities, then you should use the advanced API.
1121
1122The advanced API revolves around two container classes, which are used to store
1123the interactive examples extracted from doctest cases:
1124
Ezio Melotti0639d5a2009-12-19 23:26:38 +00001125* :class:`Example`: A single Python :term:`statement`, paired with its expected
Christian Heimesd8654cf2007-12-02 15:22:16 +00001126 output.
Georg Brandl116aa622007-08-15 14:28:22 +00001127
1128* :class:`DocTest`: A collection of :class:`Example`\ s, typically extracted
1129 from a single docstring or text file.
1130
1131Additional processing classes are defined to find, parse, and run, and check
1132doctest examples:
1133
1134* :class:`DocTestFinder`: Finds all docstrings in a given module, and uses a
1135 :class:`DocTestParser` to create a :class:`DocTest` from every docstring that
1136 contains interactive examples.
1137
1138* :class:`DocTestParser`: Creates a :class:`DocTest` object from a string (such
1139 as an object's docstring).
1140
1141* :class:`DocTestRunner`: Executes the examples in a :class:`DocTest`, and uses
1142 an :class:`OutputChecker` to verify their output.
1143
1144* :class:`OutputChecker`: Compares the actual output from a doctest example with
1145 the expected output, and decides whether they match.
1146
1147The relationships among these processing classes are summarized in the following
1148diagram::
1149
1150 list of:
1151 +------+ +---------+
1152 |module| --DocTestFinder-> | DocTest | --DocTestRunner-> results
1153 +------+ | ^ +---------+ | ^ (printed)
1154 | | | Example | | |
1155 v | | ... | v |
1156 DocTestParser | Example | OutputChecker
1157 +---------+
1158
1159
1160.. _doctest-doctest:
1161
1162DocTest Objects
1163^^^^^^^^^^^^^^^
1164
1165
1166.. class:: DocTest(examples, globs, name, filename, lineno, docstring)
1167
1168 A collection of doctest examples that should be run in a single namespace. The
Senthil Kumarana6bac952011-07-04 11:28:30 -07001169 constructor arguments are used to initialize the attributes of the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001170
Georg Brandl116aa622007-08-15 14:28:22 +00001171
Senthil Kumarana6bac952011-07-04 11:28:30 -07001172 :class:`DocTest` defines the following attributes. They are initialized by
Benjamin Petersone41251e2008-04-25 01:59:09 +00001173 the constructor, and should not be modified directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001174
1175
Benjamin Petersone41251e2008-04-25 01:59:09 +00001176 .. attribute:: examples
Georg Brandl116aa622007-08-15 14:28:22 +00001177
Benjamin Petersone41251e2008-04-25 01:59:09 +00001178 A list of :class:`Example` objects encoding the individual interactive Python
1179 examples that should be run by this test.
Georg Brandl116aa622007-08-15 14:28:22 +00001180
1181
Benjamin Petersone41251e2008-04-25 01:59:09 +00001182 .. attribute:: globs
Georg Brandl116aa622007-08-15 14:28:22 +00001183
Benjamin Petersone41251e2008-04-25 01:59:09 +00001184 The namespace (aka globals) that the examples should be run in. This is a
1185 dictionary mapping names to values. Any changes to the namespace made by the
1186 examples (such as binding new variables) will be reflected in :attr:`globs`
1187 after the test is run.
Georg Brandl116aa622007-08-15 14:28:22 +00001188
1189
Benjamin Petersone41251e2008-04-25 01:59:09 +00001190 .. attribute:: name
Georg Brandl116aa622007-08-15 14:28:22 +00001191
Benjamin Petersone41251e2008-04-25 01:59:09 +00001192 A string name identifying the :class:`DocTest`. Typically, this is the name
1193 of the object or file that the test was extracted from.
Georg Brandl116aa622007-08-15 14:28:22 +00001194
1195
Benjamin Petersone41251e2008-04-25 01:59:09 +00001196 .. attribute:: filename
Georg Brandl116aa622007-08-15 14:28:22 +00001197
Benjamin Petersone41251e2008-04-25 01:59:09 +00001198 The name of the file that this :class:`DocTest` was extracted from; or
1199 ``None`` if the filename is unknown, or if the :class:`DocTest` was not
1200 extracted from a file.
Georg Brandl116aa622007-08-15 14:28:22 +00001201
1202
Benjamin Petersone41251e2008-04-25 01:59:09 +00001203 .. attribute:: lineno
Georg Brandl116aa622007-08-15 14:28:22 +00001204
Benjamin Petersone41251e2008-04-25 01:59:09 +00001205 The line number within :attr:`filename` where this :class:`DocTest` begins, or
1206 ``None`` if the line number is unavailable. This line number is zero-based
1207 with respect to the beginning of the file.
Georg Brandl116aa622007-08-15 14:28:22 +00001208
1209
Benjamin Petersone41251e2008-04-25 01:59:09 +00001210 .. attribute:: docstring
Georg Brandl116aa622007-08-15 14:28:22 +00001211
Benjamin Petersone41251e2008-04-25 01:59:09 +00001212 The string that the test was extracted from, or 'None' if the string is
1213 unavailable, or if the test was not extracted from a string.
Georg Brandl116aa622007-08-15 14:28:22 +00001214
1215
1216.. _doctest-example:
1217
1218Example Objects
1219^^^^^^^^^^^^^^^
1220
1221
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001222.. class:: Example(source, want, exc_msg=None, lineno=0, indent=0, options=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001223
1224 A single interactive example, consisting of a Python statement and its expected
Senthil Kumarana6bac952011-07-04 11:28:30 -07001225 output. The constructor arguments are used to initialize the attributes of
1226 the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001227
Georg Brandl116aa622007-08-15 14:28:22 +00001228
Senthil Kumarana6bac952011-07-04 11:28:30 -07001229 :class:`Example` defines the following attributes. They are initialized by
Benjamin Petersone41251e2008-04-25 01:59:09 +00001230 the constructor, and should not be modified directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001231
1232
Benjamin Petersone41251e2008-04-25 01:59:09 +00001233 .. attribute:: source
Georg Brandl116aa622007-08-15 14:28:22 +00001234
Benjamin Petersone41251e2008-04-25 01:59:09 +00001235 A string containing the example's source code. This source code consists of a
1236 single Python statement, and always ends with a newline; the constructor adds
1237 a newline when necessary.
Georg Brandl116aa622007-08-15 14:28:22 +00001238
1239
Benjamin Petersone41251e2008-04-25 01:59:09 +00001240 .. attribute:: want
Georg Brandl116aa622007-08-15 14:28:22 +00001241
Benjamin Petersone41251e2008-04-25 01:59:09 +00001242 The expected output from running the example's source code (either from
1243 stdout, or a traceback in case of exception). :attr:`want` ends with a
1244 newline unless no output is expected, in which case it's an empty string. The
1245 constructor adds a newline when necessary.
Georg Brandl116aa622007-08-15 14:28:22 +00001246
1247
Benjamin Petersone41251e2008-04-25 01:59:09 +00001248 .. attribute:: exc_msg
Georg Brandl116aa622007-08-15 14:28:22 +00001249
Benjamin Petersone41251e2008-04-25 01:59:09 +00001250 The exception message generated by the example, if the example is expected to
1251 generate an exception; or ``None`` if it is not expected to generate an
1252 exception. This exception message is compared against the return value of
1253 :func:`traceback.format_exception_only`. :attr:`exc_msg` ends with a newline
1254 unless it's ``None``. The constructor adds a newline if needed.
Georg Brandl116aa622007-08-15 14:28:22 +00001255
1256
Benjamin Petersone41251e2008-04-25 01:59:09 +00001257 .. attribute:: lineno
Georg Brandl116aa622007-08-15 14:28:22 +00001258
Benjamin Petersone41251e2008-04-25 01:59:09 +00001259 The line number within the string containing this example where the example
1260 begins. This line number is zero-based with respect to the beginning of the
1261 containing string.
Georg Brandl116aa622007-08-15 14:28:22 +00001262
1263
Benjamin Petersone41251e2008-04-25 01:59:09 +00001264 .. attribute:: indent
Georg Brandl116aa622007-08-15 14:28:22 +00001265
Benjamin Petersone41251e2008-04-25 01:59:09 +00001266 The example's indentation in the containing string, i.e., the number of space
1267 characters that precede the example's first prompt.
Georg Brandl116aa622007-08-15 14:28:22 +00001268
1269
Benjamin Petersone41251e2008-04-25 01:59:09 +00001270 .. attribute:: options
Georg Brandl116aa622007-08-15 14:28:22 +00001271
Benjamin Petersone41251e2008-04-25 01:59:09 +00001272 A dictionary mapping from option flags to ``True`` or ``False``, which is used
1273 to override default options for this example. Any option flags not contained
1274 in this dictionary are left at their default value (as specified by the
1275 :class:`DocTestRunner`'s :attr:`optionflags`). By default, no options are set.
Georg Brandl116aa622007-08-15 14:28:22 +00001276
1277
1278.. _doctest-doctestfinder:
1279
1280DocTestFinder objects
1281^^^^^^^^^^^^^^^^^^^^^
1282
1283
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001284.. class:: DocTestFinder(verbose=False, parser=DocTestParser(), recurse=True, exclude_empty=True)
Georg Brandl116aa622007-08-15 14:28:22 +00001285
1286 A processing class used to extract the :class:`DocTest`\ s that are relevant to
1287 a given object, from its docstring and the docstrings of its contained objects.
1288 :class:`DocTest`\ s can currently be extracted from the following object types:
1289 modules, functions, classes, methods, staticmethods, classmethods, and
1290 properties.
1291
1292 The optional argument *verbose* can be used to display the objects searched by
1293 the finder. It defaults to ``False`` (no output).
1294
1295 The optional argument *parser* specifies the :class:`DocTestParser` object (or a
1296 drop-in replacement) that is used to extract doctests from docstrings.
1297
1298 If the optional argument *recurse* is false, then :meth:`DocTestFinder.find`
1299 will only examine the given object, and not any contained objects.
1300
1301 If the optional argument *exclude_empty* is false, then
1302 :meth:`DocTestFinder.find` will include tests for objects with empty docstrings.
1303
Georg Brandl116aa622007-08-15 14:28:22 +00001304
Benjamin Petersone41251e2008-04-25 01:59:09 +00001305 :class:`DocTestFinder` defines the following method:
Georg Brandl116aa622007-08-15 14:28:22 +00001306
1307
Benjamin Petersone41251e2008-04-25 01:59:09 +00001308 .. method:: find(obj[, name][, module][, globs][, extraglobs])
Georg Brandl116aa622007-08-15 14:28:22 +00001309
Benjamin Petersone41251e2008-04-25 01:59:09 +00001310 Return a list of the :class:`DocTest`\ s that are defined by *obj*'s
1311 docstring, or by any of its contained objects' docstrings.
Georg Brandl116aa622007-08-15 14:28:22 +00001312
Benjamin Petersone41251e2008-04-25 01:59:09 +00001313 The optional argument *name* specifies the object's name; this name will be
1314 used to construct names for the returned :class:`DocTest`\ s. If *name* is
1315 not specified, then ``obj.__name__`` is used.
Georg Brandl116aa622007-08-15 14:28:22 +00001316
Benjamin Petersone41251e2008-04-25 01:59:09 +00001317 The optional parameter *module* is the module that contains the given object.
1318 If the module is not specified or is None, then the test finder will attempt
1319 to automatically determine the correct module. The object's module is used:
Georg Brandl116aa622007-08-15 14:28:22 +00001320
Benjamin Petersone41251e2008-04-25 01:59:09 +00001321 * As a default namespace, if *globs* is not specified.
Georg Brandl116aa622007-08-15 14:28:22 +00001322
Benjamin Petersone41251e2008-04-25 01:59:09 +00001323 * To prevent the DocTestFinder from extracting DocTests from objects that are
1324 imported from other modules. (Contained objects with modules other than
1325 *module* are ignored.)
Georg Brandl116aa622007-08-15 14:28:22 +00001326
Benjamin Petersone41251e2008-04-25 01:59:09 +00001327 * To find the name of the file containing the object.
Georg Brandl116aa622007-08-15 14:28:22 +00001328
Benjamin Petersone41251e2008-04-25 01:59:09 +00001329 * To help find the line number of the object within its file.
Georg Brandl116aa622007-08-15 14:28:22 +00001330
Benjamin Petersone41251e2008-04-25 01:59:09 +00001331 If *module* is ``False``, no attempt to find the module will be made. This is
1332 obscure, of use mostly in testing doctest itself: if *module* is ``False``, or
1333 is ``None`` but cannot be found automatically, then all objects are considered
1334 to belong to the (non-existent) module, so all contained objects will
1335 (recursively) be searched for doctests.
Georg Brandl116aa622007-08-15 14:28:22 +00001336
Benjamin Petersone41251e2008-04-25 01:59:09 +00001337 The globals for each :class:`DocTest` is formed by combining *globs* and
1338 *extraglobs* (bindings in *extraglobs* override bindings in *globs*). A new
1339 shallow copy of the globals dictionary is created for each :class:`DocTest`.
1340 If *globs* is not specified, then it defaults to the module's *__dict__*, if
1341 specified, or ``{}`` otherwise. If *extraglobs* is not specified, then it
1342 defaults to ``{}``.
Georg Brandl116aa622007-08-15 14:28:22 +00001343
1344
1345.. _doctest-doctestparser:
1346
1347DocTestParser objects
1348^^^^^^^^^^^^^^^^^^^^^
1349
1350
1351.. class:: DocTestParser()
1352
1353 A processing class used to extract interactive examples from a string, and use
1354 them to create a :class:`DocTest` object.
1355
Georg Brandl116aa622007-08-15 14:28:22 +00001356
Benjamin Petersone41251e2008-04-25 01:59:09 +00001357 :class:`DocTestParser` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001358
1359
Benjamin Petersone41251e2008-04-25 01:59:09 +00001360 .. method:: get_doctest(string, globs, name, filename, lineno)
Georg Brandl116aa622007-08-15 14:28:22 +00001361
Benjamin Petersone41251e2008-04-25 01:59:09 +00001362 Extract all doctest examples from the given string, and collect them into a
1363 :class:`DocTest` object.
Georg Brandl116aa622007-08-15 14:28:22 +00001364
Benjamin Petersone41251e2008-04-25 01:59:09 +00001365 *globs*, *name*, *filename*, and *lineno* are attributes for the new
1366 :class:`DocTest` object. See the documentation for :class:`DocTest` for more
1367 information.
Georg Brandl116aa622007-08-15 14:28:22 +00001368
1369
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001370 .. method:: get_examples(string, name='<string>')
Georg Brandl116aa622007-08-15 14:28:22 +00001371
Benjamin Petersone41251e2008-04-25 01:59:09 +00001372 Extract all doctest examples from the given string, and return them as a list
1373 of :class:`Example` objects. Line numbers are 0-based. The optional argument
1374 *name* is a name identifying this string, and is only used for error messages.
Georg Brandl116aa622007-08-15 14:28:22 +00001375
1376
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001377 .. method:: parse(string, name='<string>')
Georg Brandl116aa622007-08-15 14:28:22 +00001378
Benjamin Petersone41251e2008-04-25 01:59:09 +00001379 Divide the given string into examples and intervening text, and return them as
1380 a list of alternating :class:`Example`\ s and strings. Line numbers for the
1381 :class:`Example`\ s are 0-based. The optional argument *name* is a name
1382 identifying this string, and is only used for error messages.
Georg Brandl116aa622007-08-15 14:28:22 +00001383
1384
1385.. _doctest-doctestrunner:
1386
1387DocTestRunner objects
1388^^^^^^^^^^^^^^^^^^^^^
1389
1390
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001391.. class:: DocTestRunner(checker=None, verbose=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +00001392
1393 A processing class used to execute and verify the interactive examples in a
1394 :class:`DocTest`.
1395
1396 The comparison between expected outputs and actual outputs is done by an
1397 :class:`OutputChecker`. This comparison may be customized with a number of
1398 option flags; see section :ref:`doctest-options` for more information. If the
1399 option flags are insufficient, then the comparison may also be customized by
1400 passing a subclass of :class:`OutputChecker` to the constructor.
1401
1402 The test runner's display output can be controlled in two ways. First, an output
1403 function can be passed to :meth:`TestRunner.run`; this function will be called
1404 with strings that should be displayed. It defaults to ``sys.stdout.write``. If
1405 capturing the output is not sufficient, then the display output can be also
1406 customized by subclassing DocTestRunner, and overriding the methods
1407 :meth:`report_start`, :meth:`report_success`,
1408 :meth:`report_unexpected_exception`, and :meth:`report_failure`.
1409
1410 The optional keyword argument *checker* specifies the :class:`OutputChecker`
1411 object (or drop-in replacement) that should be used to compare the expected
1412 outputs to the actual outputs of doctest examples.
1413
1414 The optional keyword argument *verbose* controls the :class:`DocTestRunner`'s
1415 verbosity. If *verbose* is ``True``, then information is printed about each
1416 example, as it is run. If *verbose* is ``False``, then only failures are
1417 printed. If *verbose* is unspecified, or ``None``, then verbose output is used
Éric Araujo713d3032010-11-18 16:38:46 +00001418 iff the command-line switch ``-v`` is used.
Georg Brandl116aa622007-08-15 14:28:22 +00001419
1420 The optional keyword argument *optionflags* can be used to control how the test
1421 runner compares expected output to actual output, and how it displays failures.
1422 For more information, see section :ref:`doctest-options`.
1423
Georg Brandl116aa622007-08-15 14:28:22 +00001424
Benjamin Petersone41251e2008-04-25 01:59:09 +00001425 :class:`DocTestParser` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001426
1427
Benjamin Petersone41251e2008-04-25 01:59:09 +00001428 .. method:: report_start(out, test, example)
Georg Brandl116aa622007-08-15 14:28:22 +00001429
Benjamin Petersone41251e2008-04-25 01:59:09 +00001430 Report that the test runner is about to process the given example. This method
1431 is provided to allow subclasses of :class:`DocTestRunner` to customize their
1432 output; it should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001433
Benjamin Petersone41251e2008-04-25 01:59:09 +00001434 *example* is the example about to be processed. *test* is the test
1435 *containing example*. *out* is the output function that was passed to
1436 :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001437
1438
Benjamin Petersone41251e2008-04-25 01:59:09 +00001439 .. method:: report_success(out, test, example, got)
Georg Brandl116aa622007-08-15 14:28:22 +00001440
Benjamin Petersone41251e2008-04-25 01:59:09 +00001441 Report that the given example ran successfully. This method is provided to
1442 allow subclasses of :class:`DocTestRunner` to customize their output; it
1443 should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001444
Benjamin Petersone41251e2008-04-25 01:59:09 +00001445 *example* is the example about to be processed. *got* is the actual output
1446 from the example. *test* is the test containing *example*. *out* is the
1447 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001448
1449
Benjamin Petersone41251e2008-04-25 01:59:09 +00001450 .. method:: report_failure(out, test, example, got)
Georg Brandl116aa622007-08-15 14:28:22 +00001451
Benjamin Petersone41251e2008-04-25 01:59:09 +00001452 Report that the given example failed. This method is provided to allow
1453 subclasses of :class:`DocTestRunner` to customize their output; it should not
1454 be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001455
Benjamin Petersone41251e2008-04-25 01:59:09 +00001456 *example* is the example about to be processed. *got* is the actual output
1457 from the example. *test* is the test containing *example*. *out* is the
1458 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001459
1460
Benjamin Petersone41251e2008-04-25 01:59:09 +00001461 .. method:: report_unexpected_exception(out, test, example, exc_info)
Georg Brandl116aa622007-08-15 14:28:22 +00001462
Benjamin Petersone41251e2008-04-25 01:59:09 +00001463 Report that the given example raised an unexpected exception. This method is
1464 provided to allow subclasses of :class:`DocTestRunner` to customize their
1465 output; it should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001466
Benjamin Petersone41251e2008-04-25 01:59:09 +00001467 *example* is the example about to be processed. *exc_info* is a tuple
1468 containing information about the unexpected exception (as returned by
1469 :func:`sys.exc_info`). *test* is the test containing *example*. *out* is the
1470 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001471
1472
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001473 .. method:: run(test, compileflags=None, out=None, clear_globs=True)
Georg Brandl116aa622007-08-15 14:28:22 +00001474
Benjamin Petersone41251e2008-04-25 01:59:09 +00001475 Run the examples in *test* (a :class:`DocTest` object), and display the
1476 results using the writer function *out*.
Georg Brandl116aa622007-08-15 14:28:22 +00001477
Benjamin Petersone41251e2008-04-25 01:59:09 +00001478 The examples are run in the namespace ``test.globs``. If *clear_globs* is
1479 true (the default), then this namespace will be cleared after the test runs,
1480 to help with garbage collection. If you would like to examine the namespace
1481 after the test completes, then use *clear_globs=False*.
Georg Brandl116aa622007-08-15 14:28:22 +00001482
Benjamin Petersone41251e2008-04-25 01:59:09 +00001483 *compileflags* gives the set of flags that should be used by the Python
1484 compiler when running the examples. If not specified, then it will default to
1485 the set of future-import flags that apply to *globs*.
Georg Brandl116aa622007-08-15 14:28:22 +00001486
Benjamin Petersone41251e2008-04-25 01:59:09 +00001487 The output of each example is checked using the :class:`DocTestRunner`'s
1488 output checker, and the results are formatted by the
1489 :meth:`DocTestRunner.report_\*` methods.
Georg Brandl116aa622007-08-15 14:28:22 +00001490
1491
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001492 .. method:: summarize(verbose=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001493
Benjamin Petersone41251e2008-04-25 01:59:09 +00001494 Print a summary of all the test cases that have been run by this DocTestRunner,
1495 and return a :term:`named tuple` ``TestResults(failed, attempted)``.
Georg Brandl116aa622007-08-15 14:28:22 +00001496
Benjamin Petersone41251e2008-04-25 01:59:09 +00001497 The optional *verbose* argument controls how detailed the summary is. If the
1498 verbosity is not specified, then the :class:`DocTestRunner`'s verbosity is
1499 used.
Georg Brandl116aa622007-08-15 14:28:22 +00001500
1501.. _doctest-outputchecker:
1502
1503OutputChecker objects
1504^^^^^^^^^^^^^^^^^^^^^
1505
1506
1507.. class:: OutputChecker()
1508
1509 A class used to check the whether the actual output from a doctest example
1510 matches the expected output. :class:`OutputChecker` defines two methods:
1511 :meth:`check_output`, which compares a given pair of outputs, and returns true
1512 if they match; and :meth:`output_difference`, which returns a string describing
1513 the differences between two outputs.
1514
Georg Brandl116aa622007-08-15 14:28:22 +00001515
Benjamin Petersone41251e2008-04-25 01:59:09 +00001516 :class:`OutputChecker` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001517
Benjamin Petersone41251e2008-04-25 01:59:09 +00001518 .. method:: check_output(want, got, optionflags)
Georg Brandl116aa622007-08-15 14:28:22 +00001519
Benjamin Petersone41251e2008-04-25 01:59:09 +00001520 Return ``True`` iff the actual output from an example (*got*) matches the
1521 expected output (*want*). These strings are always considered to match if
1522 they are identical; but depending on what option flags the test runner is
1523 using, several non-exact match types are also possible. See section
1524 :ref:`doctest-options` for more information about option flags.
Georg Brandl116aa622007-08-15 14:28:22 +00001525
1526
Benjamin Petersone41251e2008-04-25 01:59:09 +00001527 .. method:: output_difference(example, got, optionflags)
Georg Brandl116aa622007-08-15 14:28:22 +00001528
Benjamin Petersone41251e2008-04-25 01:59:09 +00001529 Return a string describing the differences between the expected output for a
1530 given example (*example*) and the actual output (*got*). *optionflags* is the
1531 set of option flags used to compare *want* and *got*.
Georg Brandl116aa622007-08-15 14:28:22 +00001532
1533
1534.. _doctest-debugging:
1535
1536Debugging
1537---------
1538
1539Doctest provides several mechanisms for debugging doctest examples:
1540
1541* Several functions convert doctests to executable Python programs, which can be
1542 run under the Python debugger, :mod:`pdb`.
1543
1544* The :class:`DebugRunner` class is a subclass of :class:`DocTestRunner` that
1545 raises an exception for the first failing example, containing information about
1546 that example. This information can be used to perform post-mortem debugging on
1547 the example.
1548
1549* The :mod:`unittest` cases generated by :func:`DocTestSuite` support the
1550 :meth:`debug` method defined by :class:`unittest.TestCase`.
1551
1552* You can add a call to :func:`pdb.set_trace` in a doctest example, and you'll
1553 drop into the Python debugger when that line is executed. Then you can inspect
1554 current values of variables, and so on. For example, suppose :file:`a.py`
1555 contains just this module docstring::
1556
1557 """
1558 >>> def f(x):
1559 ... g(x*2)
1560 >>> def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001561 ... print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001562 ... import pdb; pdb.set_trace()
1563 >>> f(3)
1564 9
1565 """
1566
1567 Then an interactive Python session may look like this::
1568
1569 >>> import a, doctest
1570 >>> doctest.testmod(a)
1571 --Return--
1572 > <doctest a[1]>(3)g()->None
1573 -> import pdb; pdb.set_trace()
1574 (Pdb) list
1575 1 def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001576 2 print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001577 3 -> import pdb; pdb.set_trace()
1578 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001579 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001580 6
1581 (Pdb) step
1582 --Return--
1583 > <doctest a[0]>(2)f()->None
1584 -> g(x*2)
1585 (Pdb) list
1586 1 def f(x):
1587 2 -> g(x*2)
1588 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001589 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001590 3
1591 (Pdb) step
1592 --Return--
1593 > <doctest a[2]>(1)?()->None
1594 -> f(3)
1595 (Pdb) cont
1596 (0, 3)
1597 >>>
1598
Georg Brandl116aa622007-08-15 14:28:22 +00001599
1600Functions that convert doctests to Python code, and possibly run the synthesized
1601code under the debugger:
1602
1603
1604.. function:: script_from_examples(s)
1605
1606 Convert text with examples to a script.
1607
1608 Argument *s* is a string containing doctest examples. The string is converted
1609 to a Python script, where doctest examples in *s* are converted to regular code,
1610 and everything else is converted to Python comments. The generated script is
1611 returned as a string. For example, ::
1612
1613 import doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001614 print(doctest.script_from_examples(r"""
Georg Brandl116aa622007-08-15 14:28:22 +00001615 Set x and y to 1 and 2.
1616 >>> x, y = 1, 2
1617
1618 Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001619 >>> print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001620 3
Georg Brandl6911e3c2007-09-04 07:15:32 +00001621 """))
Georg Brandl116aa622007-08-15 14:28:22 +00001622
1623 displays::
1624
1625 # Set x and y to 1 and 2.
1626 x, y = 1, 2
1627 #
1628 # Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001629 print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001630 # Expected:
1631 ## 3
1632
1633 This function is used internally by other functions (see below), but can also be
1634 useful when you want to transform an interactive Python session into a Python
1635 script.
1636
Georg Brandl116aa622007-08-15 14:28:22 +00001637
1638.. function:: testsource(module, name)
1639
1640 Convert the doctest for an object to a script.
1641
1642 Argument *module* is a module object, or dotted name of a module, containing the
1643 object whose doctests are of interest. Argument *name* is the name (within the
1644 module) of the object with the doctests of interest. The result is a string,
1645 containing the object's docstring converted to a Python script, as described for
1646 :func:`script_from_examples` above. For example, if module :file:`a.py`
1647 contains a top-level function :func:`f`, then ::
1648
1649 import a, doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001650 print(doctest.testsource(a, "a.f"))
Georg Brandl116aa622007-08-15 14:28:22 +00001651
1652 prints a script version of function :func:`f`'s docstring, with doctests
1653 converted to code, and the rest placed in comments.
1654
Georg Brandl116aa622007-08-15 14:28:22 +00001655
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001656.. function:: debug(module, name, pm=False)
Georg Brandl116aa622007-08-15 14:28:22 +00001657
1658 Debug the doctests for an object.
1659
1660 The *module* and *name* arguments are the same as for function
1661 :func:`testsource` above. The synthesized Python script for the named object's
1662 docstring is written to a temporary file, and then that file is run under the
1663 control of the Python debugger, :mod:`pdb`.
1664
1665 A shallow copy of ``module.__dict__`` is used for both local and global
1666 execution context.
1667
1668 Optional argument *pm* controls whether post-mortem debugging is used. If *pm*
1669 has a true value, the script file is run directly, and the debugger gets
1670 involved only if the script terminates via raising an unhandled exception. If
1671 it does, then post-mortem debugging is invoked, via :func:`pdb.post_mortem`,
1672 passing the traceback object from the unhandled exception. If *pm* is not
1673 specified, or is false, the script is run under the debugger from the start, via
1674 passing an appropriate :func:`exec` call to :func:`pdb.run`.
1675
Georg Brandl116aa622007-08-15 14:28:22 +00001676
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001677.. function:: debug_src(src, pm=False, globs=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001678
1679 Debug the doctests in a string.
1680
1681 This is like function :func:`debug` above, except that a string containing
1682 doctest examples is specified directly, via the *src* argument.
1683
1684 Optional argument *pm* has the same meaning as in function :func:`debug` above.
1685
1686 Optional argument *globs* gives a dictionary to use as both local and global
1687 execution context. If not specified, or ``None``, an empty dictionary is used.
1688 If specified, a shallow copy of the dictionary is used.
1689
Georg Brandl116aa622007-08-15 14:28:22 +00001690
1691The :class:`DebugRunner` class, and the special exceptions it may raise, are of
1692most interest to testing framework authors, and will only be sketched here. See
1693the source code, and especially :class:`DebugRunner`'s docstring (which is a
1694doctest!) for more details:
1695
1696
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001697.. class:: DebugRunner(checker=None, verbose=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +00001698
1699 A subclass of :class:`DocTestRunner` that raises an exception as soon as a
1700 failure is encountered. If an unexpected exception occurs, an
1701 :exc:`UnexpectedException` exception is raised, containing the test, the
1702 example, and the original exception. If the output doesn't match, then a
1703 :exc:`DocTestFailure` exception is raised, containing the test, the example, and
1704 the actual output.
1705
1706 For information about the constructor parameters and methods, see the
1707 documentation for :class:`DocTestRunner` in section :ref:`doctest-advanced-api`.
1708
1709There are two exceptions that may be raised by :class:`DebugRunner` instances:
1710
1711
1712.. exception:: DocTestFailure(test, example, got)
1713
Georg Brandl7cb13192010-08-03 12:06:29 +00001714 An exception raised by :class:`DocTestRunner` to signal that a doctest example's
Georg Brandl116aa622007-08-15 14:28:22 +00001715 actual output did not match its expected output. The constructor arguments are
Senthil Kumarana6bac952011-07-04 11:28:30 -07001716 used to initialize the attributes of the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001717
Senthil Kumarana6bac952011-07-04 11:28:30 -07001718:exc:`DocTestFailure` defines the following attributes:
Georg Brandl116aa622007-08-15 14:28:22 +00001719
1720
1721.. attribute:: DocTestFailure.test
1722
1723 The :class:`DocTest` object that was being run when the example failed.
1724
1725
1726.. attribute:: DocTestFailure.example
1727
1728 The :class:`Example` that failed.
1729
1730
1731.. attribute:: DocTestFailure.got
1732
1733 The example's actual output.
1734
1735
1736.. exception:: UnexpectedException(test, example, exc_info)
1737
Georg Brandl7cb13192010-08-03 12:06:29 +00001738 An exception raised by :class:`DocTestRunner` to signal that a doctest
1739 example raised an unexpected exception. The constructor arguments are used
Senthil Kumarana6bac952011-07-04 11:28:30 -07001740 to initialize the attributes of the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001741
Senthil Kumarana6bac952011-07-04 11:28:30 -07001742:exc:`UnexpectedException` defines the following attributes:
Georg Brandl116aa622007-08-15 14:28:22 +00001743
1744
1745.. attribute:: UnexpectedException.test
1746
1747 The :class:`DocTest` object that was being run when the example failed.
1748
1749
1750.. attribute:: UnexpectedException.example
1751
1752 The :class:`Example` that failed.
1753
1754
1755.. attribute:: UnexpectedException.exc_info
1756
1757 A tuple containing information about the unexpected exception, as returned by
1758 :func:`sys.exc_info`.
1759
1760
1761.. _doctest-soapbox:
1762
1763Soapbox
1764-------
1765
1766As mentioned in the introduction, :mod:`doctest` has grown to have three primary
1767uses:
1768
1769#. Checking examples in docstrings.
1770
1771#. Regression testing.
1772
1773#. Executable documentation / literate testing.
1774
1775These uses have different requirements, and it is important to distinguish them.
1776In particular, filling your docstrings with obscure test cases makes for bad
1777documentation.
1778
1779When writing a docstring, choose docstring examples with care. There's an art to
1780this that needs to be learned---it may not be natural at first. Examples should
1781add genuine value to the documentation. A good example can often be worth many
1782words. If done with care, the examples will be invaluable for your users, and
1783will pay back the time it takes to collect them many times over as the years go
1784by and things change. I'm still amazed at how often one of my :mod:`doctest`
1785examples stops working after a "harmless" change.
1786
1787Doctest also makes an excellent tool for regression testing, especially if you
1788don't skimp on explanatory text. By interleaving prose and examples, it becomes
1789much easier to keep track of what's actually being tested, and why. When a test
1790fails, good prose can make it much easier to figure out what the problem is, and
1791how it should be fixed. It's true that you could write extensive comments in
1792code-based testing, but few programmers do. Many have found that using doctest
1793approaches instead leads to much clearer tests. Perhaps this is simply because
1794doctest makes writing prose a little easier than writing code, while writing
1795comments in code is a little harder. I think it goes deeper than just that:
1796the natural attitude when writing a doctest-based test is that you want to
1797explain the fine points of your software, and illustrate them with examples.
1798This in turn naturally leads to test files that start with the simplest
1799features, and logically progress to complications and edge cases. A coherent
1800narrative is the result, instead of a collection of isolated functions that test
1801isolated bits of functionality seemingly at random. It's a different attitude,
1802and produces different results, blurring the distinction between testing and
1803explaining.
1804
1805Regression testing is best confined to dedicated objects or files. There are
1806several options for organizing tests:
1807
1808* Write text files containing test cases as interactive examples, and test the
1809 files using :func:`testfile` or :func:`DocFileSuite`. This is recommended,
1810 although is easiest to do for new projects, designed from the start to use
1811 doctest.
1812
1813* Define functions named ``_regrtest_topic`` that consist of single docstrings,
1814 containing test cases for the named topics. These functions can be included in
1815 the same file as the module, or separated out into a separate test file.
1816
1817* Define a ``__test__`` dictionary mapping from regression test topics to
1818 docstrings containing test cases.
1819
1820.. rubric:: Footnotes
1821
1822.. [#] Examples containing both expected output and an exception are not supported.
1823 Trying to guess where one ends and the other begins is too error-prone, and that
1824 also makes for a confusing test.