blob: 3ecbdd02ac33e1001eb1bae7f50b9281697512e1 [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
Georg Brandl31835852008-05-12 17:38:56 +0000166There is also a command line shortcut for running :func:`testmod`. You can
167instruct the Python interpreter to run the doctest module directly from the
168standard library and pass the module name(s) on the command line::
Georg Brandl116aa622007-08-15 14:28:22 +0000169
170 python -m doctest -v example.py
171
172This will import :file:`example.py` as a standalone module and run
173:func:`testmod` on it. Note that this may not work correctly if the file is
174part of a package and imports other submodules from that package.
175
176For more information on :func:`testmod`, see section :ref:`doctest-basic-api`.
177
178
179.. _doctest-simple-testfile:
180
181Simple Usage: Checking Examples in a Text File
182----------------------------------------------
183
184Another simple application of doctest is testing interactive examples in a text
185file. This can be done with the :func:`testfile` function::
186
187 import doctest
188 doctest.testfile("example.txt")
189
190That short script executes and verifies any interactive Python examples
191contained in the file :file:`example.txt`. The file content is treated as if it
192were a single giant docstring; the file doesn't need to contain a Python
193program! For example, perhaps :file:`example.txt` contains this::
194
195 The ``example`` module
196 ======================
197
198 Using ``factorial``
199 -------------------
200
201 This is an example text file in reStructuredText format. First import
202 ``factorial`` from the ``example`` module:
203
204 >>> from example import factorial
205
206 Now use it:
207
208 >>> factorial(6)
209 120
210
211Running ``doctest.testfile("example.txt")`` then finds the error in this
212documentation::
213
214 File "./example.txt", line 14, in example.txt
215 Failed example:
216 factorial(6)
217 Expected:
218 120
219 Got:
220 720
221
222As with :func:`testmod`, :func:`testfile` won't display anything unless an
223example fails. If an example does fail, then the failing example(s) and the
224cause(s) of the failure(s) are printed to stdout, using the same format as
225:func:`testmod`.
226
227By default, :func:`testfile` looks for files in the calling module's directory.
228See section :ref:`doctest-basic-api` for a description of the optional arguments
229that can be used to tell it to look for files in other locations.
230
231Like :func:`testmod`, :func:`testfile`'s verbosity can be set with the
232:option:`-v` command-line switch or with the optional keyword argument
233*verbose*.
234
Georg Brandl31835852008-05-12 17:38:56 +0000235There is also a command line shortcut for running :func:`testfile`. You can
236instruct the Python interpreter to run the doctest module directly from the
237standard library and pass the file name(s) on the command line::
Georg Brandl116aa622007-08-15 14:28:22 +0000238
239 python -m doctest -v example.txt
240
241Because the file name does not end with :file:`.py`, :mod:`doctest` infers that
242it must be run with :func:`testfile`, not :func:`testmod`.
243
244For more information on :func:`testfile`, see section :ref:`doctest-basic-api`.
245
246
247.. _doctest-how-it-works:
248
249How It Works
250------------
251
252This section examines in detail how doctest works: which docstrings it looks at,
253how it finds interactive examples, what execution context it uses, how it
254handles exceptions, and how option flags can be used to control its behavior.
255This is the information that you need to know to write doctest examples; for
256information about actually running doctest on these examples, see the following
257sections.
258
259
260.. _doctest-which-docstrings:
261
262Which Docstrings Are Examined?
263^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
264
265The module docstring, and all function, class and method docstrings are
266searched. Objects imported into the module are not searched.
267
268In addition, if ``M.__test__`` exists and "is true", it must be a dict, and each
269entry maps a (string) name to a function object, class object, or string.
270Function and class object docstrings found from ``M.__test__`` are searched, and
271strings are treated as if they were docstrings. In output, a key ``K`` in
272``M.__test__`` appears with name ::
273
274 <name of M>.__test__.K
275
276Any classes found are recursively searched similarly, to test docstrings in
277their contained methods and nested classes.
278
Georg Brandl116aa622007-08-15 14:28:22 +0000279
280.. _doctest-finding-examples:
281
282How are Docstring Examples Recognized?
283^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
284
285In most cases a copy-and-paste of an interactive console session works fine, but
286doctest isn't trying to do an exact emulation of any specific Python shell. All
287hard tab characters are expanded to spaces, using 8-column tab stops. If you
288don't believe tabs should mean that, too bad: don't use hard tabs, or write
289your own :class:`DocTestParser` class.
290
Georg Brandl116aa622007-08-15 14:28:22 +0000291::
292
293 >>> # comments are ignored
294 >>> x = 12
295 >>> x
296 12
297 >>> if x == 13:
Georg Brandl6911e3c2007-09-04 07:15:32 +0000298 ... print("yes")
Georg Brandl116aa622007-08-15 14:28:22 +0000299 ... else:
Georg Brandl6911e3c2007-09-04 07:15:32 +0000300 ... print("no")
301 ... print("NO")
302 ... print("NO!!!")
Georg Brandl116aa622007-08-15 14:28:22 +0000303 ...
304 no
305 NO
306 NO!!!
307 >>>
308
309Any expected output must immediately follow the final ``'>>> '`` or ``'... '``
310line containing the code, and the expected output (if any) extends to the next
311``'>>> '`` or all-whitespace line.
312
313The fine print:
314
315* Expected output cannot contain an all-whitespace line, since such a line is
316 taken to signal the end of expected output. If expected output does contain a
317 blank line, put ``<BLANKLINE>`` in your doctest example each place a blank line
318 is expected.
319
Georg Brandl116aa622007-08-15 14:28:22 +0000320* Output to stdout is captured, but not output to stderr (exception tracebacks
321 are captured via a different means).
322
323* If you continue a line via backslashing in an interactive session, or for any
324 other reason use a backslash, you should use a raw docstring, which will
325 preserve your backslashes exactly as you type them::
326
327 >>> def f(x):
328 ... r'''Backslashes in a raw docstring: m\n'''
Georg Brandl6911e3c2007-09-04 07:15:32 +0000329 >>> print(f.__doc__)
Georg Brandl116aa622007-08-15 14:28:22 +0000330 Backslashes in a raw docstring: m\n
331
332 Otherwise, the backslash will be interpreted as part of the string. For example,
333 the "\\" above would be interpreted as a newline character. Alternatively, you
334 can double each backslash in the doctest version (and not use a raw string)::
335
336 >>> def f(x):
337 ... '''Backslashes in a raw docstring: m\\n'''
Georg Brandl6911e3c2007-09-04 07:15:32 +0000338 >>> print(f.__doc__)
Georg Brandl116aa622007-08-15 14:28:22 +0000339 Backslashes in a raw docstring: m\n
340
341* The starting column doesn't matter::
342
343 >>> assert "Easy!"
344 >>> import math
345 >>> math.floor(1.9)
R. David Murray7c5714f2009-11-23 03:13:23 +0000346 1
Georg Brandl116aa622007-08-15 14:28:22 +0000347
348 and as many leading whitespace characters are stripped from the expected output
349 as appeared in the initial ``'>>> '`` line that started the example.
350
351
352.. _doctest-execution-context:
353
354What's the Execution Context?
355^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
356
357By default, each time :mod:`doctest` finds a docstring to test, it uses a
358*shallow copy* of :mod:`M`'s globals, so that running tests doesn't change the
359module's real globals, and so that one test in :mod:`M` can't leave behind
360crumbs that accidentally allow another test to work. This means examples can
361freely use any names defined at top-level in :mod:`M`, and names defined earlier
362in the docstring being run. Examples cannot see names defined in other
363docstrings.
364
365You can force use of your own dict as the execution context by passing
366``globs=your_dict`` to :func:`testmod` or :func:`testfile` instead.
367
368
369.. _doctest-exceptions:
370
371What About Exceptions?
372^^^^^^^^^^^^^^^^^^^^^^
373
374No problem, provided that the traceback is the only output produced by the
375example: just paste in the traceback. [#]_ Since tracebacks contain details
376that are likely to change rapidly (for example, exact file paths and line
377numbers), this is one case where doctest works hard to be flexible in what it
378accepts.
379
380Simple example::
381
382 >>> [1, 2, 3].remove(42)
383 Traceback (most recent call last):
384 File "<stdin>", line 1, in ?
385 ValueError: list.remove(x): x not in list
386
387That doctest succeeds if :exc:`ValueError` is raised, with the ``list.remove(x):
388x not in list`` detail as shown.
389
390The expected output for an exception must start with a traceback header, which
391may be either of the following two lines, indented the same as the first line of
392the example::
393
394 Traceback (most recent call last):
395 Traceback (innermost last):
396
397The traceback header is followed by an optional traceback stack, whose contents
398are ignored by doctest. The traceback stack is typically omitted, or copied
399verbatim from an interactive session.
400
401The traceback stack is followed by the most interesting part: the line(s)
402containing the exception type and detail. This is usually the last line of a
403traceback, but can extend across multiple lines if the exception has a
404multi-line detail::
405
406 >>> raise ValueError('multi\n line\ndetail')
407 Traceback (most recent call last):
408 File "<stdin>", line 1, in ?
409 ValueError: multi
410 line
411 detail
412
413The last three lines (starting with :exc:`ValueError`) are compared against the
414exception's type and detail, and the rest are ignored.
415
416Best practice is to omit the traceback stack, unless it adds significant
417documentation value to the example. So the last example is probably better as::
418
419 >>> raise ValueError('multi\n line\ndetail')
420 Traceback (most recent call last):
421 ...
422 ValueError: multi
423 line
424 detail
425
426Note that tracebacks are treated very specially. In particular, in the
427rewritten example, the use of ``...`` is independent of doctest's
428:const:`ELLIPSIS` option. The ellipsis in that example could be left out, or
429could just as well be three (or three hundred) commas or digits, or an indented
430transcript of a Monty Python skit.
431
432Some details you should read once, but won't need to remember:
433
434* Doctest can't guess whether your expected output came from an exception
435 traceback or from ordinary printing. So, e.g., an example that expects
436 ``ValueError: 42 is prime`` will pass whether :exc:`ValueError` is actually
437 raised or if the example merely prints that traceback text. In practice,
438 ordinary output rarely begins with a traceback header line, so this doesn't
439 create real problems.
440
441* Each line of the traceback stack (if present) must be indented further than
442 the first line of the example, *or* start with a non-alphanumeric character.
443 The first line following the traceback header indented the same and starting
444 with an alphanumeric is taken to be the start of the exception detail. Of
445 course this does the right thing for genuine tracebacks.
446
Nick Coghlan5e76e942010-06-12 13:42:46 +0000447* When the :const:`IGNORE_EXCEPTION_DETAIL` doctest option is specified,
448 everything following the leftmost colon and any module information in the
449 exception name is ignored.
Georg Brandl116aa622007-08-15 14:28:22 +0000450
451* The interactive shell omits the traceback header line for some
452 :exc:`SyntaxError`\ s. But doctest uses the traceback header line to
453 distinguish exceptions from non-exceptions. So in the rare case where you need
454 to test a :exc:`SyntaxError` that omits the traceback header, you will need to
455 manually add the traceback header line to your test example.
456
457* For some :exc:`SyntaxError`\ s, Python displays the character position of the
458 syntax error, using a ``^`` marker::
459
460 >>> 1 1
461 File "<stdin>", line 1
462 1 1
463 ^
464 SyntaxError: invalid syntax
465
466 Since the lines showing the position of the error come before the exception type
467 and detail, they are not checked by doctest. For example, the following test
468 would pass, even though it puts the ``^`` marker in the wrong location::
469
470 >>> 1 1
471 Traceback (most recent call last):
472 File "<stdin>", line 1
473 1 1
474 ^
475 SyntaxError: invalid syntax
476
Georg Brandl116aa622007-08-15 14:28:22 +0000477
478.. _doctest-options:
479
480Option Flags and Directives
481^^^^^^^^^^^^^^^^^^^^^^^^^^^
482
483A number of option flags control various aspects of doctest's behavior.
484Symbolic names for the flags are supplied as module constants, which can be
485or'ed together and passed to various functions. The names can also be used in
486doctest directives (see below).
487
488The first group of options define test semantics, controlling aspects of how
489doctest decides whether actual output matches an example's expected output:
490
491
492.. data:: DONT_ACCEPT_TRUE_FOR_1
493
494 By default, if an expected output block contains just ``1``, an actual output
495 block containing just ``1`` or just ``True`` is considered to be a match, and
496 similarly for ``0`` versus ``False``. When :const:`DONT_ACCEPT_TRUE_FOR_1` is
497 specified, neither substitution is allowed. The default behavior caters to that
498 Python changed the return type of many functions from integer to boolean;
499 doctests expecting "little integer" output still work in these cases. This
500 option will probably go away, but not for several years.
501
502
503.. data:: DONT_ACCEPT_BLANKLINE
504
505 By default, if an expected output block contains a line containing only the
506 string ``<BLANKLINE>``, then that line will match a blank line in the actual
507 output. Because a genuinely blank line delimits the expected output, this is
508 the only way to communicate that a blank line is expected. When
509 :const:`DONT_ACCEPT_BLANKLINE` is specified, this substitution is not allowed.
510
511
512.. data:: NORMALIZE_WHITESPACE
513
514 When specified, all sequences of whitespace (blanks and newlines) are treated as
515 equal. Any sequence of whitespace within the expected output will match any
516 sequence of whitespace within the actual output. By default, whitespace must
517 match exactly. :const:`NORMALIZE_WHITESPACE` is especially useful when a line of
518 expected output is very long, and you want to wrap it across multiple lines in
519 your source.
520
521
522.. data:: ELLIPSIS
523
524 When specified, an ellipsis marker (``...``) in the expected output can match
525 any substring in the actual output. This includes substrings that span line
526 boundaries, and empty substrings, so it's best to keep usage of this simple.
527 Complicated uses can lead to the same kinds of "oops, it matched too much!"
528 surprises that ``.*`` is prone to in regular expressions.
529
530
531.. data:: IGNORE_EXCEPTION_DETAIL
532
533 When specified, an example that expects an exception passes if an exception of
534 the expected type is raised, even if the exception detail does not match. For
535 example, an example expecting ``ValueError: 42`` will pass if the actual
536 exception raised is ``ValueError: 3*14``, but will fail, e.g., if
537 :exc:`TypeError` is raised.
538
Nick Coghlan5e76e942010-06-12 13:42:46 +0000539 It will also ignore the module name used in Python 3 doctest reports. Hence
540 both these variations will work regardless of whether the test is run under
541 Python 2.7 or Python 3.2 (or later versions):
542
543 >>> raise CustomError('message') #doctest: +IGNORE_EXCEPTION_DETAIL
544 Traceback (most recent call last):
545 CustomError: message
546
547 >>> raise CustomError('message') #doctest: +IGNORE_EXCEPTION_DETAIL
548 Traceback (most recent call last):
549 my_module.CustomError: message
550
551 Note that :const:`ELLIPSIS` can also be used to ignore the
552 details of the exception message, but such a test may still fail based
553 on whether or not the module details are printed as part of the
554 exception name. Using :const:`IGNORE_EXCEPTION_DETAIL` and the details
555 from Python 2.3 is also the only clear way to write a doctest that doesn't
556 care about the exception detail yet continues to pass under Python 2.3 or
557 earlier (those releases do not support doctest directives and ignore them
558 as irrelevant comments). For example, ::
Georg Brandl116aa622007-08-15 14:28:22 +0000559
560 >>> (1, 2)[3] = 'moo' #doctest: +IGNORE_EXCEPTION_DETAIL
561 Traceback (most recent call last):
562 File "<stdin>", line 1, in ?
563 TypeError: object doesn't support item assignment
564
Nick Coghlan5e76e942010-06-12 13:42:46 +0000565 passes under Python 2.3 and later Python versions, even though the detail
566 changed in Python 2.4 to say "does not" instead of "doesn't".
567
568 .. versionchanged:: 3.2
569 :const:`IGNORE_EXCEPTION_DETAIL` now also ignores any information
570 relating to the module containing the exception under test
Georg Brandl116aa622007-08-15 14:28:22 +0000571
572
573.. data:: SKIP
574
575 When specified, do not run the example at all. This can be useful in contexts
576 where doctest examples serve as both documentation and test cases, and an
577 example should be included for documentation purposes, but should not be
578 checked. E.g., the example's output might be random; or the example might
579 depend on resources which would be unavailable to the test driver.
580
581 The SKIP flag can also be used for temporarily "commenting out" examples.
582
583
584.. data:: COMPARISON_FLAGS
585
586 A bitmask or'ing together all the comparison flags above.
587
588The second group of options controls how test failures are reported:
589
590
591.. data:: REPORT_UDIFF
592
593 When specified, failures that involve multi-line expected and actual outputs are
594 displayed using a unified diff.
595
596
597.. data:: REPORT_CDIFF
598
599 When specified, failures that involve multi-line expected and actual outputs
600 will be displayed using a context diff.
601
602
603.. data:: REPORT_NDIFF
604
605 When specified, differences are computed by ``difflib.Differ``, using the same
606 algorithm as the popular :file:`ndiff.py` utility. This is the only method that
607 marks differences within lines as well as across lines. For example, if a line
608 of expected output contains digit ``1`` where actual output contains letter
609 ``l``, a line is inserted with a caret marking the mismatching column positions.
610
611
612.. data:: REPORT_ONLY_FIRST_FAILURE
613
614 When specified, display the first failing example in each doctest, but suppress
615 output for all remaining examples. This will prevent doctest from reporting
616 correct examples that break because of earlier failures; but it might also hide
617 incorrect examples that fail independently of the first failure. When
618 :const:`REPORT_ONLY_FIRST_FAILURE` is specified, the remaining examples are
619 still run, and still count towards the total number of failures reported; only
620 the output is suppressed.
621
622
623.. data:: REPORTING_FLAGS
624
625 A bitmask or'ing together all the reporting flags above.
626
627"Doctest directives" may be used to modify the option flags for individual
628examples. Doctest directives are expressed as a special Python comment
629following an example's source code:
630
631.. productionlist:: doctest
632 directive: "#" "doctest:" `directive_options`
633 directive_options: `directive_option` ("," `directive_option`)\*
634 directive_option: `on_or_off` `directive_option_name`
635 on_or_off: "+" \| "-"
636 directive_option_name: "DONT_ACCEPT_BLANKLINE" \| "NORMALIZE_WHITESPACE" \| ...
637
638Whitespace is not allowed between the ``+`` or ``-`` and the directive option
639name. The directive option name can be any of the option flag names explained
640above.
641
642An example's doctest directives modify doctest's behavior for that single
643example. Use ``+`` to enable the named behavior, or ``-`` to disable it.
644
645For example, this test passes::
646
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000647 >>> print(list(range(20))) #doctest: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000648 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9,
649 10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
650
651Without the directive it would fail, both because the actual output doesn't have
652two blanks before the single-digit list elements, and because the actual output
653is on a single line. This test also passes, and also requires a directive to do
654so::
655
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000656 >>> print(list(range(20))) # doctest: +ELLIPSIS
Georg Brandl116aa622007-08-15 14:28:22 +0000657 [0, 1, ..., 18, 19]
658
659Multiple directives can be used on a single physical line, separated by commas::
660
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000661 >>> print(list(range(20))) # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000662 [0, 1, ..., 18, 19]
663
664If multiple directive comments are used for a single example, then they are
665combined::
666
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000667 >>> print(list(range(20))) # doctest: +ELLIPSIS
668 ... # doctest: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000669 [0, 1, ..., 18, 19]
670
671As the previous example shows, you can add ``...`` lines to your example
672containing only directives. This can be useful when an example is too long for
673a directive to comfortably fit on the same line::
674
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000675 >>> print(list(range(5)) + list(range(10, 20)) + list(range(30, 40)))
Georg Brandl116aa622007-08-15 14:28:22 +0000676 ... # doctest: +ELLIPSIS
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000677 [0, ..., 4, 10, ..., 19, 30, ..., 39]
Georg Brandl116aa622007-08-15 14:28:22 +0000678
679Note that since all options are disabled by default, and directives apply only
680to the example they appear in, enabling options (via ``+`` in a directive) is
681usually the only meaningful choice. However, option flags can also be passed to
682functions that run doctests, establishing different defaults. In such cases,
683disabling an option via ``-`` in a directive can be useful.
684
Georg Brandl116aa622007-08-15 14:28:22 +0000685There's also a way to register new option flag names, although this isn't useful
686unless you intend to extend :mod:`doctest` internals via subclassing:
687
688
689.. function:: register_optionflag(name)
690
691 Create a new option flag with a given name, and return the new flag's integer
692 value. :func:`register_optionflag` can be used when subclassing
693 :class:`OutputChecker` or :class:`DocTestRunner` to create new options that are
694 supported by your subclasses. :func:`register_optionflag` should always be
695 called using the following idiom::
696
697 MY_FLAG = register_optionflag('MY_FLAG')
698
Georg Brandl116aa622007-08-15 14:28:22 +0000699
700.. _doctest-warnings:
701
702Warnings
703^^^^^^^^
704
705:mod:`doctest` is serious about requiring exact matches in expected output. If
706even a single character doesn't match, the test fails. This will probably
707surprise you a few times, as you learn exactly what Python does and doesn't
708guarantee about output. For example, when printing a dict, Python doesn't
709guarantee that the key-value pairs will be printed in any particular order, so a
Christian Heimes5b5e81c2007-12-31 16:14:33 +0000710test like ::
Georg Brandl116aa622007-08-15 14:28:22 +0000711
712 >>> foo()
713 {"Hermione": "hippogryph", "Harry": "broomstick"}
714
715is vulnerable! One workaround is to do ::
716
717 >>> foo() == {"Hermione": "hippogryph", "Harry": "broomstick"}
718 True
719
720instead. Another is to do ::
721
Ezio Melotti8f7649e2009-09-13 04:48:45 +0000722 >>> d = sorted(foo().items())
Georg Brandl116aa622007-08-15 14:28:22 +0000723 >>> d
724 [('Harry', 'broomstick'), ('Hermione', 'hippogryph')]
725
726There are others, but you get the idea.
727
728Another bad idea is to print things that embed an object address, like ::
729
730 >>> id(1.0) # certain to fail some of the time
731 7948648
732 >>> class C: pass
733 >>> C() # the default repr() for instances embeds an address
734 <__main__.C instance at 0x00AC18F0>
735
736The :const:`ELLIPSIS` directive gives a nice approach for the last example::
737
738 >>> C() #doctest: +ELLIPSIS
739 <__main__.C instance at 0x...>
740
741Floating-point numbers are also subject to small output variations across
742platforms, because Python defers to the platform C library for float formatting,
743and C libraries vary widely in quality here. ::
744
745 >>> 1./7 # risky
746 0.14285714285714285
Georg Brandl6911e3c2007-09-04 07:15:32 +0000747 >>> print(1./7) # safer
Georg Brandl116aa622007-08-15 14:28:22 +0000748 0.142857142857
Georg Brandl6911e3c2007-09-04 07:15:32 +0000749 >>> print(round(1./7, 6)) # much safer
Georg Brandl116aa622007-08-15 14:28:22 +0000750 0.142857
751
752Numbers of the form ``I/2.**J`` are safe across all platforms, and I often
753contrive doctest examples to produce numbers of that form::
754
755 >>> 3./4 # utterly safe
756 0.75
757
758Simple fractions are also easier for people to understand, and that makes for
759better documentation.
760
761
762.. _doctest-basic-api:
763
764Basic API
765---------
766
767The functions :func:`testmod` and :func:`testfile` provide a simple interface to
768doctest that should be sufficient for most basic uses. For a less formal
769introduction to these two functions, see sections :ref:`doctest-simple-testmod`
770and :ref:`doctest-simple-testfile`.
771
772
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000773.. function:: testfile(filename, module_relative=True, name=None, package=None, globs=None, verbose=None, report=True, optionflags=0, extraglobs=None, raise_on_error=False, parser=DocTestParser(), encoding=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000774
775 All arguments except *filename* are optional, and should be specified in keyword
776 form.
777
778 Test examples in the file named *filename*. Return ``(failure_count,
779 test_count)``.
780
781 Optional argument *module_relative* specifies how the filename should be
782 interpreted:
783
784 * If *module_relative* is ``True`` (the default), then *filename* specifies an
785 OS-independent module-relative path. By default, this path is relative to the
786 calling module's directory; but if the *package* argument is specified, then it
787 is relative to that package. To ensure OS-independence, *filename* should use
788 ``/`` characters to separate path segments, and may not be an absolute path
789 (i.e., it may not begin with ``/``).
790
791 * If *module_relative* is ``False``, then *filename* specifies an OS-specific
792 path. The path may be absolute or relative; relative paths are resolved with
793 respect to the current working directory.
794
795 Optional argument *name* gives the name of the test; by default, or if ``None``,
796 ``os.path.basename(filename)`` is used.
797
798 Optional argument *package* is a Python package or the name of a Python package
799 whose directory should be used as the base directory for a module-relative
800 filename. If no package is specified, then the calling module's directory is
801 used as the base directory for module-relative filenames. It is an error to
802 specify *package* if *module_relative* is ``False``.
803
804 Optional argument *globs* gives a dict to be used as the globals when executing
805 examples. A new shallow copy of this dict is created for the doctest, so its
806 examples start with a clean slate. By default, or if ``None``, a new empty dict
807 is used.
808
809 Optional argument *extraglobs* gives a dict merged into the globals used to
810 execute examples. This works like :meth:`dict.update`: if *globs* and
811 *extraglobs* have a common key, the associated value in *extraglobs* appears in
812 the combined dict. By default, or if ``None``, no extra globals are used. This
813 is an advanced feature that allows parameterization of doctests. For example, a
814 doctest can be written for a base class, using a generic name for the class,
815 then reused to test any number of subclasses by passing an *extraglobs* dict
816 mapping the generic name to the subclass to be tested.
817
818 Optional argument *verbose* prints lots of stuff if true, and prints only
819 failures if false; by default, or if ``None``, it's true if and only if ``'-v'``
820 is in ``sys.argv``.
821
822 Optional argument *report* prints a summary at the end when true, else prints
823 nothing at the end. In verbose mode, the summary is detailed, else the summary
824 is very brief (in fact, empty if all tests passed).
825
826 Optional argument *optionflags* or's together option flags. See section
827 :ref:`doctest-options`.
828
829 Optional argument *raise_on_error* defaults to false. If true, an exception is
830 raised upon the first failure or unexpected exception in an example. This
831 allows failures to be post-mortem debugged. Default behavior is to continue
832 running examples.
833
834 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass) that
835 should be used to extract tests from the files. It defaults to a normal parser
836 (i.e., ``DocTestParser()``).
837
838 Optional argument *encoding* specifies an encoding that should be used to
839 convert the file to unicode.
840
Georg Brandl116aa622007-08-15 14:28:22 +0000841
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000842.. function:: testmod(m=None, name=None, globs=None, verbose=None, report=True, optionflags=0, extraglobs=None, raise_on_error=False, exclude_empty=False)
Georg Brandl116aa622007-08-15 14:28:22 +0000843
844 All arguments are optional, and all except for *m* should be specified in
845 keyword form.
846
847 Test examples in docstrings in functions and classes reachable from module *m*
848 (or module :mod:`__main__` if *m* is not supplied or is ``None``), starting with
849 ``m.__doc__``.
850
851 Also test examples reachable from dict ``m.__test__``, if it exists and is not
852 ``None``. ``m.__test__`` maps names (strings) to functions, classes and
853 strings; function and class docstrings are searched for examples; strings are
854 searched directly, as if they were docstrings.
855
856 Only docstrings attached to objects belonging to module *m* are searched.
857
858 Return ``(failure_count, test_count)``.
859
860 Optional argument *name* gives the name of the module; by default, or if
861 ``None``, ``m.__name__`` is used.
862
863 Optional argument *exclude_empty* defaults to false. If true, objects for which
864 no doctests are found are excluded from consideration. The default is a backward
865 compatibility hack, so that code still using :meth:`doctest.master.summarize` in
866 conjunction with :func:`testmod` continues to get output for objects with no
867 tests. The *exclude_empty* argument to the newer :class:`DocTestFinder`
868 constructor defaults to true.
869
870 Optional arguments *extraglobs*, *verbose*, *report*, *optionflags*,
871 *raise_on_error*, and *globs* are the same as for function :func:`testfile`
872 above, except that *globs* defaults to ``m.__dict__``.
873
Georg Brandl116aa622007-08-15 14:28:22 +0000874
875There's also a function to run the doctests associated with a single object.
876This function is provided for backward compatibility. There are no plans to
877deprecate it, but it's rarely useful:
878
879
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000880.. function:: run_docstring_examples(f, globs, verbose=False, name="NoName", compileflags=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +0000881
882 Test examples associated with object *f*; for example, *f* may be a module,
883 function, or class object.
884
885 A shallow copy of dictionary argument *globs* is used for the execution context.
886
887 Optional argument *name* is used in failure messages, and defaults to
888 ``"NoName"``.
889
890 If optional argument *verbose* is true, output is generated even if there are no
891 failures. By default, output is generated only in case of an example failure.
892
893 Optional argument *compileflags* gives the set of flags that should be used by
894 the Python compiler when running the examples. By default, or if ``None``,
895 flags are deduced corresponding to the set of future features found in *globs*.
896
897 Optional argument *optionflags* works as for function :func:`testfile` above.
898
899
900.. _doctest-unittest-api:
901
902Unittest API
903------------
904
905As your collection of doctest'ed modules grows, you'll want a way to run all
Georg Brandl31835852008-05-12 17:38:56 +0000906their doctests systematically. :mod:`doctest` provides two functions that can
907be used to create :mod:`unittest` test suites from modules and text files
908containing doctests. These test suites can then be run using :mod:`unittest`
909test runners::
Georg Brandl116aa622007-08-15 14:28:22 +0000910
911 import unittest
912 import doctest
913 import my_module_with_doctests, and_another
914
915 suite = unittest.TestSuite()
916 for mod in my_module_with_doctests, and_another:
917 suite.addTest(doctest.DocTestSuite(mod))
918 runner = unittest.TextTestRunner()
919 runner.run(suite)
920
921There are two main functions for creating :class:`unittest.TestSuite` instances
922from text files and modules with doctests:
923
924
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000925.. function:: DocFileSuite(*paths, module_relative=True, package=None, setUp=None, tearDown=None, globs=None, optionflags=0, parser=DocTestParser(), encoding=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000926
927 Convert doctest tests from one or more text files to a
928 :class:`unittest.TestSuite`.
929
930 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
931 and runs the interactive examples in each file. If an example in any file
932 fails, then the synthesized unit test fails, and a :exc:`failureException`
933 exception is raised showing the name of the file containing the test and a
934 (sometimes approximate) line number.
935
936 Pass one or more paths (as strings) to text files to be examined.
937
938 Options may be provided as keyword arguments:
939
940 Optional argument *module_relative* specifies how the filenames in *paths*
941 should be interpreted:
942
Benjamin Petersond23f8222009-04-05 19:13:16 +0000943 * If *module_relative* is ``True`` (the default), then each filename in
944 *paths* specifies an OS-independent module-relative path. By default, this
945 path is relative to the calling module's directory; but if the *package*
946 argument is specified, then it is relative to that package. To ensure
947 OS-independence, each filename should use ``/`` characters to separate path
948 segments, and may not be an absolute path (i.e., it may not begin with
949 ``/``).
Georg Brandl116aa622007-08-15 14:28:22 +0000950
Benjamin Petersond23f8222009-04-05 19:13:16 +0000951 * If *module_relative* is ``False``, then each filename in *paths* specifies
952 an OS-specific path. The path may be absolute or relative; relative paths
953 are resolved with respect to the current working directory.
Georg Brandl116aa622007-08-15 14:28:22 +0000954
Benjamin Petersond23f8222009-04-05 19:13:16 +0000955 Optional argument *package* is a Python package or the name of a Python
956 package whose directory should be used as the base directory for
957 module-relative filenames in *paths*. If no package is specified, then the
958 calling module's directory is used as the base directory for module-relative
959 filenames. It is an error to specify *package* if *module_relative* is
960 ``False``.
Georg Brandl116aa622007-08-15 14:28:22 +0000961
Benjamin Petersond23f8222009-04-05 19:13:16 +0000962 Optional argument *setUp* specifies a set-up function for the test suite.
963 This is called before running the tests in each file. The *setUp* function
Georg Brandl116aa622007-08-15 14:28:22 +0000964 will be passed a :class:`DocTest` object. The setUp function can access the
965 test globals as the *globs* attribute of the test passed.
966
Benjamin Petersond23f8222009-04-05 19:13:16 +0000967 Optional argument *tearDown* specifies a tear-down function for the test
968 suite. This is called after running the tests in each file. The *tearDown*
969 function will be passed a :class:`DocTest` object. The setUp function can
970 access the test globals as the *globs* attribute of the test passed.
971
Georg Brandl116aa622007-08-15 14:28:22 +0000972 Optional argument *globs* is a dictionary containing the initial global
973 variables for the tests. A new copy of this dictionary is created for each
974 test. By default, *globs* is a new empty dictionary.
975
976 Optional argument *optionflags* specifies the default doctest options for the
977 tests, created by or-ing together individual option flags. See section
Benjamin Petersond23f8222009-04-05 19:13:16 +0000978 :ref:`doctest-options`. See function :func:`set_unittest_reportflags` below
979 for a better way to set reporting options.
Georg Brandl116aa622007-08-15 14:28:22 +0000980
Benjamin Petersond23f8222009-04-05 19:13:16 +0000981 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass)
982 that should be used to extract tests from the files. It defaults to a normal
983 parser (i.e., ``DocTestParser()``).
Georg Brandl116aa622007-08-15 14:28:22 +0000984
985 Optional argument *encoding* specifies an encoding that should be used to
986 convert the file to unicode.
987
Georg Brandl55ac8f02007-09-01 13:51:09 +0000988 The global ``__file__`` is added to the globals provided to doctests loaded
989 from a text file using :func:`DocFileSuite`.
Georg Brandl116aa622007-08-15 14:28:22 +0000990
991
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000992.. function:: DocTestSuite(module=None, globs=None, extraglobs=None, test_finder=None, setUp=None, tearDown=None, checker=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000993
994 Convert doctest tests for a module to a :class:`unittest.TestSuite`.
995
996 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
997 and runs each doctest in the module. If any of the doctests fail, then the
998 synthesized unit test fails, and a :exc:`failureException` exception is raised
999 showing the name of the file containing the test and a (sometimes approximate)
1000 line number.
1001
1002 Optional argument *module* provides the module to be tested. It can be a module
1003 object or a (possibly dotted) module name. If not specified, the module calling
1004 this function is used.
1005
1006 Optional argument *globs* is a dictionary containing the initial global
1007 variables for the tests. A new copy of this dictionary is created for each
1008 test. By default, *globs* is a new empty dictionary.
1009
1010 Optional argument *extraglobs* specifies an extra set of global variables, which
1011 is merged into *globs*. By default, no extra globals are used.
1012
1013 Optional argument *test_finder* is the :class:`DocTestFinder` object (or a
1014 drop-in replacement) that is used to extract doctests from the module.
1015
1016 Optional arguments *setUp*, *tearDown*, and *optionflags* are the same as for
1017 function :func:`DocFileSuite` above.
1018
Georg Brandl55ac8f02007-09-01 13:51:09 +00001019 This function uses the same search technique as :func:`testmod`.
Georg Brandl116aa622007-08-15 14:28:22 +00001020
Georg Brandl116aa622007-08-15 14:28:22 +00001021
1022Under the covers, :func:`DocTestSuite` creates a :class:`unittest.TestSuite` out
1023of :class:`doctest.DocTestCase` instances, and :class:`DocTestCase` is a
1024subclass of :class:`unittest.TestCase`. :class:`DocTestCase` isn't documented
1025here (it's an internal detail), but studying its code can answer questions about
1026the exact details of :mod:`unittest` integration.
1027
1028Similarly, :func:`DocFileSuite` creates a :class:`unittest.TestSuite` out of
1029:class:`doctest.DocFileCase` instances, and :class:`DocFileCase` is a subclass
1030of :class:`DocTestCase`.
1031
1032So both ways of creating a :class:`unittest.TestSuite` run instances of
1033:class:`DocTestCase`. This is important for a subtle reason: when you run
1034:mod:`doctest` functions yourself, you can control the :mod:`doctest` options in
1035use directly, by passing option flags to :mod:`doctest` functions. However, if
1036you're writing a :mod:`unittest` framework, :mod:`unittest` ultimately controls
1037when and how tests get run. The framework author typically wants to control
1038:mod:`doctest` reporting options (perhaps, e.g., specified by command line
1039options), but there's no way to pass options through :mod:`unittest` to
1040:mod:`doctest` test runners.
1041
1042For this reason, :mod:`doctest` also supports a notion of :mod:`doctest`
1043reporting flags specific to :mod:`unittest` support, via this function:
1044
1045
1046.. function:: set_unittest_reportflags(flags)
1047
1048 Set the :mod:`doctest` reporting flags to use.
1049
1050 Argument *flags* or's together option flags. See section
1051 :ref:`doctest-options`. Only "reporting flags" can be used.
1052
1053 This is a module-global setting, and affects all future doctests run by module
1054 :mod:`unittest`: the :meth:`runTest` method of :class:`DocTestCase` looks at
1055 the option flags specified for the test case when the :class:`DocTestCase`
1056 instance was constructed. If no reporting flags were specified (which is the
1057 typical and expected case), :mod:`doctest`'s :mod:`unittest` reporting flags are
1058 or'ed into the option flags, and the option flags so augmented are passed to the
1059 :class:`DocTestRunner` instance created to run the doctest. If any reporting
1060 flags were specified when the :class:`DocTestCase` instance was constructed,
1061 :mod:`doctest`'s :mod:`unittest` reporting flags are ignored.
1062
1063 The value of the :mod:`unittest` reporting flags in effect before the function
1064 was called is returned by the function.
1065
Georg Brandl116aa622007-08-15 14:28:22 +00001066
1067.. _doctest-advanced-api:
1068
1069Advanced API
1070------------
1071
1072The basic API is a simple wrapper that's intended to make doctest easy to use.
1073It is fairly flexible, and should meet most users' needs; however, if you
1074require more fine-grained control over testing, or wish to extend doctest's
1075capabilities, then you should use the advanced API.
1076
1077The advanced API revolves around two container classes, which are used to store
1078the interactive examples extracted from doctest cases:
1079
Ezio Melotti0639d5a2009-12-19 23:26:38 +00001080* :class:`Example`: A single Python :term:`statement`, paired with its expected
Christian Heimesd8654cf2007-12-02 15:22:16 +00001081 output.
Georg Brandl116aa622007-08-15 14:28:22 +00001082
1083* :class:`DocTest`: A collection of :class:`Example`\ s, typically extracted
1084 from a single docstring or text file.
1085
1086Additional processing classes are defined to find, parse, and run, and check
1087doctest examples:
1088
1089* :class:`DocTestFinder`: Finds all docstrings in a given module, and uses a
1090 :class:`DocTestParser` to create a :class:`DocTest` from every docstring that
1091 contains interactive examples.
1092
1093* :class:`DocTestParser`: Creates a :class:`DocTest` object from a string (such
1094 as an object's docstring).
1095
1096* :class:`DocTestRunner`: Executes the examples in a :class:`DocTest`, and uses
1097 an :class:`OutputChecker` to verify their output.
1098
1099* :class:`OutputChecker`: Compares the actual output from a doctest example with
1100 the expected output, and decides whether they match.
1101
1102The relationships among these processing classes are summarized in the following
1103diagram::
1104
1105 list of:
1106 +------+ +---------+
1107 |module| --DocTestFinder-> | DocTest | --DocTestRunner-> results
1108 +------+ | ^ +---------+ | ^ (printed)
1109 | | | Example | | |
1110 v | | ... | v |
1111 DocTestParser | Example | OutputChecker
1112 +---------+
1113
1114
1115.. _doctest-doctest:
1116
1117DocTest Objects
1118^^^^^^^^^^^^^^^
1119
1120
1121.. class:: DocTest(examples, globs, name, filename, lineno, docstring)
1122
1123 A collection of doctest examples that should be run in a single namespace. The
1124 constructor arguments are used to initialize the member variables of the same
1125 names.
1126
Georg Brandl116aa622007-08-15 14:28:22 +00001127
Benjamin Petersone41251e2008-04-25 01:59:09 +00001128 :class:`DocTest` defines the following member variables. They are initialized by
1129 the constructor, and should not be modified directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001130
1131
Benjamin Petersone41251e2008-04-25 01:59:09 +00001132 .. attribute:: examples
Georg Brandl116aa622007-08-15 14:28:22 +00001133
Benjamin Petersone41251e2008-04-25 01:59:09 +00001134 A list of :class:`Example` objects encoding the individual interactive Python
1135 examples that should be run by this test.
Georg Brandl116aa622007-08-15 14:28:22 +00001136
1137
Benjamin Petersone41251e2008-04-25 01:59:09 +00001138 .. attribute:: globs
Georg Brandl116aa622007-08-15 14:28:22 +00001139
Benjamin Petersone41251e2008-04-25 01:59:09 +00001140 The namespace (aka globals) that the examples should be run in. This is a
1141 dictionary mapping names to values. Any changes to the namespace made by the
1142 examples (such as binding new variables) will be reflected in :attr:`globs`
1143 after the test is run.
Georg Brandl116aa622007-08-15 14:28:22 +00001144
1145
Benjamin Petersone41251e2008-04-25 01:59:09 +00001146 .. attribute:: name
Georg Brandl116aa622007-08-15 14:28:22 +00001147
Benjamin Petersone41251e2008-04-25 01:59:09 +00001148 A string name identifying the :class:`DocTest`. Typically, this is the name
1149 of the object or file that the test was extracted from.
Georg Brandl116aa622007-08-15 14:28:22 +00001150
1151
Benjamin Petersone41251e2008-04-25 01:59:09 +00001152 .. attribute:: filename
Georg Brandl116aa622007-08-15 14:28:22 +00001153
Benjamin Petersone41251e2008-04-25 01:59:09 +00001154 The name of the file that this :class:`DocTest` was extracted from; or
1155 ``None`` if the filename is unknown, or if the :class:`DocTest` was not
1156 extracted from a file.
Georg Brandl116aa622007-08-15 14:28:22 +00001157
1158
Benjamin Petersone41251e2008-04-25 01:59:09 +00001159 .. attribute:: lineno
Georg Brandl116aa622007-08-15 14:28:22 +00001160
Benjamin Petersone41251e2008-04-25 01:59:09 +00001161 The line number within :attr:`filename` where this :class:`DocTest` begins, or
1162 ``None`` if the line number is unavailable. This line number is zero-based
1163 with respect to the beginning of the file.
Georg Brandl116aa622007-08-15 14:28:22 +00001164
1165
Benjamin Petersone41251e2008-04-25 01:59:09 +00001166 .. attribute:: docstring
Georg Brandl116aa622007-08-15 14:28:22 +00001167
Benjamin Petersone41251e2008-04-25 01:59:09 +00001168 The string that the test was extracted from, or 'None' if the string is
1169 unavailable, or if the test was not extracted from a string.
Georg Brandl116aa622007-08-15 14:28:22 +00001170
1171
1172.. _doctest-example:
1173
1174Example Objects
1175^^^^^^^^^^^^^^^
1176
1177
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001178.. class:: Example(source, want, exc_msg=None, lineno=0, indent=0, options=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001179
1180 A single interactive example, consisting of a Python statement and its expected
1181 output. The constructor arguments are used to initialize the member variables
1182 of the same names.
1183
Georg Brandl116aa622007-08-15 14:28:22 +00001184
Benjamin Petersone41251e2008-04-25 01:59:09 +00001185 :class:`Example` defines the following member variables. They are initialized by
1186 the constructor, and should not be modified directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001187
1188
Benjamin Petersone41251e2008-04-25 01:59:09 +00001189 .. attribute:: source
Georg Brandl116aa622007-08-15 14:28:22 +00001190
Benjamin Petersone41251e2008-04-25 01:59:09 +00001191 A string containing the example's source code. This source code consists of a
1192 single Python statement, and always ends with a newline; the constructor adds
1193 a newline when necessary.
Georg Brandl116aa622007-08-15 14:28:22 +00001194
1195
Benjamin Petersone41251e2008-04-25 01:59:09 +00001196 .. attribute:: want
Georg Brandl116aa622007-08-15 14:28:22 +00001197
Benjamin Petersone41251e2008-04-25 01:59:09 +00001198 The expected output from running the example's source code (either from
1199 stdout, or a traceback in case of exception). :attr:`want` ends with a
1200 newline unless no output is expected, in which case it's an empty string. The
1201 constructor adds a newline when necessary.
Georg Brandl116aa622007-08-15 14:28:22 +00001202
1203
Benjamin Petersone41251e2008-04-25 01:59:09 +00001204 .. attribute:: exc_msg
Georg Brandl116aa622007-08-15 14:28:22 +00001205
Benjamin Petersone41251e2008-04-25 01:59:09 +00001206 The exception message generated by the example, if the example is expected to
1207 generate an exception; or ``None`` if it is not expected to generate an
1208 exception. This exception message is compared against the return value of
1209 :func:`traceback.format_exception_only`. :attr:`exc_msg` ends with a newline
1210 unless it's ``None``. The constructor adds a newline if needed.
Georg Brandl116aa622007-08-15 14:28:22 +00001211
1212
Benjamin Petersone41251e2008-04-25 01:59:09 +00001213 .. attribute:: lineno
Georg Brandl116aa622007-08-15 14:28:22 +00001214
Benjamin Petersone41251e2008-04-25 01:59:09 +00001215 The line number within the string containing this example where the example
1216 begins. This line number is zero-based with respect to the beginning of the
1217 containing string.
Georg Brandl116aa622007-08-15 14:28:22 +00001218
1219
Benjamin Petersone41251e2008-04-25 01:59:09 +00001220 .. attribute:: indent
Georg Brandl116aa622007-08-15 14:28:22 +00001221
Benjamin Petersone41251e2008-04-25 01:59:09 +00001222 The example's indentation in the containing string, i.e., the number of space
1223 characters that precede the example's first prompt.
Georg Brandl116aa622007-08-15 14:28:22 +00001224
1225
Benjamin Petersone41251e2008-04-25 01:59:09 +00001226 .. attribute:: options
Georg Brandl116aa622007-08-15 14:28:22 +00001227
Benjamin Petersone41251e2008-04-25 01:59:09 +00001228 A dictionary mapping from option flags to ``True`` or ``False``, which is used
1229 to override default options for this example. Any option flags not contained
1230 in this dictionary are left at their default value (as specified by the
1231 :class:`DocTestRunner`'s :attr:`optionflags`). By default, no options are set.
Georg Brandl116aa622007-08-15 14:28:22 +00001232
1233
1234.. _doctest-doctestfinder:
1235
1236DocTestFinder objects
1237^^^^^^^^^^^^^^^^^^^^^
1238
1239
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001240.. class:: DocTestFinder(verbose=False, parser=DocTestParser(), recurse=True, exclude_empty=True)
Georg Brandl116aa622007-08-15 14:28:22 +00001241
1242 A processing class used to extract the :class:`DocTest`\ s that are relevant to
1243 a given object, from its docstring and the docstrings of its contained objects.
1244 :class:`DocTest`\ s can currently be extracted from the following object types:
1245 modules, functions, classes, methods, staticmethods, classmethods, and
1246 properties.
1247
1248 The optional argument *verbose* can be used to display the objects searched by
1249 the finder. It defaults to ``False`` (no output).
1250
1251 The optional argument *parser* specifies the :class:`DocTestParser` object (or a
1252 drop-in replacement) that is used to extract doctests from docstrings.
1253
1254 If the optional argument *recurse* is false, then :meth:`DocTestFinder.find`
1255 will only examine the given object, and not any contained objects.
1256
1257 If the optional argument *exclude_empty* is false, then
1258 :meth:`DocTestFinder.find` will include tests for objects with empty docstrings.
1259
Georg Brandl116aa622007-08-15 14:28:22 +00001260
Benjamin Petersone41251e2008-04-25 01:59:09 +00001261 :class:`DocTestFinder` defines the following method:
Georg Brandl116aa622007-08-15 14:28:22 +00001262
1263
Benjamin Petersone41251e2008-04-25 01:59:09 +00001264 .. method:: find(obj[, name][, module][, globs][, extraglobs])
Georg Brandl116aa622007-08-15 14:28:22 +00001265
Benjamin Petersone41251e2008-04-25 01:59:09 +00001266 Return a list of the :class:`DocTest`\ s that are defined by *obj*'s
1267 docstring, or by any of its contained objects' docstrings.
Georg Brandl116aa622007-08-15 14:28:22 +00001268
Benjamin Petersone41251e2008-04-25 01:59:09 +00001269 The optional argument *name* specifies the object's name; this name will be
1270 used to construct names for the returned :class:`DocTest`\ s. If *name* is
1271 not specified, then ``obj.__name__`` is used.
Georg Brandl116aa622007-08-15 14:28:22 +00001272
Benjamin Petersone41251e2008-04-25 01:59:09 +00001273 The optional parameter *module* is the module that contains the given object.
1274 If the module is not specified or is None, then the test finder will attempt
1275 to automatically determine the correct module. The object's module is used:
Georg Brandl116aa622007-08-15 14:28:22 +00001276
Benjamin Petersone41251e2008-04-25 01:59:09 +00001277 * As a default namespace, if *globs* is not specified.
Georg Brandl116aa622007-08-15 14:28:22 +00001278
Benjamin Petersone41251e2008-04-25 01:59:09 +00001279 * To prevent the DocTestFinder from extracting DocTests from objects that are
1280 imported from other modules. (Contained objects with modules other than
1281 *module* are ignored.)
Georg Brandl116aa622007-08-15 14:28:22 +00001282
Benjamin Petersone41251e2008-04-25 01:59:09 +00001283 * To find the name of the file containing the object.
Georg Brandl116aa622007-08-15 14:28:22 +00001284
Benjamin Petersone41251e2008-04-25 01:59:09 +00001285 * To help find the line number of the object within its file.
Georg Brandl116aa622007-08-15 14:28:22 +00001286
Benjamin Petersone41251e2008-04-25 01:59:09 +00001287 If *module* is ``False``, no attempt to find the module will be made. This is
1288 obscure, of use mostly in testing doctest itself: if *module* is ``False``, or
1289 is ``None`` but cannot be found automatically, then all objects are considered
1290 to belong to the (non-existent) module, so all contained objects will
1291 (recursively) be searched for doctests.
Georg Brandl116aa622007-08-15 14:28:22 +00001292
Benjamin Petersone41251e2008-04-25 01:59:09 +00001293 The globals for each :class:`DocTest` is formed by combining *globs* and
1294 *extraglobs* (bindings in *extraglobs* override bindings in *globs*). A new
1295 shallow copy of the globals dictionary is created for each :class:`DocTest`.
1296 If *globs* is not specified, then it defaults to the module's *__dict__*, if
1297 specified, or ``{}`` otherwise. If *extraglobs* is not specified, then it
1298 defaults to ``{}``.
Georg Brandl116aa622007-08-15 14:28:22 +00001299
1300
1301.. _doctest-doctestparser:
1302
1303DocTestParser objects
1304^^^^^^^^^^^^^^^^^^^^^
1305
1306
1307.. class:: DocTestParser()
1308
1309 A processing class used to extract interactive examples from a string, and use
1310 them to create a :class:`DocTest` object.
1311
Georg Brandl116aa622007-08-15 14:28:22 +00001312
Benjamin Petersone41251e2008-04-25 01:59:09 +00001313 :class:`DocTestParser` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001314
1315
Benjamin Petersone41251e2008-04-25 01:59:09 +00001316 .. method:: get_doctest(string, globs, name, filename, lineno)
Georg Brandl116aa622007-08-15 14:28:22 +00001317
Benjamin Petersone41251e2008-04-25 01:59:09 +00001318 Extract all doctest examples from the given string, and collect them into a
1319 :class:`DocTest` object.
Georg Brandl116aa622007-08-15 14:28:22 +00001320
Benjamin Petersone41251e2008-04-25 01:59:09 +00001321 *globs*, *name*, *filename*, and *lineno* are attributes for the new
1322 :class:`DocTest` object. See the documentation for :class:`DocTest` for more
1323 information.
Georg Brandl116aa622007-08-15 14:28:22 +00001324
1325
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001326 .. method:: get_examples(string, name='<string>')
Georg Brandl116aa622007-08-15 14:28:22 +00001327
Benjamin Petersone41251e2008-04-25 01:59:09 +00001328 Extract all doctest examples from the given string, and return them as a list
1329 of :class:`Example` objects. Line numbers are 0-based. The optional argument
1330 *name* is a name identifying this string, and is only used for error messages.
Georg Brandl116aa622007-08-15 14:28:22 +00001331
1332
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001333 .. method:: parse(string, name='<string>')
Georg Brandl116aa622007-08-15 14:28:22 +00001334
Benjamin Petersone41251e2008-04-25 01:59:09 +00001335 Divide the given string into examples and intervening text, and return them as
1336 a list of alternating :class:`Example`\ s and strings. Line numbers for the
1337 :class:`Example`\ s are 0-based. The optional argument *name* is a name
1338 identifying this string, and is only used for error messages.
Georg Brandl116aa622007-08-15 14:28:22 +00001339
1340
1341.. _doctest-doctestrunner:
1342
1343DocTestRunner objects
1344^^^^^^^^^^^^^^^^^^^^^
1345
1346
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001347.. class:: DocTestRunner(checker=None, verbose=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +00001348
1349 A processing class used to execute and verify the interactive examples in a
1350 :class:`DocTest`.
1351
1352 The comparison between expected outputs and actual outputs is done by an
1353 :class:`OutputChecker`. This comparison may be customized with a number of
1354 option flags; see section :ref:`doctest-options` for more information. If the
1355 option flags are insufficient, then the comparison may also be customized by
1356 passing a subclass of :class:`OutputChecker` to the constructor.
1357
1358 The test runner's display output can be controlled in two ways. First, an output
1359 function can be passed to :meth:`TestRunner.run`; this function will be called
1360 with strings that should be displayed. It defaults to ``sys.stdout.write``. If
1361 capturing the output is not sufficient, then the display output can be also
1362 customized by subclassing DocTestRunner, and overriding the methods
1363 :meth:`report_start`, :meth:`report_success`,
1364 :meth:`report_unexpected_exception`, and :meth:`report_failure`.
1365
1366 The optional keyword argument *checker* specifies the :class:`OutputChecker`
1367 object (or drop-in replacement) that should be used to compare the expected
1368 outputs to the actual outputs of doctest examples.
1369
1370 The optional keyword argument *verbose* controls the :class:`DocTestRunner`'s
1371 verbosity. If *verbose* is ``True``, then information is printed about each
1372 example, as it is run. If *verbose* is ``False``, then only failures are
1373 printed. If *verbose* is unspecified, or ``None``, then verbose output is used
1374 iff the command-line switch :option:`-v` is used.
1375
1376 The optional keyword argument *optionflags* can be used to control how the test
1377 runner compares expected output to actual output, and how it displays failures.
1378 For more information, see section :ref:`doctest-options`.
1379
Georg Brandl116aa622007-08-15 14:28:22 +00001380
Benjamin Petersone41251e2008-04-25 01:59:09 +00001381 :class:`DocTestParser` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001382
1383
Benjamin Petersone41251e2008-04-25 01:59:09 +00001384 .. method:: report_start(out, test, example)
Georg Brandl116aa622007-08-15 14:28:22 +00001385
Benjamin Petersone41251e2008-04-25 01:59:09 +00001386 Report that the test runner is about to process the given example. This method
1387 is provided to allow subclasses of :class:`DocTestRunner` to customize their
1388 output; it should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001389
Benjamin Petersone41251e2008-04-25 01:59:09 +00001390 *example* is the example about to be processed. *test* is the test
1391 *containing example*. *out* is the output function that was passed to
1392 :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001393
1394
Benjamin Petersone41251e2008-04-25 01:59:09 +00001395 .. method:: report_success(out, test, example, got)
Georg Brandl116aa622007-08-15 14:28:22 +00001396
Benjamin Petersone41251e2008-04-25 01:59:09 +00001397 Report that the given example ran successfully. This method is provided to
1398 allow subclasses of :class:`DocTestRunner` to customize their output; it
1399 should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001400
Benjamin Petersone41251e2008-04-25 01:59:09 +00001401 *example* is the example about to be processed. *got* is the actual output
1402 from the example. *test* is the test containing *example*. *out* is the
1403 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001404
1405
Benjamin Petersone41251e2008-04-25 01:59:09 +00001406 .. method:: report_failure(out, test, example, got)
Georg Brandl116aa622007-08-15 14:28:22 +00001407
Benjamin Petersone41251e2008-04-25 01:59:09 +00001408 Report that the given example failed. This method is provided to allow
1409 subclasses of :class:`DocTestRunner` to customize their output; it should not
1410 be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001411
Benjamin Petersone41251e2008-04-25 01:59:09 +00001412 *example* is the example about to be processed. *got* is the actual output
1413 from the example. *test* is the test containing *example*. *out* is the
1414 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001415
1416
Benjamin Petersone41251e2008-04-25 01:59:09 +00001417 .. method:: report_unexpected_exception(out, test, example, exc_info)
Georg Brandl116aa622007-08-15 14:28:22 +00001418
Benjamin Petersone41251e2008-04-25 01:59:09 +00001419 Report that the given example raised an unexpected exception. This method is
1420 provided to allow subclasses of :class:`DocTestRunner` to customize their
1421 output; it should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001422
Benjamin Petersone41251e2008-04-25 01:59:09 +00001423 *example* is the example about to be processed. *exc_info* is a tuple
1424 containing information about the unexpected exception (as returned by
1425 :func:`sys.exc_info`). *test* is the test containing *example*. *out* is the
1426 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001427
1428
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001429 .. method:: run(test, compileflags=None, out=None, clear_globs=True)
Georg Brandl116aa622007-08-15 14:28:22 +00001430
Benjamin Petersone41251e2008-04-25 01:59:09 +00001431 Run the examples in *test* (a :class:`DocTest` object), and display the
1432 results using the writer function *out*.
Georg Brandl116aa622007-08-15 14:28:22 +00001433
Benjamin Petersone41251e2008-04-25 01:59:09 +00001434 The examples are run in the namespace ``test.globs``. If *clear_globs* is
1435 true (the default), then this namespace will be cleared after the test runs,
1436 to help with garbage collection. If you would like to examine the namespace
1437 after the test completes, then use *clear_globs=False*.
Georg Brandl116aa622007-08-15 14:28:22 +00001438
Benjamin Petersone41251e2008-04-25 01:59:09 +00001439 *compileflags* gives the set of flags that should be used by the Python
1440 compiler when running the examples. If not specified, then it will default to
1441 the set of future-import flags that apply to *globs*.
Georg Brandl116aa622007-08-15 14:28:22 +00001442
Benjamin Petersone41251e2008-04-25 01:59:09 +00001443 The output of each example is checked using the :class:`DocTestRunner`'s
1444 output checker, and the results are formatted by the
1445 :meth:`DocTestRunner.report_\*` methods.
Georg Brandl116aa622007-08-15 14:28:22 +00001446
1447
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001448 .. method:: summarize(verbose=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001449
Benjamin Petersone41251e2008-04-25 01:59:09 +00001450 Print a summary of all the test cases that have been run by this DocTestRunner,
1451 and return a :term:`named tuple` ``TestResults(failed, attempted)``.
Georg Brandl116aa622007-08-15 14:28:22 +00001452
Benjamin Petersone41251e2008-04-25 01:59:09 +00001453 The optional *verbose* argument controls how detailed the summary is. If the
1454 verbosity is not specified, then the :class:`DocTestRunner`'s verbosity is
1455 used.
Georg Brandl116aa622007-08-15 14:28:22 +00001456
1457.. _doctest-outputchecker:
1458
1459OutputChecker objects
1460^^^^^^^^^^^^^^^^^^^^^
1461
1462
1463.. class:: OutputChecker()
1464
1465 A class used to check the whether the actual output from a doctest example
1466 matches the expected output. :class:`OutputChecker` defines two methods:
1467 :meth:`check_output`, which compares a given pair of outputs, and returns true
1468 if they match; and :meth:`output_difference`, which returns a string describing
1469 the differences between two outputs.
1470
Georg Brandl116aa622007-08-15 14:28:22 +00001471
Benjamin Petersone41251e2008-04-25 01:59:09 +00001472 :class:`OutputChecker` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001473
Benjamin Petersone41251e2008-04-25 01:59:09 +00001474 .. method:: check_output(want, got, optionflags)
Georg Brandl116aa622007-08-15 14:28:22 +00001475
Benjamin Petersone41251e2008-04-25 01:59:09 +00001476 Return ``True`` iff the actual output from an example (*got*) matches the
1477 expected output (*want*). These strings are always considered to match if
1478 they are identical; but depending on what option flags the test runner is
1479 using, several non-exact match types are also possible. See section
1480 :ref:`doctest-options` for more information about option flags.
Georg Brandl116aa622007-08-15 14:28:22 +00001481
1482
Benjamin Petersone41251e2008-04-25 01:59:09 +00001483 .. method:: output_difference(example, got, optionflags)
Georg Brandl116aa622007-08-15 14:28:22 +00001484
Benjamin Petersone41251e2008-04-25 01:59:09 +00001485 Return a string describing the differences between the expected output for a
1486 given example (*example*) and the actual output (*got*). *optionflags* is the
1487 set of option flags used to compare *want* and *got*.
Georg Brandl116aa622007-08-15 14:28:22 +00001488
1489
1490.. _doctest-debugging:
1491
1492Debugging
1493---------
1494
1495Doctest provides several mechanisms for debugging doctest examples:
1496
1497* Several functions convert doctests to executable Python programs, which can be
1498 run under the Python debugger, :mod:`pdb`.
1499
1500* The :class:`DebugRunner` class is a subclass of :class:`DocTestRunner` that
1501 raises an exception for the first failing example, containing information about
1502 that example. This information can be used to perform post-mortem debugging on
1503 the example.
1504
1505* The :mod:`unittest` cases generated by :func:`DocTestSuite` support the
1506 :meth:`debug` method defined by :class:`unittest.TestCase`.
1507
1508* You can add a call to :func:`pdb.set_trace` in a doctest example, and you'll
1509 drop into the Python debugger when that line is executed. Then you can inspect
1510 current values of variables, and so on. For example, suppose :file:`a.py`
1511 contains just this module docstring::
1512
1513 """
1514 >>> def f(x):
1515 ... g(x*2)
1516 >>> def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001517 ... print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001518 ... import pdb; pdb.set_trace()
1519 >>> f(3)
1520 9
1521 """
1522
1523 Then an interactive Python session may look like this::
1524
1525 >>> import a, doctest
1526 >>> doctest.testmod(a)
1527 --Return--
1528 > <doctest a[1]>(3)g()->None
1529 -> import pdb; pdb.set_trace()
1530 (Pdb) list
1531 1 def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001532 2 print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001533 3 -> import pdb; pdb.set_trace()
1534 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001535 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001536 6
1537 (Pdb) step
1538 --Return--
1539 > <doctest a[0]>(2)f()->None
1540 -> g(x*2)
1541 (Pdb) list
1542 1 def f(x):
1543 2 -> g(x*2)
1544 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001545 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001546 3
1547 (Pdb) step
1548 --Return--
1549 > <doctest a[2]>(1)?()->None
1550 -> f(3)
1551 (Pdb) cont
1552 (0, 3)
1553 >>>
1554
Georg Brandl116aa622007-08-15 14:28:22 +00001555
1556Functions that convert doctests to Python code, and possibly run the synthesized
1557code under the debugger:
1558
1559
1560.. function:: script_from_examples(s)
1561
1562 Convert text with examples to a script.
1563
1564 Argument *s* is a string containing doctest examples. The string is converted
1565 to a Python script, where doctest examples in *s* are converted to regular code,
1566 and everything else is converted to Python comments. The generated script is
1567 returned as a string. For example, ::
1568
1569 import doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001570 print(doctest.script_from_examples(r"""
Georg Brandl116aa622007-08-15 14:28:22 +00001571 Set x and y to 1 and 2.
1572 >>> x, y = 1, 2
1573
1574 Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001575 >>> print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001576 3
Georg Brandl6911e3c2007-09-04 07:15:32 +00001577 """))
Georg Brandl116aa622007-08-15 14:28:22 +00001578
1579 displays::
1580
1581 # Set x and y to 1 and 2.
1582 x, y = 1, 2
1583 #
1584 # Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001585 print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001586 # Expected:
1587 ## 3
1588
1589 This function is used internally by other functions (see below), but can also be
1590 useful when you want to transform an interactive Python session into a Python
1591 script.
1592
Georg Brandl116aa622007-08-15 14:28:22 +00001593
1594.. function:: testsource(module, name)
1595
1596 Convert the doctest for an object to a script.
1597
1598 Argument *module* is a module object, or dotted name of a module, containing the
1599 object whose doctests are of interest. Argument *name* is the name (within the
1600 module) of the object with the doctests of interest. The result is a string,
1601 containing the object's docstring converted to a Python script, as described for
1602 :func:`script_from_examples` above. For example, if module :file:`a.py`
1603 contains a top-level function :func:`f`, then ::
1604
1605 import a, doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001606 print(doctest.testsource(a, "a.f"))
Georg Brandl116aa622007-08-15 14:28:22 +00001607
1608 prints a script version of function :func:`f`'s docstring, with doctests
1609 converted to code, and the rest placed in comments.
1610
Georg Brandl116aa622007-08-15 14:28:22 +00001611
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001612.. function:: debug(module, name, pm=False)
Georg Brandl116aa622007-08-15 14:28:22 +00001613
1614 Debug the doctests for an object.
1615
1616 The *module* and *name* arguments are the same as for function
1617 :func:`testsource` above. The synthesized Python script for the named object's
1618 docstring is written to a temporary file, and then that file is run under the
1619 control of the Python debugger, :mod:`pdb`.
1620
1621 A shallow copy of ``module.__dict__`` is used for both local and global
1622 execution context.
1623
1624 Optional argument *pm* controls whether post-mortem debugging is used. If *pm*
1625 has a true value, the script file is run directly, and the debugger gets
1626 involved only if the script terminates via raising an unhandled exception. If
1627 it does, then post-mortem debugging is invoked, via :func:`pdb.post_mortem`,
1628 passing the traceback object from the unhandled exception. If *pm* is not
1629 specified, or is false, the script is run under the debugger from the start, via
1630 passing an appropriate :func:`exec` call to :func:`pdb.run`.
1631
Georg Brandl116aa622007-08-15 14:28:22 +00001632
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001633.. function:: debug_src(src, pm=False, globs=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001634
1635 Debug the doctests in a string.
1636
1637 This is like function :func:`debug` above, except that a string containing
1638 doctest examples is specified directly, via the *src* argument.
1639
1640 Optional argument *pm* has the same meaning as in function :func:`debug` above.
1641
1642 Optional argument *globs* gives a dictionary to use as both local and global
1643 execution context. If not specified, or ``None``, an empty dictionary is used.
1644 If specified, a shallow copy of the dictionary is used.
1645
Georg Brandl116aa622007-08-15 14:28:22 +00001646
1647The :class:`DebugRunner` class, and the special exceptions it may raise, are of
1648most interest to testing framework authors, and will only be sketched here. See
1649the source code, and especially :class:`DebugRunner`'s docstring (which is a
1650doctest!) for more details:
1651
1652
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001653.. class:: DebugRunner(checker=None, verbose=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +00001654
1655 A subclass of :class:`DocTestRunner` that raises an exception as soon as a
1656 failure is encountered. If an unexpected exception occurs, an
1657 :exc:`UnexpectedException` exception is raised, containing the test, the
1658 example, and the original exception. If the output doesn't match, then a
1659 :exc:`DocTestFailure` exception is raised, containing the test, the example, and
1660 the actual output.
1661
1662 For information about the constructor parameters and methods, see the
1663 documentation for :class:`DocTestRunner` in section :ref:`doctest-advanced-api`.
1664
1665There are two exceptions that may be raised by :class:`DebugRunner` instances:
1666
1667
1668.. exception:: DocTestFailure(test, example, got)
1669
1670 An exception thrown by :class:`DocTestRunner` to signal that a doctest example's
1671 actual output did not match its expected output. The constructor arguments are
1672 used to initialize the member variables of the same names.
1673
1674:exc:`DocTestFailure` defines the following member variables:
1675
1676
1677.. attribute:: DocTestFailure.test
1678
1679 The :class:`DocTest` object that was being run when the example failed.
1680
1681
1682.. attribute:: DocTestFailure.example
1683
1684 The :class:`Example` that failed.
1685
1686
1687.. attribute:: DocTestFailure.got
1688
1689 The example's actual output.
1690
1691
1692.. exception:: UnexpectedException(test, example, exc_info)
1693
1694 An exception thrown by :class:`DocTestRunner` to signal that a doctest example
1695 raised an unexpected exception. The constructor arguments are used to
1696 initialize the member variables of the same names.
1697
1698:exc:`UnexpectedException` defines the following member variables:
1699
1700
1701.. attribute:: UnexpectedException.test
1702
1703 The :class:`DocTest` object that was being run when the example failed.
1704
1705
1706.. attribute:: UnexpectedException.example
1707
1708 The :class:`Example` that failed.
1709
1710
1711.. attribute:: UnexpectedException.exc_info
1712
1713 A tuple containing information about the unexpected exception, as returned by
1714 :func:`sys.exc_info`.
1715
1716
1717.. _doctest-soapbox:
1718
1719Soapbox
1720-------
1721
1722As mentioned in the introduction, :mod:`doctest` has grown to have three primary
1723uses:
1724
1725#. Checking examples in docstrings.
1726
1727#. Regression testing.
1728
1729#. Executable documentation / literate testing.
1730
1731These uses have different requirements, and it is important to distinguish them.
1732In particular, filling your docstrings with obscure test cases makes for bad
1733documentation.
1734
1735When writing a docstring, choose docstring examples with care. There's an art to
1736this that needs to be learned---it may not be natural at first. Examples should
1737add genuine value to the documentation. A good example can often be worth many
1738words. If done with care, the examples will be invaluable for your users, and
1739will pay back the time it takes to collect them many times over as the years go
1740by and things change. I'm still amazed at how often one of my :mod:`doctest`
1741examples stops working after a "harmless" change.
1742
1743Doctest also makes an excellent tool for regression testing, especially if you
1744don't skimp on explanatory text. By interleaving prose and examples, it becomes
1745much easier to keep track of what's actually being tested, and why. When a test
1746fails, good prose can make it much easier to figure out what the problem is, and
1747how it should be fixed. It's true that you could write extensive comments in
1748code-based testing, but few programmers do. Many have found that using doctest
1749approaches instead leads to much clearer tests. Perhaps this is simply because
1750doctest makes writing prose a little easier than writing code, while writing
1751comments in code is a little harder. I think it goes deeper than just that:
1752the natural attitude when writing a doctest-based test is that you want to
1753explain the fine points of your software, and illustrate them with examples.
1754This in turn naturally leads to test files that start with the simplest
1755features, and logically progress to complications and edge cases. A coherent
1756narrative is the result, instead of a collection of isolated functions that test
1757isolated bits of functionality seemingly at random. It's a different attitude,
1758and produces different results, blurring the distinction between testing and
1759explaining.
1760
1761Regression testing is best confined to dedicated objects or files. There are
1762several options for organizing tests:
1763
1764* Write text files containing test cases as interactive examples, and test the
1765 files using :func:`testfile` or :func:`DocFileSuite`. This is recommended,
1766 although is easiest to do for new projects, designed from the start to use
1767 doctest.
1768
1769* Define functions named ``_regrtest_topic`` that consist of single docstrings,
1770 containing test cases for the named topics. These functions can be included in
1771 the same file as the module, or separated out into a separate test file.
1772
1773* Define a ``__test__`` dictionary mapping from regression test topics to
1774 docstrings containing test cases.
1775
1776.. rubric:: Footnotes
1777
1778.. [#] Examples containing both expected output and an exception are not supported.
1779 Trying to guess where one ends and the other begins is too error-prone, and that
1780 also makes for a confusing test.
1781