blob: 721d7c0f039bd6291db2cf31e1a6766aff2f97af [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
694test like
695
696.. % Hey! What happened to Monty Python examples?
697.. % Tim: ask Guido -- it's his example!
698
699::
700
701 >>> foo()
702 {"Hermione": "hippogryph", "Harry": "broomstick"}
703
704is vulnerable! One workaround is to do ::
705
706 >>> foo() == {"Hermione": "hippogryph", "Harry": "broomstick"}
707 True
708
709instead. Another is to do ::
710
711 >>> d = foo().items()
712 >>> d.sort()
713 >>> d
714 [('Harry', 'broomstick'), ('Hermione', 'hippogryph')]
715
716There are others, but you get the idea.
717
718Another bad idea is to print things that embed an object address, like ::
719
720 >>> id(1.0) # certain to fail some of the time
721 7948648
722 >>> class C: pass
723 >>> C() # the default repr() for instances embeds an address
724 <__main__.C instance at 0x00AC18F0>
725
726The :const:`ELLIPSIS` directive gives a nice approach for the last example::
727
728 >>> C() #doctest: +ELLIPSIS
729 <__main__.C instance at 0x...>
730
731Floating-point numbers are also subject to small output variations across
732platforms, because Python defers to the platform C library for float formatting,
733and C libraries vary widely in quality here. ::
734
735 >>> 1./7 # risky
736 0.14285714285714285
Georg Brandl6911e3c2007-09-04 07:15:32 +0000737 >>> print(1./7) # safer
Georg Brandl116aa622007-08-15 14:28:22 +0000738 0.142857142857
Georg Brandl6911e3c2007-09-04 07:15:32 +0000739 >>> print(round(1./7, 6)) # much safer
Georg Brandl116aa622007-08-15 14:28:22 +0000740 0.142857
741
742Numbers of the form ``I/2.**J`` are safe across all platforms, and I often
743contrive doctest examples to produce numbers of that form::
744
745 >>> 3./4 # utterly safe
746 0.75
747
748Simple fractions are also easier for people to understand, and that makes for
749better documentation.
750
751
752.. _doctest-basic-api:
753
754Basic API
755---------
756
757The functions :func:`testmod` and :func:`testfile` provide a simple interface to
758doctest that should be sufficient for most basic uses. For a less formal
759introduction to these two functions, see sections :ref:`doctest-simple-testmod`
760and :ref:`doctest-simple-testfile`.
761
762
763.. function:: testfile(filename[, module_relative][, name][, package][, globs][, verbose][, report][, optionflags][, extraglobs][, raise_on_error][, parser][, encoding])
764
765 All arguments except *filename* are optional, and should be specified in keyword
766 form.
767
768 Test examples in the file named *filename*. Return ``(failure_count,
769 test_count)``.
770
771 Optional argument *module_relative* specifies how the filename should be
772 interpreted:
773
774 * If *module_relative* is ``True`` (the default), then *filename* specifies an
775 OS-independent module-relative path. By default, this path is relative to the
776 calling module's directory; but if the *package* argument is specified, then it
777 is relative to that package. To ensure OS-independence, *filename* should use
778 ``/`` characters to separate path segments, and may not be an absolute path
779 (i.e., it may not begin with ``/``).
780
781 * If *module_relative* is ``False``, then *filename* specifies an OS-specific
782 path. The path may be absolute or relative; relative paths are resolved with
783 respect to the current working directory.
784
785 Optional argument *name* gives the name of the test; by default, or if ``None``,
786 ``os.path.basename(filename)`` is used.
787
788 Optional argument *package* is a Python package or the name of a Python package
789 whose directory should be used as the base directory for a module-relative
790 filename. If no package is specified, then the calling module's directory is
791 used as the base directory for module-relative filenames. It is an error to
792 specify *package* if *module_relative* is ``False``.
793
794 Optional argument *globs* gives a dict to be used as the globals when executing
795 examples. A new shallow copy of this dict is created for the doctest, so its
796 examples start with a clean slate. By default, or if ``None``, a new empty dict
797 is used.
798
799 Optional argument *extraglobs* gives a dict merged into the globals used to
800 execute examples. This works like :meth:`dict.update`: if *globs* and
801 *extraglobs* have a common key, the associated value in *extraglobs* appears in
802 the combined dict. By default, or if ``None``, no extra globals are used. This
803 is an advanced feature that allows parameterization of doctests. For example, a
804 doctest can be written for a base class, using a generic name for the class,
805 then reused to test any number of subclasses by passing an *extraglobs* dict
806 mapping the generic name to the subclass to be tested.
807
808 Optional argument *verbose* prints lots of stuff if true, and prints only
809 failures if false; by default, or if ``None``, it's true if and only if ``'-v'``
810 is in ``sys.argv``.
811
812 Optional argument *report* prints a summary at the end when true, else prints
813 nothing at the end. In verbose mode, the summary is detailed, else the summary
814 is very brief (in fact, empty if all tests passed).
815
816 Optional argument *optionflags* or's together option flags. See section
817 :ref:`doctest-options`.
818
819 Optional argument *raise_on_error* defaults to false. If true, an exception is
820 raised upon the first failure or unexpected exception in an example. This
821 allows failures to be post-mortem debugged. Default behavior is to continue
822 running examples.
823
824 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass) that
825 should be used to extract tests from the files. It defaults to a normal parser
826 (i.e., ``DocTestParser()``).
827
828 Optional argument *encoding* specifies an encoding that should be used to
829 convert the file to unicode.
830
Georg Brandl116aa622007-08-15 14:28:22 +0000831
832.. function:: testmod([m][, name][, globs][, verbose][, report][, optionflags][, extraglobs][, raise_on_error][, exclude_empty])
833
834 All arguments are optional, and all except for *m* should be specified in
835 keyword form.
836
837 Test examples in docstrings in functions and classes reachable from module *m*
838 (or module :mod:`__main__` if *m* is not supplied or is ``None``), starting with
839 ``m.__doc__``.
840
841 Also test examples reachable from dict ``m.__test__``, if it exists and is not
842 ``None``. ``m.__test__`` maps names (strings) to functions, classes and
843 strings; function and class docstrings are searched for examples; strings are
844 searched directly, as if they were docstrings.
845
846 Only docstrings attached to objects belonging to module *m* are searched.
847
848 Return ``(failure_count, test_count)``.
849
850 Optional argument *name* gives the name of the module; by default, or if
851 ``None``, ``m.__name__`` is used.
852
853 Optional argument *exclude_empty* defaults to false. If true, objects for which
854 no doctests are found are excluded from consideration. The default is a backward
855 compatibility hack, so that code still using :meth:`doctest.master.summarize` in
856 conjunction with :func:`testmod` continues to get output for objects with no
857 tests. The *exclude_empty* argument to the newer :class:`DocTestFinder`
858 constructor defaults to true.
859
860 Optional arguments *extraglobs*, *verbose*, *report*, *optionflags*,
861 *raise_on_error*, and *globs* are the same as for function :func:`testfile`
862 above, except that *globs* defaults to ``m.__dict__``.
863
Georg Brandl116aa622007-08-15 14:28:22 +0000864
865There's also a function to run the doctests associated with a single object.
866This function is provided for backward compatibility. There are no plans to
867deprecate it, but it's rarely useful:
868
869
870.. function:: run_docstring_examples(f, globs[, verbose][, name][, compileflags][, optionflags])
871
872 Test examples associated with object *f*; for example, *f* may be a module,
873 function, or class object.
874
875 A shallow copy of dictionary argument *globs* is used for the execution context.
876
877 Optional argument *name* is used in failure messages, and defaults to
878 ``"NoName"``.
879
880 If optional argument *verbose* is true, output is generated even if there are no
881 failures. By default, output is generated only in case of an example failure.
882
883 Optional argument *compileflags* gives the set of flags that should be used by
884 the Python compiler when running the examples. By default, or if ``None``,
885 flags are deduced corresponding to the set of future features found in *globs*.
886
887 Optional argument *optionflags* works as for function :func:`testfile` above.
888
889
890.. _doctest-unittest-api:
891
892Unittest API
893------------
894
895As your collection of doctest'ed modules grows, you'll want a way to run all
896their doctests systematically. Prior to Python 2.4, :mod:`doctest` had a barely
897documented :class:`Tester` class that supplied a rudimentary way to combine
898doctests from multiple modules. :class:`Tester` was feeble, and in practice most
899serious Python testing frameworks build on the :mod:`unittest` module, which
900supplies many flexible ways to combine tests from multiple sources. So, in
901Python 2.4, :mod:`doctest`'s :class:`Tester` class is deprecated, and
902:mod:`doctest` provides two functions that can be used to create :mod:`unittest`
903test suites from modules and text files containing doctests. These test suites
904can then be run using :mod:`unittest` test runners::
905
906 import unittest
907 import doctest
908 import my_module_with_doctests, and_another
909
910 suite = unittest.TestSuite()
911 for mod in my_module_with_doctests, and_another:
912 suite.addTest(doctest.DocTestSuite(mod))
913 runner = unittest.TextTestRunner()
914 runner.run(suite)
915
916There are two main functions for creating :class:`unittest.TestSuite` instances
917from text files and modules with doctests:
918
919
920.. function:: DocFileSuite([module_relative][, package][, setUp][, tearDown][, globs][, optionflags][, parser][, encoding])
921
922 Convert doctest tests from one or more text files to a
923 :class:`unittest.TestSuite`.
924
925 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
926 and runs the interactive examples in each file. If an example in any file
927 fails, then the synthesized unit test fails, and a :exc:`failureException`
928 exception is raised showing the name of the file containing the test and a
929 (sometimes approximate) line number.
930
931 Pass one or more paths (as strings) to text files to be examined.
932
933 Options may be provided as keyword arguments:
934
935 Optional argument *module_relative* specifies how the filenames in *paths*
936 should be interpreted:
937
938 * If *module_relative* is ``True`` (the default), then each filename specifies
939 an OS-independent module-relative path. By default, this path is relative to
940 the calling module's directory; but if the *package* argument is specified, then
941 it is relative to that package. To ensure OS-independence, each filename should
942 use ``/`` characters to separate path segments, and may not be an absolute path
943 (i.e., it may not begin with ``/``).
944
945 * If *module_relative* is ``False``, then each filename specifies an OS-specific
946 path. The path may be absolute or relative; relative paths are resolved with
947 respect to the current working directory.
948
949 Optional argument *package* is a Python package or the name of a Python package
950 whose directory should be used as the base directory for module-relative
951 filenames. If no package is specified, then the calling module's directory is
952 used as the base directory for module-relative filenames. It is an error to
953 specify *package* if *module_relative* is ``False``.
954
955 Optional argument *setUp* specifies a set-up function for the test suite. This
956 is called before running the tests in each file. The *setUp* function will be
957 passed a :class:`DocTest` object. The setUp function can access the test
958 globals as the *globs* attribute of the test passed.
959
960 Optional argument *tearDown* specifies a tear-down function for the test suite.
961 This is called after running the tests in each file. The *tearDown* function
962 will be passed a :class:`DocTest` object. The setUp function can access the
963 test globals as the *globs* attribute of the test passed.
964
965 Optional argument *globs* is a dictionary containing the initial global
966 variables for the tests. A new copy of this dictionary is created for each
967 test. By default, *globs* is a new empty dictionary.
968
969 Optional argument *optionflags* specifies the default doctest options for the
970 tests, created by or-ing together individual option flags. See section
971 :ref:`doctest-options`. See function :func:`set_unittest_reportflags` below for
972 a better way to set reporting options.
973
974 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass) that
975 should be used to extract tests from the files. It defaults to a normal parser
976 (i.e., ``DocTestParser()``).
977
978 Optional argument *encoding* specifies an encoding that should be used to
979 convert the file to unicode.
980
Georg Brandl55ac8f02007-09-01 13:51:09 +0000981 The global ``__file__`` is added to the globals provided to doctests loaded
982 from a text file using :func:`DocFileSuite`.
Georg Brandl116aa622007-08-15 14:28:22 +0000983
984
985.. function:: DocTestSuite([module][, globs][, extraglobs][, test_finder][, setUp][, tearDown][, checker])
986
987 Convert doctest tests for a module to a :class:`unittest.TestSuite`.
988
989 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
990 and runs each doctest in the module. If any of the doctests fail, then the
991 synthesized unit test fails, and a :exc:`failureException` exception is raised
992 showing the name of the file containing the test and a (sometimes approximate)
993 line number.
994
995 Optional argument *module* provides the module to be tested. It can be a module
996 object or a (possibly dotted) module name. If not specified, the module calling
997 this function is used.
998
999 Optional argument *globs* is a dictionary containing the initial global
1000 variables for the tests. A new copy of this dictionary is created for each
1001 test. By default, *globs* is a new empty dictionary.
1002
1003 Optional argument *extraglobs* specifies an extra set of global variables, which
1004 is merged into *globs*. By default, no extra globals are used.
1005
1006 Optional argument *test_finder* is the :class:`DocTestFinder` object (or a
1007 drop-in replacement) that is used to extract doctests from the module.
1008
1009 Optional arguments *setUp*, *tearDown*, and *optionflags* are the same as for
1010 function :func:`DocFileSuite` above.
1011
Georg Brandl55ac8f02007-09-01 13:51:09 +00001012 This function uses the same search technique as :func:`testmod`.
Georg Brandl116aa622007-08-15 14:28:22 +00001013
Georg Brandl116aa622007-08-15 14:28:22 +00001014
1015Under the covers, :func:`DocTestSuite` creates a :class:`unittest.TestSuite` out
1016of :class:`doctest.DocTestCase` instances, and :class:`DocTestCase` is a
1017subclass of :class:`unittest.TestCase`. :class:`DocTestCase` isn't documented
1018here (it's an internal detail), but studying its code can answer questions about
1019the exact details of :mod:`unittest` integration.
1020
1021Similarly, :func:`DocFileSuite` creates a :class:`unittest.TestSuite` out of
1022:class:`doctest.DocFileCase` instances, and :class:`DocFileCase` is a subclass
1023of :class:`DocTestCase`.
1024
1025So both ways of creating a :class:`unittest.TestSuite` run instances of
1026:class:`DocTestCase`. This is important for a subtle reason: when you run
1027:mod:`doctest` functions yourself, you can control the :mod:`doctest` options in
1028use directly, by passing option flags to :mod:`doctest` functions. However, if
1029you're writing a :mod:`unittest` framework, :mod:`unittest` ultimately controls
1030when and how tests get run. The framework author typically wants to control
1031:mod:`doctest` reporting options (perhaps, e.g., specified by command line
1032options), but there's no way to pass options through :mod:`unittest` to
1033:mod:`doctest` test runners.
1034
1035For this reason, :mod:`doctest` also supports a notion of :mod:`doctest`
1036reporting flags specific to :mod:`unittest` support, via this function:
1037
1038
1039.. function:: set_unittest_reportflags(flags)
1040
1041 Set the :mod:`doctest` reporting flags to use.
1042
1043 Argument *flags* or's together option flags. See section
1044 :ref:`doctest-options`. Only "reporting flags" can be used.
1045
1046 This is a module-global setting, and affects all future doctests run by module
1047 :mod:`unittest`: the :meth:`runTest` method of :class:`DocTestCase` looks at
1048 the option flags specified for the test case when the :class:`DocTestCase`
1049 instance was constructed. If no reporting flags were specified (which is the
1050 typical and expected case), :mod:`doctest`'s :mod:`unittest` reporting flags are
1051 or'ed into the option flags, and the option flags so augmented are passed to the
1052 :class:`DocTestRunner` instance created to run the doctest. If any reporting
1053 flags were specified when the :class:`DocTestCase` instance was constructed,
1054 :mod:`doctest`'s :mod:`unittest` reporting flags are ignored.
1055
1056 The value of the :mod:`unittest` reporting flags in effect before the function
1057 was called is returned by the function.
1058
Georg Brandl116aa622007-08-15 14:28:22 +00001059
1060.. _doctest-advanced-api:
1061
1062Advanced API
1063------------
1064
1065The basic API is a simple wrapper that's intended to make doctest easy to use.
1066It is fairly flexible, and should meet most users' needs; however, if you
1067require more fine-grained control over testing, or wish to extend doctest's
1068capabilities, then you should use the advanced API.
1069
1070The advanced API revolves around two container classes, which are used to store
1071the interactive examples extracted from doctest cases:
1072
1073* :class:`Example`: A single python statement, paired with its expected output.
1074
1075* :class:`DocTest`: A collection of :class:`Example`\ s, typically extracted
1076 from a single docstring or text file.
1077
1078Additional processing classes are defined to find, parse, and run, and check
1079doctest examples:
1080
1081* :class:`DocTestFinder`: Finds all docstrings in a given module, and uses a
1082 :class:`DocTestParser` to create a :class:`DocTest` from every docstring that
1083 contains interactive examples.
1084
1085* :class:`DocTestParser`: Creates a :class:`DocTest` object from a string (such
1086 as an object's docstring).
1087
1088* :class:`DocTestRunner`: Executes the examples in a :class:`DocTest`, and uses
1089 an :class:`OutputChecker` to verify their output.
1090
1091* :class:`OutputChecker`: Compares the actual output from a doctest example with
1092 the expected output, and decides whether they match.
1093
1094The relationships among these processing classes are summarized in the following
1095diagram::
1096
1097 list of:
1098 +------+ +---------+
1099 |module| --DocTestFinder-> | DocTest | --DocTestRunner-> results
1100 +------+ | ^ +---------+ | ^ (printed)
1101 | | | Example | | |
1102 v | | ... | v |
1103 DocTestParser | Example | OutputChecker
1104 +---------+
1105
1106
1107.. _doctest-doctest:
1108
1109DocTest Objects
1110^^^^^^^^^^^^^^^
1111
1112
1113.. class:: DocTest(examples, globs, name, filename, lineno, docstring)
1114
1115 A collection of doctest examples that should be run in a single namespace. The
1116 constructor arguments are used to initialize the member variables of the same
1117 names.
1118
Georg Brandl116aa622007-08-15 14:28:22 +00001119
1120:class:`DocTest` defines the following member variables. They are initialized
1121by the constructor, and should not be modified directly.
1122
1123
1124.. attribute:: DocTest.examples
1125
1126 A list of :class:`Example` objects encoding the individual interactive Python
1127 examples that should be run by this test.
1128
1129
1130.. attribute:: DocTest.globs
1131
1132 The namespace (aka globals) that the examples should be run in. This is a
1133 dictionary mapping names to values. Any changes to the namespace made by the
1134 examples (such as binding new variables) will be reflected in :attr:`globs`
1135 after the test is run.
1136
1137
1138.. attribute:: DocTest.name
1139
1140 A string name identifying the :class:`DocTest`. Typically, this is the name of
1141 the object or file that the test was extracted from.
1142
1143
1144.. attribute:: DocTest.filename
1145
1146 The name of the file that this :class:`DocTest` was extracted from; or ``None``
1147 if the filename is unknown, or if the :class:`DocTest` was not extracted from a
1148 file.
1149
1150
1151.. attribute:: DocTest.lineno
1152
1153 The line number within :attr:`filename` where this :class:`DocTest` begins, or
1154 ``None`` if the line number is unavailable. This line number is zero-based with
1155 respect to the beginning of the file.
1156
1157
1158.. attribute:: DocTest.docstring
1159
1160 The string that the test was extracted from, or 'None' if the string is
1161 unavailable, or if the test was not extracted from a string.
1162
1163
1164.. _doctest-example:
1165
1166Example Objects
1167^^^^^^^^^^^^^^^
1168
1169
1170.. class:: Example(source, want[, exc_msg][, lineno][, indent][, options])
1171
1172 A single interactive example, consisting of a Python statement and its expected
1173 output. The constructor arguments are used to initialize the member variables
1174 of the same names.
1175
Georg Brandl116aa622007-08-15 14:28:22 +00001176
1177:class:`Example` defines the following member variables. They are initialized
1178by the constructor, and should not be modified directly.
1179
1180
1181.. attribute:: Example.source
1182
1183 A string containing the example's source code. This source code consists of a
1184 single Python statement, and always ends with a newline; the constructor adds a
1185 newline when necessary.
1186
1187
1188.. attribute:: Example.want
1189
1190 The expected output from running the example's source code (either from stdout,
1191 or a traceback in case of exception). :attr:`want` ends with a newline unless
1192 no output is expected, in which case it's an empty string. The constructor adds
1193 a newline when necessary.
1194
1195
1196.. attribute:: Example.exc_msg
1197
1198 The exception message generated by the example, if the example is expected to
1199 generate an exception; or ``None`` if it is not expected to generate an
1200 exception. This exception message is compared against the return value of
1201 :func:`traceback.format_exception_only`. :attr:`exc_msg` ends with a newline
1202 unless it's ``None``. The constructor adds a newline if needed.
1203
1204
1205.. attribute:: Example.lineno
1206
1207 The line number within the string containing this example where the example
1208 begins. This line number is zero-based with respect to the beginning of the
1209 containing string.
1210
1211
1212.. attribute:: Example.indent
1213
1214 The example's indentation in the containing string, i.e., the number of space
1215 characters that precede the example's first prompt.
1216
1217
1218.. attribute:: Example.options
1219
1220 A dictionary mapping from option flags to ``True`` or ``False``, which is used
1221 to override default options for this example. Any option flags not contained in
1222 this dictionary are left at their default value (as specified by the
1223 :class:`DocTestRunner`'s :attr:`optionflags`). By default, no options are set.
1224
1225
1226.. _doctest-doctestfinder:
1227
1228DocTestFinder objects
1229^^^^^^^^^^^^^^^^^^^^^
1230
1231
1232.. class:: DocTestFinder([verbose][, parser][, recurse][, exclude_empty])
1233
1234 A processing class used to extract the :class:`DocTest`\ s that are relevant to
1235 a given object, from its docstring and the docstrings of its contained objects.
1236 :class:`DocTest`\ s can currently be extracted from the following object types:
1237 modules, functions, classes, methods, staticmethods, classmethods, and
1238 properties.
1239
1240 The optional argument *verbose* can be used to display the objects searched by
1241 the finder. It defaults to ``False`` (no output).
1242
1243 The optional argument *parser* specifies the :class:`DocTestParser` object (or a
1244 drop-in replacement) that is used to extract doctests from docstrings.
1245
1246 If the optional argument *recurse* is false, then :meth:`DocTestFinder.find`
1247 will only examine the given object, and not any contained objects.
1248
1249 If the optional argument *exclude_empty* is false, then
1250 :meth:`DocTestFinder.find` will include tests for objects with empty docstrings.
1251
Georg Brandl116aa622007-08-15 14:28:22 +00001252
1253:class:`DocTestFinder` defines the following method:
1254
1255
1256.. method:: DocTestFinder.find(obj[, name][, module][, globs][, extraglobs])
1257
1258 Return a list of the :class:`DocTest`\ s that are defined by *obj*'s docstring,
1259 or by any of its contained objects' docstrings.
1260
1261 The optional argument *name* specifies the object's name; this name will be used
1262 to construct names for the returned :class:`DocTest`\ s. If *name* is not
1263 specified, then ``obj.__name__`` is used.
1264
1265 The optional parameter *module* is the module that contains the given object.
1266 If the module is not specified or is None, then the test finder will attempt to
1267 automatically determine the correct module. The object's module is used:
1268
1269 * As a default namespace, if *globs* is not specified.
1270
1271 * To prevent the DocTestFinder from extracting DocTests from objects that are
1272 imported from other modules. (Contained objects with modules other than
1273 *module* are ignored.)
1274
1275 * To find the name of the file containing the object.
1276
1277 * To help find the line number of the object within its file.
1278
1279 If *module* is ``False``, no attempt to find the module will be made. This is
1280 obscure, of use mostly in testing doctest itself: if *module* is ``False``, or
1281 is ``None`` but cannot be found automatically, then all objects are considered
1282 to belong to the (non-existent) module, so all contained objects will
1283 (recursively) be searched for doctests.
1284
1285 The globals for each :class:`DocTest` is formed by combining *globs* and
1286 *extraglobs* (bindings in *extraglobs* override bindings in *globs*). A new
1287 shallow copy of the globals dictionary is created for each :class:`DocTest`. If
1288 *globs* is not specified, then it defaults to the module's *__dict__*, if
1289 specified, or ``{}`` otherwise. If *extraglobs* is not specified, then it
1290 defaults to ``{}``.
1291
1292
1293.. _doctest-doctestparser:
1294
1295DocTestParser objects
1296^^^^^^^^^^^^^^^^^^^^^
1297
1298
1299.. class:: DocTestParser()
1300
1301 A processing class used to extract interactive examples from a string, and use
1302 them to create a :class:`DocTest` object.
1303
Georg Brandl116aa622007-08-15 14:28:22 +00001304
1305:class:`DocTestParser` defines the following methods:
1306
1307
1308.. method:: DocTestParser.get_doctest(string, globs, name, filename, lineno)
1309
1310 Extract all doctest examples from the given string, and collect them into a
1311 :class:`DocTest` object.
1312
1313 *globs*, *name*, *filename*, and *lineno* are attributes for the new
1314 :class:`DocTest` object. See the documentation for :class:`DocTest` for more
1315 information.
1316
1317
1318.. method:: DocTestParser.get_examples(string[, name])
1319
1320 Extract all doctest examples from the given string, and return them as a list of
1321 :class:`Example` objects. Line numbers are 0-based. The optional argument
1322 *name* is a name identifying this string, and is only used for error messages.
1323
1324
1325.. method:: DocTestParser.parse(string[, name])
1326
1327 Divide the given string into examples and intervening text, and return them as a
1328 list of alternating :class:`Example`\ s and strings. Line numbers for the
1329 :class:`Example`\ s are 0-based. The optional argument *name* is a name
1330 identifying this string, and is only used for error messages.
1331
1332
1333.. _doctest-doctestrunner:
1334
1335DocTestRunner objects
1336^^^^^^^^^^^^^^^^^^^^^
1337
1338
1339.. class:: DocTestRunner([checker][, verbose][, optionflags])
1340
1341 A processing class used to execute and verify the interactive examples in a
1342 :class:`DocTest`.
1343
1344 The comparison between expected outputs and actual outputs is done by an
1345 :class:`OutputChecker`. This comparison may be customized with a number of
1346 option flags; see section :ref:`doctest-options` for more information. If the
1347 option flags are insufficient, then the comparison may also be customized by
1348 passing a subclass of :class:`OutputChecker` to the constructor.
1349
1350 The test runner's display output can be controlled in two ways. First, an output
1351 function can be passed to :meth:`TestRunner.run`; this function will be called
1352 with strings that should be displayed. It defaults to ``sys.stdout.write``. If
1353 capturing the output is not sufficient, then the display output can be also
1354 customized by subclassing DocTestRunner, and overriding the methods
1355 :meth:`report_start`, :meth:`report_success`,
1356 :meth:`report_unexpected_exception`, and :meth:`report_failure`.
1357
1358 The optional keyword argument *checker* specifies the :class:`OutputChecker`
1359 object (or drop-in replacement) that should be used to compare the expected
1360 outputs to the actual outputs of doctest examples.
1361
1362 The optional keyword argument *verbose* controls the :class:`DocTestRunner`'s
1363 verbosity. If *verbose* is ``True``, then information is printed about each
1364 example, as it is run. If *verbose* is ``False``, then only failures are
1365 printed. If *verbose* is unspecified, or ``None``, then verbose output is used
1366 iff the command-line switch :option:`-v` is used.
1367
1368 The optional keyword argument *optionflags* can be used to control how the test
1369 runner compares expected output to actual output, and how it displays failures.
1370 For more information, see section :ref:`doctest-options`.
1371
Georg Brandl116aa622007-08-15 14:28:22 +00001372
1373:class:`DocTestParser` defines the following methods:
1374
1375
1376.. method:: DocTestRunner.report_start(out, test, example)
1377
1378 Report that the test runner is about to process the given example. This method
1379 is provided to allow subclasses of :class:`DocTestRunner` to customize their
1380 output; it should not be called directly.
1381
1382 *example* is the example about to be processed. *test* is the test containing
1383 *example*. *out* is the output function that was passed to
1384 :meth:`DocTestRunner.run`.
1385
1386
1387.. method:: DocTestRunner.report_success(out, test, example, got)
1388
1389 Report that the given example ran successfully. This method is provided to
1390 allow subclasses of :class:`DocTestRunner` to customize their output; it should
1391 not be called directly.
1392
1393 *example* is the example about to be processed. *got* is the actual output from
1394 the example. *test* is the test containing *example*. *out* is the output
1395 function that was passed to :meth:`DocTestRunner.run`.
1396
1397
1398.. method:: DocTestRunner.report_failure(out, test, example, got)
1399
1400 Report that the given example failed. This method is provided to allow
1401 subclasses of :class:`DocTestRunner` to customize their output; it should not be
1402 called directly.
1403
1404 *example* is the example about to be processed. *got* is the actual output from
1405 the example. *test* is the test containing *example*. *out* is the output
1406 function that was passed to :meth:`DocTestRunner.run`.
1407
1408
1409.. method:: DocTestRunner.report_unexpected_exception(out, test, example, exc_info)
1410
1411 Report that the given example raised an unexpected exception. This method is
1412 provided to allow subclasses of :class:`DocTestRunner` to customize their
1413 output; it should not be called directly.
1414
1415 *example* is the example about to be processed. *exc_info* is a tuple containing
1416 information about the unexpected exception (as returned by
1417 :func:`sys.exc_info`). *test* is the test containing *example*. *out* is the
1418 output function that was passed to :meth:`DocTestRunner.run`.
1419
1420
1421.. method:: DocTestRunner.run(test[, compileflags][, out][, clear_globs])
1422
1423 Run the examples in *test* (a :class:`DocTest` object), and display the results
1424 using the writer function *out*.
1425
1426 The examples are run in the namespace ``test.globs``. If *clear_globs* is true
1427 (the default), then this namespace will be cleared after the test runs, to help
1428 with garbage collection. If you would like to examine the namespace after the
1429 test completes, then use *clear_globs=False*.
1430
1431 *compileflags* gives the set of flags that should be used by the Python compiler
1432 when running the examples. If not specified, then it will default to the set of
1433 future-import flags that apply to *globs*.
1434
1435 The output of each example is checked using the :class:`DocTestRunner`'s output
1436 checker, and the results are formatted by the :meth:`DocTestRunner.report_\*`
1437 methods.
1438
1439
1440.. method:: DocTestRunner.summarize([verbose])
1441
1442 Print a summary of all the test cases that have been run by this DocTestRunner,
1443 and return a tuple ``(failure_count, test_count)``.
1444
1445 The optional *verbose* argument controls how detailed the summary is. If the
1446 verbosity is not specified, then the :class:`DocTestRunner`'s verbosity is used.
1447
1448
1449.. _doctest-outputchecker:
1450
1451OutputChecker objects
1452^^^^^^^^^^^^^^^^^^^^^
1453
1454
1455.. class:: OutputChecker()
1456
1457 A class used to check the whether the actual output from a doctest example
1458 matches the expected output. :class:`OutputChecker` defines two methods:
1459 :meth:`check_output`, which compares a given pair of outputs, and returns true
1460 if they match; and :meth:`output_difference`, which returns a string describing
1461 the differences between two outputs.
1462
Georg Brandl116aa622007-08-15 14:28:22 +00001463
1464:class:`OutputChecker` defines the following methods:
1465
Georg Brandl116aa622007-08-15 14:28:22 +00001466.. method:: OutputChecker.check_output(want, got, optionflags)
1467
1468 Return ``True`` iff the actual output from an example (*got*) matches the
1469 expected output (*want*). These strings are always considered to match if they
1470 are identical; but depending on what option flags the test runner is using,
1471 several non-exact match types are also possible. See section
1472 :ref:`doctest-options` for more information about option flags.
1473
1474
1475.. method:: OutputChecker.output_difference(example, got, optionflags)
1476
1477 Return a string describing the differences between the expected output for a
1478 given example (*example*) and the actual output (*got*). *optionflags* is the
1479 set of option flags used to compare *want* and *got*.
1480
1481
1482.. _doctest-debugging:
1483
1484Debugging
1485---------
1486
1487Doctest provides several mechanisms for debugging doctest examples:
1488
1489* Several functions convert doctests to executable Python programs, which can be
1490 run under the Python debugger, :mod:`pdb`.
1491
1492* The :class:`DebugRunner` class is a subclass of :class:`DocTestRunner` that
1493 raises an exception for the first failing example, containing information about
1494 that example. This information can be used to perform post-mortem debugging on
1495 the example.
1496
1497* The :mod:`unittest` cases generated by :func:`DocTestSuite` support the
1498 :meth:`debug` method defined by :class:`unittest.TestCase`.
1499
1500* You can add a call to :func:`pdb.set_trace` in a doctest example, and you'll
1501 drop into the Python debugger when that line is executed. Then you can inspect
1502 current values of variables, and so on. For example, suppose :file:`a.py`
1503 contains just this module docstring::
1504
1505 """
1506 >>> def f(x):
1507 ... g(x*2)
1508 >>> def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001509 ... print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001510 ... import pdb; pdb.set_trace()
1511 >>> f(3)
1512 9
1513 """
1514
1515 Then an interactive Python session may look like this::
1516
1517 >>> import a, doctest
1518 >>> doctest.testmod(a)
1519 --Return--
1520 > <doctest a[1]>(3)g()->None
1521 -> import pdb; pdb.set_trace()
1522 (Pdb) list
1523 1 def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001524 2 print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001525 3 -> import pdb; pdb.set_trace()
1526 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001527 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001528 6
1529 (Pdb) step
1530 --Return--
1531 > <doctest a[0]>(2)f()->None
1532 -> g(x*2)
1533 (Pdb) list
1534 1 def f(x):
1535 2 -> g(x*2)
1536 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001537 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001538 3
1539 (Pdb) step
1540 --Return--
1541 > <doctest a[2]>(1)?()->None
1542 -> f(3)
1543 (Pdb) cont
1544 (0, 3)
1545 >>>
1546
Georg Brandl116aa622007-08-15 14:28:22 +00001547
1548Functions that convert doctests to Python code, and possibly run the synthesized
1549code under the debugger:
1550
1551
1552.. function:: script_from_examples(s)
1553
1554 Convert text with examples to a script.
1555
1556 Argument *s* is a string containing doctest examples. The string is converted
1557 to a Python script, where doctest examples in *s* are converted to regular code,
1558 and everything else is converted to Python comments. The generated script is
1559 returned as a string. For example, ::
1560
1561 import doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001562 print(doctest.script_from_examples(r"""
Georg Brandl116aa622007-08-15 14:28:22 +00001563 Set x and y to 1 and 2.
1564 >>> x, y = 1, 2
1565
1566 Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001567 >>> print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001568 3
Georg Brandl6911e3c2007-09-04 07:15:32 +00001569 """))
Georg Brandl116aa622007-08-15 14:28:22 +00001570
1571 displays::
1572
1573 # Set x and y to 1 and 2.
1574 x, y = 1, 2
1575 #
1576 # Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001577 print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001578 # Expected:
1579 ## 3
1580
1581 This function is used internally by other functions (see below), but can also be
1582 useful when you want to transform an interactive Python session into a Python
1583 script.
1584
Georg Brandl116aa622007-08-15 14:28:22 +00001585
1586.. function:: testsource(module, name)
1587
1588 Convert the doctest for an object to a script.
1589
1590 Argument *module* is a module object, or dotted name of a module, containing the
1591 object whose doctests are of interest. Argument *name* is the name (within the
1592 module) of the object with the doctests of interest. The result is a string,
1593 containing the object's docstring converted to a Python script, as described for
1594 :func:`script_from_examples` above. For example, if module :file:`a.py`
1595 contains a top-level function :func:`f`, then ::
1596
1597 import a, doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001598 print(doctest.testsource(a, "a.f"))
Georg Brandl116aa622007-08-15 14:28:22 +00001599
1600 prints a script version of function :func:`f`'s docstring, with doctests
1601 converted to code, and the rest placed in comments.
1602
Georg Brandl116aa622007-08-15 14:28:22 +00001603
1604.. function:: debug(module, name[, pm])
1605
1606 Debug the doctests for an object.
1607
1608 The *module* and *name* arguments are the same as for function
1609 :func:`testsource` above. The synthesized Python script for the named object's
1610 docstring is written to a temporary file, and then that file is run under the
1611 control of the Python debugger, :mod:`pdb`.
1612
1613 A shallow copy of ``module.__dict__`` is used for both local and global
1614 execution context.
1615
1616 Optional argument *pm* controls whether post-mortem debugging is used. If *pm*
1617 has a true value, the script file is run directly, and the debugger gets
1618 involved only if the script terminates via raising an unhandled exception. If
1619 it does, then post-mortem debugging is invoked, via :func:`pdb.post_mortem`,
1620 passing the traceback object from the unhandled exception. If *pm* is not
1621 specified, or is false, the script is run under the debugger from the start, via
1622 passing an appropriate :func:`exec` call to :func:`pdb.run`.
1623
Georg Brandl116aa622007-08-15 14:28:22 +00001624
1625.. function:: debug_src(src[, pm][, globs])
1626
1627 Debug the doctests in a string.
1628
1629 This is like function :func:`debug` above, except that a string containing
1630 doctest examples is specified directly, via the *src* argument.
1631
1632 Optional argument *pm* has the same meaning as in function :func:`debug` above.
1633
1634 Optional argument *globs* gives a dictionary to use as both local and global
1635 execution context. If not specified, or ``None``, an empty dictionary is used.
1636 If specified, a shallow copy of the dictionary is used.
1637
Georg Brandl116aa622007-08-15 14:28:22 +00001638
1639The :class:`DebugRunner` class, and the special exceptions it may raise, are of
1640most interest to testing framework authors, and will only be sketched here. See
1641the source code, and especially :class:`DebugRunner`'s docstring (which is a
1642doctest!) for more details:
1643
1644
1645.. class:: DebugRunner([checker][, verbose][, optionflags])
1646
1647 A subclass of :class:`DocTestRunner` that raises an exception as soon as a
1648 failure is encountered. If an unexpected exception occurs, an
1649 :exc:`UnexpectedException` exception is raised, containing the test, the
1650 example, and the original exception. If the output doesn't match, then a
1651 :exc:`DocTestFailure` exception is raised, containing the test, the example, and
1652 the actual output.
1653
1654 For information about the constructor parameters and methods, see the
1655 documentation for :class:`DocTestRunner` in section :ref:`doctest-advanced-api`.
1656
1657There are two exceptions that may be raised by :class:`DebugRunner` instances:
1658
1659
1660.. exception:: DocTestFailure(test, example, got)
1661
1662 An exception thrown by :class:`DocTestRunner` to signal that a doctest example's
1663 actual output did not match its expected output. The constructor arguments are
1664 used to initialize the member variables of the same names.
1665
1666:exc:`DocTestFailure` defines the following member variables:
1667
1668
1669.. attribute:: DocTestFailure.test
1670
1671 The :class:`DocTest` object that was being run when the example failed.
1672
1673
1674.. attribute:: DocTestFailure.example
1675
1676 The :class:`Example` that failed.
1677
1678
1679.. attribute:: DocTestFailure.got
1680
1681 The example's actual output.
1682
1683
1684.. exception:: UnexpectedException(test, example, exc_info)
1685
1686 An exception thrown by :class:`DocTestRunner` to signal that a doctest example
1687 raised an unexpected exception. The constructor arguments are used to
1688 initialize the member variables of the same names.
1689
1690:exc:`UnexpectedException` defines the following member variables:
1691
1692
1693.. attribute:: UnexpectedException.test
1694
1695 The :class:`DocTest` object that was being run when the example failed.
1696
1697
1698.. attribute:: UnexpectedException.example
1699
1700 The :class:`Example` that failed.
1701
1702
1703.. attribute:: UnexpectedException.exc_info
1704
1705 A tuple containing information about the unexpected exception, as returned by
1706 :func:`sys.exc_info`.
1707
1708
1709.. _doctest-soapbox:
1710
1711Soapbox
1712-------
1713
1714As mentioned in the introduction, :mod:`doctest` has grown to have three primary
1715uses:
1716
1717#. Checking examples in docstrings.
1718
1719#. Regression testing.
1720
1721#. Executable documentation / literate testing.
1722
1723These uses have different requirements, and it is important to distinguish them.
1724In particular, filling your docstrings with obscure test cases makes for bad
1725documentation.
1726
1727When writing a docstring, choose docstring examples with care. There's an art to
1728this that needs to be learned---it may not be natural at first. Examples should
1729add genuine value to the documentation. A good example can often be worth many
1730words. If done with care, the examples will be invaluable for your users, and
1731will pay back the time it takes to collect them many times over as the years go
1732by and things change. I'm still amazed at how often one of my :mod:`doctest`
1733examples stops working after a "harmless" change.
1734
1735Doctest also makes an excellent tool for regression testing, especially if you
1736don't skimp on explanatory text. By interleaving prose and examples, it becomes
1737much easier to keep track of what's actually being tested, and why. When a test
1738fails, good prose can make it much easier to figure out what the problem is, and
1739how it should be fixed. It's true that you could write extensive comments in
1740code-based testing, but few programmers do. Many have found that using doctest
1741approaches instead leads to much clearer tests. Perhaps this is simply because
1742doctest makes writing prose a little easier than writing code, while writing
1743comments in code is a little harder. I think it goes deeper than just that:
1744the natural attitude when writing a doctest-based test is that you want to
1745explain the fine points of your software, and illustrate them with examples.
1746This in turn naturally leads to test files that start with the simplest
1747features, and logically progress to complications and edge cases. A coherent
1748narrative is the result, instead of a collection of isolated functions that test
1749isolated bits of functionality seemingly at random. It's a different attitude,
1750and produces different results, blurring the distinction between testing and
1751explaining.
1752
1753Regression testing is best confined to dedicated objects or files. There are
1754several options for organizing tests:
1755
1756* Write text files containing test cases as interactive examples, and test the
1757 files using :func:`testfile` or :func:`DocFileSuite`. This is recommended,
1758 although is easiest to do for new projects, designed from the start to use
1759 doctest.
1760
1761* Define functions named ``_regrtest_topic`` that consist of single docstrings,
1762 containing test cases for the named topics. These functions can be included in
1763 the same file as the module, or separated out into a separate test file.
1764
1765* Define a ``__test__`` dictionary mapping from regression test topics to
1766 docstrings containing test cases.
1767
1768.. rubric:: Footnotes
1769
1770.. [#] Examples containing both expected output and an exception are not supported.
1771 Trying to guess where one ends and the other begins is too error-prone, and that
1772 also makes for a confusing test.
1773