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