blob: e7c0033eb6bc45af291d6bcb9802fae60adc7333 [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 Storchakaddb961d2018-10-26 09:00:49 +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 Storchaka913876d2018-10-28 13:41:26 +0200488.. index:: single: ^ (caret); marker
Serhiy Storchakaddb961d2018-10-26 09:00:49 +0300489
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 Storchakaddb961d2018-10-26 09:00:49 +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 Storchakaddb961d2018-10-26 09:00:49 +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 Storchakaddb961d2018-10-26 09:00:49 +0300697.. index::
Serhiy Storchaka913876d2018-10-28 13:41:26 +0200698 single: # (hash); in doctests
699 single: + (plus); in doctests
700 single: - (minus); 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
Stéphane Wirtel0522fd82018-10-20 10:43:32 +0200774guarantee about output. For example, when printing a set, Python doesn't
775guarantee that the element is printed in any particular order, so a test like ::
Georg Brandl116aa622007-08-15 14:28:22 +0000776
777 >>> foo()
Stéphane Wirtel0522fd82018-10-20 10:43:32 +0200778 {"Hermione", "Harry"}
Georg Brandl116aa622007-08-15 14:28:22 +0000779
780is vulnerable! One workaround is to do ::
781
Stéphane Wirtel0522fd82018-10-20 10:43:32 +0200782 >>> foo() == {"Hermione", "Harry"}
Georg Brandl116aa622007-08-15 14:28:22 +0000783 True
784
785instead. Another is to do ::
786
Stéphane Wirtel0522fd82018-10-20 10:43:32 +0200787 >>> d = sorted(foo())
Georg Brandl116aa622007-08-15 14:28:22 +0000788 >>> d
Stéphane Wirtel0522fd82018-10-20 10:43:32 +0200789 ['Harry', 'Hermione']
790
791.. note::
792
793 Before Python 3.6, when printing a dict, Python did not guarantee that
794 the key-value pairs was printed in any particular order.
Georg Brandl116aa622007-08-15 14:28:22 +0000795
796There are others, but you get the idea.
797
798Another bad idea is to print things that embed an object address, like ::
799
800 >>> id(1.0) # certain to fail some of the time
801 7948648
802 >>> class C: pass
803 >>> C() # the default repr() for instances embeds an address
804 <__main__.C instance at 0x00AC18F0>
805
Georg Brandl23a87de2012-10-10 16:56:15 +0200806The :const:`ELLIPSIS` directive gives a nice approach for the last example::
Georg Brandl116aa622007-08-15 14:28:22 +0000807
808 >>> C() #doctest: +ELLIPSIS
809 <__main__.C instance at 0x...>
810
811Floating-point numbers are also subject to small output variations across
812platforms, because Python defers to the platform C library for float formatting,
813and C libraries vary widely in quality here. ::
814
815 >>> 1./7 # risky
816 0.14285714285714285
Georg Brandl6911e3c2007-09-04 07:15:32 +0000817 >>> print(1./7) # safer
Georg Brandl116aa622007-08-15 14:28:22 +0000818 0.142857142857
Georg Brandl6911e3c2007-09-04 07:15:32 +0000819 >>> print(round(1./7, 6)) # much safer
Georg Brandl116aa622007-08-15 14:28:22 +0000820 0.142857
821
822Numbers of the form ``I/2.**J`` are safe across all platforms, and I often
823contrive doctest examples to produce numbers of that form::
824
825 >>> 3./4 # utterly safe
826 0.75
827
828Simple fractions are also easier for people to understand, and that makes for
829better documentation.
830
831
832.. _doctest-basic-api:
833
834Basic API
835---------
836
837The functions :func:`testmod` and :func:`testfile` provide a simple interface to
838doctest that should be sufficient for most basic uses. For a less formal
839introduction to these two functions, see sections :ref:`doctest-simple-testmod`
840and :ref:`doctest-simple-testfile`.
841
842
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000843.. 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 +0000844
845 All arguments except *filename* are optional, and should be specified in keyword
846 form.
847
848 Test examples in the file named *filename*. Return ``(failure_count,
849 test_count)``.
850
851 Optional argument *module_relative* specifies how the filename should be
852 interpreted:
853
854 * If *module_relative* is ``True`` (the default), then *filename* specifies an
855 OS-independent module-relative path. By default, this path is relative to the
856 calling module's directory; but if the *package* argument is specified, then it
857 is relative to that package. To ensure OS-independence, *filename* should use
858 ``/`` characters to separate path segments, and may not be an absolute path
859 (i.e., it may not begin with ``/``).
860
861 * If *module_relative* is ``False``, then *filename* specifies an OS-specific
862 path. The path may be absolute or relative; relative paths are resolved with
863 respect to the current working directory.
864
865 Optional argument *name* gives the name of the test; by default, or if ``None``,
866 ``os.path.basename(filename)`` is used.
867
868 Optional argument *package* is a Python package or the name of a Python package
869 whose directory should be used as the base directory for a module-relative
870 filename. If no package is specified, then the calling module's directory is
871 used as the base directory for module-relative filenames. It is an error to
872 specify *package* if *module_relative* is ``False``.
873
874 Optional argument *globs* gives a dict to be used as the globals when executing
875 examples. A new shallow copy of this dict is created for the doctest, so its
876 examples start with a clean slate. By default, or if ``None``, a new empty dict
877 is used.
878
879 Optional argument *extraglobs* gives a dict merged into the globals used to
880 execute examples. This works like :meth:`dict.update`: if *globs* and
881 *extraglobs* have a common key, the associated value in *extraglobs* appears in
882 the combined dict. By default, or if ``None``, no extra globals are used. This
883 is an advanced feature that allows parameterization of doctests. For example, a
884 doctest can be written for a base class, using a generic name for the class,
885 then reused to test any number of subclasses by passing an *extraglobs* dict
886 mapping the generic name to the subclass to be tested.
887
888 Optional argument *verbose* prints lots of stuff if true, and prints only
889 failures if false; by default, or if ``None``, it's true if and only if ``'-v'``
890 is in ``sys.argv``.
891
892 Optional argument *report* prints a summary at the end when true, else prints
893 nothing at the end. In verbose mode, the summary is detailed, else the summary
894 is very brief (in fact, empty if all tests passed).
895
Mariatta Wijaya81b89772017-02-06 20:15:01 -0800896 Optional argument *optionflags* (default value 0) takes the
897 :ref:`bitwise OR <bitwise>` of option flags.
898 See section :ref:`doctest-options`.
Georg Brandl116aa622007-08-15 14:28:22 +0000899
900 Optional argument *raise_on_error* defaults to false. If true, an exception is
901 raised upon the first failure or unexpected exception in an example. This
902 allows failures to be post-mortem debugged. Default behavior is to continue
903 running examples.
904
905 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass) that
906 should be used to extract tests from the files. It defaults to a normal parser
907 (i.e., ``DocTestParser()``).
908
909 Optional argument *encoding* specifies an encoding that should be used to
910 convert the file to unicode.
911
Georg Brandl116aa622007-08-15 14:28:22 +0000912
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000913.. 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 +0000914
915 All arguments are optional, and all except for *m* should be specified in
916 keyword form.
917
918 Test examples in docstrings in functions and classes reachable from module *m*
919 (or module :mod:`__main__` if *m* is not supplied or is ``None``), starting with
920 ``m.__doc__``.
921
922 Also test examples reachable from dict ``m.__test__``, if it exists and is not
923 ``None``. ``m.__test__`` maps names (strings) to functions, classes and
924 strings; function and class docstrings are searched for examples; strings are
925 searched directly, as if they were docstrings.
926
927 Only docstrings attached to objects belonging to module *m* are searched.
928
929 Return ``(failure_count, test_count)``.
930
931 Optional argument *name* gives the name of the module; by default, or if
932 ``None``, ``m.__name__`` is used.
933
934 Optional argument *exclude_empty* defaults to false. If true, objects for which
935 no doctests are found are excluded from consideration. The default is a backward
936 compatibility hack, so that code still using :meth:`doctest.master.summarize` in
937 conjunction with :func:`testmod` continues to get output for objects with no
938 tests. The *exclude_empty* argument to the newer :class:`DocTestFinder`
939 constructor defaults to true.
940
941 Optional arguments *extraglobs*, *verbose*, *report*, *optionflags*,
942 *raise_on_error*, and *globs* are the same as for function :func:`testfile`
943 above, except that *globs* defaults to ``m.__dict__``.
944
Georg Brandl116aa622007-08-15 14:28:22 +0000945
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000946.. function:: run_docstring_examples(f, globs, verbose=False, name="NoName", compileflags=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +0000947
Ethan Furman2a5f9da2015-09-17 22:20:41 -0700948 Test examples associated with object *f*; for example, *f* may be a string,
949 a module, a function, or a class object.
Georg Brandl116aa622007-08-15 14:28:22 +0000950
951 A shallow copy of dictionary argument *globs* is used for the execution context.
952
953 Optional argument *name* is used in failure messages, and defaults to
954 ``"NoName"``.
955
956 If optional argument *verbose* is true, output is generated even if there are no
957 failures. By default, output is generated only in case of an example failure.
958
959 Optional argument *compileflags* gives the set of flags that should be used by
960 the Python compiler when running the examples. By default, or if ``None``,
961 flags are deduced corresponding to the set of future features found in *globs*.
962
963 Optional argument *optionflags* works as for function :func:`testfile` above.
964
965
966.. _doctest-unittest-api:
967
968Unittest API
969------------
970
971As your collection of doctest'ed modules grows, you'll want a way to run all
Georg Brandl31835852008-05-12 17:38:56 +0000972their doctests systematically. :mod:`doctest` provides two functions that can
973be used to create :mod:`unittest` test suites from modules and text files
Georg Brandla8514832010-07-10 12:20:38 +0000974containing doctests. To integrate with :mod:`unittest` test discovery, include
975a :func:`load_tests` function in your test module::
Georg Brandl116aa622007-08-15 14:28:22 +0000976
977 import unittest
978 import doctest
Georg Brandla8514832010-07-10 12:20:38 +0000979 import my_module_with_doctests
Georg Brandl116aa622007-08-15 14:28:22 +0000980
Georg Brandla8514832010-07-10 12:20:38 +0000981 def load_tests(loader, tests, ignore):
982 tests.addTests(doctest.DocTestSuite(my_module_with_doctests))
R. David Murray796343b2010-12-13 22:50:30 +0000983 return tests
Georg Brandl116aa622007-08-15 14:28:22 +0000984
985There are two main functions for creating :class:`unittest.TestSuite` instances
986from text files and modules with doctests:
987
988
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000989.. 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 +0000990
991 Convert doctest tests from one or more text files to a
992 :class:`unittest.TestSuite`.
993
994 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
995 and runs the interactive examples in each file. If an example in any file
996 fails, then the synthesized unit test fails, and a :exc:`failureException`
997 exception is raised showing the name of the file containing the test and a
998 (sometimes approximate) line number.
999
1000 Pass one or more paths (as strings) to text files to be examined.
1001
1002 Options may be provided as keyword arguments:
1003
1004 Optional argument *module_relative* specifies how the filenames in *paths*
1005 should be interpreted:
1006
Benjamin Petersond23f8222009-04-05 19:13:16 +00001007 * If *module_relative* is ``True`` (the default), then each filename in
1008 *paths* specifies an OS-independent module-relative path. By default, this
1009 path is relative to the calling module's directory; but if the *package*
1010 argument is specified, then it is relative to that package. To ensure
1011 OS-independence, each filename should use ``/`` characters to separate path
1012 segments, and may not be an absolute path (i.e., it may not begin with
1013 ``/``).
Georg Brandl116aa622007-08-15 14:28:22 +00001014
Benjamin Petersond23f8222009-04-05 19:13:16 +00001015 * If *module_relative* is ``False``, then each filename in *paths* specifies
1016 an OS-specific path. The path may be absolute or relative; relative paths
1017 are resolved with respect to the current working directory.
Georg Brandl116aa622007-08-15 14:28:22 +00001018
Benjamin Petersond23f8222009-04-05 19:13:16 +00001019 Optional argument *package* is a Python package or the name of a Python
1020 package whose directory should be used as the base directory for
1021 module-relative filenames in *paths*. If no package is specified, then the
1022 calling module's directory is used as the base directory for module-relative
1023 filenames. It is an error to specify *package* if *module_relative* is
1024 ``False``.
Georg Brandl116aa622007-08-15 14:28:22 +00001025
Benjamin Petersond23f8222009-04-05 19:13:16 +00001026 Optional argument *setUp* specifies a set-up function for the test suite.
1027 This is called before running the tests in each file. The *setUp* function
Georg Brandl116aa622007-08-15 14:28:22 +00001028 will be passed a :class:`DocTest` object. The setUp function can access the
1029 test globals as the *globs* attribute of the test passed.
1030
Benjamin Petersond23f8222009-04-05 19:13:16 +00001031 Optional argument *tearDown* specifies a tear-down function for the test
1032 suite. This is called after running the tests in each file. The *tearDown*
1033 function will be passed a :class:`DocTest` object. The setUp function can
1034 access the test globals as the *globs* attribute of the test passed.
1035
Georg Brandl116aa622007-08-15 14:28:22 +00001036 Optional argument *globs* is a dictionary containing the initial global
1037 variables for the tests. A new copy of this dictionary is created for each
1038 test. By default, *globs* is a new empty dictionary.
1039
1040 Optional argument *optionflags* specifies the default doctest options for the
1041 tests, created by or-ing together individual option flags. See section
Benjamin Petersond23f8222009-04-05 19:13:16 +00001042 :ref:`doctest-options`. See function :func:`set_unittest_reportflags` below
1043 for a better way to set reporting options.
Georg Brandl116aa622007-08-15 14:28:22 +00001044
Benjamin Petersond23f8222009-04-05 19:13:16 +00001045 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass)
1046 that should be used to extract tests from the files. It defaults to a normal
1047 parser (i.e., ``DocTestParser()``).
Georg Brandl116aa622007-08-15 14:28:22 +00001048
1049 Optional argument *encoding* specifies an encoding that should be used to
1050 convert the file to unicode.
1051
Georg Brandl55ac8f02007-09-01 13:51:09 +00001052 The global ``__file__`` is added to the globals provided to doctests loaded
1053 from a text file using :func:`DocFileSuite`.
Georg Brandl116aa622007-08-15 14:28:22 +00001054
1055
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001056.. function:: DocTestSuite(module=None, globs=None, extraglobs=None, test_finder=None, setUp=None, tearDown=None, checker=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001057
1058 Convert doctest tests for a module to a :class:`unittest.TestSuite`.
1059
1060 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
1061 and runs each doctest in the module. If any of the doctests fail, then the
1062 synthesized unit test fails, and a :exc:`failureException` exception is raised
1063 showing the name of the file containing the test and a (sometimes approximate)
1064 line number.
1065
1066 Optional argument *module* provides the module to be tested. It can be a module
1067 object or a (possibly dotted) module name. If not specified, the module calling
1068 this function is used.
1069
1070 Optional argument *globs* is a dictionary containing the initial global
1071 variables for the tests. A new copy of this dictionary is created for each
1072 test. By default, *globs* is a new empty dictionary.
1073
1074 Optional argument *extraglobs* specifies an extra set of global variables, which
1075 is merged into *globs*. By default, no extra globals are used.
1076
1077 Optional argument *test_finder* is the :class:`DocTestFinder` object (or a
1078 drop-in replacement) that is used to extract doctests from the module.
1079
1080 Optional arguments *setUp*, *tearDown*, and *optionflags* are the same as for
1081 function :func:`DocFileSuite` above.
1082
Georg Brandl55ac8f02007-09-01 13:51:09 +00001083 This function uses the same search technique as :func:`testmod`.
Georg Brandl116aa622007-08-15 14:28:22 +00001084
R David Murray1976d9b2014-04-14 20:28:36 -04001085 .. versionchanged:: 3.5
1086 :func:`DocTestSuite` returns an empty :class:`unittest.TestSuite` if *module*
1087 contains no docstrings instead of raising :exc:`ValueError`.
R David Murray5abd76a2012-09-10 10:15:58 -04001088
Georg Brandl116aa622007-08-15 14:28:22 +00001089
1090Under the covers, :func:`DocTestSuite` creates a :class:`unittest.TestSuite` out
1091of :class:`doctest.DocTestCase` instances, and :class:`DocTestCase` is a
1092subclass of :class:`unittest.TestCase`. :class:`DocTestCase` isn't documented
1093here (it's an internal detail), but studying its code can answer questions about
1094the exact details of :mod:`unittest` integration.
1095
1096Similarly, :func:`DocFileSuite` creates a :class:`unittest.TestSuite` out of
1097:class:`doctest.DocFileCase` instances, and :class:`DocFileCase` is a subclass
1098of :class:`DocTestCase`.
1099
1100So both ways of creating a :class:`unittest.TestSuite` run instances of
1101:class:`DocTestCase`. This is important for a subtle reason: when you run
1102:mod:`doctest` functions yourself, you can control the :mod:`doctest` options in
1103use directly, by passing option flags to :mod:`doctest` functions. However, if
1104you're writing a :mod:`unittest` framework, :mod:`unittest` ultimately controls
1105when and how tests get run. The framework author typically wants to control
1106:mod:`doctest` reporting options (perhaps, e.g., specified by command line
1107options), but there's no way to pass options through :mod:`unittest` to
1108:mod:`doctest` test runners.
1109
1110For this reason, :mod:`doctest` also supports a notion of :mod:`doctest`
1111reporting flags specific to :mod:`unittest` support, via this function:
1112
1113
1114.. function:: set_unittest_reportflags(flags)
1115
1116 Set the :mod:`doctest` reporting flags to use.
1117
Mariatta Wijaya81b89772017-02-06 20:15:01 -08001118 Argument *flags* takes the :ref:`bitwise OR <bitwise>` of option flags. See
1119 section :ref:`doctest-options`. Only "reporting flags" can be used.
Georg Brandl116aa622007-08-15 14:28:22 +00001120
1121 This is a module-global setting, and affects all future doctests run by module
1122 :mod:`unittest`: the :meth:`runTest` method of :class:`DocTestCase` looks at
1123 the option flags specified for the test case when the :class:`DocTestCase`
1124 instance was constructed. If no reporting flags were specified (which is the
1125 typical and expected case), :mod:`doctest`'s :mod:`unittest` reporting flags are
Mariatta Wijaya81b89772017-02-06 20:15:01 -08001126 :ref:`bitwise ORed <bitwise>` into the option flags, and the option flags
1127 so augmented are passed to the :class:`DocTestRunner` instance created to
1128 run the doctest. If any reporting flags were specified when the
1129 :class:`DocTestCase` instance was constructed, :mod:`doctest`'s
1130 :mod:`unittest` reporting flags are ignored.
Georg Brandl116aa622007-08-15 14:28:22 +00001131
1132 The value of the :mod:`unittest` reporting flags in effect before the function
1133 was called is returned by the function.
1134
Georg Brandl116aa622007-08-15 14:28:22 +00001135
1136.. _doctest-advanced-api:
1137
1138Advanced API
1139------------
1140
1141The basic API is a simple wrapper that's intended to make doctest easy to use.
1142It is fairly flexible, and should meet most users' needs; however, if you
1143require more fine-grained control over testing, or wish to extend doctest's
1144capabilities, then you should use the advanced API.
1145
1146The advanced API revolves around two container classes, which are used to store
1147the interactive examples extracted from doctest cases:
1148
Ezio Melotti0639d5a2009-12-19 23:26:38 +00001149* :class:`Example`: A single Python :term:`statement`, paired with its expected
Christian Heimesd8654cf2007-12-02 15:22:16 +00001150 output.
Georg Brandl116aa622007-08-15 14:28:22 +00001151
1152* :class:`DocTest`: A collection of :class:`Example`\ s, typically extracted
1153 from a single docstring or text file.
1154
1155Additional processing classes are defined to find, parse, and run, and check
1156doctest examples:
1157
1158* :class:`DocTestFinder`: Finds all docstrings in a given module, and uses a
1159 :class:`DocTestParser` to create a :class:`DocTest` from every docstring that
1160 contains interactive examples.
1161
1162* :class:`DocTestParser`: Creates a :class:`DocTest` object from a string (such
1163 as an object's docstring).
1164
1165* :class:`DocTestRunner`: Executes the examples in a :class:`DocTest`, and uses
1166 an :class:`OutputChecker` to verify their output.
1167
1168* :class:`OutputChecker`: Compares the actual output from a doctest example with
1169 the expected output, and decides whether they match.
1170
1171The relationships among these processing classes are summarized in the following
1172diagram::
1173
1174 list of:
1175 +------+ +---------+
1176 |module| --DocTestFinder-> | DocTest | --DocTestRunner-> results
1177 +------+ | ^ +---------+ | ^ (printed)
1178 | | | Example | | |
1179 v | | ... | v |
1180 DocTestParser | Example | OutputChecker
1181 +---------+
1182
1183
1184.. _doctest-doctest:
1185
1186DocTest Objects
1187^^^^^^^^^^^^^^^
1188
1189
1190.. class:: DocTest(examples, globs, name, filename, lineno, docstring)
1191
1192 A collection of doctest examples that should be run in a single namespace. The
Senthil Kumarana6bac952011-07-04 11:28:30 -07001193 constructor arguments are used to initialize the attributes of the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001194
Georg Brandl116aa622007-08-15 14:28:22 +00001195
Senthil Kumarana6bac952011-07-04 11:28:30 -07001196 :class:`DocTest` defines the following attributes. They are initialized by
Benjamin Petersone41251e2008-04-25 01:59:09 +00001197 the constructor, and should not be modified directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001198
1199
Benjamin Petersone41251e2008-04-25 01:59:09 +00001200 .. attribute:: examples
Georg Brandl116aa622007-08-15 14:28:22 +00001201
Benjamin Petersone41251e2008-04-25 01:59:09 +00001202 A list of :class:`Example` objects encoding the individual interactive Python
1203 examples that should be run by this test.
Georg Brandl116aa622007-08-15 14:28:22 +00001204
1205
Benjamin Petersone41251e2008-04-25 01:59:09 +00001206 .. attribute:: globs
Georg Brandl116aa622007-08-15 14:28:22 +00001207
Benjamin Petersone41251e2008-04-25 01:59:09 +00001208 The namespace (aka globals) that the examples should be run in. This is a
1209 dictionary mapping names to values. Any changes to the namespace made by the
1210 examples (such as binding new variables) will be reflected in :attr:`globs`
1211 after the test is run.
Georg Brandl116aa622007-08-15 14:28:22 +00001212
1213
Benjamin Petersone41251e2008-04-25 01:59:09 +00001214 .. attribute:: name
Georg Brandl116aa622007-08-15 14:28:22 +00001215
Benjamin Petersone41251e2008-04-25 01:59:09 +00001216 A string name identifying the :class:`DocTest`. Typically, this is the name
1217 of the object or file that the test was extracted from.
Georg Brandl116aa622007-08-15 14:28:22 +00001218
1219
Benjamin Petersone41251e2008-04-25 01:59:09 +00001220 .. attribute:: filename
Georg Brandl116aa622007-08-15 14:28:22 +00001221
Benjamin Petersone41251e2008-04-25 01:59:09 +00001222 The name of the file that this :class:`DocTest` was extracted from; or
1223 ``None`` if the filename is unknown, or if the :class:`DocTest` was not
1224 extracted from a file.
Georg Brandl116aa622007-08-15 14:28:22 +00001225
1226
Benjamin Petersone41251e2008-04-25 01:59:09 +00001227 .. attribute:: lineno
Georg Brandl116aa622007-08-15 14:28:22 +00001228
Benjamin Petersone41251e2008-04-25 01:59:09 +00001229 The line number within :attr:`filename` where this :class:`DocTest` begins, or
1230 ``None`` if the line number is unavailable. This line number is zero-based
1231 with respect to the beginning of the file.
Georg Brandl116aa622007-08-15 14:28:22 +00001232
1233
Benjamin Petersone41251e2008-04-25 01:59:09 +00001234 .. attribute:: docstring
Georg Brandl116aa622007-08-15 14:28:22 +00001235
Serhiy Storchakaecf41da2016-10-19 16:29:26 +03001236 The string that the test was extracted from, or ``None`` if the string is
Benjamin Petersone41251e2008-04-25 01:59:09 +00001237 unavailable, or if the test was not extracted from a string.
Georg Brandl116aa622007-08-15 14:28:22 +00001238
1239
1240.. _doctest-example:
1241
1242Example Objects
1243^^^^^^^^^^^^^^^
1244
1245
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001246.. class:: Example(source, want, exc_msg=None, lineno=0, indent=0, options=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001247
1248 A single interactive example, consisting of a Python statement and its expected
Senthil Kumarana6bac952011-07-04 11:28:30 -07001249 output. The constructor arguments are used to initialize the attributes of
1250 the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001251
Georg Brandl116aa622007-08-15 14:28:22 +00001252
Senthil Kumarana6bac952011-07-04 11:28:30 -07001253 :class:`Example` defines the following attributes. They are initialized by
Benjamin Petersone41251e2008-04-25 01:59:09 +00001254 the constructor, and should not be modified directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001255
1256
Benjamin Petersone41251e2008-04-25 01:59:09 +00001257 .. attribute:: source
Georg Brandl116aa622007-08-15 14:28:22 +00001258
Benjamin Petersone41251e2008-04-25 01:59:09 +00001259 A string containing the example's source code. This source code consists of a
1260 single Python statement, and always ends with a newline; the constructor adds
1261 a newline when necessary.
Georg Brandl116aa622007-08-15 14:28:22 +00001262
1263
Benjamin Petersone41251e2008-04-25 01:59:09 +00001264 .. attribute:: want
Georg Brandl116aa622007-08-15 14:28:22 +00001265
Benjamin Petersone41251e2008-04-25 01:59:09 +00001266 The expected output from running the example's source code (either from
1267 stdout, or a traceback in case of exception). :attr:`want` ends with a
1268 newline unless no output is expected, in which case it's an empty string. The
1269 constructor adds a newline when necessary.
Georg Brandl116aa622007-08-15 14:28:22 +00001270
1271
Benjamin Petersone41251e2008-04-25 01:59:09 +00001272 .. attribute:: exc_msg
Georg Brandl116aa622007-08-15 14:28:22 +00001273
Benjamin Petersone41251e2008-04-25 01:59:09 +00001274 The exception message generated by the example, if the example is expected to
1275 generate an exception; or ``None`` if it is not expected to generate an
1276 exception. This exception message is compared against the return value of
1277 :func:`traceback.format_exception_only`. :attr:`exc_msg` ends with a newline
1278 unless it's ``None``. The constructor adds a newline if needed.
Georg Brandl116aa622007-08-15 14:28:22 +00001279
1280
Benjamin Petersone41251e2008-04-25 01:59:09 +00001281 .. attribute:: lineno
Georg Brandl116aa622007-08-15 14:28:22 +00001282
Benjamin Petersone41251e2008-04-25 01:59:09 +00001283 The line number within the string containing this example where the example
1284 begins. This line number is zero-based with respect to the beginning of the
1285 containing string.
Georg Brandl116aa622007-08-15 14:28:22 +00001286
1287
Benjamin Petersone41251e2008-04-25 01:59:09 +00001288 .. attribute:: indent
Georg Brandl116aa622007-08-15 14:28:22 +00001289
Benjamin Petersone41251e2008-04-25 01:59:09 +00001290 The example's indentation in the containing string, i.e., the number of space
1291 characters that precede the example's first prompt.
Georg Brandl116aa622007-08-15 14:28:22 +00001292
1293
Benjamin Petersone41251e2008-04-25 01:59:09 +00001294 .. attribute:: options
Georg Brandl116aa622007-08-15 14:28:22 +00001295
Benjamin Petersone41251e2008-04-25 01:59:09 +00001296 A dictionary mapping from option flags to ``True`` or ``False``, which is used
1297 to override default options for this example. Any option flags not contained
1298 in this dictionary are left at their default value (as specified by the
1299 :class:`DocTestRunner`'s :attr:`optionflags`). By default, no options are set.
Georg Brandl116aa622007-08-15 14:28:22 +00001300
1301
1302.. _doctest-doctestfinder:
1303
1304DocTestFinder objects
1305^^^^^^^^^^^^^^^^^^^^^
1306
1307
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001308.. class:: DocTestFinder(verbose=False, parser=DocTestParser(), recurse=True, exclude_empty=True)
Georg Brandl116aa622007-08-15 14:28:22 +00001309
1310 A processing class used to extract the :class:`DocTest`\ s that are relevant to
1311 a given object, from its docstring and the docstrings of its contained objects.
Zachary Warea4b7a752013-11-24 01:19:09 -06001312 :class:`DocTest`\ s can be extracted from modules, classes, functions,
1313 methods, staticmethods, classmethods, and properties.
Georg Brandl116aa622007-08-15 14:28:22 +00001314
1315 The optional argument *verbose* can be used to display the objects searched by
1316 the finder. It defaults to ``False`` (no output).
1317
1318 The optional argument *parser* specifies the :class:`DocTestParser` object (or a
1319 drop-in replacement) that is used to extract doctests from docstrings.
1320
1321 If the optional argument *recurse* is false, then :meth:`DocTestFinder.find`
1322 will only examine the given object, and not any contained objects.
1323
1324 If the optional argument *exclude_empty* is false, then
1325 :meth:`DocTestFinder.find` will include tests for objects with empty docstrings.
1326
Georg Brandl116aa622007-08-15 14:28:22 +00001327
Benjamin Petersone41251e2008-04-25 01:59:09 +00001328 :class:`DocTestFinder` defines the following method:
Georg Brandl116aa622007-08-15 14:28:22 +00001329
1330
Benjamin Petersone41251e2008-04-25 01:59:09 +00001331 .. method:: find(obj[, name][, module][, globs][, extraglobs])
Georg Brandl116aa622007-08-15 14:28:22 +00001332
Benjamin Petersone41251e2008-04-25 01:59:09 +00001333 Return a list of the :class:`DocTest`\ s that are defined by *obj*'s
1334 docstring, or by any of its contained objects' docstrings.
Georg Brandl116aa622007-08-15 14:28:22 +00001335
Benjamin Petersone41251e2008-04-25 01:59:09 +00001336 The optional argument *name* specifies the object's name; this name will be
1337 used to construct names for the returned :class:`DocTest`\ s. If *name* is
1338 not specified, then ``obj.__name__`` is used.
Georg Brandl116aa622007-08-15 14:28:22 +00001339
Benjamin Petersone41251e2008-04-25 01:59:09 +00001340 The optional parameter *module* is the module that contains the given object.
Serhiy Storchakaecf41da2016-10-19 16:29:26 +03001341 If the module is not specified or is ``None``, then the test finder will attempt
Benjamin Petersone41251e2008-04-25 01:59:09 +00001342 to automatically determine the correct module. The object's module is used:
Georg Brandl116aa622007-08-15 14:28:22 +00001343
Benjamin Petersone41251e2008-04-25 01:59:09 +00001344 * As a default namespace, if *globs* is not specified.
Georg Brandl116aa622007-08-15 14:28:22 +00001345
Benjamin Petersone41251e2008-04-25 01:59:09 +00001346 * To prevent the DocTestFinder from extracting DocTests from objects that are
1347 imported from other modules. (Contained objects with modules other than
1348 *module* are ignored.)
Georg Brandl116aa622007-08-15 14:28:22 +00001349
Benjamin Petersone41251e2008-04-25 01:59:09 +00001350 * To find the name of the file containing the object.
Georg Brandl116aa622007-08-15 14:28:22 +00001351
Benjamin Petersone41251e2008-04-25 01:59:09 +00001352 * To help find the line number of the object within its file.
Georg Brandl116aa622007-08-15 14:28:22 +00001353
Benjamin Petersone41251e2008-04-25 01:59:09 +00001354 If *module* is ``False``, no attempt to find the module will be made. This is
1355 obscure, of use mostly in testing doctest itself: if *module* is ``False``, or
1356 is ``None`` but cannot be found automatically, then all objects are considered
1357 to belong to the (non-existent) module, so all contained objects will
1358 (recursively) be searched for doctests.
Georg Brandl116aa622007-08-15 14:28:22 +00001359
Benjamin Petersone41251e2008-04-25 01:59:09 +00001360 The globals for each :class:`DocTest` is formed by combining *globs* and
1361 *extraglobs* (bindings in *extraglobs* override bindings in *globs*). A new
1362 shallow copy of the globals dictionary is created for each :class:`DocTest`.
1363 If *globs* is not specified, then it defaults to the module's *__dict__*, if
1364 specified, or ``{}`` otherwise. If *extraglobs* is not specified, then it
1365 defaults to ``{}``.
Georg Brandl116aa622007-08-15 14:28:22 +00001366
1367
1368.. _doctest-doctestparser:
1369
1370DocTestParser objects
1371^^^^^^^^^^^^^^^^^^^^^
1372
1373
1374.. class:: DocTestParser()
1375
1376 A processing class used to extract interactive examples from a string, and use
1377 them to create a :class:`DocTest` object.
1378
Georg Brandl116aa622007-08-15 14:28:22 +00001379
Benjamin Petersone41251e2008-04-25 01:59:09 +00001380 :class:`DocTestParser` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001381
1382
Benjamin Petersone41251e2008-04-25 01:59:09 +00001383 .. method:: get_doctest(string, globs, name, filename, lineno)
Georg Brandl116aa622007-08-15 14:28:22 +00001384
Benjamin Petersone41251e2008-04-25 01:59:09 +00001385 Extract all doctest examples from the given string, and collect them into a
1386 :class:`DocTest` object.
Georg Brandl116aa622007-08-15 14:28:22 +00001387
Benjamin Petersone41251e2008-04-25 01:59:09 +00001388 *globs*, *name*, *filename*, and *lineno* are attributes for the new
1389 :class:`DocTest` object. See the documentation for :class:`DocTest` for more
1390 information.
Georg Brandl116aa622007-08-15 14:28:22 +00001391
1392
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001393 .. method:: get_examples(string, name='<string>')
Georg Brandl116aa622007-08-15 14:28:22 +00001394
Benjamin Petersone41251e2008-04-25 01:59:09 +00001395 Extract all doctest examples from the given string, and return them as a list
1396 of :class:`Example` objects. Line numbers are 0-based. The optional argument
1397 *name* is a name identifying this string, and is only used for error messages.
Georg Brandl116aa622007-08-15 14:28:22 +00001398
1399
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001400 .. method:: parse(string, name='<string>')
Georg Brandl116aa622007-08-15 14:28:22 +00001401
Benjamin Petersone41251e2008-04-25 01:59:09 +00001402 Divide the given string into examples and intervening text, and return them as
1403 a list of alternating :class:`Example`\ s and strings. Line numbers for the
1404 :class:`Example`\ s are 0-based. The optional argument *name* is a name
1405 identifying this string, and is only used for error messages.
Georg Brandl116aa622007-08-15 14:28:22 +00001406
1407
1408.. _doctest-doctestrunner:
1409
1410DocTestRunner objects
1411^^^^^^^^^^^^^^^^^^^^^
1412
1413
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001414.. class:: DocTestRunner(checker=None, verbose=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +00001415
1416 A processing class used to execute and verify the interactive examples in a
1417 :class:`DocTest`.
1418
1419 The comparison between expected outputs and actual outputs is done by an
1420 :class:`OutputChecker`. This comparison may be customized with a number of
1421 option flags; see section :ref:`doctest-options` for more information. If the
1422 option flags are insufficient, then the comparison may also be customized by
1423 passing a subclass of :class:`OutputChecker` to the constructor.
1424
1425 The test runner's display output can be controlled in two ways. First, an output
1426 function can be passed to :meth:`TestRunner.run`; this function will be called
1427 with strings that should be displayed. It defaults to ``sys.stdout.write``. If
1428 capturing the output is not sufficient, then the display output can be also
1429 customized by subclassing DocTestRunner, and overriding the methods
1430 :meth:`report_start`, :meth:`report_success`,
1431 :meth:`report_unexpected_exception`, and :meth:`report_failure`.
1432
1433 The optional keyword argument *checker* specifies the :class:`OutputChecker`
1434 object (or drop-in replacement) that should be used to compare the expected
1435 outputs to the actual outputs of doctest examples.
1436
1437 The optional keyword argument *verbose* controls the :class:`DocTestRunner`'s
1438 verbosity. If *verbose* is ``True``, then information is printed about each
1439 example, as it is run. If *verbose* is ``False``, then only failures are
1440 printed. If *verbose* is unspecified, or ``None``, then verbose output is used
Éric Araujo713d3032010-11-18 16:38:46 +00001441 iff the command-line switch ``-v`` is used.
Georg Brandl116aa622007-08-15 14:28:22 +00001442
1443 The optional keyword argument *optionflags* can be used to control how the test
1444 runner compares expected output to actual output, and how it displays failures.
1445 For more information, see section :ref:`doctest-options`.
1446
Georg Brandl116aa622007-08-15 14:28:22 +00001447
Benjamin Petersone41251e2008-04-25 01:59:09 +00001448 :class:`DocTestParser` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001449
1450
Benjamin Petersone41251e2008-04-25 01:59:09 +00001451 .. method:: report_start(out, test, example)
Georg Brandl116aa622007-08-15 14:28:22 +00001452
Benjamin Petersone41251e2008-04-25 01:59:09 +00001453 Report that the test runner is about to process the given example. This method
1454 is provided to allow subclasses of :class:`DocTestRunner` to customize their
1455 output; it should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001456
Benjamin Petersone41251e2008-04-25 01:59:09 +00001457 *example* is the example about to be processed. *test* is the test
1458 *containing example*. *out* is the output function that was passed to
1459 :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001460
1461
Benjamin Petersone41251e2008-04-25 01:59:09 +00001462 .. method:: report_success(out, test, example, got)
Georg Brandl116aa622007-08-15 14:28:22 +00001463
Benjamin Petersone41251e2008-04-25 01:59:09 +00001464 Report that the given example ran successfully. This method is provided to
1465 allow subclasses of :class:`DocTestRunner` to customize their output; it
1466 should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001467
Benjamin Petersone41251e2008-04-25 01:59:09 +00001468 *example* is the example about to be processed. *got* is the actual output
1469 from the example. *test* is the test containing *example*. *out* is the
1470 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001471
1472
Benjamin Petersone41251e2008-04-25 01:59:09 +00001473 .. method:: report_failure(out, test, example, got)
Georg Brandl116aa622007-08-15 14:28:22 +00001474
Benjamin Petersone41251e2008-04-25 01:59:09 +00001475 Report that the given example failed. This method is provided to allow
1476 subclasses of :class:`DocTestRunner` to customize their output; it should not
1477 be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001478
Benjamin Petersone41251e2008-04-25 01:59:09 +00001479 *example* is the example about to be processed. *got* is the actual output
1480 from the example. *test* is the test containing *example*. *out* is the
1481 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001482
1483
Benjamin Petersone41251e2008-04-25 01:59:09 +00001484 .. method:: report_unexpected_exception(out, test, example, exc_info)
Georg Brandl116aa622007-08-15 14:28:22 +00001485
Benjamin Petersone41251e2008-04-25 01:59:09 +00001486 Report that the given example raised an unexpected exception. This method is
1487 provided to allow subclasses of :class:`DocTestRunner` to customize their
1488 output; it should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001489
Benjamin Petersone41251e2008-04-25 01:59:09 +00001490 *example* is the example about to be processed. *exc_info* is a tuple
1491 containing information about the unexpected exception (as returned by
1492 :func:`sys.exc_info`). *test* is the test containing *example*. *out* is the
1493 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001494
1495
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001496 .. method:: run(test, compileflags=None, out=None, clear_globs=True)
Georg Brandl116aa622007-08-15 14:28:22 +00001497
Benjamin Petersone41251e2008-04-25 01:59:09 +00001498 Run the examples in *test* (a :class:`DocTest` object), and display the
1499 results using the writer function *out*.
Georg Brandl116aa622007-08-15 14:28:22 +00001500
Benjamin Petersone41251e2008-04-25 01:59:09 +00001501 The examples are run in the namespace ``test.globs``. If *clear_globs* is
1502 true (the default), then this namespace will be cleared after the test runs,
1503 to help with garbage collection. If you would like to examine the namespace
1504 after the test completes, then use *clear_globs=False*.
Georg Brandl116aa622007-08-15 14:28:22 +00001505
Benjamin Petersone41251e2008-04-25 01:59:09 +00001506 *compileflags* gives the set of flags that should be used by the Python
1507 compiler when running the examples. If not specified, then it will default to
1508 the set of future-import flags that apply to *globs*.
Georg Brandl116aa622007-08-15 14:28:22 +00001509
Benjamin Petersone41251e2008-04-25 01:59:09 +00001510 The output of each example is checked using the :class:`DocTestRunner`'s
1511 output checker, and the results are formatted by the
1512 :meth:`DocTestRunner.report_\*` methods.
Georg Brandl116aa622007-08-15 14:28:22 +00001513
1514
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001515 .. method:: summarize(verbose=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001516
Benjamin Petersone41251e2008-04-25 01:59:09 +00001517 Print a summary of all the test cases that have been run by this DocTestRunner,
1518 and return a :term:`named tuple` ``TestResults(failed, attempted)``.
Georg Brandl116aa622007-08-15 14:28:22 +00001519
Benjamin Petersone41251e2008-04-25 01:59:09 +00001520 The optional *verbose* argument controls how detailed the summary is. If the
1521 verbosity is not specified, then the :class:`DocTestRunner`'s verbosity is
1522 used.
Georg Brandl116aa622007-08-15 14:28:22 +00001523
1524.. _doctest-outputchecker:
1525
1526OutputChecker objects
1527^^^^^^^^^^^^^^^^^^^^^
1528
1529
1530.. class:: OutputChecker()
1531
1532 A class used to check the whether the actual output from a doctest example
1533 matches the expected output. :class:`OutputChecker` defines two methods:
1534 :meth:`check_output`, which compares a given pair of outputs, and returns true
1535 if they match; and :meth:`output_difference`, which returns a string describing
1536 the differences between two outputs.
1537
Georg Brandl116aa622007-08-15 14:28:22 +00001538
Benjamin Petersone41251e2008-04-25 01:59:09 +00001539 :class:`OutputChecker` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001540
Benjamin Petersone41251e2008-04-25 01:59:09 +00001541 .. method:: check_output(want, got, optionflags)
Georg Brandl116aa622007-08-15 14:28:22 +00001542
Benjamin Petersone41251e2008-04-25 01:59:09 +00001543 Return ``True`` iff the actual output from an example (*got*) matches the
1544 expected output (*want*). These strings are always considered to match if
1545 they are identical; but depending on what option flags the test runner is
1546 using, several non-exact match types are also possible. See section
1547 :ref:`doctest-options` for more information about option flags.
Georg Brandl116aa622007-08-15 14:28:22 +00001548
1549
Benjamin Petersone41251e2008-04-25 01:59:09 +00001550 .. method:: output_difference(example, got, optionflags)
Georg Brandl116aa622007-08-15 14:28:22 +00001551
Benjamin Petersone41251e2008-04-25 01:59:09 +00001552 Return a string describing the differences between the expected output for a
1553 given example (*example*) and the actual output (*got*). *optionflags* is the
1554 set of option flags used to compare *want* and *got*.
Georg Brandl116aa622007-08-15 14:28:22 +00001555
1556
1557.. _doctest-debugging:
1558
1559Debugging
1560---------
1561
1562Doctest provides several mechanisms for debugging doctest examples:
1563
1564* Several functions convert doctests to executable Python programs, which can be
1565 run under the Python debugger, :mod:`pdb`.
1566
1567* The :class:`DebugRunner` class is a subclass of :class:`DocTestRunner` that
1568 raises an exception for the first failing example, containing information about
1569 that example. This information can be used to perform post-mortem debugging on
1570 the example.
1571
1572* The :mod:`unittest` cases generated by :func:`DocTestSuite` support the
1573 :meth:`debug` method defined by :class:`unittest.TestCase`.
1574
1575* You can add a call to :func:`pdb.set_trace` in a doctest example, and you'll
1576 drop into the Python debugger when that line is executed. Then you can inspect
1577 current values of variables, and so on. For example, suppose :file:`a.py`
1578 contains just this module docstring::
1579
1580 """
1581 >>> def f(x):
1582 ... g(x*2)
1583 >>> def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001584 ... print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001585 ... import pdb; pdb.set_trace()
1586 >>> f(3)
1587 9
1588 """
1589
1590 Then an interactive Python session may look like this::
1591
1592 >>> import a, doctest
1593 >>> doctest.testmod(a)
1594 --Return--
1595 > <doctest a[1]>(3)g()->None
1596 -> import pdb; pdb.set_trace()
1597 (Pdb) list
1598 1 def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001599 2 print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001600 3 -> import pdb; pdb.set_trace()
1601 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001602 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001603 6
1604 (Pdb) step
1605 --Return--
1606 > <doctest a[0]>(2)f()->None
1607 -> g(x*2)
1608 (Pdb) list
1609 1 def f(x):
1610 2 -> g(x*2)
1611 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001612 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001613 3
1614 (Pdb) step
1615 --Return--
1616 > <doctest a[2]>(1)?()->None
1617 -> f(3)
1618 (Pdb) cont
1619 (0, 3)
1620 >>>
1621
Georg Brandl116aa622007-08-15 14:28:22 +00001622
1623Functions that convert doctests to Python code, and possibly run the synthesized
1624code under the debugger:
1625
1626
1627.. function:: script_from_examples(s)
1628
1629 Convert text with examples to a script.
1630
1631 Argument *s* is a string containing doctest examples. The string is converted
1632 to a Python script, where doctest examples in *s* are converted to regular code,
1633 and everything else is converted to Python comments. The generated script is
1634 returned as a string. For example, ::
1635
1636 import doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001637 print(doctest.script_from_examples(r"""
Georg Brandl116aa622007-08-15 14:28:22 +00001638 Set x and y to 1 and 2.
1639 >>> x, y = 1, 2
1640
1641 Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001642 >>> print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001643 3
Georg Brandl6911e3c2007-09-04 07:15:32 +00001644 """))
Georg Brandl116aa622007-08-15 14:28:22 +00001645
1646 displays::
1647
1648 # Set x and y to 1 and 2.
1649 x, y = 1, 2
1650 #
1651 # Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001652 print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001653 # Expected:
1654 ## 3
1655
1656 This function is used internally by other functions (see below), but can also be
1657 useful when you want to transform an interactive Python session into a Python
1658 script.
1659
Georg Brandl116aa622007-08-15 14:28:22 +00001660
1661.. function:: testsource(module, name)
1662
1663 Convert the doctest for an object to a script.
1664
1665 Argument *module* is a module object, or dotted name of a module, containing the
1666 object whose doctests are of interest. Argument *name* is the name (within the
1667 module) of the object with the doctests of interest. The result is a string,
1668 containing the object's docstring converted to a Python script, as described for
1669 :func:`script_from_examples` above. For example, if module :file:`a.py`
1670 contains a top-level function :func:`f`, then ::
1671
1672 import a, doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001673 print(doctest.testsource(a, "a.f"))
Georg Brandl116aa622007-08-15 14:28:22 +00001674
1675 prints a script version of function :func:`f`'s docstring, with doctests
1676 converted to code, and the rest placed in comments.
1677
Georg Brandl116aa622007-08-15 14:28:22 +00001678
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001679.. function:: debug(module, name, pm=False)
Georg Brandl116aa622007-08-15 14:28:22 +00001680
1681 Debug the doctests for an object.
1682
1683 The *module* and *name* arguments are the same as for function
1684 :func:`testsource` above. The synthesized Python script for the named object's
1685 docstring is written to a temporary file, and then that file is run under the
1686 control of the Python debugger, :mod:`pdb`.
1687
1688 A shallow copy of ``module.__dict__`` is used for both local and global
1689 execution context.
1690
1691 Optional argument *pm* controls whether post-mortem debugging is used. If *pm*
1692 has a true value, the script file is run directly, and the debugger gets
1693 involved only if the script terminates via raising an unhandled exception. If
1694 it does, then post-mortem debugging is invoked, via :func:`pdb.post_mortem`,
1695 passing the traceback object from the unhandled exception. If *pm* is not
1696 specified, or is false, the script is run under the debugger from the start, via
1697 passing an appropriate :func:`exec` call to :func:`pdb.run`.
1698
Georg Brandl116aa622007-08-15 14:28:22 +00001699
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001700.. function:: debug_src(src, pm=False, globs=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001701
1702 Debug the doctests in a string.
1703
1704 This is like function :func:`debug` above, except that a string containing
1705 doctest examples is specified directly, via the *src* argument.
1706
1707 Optional argument *pm* has the same meaning as in function :func:`debug` above.
1708
1709 Optional argument *globs* gives a dictionary to use as both local and global
1710 execution context. If not specified, or ``None``, an empty dictionary is used.
1711 If specified, a shallow copy of the dictionary is used.
1712
Georg Brandl116aa622007-08-15 14:28:22 +00001713
1714The :class:`DebugRunner` class, and the special exceptions it may raise, are of
1715most interest to testing framework authors, and will only be sketched here. See
1716the source code, and especially :class:`DebugRunner`'s docstring (which is a
1717doctest!) for more details:
1718
1719
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001720.. class:: DebugRunner(checker=None, verbose=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +00001721
1722 A subclass of :class:`DocTestRunner` that raises an exception as soon as a
1723 failure is encountered. If an unexpected exception occurs, an
1724 :exc:`UnexpectedException` exception is raised, containing the test, the
1725 example, and the original exception. If the output doesn't match, then a
1726 :exc:`DocTestFailure` exception is raised, containing the test, the example, and
1727 the actual output.
1728
1729 For information about the constructor parameters and methods, see the
1730 documentation for :class:`DocTestRunner` in section :ref:`doctest-advanced-api`.
1731
1732There are two exceptions that may be raised by :class:`DebugRunner` instances:
1733
1734
1735.. exception:: DocTestFailure(test, example, got)
1736
Georg Brandl7cb13192010-08-03 12:06:29 +00001737 An exception raised by :class:`DocTestRunner` to signal that a doctest example's
Georg Brandl116aa622007-08-15 14:28:22 +00001738 actual output did not match its expected output. The constructor arguments are
Senthil Kumarana6bac952011-07-04 11:28:30 -07001739 used to initialize the attributes of the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001740
Senthil Kumarana6bac952011-07-04 11:28:30 -07001741:exc:`DocTestFailure` defines the following attributes:
Georg Brandl116aa622007-08-15 14:28:22 +00001742
1743
1744.. attribute:: DocTestFailure.test
1745
1746 The :class:`DocTest` object that was being run when the example failed.
1747
1748
1749.. attribute:: DocTestFailure.example
1750
1751 The :class:`Example` that failed.
1752
1753
1754.. attribute:: DocTestFailure.got
1755
1756 The example's actual output.
1757
1758
1759.. exception:: UnexpectedException(test, example, exc_info)
1760
Georg Brandl7cb13192010-08-03 12:06:29 +00001761 An exception raised by :class:`DocTestRunner` to signal that a doctest
1762 example raised an unexpected exception. The constructor arguments are used
Senthil Kumarana6bac952011-07-04 11:28:30 -07001763 to initialize the attributes of the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001764
Senthil Kumarana6bac952011-07-04 11:28:30 -07001765:exc:`UnexpectedException` defines the following attributes:
Georg Brandl116aa622007-08-15 14:28:22 +00001766
1767
1768.. attribute:: UnexpectedException.test
1769
1770 The :class:`DocTest` object that was being run when the example failed.
1771
1772
1773.. attribute:: UnexpectedException.example
1774
1775 The :class:`Example` that failed.
1776
1777
1778.. attribute:: UnexpectedException.exc_info
1779
1780 A tuple containing information about the unexpected exception, as returned by
1781 :func:`sys.exc_info`.
1782
1783
1784.. _doctest-soapbox:
1785
1786Soapbox
1787-------
1788
1789As mentioned in the introduction, :mod:`doctest` has grown to have three primary
1790uses:
1791
1792#. Checking examples in docstrings.
1793
1794#. Regression testing.
1795
1796#. Executable documentation / literate testing.
1797
1798These uses have different requirements, and it is important to distinguish them.
1799In particular, filling your docstrings with obscure test cases makes for bad
1800documentation.
1801
1802When writing a docstring, choose docstring examples with care. There's an art to
1803this that needs to be learned---it may not be natural at first. Examples should
1804add genuine value to the documentation. A good example can often be worth many
1805words. If done with care, the examples will be invaluable for your users, and
1806will pay back the time it takes to collect them many times over as the years go
1807by and things change. I'm still amazed at how often one of my :mod:`doctest`
1808examples stops working after a "harmless" change.
1809
1810Doctest also makes an excellent tool for regression testing, especially if you
1811don't skimp on explanatory text. By interleaving prose and examples, it becomes
1812much easier to keep track of what's actually being tested, and why. When a test
1813fails, good prose can make it much easier to figure out what the problem is, and
1814how it should be fixed. It's true that you could write extensive comments in
1815code-based testing, but few programmers do. Many have found that using doctest
1816approaches instead leads to much clearer tests. Perhaps this is simply because
1817doctest makes writing prose a little easier than writing code, while writing
1818comments in code is a little harder. I think it goes deeper than just that:
1819the natural attitude when writing a doctest-based test is that you want to
1820explain the fine points of your software, and illustrate them with examples.
1821This in turn naturally leads to test files that start with the simplest
1822features, and logically progress to complications and edge cases. A coherent
1823narrative is the result, instead of a collection of isolated functions that test
1824isolated bits of functionality seemingly at random. It's a different attitude,
1825and produces different results, blurring the distinction between testing and
1826explaining.
1827
1828Regression testing is best confined to dedicated objects or files. There are
1829several options for organizing tests:
1830
1831* Write text files containing test cases as interactive examples, and test the
1832 files using :func:`testfile` or :func:`DocFileSuite`. This is recommended,
1833 although is easiest to do for new projects, designed from the start to use
1834 doctest.
1835
1836* Define functions named ``_regrtest_topic`` that consist of single docstrings,
1837 containing test cases for the named topics. These functions can be included in
1838 the same file as the module, or separated out into a separate test file.
1839
1840* Define a ``__test__`` dictionary mapping from regression test topics to
1841 docstrings containing test cases.
1842
Ethan Furman2a5f9da2015-09-17 22:20:41 -07001843When you have placed your tests in a module, the module can itself be the test
1844runner. When a test fails, you can arrange for your test runner to re-run only
1845the failing doctest while you debug the problem. Here is a minimal example of
1846such a test runner::
1847
1848 if __name__ == '__main__':
1849 import doctest
1850 flags = doctest.REPORT_NDIFF|doctest.FAIL_FAST
1851 if len(sys.argv) > 1:
1852 name = sys.argv[1]
1853 if name in globals():
1854 obj = globals()[name]
1855 else:
1856 obj = __test__[name]
1857 doctest.run_docstring_examples(obj, globals(), name=name,
1858 optionflags=flags)
1859 else:
1860 fail, total = doctest.testmod(optionflags=flags)
1861 print("{} failures out of {} tests".format(fail, total))
1862
1863
Georg Brandl116aa622007-08-15 14:28:22 +00001864.. rubric:: Footnotes
1865
1866.. [#] Examples containing both expected output and an exception are not supported.
1867 Trying to guess where one ends and the other begins is too error-prone, and that
1868 also makes for a confusing test.