blob: d463356e7adc5002708d5ad003f4d899152b387d [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.
Terry Jan Reedyfa089b92016-06-11 15:02:54 -04008
Georg Brandl116aa622007-08-15 14:28:22 +00009.. moduleauthor:: Tim Peters <tim@python.org>
10.. sectionauthor:: Tim Peters <tim@python.org>
11.. sectionauthor:: Moshe Zadka <moshez@debian.org>
12.. sectionauthor:: Edward Loper <edloper@users.sourceforge.net>
13
Terry Jan Reedyfa089b92016-06-11 15:02:54 -040014**Source code:** :source:`Lib/doctest.py`
15
16--------------
Georg Brandl116aa622007-08-15 14:28:22 +000017
18The :mod:`doctest` module searches for pieces of text that look like interactive
19Python sessions, and then executes those sessions to verify that they work
20exactly as shown. There are several common ways to use doctest:
21
22* To check that a module's docstrings are up-to-date by verifying that all
23 interactive examples still work as documented.
24
25* To perform regression testing by verifying that interactive examples from a
26 test file or a test object work as expected.
27
28* To write tutorial documentation for a package, liberally illustrated with
29 input-output examples. Depending on whether the examples or the expository text
30 are emphasized, this has the flavor of "literate testing" or "executable
31 documentation".
32
33Here's a complete but small example module::
34
35 """
36 This is the "example" module.
37
38 The example module supplies one function, factorial(). For example,
39
40 >>> factorial(5)
41 120
42 """
43
44 def factorial(n):
45 """Return the factorial of n, an exact integer >= 0.
46
Georg Brandl116aa622007-08-15 14:28:22 +000047 >>> [factorial(n) for n in range(6)]
48 [1, 1, 2, 6, 24, 120]
Georg Brandl116aa622007-08-15 14:28:22 +000049 >>> factorial(30)
Georg Brandl5c106642007-11-29 17:41:05 +000050 265252859812191058636308480000000
Georg Brandl116aa622007-08-15 14:28:22 +000051 >>> factorial(-1)
52 Traceback (most recent call last):
53 ...
54 ValueError: n must be >= 0
55
56 Factorials of floats are OK, but the float must be an exact integer:
57 >>> factorial(30.1)
58 Traceback (most recent call last):
59 ...
60 ValueError: n must be exact integer
61 >>> factorial(30.0)
Georg Brandl5c106642007-11-29 17:41:05 +000062 265252859812191058636308480000000
Georg Brandl116aa622007-08-15 14:28:22 +000063
64 It must also not be ridiculously large:
65 >>> factorial(1e100)
66 Traceback (most recent call last):
67 ...
68 OverflowError: n too large
69 """
70
Georg Brandl116aa622007-08-15 14:28:22 +000071 import math
72 if not n >= 0:
73 raise ValueError("n must be >= 0")
74 if math.floor(n) != n:
75 raise ValueError("n must be exact integer")
76 if n+1 == n: # catch a value like 1e300
77 raise OverflowError("n too large")
78 result = 1
79 factor = 2
80 while factor <= n:
81 result *= factor
82 factor += 1
83 return result
84
Georg Brandl116aa622007-08-15 14:28:22 +000085
86 if __name__ == "__main__":
Guido van Rossum04110fb2007-08-24 16:32:05 +000087 import doctest
88 doctest.testmod()
Georg Brandl116aa622007-08-15 14:28:22 +000089
90If you run :file:`example.py` directly from the command line, :mod:`doctest`
Martin Panter1050d2d2016-07-26 11:18:21 +020091works its magic:
92
93.. code-block:: shell-session
Georg Brandl116aa622007-08-15 14:28:22 +000094
95 $ python example.py
96 $
97
98There's no output! That's normal, and it means all the examples worked. Pass
Éric Araujo713d3032010-11-18 16:38:46 +000099``-v`` to the script, and :mod:`doctest` prints a detailed log of what
Martin Panter1050d2d2016-07-26 11:18:21 +0200100it's trying, and prints a summary at the end:
101
102.. code-block:: shell-session
Georg Brandl116aa622007-08-15 14:28:22 +0000103
104 $ python example.py -v
105 Trying:
106 factorial(5)
107 Expecting:
108 120
109 ok
110 Trying:
111 [factorial(n) for n in range(6)]
112 Expecting:
113 [1, 1, 2, 6, 24, 120]
114 ok
Georg Brandl116aa622007-08-15 14:28:22 +0000115
Martin Panter1050d2d2016-07-26 11:18:21 +0200116And so on, eventually ending with:
117
118.. code-block:: none
Georg Brandl116aa622007-08-15 14:28:22 +0000119
120 Trying:
121 factorial(1e100)
122 Expecting:
123 Traceback (most recent call last):
124 ...
125 OverflowError: n too large
126 ok
Georg Brandl116aa622007-08-15 14:28:22 +0000127 2 items passed all tests:
128 1 tests in __main__
129 8 tests in __main__.factorial
Guido van Rossum04110fb2007-08-24 16:32:05 +0000130 9 tests in 2 items.
Georg Brandl116aa622007-08-15 14:28:22 +0000131 9 passed and 0 failed.
132 Test passed.
133 $
134
135That's all you need to know to start making productive use of :mod:`doctest`!
136Jump in. The following sections provide full details. Note that there are many
137examples of doctests in the standard Python test suite and libraries.
138Especially useful examples can be found in the standard test file
139:file:`Lib/test/test_doctest.py`.
140
141
142.. _doctest-simple-testmod:
143
144Simple Usage: Checking Examples in Docstrings
145---------------------------------------------
146
147The simplest way to start using doctest (but not necessarily the way you'll
148continue to do it) is to end each module :mod:`M` with::
149
Guido van Rossum04110fb2007-08-24 16:32:05 +0000150 if __name__ == "__main__":
Georg Brandl116aa622007-08-15 14:28:22 +0000151 import doctest
152 doctest.testmod()
153
Georg Brandl116aa622007-08-15 14:28:22 +0000154:mod:`doctest` then examines docstrings in module :mod:`M`.
155
156Running the module as a script causes the examples in the docstrings to get
157executed and verified::
158
159 python M.py
160
161This won't display anything unless an example fails, in which case the failing
162example(s) and the cause(s) of the failure(s) are printed to stdout, and the
163final line of output is ``***Test Failed*** N failures.``, where *N* is the
164number of examples that failed.
165
Éric Araujo713d3032010-11-18 16:38:46 +0000166Run it with the ``-v`` switch instead::
Georg Brandl116aa622007-08-15 14:28:22 +0000167
168 python M.py -v
169
170and a detailed report of all examples tried is printed to standard output, along
171with assorted summaries at the end.
172
173You can force verbose mode by passing ``verbose=True`` to :func:`testmod`, or
174prohibit it by passing ``verbose=False``. In either of those cases,
Éric Araujo713d3032010-11-18 16:38:46 +0000175``sys.argv`` is not examined by :func:`testmod` (so passing ``-v`` or not
Georg Brandl116aa622007-08-15 14:28:22 +0000176has no effect).
177
Georg Brandl31835852008-05-12 17:38:56 +0000178There is also a command line shortcut for running :func:`testmod`. You can
179instruct the Python interpreter to run the doctest module directly from the
180standard library and pass the module name(s) on the command line::
Georg Brandl116aa622007-08-15 14:28:22 +0000181
182 python -m doctest -v example.py
183
184This will import :file:`example.py` as a standalone module and run
185:func:`testmod` on it. Note that this may not work correctly if the file is
186part of a package and imports other submodules from that package.
187
188For more information on :func:`testmod`, see section :ref:`doctest-basic-api`.
189
190
191.. _doctest-simple-testfile:
192
193Simple Usage: Checking Examples in a Text File
194----------------------------------------------
195
196Another simple application of doctest is testing interactive examples in a text
197file. This can be done with the :func:`testfile` function::
198
199 import doctest
200 doctest.testfile("example.txt")
201
202That short script executes and verifies any interactive Python examples
203contained in the file :file:`example.txt`. The file content is treated as if it
204were a single giant docstring; the file doesn't need to contain a Python
Martin Panter1050d2d2016-07-26 11:18:21 +0200205program! For example, perhaps :file:`example.txt` contains this:
206
207.. code-block:: none
Georg Brandl116aa622007-08-15 14:28:22 +0000208
209 The ``example`` module
210 ======================
211
212 Using ``factorial``
213 -------------------
214
215 This is an example text file in reStructuredText format. First import
216 ``factorial`` from the ``example`` module:
217
218 >>> from example import factorial
219
220 Now use it:
221
222 >>> factorial(6)
223 120
224
225Running ``doctest.testfile("example.txt")`` then finds the error in this
226documentation::
227
228 File "./example.txt", line 14, in example.txt
229 Failed example:
230 factorial(6)
231 Expected:
232 120
233 Got:
234 720
235
236As with :func:`testmod`, :func:`testfile` won't display anything unless an
237example fails. If an example does fail, then the failing example(s) and the
238cause(s) of the failure(s) are printed to stdout, using the same format as
239:func:`testmod`.
240
241By default, :func:`testfile` looks for files in the calling module's directory.
242See section :ref:`doctest-basic-api` for a description of the optional arguments
243that can be used to tell it to look for files in other locations.
244
245Like :func:`testmod`, :func:`testfile`'s verbosity can be set with the
Éric Araujo713d3032010-11-18 16:38:46 +0000246``-v`` command-line switch or with the optional keyword argument
Georg Brandl116aa622007-08-15 14:28:22 +0000247*verbose*.
248
Georg Brandl31835852008-05-12 17:38:56 +0000249There is also a command line shortcut for running :func:`testfile`. You can
250instruct the Python interpreter to run the doctest module directly from the
251standard library and pass the file name(s) on the command line::
Georg Brandl116aa622007-08-15 14:28:22 +0000252
253 python -m doctest -v example.txt
254
255Because the file name does not end with :file:`.py`, :mod:`doctest` infers that
256it must be run with :func:`testfile`, not :func:`testmod`.
257
258For more information on :func:`testfile`, see section :ref:`doctest-basic-api`.
259
260
261.. _doctest-how-it-works:
262
263How It Works
264------------
265
266This section examines in detail how doctest works: which docstrings it looks at,
267how it finds interactive examples, what execution context it uses, how it
268handles exceptions, and how option flags can be used to control its behavior.
269This is the information that you need to know to write doctest examples; for
270information about actually running doctest on these examples, see the following
271sections.
272
273
274.. _doctest-which-docstrings:
275
276Which Docstrings Are Examined?
277^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
278
279The module docstring, and all function, class and method docstrings are
280searched. Objects imported into the module are not searched.
281
282In addition, if ``M.__test__`` exists and "is true", it must be a dict, and each
283entry maps a (string) name to a function object, class object, or string.
284Function and class object docstrings found from ``M.__test__`` are searched, and
285strings are treated as if they were docstrings. In output, a key ``K`` in
286``M.__test__`` appears with name ::
287
288 <name of M>.__test__.K
289
290Any classes found are recursively searched similarly, to test docstrings in
291their contained methods and nested classes.
292
Zachary Warea4b7a752013-11-24 01:19:09 -0600293.. impl-detail::
294 Prior to version 3.4, extension modules written in C were not fully
295 searched by doctest.
296
Georg Brandl116aa622007-08-15 14:28:22 +0000297
298.. _doctest-finding-examples:
299
300How are Docstring Examples Recognized?
301^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
302
R. David Murray9691e592010-06-15 23:46:40 +0000303In most cases a copy-and-paste of an interactive console session works fine,
304but doctest isn't trying to do an exact emulation of any specific Python shell.
Georg Brandl116aa622007-08-15 14:28:22 +0000305
Georg Brandl116aa622007-08-15 14:28:22 +0000306::
307
308 >>> # comments are ignored
309 >>> x = 12
310 >>> x
311 12
312 >>> if x == 13:
Georg Brandl6911e3c2007-09-04 07:15:32 +0000313 ... print("yes")
Georg Brandl116aa622007-08-15 14:28:22 +0000314 ... else:
Georg Brandl6911e3c2007-09-04 07:15:32 +0000315 ... print("no")
316 ... print("NO")
317 ... print("NO!!!")
Georg Brandl116aa622007-08-15 14:28:22 +0000318 ...
319 no
320 NO
321 NO!!!
322 >>>
323
Serhiy Storchaka9a75b842018-10-26 11:18:42 +0300324.. index::
325 single: >>>; interpreter prompt
326 single: ...; interpreter prompt
327
Georg Brandl116aa622007-08-15 14:28:22 +0000328Any expected output must immediately follow the final ``'>>> '`` or ``'... '``
329line containing the code, and the expected output (if any) extends to the next
330``'>>> '`` or all-whitespace line.
331
332The fine print:
333
334* Expected output cannot contain an all-whitespace line, since such a line is
335 taken to signal the end of expected output. If expected output does contain a
336 blank line, put ``<BLANKLINE>`` in your doctest example each place a blank line
337 is expected.
338
R. David Murray9691e592010-06-15 23:46:40 +0000339* All hard tab characters are expanded to spaces, using 8-column tab stops.
340 Tabs in output generated by the tested code are not modified. Because any
341 hard tabs in the sample output *are* expanded, this means that if the code
342 output includes hard tabs, the only way the doctest can pass is if the
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700343 :const:`NORMALIZE_WHITESPACE` option or :ref:`directive <doctest-directives>`
344 is in effect.
R. David Murray9691e592010-06-15 23:46:40 +0000345 Alternatively, the test can be rewritten to capture the output and compare it
346 to an expected value as part of the test. This handling of tabs in the
347 source was arrived at through trial and error, and has proven to be the least
348 error prone way of handling them. It is possible to use a different
349 algorithm for handling tabs by writing a custom :class:`DocTestParser` class.
350
Georg Brandl116aa622007-08-15 14:28:22 +0000351* Output to stdout is captured, but not output to stderr (exception tracebacks
352 are captured via a different means).
353
354* If you continue a line via backslashing in an interactive session, or for any
355 other reason use a backslash, you should use a raw docstring, which will
356 preserve your backslashes exactly as you type them::
357
358 >>> def f(x):
359 ... r'''Backslashes in a raw docstring: m\n'''
Georg Brandl6911e3c2007-09-04 07:15:32 +0000360 >>> print(f.__doc__)
Georg Brandl116aa622007-08-15 14:28:22 +0000361 Backslashes in a raw docstring: m\n
362
363 Otherwise, the backslash will be interpreted as part of the string. For example,
Ezio Melotti694f2332012-09-20 09:47:03 +0300364 the ``\n`` above would be interpreted as a newline character. Alternatively, you
Georg Brandl116aa622007-08-15 14:28:22 +0000365 can double each backslash in the doctest version (and not use a raw string)::
366
367 >>> def f(x):
368 ... '''Backslashes in a raw docstring: m\\n'''
Georg Brandl6911e3c2007-09-04 07:15:32 +0000369 >>> print(f.__doc__)
Georg Brandl116aa622007-08-15 14:28:22 +0000370 Backslashes in a raw docstring: m\n
371
372* The starting column doesn't matter::
373
374 >>> assert "Easy!"
375 >>> import math
376 >>> math.floor(1.9)
R. David Murray7c5714f2009-11-23 03:13:23 +0000377 1
Georg Brandl116aa622007-08-15 14:28:22 +0000378
379 and as many leading whitespace characters are stripped from the expected output
380 as appeared in the initial ``'>>> '`` line that started the example.
381
382
383.. _doctest-execution-context:
384
385What's the Execution Context?
386^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
387
388By default, each time :mod:`doctest` finds a docstring to test, it uses a
389*shallow copy* of :mod:`M`'s globals, so that running tests doesn't change the
390module's real globals, and so that one test in :mod:`M` can't leave behind
391crumbs that accidentally allow another test to work. This means examples can
392freely use any names defined at top-level in :mod:`M`, and names defined earlier
393in the docstring being run. Examples cannot see names defined in other
394docstrings.
395
396You can force use of your own dict as the execution context by passing
397``globs=your_dict`` to :func:`testmod` or :func:`testfile` instead.
398
399
400.. _doctest-exceptions:
401
402What About Exceptions?
403^^^^^^^^^^^^^^^^^^^^^^
404
405No problem, provided that the traceback is the only output produced by the
406example: just paste in the traceback. [#]_ Since tracebacks contain details
407that are likely to change rapidly (for example, exact file paths and line
408numbers), this is one case where doctest works hard to be flexible in what it
409accepts.
410
411Simple example::
412
413 >>> [1, 2, 3].remove(42)
414 Traceback (most recent call last):
UltimateCoder88569402017-05-03 22:16:45 +0530415 File "<stdin>", line 1, in <module>
Georg Brandl116aa622007-08-15 14:28:22 +0000416 ValueError: list.remove(x): x not in list
417
418That doctest succeeds if :exc:`ValueError` is raised, with the ``list.remove(x):
419x not in list`` detail as shown.
420
421The expected output for an exception must start with a traceback header, which
422may be either of the following two lines, indented the same as the first line of
423the example::
424
425 Traceback (most recent call last):
426 Traceback (innermost last):
427
428The traceback header is followed by an optional traceback stack, whose contents
429are ignored by doctest. The traceback stack is typically omitted, or copied
430verbatim from an interactive session.
431
432The traceback stack is followed by the most interesting part: the line(s)
433containing the exception type and detail. This is usually the last line of a
434traceback, but can extend across multiple lines if the exception has a
435multi-line detail::
436
437 >>> raise ValueError('multi\n line\ndetail')
438 Traceback (most recent call last):
UltimateCoder88569402017-05-03 22:16:45 +0530439 File "<stdin>", line 1, in <module>
Georg Brandl116aa622007-08-15 14:28:22 +0000440 ValueError: multi
441 line
442 detail
443
444The last three lines (starting with :exc:`ValueError`) are compared against the
445exception's type and detail, and the rest are ignored.
446
447Best practice is to omit the traceback stack, unless it adds significant
448documentation value to the example. So the last example is probably better as::
449
450 >>> raise ValueError('multi\n line\ndetail')
451 Traceback (most recent call last):
452 ...
453 ValueError: multi
454 line
455 detail
456
457Note that tracebacks are treated very specially. In particular, in the
458rewritten example, the use of ``...`` is independent of doctest's
459:const:`ELLIPSIS` option. The ellipsis in that example could be left out, or
460could just as well be three (or three hundred) commas or digits, or an indented
461transcript of a Monty Python skit.
462
463Some details you should read once, but won't need to remember:
464
465* Doctest can't guess whether your expected output came from an exception
466 traceback or from ordinary printing. So, e.g., an example that expects
467 ``ValueError: 42 is prime`` will pass whether :exc:`ValueError` is actually
468 raised or if the example merely prints that traceback text. In practice,
469 ordinary output rarely begins with a traceback header line, so this doesn't
470 create real problems.
471
472* Each line of the traceback stack (if present) must be indented further than
473 the first line of the example, *or* start with a non-alphanumeric character.
474 The first line following the traceback header indented the same and starting
475 with an alphanumeric is taken to be the start of the exception detail. Of
476 course this does the right thing for genuine tracebacks.
477
Nick Coghlan5e76e942010-06-12 13:42:46 +0000478* When the :const:`IGNORE_EXCEPTION_DETAIL` doctest option is specified,
479 everything following the leftmost colon and any module information in the
480 exception name is ignored.
Georg Brandl116aa622007-08-15 14:28:22 +0000481
482* The interactive shell omits the traceback header line for some
483 :exc:`SyntaxError`\ s. But doctest uses the traceback header line to
484 distinguish exceptions from non-exceptions. So in the rare case where you need
485 to test a :exc:`SyntaxError` that omits the traceback header, you will need to
486 manually add the traceback header line to your test example.
487
Serhiy Storchaka9a75b842018-10-26 11:18:42 +0300488.. index:: single: ^; caret
489
Georg Brandl116aa622007-08-15 14:28:22 +0000490* For some :exc:`SyntaxError`\ s, Python displays the character position of the
491 syntax error, using a ``^`` marker::
492
493 >>> 1 1
494 File "<stdin>", line 1
495 1 1
496 ^
497 SyntaxError: invalid syntax
498
499 Since the lines showing the position of the error come before the exception type
500 and detail, they are not checked by doctest. For example, the following test
501 would pass, even though it puts the ``^`` marker in the wrong location::
502
503 >>> 1 1
504 Traceback (most recent call last):
505 File "<stdin>", line 1
506 1 1
507 ^
508 SyntaxError: invalid syntax
509
Georg Brandl116aa622007-08-15 14:28:22 +0000510
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700511.. _option-flags-and-directives:
Georg Brandl116aa622007-08-15 14:28:22 +0000512.. _doctest-options:
513
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700514Option Flags
515^^^^^^^^^^^^
Georg Brandl116aa622007-08-15 14:28:22 +0000516
517A number of option flags control various aspects of doctest's behavior.
518Symbolic names for the flags are supplied as module constants, which can be
Mariatta Wijaya81b89772017-02-06 20:15:01 -0800519:ref:`bitwise ORed <bitwise>` together and passed to various functions.
520The names can also be used in :ref:`doctest directives <doctest-directives>`,
521and may be passed to the doctest command line interface via the ``-o`` option.
R David Murray5707d502013-06-23 14:24:13 -0400522
Georg Brandldf48b972014-03-24 09:06:18 +0100523.. versionadded:: 3.4
524 The ``-o`` command line option.
Georg Brandl116aa622007-08-15 14:28:22 +0000525
526The first group of options define test semantics, controlling aspects of how
527doctest decides whether actual output matches an example's expected output:
528
529
530.. data:: DONT_ACCEPT_TRUE_FOR_1
531
532 By default, if an expected output block contains just ``1``, an actual output
533 block containing just ``1`` or just ``True`` is considered to be a match, and
534 similarly for ``0`` versus ``False``. When :const:`DONT_ACCEPT_TRUE_FOR_1` is
535 specified, neither substitution is allowed. The default behavior caters to that
536 Python changed the return type of many functions from integer to boolean;
537 doctests expecting "little integer" output still work in these cases. This
538 option will probably go away, but not for several years.
539
540
Serhiy Storchaka9a75b842018-10-26 11:18:42 +0300541.. index:: single: <BLANKLINE>
Georg Brandl116aa622007-08-15 14:28:22 +0000542.. data:: DONT_ACCEPT_BLANKLINE
543
544 By default, if an expected output block contains a line containing only the
545 string ``<BLANKLINE>``, then that line will match a blank line in the actual
546 output. Because a genuinely blank line delimits the expected output, this is
547 the only way to communicate that a blank line is expected. When
548 :const:`DONT_ACCEPT_BLANKLINE` is specified, this substitution is not allowed.
549
550
551.. data:: NORMALIZE_WHITESPACE
552
553 When specified, all sequences of whitespace (blanks and newlines) are treated as
554 equal. Any sequence of whitespace within the expected output will match any
555 sequence of whitespace within the actual output. By default, whitespace must
556 match exactly. :const:`NORMALIZE_WHITESPACE` is especially useful when a line of
557 expected output is very long, and you want to wrap it across multiple lines in
558 your source.
559
560
Serhiy Storchaka9a75b842018-10-26 11:18:42 +0300561.. index:: single: ...; in doctests
Georg Brandl116aa622007-08-15 14:28:22 +0000562.. data:: ELLIPSIS
563
564 When specified, an ellipsis marker (``...``) in the expected output can match
565 any substring in the actual output. This includes substrings that span line
566 boundaries, and empty substrings, so it's best to keep usage of this simple.
567 Complicated uses can lead to the same kinds of "oops, it matched too much!"
568 surprises that ``.*`` is prone to in regular expressions.
569
570
571.. data:: IGNORE_EXCEPTION_DETAIL
572
573 When specified, an example that expects an exception passes if an exception of
574 the expected type is raised, even if the exception detail does not match. For
575 example, an example expecting ``ValueError: 42`` will pass if the actual
576 exception raised is ``ValueError: 3*14``, but will fail, e.g., if
577 :exc:`TypeError` is raised.
578
Nick Coghlan5e76e942010-06-12 13:42:46 +0000579 It will also ignore the module name used in Python 3 doctest reports. Hence
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700580 both of these variations will work with the flag specified, regardless of
581 whether the test is run under Python 2.7 or Python 3.2 (or later versions)::
Nick Coghlan5e76e942010-06-12 13:42:46 +0000582
Chris Jerdonek3fa8c592012-10-10 08:34:38 -0700583 >>> raise CustomError('message')
Nick Coghlan5e76e942010-06-12 13:42:46 +0000584 Traceback (most recent call last):
585 CustomError: message
586
Chris Jerdonek3fa8c592012-10-10 08:34:38 -0700587 >>> raise CustomError('message')
Nick Coghlan5e76e942010-06-12 13:42:46 +0000588 Traceback (most recent call last):
589 my_module.CustomError: message
590
591 Note that :const:`ELLIPSIS` can also be used to ignore the
592 details of the exception message, but such a test may still fail based
593 on whether or not the module details are printed as part of the
594 exception name. Using :const:`IGNORE_EXCEPTION_DETAIL` and the details
595 from Python 2.3 is also the only clear way to write a doctest that doesn't
596 care about the exception detail yet continues to pass under Python 2.3 or
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700597 earlier (those releases do not support :ref:`doctest directives
598 <doctest-directives>` and ignore them as irrelevant comments). For example::
Georg Brandl116aa622007-08-15 14:28:22 +0000599
Chris Jerdonek3fa8c592012-10-10 08:34:38 -0700600 >>> (1, 2)[3] = 'moo'
Georg Brandl116aa622007-08-15 14:28:22 +0000601 Traceback (most recent call last):
UltimateCoder88569402017-05-03 22:16:45 +0530602 File "<stdin>", line 1, in <module>
Georg Brandl116aa622007-08-15 14:28:22 +0000603 TypeError: object doesn't support item assignment
604
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700605 passes under Python 2.3 and later Python versions with the flag specified,
606 even though the detail
Nick Coghlan5e76e942010-06-12 13:42:46 +0000607 changed in Python 2.4 to say "does not" instead of "doesn't".
608
609 .. versionchanged:: 3.2
Georg Brandl67b21b72010-08-17 15:07:14 +0000610 :const:`IGNORE_EXCEPTION_DETAIL` now also ignores any information relating
611 to the module containing the exception under test.
Georg Brandl116aa622007-08-15 14:28:22 +0000612
613
614.. data:: SKIP
615
616 When specified, do not run the example at all. This can be useful in contexts
617 where doctest examples serve as both documentation and test cases, and an
618 example should be included for documentation purposes, but should not be
619 checked. E.g., the example's output might be random; or the example might
620 depend on resources which would be unavailable to the test driver.
621
622 The SKIP flag can also be used for temporarily "commenting out" examples.
623
624
625.. data:: COMPARISON_FLAGS
626
627 A bitmask or'ing together all the comparison flags above.
628
629The second group of options controls how test failures are reported:
630
631
632.. data:: REPORT_UDIFF
633
634 When specified, failures that involve multi-line expected and actual outputs are
635 displayed using a unified diff.
636
637
638.. data:: REPORT_CDIFF
639
640 When specified, failures that involve multi-line expected and actual outputs
641 will be displayed using a context diff.
642
643
644.. data:: REPORT_NDIFF
645
646 When specified, differences are computed by ``difflib.Differ``, using the same
647 algorithm as the popular :file:`ndiff.py` utility. This is the only method that
648 marks differences within lines as well as across lines. For example, if a line
649 of expected output contains digit ``1`` where actual output contains letter
650 ``l``, a line is inserted with a caret marking the mismatching column positions.
651
652
653.. data:: REPORT_ONLY_FIRST_FAILURE
654
655 When specified, display the first failing example in each doctest, but suppress
656 output for all remaining examples. This will prevent doctest from reporting
657 correct examples that break because of earlier failures; but it might also hide
658 incorrect examples that fail independently of the first failure. When
659 :const:`REPORT_ONLY_FIRST_FAILURE` is specified, the remaining examples are
660 still run, and still count towards the total number of failures reported; only
661 the output is suppressed.
662
663
R David Murray5a9d7062012-11-21 15:09:21 -0500664.. data:: FAIL_FAST
665
666 When specified, exit after the first failing example and don't attempt to run
R David Murray60dd6e52012-11-22 06:22:41 -0500667 the remaining examples. Thus, the number of failures reported will be at most
668 1. This flag may be useful during debugging, since examples after the first
R David Murray5a9d7062012-11-21 15:09:21 -0500669 failure won't even produce debugging output.
670
R David Murray5707d502013-06-23 14:24:13 -0400671 The doctest command line accepts the option ``-f`` as a shorthand for ``-o
672 FAIL_FAST``.
673
R David Murray5a9d7062012-11-21 15:09:21 -0500674 .. versionadded:: 3.4
675
676
Georg Brandl116aa622007-08-15 14:28:22 +0000677.. data:: REPORTING_FLAGS
678
679 A bitmask or'ing together all the reporting flags above.
680
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700681
682There is also a way to register new option flag names, though this isn't
683useful unless you intend to extend :mod:`doctest` internals via subclassing:
684
685
686.. function:: register_optionflag(name)
687
688 Create a new option flag with a given name, and return the new flag's integer
689 value. :func:`register_optionflag` can be used when subclassing
690 :class:`OutputChecker` or :class:`DocTestRunner` to create new options that are
691 supported by your subclasses. :func:`register_optionflag` should always be
692 called using the following idiom::
693
694 MY_FLAG = register_optionflag('MY_FLAG')
695
696
Serhiy Storchaka9a75b842018-10-26 11:18:42 +0300697.. index::
698 single: #; in doctests
699 single: +; in doctests
700 single: -; in doctests
Chris Jerdonek3650ea22012-10-10 06:52:08 -0700701.. _doctest-directives:
702
703Directives
704^^^^^^^^^^
705
706Doctest directives may be used to modify the :ref:`option flags
707<doctest-options>` for an individual example. Doctest directives are
708special Python comments following an example's source code:
Georg Brandl116aa622007-08-15 14:28:22 +0000709
710.. productionlist:: doctest
711 directive: "#" "doctest:" `directive_options`
712 directive_options: `directive_option` ("," `directive_option`)\*
713 directive_option: `on_or_off` `directive_option_name`
714 on_or_off: "+" \| "-"
715 directive_option_name: "DONT_ACCEPT_BLANKLINE" \| "NORMALIZE_WHITESPACE" \| ...
716
717Whitespace is not allowed between the ``+`` or ``-`` and the directive option
718name. The directive option name can be any of the option flag names explained
719above.
720
721An example's doctest directives modify doctest's behavior for that single
722example. Use ``+`` to enable the named behavior, or ``-`` to disable it.
723
Georg Brandl83e51f42012-10-10 16:45:11 +0200724For example, this test passes::
Nick Coghlan8f80e0a2012-10-03 12:21:44 +0530725
Georg Brandl83e51f42012-10-10 16:45:11 +0200726 >>> print(list(range(20))) # doctest: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000727 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9,
728 10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
729
730Without the directive it would fail, both because the actual output doesn't have
731two blanks before the single-digit list elements, and because the actual output
732is on a single line. This test also passes, and also requires a directive to do
Georg Brandl83e51f42012-10-10 16:45:11 +0200733so::
Georg Brandl116aa622007-08-15 14:28:22 +0000734
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000735 >>> print(list(range(20))) # doctest: +ELLIPSIS
Georg Brandl116aa622007-08-15 14:28:22 +0000736 [0, 1, ..., 18, 19]
737
Nick Coghlan0b26ccf2012-10-03 13:52:48 +0530738Multiple directives can be used on a single physical line, separated by
Georg Brandl83e51f42012-10-10 16:45:11 +0200739commas::
Georg Brandl116aa622007-08-15 14:28:22 +0000740
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000741 >>> print(list(range(20))) # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000742 [0, 1, ..., 18, 19]
743
744If multiple directive comments are used for a single example, then they are
Georg Brandl83e51f42012-10-10 16:45:11 +0200745combined::
Georg Brandl116aa622007-08-15 14:28:22 +0000746
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000747 >>> print(list(range(20))) # doctest: +ELLIPSIS
748 ... # doctest: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000749 [0, 1, ..., 18, 19]
750
751As the previous example shows, you can add ``...`` lines to your example
752containing only directives. This can be useful when an example is too long for
Georg Brandl83e51f42012-10-10 16:45:11 +0200753a directive to comfortably fit on the same line::
Georg Brandl116aa622007-08-15 14:28:22 +0000754
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000755 >>> print(list(range(5)) + list(range(10, 20)) + list(range(30, 40)))
Georg Brandl116aa622007-08-15 14:28:22 +0000756 ... # doctest: +ELLIPSIS
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000757 [0, ..., 4, 10, ..., 19, 30, ..., 39]
Georg Brandl116aa622007-08-15 14:28:22 +0000758
759Note that since all options are disabled by default, and directives apply only
760to the example they appear in, enabling options (via ``+`` in a directive) is
761usually the only meaningful choice. However, option flags can also be passed to
762functions that run doctests, establishing different defaults. In such cases,
763disabling an option via ``-`` in a directive can be useful.
764
Georg Brandl116aa622007-08-15 14:28:22 +0000765
766.. _doctest-warnings:
767
768Warnings
769^^^^^^^^
770
771:mod:`doctest` is serious about requiring exact matches in expected output. If
772even a single character doesn't match, the test fails. This will probably
773surprise you a few times, as you learn exactly what Python does and doesn't
774guarantee about output. For example, when printing a dict, Python doesn't
775guarantee that the key-value pairs will be printed in any particular order, so a
Christian Heimes5b5e81c2007-12-31 16:14:33 +0000776test like ::
Georg Brandl116aa622007-08-15 14:28:22 +0000777
778 >>> foo()
779 {"Hermione": "hippogryph", "Harry": "broomstick"}
780
781is vulnerable! One workaround is to do ::
782
783 >>> foo() == {"Hermione": "hippogryph", "Harry": "broomstick"}
784 True
785
786instead. Another is to do ::
787
Ezio Melotti8f7649e2009-09-13 04:48:45 +0000788 >>> d = sorted(foo().items())
Georg Brandl116aa622007-08-15 14:28:22 +0000789 >>> d
790 [('Harry', 'broomstick'), ('Hermione', 'hippogryph')]
791
792There are others, but you get the idea.
793
794Another bad idea is to print things that embed an object address, like ::
795
796 >>> id(1.0) # certain to fail some of the time
797 7948648
798 >>> class C: pass
799 >>> C() # the default repr() for instances embeds an address
800 <__main__.C instance at 0x00AC18F0>
801
Georg Brandl23a87de2012-10-10 16:56:15 +0200802The :const:`ELLIPSIS` directive gives a nice approach for the last example::
Georg Brandl116aa622007-08-15 14:28:22 +0000803
804 >>> C() #doctest: +ELLIPSIS
805 <__main__.C instance at 0x...>
806
807Floating-point numbers are also subject to small output variations across
808platforms, because Python defers to the platform C library for float formatting,
809and C libraries vary widely in quality here. ::
810
811 >>> 1./7 # risky
812 0.14285714285714285
Georg Brandl6911e3c2007-09-04 07:15:32 +0000813 >>> print(1./7) # safer
Georg Brandl116aa622007-08-15 14:28:22 +0000814 0.142857142857
Georg Brandl6911e3c2007-09-04 07:15:32 +0000815 >>> print(round(1./7, 6)) # much safer
Georg Brandl116aa622007-08-15 14:28:22 +0000816 0.142857
817
818Numbers of the form ``I/2.**J`` are safe across all platforms, and I often
819contrive doctest examples to produce numbers of that form::
820
821 >>> 3./4 # utterly safe
822 0.75
823
824Simple fractions are also easier for people to understand, and that makes for
825better documentation.
826
827
828.. _doctest-basic-api:
829
830Basic API
831---------
832
833The functions :func:`testmod` and :func:`testfile` provide a simple interface to
834doctest that should be sufficient for most basic uses. For a less formal
835introduction to these two functions, see sections :ref:`doctest-simple-testmod`
836and :ref:`doctest-simple-testfile`.
837
838
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000839.. 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 +0000840
841 All arguments except *filename* are optional, and should be specified in keyword
842 form.
843
844 Test examples in the file named *filename*. Return ``(failure_count,
845 test_count)``.
846
847 Optional argument *module_relative* specifies how the filename should be
848 interpreted:
849
850 * If *module_relative* is ``True`` (the default), then *filename* specifies an
851 OS-independent module-relative path. By default, this path is relative to the
852 calling module's directory; but if the *package* argument is specified, then it
853 is relative to that package. To ensure OS-independence, *filename* should use
854 ``/`` characters to separate path segments, and may not be an absolute path
855 (i.e., it may not begin with ``/``).
856
857 * If *module_relative* is ``False``, then *filename* specifies an OS-specific
858 path. The path may be absolute or relative; relative paths are resolved with
859 respect to the current working directory.
860
861 Optional argument *name* gives the name of the test; by default, or if ``None``,
862 ``os.path.basename(filename)`` is used.
863
864 Optional argument *package* is a Python package or the name of a Python package
865 whose directory should be used as the base directory for a module-relative
866 filename. If no package is specified, then the calling module's directory is
867 used as the base directory for module-relative filenames. It is an error to
868 specify *package* if *module_relative* is ``False``.
869
870 Optional argument *globs* gives a dict to be used as the globals when executing
871 examples. A new shallow copy of this dict is created for the doctest, so its
872 examples start with a clean slate. By default, or if ``None``, a new empty dict
873 is used.
874
875 Optional argument *extraglobs* gives a dict merged into the globals used to
876 execute examples. This works like :meth:`dict.update`: if *globs* and
877 *extraglobs* have a common key, the associated value in *extraglobs* appears in
878 the combined dict. By default, or if ``None``, no extra globals are used. This
879 is an advanced feature that allows parameterization of doctests. For example, a
880 doctest can be written for a base class, using a generic name for the class,
881 then reused to test any number of subclasses by passing an *extraglobs* dict
882 mapping the generic name to the subclass to be tested.
883
884 Optional argument *verbose* prints lots of stuff if true, and prints only
885 failures if false; by default, or if ``None``, it's true if and only if ``'-v'``
886 is in ``sys.argv``.
887
888 Optional argument *report* prints a summary at the end when true, else prints
889 nothing at the end. In verbose mode, the summary is detailed, else the summary
890 is very brief (in fact, empty if all tests passed).
891
Mariatta Wijaya81b89772017-02-06 20:15:01 -0800892 Optional argument *optionflags* (default value 0) takes the
893 :ref:`bitwise OR <bitwise>` of option flags.
894 See section :ref:`doctest-options`.
Georg Brandl116aa622007-08-15 14:28:22 +0000895
896 Optional argument *raise_on_error* defaults to false. If true, an exception is
897 raised upon the first failure or unexpected exception in an example. This
898 allows failures to be post-mortem debugged. Default behavior is to continue
899 running examples.
900
901 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass) that
902 should be used to extract tests from the files. It defaults to a normal parser
903 (i.e., ``DocTestParser()``).
904
905 Optional argument *encoding* specifies an encoding that should be used to
906 convert the file to unicode.
907
Georg Brandl116aa622007-08-15 14:28:22 +0000908
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000909.. 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 +0000910
911 All arguments are optional, and all except for *m* should be specified in
912 keyword form.
913
914 Test examples in docstrings in functions and classes reachable from module *m*
915 (or module :mod:`__main__` if *m* is not supplied or is ``None``), starting with
916 ``m.__doc__``.
917
918 Also test examples reachable from dict ``m.__test__``, if it exists and is not
919 ``None``. ``m.__test__`` maps names (strings) to functions, classes and
920 strings; function and class docstrings are searched for examples; strings are
921 searched directly, as if they were docstrings.
922
923 Only docstrings attached to objects belonging to module *m* are searched.
924
925 Return ``(failure_count, test_count)``.
926
927 Optional argument *name* gives the name of the module; by default, or if
928 ``None``, ``m.__name__`` is used.
929
930 Optional argument *exclude_empty* defaults to false. If true, objects for which
931 no doctests are found are excluded from consideration. The default is a backward
932 compatibility hack, so that code still using :meth:`doctest.master.summarize` in
933 conjunction with :func:`testmod` continues to get output for objects with no
934 tests. The *exclude_empty* argument to the newer :class:`DocTestFinder`
935 constructor defaults to true.
936
937 Optional arguments *extraglobs*, *verbose*, *report*, *optionflags*,
938 *raise_on_error*, and *globs* are the same as for function :func:`testfile`
939 above, except that *globs* defaults to ``m.__dict__``.
940
Georg Brandl116aa622007-08-15 14:28:22 +0000941
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000942.. function:: run_docstring_examples(f, globs, verbose=False, name="NoName", compileflags=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +0000943
Ethan Furman2a5f9da2015-09-17 22:20:41 -0700944 Test examples associated with object *f*; for example, *f* may be a string,
945 a module, a function, or a class object.
Georg Brandl116aa622007-08-15 14:28:22 +0000946
947 A shallow copy of dictionary argument *globs* is used for the execution context.
948
949 Optional argument *name* is used in failure messages, and defaults to
950 ``"NoName"``.
951
952 If optional argument *verbose* is true, output is generated even if there are no
953 failures. By default, output is generated only in case of an example failure.
954
955 Optional argument *compileflags* gives the set of flags that should be used by
956 the Python compiler when running the examples. By default, or if ``None``,
957 flags are deduced corresponding to the set of future features found in *globs*.
958
959 Optional argument *optionflags* works as for function :func:`testfile` above.
960
961
962.. _doctest-unittest-api:
963
964Unittest API
965------------
966
967As your collection of doctest'ed modules grows, you'll want a way to run all
Georg Brandl31835852008-05-12 17:38:56 +0000968their doctests systematically. :mod:`doctest` provides two functions that can
969be used to create :mod:`unittest` test suites from modules and text files
Georg Brandla8514832010-07-10 12:20:38 +0000970containing doctests. To integrate with :mod:`unittest` test discovery, include
971a :func:`load_tests` function in your test module::
Georg Brandl116aa622007-08-15 14:28:22 +0000972
973 import unittest
974 import doctest
Georg Brandla8514832010-07-10 12:20:38 +0000975 import my_module_with_doctests
Georg Brandl116aa622007-08-15 14:28:22 +0000976
Georg Brandla8514832010-07-10 12:20:38 +0000977 def load_tests(loader, tests, ignore):
978 tests.addTests(doctest.DocTestSuite(my_module_with_doctests))
R. David Murray796343b2010-12-13 22:50:30 +0000979 return tests
Georg Brandl116aa622007-08-15 14:28:22 +0000980
981There are two main functions for creating :class:`unittest.TestSuite` instances
982from text files and modules with doctests:
983
984
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000985.. 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 +0000986
987 Convert doctest tests from one or more text files to a
988 :class:`unittest.TestSuite`.
989
990 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
991 and runs the interactive examples in each file. If an example in any file
992 fails, then the synthesized unit test fails, and a :exc:`failureException`
993 exception is raised showing the name of the file containing the test and a
994 (sometimes approximate) line number.
995
996 Pass one or more paths (as strings) to text files to be examined.
997
998 Options may be provided as keyword arguments:
999
1000 Optional argument *module_relative* specifies how the filenames in *paths*
1001 should be interpreted:
1002
Benjamin Petersond23f8222009-04-05 19:13:16 +00001003 * If *module_relative* is ``True`` (the default), then each filename in
1004 *paths* specifies an OS-independent module-relative path. By default, this
1005 path is relative to the calling module's directory; but if the *package*
1006 argument is specified, then it is relative to that package. To ensure
1007 OS-independence, each filename should use ``/`` characters to separate path
1008 segments, and may not be an absolute path (i.e., it may not begin with
1009 ``/``).
Georg Brandl116aa622007-08-15 14:28:22 +00001010
Benjamin Petersond23f8222009-04-05 19:13:16 +00001011 * If *module_relative* is ``False``, then each filename in *paths* specifies
1012 an OS-specific path. The path may be absolute or relative; relative paths
1013 are resolved with respect to the current working directory.
Georg Brandl116aa622007-08-15 14:28:22 +00001014
Benjamin Petersond23f8222009-04-05 19:13:16 +00001015 Optional argument *package* is a Python package or the name of a Python
1016 package whose directory should be used as the base directory for
1017 module-relative filenames in *paths*. If no package is specified, then the
1018 calling module's directory is used as the base directory for module-relative
1019 filenames. It is an error to specify *package* if *module_relative* is
1020 ``False``.
Georg Brandl116aa622007-08-15 14:28:22 +00001021
Benjamin Petersond23f8222009-04-05 19:13:16 +00001022 Optional argument *setUp* specifies a set-up function for the test suite.
1023 This is called before running the tests in each file. The *setUp* function
Georg Brandl116aa622007-08-15 14:28:22 +00001024 will be passed a :class:`DocTest` object. The setUp function can access the
1025 test globals as the *globs* attribute of the test passed.
1026
Benjamin Petersond23f8222009-04-05 19:13:16 +00001027 Optional argument *tearDown* specifies a tear-down function for the test
1028 suite. This is called after running the tests in each file. The *tearDown*
1029 function will be passed a :class:`DocTest` object. The setUp function can
1030 access the test globals as the *globs* attribute of the test passed.
1031
Georg Brandl116aa622007-08-15 14:28:22 +00001032 Optional argument *globs* is a dictionary containing the initial global
1033 variables for the tests. A new copy of this dictionary is created for each
1034 test. By default, *globs* is a new empty dictionary.
1035
1036 Optional argument *optionflags* specifies the default doctest options for the
1037 tests, created by or-ing together individual option flags. See section
Benjamin Petersond23f8222009-04-05 19:13:16 +00001038 :ref:`doctest-options`. See function :func:`set_unittest_reportflags` below
1039 for a better way to set reporting options.
Georg Brandl116aa622007-08-15 14:28:22 +00001040
Benjamin Petersond23f8222009-04-05 19:13:16 +00001041 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass)
1042 that should be used to extract tests from the files. It defaults to a normal
1043 parser (i.e., ``DocTestParser()``).
Georg Brandl116aa622007-08-15 14:28:22 +00001044
1045 Optional argument *encoding* specifies an encoding that should be used to
1046 convert the file to unicode.
1047
Georg Brandl55ac8f02007-09-01 13:51:09 +00001048 The global ``__file__`` is added to the globals provided to doctests loaded
1049 from a text file using :func:`DocFileSuite`.
Georg Brandl116aa622007-08-15 14:28:22 +00001050
1051
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001052.. function:: DocTestSuite(module=None, globs=None, extraglobs=None, test_finder=None, setUp=None, tearDown=None, checker=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001053
1054 Convert doctest tests for a module to a :class:`unittest.TestSuite`.
1055
1056 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
1057 and runs each doctest in the module. If any of the doctests fail, then the
1058 synthesized unit test fails, and a :exc:`failureException` exception is raised
1059 showing the name of the file containing the test and a (sometimes approximate)
1060 line number.
1061
1062 Optional argument *module* provides the module to be tested. It can be a module
1063 object or a (possibly dotted) module name. If not specified, the module calling
1064 this function is used.
1065
1066 Optional argument *globs* is a dictionary containing the initial global
1067 variables for the tests. A new copy of this dictionary is created for each
1068 test. By default, *globs* is a new empty dictionary.
1069
1070 Optional argument *extraglobs* specifies an extra set of global variables, which
1071 is merged into *globs*. By default, no extra globals are used.
1072
1073 Optional argument *test_finder* is the :class:`DocTestFinder` object (or a
1074 drop-in replacement) that is used to extract doctests from the module.
1075
1076 Optional arguments *setUp*, *tearDown*, and *optionflags* are the same as for
1077 function :func:`DocFileSuite` above.
1078
Georg Brandl55ac8f02007-09-01 13:51:09 +00001079 This function uses the same search technique as :func:`testmod`.
Georg Brandl116aa622007-08-15 14:28:22 +00001080
R David Murray1976d9b2014-04-14 20:28:36 -04001081 .. versionchanged:: 3.5
1082 :func:`DocTestSuite` returns an empty :class:`unittest.TestSuite` if *module*
1083 contains no docstrings instead of raising :exc:`ValueError`.
R David Murray5abd76a2012-09-10 10:15:58 -04001084
Georg Brandl116aa622007-08-15 14:28:22 +00001085
1086Under the covers, :func:`DocTestSuite` creates a :class:`unittest.TestSuite` out
1087of :class:`doctest.DocTestCase` instances, and :class:`DocTestCase` is a
1088subclass of :class:`unittest.TestCase`. :class:`DocTestCase` isn't documented
1089here (it's an internal detail), but studying its code can answer questions about
1090the exact details of :mod:`unittest` integration.
1091
1092Similarly, :func:`DocFileSuite` creates a :class:`unittest.TestSuite` out of
1093:class:`doctest.DocFileCase` instances, and :class:`DocFileCase` is a subclass
1094of :class:`DocTestCase`.
1095
1096So both ways of creating a :class:`unittest.TestSuite` run instances of
1097:class:`DocTestCase`. This is important for a subtle reason: when you run
1098:mod:`doctest` functions yourself, you can control the :mod:`doctest` options in
1099use directly, by passing option flags to :mod:`doctest` functions. However, if
1100you're writing a :mod:`unittest` framework, :mod:`unittest` ultimately controls
1101when and how tests get run. The framework author typically wants to control
1102:mod:`doctest` reporting options (perhaps, e.g., specified by command line
1103options), but there's no way to pass options through :mod:`unittest` to
1104:mod:`doctest` test runners.
1105
1106For this reason, :mod:`doctest` also supports a notion of :mod:`doctest`
1107reporting flags specific to :mod:`unittest` support, via this function:
1108
1109
1110.. function:: set_unittest_reportflags(flags)
1111
1112 Set the :mod:`doctest` reporting flags to use.
1113
Mariatta Wijaya81b89772017-02-06 20:15:01 -08001114 Argument *flags* takes the :ref:`bitwise OR <bitwise>` of option flags. See
1115 section :ref:`doctest-options`. Only "reporting flags" can be used.
Georg Brandl116aa622007-08-15 14:28:22 +00001116
1117 This is a module-global setting, and affects all future doctests run by module
1118 :mod:`unittest`: the :meth:`runTest` method of :class:`DocTestCase` looks at
1119 the option flags specified for the test case when the :class:`DocTestCase`
1120 instance was constructed. If no reporting flags were specified (which is the
1121 typical and expected case), :mod:`doctest`'s :mod:`unittest` reporting flags are
Mariatta Wijaya81b89772017-02-06 20:15:01 -08001122 :ref:`bitwise ORed <bitwise>` into the option flags, and the option flags
1123 so augmented are passed to the :class:`DocTestRunner` instance created to
1124 run the doctest. If any reporting flags were specified when the
1125 :class:`DocTestCase` instance was constructed, :mod:`doctest`'s
1126 :mod:`unittest` reporting flags are ignored.
Georg Brandl116aa622007-08-15 14:28:22 +00001127
1128 The value of the :mod:`unittest` reporting flags in effect before the function
1129 was called is returned by the function.
1130
Georg Brandl116aa622007-08-15 14:28:22 +00001131
1132.. _doctest-advanced-api:
1133
1134Advanced API
1135------------
1136
1137The basic API is a simple wrapper that's intended to make doctest easy to use.
1138It is fairly flexible, and should meet most users' needs; however, if you
1139require more fine-grained control over testing, or wish to extend doctest's
1140capabilities, then you should use the advanced API.
1141
1142The advanced API revolves around two container classes, which are used to store
1143the interactive examples extracted from doctest cases:
1144
Ezio Melotti0639d5a2009-12-19 23:26:38 +00001145* :class:`Example`: A single Python :term:`statement`, paired with its expected
Christian Heimesd8654cf2007-12-02 15:22:16 +00001146 output.
Georg Brandl116aa622007-08-15 14:28:22 +00001147
1148* :class:`DocTest`: A collection of :class:`Example`\ s, typically extracted
1149 from a single docstring or text file.
1150
1151Additional processing classes are defined to find, parse, and run, and check
1152doctest examples:
1153
1154* :class:`DocTestFinder`: Finds all docstrings in a given module, and uses a
1155 :class:`DocTestParser` to create a :class:`DocTest` from every docstring that
1156 contains interactive examples.
1157
1158* :class:`DocTestParser`: Creates a :class:`DocTest` object from a string (such
1159 as an object's docstring).
1160
1161* :class:`DocTestRunner`: Executes the examples in a :class:`DocTest`, and uses
1162 an :class:`OutputChecker` to verify their output.
1163
1164* :class:`OutputChecker`: Compares the actual output from a doctest example with
1165 the expected output, and decides whether they match.
1166
1167The relationships among these processing classes are summarized in the following
1168diagram::
1169
1170 list of:
1171 +------+ +---------+
1172 |module| --DocTestFinder-> | DocTest | --DocTestRunner-> results
1173 +------+ | ^ +---------+ | ^ (printed)
1174 | | | Example | | |
1175 v | | ... | v |
1176 DocTestParser | Example | OutputChecker
1177 +---------+
1178
1179
1180.. _doctest-doctest:
1181
1182DocTest Objects
1183^^^^^^^^^^^^^^^
1184
1185
1186.. class:: DocTest(examples, globs, name, filename, lineno, docstring)
1187
1188 A collection of doctest examples that should be run in a single namespace. The
Senthil Kumarana6bac952011-07-04 11:28:30 -07001189 constructor arguments are used to initialize the attributes of the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001190
Georg Brandl116aa622007-08-15 14:28:22 +00001191
Senthil Kumarana6bac952011-07-04 11:28:30 -07001192 :class:`DocTest` defines the following attributes. They are initialized by
Benjamin Petersone41251e2008-04-25 01:59:09 +00001193 the constructor, and should not be modified directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001194
1195
Benjamin Petersone41251e2008-04-25 01:59:09 +00001196 .. attribute:: examples
Georg Brandl116aa622007-08-15 14:28:22 +00001197
Benjamin Petersone41251e2008-04-25 01:59:09 +00001198 A list of :class:`Example` objects encoding the individual interactive Python
1199 examples that should be run by this test.
Georg Brandl116aa622007-08-15 14:28:22 +00001200
1201
Benjamin Petersone41251e2008-04-25 01:59:09 +00001202 .. attribute:: globs
Georg Brandl116aa622007-08-15 14:28:22 +00001203
Benjamin Petersone41251e2008-04-25 01:59:09 +00001204 The namespace (aka globals) that the examples should be run in. This is a
1205 dictionary mapping names to values. Any changes to the namespace made by the
1206 examples (such as binding new variables) will be reflected in :attr:`globs`
1207 after the test is run.
Georg Brandl116aa622007-08-15 14:28:22 +00001208
1209
Benjamin Petersone41251e2008-04-25 01:59:09 +00001210 .. attribute:: name
Georg Brandl116aa622007-08-15 14:28:22 +00001211
Benjamin Petersone41251e2008-04-25 01:59:09 +00001212 A string name identifying the :class:`DocTest`. Typically, this is the name
1213 of the object or file that the test was extracted from.
Georg Brandl116aa622007-08-15 14:28:22 +00001214
1215
Benjamin Petersone41251e2008-04-25 01:59:09 +00001216 .. attribute:: filename
Georg Brandl116aa622007-08-15 14:28:22 +00001217
Benjamin Petersone41251e2008-04-25 01:59:09 +00001218 The name of the file that this :class:`DocTest` was extracted from; or
1219 ``None`` if the filename is unknown, or if the :class:`DocTest` was not
1220 extracted from a file.
Georg Brandl116aa622007-08-15 14:28:22 +00001221
1222
Benjamin Petersone41251e2008-04-25 01:59:09 +00001223 .. attribute:: lineno
Georg Brandl116aa622007-08-15 14:28:22 +00001224
Benjamin Petersone41251e2008-04-25 01:59:09 +00001225 The line number within :attr:`filename` where this :class:`DocTest` begins, or
1226 ``None`` if the line number is unavailable. This line number is zero-based
1227 with respect to the beginning of the file.
Georg Brandl116aa622007-08-15 14:28:22 +00001228
1229
Benjamin Petersone41251e2008-04-25 01:59:09 +00001230 .. attribute:: docstring
Georg Brandl116aa622007-08-15 14:28:22 +00001231
Serhiy Storchakaecf41da2016-10-19 16:29:26 +03001232 The string that the test was extracted from, or ``None`` if the string is
Benjamin Petersone41251e2008-04-25 01:59:09 +00001233 unavailable, or if the test was not extracted from a string.
Georg Brandl116aa622007-08-15 14:28:22 +00001234
1235
1236.. _doctest-example:
1237
1238Example Objects
1239^^^^^^^^^^^^^^^
1240
1241
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001242.. class:: Example(source, want, exc_msg=None, lineno=0, indent=0, options=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001243
1244 A single interactive example, consisting of a Python statement and its expected
Senthil Kumarana6bac952011-07-04 11:28:30 -07001245 output. The constructor arguments are used to initialize the attributes of
1246 the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001247
Georg Brandl116aa622007-08-15 14:28:22 +00001248
Senthil Kumarana6bac952011-07-04 11:28:30 -07001249 :class:`Example` defines the following attributes. They are initialized by
Benjamin Petersone41251e2008-04-25 01:59:09 +00001250 the constructor, and should not be modified directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001251
1252
Benjamin Petersone41251e2008-04-25 01:59:09 +00001253 .. attribute:: source
Georg Brandl116aa622007-08-15 14:28:22 +00001254
Benjamin Petersone41251e2008-04-25 01:59:09 +00001255 A string containing the example's source code. This source code consists of a
1256 single Python statement, and always ends with a newline; the constructor adds
1257 a newline when necessary.
Georg Brandl116aa622007-08-15 14:28:22 +00001258
1259
Benjamin Petersone41251e2008-04-25 01:59:09 +00001260 .. attribute:: want
Georg Brandl116aa622007-08-15 14:28:22 +00001261
Benjamin Petersone41251e2008-04-25 01:59:09 +00001262 The expected output from running the example's source code (either from
1263 stdout, or a traceback in case of exception). :attr:`want` ends with a
1264 newline unless no output is expected, in which case it's an empty string. The
1265 constructor adds a newline when necessary.
Georg Brandl116aa622007-08-15 14:28:22 +00001266
1267
Benjamin Petersone41251e2008-04-25 01:59:09 +00001268 .. attribute:: exc_msg
Georg Brandl116aa622007-08-15 14:28:22 +00001269
Benjamin Petersone41251e2008-04-25 01:59:09 +00001270 The exception message generated by the example, if the example is expected to
1271 generate an exception; or ``None`` if it is not expected to generate an
1272 exception. This exception message is compared against the return value of
1273 :func:`traceback.format_exception_only`. :attr:`exc_msg` ends with a newline
1274 unless it's ``None``. The constructor adds a newline if needed.
Georg Brandl116aa622007-08-15 14:28:22 +00001275
1276
Benjamin Petersone41251e2008-04-25 01:59:09 +00001277 .. attribute:: lineno
Georg Brandl116aa622007-08-15 14:28:22 +00001278
Benjamin Petersone41251e2008-04-25 01:59:09 +00001279 The line number within the string containing this example where the example
1280 begins. This line number is zero-based with respect to the beginning of the
1281 containing string.
Georg Brandl116aa622007-08-15 14:28:22 +00001282
1283
Benjamin Petersone41251e2008-04-25 01:59:09 +00001284 .. attribute:: indent
Georg Brandl116aa622007-08-15 14:28:22 +00001285
Benjamin Petersone41251e2008-04-25 01:59:09 +00001286 The example's indentation in the containing string, i.e., the number of space
1287 characters that precede the example's first prompt.
Georg Brandl116aa622007-08-15 14:28:22 +00001288
1289
Benjamin Petersone41251e2008-04-25 01:59:09 +00001290 .. attribute:: options
Georg Brandl116aa622007-08-15 14:28:22 +00001291
Benjamin Petersone41251e2008-04-25 01:59:09 +00001292 A dictionary mapping from option flags to ``True`` or ``False``, which is used
1293 to override default options for this example. Any option flags not contained
1294 in this dictionary are left at their default value (as specified by the
1295 :class:`DocTestRunner`'s :attr:`optionflags`). By default, no options are set.
Georg Brandl116aa622007-08-15 14:28:22 +00001296
1297
1298.. _doctest-doctestfinder:
1299
1300DocTestFinder objects
1301^^^^^^^^^^^^^^^^^^^^^
1302
1303
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001304.. class:: DocTestFinder(verbose=False, parser=DocTestParser(), recurse=True, exclude_empty=True)
Georg Brandl116aa622007-08-15 14:28:22 +00001305
1306 A processing class used to extract the :class:`DocTest`\ s that are relevant to
1307 a given object, from its docstring and the docstrings of its contained objects.
Zachary Warea4b7a752013-11-24 01:19:09 -06001308 :class:`DocTest`\ s can be extracted from modules, classes, functions,
1309 methods, staticmethods, classmethods, and properties.
Georg Brandl116aa622007-08-15 14:28:22 +00001310
1311 The optional argument *verbose* can be used to display the objects searched by
1312 the finder. It defaults to ``False`` (no output).
1313
1314 The optional argument *parser* specifies the :class:`DocTestParser` object (or a
1315 drop-in replacement) that is used to extract doctests from docstrings.
1316
1317 If the optional argument *recurse* is false, then :meth:`DocTestFinder.find`
1318 will only examine the given object, and not any contained objects.
1319
1320 If the optional argument *exclude_empty* is false, then
1321 :meth:`DocTestFinder.find` will include tests for objects with empty docstrings.
1322
Georg Brandl116aa622007-08-15 14:28:22 +00001323
Benjamin Petersone41251e2008-04-25 01:59:09 +00001324 :class:`DocTestFinder` defines the following method:
Georg Brandl116aa622007-08-15 14:28:22 +00001325
1326
Benjamin Petersone41251e2008-04-25 01:59:09 +00001327 .. method:: find(obj[, name][, module][, globs][, extraglobs])
Georg Brandl116aa622007-08-15 14:28:22 +00001328
Benjamin Petersone41251e2008-04-25 01:59:09 +00001329 Return a list of the :class:`DocTest`\ s that are defined by *obj*'s
1330 docstring, or by any of its contained objects' docstrings.
Georg Brandl116aa622007-08-15 14:28:22 +00001331
Benjamin Petersone41251e2008-04-25 01:59:09 +00001332 The optional argument *name* specifies the object's name; this name will be
1333 used to construct names for the returned :class:`DocTest`\ s. If *name* is
1334 not specified, then ``obj.__name__`` is used.
Georg Brandl116aa622007-08-15 14:28:22 +00001335
Benjamin Petersone41251e2008-04-25 01:59:09 +00001336 The optional parameter *module* is the module that contains the given object.
Serhiy Storchakaecf41da2016-10-19 16:29:26 +03001337 If the module is not specified or is ``None``, then the test finder will attempt
Benjamin Petersone41251e2008-04-25 01:59:09 +00001338 to automatically determine the correct module. The object's module is used:
Georg Brandl116aa622007-08-15 14:28:22 +00001339
Benjamin Petersone41251e2008-04-25 01:59:09 +00001340 * As a default namespace, if *globs* is not specified.
Georg Brandl116aa622007-08-15 14:28:22 +00001341
Benjamin Petersone41251e2008-04-25 01:59:09 +00001342 * To prevent the DocTestFinder from extracting DocTests from objects that are
1343 imported from other modules. (Contained objects with modules other than
1344 *module* are ignored.)
Georg Brandl116aa622007-08-15 14:28:22 +00001345
Benjamin Petersone41251e2008-04-25 01:59:09 +00001346 * To find the name of the file containing the object.
Georg Brandl116aa622007-08-15 14:28:22 +00001347
Benjamin Petersone41251e2008-04-25 01:59:09 +00001348 * To help find the line number of the object within its file.
Georg Brandl116aa622007-08-15 14:28:22 +00001349
Benjamin Petersone41251e2008-04-25 01:59:09 +00001350 If *module* is ``False``, no attempt to find the module will be made. This is
1351 obscure, of use mostly in testing doctest itself: if *module* is ``False``, or
1352 is ``None`` but cannot be found automatically, then all objects are considered
1353 to belong to the (non-existent) module, so all contained objects will
1354 (recursively) be searched for doctests.
Georg Brandl116aa622007-08-15 14:28:22 +00001355
Benjamin Petersone41251e2008-04-25 01:59:09 +00001356 The globals for each :class:`DocTest` is formed by combining *globs* and
1357 *extraglobs* (bindings in *extraglobs* override bindings in *globs*). A new
1358 shallow copy of the globals dictionary is created for each :class:`DocTest`.
1359 If *globs* is not specified, then it defaults to the module's *__dict__*, if
1360 specified, or ``{}`` otherwise. If *extraglobs* is not specified, then it
1361 defaults to ``{}``.
Georg Brandl116aa622007-08-15 14:28:22 +00001362
1363
1364.. _doctest-doctestparser:
1365
1366DocTestParser objects
1367^^^^^^^^^^^^^^^^^^^^^
1368
1369
1370.. class:: DocTestParser()
1371
1372 A processing class used to extract interactive examples from a string, and use
1373 them to create a :class:`DocTest` object.
1374
Georg Brandl116aa622007-08-15 14:28:22 +00001375
Benjamin Petersone41251e2008-04-25 01:59:09 +00001376 :class:`DocTestParser` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001377
1378
Benjamin Petersone41251e2008-04-25 01:59:09 +00001379 .. method:: get_doctest(string, globs, name, filename, lineno)
Georg Brandl116aa622007-08-15 14:28:22 +00001380
Benjamin Petersone41251e2008-04-25 01:59:09 +00001381 Extract all doctest examples from the given string, and collect them into a
1382 :class:`DocTest` object.
Georg Brandl116aa622007-08-15 14:28:22 +00001383
Benjamin Petersone41251e2008-04-25 01:59:09 +00001384 *globs*, *name*, *filename*, and *lineno* are attributes for the new
1385 :class:`DocTest` object. See the documentation for :class:`DocTest` for more
1386 information.
Georg Brandl116aa622007-08-15 14:28:22 +00001387
1388
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001389 .. method:: get_examples(string, name='<string>')
Georg Brandl116aa622007-08-15 14:28:22 +00001390
Benjamin Petersone41251e2008-04-25 01:59:09 +00001391 Extract all doctest examples from the given string, and return them as a list
1392 of :class:`Example` objects. Line numbers are 0-based. The optional argument
1393 *name* is a name identifying this string, and is only used for error messages.
Georg Brandl116aa622007-08-15 14:28:22 +00001394
1395
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001396 .. method:: parse(string, name='<string>')
Georg Brandl116aa622007-08-15 14:28:22 +00001397
Benjamin Petersone41251e2008-04-25 01:59:09 +00001398 Divide the given string into examples and intervening text, and return them as
1399 a list of alternating :class:`Example`\ s and strings. Line numbers for the
1400 :class:`Example`\ s are 0-based. The optional argument *name* is a name
1401 identifying this string, and is only used for error messages.
Georg Brandl116aa622007-08-15 14:28:22 +00001402
1403
1404.. _doctest-doctestrunner:
1405
1406DocTestRunner objects
1407^^^^^^^^^^^^^^^^^^^^^
1408
1409
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001410.. class:: DocTestRunner(checker=None, verbose=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +00001411
1412 A processing class used to execute and verify the interactive examples in a
1413 :class:`DocTest`.
1414
1415 The comparison between expected outputs and actual outputs is done by an
1416 :class:`OutputChecker`. This comparison may be customized with a number of
1417 option flags; see section :ref:`doctest-options` for more information. If the
1418 option flags are insufficient, then the comparison may also be customized by
1419 passing a subclass of :class:`OutputChecker` to the constructor.
1420
1421 The test runner's display output can be controlled in two ways. First, an output
1422 function can be passed to :meth:`TestRunner.run`; this function will be called
1423 with strings that should be displayed. It defaults to ``sys.stdout.write``. If
1424 capturing the output is not sufficient, then the display output can be also
1425 customized by subclassing DocTestRunner, and overriding the methods
1426 :meth:`report_start`, :meth:`report_success`,
1427 :meth:`report_unexpected_exception`, and :meth:`report_failure`.
1428
1429 The optional keyword argument *checker* specifies the :class:`OutputChecker`
1430 object (or drop-in replacement) that should be used to compare the expected
1431 outputs to the actual outputs of doctest examples.
1432
1433 The optional keyword argument *verbose* controls the :class:`DocTestRunner`'s
1434 verbosity. If *verbose* is ``True``, then information is printed about each
1435 example, as it is run. If *verbose* is ``False``, then only failures are
1436 printed. If *verbose* is unspecified, or ``None``, then verbose output is used
Éric Araujo713d3032010-11-18 16:38:46 +00001437 iff the command-line switch ``-v`` is used.
Georg Brandl116aa622007-08-15 14:28:22 +00001438
1439 The optional keyword argument *optionflags* can be used to control how the test
1440 runner compares expected output to actual output, and how it displays failures.
1441 For more information, see section :ref:`doctest-options`.
1442
Georg Brandl116aa622007-08-15 14:28:22 +00001443
Benjamin Petersone41251e2008-04-25 01:59:09 +00001444 :class:`DocTestParser` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001445
1446
Benjamin Petersone41251e2008-04-25 01:59:09 +00001447 .. method:: report_start(out, test, example)
Georg Brandl116aa622007-08-15 14:28:22 +00001448
Benjamin Petersone41251e2008-04-25 01:59:09 +00001449 Report that the test runner is about to process the given example. This method
1450 is provided to allow subclasses of :class:`DocTestRunner` to customize their
1451 output; it should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001452
Benjamin Petersone41251e2008-04-25 01:59:09 +00001453 *example* is the example about to be processed. *test* is the test
1454 *containing example*. *out* is the output function that was passed to
1455 :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001456
1457
Benjamin Petersone41251e2008-04-25 01:59:09 +00001458 .. method:: report_success(out, test, example, got)
Georg Brandl116aa622007-08-15 14:28:22 +00001459
Benjamin Petersone41251e2008-04-25 01:59:09 +00001460 Report that the given example ran successfully. This method is provided to
1461 allow subclasses of :class:`DocTestRunner` to customize their output; it
1462 should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001463
Benjamin Petersone41251e2008-04-25 01:59:09 +00001464 *example* is the example about to be processed. *got* is the actual output
1465 from the example. *test* is the test containing *example*. *out* is the
1466 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001467
1468
Benjamin Petersone41251e2008-04-25 01:59:09 +00001469 .. method:: report_failure(out, test, example, got)
Georg Brandl116aa622007-08-15 14:28:22 +00001470
Benjamin Petersone41251e2008-04-25 01:59:09 +00001471 Report that the given example failed. This method is provided to allow
1472 subclasses of :class:`DocTestRunner` to customize their output; it should not
1473 be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001474
Benjamin Petersone41251e2008-04-25 01:59:09 +00001475 *example* is the example about to be processed. *got* is the actual output
1476 from the example. *test* is the test containing *example*. *out* is the
1477 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001478
1479
Benjamin Petersone41251e2008-04-25 01:59:09 +00001480 .. method:: report_unexpected_exception(out, test, example, exc_info)
Georg Brandl116aa622007-08-15 14:28:22 +00001481
Benjamin Petersone41251e2008-04-25 01:59:09 +00001482 Report that the given example raised an unexpected exception. This method is
1483 provided to allow subclasses of :class:`DocTestRunner` to customize their
1484 output; it should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001485
Benjamin Petersone41251e2008-04-25 01:59:09 +00001486 *example* is the example about to be processed. *exc_info* is a tuple
1487 containing information about the unexpected exception (as returned by
1488 :func:`sys.exc_info`). *test* is the test containing *example*. *out* is the
1489 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001490
1491
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001492 .. method:: run(test, compileflags=None, out=None, clear_globs=True)
Georg Brandl116aa622007-08-15 14:28:22 +00001493
Benjamin Petersone41251e2008-04-25 01:59:09 +00001494 Run the examples in *test* (a :class:`DocTest` object), and display the
1495 results using the writer function *out*.
Georg Brandl116aa622007-08-15 14:28:22 +00001496
Benjamin Petersone41251e2008-04-25 01:59:09 +00001497 The examples are run in the namespace ``test.globs``. If *clear_globs* is
1498 true (the default), then this namespace will be cleared after the test runs,
1499 to help with garbage collection. If you would like to examine the namespace
1500 after the test completes, then use *clear_globs=False*.
Georg Brandl116aa622007-08-15 14:28:22 +00001501
Benjamin Petersone41251e2008-04-25 01:59:09 +00001502 *compileflags* gives the set of flags that should be used by the Python
1503 compiler when running the examples. If not specified, then it will default to
1504 the set of future-import flags that apply to *globs*.
Georg Brandl116aa622007-08-15 14:28:22 +00001505
Benjamin Petersone41251e2008-04-25 01:59:09 +00001506 The output of each example is checked using the :class:`DocTestRunner`'s
1507 output checker, and the results are formatted by the
1508 :meth:`DocTestRunner.report_\*` methods.
Georg Brandl116aa622007-08-15 14:28:22 +00001509
1510
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001511 .. method:: summarize(verbose=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001512
Benjamin Petersone41251e2008-04-25 01:59:09 +00001513 Print a summary of all the test cases that have been run by this DocTestRunner,
1514 and return a :term:`named tuple` ``TestResults(failed, attempted)``.
Georg Brandl116aa622007-08-15 14:28:22 +00001515
Benjamin Petersone41251e2008-04-25 01:59:09 +00001516 The optional *verbose* argument controls how detailed the summary is. If the
1517 verbosity is not specified, then the :class:`DocTestRunner`'s verbosity is
1518 used.
Georg Brandl116aa622007-08-15 14:28:22 +00001519
1520.. _doctest-outputchecker:
1521
1522OutputChecker objects
1523^^^^^^^^^^^^^^^^^^^^^
1524
1525
1526.. class:: OutputChecker()
1527
1528 A class used to check the whether the actual output from a doctest example
1529 matches the expected output. :class:`OutputChecker` defines two methods:
1530 :meth:`check_output`, which compares a given pair of outputs, and returns true
1531 if they match; and :meth:`output_difference`, which returns a string describing
1532 the differences between two outputs.
1533
Georg Brandl116aa622007-08-15 14:28:22 +00001534
Benjamin Petersone41251e2008-04-25 01:59:09 +00001535 :class:`OutputChecker` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001536
Benjamin Petersone41251e2008-04-25 01:59:09 +00001537 .. method:: check_output(want, got, optionflags)
Georg Brandl116aa622007-08-15 14:28:22 +00001538
Benjamin Petersone41251e2008-04-25 01:59:09 +00001539 Return ``True`` iff the actual output from an example (*got*) matches the
1540 expected output (*want*). These strings are always considered to match if
1541 they are identical; but depending on what option flags the test runner is
1542 using, several non-exact match types are also possible. See section
1543 :ref:`doctest-options` for more information about option flags.
Georg Brandl116aa622007-08-15 14:28:22 +00001544
1545
Benjamin Petersone41251e2008-04-25 01:59:09 +00001546 .. method:: output_difference(example, got, optionflags)
Georg Brandl116aa622007-08-15 14:28:22 +00001547
Benjamin Petersone41251e2008-04-25 01:59:09 +00001548 Return a string describing the differences between the expected output for a
1549 given example (*example*) and the actual output (*got*). *optionflags* is the
1550 set of option flags used to compare *want* and *got*.
Georg Brandl116aa622007-08-15 14:28:22 +00001551
1552
1553.. _doctest-debugging:
1554
1555Debugging
1556---------
1557
1558Doctest provides several mechanisms for debugging doctest examples:
1559
1560* Several functions convert doctests to executable Python programs, which can be
1561 run under the Python debugger, :mod:`pdb`.
1562
1563* The :class:`DebugRunner` class is a subclass of :class:`DocTestRunner` that
1564 raises an exception for the first failing example, containing information about
1565 that example. This information can be used to perform post-mortem debugging on
1566 the example.
1567
1568* The :mod:`unittest` cases generated by :func:`DocTestSuite` support the
1569 :meth:`debug` method defined by :class:`unittest.TestCase`.
1570
1571* You can add a call to :func:`pdb.set_trace` in a doctest example, and you'll
1572 drop into the Python debugger when that line is executed. Then you can inspect
1573 current values of variables, and so on. For example, suppose :file:`a.py`
1574 contains just this module docstring::
1575
1576 """
1577 >>> def f(x):
1578 ... g(x*2)
1579 >>> def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001580 ... print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001581 ... import pdb; pdb.set_trace()
1582 >>> f(3)
1583 9
1584 """
1585
1586 Then an interactive Python session may look like this::
1587
1588 >>> import a, doctest
1589 >>> doctest.testmod(a)
1590 --Return--
1591 > <doctest a[1]>(3)g()->None
1592 -> import pdb; pdb.set_trace()
1593 (Pdb) list
1594 1 def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001595 2 print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001596 3 -> import pdb; pdb.set_trace()
1597 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001598 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001599 6
1600 (Pdb) step
1601 --Return--
1602 > <doctest a[0]>(2)f()->None
1603 -> g(x*2)
1604 (Pdb) list
1605 1 def f(x):
1606 2 -> g(x*2)
1607 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001608 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001609 3
1610 (Pdb) step
1611 --Return--
1612 > <doctest a[2]>(1)?()->None
1613 -> f(3)
1614 (Pdb) cont
1615 (0, 3)
1616 >>>
1617
Georg Brandl116aa622007-08-15 14:28:22 +00001618
1619Functions that convert doctests to Python code, and possibly run the synthesized
1620code under the debugger:
1621
1622
1623.. function:: script_from_examples(s)
1624
1625 Convert text with examples to a script.
1626
1627 Argument *s* is a string containing doctest examples. The string is converted
1628 to a Python script, where doctest examples in *s* are converted to regular code,
1629 and everything else is converted to Python comments. The generated script is
1630 returned as a string. For example, ::
1631
1632 import doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001633 print(doctest.script_from_examples(r"""
Georg Brandl116aa622007-08-15 14:28:22 +00001634 Set x and y to 1 and 2.
1635 >>> x, y = 1, 2
1636
1637 Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001638 >>> print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001639 3
Georg Brandl6911e3c2007-09-04 07:15:32 +00001640 """))
Georg Brandl116aa622007-08-15 14:28:22 +00001641
1642 displays::
1643
1644 # Set x and y to 1 and 2.
1645 x, y = 1, 2
1646 #
1647 # Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001648 print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001649 # Expected:
1650 ## 3
1651
1652 This function is used internally by other functions (see below), but can also be
1653 useful when you want to transform an interactive Python session into a Python
1654 script.
1655
Georg Brandl116aa622007-08-15 14:28:22 +00001656
1657.. function:: testsource(module, name)
1658
1659 Convert the doctest for an object to a script.
1660
1661 Argument *module* is a module object, or dotted name of a module, containing the
1662 object whose doctests are of interest. Argument *name* is the name (within the
1663 module) of the object with the doctests of interest. The result is a string,
1664 containing the object's docstring converted to a Python script, as described for
1665 :func:`script_from_examples` above. For example, if module :file:`a.py`
1666 contains a top-level function :func:`f`, then ::
1667
1668 import a, doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001669 print(doctest.testsource(a, "a.f"))
Georg Brandl116aa622007-08-15 14:28:22 +00001670
1671 prints a script version of function :func:`f`'s docstring, with doctests
1672 converted to code, and the rest placed in comments.
1673
Georg Brandl116aa622007-08-15 14:28:22 +00001674
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001675.. function:: debug(module, name, pm=False)
Georg Brandl116aa622007-08-15 14:28:22 +00001676
1677 Debug the doctests for an object.
1678
1679 The *module* and *name* arguments are the same as for function
1680 :func:`testsource` above. The synthesized Python script for the named object's
1681 docstring is written to a temporary file, and then that file is run under the
1682 control of the Python debugger, :mod:`pdb`.
1683
1684 A shallow copy of ``module.__dict__`` is used for both local and global
1685 execution context.
1686
1687 Optional argument *pm* controls whether post-mortem debugging is used. If *pm*
1688 has a true value, the script file is run directly, and the debugger gets
1689 involved only if the script terminates via raising an unhandled exception. If
1690 it does, then post-mortem debugging is invoked, via :func:`pdb.post_mortem`,
1691 passing the traceback object from the unhandled exception. If *pm* is not
1692 specified, or is false, the script is run under the debugger from the start, via
1693 passing an appropriate :func:`exec` call to :func:`pdb.run`.
1694
Georg Brandl116aa622007-08-15 14:28:22 +00001695
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001696.. function:: debug_src(src, pm=False, globs=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001697
1698 Debug the doctests in a string.
1699
1700 This is like function :func:`debug` above, except that a string containing
1701 doctest examples is specified directly, via the *src* argument.
1702
1703 Optional argument *pm* has the same meaning as in function :func:`debug` above.
1704
1705 Optional argument *globs* gives a dictionary to use as both local and global
1706 execution context. If not specified, or ``None``, an empty dictionary is used.
1707 If specified, a shallow copy of the dictionary is used.
1708
Georg Brandl116aa622007-08-15 14:28:22 +00001709
1710The :class:`DebugRunner` class, and the special exceptions it may raise, are of
1711most interest to testing framework authors, and will only be sketched here. See
1712the source code, and especially :class:`DebugRunner`'s docstring (which is a
1713doctest!) for more details:
1714
1715
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001716.. class:: DebugRunner(checker=None, verbose=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +00001717
1718 A subclass of :class:`DocTestRunner` that raises an exception as soon as a
1719 failure is encountered. If an unexpected exception occurs, an
1720 :exc:`UnexpectedException` exception is raised, containing the test, the
1721 example, and the original exception. If the output doesn't match, then a
1722 :exc:`DocTestFailure` exception is raised, containing the test, the example, and
1723 the actual output.
1724
1725 For information about the constructor parameters and methods, see the
1726 documentation for :class:`DocTestRunner` in section :ref:`doctest-advanced-api`.
1727
1728There are two exceptions that may be raised by :class:`DebugRunner` instances:
1729
1730
1731.. exception:: DocTestFailure(test, example, got)
1732
Georg Brandl7cb13192010-08-03 12:06:29 +00001733 An exception raised by :class:`DocTestRunner` to signal that a doctest example's
Georg Brandl116aa622007-08-15 14:28:22 +00001734 actual output did not match its expected output. The constructor arguments are
Senthil Kumarana6bac952011-07-04 11:28:30 -07001735 used to initialize the attributes of the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001736
Senthil Kumarana6bac952011-07-04 11:28:30 -07001737:exc:`DocTestFailure` defines the following attributes:
Georg Brandl116aa622007-08-15 14:28:22 +00001738
1739
1740.. attribute:: DocTestFailure.test
1741
1742 The :class:`DocTest` object that was being run when the example failed.
1743
1744
1745.. attribute:: DocTestFailure.example
1746
1747 The :class:`Example` that failed.
1748
1749
1750.. attribute:: DocTestFailure.got
1751
1752 The example's actual output.
1753
1754
1755.. exception:: UnexpectedException(test, example, exc_info)
1756
Georg Brandl7cb13192010-08-03 12:06:29 +00001757 An exception raised by :class:`DocTestRunner` to signal that a doctest
1758 example raised an unexpected exception. The constructor arguments are used
Senthil Kumarana6bac952011-07-04 11:28:30 -07001759 to initialize the attributes of the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001760
Senthil Kumarana6bac952011-07-04 11:28:30 -07001761:exc:`UnexpectedException` defines the following attributes:
Georg Brandl116aa622007-08-15 14:28:22 +00001762
1763
1764.. attribute:: UnexpectedException.test
1765
1766 The :class:`DocTest` object that was being run when the example failed.
1767
1768
1769.. attribute:: UnexpectedException.example
1770
1771 The :class:`Example` that failed.
1772
1773
1774.. attribute:: UnexpectedException.exc_info
1775
1776 A tuple containing information about the unexpected exception, as returned by
1777 :func:`sys.exc_info`.
1778
1779
1780.. _doctest-soapbox:
1781
1782Soapbox
1783-------
1784
1785As mentioned in the introduction, :mod:`doctest` has grown to have three primary
1786uses:
1787
1788#. Checking examples in docstrings.
1789
1790#. Regression testing.
1791
1792#. Executable documentation / literate testing.
1793
1794These uses have different requirements, and it is important to distinguish them.
1795In particular, filling your docstrings with obscure test cases makes for bad
1796documentation.
1797
1798When writing a docstring, choose docstring examples with care. There's an art to
1799this that needs to be learned---it may not be natural at first. Examples should
1800add genuine value to the documentation. A good example can often be worth many
1801words. If done with care, the examples will be invaluable for your users, and
1802will pay back the time it takes to collect them many times over as the years go
1803by and things change. I'm still amazed at how often one of my :mod:`doctest`
1804examples stops working after a "harmless" change.
1805
1806Doctest also makes an excellent tool for regression testing, especially if you
1807don't skimp on explanatory text. By interleaving prose and examples, it becomes
1808much easier to keep track of what's actually being tested, and why. When a test
1809fails, good prose can make it much easier to figure out what the problem is, and
1810how it should be fixed. It's true that you could write extensive comments in
1811code-based testing, but few programmers do. Many have found that using doctest
1812approaches instead leads to much clearer tests. Perhaps this is simply because
1813doctest makes writing prose a little easier than writing code, while writing
1814comments in code is a little harder. I think it goes deeper than just that:
1815the natural attitude when writing a doctest-based test is that you want to
1816explain the fine points of your software, and illustrate them with examples.
1817This in turn naturally leads to test files that start with the simplest
1818features, and logically progress to complications and edge cases. A coherent
1819narrative is the result, instead of a collection of isolated functions that test
1820isolated bits of functionality seemingly at random. It's a different attitude,
1821and produces different results, blurring the distinction between testing and
1822explaining.
1823
1824Regression testing is best confined to dedicated objects or files. There are
1825several options for organizing tests:
1826
1827* Write text files containing test cases as interactive examples, and test the
1828 files using :func:`testfile` or :func:`DocFileSuite`. This is recommended,
1829 although is easiest to do for new projects, designed from the start to use
1830 doctest.
1831
1832* Define functions named ``_regrtest_topic`` that consist of single docstrings,
1833 containing test cases for the named topics. These functions can be included in
1834 the same file as the module, or separated out into a separate test file.
1835
1836* Define a ``__test__`` dictionary mapping from regression test topics to
1837 docstrings containing test cases.
1838
Ethan Furman2a5f9da2015-09-17 22:20:41 -07001839When you have placed your tests in a module, the module can itself be the test
1840runner. When a test fails, you can arrange for your test runner to re-run only
1841the failing doctest while you debug the problem. Here is a minimal example of
1842such a test runner::
1843
1844 if __name__ == '__main__':
1845 import doctest
1846 flags = doctest.REPORT_NDIFF|doctest.FAIL_FAST
1847 if len(sys.argv) > 1:
1848 name = sys.argv[1]
1849 if name in globals():
1850 obj = globals()[name]
1851 else:
1852 obj = __test__[name]
1853 doctest.run_docstring_examples(obj, globals(), name=name,
1854 optionflags=flags)
1855 else:
1856 fail, total = doctest.testmod(optionflags=flags)
1857 print("{} failures out of {} tests".format(fail, total))
1858
1859
Georg Brandl116aa622007-08-15 14:28:22 +00001860.. rubric:: Footnotes
1861
1862.. [#] Examples containing both expected output and an exception are not supported.
1863 Trying to guess where one ends and the other begins is too error-prone, and that
1864 also makes for a confusing test.