blob: 04bc219fe746dd1ee3c0c4110ce515f867a6eefc [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001:mod:`doctest` --- Test interactive Python examples
2===================================================
3
4.. module:: doctest
5 :synopsis: Test pieces of code within docstrings.
6.. moduleauthor:: Tim Peters <tim@python.org>
7.. sectionauthor:: Tim Peters <tim@python.org>
8.. sectionauthor:: Moshe Zadka <moshez@debian.org>
9.. sectionauthor:: Edward Loper <edloper@users.sourceforge.net>
10
11
12The :mod:`doctest` module searches for pieces of text that look like interactive
13Python sessions, and then executes those sessions to verify that they work
14exactly as shown. There are several common ways to use doctest:
15
16* To check that a module's docstrings are up-to-date by verifying that all
17 interactive examples still work as documented.
18
19* To perform regression testing by verifying that interactive examples from a
20 test file or a test object work as expected.
21
22* To write tutorial documentation for a package, liberally illustrated with
23 input-output examples. Depending on whether the examples or the expository text
24 are emphasized, this has the flavor of "literate testing" or "executable
25 documentation".
26
27Here's a complete but small example module::
28
29 """
30 This is the "example" module.
31
32 The example module supplies one function, factorial(). For example,
33
34 >>> factorial(5)
35 120
36 """
37
38 def factorial(n):
39 """Return the factorial of n, an exact integer >= 0.
40
Georg Brandl116aa622007-08-15 14:28:22 +000041 >>> [factorial(n) for n in range(6)]
42 [1, 1, 2, 6, 24, 120]
Georg Brandl116aa622007-08-15 14:28:22 +000043 >>> factorial(30)
Georg Brandl5c106642007-11-29 17:41:05 +000044 265252859812191058636308480000000
Georg Brandl116aa622007-08-15 14:28:22 +000045 >>> factorial(-1)
46 Traceback (most recent call last):
47 ...
48 ValueError: n must be >= 0
49
50 Factorials of floats are OK, but the float must be an exact integer:
51 >>> factorial(30.1)
52 Traceback (most recent call last):
53 ...
54 ValueError: n must be exact integer
55 >>> factorial(30.0)
Georg Brandl5c106642007-11-29 17:41:05 +000056 265252859812191058636308480000000
Georg Brandl116aa622007-08-15 14:28:22 +000057
58 It must also not be ridiculously large:
59 >>> factorial(1e100)
60 Traceback (most recent call last):
61 ...
62 OverflowError: n too large
63 """
64
Georg Brandl116aa622007-08-15 14:28:22 +000065 import math
66 if not n >= 0:
67 raise ValueError("n must be >= 0")
68 if math.floor(n) != n:
69 raise ValueError("n must be exact integer")
70 if n+1 == n: # catch a value like 1e300
71 raise OverflowError("n too large")
72 result = 1
73 factor = 2
74 while factor <= n:
75 result *= factor
76 factor += 1
77 return result
78
Georg Brandl116aa622007-08-15 14:28:22 +000079
80 if __name__ == "__main__":
Guido van Rossum04110fb2007-08-24 16:32:05 +000081 import doctest
82 doctest.testmod()
Georg Brandl116aa622007-08-15 14:28:22 +000083
84If you run :file:`example.py` directly from the command line, :mod:`doctest`
85works its magic::
86
87 $ python example.py
88 $
89
90There's no output! That's normal, and it means all the examples worked. Pass
91:option:`-v` to the script, and :mod:`doctest` prints a detailed log of what
92it's trying, and prints a summary at the end::
93
94 $ python example.py -v
95 Trying:
96 factorial(5)
97 Expecting:
98 120
99 ok
100 Trying:
101 [factorial(n) for n in range(6)]
102 Expecting:
103 [1, 1, 2, 6, 24, 120]
104 ok
Georg Brandl116aa622007-08-15 14:28:22 +0000105
106And so on, eventually ending with::
107
108 Trying:
109 factorial(1e100)
110 Expecting:
111 Traceback (most recent call last):
112 ...
113 OverflowError: n too large
114 ok
Georg Brandl116aa622007-08-15 14:28:22 +0000115 2 items passed all tests:
116 1 tests in __main__
117 8 tests in __main__.factorial
Guido van Rossum04110fb2007-08-24 16:32:05 +0000118 9 tests in 2 items.
Georg Brandl116aa622007-08-15 14:28:22 +0000119 9 passed and 0 failed.
120 Test passed.
121 $
122
123That's all you need to know to start making productive use of :mod:`doctest`!
124Jump in. The following sections provide full details. Note that there are many
125examples of doctests in the standard Python test suite and libraries.
126Especially useful examples can be found in the standard test file
127:file:`Lib/test/test_doctest.py`.
128
129
130.. _doctest-simple-testmod:
131
132Simple Usage: Checking Examples in Docstrings
133---------------------------------------------
134
135The simplest way to start using doctest (but not necessarily the way you'll
136continue to do it) is to end each module :mod:`M` with::
137
Guido van Rossum04110fb2007-08-24 16:32:05 +0000138 if __name__ == "__main__":
Georg Brandl116aa622007-08-15 14:28:22 +0000139 import doctest
140 doctest.testmod()
141
Georg Brandl116aa622007-08-15 14:28:22 +0000142:mod:`doctest` then examines docstrings in module :mod:`M`.
143
144Running the module as a script causes the examples in the docstrings to get
145executed and verified::
146
147 python M.py
148
149This won't display anything unless an example fails, in which case the failing
150example(s) and the cause(s) of the failure(s) are printed to stdout, and the
151final line of output is ``***Test Failed*** N failures.``, where *N* is the
152number of examples that failed.
153
154Run it with the :option:`-v` switch instead::
155
156 python M.py -v
157
158and a detailed report of all examples tried is printed to standard output, along
159with assorted summaries at the end.
160
161You can force verbose mode by passing ``verbose=True`` to :func:`testmod`, or
162prohibit it by passing ``verbose=False``. In either of those cases,
163``sys.argv`` is not examined by :func:`testmod` (so passing :option:`-v` or not
164has no effect).
165
166Since Python 2.6, there is also a command line shortcut for running
167:func:`testmod`. You can instruct the Python interpreter to run the doctest
168module directly from the standard library and pass the module name(s) on the
169command line::
170
171 python -m doctest -v example.py
172
173This will import :file:`example.py` as a standalone module and run
174:func:`testmod` on it. Note that this may not work correctly if the file is
175part of a package and imports other submodules from that package.
176
177For more information on :func:`testmod`, see section :ref:`doctest-basic-api`.
178
179
180.. _doctest-simple-testfile:
181
182Simple Usage: Checking Examples in a Text File
183----------------------------------------------
184
185Another simple application of doctest is testing interactive examples in a text
186file. This can be done with the :func:`testfile` function::
187
188 import doctest
189 doctest.testfile("example.txt")
190
191That short script executes and verifies any interactive Python examples
192contained in the file :file:`example.txt`. The file content is treated as if it
193were a single giant docstring; the file doesn't need to contain a Python
194program! For example, perhaps :file:`example.txt` contains this::
195
196 The ``example`` module
197 ======================
198
199 Using ``factorial``
200 -------------------
201
202 This is an example text file in reStructuredText format. First import
203 ``factorial`` from the ``example`` module:
204
205 >>> from example import factorial
206
207 Now use it:
208
209 >>> factorial(6)
210 120
211
212Running ``doctest.testfile("example.txt")`` then finds the error in this
213documentation::
214
215 File "./example.txt", line 14, in example.txt
216 Failed example:
217 factorial(6)
218 Expected:
219 120
220 Got:
221 720
222
223As with :func:`testmod`, :func:`testfile` won't display anything unless an
224example fails. If an example does fail, then the failing example(s) and the
225cause(s) of the failure(s) are printed to stdout, using the same format as
226:func:`testmod`.
227
228By default, :func:`testfile` looks for files in the calling module's directory.
229See section :ref:`doctest-basic-api` for a description of the optional arguments
230that can be used to tell it to look for files in other locations.
231
232Like :func:`testmod`, :func:`testfile`'s verbosity can be set with the
233:option:`-v` command-line switch or with the optional keyword argument
234*verbose*.
235
236Since Python 2.6, there is also a command line shortcut for running
237:func:`testfile`. You can instruct the Python interpreter to run the doctest
238module directly from the standard library and pass the file name(s) on the
239command line::
240
241 python -m doctest -v example.txt
242
243Because the file name does not end with :file:`.py`, :mod:`doctest` infers that
244it must be run with :func:`testfile`, not :func:`testmod`.
245
246For more information on :func:`testfile`, see section :ref:`doctest-basic-api`.
247
248
249.. _doctest-how-it-works:
250
251How It Works
252------------
253
254This section examines in detail how doctest works: which docstrings it looks at,
255how it finds interactive examples, what execution context it uses, how it
256handles exceptions, and how option flags can be used to control its behavior.
257This is the information that you need to know to write doctest examples; for
258information about actually running doctest on these examples, see the following
259sections.
260
261
262.. _doctest-which-docstrings:
263
264Which Docstrings Are Examined?
265^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
266
267The module docstring, and all function, class and method docstrings are
268searched. Objects imported into the module are not searched.
269
270In addition, if ``M.__test__`` exists and "is true", it must be a dict, and each
271entry maps a (string) name to a function object, class object, or string.
272Function and class object docstrings found from ``M.__test__`` are searched, and
273strings are treated as if they were docstrings. In output, a key ``K`` in
274``M.__test__`` appears with name ::
275
276 <name of M>.__test__.K
277
278Any classes found are recursively searched similarly, to test docstrings in
279their contained methods and nested classes.
280
Georg Brandl116aa622007-08-15 14:28:22 +0000281
282.. _doctest-finding-examples:
283
284How are Docstring Examples Recognized?
285^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
286
287In most cases a copy-and-paste of an interactive console session works fine, but
288doctest isn't trying to do an exact emulation of any specific Python shell. All
289hard tab characters are expanded to spaces, using 8-column tab stops. If you
290don't believe tabs should mean that, too bad: don't use hard tabs, or write
291your own :class:`DocTestParser` class.
292
Georg Brandl116aa622007-08-15 14:28:22 +0000293::
294
295 >>> # comments are ignored
296 >>> x = 12
297 >>> x
298 12
299 >>> if x == 13:
Georg Brandl6911e3c2007-09-04 07:15:32 +0000300 ... print("yes")
Georg Brandl116aa622007-08-15 14:28:22 +0000301 ... else:
Georg Brandl6911e3c2007-09-04 07:15:32 +0000302 ... print("no")
303 ... print("NO")
304 ... print("NO!!!")
Georg Brandl116aa622007-08-15 14:28:22 +0000305 ...
306 no
307 NO
308 NO!!!
309 >>>
310
311Any expected output must immediately follow the final ``'>>> '`` or ``'... '``
312line containing the code, and the expected output (if any) extends to the next
313``'>>> '`` or all-whitespace line.
314
315The fine print:
316
317* Expected output cannot contain an all-whitespace line, since such a line is
318 taken to signal the end of expected output. If expected output does contain a
319 blank line, put ``<BLANKLINE>`` in your doctest example each place a blank line
320 is expected.
321
Georg Brandl116aa622007-08-15 14:28:22 +0000322* Output to stdout is captured, but not output to stderr (exception tracebacks
323 are captured via a different means).
324
325* If you continue a line via backslashing in an interactive session, or for any
326 other reason use a backslash, you should use a raw docstring, which will
327 preserve your backslashes exactly as you type them::
328
329 >>> def f(x):
330 ... r'''Backslashes in a raw docstring: m\n'''
Georg Brandl6911e3c2007-09-04 07:15:32 +0000331 >>> print(f.__doc__)
Georg Brandl116aa622007-08-15 14:28:22 +0000332 Backslashes in a raw docstring: m\n
333
334 Otherwise, the backslash will be interpreted as part of the string. For example,
335 the "\\" above would be interpreted as a newline character. Alternatively, you
336 can double each backslash in the doctest version (and not use a raw string)::
337
338 >>> def f(x):
339 ... '''Backslashes in a raw docstring: m\\n'''
Georg Brandl6911e3c2007-09-04 07:15:32 +0000340 >>> print(f.__doc__)
Georg Brandl116aa622007-08-15 14:28:22 +0000341 Backslashes in a raw docstring: m\n
342
343* The starting column doesn't matter::
344
345 >>> assert "Easy!"
346 >>> import math
347 >>> math.floor(1.9)
348 1.0
349
350 and as many leading whitespace characters are stripped from the expected output
351 as appeared in the initial ``'>>> '`` line that started the example.
352
353
354.. _doctest-execution-context:
355
356What's the Execution Context?
357^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
358
359By default, each time :mod:`doctest` finds a docstring to test, it uses a
360*shallow copy* of :mod:`M`'s globals, so that running tests doesn't change the
361module's real globals, and so that one test in :mod:`M` can't leave behind
362crumbs that accidentally allow another test to work. This means examples can
363freely use any names defined at top-level in :mod:`M`, and names defined earlier
364in the docstring being run. Examples cannot see names defined in other
365docstrings.
366
367You can force use of your own dict as the execution context by passing
368``globs=your_dict`` to :func:`testmod` or :func:`testfile` instead.
369
370
371.. _doctest-exceptions:
372
373What About Exceptions?
374^^^^^^^^^^^^^^^^^^^^^^
375
376No problem, provided that the traceback is the only output produced by the
377example: just paste in the traceback. [#]_ Since tracebacks contain details
378that are likely to change rapidly (for example, exact file paths and line
379numbers), this is one case where doctest works hard to be flexible in what it
380accepts.
381
382Simple example::
383
384 >>> [1, 2, 3].remove(42)
385 Traceback (most recent call last):
386 File "<stdin>", line 1, in ?
387 ValueError: list.remove(x): x not in list
388
389That doctest succeeds if :exc:`ValueError` is raised, with the ``list.remove(x):
390x not in list`` detail as shown.
391
392The expected output for an exception must start with a traceback header, which
393may be either of the following two lines, indented the same as the first line of
394the example::
395
396 Traceback (most recent call last):
397 Traceback (innermost last):
398
399The traceback header is followed by an optional traceback stack, whose contents
400are ignored by doctest. The traceback stack is typically omitted, or copied
401verbatim from an interactive session.
402
403The traceback stack is followed by the most interesting part: the line(s)
404containing the exception type and detail. This is usually the last line of a
405traceback, but can extend across multiple lines if the exception has a
406multi-line detail::
407
408 >>> raise ValueError('multi\n line\ndetail')
409 Traceback (most recent call last):
410 File "<stdin>", line 1, in ?
411 ValueError: multi
412 line
413 detail
414
415The last three lines (starting with :exc:`ValueError`) are compared against the
416exception's type and detail, and the rest are ignored.
417
418Best practice is to omit the traceback stack, unless it adds significant
419documentation value to the example. So the last example is probably better as::
420
421 >>> raise ValueError('multi\n line\ndetail')
422 Traceback (most recent call last):
423 ...
424 ValueError: multi
425 line
426 detail
427
428Note that tracebacks are treated very specially. In particular, in the
429rewritten example, the use of ``...`` is independent of doctest's
430:const:`ELLIPSIS` option. The ellipsis in that example could be left out, or
431could just as well be three (or three hundred) commas or digits, or an indented
432transcript of a Monty Python skit.
433
434Some details you should read once, but won't need to remember:
435
436* Doctest can't guess whether your expected output came from an exception
437 traceback or from ordinary printing. So, e.g., an example that expects
438 ``ValueError: 42 is prime`` will pass whether :exc:`ValueError` is actually
439 raised or if the example merely prints that traceback text. In practice,
440 ordinary output rarely begins with a traceback header line, so this doesn't
441 create real problems.
442
443* Each line of the traceback stack (if present) must be indented further than
444 the first line of the example, *or* start with a non-alphanumeric character.
445 The first line following the traceback header indented the same and starting
446 with an alphanumeric is taken to be the start of the exception detail. Of
447 course this does the right thing for genuine tracebacks.
448
449* When the :const:`IGNORE_EXCEPTION_DETAIL` doctest option is is specified,
450 everything following the leftmost colon is ignored.
451
452* The interactive shell omits the traceback header line for some
453 :exc:`SyntaxError`\ s. But doctest uses the traceback header line to
454 distinguish exceptions from non-exceptions. So in the rare case where you need
455 to test a :exc:`SyntaxError` that omits the traceback header, you will need to
456 manually add the traceback header line to your test example.
457
458* For some :exc:`SyntaxError`\ s, Python displays the character position of the
459 syntax error, using a ``^`` marker::
460
461 >>> 1 1
462 File "<stdin>", line 1
463 1 1
464 ^
465 SyntaxError: invalid syntax
466
467 Since the lines showing the position of the error come before the exception type
468 and detail, they are not checked by doctest. For example, the following test
469 would pass, even though it puts the ``^`` marker in the wrong location::
470
471 >>> 1 1
472 Traceback (most recent call last):
473 File "<stdin>", line 1
474 1 1
475 ^
476 SyntaxError: invalid syntax
477
Georg Brandl116aa622007-08-15 14:28:22 +0000478
479.. _doctest-options:
480
481Option Flags and Directives
482^^^^^^^^^^^^^^^^^^^^^^^^^^^
483
484A number of option flags control various aspects of doctest's behavior.
485Symbolic names for the flags are supplied as module constants, which can be
486or'ed together and passed to various functions. The names can also be used in
487doctest directives (see below).
488
489The first group of options define test semantics, controlling aspects of how
490doctest decides whether actual output matches an example's expected output:
491
492
493.. data:: DONT_ACCEPT_TRUE_FOR_1
494
495 By default, if an expected output block contains just ``1``, an actual output
496 block containing just ``1`` or just ``True`` is considered to be a match, and
497 similarly for ``0`` versus ``False``. When :const:`DONT_ACCEPT_TRUE_FOR_1` is
498 specified, neither substitution is allowed. The default behavior caters to that
499 Python changed the return type of many functions from integer to boolean;
500 doctests expecting "little integer" output still work in these cases. This
501 option will probably go away, but not for several years.
502
503
504.. data:: DONT_ACCEPT_BLANKLINE
505
506 By default, if an expected output block contains a line containing only the
507 string ``<BLANKLINE>``, then that line will match a blank line in the actual
508 output. Because a genuinely blank line delimits the expected output, this is
509 the only way to communicate that a blank line is expected. When
510 :const:`DONT_ACCEPT_BLANKLINE` is specified, this substitution is not allowed.
511
512
513.. data:: NORMALIZE_WHITESPACE
514
515 When specified, all sequences of whitespace (blanks and newlines) are treated as
516 equal. Any sequence of whitespace within the expected output will match any
517 sequence of whitespace within the actual output. By default, whitespace must
518 match exactly. :const:`NORMALIZE_WHITESPACE` is especially useful when a line of
519 expected output is very long, and you want to wrap it across multiple lines in
520 your source.
521
522
523.. data:: ELLIPSIS
524
525 When specified, an ellipsis marker (``...``) in the expected output can match
526 any substring in the actual output. This includes substrings that span line
527 boundaries, and empty substrings, so it's best to keep usage of this simple.
528 Complicated uses can lead to the same kinds of "oops, it matched too much!"
529 surprises that ``.*`` is prone to in regular expressions.
530
531
532.. data:: IGNORE_EXCEPTION_DETAIL
533
534 When specified, an example that expects an exception passes if an exception of
535 the expected type is raised, even if the exception detail does not match. For
536 example, an example expecting ``ValueError: 42`` will pass if the actual
537 exception raised is ``ValueError: 3*14``, but will fail, e.g., if
538 :exc:`TypeError` is raised.
539
540 Note that a similar effect can be obtained using :const:`ELLIPSIS`, and
541 :const:`IGNORE_EXCEPTION_DETAIL` may go away when Python releases prior to 2.4
542 become uninteresting. Until then, :const:`IGNORE_EXCEPTION_DETAIL` is the only
543 clear way to write a doctest that doesn't care about the exception detail yet
544 continues to pass under Python releases prior to 2.4 (doctest directives appear
545 to be comments to them). For example, ::
546
547 >>> (1, 2)[3] = 'moo' #doctest: +IGNORE_EXCEPTION_DETAIL
548 Traceback (most recent call last):
549 File "<stdin>", line 1, in ?
550 TypeError: object doesn't support item assignment
551
552 passes under Python 2.4 and Python 2.3. The detail changed in 2.4, to say "does
553 not" instead of "doesn't".
554
555
556.. data:: SKIP
557
558 When specified, do not run the example at all. This can be useful in contexts
559 where doctest examples serve as both documentation and test cases, and an
560 example should be included for documentation purposes, but should not be
561 checked. E.g., the example's output might be random; or the example might
562 depend on resources which would be unavailable to the test driver.
563
564 The SKIP flag can also be used for temporarily "commenting out" examples.
565
566
567.. data:: COMPARISON_FLAGS
568
569 A bitmask or'ing together all the comparison flags above.
570
571The second group of options controls how test failures are reported:
572
573
574.. data:: REPORT_UDIFF
575
576 When specified, failures that involve multi-line expected and actual outputs are
577 displayed using a unified diff.
578
579
580.. data:: REPORT_CDIFF
581
582 When specified, failures that involve multi-line expected and actual outputs
583 will be displayed using a context diff.
584
585
586.. data:: REPORT_NDIFF
587
588 When specified, differences are computed by ``difflib.Differ``, using the same
589 algorithm as the popular :file:`ndiff.py` utility. This is the only method that
590 marks differences within lines as well as across lines. For example, if a line
591 of expected output contains digit ``1`` where actual output contains letter
592 ``l``, a line is inserted with a caret marking the mismatching column positions.
593
594
595.. data:: REPORT_ONLY_FIRST_FAILURE
596
597 When specified, display the first failing example in each doctest, but suppress
598 output for all remaining examples. This will prevent doctest from reporting
599 correct examples that break because of earlier failures; but it might also hide
600 incorrect examples that fail independently of the first failure. When
601 :const:`REPORT_ONLY_FIRST_FAILURE` is specified, the remaining examples are
602 still run, and still count towards the total number of failures reported; only
603 the output is suppressed.
604
605
606.. data:: REPORTING_FLAGS
607
608 A bitmask or'ing together all the reporting flags above.
609
610"Doctest directives" may be used to modify the option flags for individual
611examples. Doctest directives are expressed as a special Python comment
612following an example's source code:
613
614.. productionlist:: doctest
615 directive: "#" "doctest:" `directive_options`
616 directive_options: `directive_option` ("," `directive_option`)\*
617 directive_option: `on_or_off` `directive_option_name`
618 on_or_off: "+" \| "-"
619 directive_option_name: "DONT_ACCEPT_BLANKLINE" \| "NORMALIZE_WHITESPACE" \| ...
620
621Whitespace is not allowed between the ``+`` or ``-`` and the directive option
622name. The directive option name can be any of the option flag names explained
623above.
624
625An example's doctest directives modify doctest's behavior for that single
626example. Use ``+`` to enable the named behavior, or ``-`` to disable it.
627
628For example, this test passes::
629
Georg Brandl6911e3c2007-09-04 07:15:32 +0000630 >>> print(range(20)) #doctest: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000631 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9,
632 10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
633
634Without the directive it would fail, both because the actual output doesn't have
635two blanks before the single-digit list elements, and because the actual output
636is on a single line. This test also passes, and also requires a directive to do
637so::
638
Georg Brandl6911e3c2007-09-04 07:15:32 +0000639 >>> print(range(20)) # doctest: +ELLIPSIS
Georg Brandl116aa622007-08-15 14:28:22 +0000640 [0, 1, ..., 18, 19]
641
642Multiple directives can be used on a single physical line, separated by commas::
643
Georg Brandl6911e3c2007-09-04 07:15:32 +0000644 >>> print(range(20)) # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000645 [0, 1, ..., 18, 19]
646
647If multiple directive comments are used for a single example, then they are
648combined::
649
Georg Brandl6911e3c2007-09-04 07:15:32 +0000650 >>> print(range(20)) # doctest: +ELLIPSIS
Georg Brandl116aa622007-08-15 14:28:22 +0000651 ... # doctest: +NORMALIZE_WHITESPACE
652 [0, 1, ..., 18, 19]
653
654As the previous example shows, you can add ``...`` lines to your example
655containing only directives. This can be useful when an example is too long for
656a directive to comfortably fit on the same line::
657
Georg Brandl6911e3c2007-09-04 07:15:32 +0000658 >>> print(range(5) + range(10,20) + range(30,40) + range(50,60))
Georg Brandl116aa622007-08-15 14:28:22 +0000659 ... # doctest: +ELLIPSIS
660 [0, ..., 4, 10, ..., 19, 30, ..., 39, 50, ..., 59]
661
662Note that since all options are disabled by default, and directives apply only
663to the example they appear in, enabling options (via ``+`` in a directive) is
664usually the only meaningful choice. However, option flags can also be passed to
665functions that run doctests, establishing different defaults. In such cases,
666disabling an option via ``-`` in a directive can be useful.
667
Georg Brandl116aa622007-08-15 14:28:22 +0000668
669There's also a way to register new option flag names, although this isn't useful
670unless you intend to extend :mod:`doctest` internals via subclassing:
671
672
673.. function:: register_optionflag(name)
674
675 Create a new option flag with a given name, and return the new flag's integer
676 value. :func:`register_optionflag` can be used when subclassing
677 :class:`OutputChecker` or :class:`DocTestRunner` to create new options that are
678 supported by your subclasses. :func:`register_optionflag` should always be
679 called using the following idiom::
680
681 MY_FLAG = register_optionflag('MY_FLAG')
682
Georg Brandl116aa622007-08-15 14:28:22 +0000683
684.. _doctest-warnings:
685
686Warnings
687^^^^^^^^
688
689:mod:`doctest` is serious about requiring exact matches in expected output. If
690even a single character doesn't match, the test fails. This will probably
691surprise you a few times, as you learn exactly what Python does and doesn't
692guarantee about output. For example, when printing a dict, Python doesn't
693guarantee that the key-value pairs will be printed in any particular order, so a
Christian Heimes5b5e81c2007-12-31 16:14:33 +0000694test like ::
Georg Brandl116aa622007-08-15 14:28:22 +0000695
696 >>> foo()
697 {"Hermione": "hippogryph", "Harry": "broomstick"}
698
699is vulnerable! One workaround is to do ::
700
701 >>> foo() == {"Hermione": "hippogryph", "Harry": "broomstick"}
702 True
703
704instead. Another is to do ::
705
706 >>> d = foo().items()
707 >>> d.sort()
708 >>> d
709 [('Harry', 'broomstick'), ('Hermione', 'hippogryph')]
710
711There are others, but you get the idea.
712
713Another bad idea is to print things that embed an object address, like ::
714
715 >>> id(1.0) # certain to fail some of the time
716 7948648
717 >>> class C: pass
718 >>> C() # the default repr() for instances embeds an address
719 <__main__.C instance at 0x00AC18F0>
720
721The :const:`ELLIPSIS` directive gives a nice approach for the last example::
722
723 >>> C() #doctest: +ELLIPSIS
724 <__main__.C instance at 0x...>
725
726Floating-point numbers are also subject to small output variations across
727platforms, because Python defers to the platform C library for float formatting,
728and C libraries vary widely in quality here. ::
729
730 >>> 1./7 # risky
731 0.14285714285714285
Georg Brandl6911e3c2007-09-04 07:15:32 +0000732 >>> print(1./7) # safer
Georg Brandl116aa622007-08-15 14:28:22 +0000733 0.142857142857
Georg Brandl6911e3c2007-09-04 07:15:32 +0000734 >>> print(round(1./7, 6)) # much safer
Georg Brandl116aa622007-08-15 14:28:22 +0000735 0.142857
736
737Numbers of the form ``I/2.**J`` are safe across all platforms, and I often
738contrive doctest examples to produce numbers of that form::
739
740 >>> 3./4 # utterly safe
741 0.75
742
743Simple fractions are also easier for people to understand, and that makes for
744better documentation.
745
746
747.. _doctest-basic-api:
748
749Basic API
750---------
751
752The functions :func:`testmod` and :func:`testfile` provide a simple interface to
753doctest that should be sufficient for most basic uses. For a less formal
754introduction to these two functions, see sections :ref:`doctest-simple-testmod`
755and :ref:`doctest-simple-testfile`.
756
757
758.. function:: testfile(filename[, module_relative][, name][, package][, globs][, verbose][, report][, optionflags][, extraglobs][, raise_on_error][, parser][, encoding])
759
760 All arguments except *filename* are optional, and should be specified in keyword
761 form.
762
763 Test examples in the file named *filename*. Return ``(failure_count,
764 test_count)``.
765
766 Optional argument *module_relative* specifies how the filename should be
767 interpreted:
768
769 * If *module_relative* is ``True`` (the default), then *filename* specifies an
770 OS-independent module-relative path. By default, this path is relative to the
771 calling module's directory; but if the *package* argument is specified, then it
772 is relative to that package. To ensure OS-independence, *filename* should use
773 ``/`` characters to separate path segments, and may not be an absolute path
774 (i.e., it may not begin with ``/``).
775
776 * If *module_relative* is ``False``, then *filename* specifies an OS-specific
777 path. The path may be absolute or relative; relative paths are resolved with
778 respect to the current working directory.
779
780 Optional argument *name* gives the name of the test; by default, or if ``None``,
781 ``os.path.basename(filename)`` is used.
782
783 Optional argument *package* is a Python package or the name of a Python package
784 whose directory should be used as the base directory for a module-relative
785 filename. If no package is specified, then the calling module's directory is
786 used as the base directory for module-relative filenames. It is an error to
787 specify *package* if *module_relative* is ``False``.
788
789 Optional argument *globs* gives a dict to be used as the globals when executing
790 examples. A new shallow copy of this dict is created for the doctest, so its
791 examples start with a clean slate. By default, or if ``None``, a new empty dict
792 is used.
793
794 Optional argument *extraglobs* gives a dict merged into the globals used to
795 execute examples. This works like :meth:`dict.update`: if *globs* and
796 *extraglobs* have a common key, the associated value in *extraglobs* appears in
797 the combined dict. By default, or if ``None``, no extra globals are used. This
798 is an advanced feature that allows parameterization of doctests. For example, a
799 doctest can be written for a base class, using a generic name for the class,
800 then reused to test any number of subclasses by passing an *extraglobs* dict
801 mapping the generic name to the subclass to be tested.
802
803 Optional argument *verbose* prints lots of stuff if true, and prints only
804 failures if false; by default, or if ``None``, it's true if and only if ``'-v'``
805 is in ``sys.argv``.
806
807 Optional argument *report* prints a summary at the end when true, else prints
808 nothing at the end. In verbose mode, the summary is detailed, else the summary
809 is very brief (in fact, empty if all tests passed).
810
811 Optional argument *optionflags* or's together option flags. See section
812 :ref:`doctest-options`.
813
814 Optional argument *raise_on_error* defaults to false. If true, an exception is
815 raised upon the first failure or unexpected exception in an example. This
816 allows failures to be post-mortem debugged. Default behavior is to continue
817 running examples.
818
819 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass) that
820 should be used to extract tests from the files. It defaults to a normal parser
821 (i.e., ``DocTestParser()``).
822
823 Optional argument *encoding* specifies an encoding that should be used to
824 convert the file to unicode.
825
Georg Brandl116aa622007-08-15 14:28:22 +0000826
827.. function:: testmod([m][, name][, globs][, verbose][, report][, optionflags][, extraglobs][, raise_on_error][, exclude_empty])
828
829 All arguments are optional, and all except for *m* should be specified in
830 keyword form.
831
832 Test examples in docstrings in functions and classes reachable from module *m*
833 (or module :mod:`__main__` if *m* is not supplied or is ``None``), starting with
834 ``m.__doc__``.
835
836 Also test examples reachable from dict ``m.__test__``, if it exists and is not
837 ``None``. ``m.__test__`` maps names (strings) to functions, classes and
838 strings; function and class docstrings are searched for examples; strings are
839 searched directly, as if they were docstrings.
840
841 Only docstrings attached to objects belonging to module *m* are searched.
842
843 Return ``(failure_count, test_count)``.
844
845 Optional argument *name* gives the name of the module; by default, or if
846 ``None``, ``m.__name__`` is used.
847
848 Optional argument *exclude_empty* defaults to false. If true, objects for which
849 no doctests are found are excluded from consideration. The default is a backward
850 compatibility hack, so that code still using :meth:`doctest.master.summarize` in
851 conjunction with :func:`testmod` continues to get output for objects with no
852 tests. The *exclude_empty* argument to the newer :class:`DocTestFinder`
853 constructor defaults to true.
854
855 Optional arguments *extraglobs*, *verbose*, *report*, *optionflags*,
856 *raise_on_error*, and *globs* are the same as for function :func:`testfile`
857 above, except that *globs* defaults to ``m.__dict__``.
858
Georg Brandl116aa622007-08-15 14:28:22 +0000859
860There's also a function to run the doctests associated with a single object.
861This function is provided for backward compatibility. There are no plans to
862deprecate it, but it's rarely useful:
863
864
865.. function:: run_docstring_examples(f, globs[, verbose][, name][, compileflags][, optionflags])
866
867 Test examples associated with object *f*; for example, *f* may be a module,
868 function, or class object.
869
870 A shallow copy of dictionary argument *globs* is used for the execution context.
871
872 Optional argument *name* is used in failure messages, and defaults to
873 ``"NoName"``.
874
875 If optional argument *verbose* is true, output is generated even if there are no
876 failures. By default, output is generated only in case of an example failure.
877
878 Optional argument *compileflags* gives the set of flags that should be used by
879 the Python compiler when running the examples. By default, or if ``None``,
880 flags are deduced corresponding to the set of future features found in *globs*.
881
882 Optional argument *optionflags* works as for function :func:`testfile` above.
883
884
885.. _doctest-unittest-api:
886
887Unittest API
888------------
889
890As your collection of doctest'ed modules grows, you'll want a way to run all
891their doctests systematically. Prior to Python 2.4, :mod:`doctest` had a barely
892documented :class:`Tester` class that supplied a rudimentary way to combine
893doctests from multiple modules. :class:`Tester` was feeble, and in practice most
894serious Python testing frameworks build on the :mod:`unittest` module, which
895supplies many flexible ways to combine tests from multiple sources. So, in
896Python 2.4, :mod:`doctest`'s :class:`Tester` class is deprecated, and
897:mod:`doctest` provides two functions that can be used to create :mod:`unittest`
898test suites from modules and text files containing doctests. These test suites
899can then be run using :mod:`unittest` test runners::
900
901 import unittest
902 import doctest
903 import my_module_with_doctests, and_another
904
905 suite = unittest.TestSuite()
906 for mod in my_module_with_doctests, and_another:
907 suite.addTest(doctest.DocTestSuite(mod))
908 runner = unittest.TextTestRunner()
909 runner.run(suite)
910
911There are two main functions for creating :class:`unittest.TestSuite` instances
912from text files and modules with doctests:
913
914
915.. function:: DocFileSuite([module_relative][, package][, setUp][, tearDown][, globs][, optionflags][, parser][, encoding])
916
917 Convert doctest tests from one or more text files to a
918 :class:`unittest.TestSuite`.
919
920 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
921 and runs the interactive examples in each file. If an example in any file
922 fails, then the synthesized unit test fails, and a :exc:`failureException`
923 exception is raised showing the name of the file containing the test and a
924 (sometimes approximate) line number.
925
926 Pass one or more paths (as strings) to text files to be examined.
927
928 Options may be provided as keyword arguments:
929
930 Optional argument *module_relative* specifies how the filenames in *paths*
931 should be interpreted:
932
933 * If *module_relative* is ``True`` (the default), then each filename specifies
934 an OS-independent module-relative path. By default, this path is relative to
935 the calling module's directory; but if the *package* argument is specified, then
936 it is relative to that package. To ensure OS-independence, each filename should
937 use ``/`` characters to separate path segments, and may not be an absolute path
938 (i.e., it may not begin with ``/``).
939
940 * If *module_relative* is ``False``, then each filename specifies an OS-specific
941 path. The path may be absolute or relative; relative paths are resolved with
942 respect to the current working directory.
943
944 Optional argument *package* is a Python package or the name of a Python package
945 whose directory should be used as the base directory for module-relative
946 filenames. If no package is specified, then the calling module's directory is
947 used as the base directory for module-relative filenames. It is an error to
948 specify *package* if *module_relative* is ``False``.
949
950 Optional argument *setUp* specifies a set-up function for the test suite. This
951 is called before running the tests in each file. The *setUp* function will be
952 passed a :class:`DocTest` object. The setUp function can access the test
953 globals as the *globs* attribute of the test passed.
954
955 Optional argument *tearDown* specifies a tear-down function for the test suite.
956 This is called after running the tests in each file. The *tearDown* function
957 will be passed a :class:`DocTest` object. The setUp function can access the
958 test globals as the *globs* attribute of the test passed.
959
960 Optional argument *globs* is a dictionary containing the initial global
961 variables for the tests. A new copy of this dictionary is created for each
962 test. By default, *globs* is a new empty dictionary.
963
964 Optional argument *optionflags* specifies the default doctest options for the
965 tests, created by or-ing together individual option flags. See section
966 :ref:`doctest-options`. See function :func:`set_unittest_reportflags` below for
967 a better way to set reporting options.
968
969 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass) that
970 should be used to extract tests from the files. It defaults to a normal parser
971 (i.e., ``DocTestParser()``).
972
973 Optional argument *encoding* specifies an encoding that should be used to
974 convert the file to unicode.
975
Georg Brandl55ac8f02007-09-01 13:51:09 +0000976 The global ``__file__`` is added to the globals provided to doctests loaded
977 from a text file using :func:`DocFileSuite`.
Georg Brandl116aa622007-08-15 14:28:22 +0000978
979
980.. function:: DocTestSuite([module][, globs][, extraglobs][, test_finder][, setUp][, tearDown][, checker])
981
982 Convert doctest tests for a module to a :class:`unittest.TestSuite`.
983
984 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
985 and runs each doctest in the module. If any of the doctests fail, then the
986 synthesized unit test fails, and a :exc:`failureException` exception is raised
987 showing the name of the file containing the test and a (sometimes approximate)
988 line number.
989
990 Optional argument *module* provides the module to be tested. It can be a module
991 object or a (possibly dotted) module name. If not specified, the module calling
992 this function is used.
993
994 Optional argument *globs* is a dictionary containing the initial global
995 variables for the tests. A new copy of this dictionary is created for each
996 test. By default, *globs* is a new empty dictionary.
997
998 Optional argument *extraglobs* specifies an extra set of global variables, which
999 is merged into *globs*. By default, no extra globals are used.
1000
1001 Optional argument *test_finder* is the :class:`DocTestFinder` object (or a
1002 drop-in replacement) that is used to extract doctests from the module.
1003
1004 Optional arguments *setUp*, *tearDown*, and *optionflags* are the same as for
1005 function :func:`DocFileSuite` above.
1006
Georg Brandl55ac8f02007-09-01 13:51:09 +00001007 This function uses the same search technique as :func:`testmod`.
Georg Brandl116aa622007-08-15 14:28:22 +00001008
Georg Brandl116aa622007-08-15 14:28:22 +00001009
1010Under the covers, :func:`DocTestSuite` creates a :class:`unittest.TestSuite` out
1011of :class:`doctest.DocTestCase` instances, and :class:`DocTestCase` is a
1012subclass of :class:`unittest.TestCase`. :class:`DocTestCase` isn't documented
1013here (it's an internal detail), but studying its code can answer questions about
1014the exact details of :mod:`unittest` integration.
1015
1016Similarly, :func:`DocFileSuite` creates a :class:`unittest.TestSuite` out of
1017:class:`doctest.DocFileCase` instances, and :class:`DocFileCase` is a subclass
1018of :class:`DocTestCase`.
1019
1020So both ways of creating a :class:`unittest.TestSuite` run instances of
1021:class:`DocTestCase`. This is important for a subtle reason: when you run
1022:mod:`doctest` functions yourself, you can control the :mod:`doctest` options in
1023use directly, by passing option flags to :mod:`doctest` functions. However, if
1024you're writing a :mod:`unittest` framework, :mod:`unittest` ultimately controls
1025when and how tests get run. The framework author typically wants to control
1026:mod:`doctest` reporting options (perhaps, e.g., specified by command line
1027options), but there's no way to pass options through :mod:`unittest` to
1028:mod:`doctest` test runners.
1029
1030For this reason, :mod:`doctest` also supports a notion of :mod:`doctest`
1031reporting flags specific to :mod:`unittest` support, via this function:
1032
1033
1034.. function:: set_unittest_reportflags(flags)
1035
1036 Set the :mod:`doctest` reporting flags to use.
1037
1038 Argument *flags* or's together option flags. See section
1039 :ref:`doctest-options`. Only "reporting flags" can be used.
1040
1041 This is a module-global setting, and affects all future doctests run by module
1042 :mod:`unittest`: the :meth:`runTest` method of :class:`DocTestCase` looks at
1043 the option flags specified for the test case when the :class:`DocTestCase`
1044 instance was constructed. If no reporting flags were specified (which is the
1045 typical and expected case), :mod:`doctest`'s :mod:`unittest` reporting flags are
1046 or'ed into the option flags, and the option flags so augmented are passed to the
1047 :class:`DocTestRunner` instance created to run the doctest. If any reporting
1048 flags were specified when the :class:`DocTestCase` instance was constructed,
1049 :mod:`doctest`'s :mod:`unittest` reporting flags are ignored.
1050
1051 The value of the :mod:`unittest` reporting flags in effect before the function
1052 was called is returned by the function.
1053
Georg Brandl116aa622007-08-15 14:28:22 +00001054
1055.. _doctest-advanced-api:
1056
1057Advanced API
1058------------
1059
1060The basic API is a simple wrapper that's intended to make doctest easy to use.
1061It is fairly flexible, and should meet most users' needs; however, if you
1062require more fine-grained control over testing, or wish to extend doctest's
1063capabilities, then you should use the advanced API.
1064
1065The advanced API revolves around two container classes, which are used to store
1066the interactive examples extracted from doctest cases:
1067
Christian Heimesd8654cf2007-12-02 15:22:16 +00001068* :class:`Example`: A single python :term:`statement`, paired with its expected
1069 output.
Georg Brandl116aa622007-08-15 14:28:22 +00001070
1071* :class:`DocTest`: A collection of :class:`Example`\ s, typically extracted
1072 from a single docstring or text file.
1073
1074Additional processing classes are defined to find, parse, and run, and check
1075doctest examples:
1076
1077* :class:`DocTestFinder`: Finds all docstrings in a given module, and uses a
1078 :class:`DocTestParser` to create a :class:`DocTest` from every docstring that
1079 contains interactive examples.
1080
1081* :class:`DocTestParser`: Creates a :class:`DocTest` object from a string (such
1082 as an object's docstring).
1083
1084* :class:`DocTestRunner`: Executes the examples in a :class:`DocTest`, and uses
1085 an :class:`OutputChecker` to verify their output.
1086
1087* :class:`OutputChecker`: Compares the actual output from a doctest example with
1088 the expected output, and decides whether they match.
1089
1090The relationships among these processing classes are summarized in the following
1091diagram::
1092
1093 list of:
1094 +------+ +---------+
1095 |module| --DocTestFinder-> | DocTest | --DocTestRunner-> results
1096 +------+ | ^ +---------+ | ^ (printed)
1097 | | | Example | | |
1098 v | | ... | v |
1099 DocTestParser | Example | OutputChecker
1100 +---------+
1101
1102
1103.. _doctest-doctest:
1104
1105DocTest Objects
1106^^^^^^^^^^^^^^^
1107
1108
1109.. class:: DocTest(examples, globs, name, filename, lineno, docstring)
1110
1111 A collection of doctest examples that should be run in a single namespace. The
1112 constructor arguments are used to initialize the member variables of the same
1113 names.
1114
Georg Brandl116aa622007-08-15 14:28:22 +00001115
1116:class:`DocTest` defines the following member variables. They are initialized
1117by the constructor, and should not be modified directly.
1118
1119
1120.. attribute:: DocTest.examples
1121
1122 A list of :class:`Example` objects encoding the individual interactive Python
1123 examples that should be run by this test.
1124
1125
1126.. attribute:: DocTest.globs
1127
1128 The namespace (aka globals) that the examples should be run in. This is a
1129 dictionary mapping names to values. Any changes to the namespace made by the
1130 examples (such as binding new variables) will be reflected in :attr:`globs`
1131 after the test is run.
1132
1133
1134.. attribute:: DocTest.name
1135
1136 A string name identifying the :class:`DocTest`. Typically, this is the name of
1137 the object or file that the test was extracted from.
1138
1139
1140.. attribute:: DocTest.filename
1141
1142 The name of the file that this :class:`DocTest` was extracted from; or ``None``
1143 if the filename is unknown, or if the :class:`DocTest` was not extracted from a
1144 file.
1145
1146
1147.. attribute:: DocTest.lineno
1148
1149 The line number within :attr:`filename` where this :class:`DocTest` begins, or
1150 ``None`` if the line number is unavailable. This line number is zero-based with
1151 respect to the beginning of the file.
1152
1153
1154.. attribute:: DocTest.docstring
1155
1156 The string that the test was extracted from, or 'None' if the string is
1157 unavailable, or if the test was not extracted from a string.
1158
1159
1160.. _doctest-example:
1161
1162Example Objects
1163^^^^^^^^^^^^^^^
1164
1165
1166.. class:: Example(source, want[, exc_msg][, lineno][, indent][, options])
1167
1168 A single interactive example, consisting of a Python statement and its expected
1169 output. The constructor arguments are used to initialize the member variables
1170 of the same names.
1171
Georg Brandl116aa622007-08-15 14:28:22 +00001172
1173:class:`Example` defines the following member variables. They are initialized
1174by the constructor, and should not be modified directly.
1175
1176
1177.. attribute:: Example.source
1178
1179 A string containing the example's source code. This source code consists of a
1180 single Python statement, and always ends with a newline; the constructor adds a
1181 newline when necessary.
1182
1183
1184.. attribute:: Example.want
1185
1186 The expected output from running the example's source code (either from stdout,
1187 or a traceback in case of exception). :attr:`want` ends with a newline unless
1188 no output is expected, in which case it's an empty string. The constructor adds
1189 a newline when necessary.
1190
1191
1192.. attribute:: Example.exc_msg
1193
1194 The exception message generated by the example, if the example is expected to
1195 generate an exception; or ``None`` if it is not expected to generate an
1196 exception. This exception message is compared against the return value of
1197 :func:`traceback.format_exception_only`. :attr:`exc_msg` ends with a newline
1198 unless it's ``None``. The constructor adds a newline if needed.
1199
1200
1201.. attribute:: Example.lineno
1202
1203 The line number within the string containing this example where the example
1204 begins. This line number is zero-based with respect to the beginning of the
1205 containing string.
1206
1207
1208.. attribute:: Example.indent
1209
1210 The example's indentation in the containing string, i.e., the number of space
1211 characters that precede the example's first prompt.
1212
1213
1214.. attribute:: Example.options
1215
1216 A dictionary mapping from option flags to ``True`` or ``False``, which is used
1217 to override default options for this example. Any option flags not contained in
1218 this dictionary are left at their default value (as specified by the
1219 :class:`DocTestRunner`'s :attr:`optionflags`). By default, no options are set.
1220
1221
1222.. _doctest-doctestfinder:
1223
1224DocTestFinder objects
1225^^^^^^^^^^^^^^^^^^^^^
1226
1227
1228.. class:: DocTestFinder([verbose][, parser][, recurse][, exclude_empty])
1229
1230 A processing class used to extract the :class:`DocTest`\ s that are relevant to
1231 a given object, from its docstring and the docstrings of its contained objects.
1232 :class:`DocTest`\ s can currently be extracted from the following object types:
1233 modules, functions, classes, methods, staticmethods, classmethods, and
1234 properties.
1235
1236 The optional argument *verbose* can be used to display the objects searched by
1237 the finder. It defaults to ``False`` (no output).
1238
1239 The optional argument *parser* specifies the :class:`DocTestParser` object (or a
1240 drop-in replacement) that is used to extract doctests from docstrings.
1241
1242 If the optional argument *recurse* is false, then :meth:`DocTestFinder.find`
1243 will only examine the given object, and not any contained objects.
1244
1245 If the optional argument *exclude_empty* is false, then
1246 :meth:`DocTestFinder.find` will include tests for objects with empty docstrings.
1247
Georg Brandl116aa622007-08-15 14:28:22 +00001248
1249:class:`DocTestFinder` defines the following method:
1250
1251
1252.. method:: DocTestFinder.find(obj[, name][, module][, globs][, extraglobs])
1253
1254 Return a list of the :class:`DocTest`\ s that are defined by *obj*'s docstring,
1255 or by any of its contained objects' docstrings.
1256
1257 The optional argument *name* specifies the object's name; this name will be used
1258 to construct names for the returned :class:`DocTest`\ s. If *name* is not
1259 specified, then ``obj.__name__`` is used.
1260
1261 The optional parameter *module* is the module that contains the given object.
1262 If the module is not specified or is None, then the test finder will attempt to
1263 automatically determine the correct module. The object's module is used:
1264
1265 * As a default namespace, if *globs* is not specified.
1266
1267 * To prevent the DocTestFinder from extracting DocTests from objects that are
1268 imported from other modules. (Contained objects with modules other than
1269 *module* are ignored.)
1270
1271 * To find the name of the file containing the object.
1272
1273 * To help find the line number of the object within its file.
1274
1275 If *module* is ``False``, no attempt to find the module will be made. This is
1276 obscure, of use mostly in testing doctest itself: if *module* is ``False``, or
1277 is ``None`` but cannot be found automatically, then all objects are considered
1278 to belong to the (non-existent) module, so all contained objects will
1279 (recursively) be searched for doctests.
1280
1281 The globals for each :class:`DocTest` is formed by combining *globs* and
1282 *extraglobs* (bindings in *extraglobs* override bindings in *globs*). A new
1283 shallow copy of the globals dictionary is created for each :class:`DocTest`. If
1284 *globs* is not specified, then it defaults to the module's *__dict__*, if
1285 specified, or ``{}`` otherwise. If *extraglobs* is not specified, then it
1286 defaults to ``{}``.
1287
1288
1289.. _doctest-doctestparser:
1290
1291DocTestParser objects
1292^^^^^^^^^^^^^^^^^^^^^
1293
1294
1295.. class:: DocTestParser()
1296
1297 A processing class used to extract interactive examples from a string, and use
1298 them to create a :class:`DocTest` object.
1299
Georg Brandl116aa622007-08-15 14:28:22 +00001300
1301:class:`DocTestParser` defines the following methods:
1302
1303
1304.. method:: DocTestParser.get_doctest(string, globs, name, filename, lineno)
1305
1306 Extract all doctest examples from the given string, and collect them into a
1307 :class:`DocTest` object.
1308
1309 *globs*, *name*, *filename*, and *lineno* are attributes for the new
1310 :class:`DocTest` object. See the documentation for :class:`DocTest` for more
1311 information.
1312
1313
1314.. method:: DocTestParser.get_examples(string[, name])
1315
1316 Extract all doctest examples from the given string, and return them as a list of
1317 :class:`Example` objects. Line numbers are 0-based. The optional argument
1318 *name* is a name identifying this string, and is only used for error messages.
1319
1320
1321.. method:: DocTestParser.parse(string[, name])
1322
1323 Divide the given string into examples and intervening text, and return them as a
1324 list of alternating :class:`Example`\ s and strings. Line numbers for the
1325 :class:`Example`\ s are 0-based. The optional argument *name* is a name
1326 identifying this string, and is only used for error messages.
1327
1328
1329.. _doctest-doctestrunner:
1330
1331DocTestRunner objects
1332^^^^^^^^^^^^^^^^^^^^^
1333
1334
1335.. class:: DocTestRunner([checker][, verbose][, optionflags])
1336
1337 A processing class used to execute and verify the interactive examples in a
1338 :class:`DocTest`.
1339
1340 The comparison between expected outputs and actual outputs is done by an
1341 :class:`OutputChecker`. This comparison may be customized with a number of
1342 option flags; see section :ref:`doctest-options` for more information. If the
1343 option flags are insufficient, then the comparison may also be customized by
1344 passing a subclass of :class:`OutputChecker` to the constructor.
1345
1346 The test runner's display output can be controlled in two ways. First, an output
1347 function can be passed to :meth:`TestRunner.run`; this function will be called
1348 with strings that should be displayed. It defaults to ``sys.stdout.write``. If
1349 capturing the output is not sufficient, then the display output can be also
1350 customized by subclassing DocTestRunner, and overriding the methods
1351 :meth:`report_start`, :meth:`report_success`,
1352 :meth:`report_unexpected_exception`, and :meth:`report_failure`.
1353
1354 The optional keyword argument *checker* specifies the :class:`OutputChecker`
1355 object (or drop-in replacement) that should be used to compare the expected
1356 outputs to the actual outputs of doctest examples.
1357
1358 The optional keyword argument *verbose* controls the :class:`DocTestRunner`'s
1359 verbosity. If *verbose* is ``True``, then information is printed about each
1360 example, as it is run. If *verbose* is ``False``, then only failures are
1361 printed. If *verbose* is unspecified, or ``None``, then verbose output is used
1362 iff the command-line switch :option:`-v` is used.
1363
1364 The optional keyword argument *optionflags* can be used to control how the test
1365 runner compares expected output to actual output, and how it displays failures.
1366 For more information, see section :ref:`doctest-options`.
1367
Georg Brandl116aa622007-08-15 14:28:22 +00001368
1369:class:`DocTestParser` defines the following methods:
1370
1371
1372.. method:: DocTestRunner.report_start(out, test, example)
1373
1374 Report that the test runner is about to process the given example. This method
1375 is provided to allow subclasses of :class:`DocTestRunner` to customize their
1376 output; it should not be called directly.
1377
1378 *example* is the example about to be processed. *test* is the test containing
1379 *example*. *out* is the output function that was passed to
1380 :meth:`DocTestRunner.run`.
1381
1382
1383.. method:: DocTestRunner.report_success(out, test, example, got)
1384
1385 Report that the given example ran successfully. This method is provided to
1386 allow subclasses of :class:`DocTestRunner` to customize their output; it should
1387 not be called directly.
1388
1389 *example* is the example about to be processed. *got* is the actual output from
1390 the example. *test* is the test containing *example*. *out* is the output
1391 function that was passed to :meth:`DocTestRunner.run`.
1392
1393
1394.. method:: DocTestRunner.report_failure(out, test, example, got)
1395
1396 Report that the given example failed. This method is provided to allow
1397 subclasses of :class:`DocTestRunner` to customize their output; it should not be
1398 called directly.
1399
1400 *example* is the example about to be processed. *got* is the actual output from
1401 the example. *test* is the test containing *example*. *out* is the output
1402 function that was passed to :meth:`DocTestRunner.run`.
1403
1404
1405.. method:: DocTestRunner.report_unexpected_exception(out, test, example, exc_info)
1406
1407 Report that the given example raised an unexpected exception. This method is
1408 provided to allow subclasses of :class:`DocTestRunner` to customize their
1409 output; it should not be called directly.
1410
1411 *example* is the example about to be processed. *exc_info* is a tuple containing
1412 information about the unexpected exception (as returned by
1413 :func:`sys.exc_info`). *test* is the test containing *example*. *out* is the
1414 output function that was passed to :meth:`DocTestRunner.run`.
1415
1416
1417.. method:: DocTestRunner.run(test[, compileflags][, out][, clear_globs])
1418
1419 Run the examples in *test* (a :class:`DocTest` object), and display the results
1420 using the writer function *out*.
1421
1422 The examples are run in the namespace ``test.globs``. If *clear_globs* is true
1423 (the default), then this namespace will be cleared after the test runs, to help
1424 with garbage collection. If you would like to examine the namespace after the
1425 test completes, then use *clear_globs=False*.
1426
1427 *compileflags* gives the set of flags that should be used by the Python compiler
1428 when running the examples. If not specified, then it will default to the set of
1429 future-import flags that apply to *globs*.
1430
1431 The output of each example is checked using the :class:`DocTestRunner`'s output
1432 checker, and the results are formatted by the :meth:`DocTestRunner.report_\*`
1433 methods.
1434
1435
1436.. method:: DocTestRunner.summarize([verbose])
1437
1438 Print a summary of all the test cases that have been run by this DocTestRunner,
1439 and return a tuple ``(failure_count, test_count)``.
1440
1441 The optional *verbose* argument controls how detailed the summary is. If the
1442 verbosity is not specified, then the :class:`DocTestRunner`'s verbosity is used.
1443
1444
1445.. _doctest-outputchecker:
1446
1447OutputChecker objects
1448^^^^^^^^^^^^^^^^^^^^^
1449
1450
1451.. class:: OutputChecker()
1452
1453 A class used to check the whether the actual output from a doctest example
1454 matches the expected output. :class:`OutputChecker` defines two methods:
1455 :meth:`check_output`, which compares a given pair of outputs, and returns true
1456 if they match; and :meth:`output_difference`, which returns a string describing
1457 the differences between two outputs.
1458
Georg Brandl116aa622007-08-15 14:28:22 +00001459
1460:class:`OutputChecker` defines the following methods:
1461
Georg Brandl116aa622007-08-15 14:28:22 +00001462.. method:: OutputChecker.check_output(want, got, optionflags)
1463
1464 Return ``True`` iff the actual output from an example (*got*) matches the
1465 expected output (*want*). These strings are always considered to match if they
1466 are identical; but depending on what option flags the test runner is using,
1467 several non-exact match types are also possible. See section
1468 :ref:`doctest-options` for more information about option flags.
1469
1470
1471.. method:: OutputChecker.output_difference(example, got, optionflags)
1472
1473 Return a string describing the differences between the expected output for a
1474 given example (*example*) and the actual output (*got*). *optionflags* is the
1475 set of option flags used to compare *want* and *got*.
1476
1477
1478.. _doctest-debugging:
1479
1480Debugging
1481---------
1482
1483Doctest provides several mechanisms for debugging doctest examples:
1484
1485* Several functions convert doctests to executable Python programs, which can be
1486 run under the Python debugger, :mod:`pdb`.
1487
1488* The :class:`DebugRunner` class is a subclass of :class:`DocTestRunner` that
1489 raises an exception for the first failing example, containing information about
1490 that example. This information can be used to perform post-mortem debugging on
1491 the example.
1492
1493* The :mod:`unittest` cases generated by :func:`DocTestSuite` support the
1494 :meth:`debug` method defined by :class:`unittest.TestCase`.
1495
1496* You can add a call to :func:`pdb.set_trace` in a doctest example, and you'll
1497 drop into the Python debugger when that line is executed. Then you can inspect
1498 current values of variables, and so on. For example, suppose :file:`a.py`
1499 contains just this module docstring::
1500
1501 """
1502 >>> def f(x):
1503 ... g(x*2)
1504 >>> def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001505 ... print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001506 ... import pdb; pdb.set_trace()
1507 >>> f(3)
1508 9
1509 """
1510
1511 Then an interactive Python session may look like this::
1512
1513 >>> import a, doctest
1514 >>> doctest.testmod(a)
1515 --Return--
1516 > <doctest a[1]>(3)g()->None
1517 -> import pdb; pdb.set_trace()
1518 (Pdb) list
1519 1 def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001520 2 print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001521 3 -> import pdb; pdb.set_trace()
1522 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001523 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001524 6
1525 (Pdb) step
1526 --Return--
1527 > <doctest a[0]>(2)f()->None
1528 -> g(x*2)
1529 (Pdb) list
1530 1 def f(x):
1531 2 -> g(x*2)
1532 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001533 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001534 3
1535 (Pdb) step
1536 --Return--
1537 > <doctest a[2]>(1)?()->None
1538 -> f(3)
1539 (Pdb) cont
1540 (0, 3)
1541 >>>
1542
Georg Brandl116aa622007-08-15 14:28:22 +00001543
1544Functions that convert doctests to Python code, and possibly run the synthesized
1545code under the debugger:
1546
1547
1548.. function:: script_from_examples(s)
1549
1550 Convert text with examples to a script.
1551
1552 Argument *s* is a string containing doctest examples. The string is converted
1553 to a Python script, where doctest examples in *s* are converted to regular code,
1554 and everything else is converted to Python comments. The generated script is
1555 returned as a string. For example, ::
1556
1557 import doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001558 print(doctest.script_from_examples(r"""
Georg Brandl116aa622007-08-15 14:28:22 +00001559 Set x and y to 1 and 2.
1560 >>> x, y = 1, 2
1561
1562 Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001563 >>> print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001564 3
Georg Brandl6911e3c2007-09-04 07:15:32 +00001565 """))
Georg Brandl116aa622007-08-15 14:28:22 +00001566
1567 displays::
1568
1569 # Set x and y to 1 and 2.
1570 x, y = 1, 2
1571 #
1572 # Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001573 print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001574 # Expected:
1575 ## 3
1576
1577 This function is used internally by other functions (see below), but can also be
1578 useful when you want to transform an interactive Python session into a Python
1579 script.
1580
Georg Brandl116aa622007-08-15 14:28:22 +00001581
1582.. function:: testsource(module, name)
1583
1584 Convert the doctest for an object to a script.
1585
1586 Argument *module* is a module object, or dotted name of a module, containing the
1587 object whose doctests are of interest. Argument *name* is the name (within the
1588 module) of the object with the doctests of interest. The result is a string,
1589 containing the object's docstring converted to a Python script, as described for
1590 :func:`script_from_examples` above. For example, if module :file:`a.py`
1591 contains a top-level function :func:`f`, then ::
1592
1593 import a, doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001594 print(doctest.testsource(a, "a.f"))
Georg Brandl116aa622007-08-15 14:28:22 +00001595
1596 prints a script version of function :func:`f`'s docstring, with doctests
1597 converted to code, and the rest placed in comments.
1598
Georg Brandl116aa622007-08-15 14:28:22 +00001599
1600.. function:: debug(module, name[, pm])
1601
1602 Debug the doctests for an object.
1603
1604 The *module* and *name* arguments are the same as for function
1605 :func:`testsource` above. The synthesized Python script for the named object's
1606 docstring is written to a temporary file, and then that file is run under the
1607 control of the Python debugger, :mod:`pdb`.
1608
1609 A shallow copy of ``module.__dict__`` is used for both local and global
1610 execution context.
1611
1612 Optional argument *pm* controls whether post-mortem debugging is used. If *pm*
1613 has a true value, the script file is run directly, and the debugger gets
1614 involved only if the script terminates via raising an unhandled exception. If
1615 it does, then post-mortem debugging is invoked, via :func:`pdb.post_mortem`,
1616 passing the traceback object from the unhandled exception. If *pm* is not
1617 specified, or is false, the script is run under the debugger from the start, via
1618 passing an appropriate :func:`exec` call to :func:`pdb.run`.
1619
Georg Brandl116aa622007-08-15 14:28:22 +00001620
1621.. function:: debug_src(src[, pm][, globs])
1622
1623 Debug the doctests in a string.
1624
1625 This is like function :func:`debug` above, except that a string containing
1626 doctest examples is specified directly, via the *src* argument.
1627
1628 Optional argument *pm* has the same meaning as in function :func:`debug` above.
1629
1630 Optional argument *globs* gives a dictionary to use as both local and global
1631 execution context. If not specified, or ``None``, an empty dictionary is used.
1632 If specified, a shallow copy of the dictionary is used.
1633
Georg Brandl116aa622007-08-15 14:28:22 +00001634
1635The :class:`DebugRunner` class, and the special exceptions it may raise, are of
1636most interest to testing framework authors, and will only be sketched here. See
1637the source code, and especially :class:`DebugRunner`'s docstring (which is a
1638doctest!) for more details:
1639
1640
1641.. class:: DebugRunner([checker][, verbose][, optionflags])
1642
1643 A subclass of :class:`DocTestRunner` that raises an exception as soon as a
1644 failure is encountered. If an unexpected exception occurs, an
1645 :exc:`UnexpectedException` exception is raised, containing the test, the
1646 example, and the original exception. If the output doesn't match, then a
1647 :exc:`DocTestFailure` exception is raised, containing the test, the example, and
1648 the actual output.
1649
1650 For information about the constructor parameters and methods, see the
1651 documentation for :class:`DocTestRunner` in section :ref:`doctest-advanced-api`.
1652
1653There are two exceptions that may be raised by :class:`DebugRunner` instances:
1654
1655
1656.. exception:: DocTestFailure(test, example, got)
1657
1658 An exception thrown by :class:`DocTestRunner` to signal that a doctest example's
1659 actual output did not match its expected output. The constructor arguments are
1660 used to initialize the member variables of the same names.
1661
1662:exc:`DocTestFailure` defines the following member variables:
1663
1664
1665.. attribute:: DocTestFailure.test
1666
1667 The :class:`DocTest` object that was being run when the example failed.
1668
1669
1670.. attribute:: DocTestFailure.example
1671
1672 The :class:`Example` that failed.
1673
1674
1675.. attribute:: DocTestFailure.got
1676
1677 The example's actual output.
1678
1679
1680.. exception:: UnexpectedException(test, example, exc_info)
1681
1682 An exception thrown by :class:`DocTestRunner` to signal that a doctest example
1683 raised an unexpected exception. The constructor arguments are used to
1684 initialize the member variables of the same names.
1685
1686:exc:`UnexpectedException` defines the following member variables:
1687
1688
1689.. attribute:: UnexpectedException.test
1690
1691 The :class:`DocTest` object that was being run when the example failed.
1692
1693
1694.. attribute:: UnexpectedException.example
1695
1696 The :class:`Example` that failed.
1697
1698
1699.. attribute:: UnexpectedException.exc_info
1700
1701 A tuple containing information about the unexpected exception, as returned by
1702 :func:`sys.exc_info`.
1703
1704
1705.. _doctest-soapbox:
1706
1707Soapbox
1708-------
1709
1710As mentioned in the introduction, :mod:`doctest` has grown to have three primary
1711uses:
1712
1713#. Checking examples in docstrings.
1714
1715#. Regression testing.
1716
1717#. Executable documentation / literate testing.
1718
1719These uses have different requirements, and it is important to distinguish them.
1720In particular, filling your docstrings with obscure test cases makes for bad
1721documentation.
1722
1723When writing a docstring, choose docstring examples with care. There's an art to
1724this that needs to be learned---it may not be natural at first. Examples should
1725add genuine value to the documentation. A good example can often be worth many
1726words. If done with care, the examples will be invaluable for your users, and
1727will pay back the time it takes to collect them many times over as the years go
1728by and things change. I'm still amazed at how often one of my :mod:`doctest`
1729examples stops working after a "harmless" change.
1730
1731Doctest also makes an excellent tool for regression testing, especially if you
1732don't skimp on explanatory text. By interleaving prose and examples, it becomes
1733much easier to keep track of what's actually being tested, and why. When a test
1734fails, good prose can make it much easier to figure out what the problem is, and
1735how it should be fixed. It's true that you could write extensive comments in
1736code-based testing, but few programmers do. Many have found that using doctest
1737approaches instead leads to much clearer tests. Perhaps this is simply because
1738doctest makes writing prose a little easier than writing code, while writing
1739comments in code is a little harder. I think it goes deeper than just that:
1740the natural attitude when writing a doctest-based test is that you want to
1741explain the fine points of your software, and illustrate them with examples.
1742This in turn naturally leads to test files that start with the simplest
1743features, and logically progress to complications and edge cases. A coherent
1744narrative is the result, instead of a collection of isolated functions that test
1745isolated bits of functionality seemingly at random. It's a different attitude,
1746and produces different results, blurring the distinction between testing and
1747explaining.
1748
1749Regression testing is best confined to dedicated objects or files. There are
1750several options for organizing tests:
1751
1752* Write text files containing test cases as interactive examples, and test the
1753 files using :func:`testfile` or :func:`DocFileSuite`. This is recommended,
1754 although is easiest to do for new projects, designed from the start to use
1755 doctest.
1756
1757* Define functions named ``_regrtest_topic`` that consist of single docstrings,
1758 containing test cases for the named topics. These functions can be included in
1759 the same file as the module, or separated out into a separate test file.
1760
1761* Define a ``__test__`` dictionary mapping from regression test topics to
1762 docstrings containing test cases.
1763
1764.. rubric:: Footnotes
1765
1766.. [#] Examples containing both expected output and an exception are not supported.
1767 Trying to guess where one ends and the other begins is too error-prone, and that
1768 also makes for a confusing test.
1769