blob: 5d9759a64dd7b5898e012ba3e6ad4a17ceae4758 [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
R. David Murray24d83872010-06-16 00:37:16 +0000285In most cases a copy-and-paste of an interactive console session works fine,
286but doctest isn't trying to do an exact emulation of any specific Python shell.
Georg Brandl116aa622007-08-15 14:28:22 +0000287
Georg Brandl116aa622007-08-15 14:28:22 +0000288::
289
290 >>> # comments are ignored
291 >>> x = 12
292 >>> x
293 12
294 >>> if x == 13:
Georg Brandl6911e3c2007-09-04 07:15:32 +0000295 ... print("yes")
Georg Brandl116aa622007-08-15 14:28:22 +0000296 ... else:
Georg Brandl6911e3c2007-09-04 07:15:32 +0000297 ... print("no")
298 ... print("NO")
299 ... print("NO!!!")
Georg Brandl116aa622007-08-15 14:28:22 +0000300 ...
301 no
302 NO
303 NO!!!
304 >>>
305
306Any expected output must immediately follow the final ``'>>> '`` or ``'... '``
307line containing the code, and the expected output (if any) extends to the next
308``'>>> '`` or all-whitespace line.
309
310The fine print:
311
312* Expected output cannot contain an all-whitespace line, since such a line is
313 taken to signal the end of expected output. If expected output does contain a
314 blank line, put ``<BLANKLINE>`` in your doctest example each place a blank line
315 is expected.
316
R. David Murray24d83872010-06-16 00:37:16 +0000317* All hard tab characters are expanded to spaces, using 8-column tab stops.
318 Tabs in output generated by the tested code are not modified. Because any
319 hard tabs in the sample output *are* expanded, this means that if the code
320 output includes hard tabs, the only way the doctest can pass is if the
321 :const:`NORMALIZE_WHITESPACE` option or directive is in effect.
322 Alternatively, the test can be rewritten to capture the output and compare it
323 to an expected value as part of the test. This handling of tabs in the
324 source was arrived at through trial and error, and has proven to be the least
325 error prone way of handling them. It is possible to use a different
326 algorithm for handling tabs by writing a custom :class:`DocTestParser` class.
327
328 .. versionchanged:: 2.4
329 Expanding tabs to spaces is new; previous versions tried to preserve hard tabs,
330 with confusing results.
331
Georg Brandl116aa622007-08-15 14:28:22 +0000332* Output to stdout is captured, but not output to stderr (exception tracebacks
333 are captured via a different means).
334
335* If you continue a line via backslashing in an interactive session, or for any
336 other reason use a backslash, you should use a raw docstring, which will
337 preserve your backslashes exactly as you type them::
338
339 >>> def f(x):
340 ... r'''Backslashes in a raw docstring: m\n'''
Georg Brandl6911e3c2007-09-04 07:15:32 +0000341 >>> print(f.__doc__)
Georg Brandl116aa622007-08-15 14:28:22 +0000342 Backslashes in a raw docstring: m\n
343
344 Otherwise, the backslash will be interpreted as part of the string. For example,
345 the "\\" above would be interpreted as a newline character. Alternatively, you
346 can double each backslash in the doctest version (and not use a raw string)::
347
348 >>> def f(x):
349 ... '''Backslashes in a raw docstring: m\\n'''
Georg Brandl6911e3c2007-09-04 07:15:32 +0000350 >>> print(f.__doc__)
Georg Brandl116aa622007-08-15 14:28:22 +0000351 Backslashes in a raw docstring: m\n
352
353* The starting column doesn't matter::
354
355 >>> assert "Easy!"
356 >>> import math
357 >>> math.floor(1.9)
R. David Murray4c380592009-11-23 03:14:51 +0000358 1
Georg Brandl116aa622007-08-15 14:28:22 +0000359
360 and as many leading whitespace characters are stripped from the expected output
361 as appeared in the initial ``'>>> '`` line that started the example.
362
363
364.. _doctest-execution-context:
365
366What's the Execution Context?
367^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
368
369By default, each time :mod:`doctest` finds a docstring to test, it uses a
370*shallow copy* of :mod:`M`'s globals, so that running tests doesn't change the
371module's real globals, and so that one test in :mod:`M` can't leave behind
372crumbs that accidentally allow another test to work. This means examples can
373freely use any names defined at top-level in :mod:`M`, and names defined earlier
374in the docstring being run. Examples cannot see names defined in other
375docstrings.
376
377You can force use of your own dict as the execution context by passing
378``globs=your_dict`` to :func:`testmod` or :func:`testfile` instead.
379
380
381.. _doctest-exceptions:
382
383What About Exceptions?
384^^^^^^^^^^^^^^^^^^^^^^
385
386No problem, provided that the traceback is the only output produced by the
387example: just paste in the traceback. [#]_ Since tracebacks contain details
388that are likely to change rapidly (for example, exact file paths and line
389numbers), this is one case where doctest works hard to be flexible in what it
390accepts.
391
392Simple example::
393
394 >>> [1, 2, 3].remove(42)
395 Traceback (most recent call last):
396 File "<stdin>", line 1, in ?
397 ValueError: list.remove(x): x not in list
398
399That doctest succeeds if :exc:`ValueError` is raised, with the ``list.remove(x):
400x not in list`` detail as shown.
401
402The expected output for an exception must start with a traceback header, which
403may be either of the following two lines, indented the same as the first line of
404the example::
405
406 Traceback (most recent call last):
407 Traceback (innermost last):
408
409The traceback header is followed by an optional traceback stack, whose contents
410are ignored by doctest. The traceback stack is typically omitted, or copied
411verbatim from an interactive session.
412
413The traceback stack is followed by the most interesting part: the line(s)
414containing the exception type and detail. This is usually the last line of a
415traceback, but can extend across multiple lines if the exception has a
416multi-line detail::
417
418 >>> raise ValueError('multi\n line\ndetail')
419 Traceback (most recent call last):
420 File "<stdin>", line 1, in ?
421 ValueError: multi
422 line
423 detail
424
425The last three lines (starting with :exc:`ValueError`) are compared against the
426exception's type and detail, and the rest are ignored.
427
428Best practice is to omit the traceback stack, unless it adds significant
429documentation value to the example. So the last example is probably better as::
430
431 >>> raise ValueError('multi\n line\ndetail')
432 Traceback (most recent call last):
433 ...
434 ValueError: multi
435 line
436 detail
437
438Note that tracebacks are treated very specially. In particular, in the
439rewritten example, the use of ``...`` is independent of doctest's
440:const:`ELLIPSIS` option. The ellipsis in that example could be left out, or
441could just as well be three (or three hundred) commas or digits, or an indented
442transcript of a Monty Python skit.
443
444Some details you should read once, but won't need to remember:
445
446* Doctest can't guess whether your expected output came from an exception
447 traceback or from ordinary printing. So, e.g., an example that expects
448 ``ValueError: 42 is prime`` will pass whether :exc:`ValueError` is actually
449 raised or if the example merely prints that traceback text. In practice,
450 ordinary output rarely begins with a traceback header line, so this doesn't
451 create real problems.
452
453* Each line of the traceback stack (if present) must be indented further than
454 the first line of the example, *or* start with a non-alphanumeric character.
455 The first line following the traceback header indented the same and starting
456 with an alphanumeric is taken to be the start of the exception detail. Of
457 course this does the right thing for genuine tracebacks.
458
459* When the :const:`IGNORE_EXCEPTION_DETAIL` doctest option is is specified,
460 everything following the leftmost colon is ignored.
461
462* The interactive shell omits the traceback header line for some
463 :exc:`SyntaxError`\ s. But doctest uses the traceback header line to
464 distinguish exceptions from non-exceptions. So in the rare case where you need
465 to test a :exc:`SyntaxError` that omits the traceback header, you will need to
466 manually add the traceback header line to your test example.
467
468* For some :exc:`SyntaxError`\ s, Python displays the character position of the
469 syntax error, using a ``^`` marker::
470
471 >>> 1 1
472 File "<stdin>", line 1
473 1 1
474 ^
475 SyntaxError: invalid syntax
476
477 Since the lines showing the position of the error come before the exception type
478 and detail, they are not checked by doctest. For example, the following test
479 would pass, even though it puts the ``^`` marker in the wrong location::
480
481 >>> 1 1
482 Traceback (most recent call last):
483 File "<stdin>", line 1
484 1 1
485 ^
486 SyntaxError: invalid syntax
487
Georg Brandl116aa622007-08-15 14:28:22 +0000488
489.. _doctest-options:
490
491Option Flags and Directives
492^^^^^^^^^^^^^^^^^^^^^^^^^^^
493
494A number of option flags control various aspects of doctest's behavior.
495Symbolic names for the flags are supplied as module constants, which can be
496or'ed together and passed to various functions. The names can also be used in
497doctest directives (see below).
498
499The first group of options define test semantics, controlling aspects of how
500doctest decides whether actual output matches an example's expected output:
501
502
503.. data:: DONT_ACCEPT_TRUE_FOR_1
504
505 By default, if an expected output block contains just ``1``, an actual output
506 block containing just ``1`` or just ``True`` is considered to be a match, and
507 similarly for ``0`` versus ``False``. When :const:`DONT_ACCEPT_TRUE_FOR_1` is
508 specified, neither substitution is allowed. The default behavior caters to that
509 Python changed the return type of many functions from integer to boolean;
510 doctests expecting "little integer" output still work in these cases. This
511 option will probably go away, but not for several years.
512
513
514.. data:: DONT_ACCEPT_BLANKLINE
515
516 By default, if an expected output block contains a line containing only the
517 string ``<BLANKLINE>``, then that line will match a blank line in the actual
518 output. Because a genuinely blank line delimits the expected output, this is
519 the only way to communicate that a blank line is expected. When
520 :const:`DONT_ACCEPT_BLANKLINE` is specified, this substitution is not allowed.
521
522
523.. data:: NORMALIZE_WHITESPACE
524
525 When specified, all sequences of whitespace (blanks and newlines) are treated as
526 equal. Any sequence of whitespace within the expected output will match any
527 sequence of whitespace within the actual output. By default, whitespace must
528 match exactly. :const:`NORMALIZE_WHITESPACE` is especially useful when a line of
529 expected output is very long, and you want to wrap it across multiple lines in
530 your source.
531
532
533.. data:: ELLIPSIS
534
535 When specified, an ellipsis marker (``...``) in the expected output can match
536 any substring in the actual output. This includes substrings that span line
537 boundaries, and empty substrings, so it's best to keep usage of this simple.
538 Complicated uses can lead to the same kinds of "oops, it matched too much!"
539 surprises that ``.*`` is prone to in regular expressions.
540
541
542.. data:: IGNORE_EXCEPTION_DETAIL
543
544 When specified, an example that expects an exception passes if an exception of
545 the expected type is raised, even if the exception detail does not match. For
546 example, an example expecting ``ValueError: 42`` will pass if the actual
547 exception raised is ``ValueError: 3*14``, but will fail, e.g., if
548 :exc:`TypeError` is raised.
549
550 Note that a similar effect can be obtained using :const:`ELLIPSIS`, and
551 :const:`IGNORE_EXCEPTION_DETAIL` may go away when Python releases prior to 2.4
552 become uninteresting. Until then, :const:`IGNORE_EXCEPTION_DETAIL` is the only
553 clear way to write a doctest that doesn't care about the exception detail yet
554 continues to pass under Python releases prior to 2.4 (doctest directives appear
555 to be comments to them). For example, ::
556
557 >>> (1, 2)[3] = 'moo' #doctest: +IGNORE_EXCEPTION_DETAIL
558 Traceback (most recent call last):
559 File "<stdin>", line 1, in ?
560 TypeError: object doesn't support item assignment
561
562 passes under Python 2.4 and Python 2.3. The detail changed in 2.4, to say "does
563 not" instead of "doesn't".
564
565
566.. data:: SKIP
567
568 When specified, do not run the example at all. This can be useful in contexts
569 where doctest examples serve as both documentation and test cases, and an
570 example should be included for documentation purposes, but should not be
571 checked. E.g., the example's output might be random; or the example might
572 depend on resources which would be unavailable to the test driver.
573
574 The SKIP flag can also be used for temporarily "commenting out" examples.
575
576
577.. data:: COMPARISON_FLAGS
578
579 A bitmask or'ing together all the comparison flags above.
580
581The second group of options controls how test failures are reported:
582
583
584.. data:: REPORT_UDIFF
585
586 When specified, failures that involve multi-line expected and actual outputs are
587 displayed using a unified diff.
588
589
590.. data:: REPORT_CDIFF
591
592 When specified, failures that involve multi-line expected and actual outputs
593 will be displayed using a context diff.
594
595
596.. data:: REPORT_NDIFF
597
598 When specified, differences are computed by ``difflib.Differ``, using the same
599 algorithm as the popular :file:`ndiff.py` utility. This is the only method that
600 marks differences within lines as well as across lines. For example, if a line
601 of expected output contains digit ``1`` where actual output contains letter
602 ``l``, a line is inserted with a caret marking the mismatching column positions.
603
604
605.. data:: REPORT_ONLY_FIRST_FAILURE
606
607 When specified, display the first failing example in each doctest, but suppress
608 output for all remaining examples. This will prevent doctest from reporting
609 correct examples that break because of earlier failures; but it might also hide
610 incorrect examples that fail independently of the first failure. When
611 :const:`REPORT_ONLY_FIRST_FAILURE` is specified, the remaining examples are
612 still run, and still count towards the total number of failures reported; only
613 the output is suppressed.
614
615
616.. data:: REPORTING_FLAGS
617
618 A bitmask or'ing together all the reporting flags above.
619
620"Doctest directives" may be used to modify the option flags for individual
621examples. Doctest directives are expressed as a special Python comment
622following an example's source code:
623
624.. productionlist:: doctest
625 directive: "#" "doctest:" `directive_options`
626 directive_options: `directive_option` ("," `directive_option`)\*
627 directive_option: `on_or_off` `directive_option_name`
628 on_or_off: "+" \| "-"
629 directive_option_name: "DONT_ACCEPT_BLANKLINE" \| "NORMALIZE_WHITESPACE" \| ...
630
631Whitespace is not allowed between the ``+`` or ``-`` and the directive option
632name. The directive option name can be any of the option flag names explained
633above.
634
635An example's doctest directives modify doctest's behavior for that single
636example. Use ``+`` to enable the named behavior, or ``-`` to disable it.
637
638For example, this test passes::
639
Georg Brandl6911e3c2007-09-04 07:15:32 +0000640 >>> print(range(20)) #doctest: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000641 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9,
642 10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
643
644Without the directive it would fail, both because the actual output doesn't have
645two blanks before the single-digit list elements, and because the actual output
646is on a single line. This test also passes, and also requires a directive to do
647so::
648
Georg Brandl6911e3c2007-09-04 07:15:32 +0000649 >>> print(range(20)) # doctest: +ELLIPSIS
Georg Brandl116aa622007-08-15 14:28:22 +0000650 [0, 1, ..., 18, 19]
651
652Multiple directives can be used on a single physical line, separated by commas::
653
Georg Brandl6911e3c2007-09-04 07:15:32 +0000654 >>> print(range(20)) # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000655 [0, 1, ..., 18, 19]
656
657If multiple directive comments are used for a single example, then they are
658combined::
659
Georg Brandl6911e3c2007-09-04 07:15:32 +0000660 >>> print(range(20)) # doctest: +ELLIPSIS
Georg Brandl116aa622007-08-15 14:28:22 +0000661 ... # doctest: +NORMALIZE_WHITESPACE
662 [0, 1, ..., 18, 19]
663
664As the previous example shows, you can add ``...`` lines to your example
665containing only directives. This can be useful when an example is too long for
666a directive to comfortably fit on the same line::
667
Georg Brandl6911e3c2007-09-04 07:15:32 +0000668 >>> print(range(5) + range(10,20) + range(30,40) + range(50,60))
Georg Brandl116aa622007-08-15 14:28:22 +0000669 ... # doctest: +ELLIPSIS
670 [0, ..., 4, 10, ..., 19, 30, ..., 39, 50, ..., 59]
671
672Note that since all options are disabled by default, and directives apply only
673to the example they appear in, enabling options (via ``+`` in a directive) is
674usually the only meaningful choice. However, option flags can also be passed to
675functions that run doctests, establishing different defaults. In such cases,
676disabling an option via ``-`` in a directive can be useful.
677
Georg Brandl116aa622007-08-15 14:28:22 +0000678
679There's also a way to register new option flag names, although this isn't useful
680unless you intend to extend :mod:`doctest` internals via subclassing:
681
682
683.. function:: register_optionflag(name)
684
685 Create a new option flag with a given name, and return the new flag's integer
686 value. :func:`register_optionflag` can be used when subclassing
687 :class:`OutputChecker` or :class:`DocTestRunner` to create new options that are
688 supported by your subclasses. :func:`register_optionflag` should always be
689 called using the following idiom::
690
691 MY_FLAG = register_optionflag('MY_FLAG')
692
Georg Brandl116aa622007-08-15 14:28:22 +0000693
694.. _doctest-warnings:
695
696Warnings
697^^^^^^^^
698
699:mod:`doctest` is serious about requiring exact matches in expected output. If
700even a single character doesn't match, the test fails. This will probably
701surprise you a few times, as you learn exactly what Python does and doesn't
702guarantee about output. For example, when printing a dict, Python doesn't
703guarantee that the key-value pairs will be printed in any particular order, so a
Christian Heimes5b5e81c2007-12-31 16:14:33 +0000704test like ::
Georg Brandl116aa622007-08-15 14:28:22 +0000705
706 >>> foo()
707 {"Hermione": "hippogryph", "Harry": "broomstick"}
708
709is vulnerable! One workaround is to do ::
710
711 >>> foo() == {"Hermione": "hippogryph", "Harry": "broomstick"}
712 True
713
714instead. Another is to do ::
715
Ezio Melotti2befd9a2009-09-13 08:08:32 +0000716 >>> d = sorted(foo().items())
Georg Brandl116aa622007-08-15 14:28:22 +0000717 >>> d
718 [('Harry', 'broomstick'), ('Hermione', 'hippogryph')]
719
720There are others, but you get the idea.
721
722Another bad idea is to print things that embed an object address, like ::
723
724 >>> id(1.0) # certain to fail some of the time
725 7948648
726 >>> class C: pass
727 >>> C() # the default repr() for instances embeds an address
728 <__main__.C instance at 0x00AC18F0>
729
730The :const:`ELLIPSIS` directive gives a nice approach for the last example::
731
732 >>> C() #doctest: +ELLIPSIS
733 <__main__.C instance at 0x...>
734
735Floating-point numbers are also subject to small output variations across
736platforms, because Python defers to the platform C library for float formatting,
737and C libraries vary widely in quality here. ::
738
739 >>> 1./7 # risky
740 0.14285714285714285
Georg Brandl6911e3c2007-09-04 07:15:32 +0000741 >>> print(1./7) # safer
Georg Brandl116aa622007-08-15 14:28:22 +0000742 0.142857142857
Georg Brandl6911e3c2007-09-04 07:15:32 +0000743 >>> print(round(1./7, 6)) # much safer
Georg Brandl116aa622007-08-15 14:28:22 +0000744 0.142857
745
746Numbers of the form ``I/2.**J`` are safe across all platforms, and I often
747contrive doctest examples to produce numbers of that form::
748
749 >>> 3./4 # utterly safe
750 0.75
751
752Simple fractions are also easier for people to understand, and that makes for
753better documentation.
754
755
756.. _doctest-basic-api:
757
758Basic API
759---------
760
761The functions :func:`testmod` and :func:`testfile` provide a simple interface to
762doctest that should be sufficient for most basic uses. For a less formal
763introduction to these two functions, see sections :ref:`doctest-simple-testmod`
764and :ref:`doctest-simple-testfile`.
765
766
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000767.. 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 +0000768
769 All arguments except *filename* are optional, and should be specified in keyword
770 form.
771
772 Test examples in the file named *filename*. Return ``(failure_count,
773 test_count)``.
774
775 Optional argument *module_relative* specifies how the filename should be
776 interpreted:
777
778 * If *module_relative* is ``True`` (the default), then *filename* specifies an
779 OS-independent module-relative path. By default, this path is relative to the
780 calling module's directory; but if the *package* argument is specified, then it
781 is relative to that package. To ensure OS-independence, *filename* should use
782 ``/`` characters to separate path segments, and may not be an absolute path
783 (i.e., it may not begin with ``/``).
784
785 * If *module_relative* is ``False``, then *filename* specifies an OS-specific
786 path. The path may be absolute or relative; relative paths are resolved with
787 respect to the current working directory.
788
789 Optional argument *name* gives the name of the test; by default, or if ``None``,
790 ``os.path.basename(filename)`` is used.
791
792 Optional argument *package* is a Python package or the name of a Python package
793 whose directory should be used as the base directory for a module-relative
794 filename. If no package is specified, then the calling module's directory is
795 used as the base directory for module-relative filenames. It is an error to
796 specify *package* if *module_relative* is ``False``.
797
798 Optional argument *globs* gives a dict to be used as the globals when executing
799 examples. A new shallow copy of this dict is created for the doctest, so its
800 examples start with a clean slate. By default, or if ``None``, a new empty dict
801 is used.
802
803 Optional argument *extraglobs* gives a dict merged into the globals used to
804 execute examples. This works like :meth:`dict.update`: if *globs* and
805 *extraglobs* have a common key, the associated value in *extraglobs* appears in
806 the combined dict. By default, or if ``None``, no extra globals are used. This
807 is an advanced feature that allows parameterization of doctests. For example, a
808 doctest can be written for a base class, using a generic name for the class,
809 then reused to test any number of subclasses by passing an *extraglobs* dict
810 mapping the generic name to the subclass to be tested.
811
812 Optional argument *verbose* prints lots of stuff if true, and prints only
813 failures if false; by default, or if ``None``, it's true if and only if ``'-v'``
814 is in ``sys.argv``.
815
816 Optional argument *report* prints a summary at the end when true, else prints
817 nothing at the end. In verbose mode, the summary is detailed, else the summary
818 is very brief (in fact, empty if all tests passed).
819
820 Optional argument *optionflags* or's together option flags. See section
821 :ref:`doctest-options`.
822
823 Optional argument *raise_on_error* defaults to false. If true, an exception is
824 raised upon the first failure or unexpected exception in an example. This
825 allows failures to be post-mortem debugged. Default behavior is to continue
826 running examples.
827
828 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass) that
829 should be used to extract tests from the files. It defaults to a normal parser
830 (i.e., ``DocTestParser()``).
831
832 Optional argument *encoding* specifies an encoding that should be used to
833 convert the file to unicode.
834
Georg Brandl116aa622007-08-15 14:28:22 +0000835
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000836.. 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 +0000837
838 All arguments are optional, and all except for *m* should be specified in
839 keyword form.
840
841 Test examples in docstrings in functions and classes reachable from module *m*
842 (or module :mod:`__main__` if *m* is not supplied or is ``None``), starting with
843 ``m.__doc__``.
844
845 Also test examples reachable from dict ``m.__test__``, if it exists and is not
846 ``None``. ``m.__test__`` maps names (strings) to functions, classes and
847 strings; function and class docstrings are searched for examples; strings are
848 searched directly, as if they were docstrings.
849
850 Only docstrings attached to objects belonging to module *m* are searched.
851
852 Return ``(failure_count, test_count)``.
853
854 Optional argument *name* gives the name of the module; by default, or if
855 ``None``, ``m.__name__`` is used.
856
857 Optional argument *exclude_empty* defaults to false. If true, objects for which
858 no doctests are found are excluded from consideration. The default is a backward
859 compatibility hack, so that code still using :meth:`doctest.master.summarize` in
860 conjunction with :func:`testmod` continues to get output for objects with no
861 tests. The *exclude_empty* argument to the newer :class:`DocTestFinder`
862 constructor defaults to true.
863
864 Optional arguments *extraglobs*, *verbose*, *report*, *optionflags*,
865 *raise_on_error*, and *globs* are the same as for function :func:`testfile`
866 above, except that *globs* defaults to ``m.__dict__``.
867
Georg Brandl116aa622007-08-15 14:28:22 +0000868
869There's also a function to run the doctests associated with a single object.
870This function is provided for backward compatibility. There are no plans to
871deprecate it, but it's rarely useful:
872
873
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000874.. function:: run_docstring_examples(f, globs, verbose=False, name="NoName", compileflags=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +0000875
876 Test examples associated with object *f*; for example, *f* may be a module,
877 function, or class object.
878
879 A shallow copy of dictionary argument *globs* is used for the execution context.
880
881 Optional argument *name* is used in failure messages, and defaults to
882 ``"NoName"``.
883
884 If optional argument *verbose* is true, output is generated even if there are no
885 failures. By default, output is generated only in case of an example failure.
886
887 Optional argument *compileflags* gives the set of flags that should be used by
888 the Python compiler when running the examples. By default, or if ``None``,
889 flags are deduced corresponding to the set of future features found in *globs*.
890
891 Optional argument *optionflags* works as for function :func:`testfile` above.
892
893
894.. _doctest-unittest-api:
895
896Unittest API
897------------
898
899As your collection of doctest'ed modules grows, you'll want a way to run all
Georg Brandl31835852008-05-12 17:38:56 +0000900their doctests systematically. :mod:`doctest` provides two functions that can
901be used to create :mod:`unittest` test suites from modules and text files
902containing doctests. These test suites can then be run using :mod:`unittest`
903test runners::
Georg Brandl116aa622007-08-15 14:28:22 +0000904
905 import unittest
906 import doctest
907 import my_module_with_doctests, and_another
908
909 suite = unittest.TestSuite()
910 for mod in my_module_with_doctests, and_another:
911 suite.addTest(doctest.DocTestSuite(mod))
912 runner = unittest.TextTestRunner()
913 runner.run(suite)
914
915There are two main functions for creating :class:`unittest.TestSuite` instances
916from text files and modules with doctests:
917
918
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000919.. 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 +0000920
921 Convert doctest tests from one or more text files to a
922 :class:`unittest.TestSuite`.
923
924 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
925 and runs the interactive examples in each file. If an example in any file
926 fails, then the synthesized unit test fails, and a :exc:`failureException`
927 exception is raised showing the name of the file containing the test and a
928 (sometimes approximate) line number.
929
930 Pass one or more paths (as strings) to text files to be examined.
931
932 Options may be provided as keyword arguments:
933
934 Optional argument *module_relative* specifies how the filenames in *paths*
935 should be interpreted:
936
Benjamin Petersond23f8222009-04-05 19:13:16 +0000937 * If *module_relative* is ``True`` (the default), then each filename in
938 *paths* specifies an OS-independent module-relative path. By default, this
939 path is relative to the calling module's directory; but if the *package*
940 argument is specified, then it is relative to that package. To ensure
941 OS-independence, each filename should use ``/`` characters to separate path
942 segments, and may not be an absolute path (i.e., it may not begin with
943 ``/``).
Georg Brandl116aa622007-08-15 14:28:22 +0000944
Benjamin Petersond23f8222009-04-05 19:13:16 +0000945 * If *module_relative* is ``False``, then each filename in *paths* specifies
946 an OS-specific path. The path may be absolute or relative; relative paths
947 are resolved with respect to the current working directory.
Georg Brandl116aa622007-08-15 14:28:22 +0000948
Benjamin Petersond23f8222009-04-05 19:13:16 +0000949 Optional argument *package* is a Python package or the name of a Python
950 package whose directory should be used as the base directory for
951 module-relative filenames in *paths*. If no package is specified, then the
952 calling module's directory is used as the base directory for module-relative
953 filenames. It is an error to specify *package* if *module_relative* is
954 ``False``.
Georg Brandl116aa622007-08-15 14:28:22 +0000955
Benjamin Petersond23f8222009-04-05 19:13:16 +0000956 Optional argument *setUp* specifies a set-up function for the test suite.
957 This is called before running the tests in each file. The *setUp* function
Georg Brandl116aa622007-08-15 14:28:22 +0000958 will be passed a :class:`DocTest` object. The setUp function can access the
959 test globals as the *globs* attribute of the test passed.
960
Benjamin Petersond23f8222009-04-05 19:13:16 +0000961 Optional argument *tearDown* specifies a tear-down function for the test
962 suite. This is called after running the tests in each file. The *tearDown*
963 function will be passed a :class:`DocTest` object. The setUp function can
964 access the test globals as the *globs* attribute of the test passed.
965
Georg Brandl116aa622007-08-15 14:28:22 +0000966 Optional argument *globs* is a dictionary containing the initial global
967 variables for the tests. A new copy of this dictionary is created for each
968 test. By default, *globs* is a new empty dictionary.
969
970 Optional argument *optionflags* specifies the default doctest options for the
971 tests, created by or-ing together individual option flags. See section
Benjamin Petersond23f8222009-04-05 19:13:16 +0000972 :ref:`doctest-options`. See function :func:`set_unittest_reportflags` below
973 for a better way to set reporting options.
Georg Brandl116aa622007-08-15 14:28:22 +0000974
Benjamin Petersond23f8222009-04-05 19:13:16 +0000975 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass)
976 that should be used to extract tests from the files. It defaults to a normal
977 parser (i.e., ``DocTestParser()``).
Georg Brandl116aa622007-08-15 14:28:22 +0000978
979 Optional argument *encoding* specifies an encoding that should be used to
980 convert the file to unicode.
981
Georg Brandl55ac8f02007-09-01 13:51:09 +0000982 The global ``__file__`` is added to the globals provided to doctests loaded
983 from a text file using :func:`DocFileSuite`.
Georg Brandl116aa622007-08-15 14:28:22 +0000984
985
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000986.. function:: DocTestSuite(module=None, globs=None, extraglobs=None, test_finder=None, setUp=None, tearDown=None, checker=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000987
988 Convert doctest tests for a module to a :class:`unittest.TestSuite`.
989
990 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
991 and runs each doctest in the module. If any of the doctests fail, then the
992 synthesized unit test fails, and a :exc:`failureException` exception is raised
993 showing the name of the file containing the test and a (sometimes approximate)
994 line number.
995
996 Optional argument *module* provides the module to be tested. It can be a module
997 object or a (possibly dotted) module name. If not specified, the module calling
998 this function is used.
999
1000 Optional argument *globs* is a dictionary containing the initial global
1001 variables for the tests. A new copy of this dictionary is created for each
1002 test. By default, *globs* is a new empty dictionary.
1003
1004 Optional argument *extraglobs* specifies an extra set of global variables, which
1005 is merged into *globs*. By default, no extra globals are used.
1006
1007 Optional argument *test_finder* is the :class:`DocTestFinder` object (or a
1008 drop-in replacement) that is used to extract doctests from the module.
1009
1010 Optional arguments *setUp*, *tearDown*, and *optionflags* are the same as for
1011 function :func:`DocFileSuite` above.
1012
Georg Brandl55ac8f02007-09-01 13:51:09 +00001013 This function uses the same search technique as :func:`testmod`.
Georg Brandl116aa622007-08-15 14:28:22 +00001014
Georg Brandl116aa622007-08-15 14:28:22 +00001015
1016Under the covers, :func:`DocTestSuite` creates a :class:`unittest.TestSuite` out
1017of :class:`doctest.DocTestCase` instances, and :class:`DocTestCase` is a
1018subclass of :class:`unittest.TestCase`. :class:`DocTestCase` isn't documented
1019here (it's an internal detail), but studying its code can answer questions about
1020the exact details of :mod:`unittest` integration.
1021
1022Similarly, :func:`DocFileSuite` creates a :class:`unittest.TestSuite` out of
1023:class:`doctest.DocFileCase` instances, and :class:`DocFileCase` is a subclass
1024of :class:`DocTestCase`.
1025
1026So both ways of creating a :class:`unittest.TestSuite` run instances of
1027:class:`DocTestCase`. This is important for a subtle reason: when you run
1028:mod:`doctest` functions yourself, you can control the :mod:`doctest` options in
1029use directly, by passing option flags to :mod:`doctest` functions. However, if
1030you're writing a :mod:`unittest` framework, :mod:`unittest` ultimately controls
1031when and how tests get run. The framework author typically wants to control
1032:mod:`doctest` reporting options (perhaps, e.g., specified by command line
1033options), but there's no way to pass options through :mod:`unittest` to
1034:mod:`doctest` test runners.
1035
1036For this reason, :mod:`doctest` also supports a notion of :mod:`doctest`
1037reporting flags specific to :mod:`unittest` support, via this function:
1038
1039
1040.. function:: set_unittest_reportflags(flags)
1041
1042 Set the :mod:`doctest` reporting flags to use.
1043
1044 Argument *flags* or's together option flags. See section
1045 :ref:`doctest-options`. Only "reporting flags" can be used.
1046
1047 This is a module-global setting, and affects all future doctests run by module
1048 :mod:`unittest`: the :meth:`runTest` method of :class:`DocTestCase` looks at
1049 the option flags specified for the test case when the :class:`DocTestCase`
1050 instance was constructed. If no reporting flags were specified (which is the
1051 typical and expected case), :mod:`doctest`'s :mod:`unittest` reporting flags are
1052 or'ed into the option flags, and the option flags so augmented are passed to the
1053 :class:`DocTestRunner` instance created to run the doctest. If any reporting
1054 flags were specified when the :class:`DocTestCase` instance was constructed,
1055 :mod:`doctest`'s :mod:`unittest` reporting flags are ignored.
1056
1057 The value of the :mod:`unittest` reporting flags in effect before the function
1058 was called is returned by the function.
1059
Georg Brandl116aa622007-08-15 14:28:22 +00001060
1061.. _doctest-advanced-api:
1062
1063Advanced API
1064------------
1065
1066The basic API is a simple wrapper that's intended to make doctest easy to use.
1067It is fairly flexible, and should meet most users' needs; however, if you
1068require more fine-grained control over testing, or wish to extend doctest's
1069capabilities, then you should use the advanced API.
1070
1071The advanced API revolves around two container classes, which are used to store
1072the interactive examples extracted from doctest cases:
1073
Ezio Melotti890c1932009-12-19 23:33:46 +00001074* :class:`Example`: A single Python :term:`statement`, paired with its expected
Christian Heimesd8654cf2007-12-02 15:22:16 +00001075 output.
Georg Brandl116aa622007-08-15 14:28:22 +00001076
1077* :class:`DocTest`: A collection of :class:`Example`\ s, typically extracted
1078 from a single docstring or text file.
1079
1080Additional processing classes are defined to find, parse, and run, and check
1081doctest examples:
1082
1083* :class:`DocTestFinder`: Finds all docstrings in a given module, and uses a
1084 :class:`DocTestParser` to create a :class:`DocTest` from every docstring that
1085 contains interactive examples.
1086
1087* :class:`DocTestParser`: Creates a :class:`DocTest` object from a string (such
1088 as an object's docstring).
1089
1090* :class:`DocTestRunner`: Executes the examples in a :class:`DocTest`, and uses
1091 an :class:`OutputChecker` to verify their output.
1092
1093* :class:`OutputChecker`: Compares the actual output from a doctest example with
1094 the expected output, and decides whether they match.
1095
1096The relationships among these processing classes are summarized in the following
1097diagram::
1098
1099 list of:
1100 +------+ +---------+
1101 |module| --DocTestFinder-> | DocTest | --DocTestRunner-> results
1102 +------+ | ^ +---------+ | ^ (printed)
1103 | | | Example | | |
1104 v | | ... | v |
1105 DocTestParser | Example | OutputChecker
1106 +---------+
1107
1108
1109.. _doctest-doctest:
1110
1111DocTest Objects
1112^^^^^^^^^^^^^^^
1113
1114
1115.. class:: DocTest(examples, globs, name, filename, lineno, docstring)
1116
1117 A collection of doctest examples that should be run in a single namespace. The
1118 constructor arguments are used to initialize the member variables of the same
1119 names.
1120
Georg Brandl116aa622007-08-15 14:28:22 +00001121
Benjamin Petersone41251e2008-04-25 01:59:09 +00001122 :class:`DocTest` defines the following member variables. They are initialized by
1123 the constructor, and should not be modified directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001124
1125
Benjamin Petersone41251e2008-04-25 01:59:09 +00001126 .. attribute:: examples
Georg Brandl116aa622007-08-15 14:28:22 +00001127
Benjamin Petersone41251e2008-04-25 01:59:09 +00001128 A list of :class:`Example` objects encoding the individual interactive Python
1129 examples that should be run by this test.
Georg Brandl116aa622007-08-15 14:28:22 +00001130
1131
Benjamin Petersone41251e2008-04-25 01:59:09 +00001132 .. attribute:: globs
Georg Brandl116aa622007-08-15 14:28:22 +00001133
Benjamin Petersone41251e2008-04-25 01:59:09 +00001134 The namespace (aka globals) that the examples should be run in. This is a
1135 dictionary mapping names to values. Any changes to the namespace made by the
1136 examples (such as binding new variables) will be reflected in :attr:`globs`
1137 after the test is run.
Georg Brandl116aa622007-08-15 14:28:22 +00001138
1139
Benjamin Petersone41251e2008-04-25 01:59:09 +00001140 .. attribute:: name
Georg Brandl116aa622007-08-15 14:28:22 +00001141
Benjamin Petersone41251e2008-04-25 01:59:09 +00001142 A string name identifying the :class:`DocTest`. Typically, this is the name
1143 of the object or file that the test was extracted from.
Georg Brandl116aa622007-08-15 14:28:22 +00001144
1145
Benjamin Petersone41251e2008-04-25 01:59:09 +00001146 .. attribute:: filename
Georg Brandl116aa622007-08-15 14:28:22 +00001147
Benjamin Petersone41251e2008-04-25 01:59:09 +00001148 The name of the file that this :class:`DocTest` was extracted from; or
1149 ``None`` if the filename is unknown, or if the :class:`DocTest` was not
1150 extracted from a file.
Georg Brandl116aa622007-08-15 14:28:22 +00001151
1152
Benjamin Petersone41251e2008-04-25 01:59:09 +00001153 .. attribute:: lineno
Georg Brandl116aa622007-08-15 14:28:22 +00001154
Benjamin Petersone41251e2008-04-25 01:59:09 +00001155 The line number within :attr:`filename` where this :class:`DocTest` begins, or
1156 ``None`` if the line number is unavailable. This line number is zero-based
1157 with respect to the beginning of the file.
Georg Brandl116aa622007-08-15 14:28:22 +00001158
1159
Benjamin Petersone41251e2008-04-25 01:59:09 +00001160 .. attribute:: docstring
Georg Brandl116aa622007-08-15 14:28:22 +00001161
Benjamin Petersone41251e2008-04-25 01:59:09 +00001162 The string that the test was extracted from, or 'None' if the string is
1163 unavailable, or if the test was not extracted from a string.
Georg Brandl116aa622007-08-15 14:28:22 +00001164
1165
1166.. _doctest-example:
1167
1168Example Objects
1169^^^^^^^^^^^^^^^
1170
1171
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001172.. class:: Example(source, want, exc_msg=None, lineno=0, indent=0, options=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001173
1174 A single interactive example, consisting of a Python statement and its expected
1175 output. The constructor arguments are used to initialize the member variables
1176 of the same names.
1177
Georg Brandl116aa622007-08-15 14:28:22 +00001178
Benjamin Petersone41251e2008-04-25 01:59:09 +00001179 :class:`Example` defines the following member variables. They are initialized by
1180 the constructor, and should not be modified directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001181
1182
Benjamin Petersone41251e2008-04-25 01:59:09 +00001183 .. attribute:: source
Georg Brandl116aa622007-08-15 14:28:22 +00001184
Benjamin Petersone41251e2008-04-25 01:59:09 +00001185 A string containing the example's source code. This source code consists of a
1186 single Python statement, and always ends with a newline; the constructor adds
1187 a newline when necessary.
Georg Brandl116aa622007-08-15 14:28:22 +00001188
1189
Benjamin Petersone41251e2008-04-25 01:59:09 +00001190 .. attribute:: want
Georg Brandl116aa622007-08-15 14:28:22 +00001191
Benjamin Petersone41251e2008-04-25 01:59:09 +00001192 The expected output from running the example's source code (either from
1193 stdout, or a traceback in case of exception). :attr:`want` ends with a
1194 newline unless no output is expected, in which case it's an empty string. The
1195 constructor adds a newline when necessary.
Georg Brandl116aa622007-08-15 14:28:22 +00001196
1197
Benjamin Petersone41251e2008-04-25 01:59:09 +00001198 .. attribute:: exc_msg
Georg Brandl116aa622007-08-15 14:28:22 +00001199
Benjamin Petersone41251e2008-04-25 01:59:09 +00001200 The exception message generated by the example, if the example is expected to
1201 generate an exception; or ``None`` if it is not expected to generate an
1202 exception. This exception message is compared against the return value of
1203 :func:`traceback.format_exception_only`. :attr:`exc_msg` ends with a newline
1204 unless it's ``None``. The constructor adds a newline if needed.
Georg Brandl116aa622007-08-15 14:28:22 +00001205
1206
Benjamin Petersone41251e2008-04-25 01:59:09 +00001207 .. attribute:: lineno
Georg Brandl116aa622007-08-15 14:28:22 +00001208
Benjamin Petersone41251e2008-04-25 01:59:09 +00001209 The line number within the string containing this example where the example
1210 begins. This line number is zero-based with respect to the beginning of the
1211 containing string.
Georg Brandl116aa622007-08-15 14:28:22 +00001212
1213
Benjamin Petersone41251e2008-04-25 01:59:09 +00001214 .. attribute:: indent
Georg Brandl116aa622007-08-15 14:28:22 +00001215
Benjamin Petersone41251e2008-04-25 01:59:09 +00001216 The example's indentation in the containing string, i.e., the number of space
1217 characters that precede the example's first prompt.
Georg Brandl116aa622007-08-15 14:28:22 +00001218
1219
Benjamin Petersone41251e2008-04-25 01:59:09 +00001220 .. attribute:: options
Georg Brandl116aa622007-08-15 14:28:22 +00001221
Benjamin Petersone41251e2008-04-25 01:59:09 +00001222 A dictionary mapping from option flags to ``True`` or ``False``, which is used
1223 to override default options for this example. Any option flags not contained
1224 in this dictionary are left at their default value (as specified by the
1225 :class:`DocTestRunner`'s :attr:`optionflags`). By default, no options are set.
Georg Brandl116aa622007-08-15 14:28:22 +00001226
1227
1228.. _doctest-doctestfinder:
1229
1230DocTestFinder objects
1231^^^^^^^^^^^^^^^^^^^^^
1232
1233
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001234.. class:: DocTestFinder(verbose=False, parser=DocTestParser(), recurse=True, exclude_empty=True)
Georg Brandl116aa622007-08-15 14:28:22 +00001235
1236 A processing class used to extract the :class:`DocTest`\ s that are relevant to
1237 a given object, from its docstring and the docstrings of its contained objects.
1238 :class:`DocTest`\ s can currently be extracted from the following object types:
1239 modules, functions, classes, methods, staticmethods, classmethods, and
1240 properties.
1241
1242 The optional argument *verbose* can be used to display the objects searched by
1243 the finder. It defaults to ``False`` (no output).
1244
1245 The optional argument *parser* specifies the :class:`DocTestParser` object (or a
1246 drop-in replacement) that is used to extract doctests from docstrings.
1247
1248 If the optional argument *recurse* is false, then :meth:`DocTestFinder.find`
1249 will only examine the given object, and not any contained objects.
1250
1251 If the optional argument *exclude_empty* is false, then
1252 :meth:`DocTestFinder.find` will include tests for objects with empty docstrings.
1253
Georg Brandl116aa622007-08-15 14:28:22 +00001254
Benjamin Petersone41251e2008-04-25 01:59:09 +00001255 :class:`DocTestFinder` defines the following method:
Georg Brandl116aa622007-08-15 14:28:22 +00001256
1257
Benjamin Petersone41251e2008-04-25 01:59:09 +00001258 .. method:: find(obj[, name][, module][, globs][, extraglobs])
Georg Brandl116aa622007-08-15 14:28:22 +00001259
Benjamin Petersone41251e2008-04-25 01:59:09 +00001260 Return a list of the :class:`DocTest`\ s that are defined by *obj*'s
1261 docstring, or by any of its contained objects' docstrings.
Georg Brandl116aa622007-08-15 14:28:22 +00001262
Benjamin Petersone41251e2008-04-25 01:59:09 +00001263 The optional argument *name* specifies the object's name; this name will be
1264 used to construct names for the returned :class:`DocTest`\ s. If *name* is
1265 not specified, then ``obj.__name__`` is used.
Georg Brandl116aa622007-08-15 14:28:22 +00001266
Benjamin Petersone41251e2008-04-25 01:59:09 +00001267 The optional parameter *module* is the module that contains the given object.
1268 If the module is not specified or is None, then the test finder will attempt
1269 to automatically determine the correct module. The object's module is used:
Georg Brandl116aa622007-08-15 14:28:22 +00001270
Benjamin Petersone41251e2008-04-25 01:59:09 +00001271 * As a default namespace, if *globs* is not specified.
Georg Brandl116aa622007-08-15 14:28:22 +00001272
Benjamin Petersone41251e2008-04-25 01:59:09 +00001273 * To prevent the DocTestFinder from extracting DocTests from objects that are
1274 imported from other modules. (Contained objects with modules other than
1275 *module* are ignored.)
Georg Brandl116aa622007-08-15 14:28:22 +00001276
Benjamin Petersone41251e2008-04-25 01:59:09 +00001277 * To find the name of the file containing the object.
Georg Brandl116aa622007-08-15 14:28:22 +00001278
Benjamin Petersone41251e2008-04-25 01:59:09 +00001279 * To help find the line number of the object within its file.
Georg Brandl116aa622007-08-15 14:28:22 +00001280
Benjamin Petersone41251e2008-04-25 01:59:09 +00001281 If *module* is ``False``, no attempt to find the module will be made. This is
1282 obscure, of use mostly in testing doctest itself: if *module* is ``False``, or
1283 is ``None`` but cannot be found automatically, then all objects are considered
1284 to belong to the (non-existent) module, so all contained objects will
1285 (recursively) be searched for doctests.
Georg Brandl116aa622007-08-15 14:28:22 +00001286
Benjamin Petersone41251e2008-04-25 01:59:09 +00001287 The globals for each :class:`DocTest` is formed by combining *globs* and
1288 *extraglobs* (bindings in *extraglobs* override bindings in *globs*). A new
1289 shallow copy of the globals dictionary is created for each :class:`DocTest`.
1290 If *globs* is not specified, then it defaults to the module's *__dict__*, if
1291 specified, or ``{}`` otherwise. If *extraglobs* is not specified, then it
1292 defaults to ``{}``.
Georg Brandl116aa622007-08-15 14:28:22 +00001293
1294
1295.. _doctest-doctestparser:
1296
1297DocTestParser objects
1298^^^^^^^^^^^^^^^^^^^^^
1299
1300
1301.. class:: DocTestParser()
1302
1303 A processing class used to extract interactive examples from a string, and use
1304 them to create a :class:`DocTest` object.
1305
Georg Brandl116aa622007-08-15 14:28:22 +00001306
Benjamin Petersone41251e2008-04-25 01:59:09 +00001307 :class:`DocTestParser` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001308
1309
Benjamin Petersone41251e2008-04-25 01:59:09 +00001310 .. method:: get_doctest(string, globs, name, filename, lineno)
Georg Brandl116aa622007-08-15 14:28:22 +00001311
Benjamin Petersone41251e2008-04-25 01:59:09 +00001312 Extract all doctest examples from the given string, and collect them into a
1313 :class:`DocTest` object.
Georg Brandl116aa622007-08-15 14:28:22 +00001314
Benjamin Petersone41251e2008-04-25 01:59:09 +00001315 *globs*, *name*, *filename*, and *lineno* are attributes for the new
1316 :class:`DocTest` object. See the documentation for :class:`DocTest` for more
1317 information.
Georg Brandl116aa622007-08-15 14:28:22 +00001318
1319
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001320 .. method:: get_examples(string, name='<string>')
Georg Brandl116aa622007-08-15 14:28:22 +00001321
Benjamin Petersone41251e2008-04-25 01:59:09 +00001322 Extract all doctest examples from the given string, and return them as a list
1323 of :class:`Example` objects. Line numbers are 0-based. The optional argument
1324 *name* is a name identifying this string, and is only used for error messages.
Georg Brandl116aa622007-08-15 14:28:22 +00001325
1326
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001327 .. method:: parse(string, name='<string>')
Georg Brandl116aa622007-08-15 14:28:22 +00001328
Benjamin Petersone41251e2008-04-25 01:59:09 +00001329 Divide the given string into examples and intervening text, and return them as
1330 a list of alternating :class:`Example`\ s and strings. Line numbers for the
1331 :class:`Example`\ s are 0-based. The optional argument *name* is a name
1332 identifying this string, and is only used for error messages.
Georg Brandl116aa622007-08-15 14:28:22 +00001333
1334
1335.. _doctest-doctestrunner:
1336
1337DocTestRunner objects
1338^^^^^^^^^^^^^^^^^^^^^
1339
1340
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001341.. class:: DocTestRunner(checker=None, verbose=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +00001342
1343 A processing class used to execute and verify the interactive examples in a
1344 :class:`DocTest`.
1345
1346 The comparison between expected outputs and actual outputs is done by an
1347 :class:`OutputChecker`. This comparison may be customized with a number of
1348 option flags; see section :ref:`doctest-options` for more information. If the
1349 option flags are insufficient, then the comparison may also be customized by
1350 passing a subclass of :class:`OutputChecker` to the constructor.
1351
1352 The test runner's display output can be controlled in two ways. First, an output
1353 function can be passed to :meth:`TestRunner.run`; this function will be called
1354 with strings that should be displayed. It defaults to ``sys.stdout.write``. If
1355 capturing the output is not sufficient, then the display output can be also
1356 customized by subclassing DocTestRunner, and overriding the methods
1357 :meth:`report_start`, :meth:`report_success`,
1358 :meth:`report_unexpected_exception`, and :meth:`report_failure`.
1359
1360 The optional keyword argument *checker* specifies the :class:`OutputChecker`
1361 object (or drop-in replacement) that should be used to compare the expected
1362 outputs to the actual outputs of doctest examples.
1363
1364 The optional keyword argument *verbose* controls the :class:`DocTestRunner`'s
1365 verbosity. If *verbose* is ``True``, then information is printed about each
1366 example, as it is run. If *verbose* is ``False``, then only failures are
1367 printed. If *verbose* is unspecified, or ``None``, then verbose output is used
1368 iff the command-line switch :option:`-v` is used.
1369
1370 The optional keyword argument *optionflags* can be used to control how the test
1371 runner compares expected output to actual output, and how it displays failures.
1372 For more information, see section :ref:`doctest-options`.
1373
Georg Brandl116aa622007-08-15 14:28:22 +00001374
Benjamin Petersone41251e2008-04-25 01:59:09 +00001375 :class:`DocTestParser` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001376
1377
Benjamin Petersone41251e2008-04-25 01:59:09 +00001378 .. method:: report_start(out, test, example)
Georg Brandl116aa622007-08-15 14:28:22 +00001379
Benjamin Petersone41251e2008-04-25 01:59:09 +00001380 Report that the test runner is about to process the given example. This method
1381 is provided to allow subclasses of :class:`DocTestRunner` to customize their
1382 output; it should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001383
Benjamin Petersone41251e2008-04-25 01:59:09 +00001384 *example* is the example about to be processed. *test* is the test
1385 *containing example*. *out* is the output function that was passed to
1386 :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001387
1388
Benjamin Petersone41251e2008-04-25 01:59:09 +00001389 .. method:: report_success(out, test, example, got)
Georg Brandl116aa622007-08-15 14:28:22 +00001390
Benjamin Petersone41251e2008-04-25 01:59:09 +00001391 Report that the given example ran successfully. This method is provided to
1392 allow subclasses of :class:`DocTestRunner` to customize their output; it
1393 should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001394
Benjamin Petersone41251e2008-04-25 01:59:09 +00001395 *example* is the example about to be processed. *got* is the actual output
1396 from the example. *test* is the test containing *example*. *out* is the
1397 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001398
1399
Benjamin Petersone41251e2008-04-25 01:59:09 +00001400 .. method:: report_failure(out, test, example, got)
Georg Brandl116aa622007-08-15 14:28:22 +00001401
Benjamin Petersone41251e2008-04-25 01:59:09 +00001402 Report that the given example failed. This method is provided to allow
1403 subclasses of :class:`DocTestRunner` to customize their output; it should not
1404 be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001405
Benjamin Petersone41251e2008-04-25 01:59:09 +00001406 *example* is the example about to be processed. *got* is the actual output
1407 from the example. *test* is the test containing *example*. *out* is the
1408 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001409
1410
Benjamin Petersone41251e2008-04-25 01:59:09 +00001411 .. method:: report_unexpected_exception(out, test, example, exc_info)
Georg Brandl116aa622007-08-15 14:28:22 +00001412
Benjamin Petersone41251e2008-04-25 01:59:09 +00001413 Report that the given example raised an unexpected exception. This method is
1414 provided to allow subclasses of :class:`DocTestRunner` to customize their
1415 output; it should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001416
Benjamin Petersone41251e2008-04-25 01:59:09 +00001417 *example* is the example about to be processed. *exc_info* is a tuple
1418 containing information about the unexpected exception (as returned by
1419 :func:`sys.exc_info`). *test* is the test containing *example*. *out* is the
1420 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001421
1422
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001423 .. method:: run(test, compileflags=None, out=None, clear_globs=True)
Georg Brandl116aa622007-08-15 14:28:22 +00001424
Benjamin Petersone41251e2008-04-25 01:59:09 +00001425 Run the examples in *test* (a :class:`DocTest` object), and display the
1426 results using the writer function *out*.
Georg Brandl116aa622007-08-15 14:28:22 +00001427
Benjamin Petersone41251e2008-04-25 01:59:09 +00001428 The examples are run in the namespace ``test.globs``. If *clear_globs* is
1429 true (the default), then this namespace will be cleared after the test runs,
1430 to help with garbage collection. If you would like to examine the namespace
1431 after the test completes, then use *clear_globs=False*.
Georg Brandl116aa622007-08-15 14:28:22 +00001432
Benjamin Petersone41251e2008-04-25 01:59:09 +00001433 *compileflags* gives the set of flags that should be used by the Python
1434 compiler when running the examples. If not specified, then it will default to
1435 the set of future-import flags that apply to *globs*.
Georg Brandl116aa622007-08-15 14:28:22 +00001436
Benjamin Petersone41251e2008-04-25 01:59:09 +00001437 The output of each example is checked using the :class:`DocTestRunner`'s
1438 output checker, and the results are formatted by the
1439 :meth:`DocTestRunner.report_\*` methods.
Georg Brandl116aa622007-08-15 14:28:22 +00001440
1441
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001442 .. method:: summarize(verbose=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001443
Benjamin Petersone41251e2008-04-25 01:59:09 +00001444 Print a summary of all the test cases that have been run by this DocTestRunner,
1445 and return a :term:`named tuple` ``TestResults(failed, attempted)``.
Georg Brandl116aa622007-08-15 14:28:22 +00001446
Benjamin Petersone41251e2008-04-25 01:59:09 +00001447 The optional *verbose* argument controls how detailed the summary is. If the
1448 verbosity is not specified, then the :class:`DocTestRunner`'s verbosity is
1449 used.
Georg Brandl116aa622007-08-15 14:28:22 +00001450
1451.. _doctest-outputchecker:
1452
1453OutputChecker objects
1454^^^^^^^^^^^^^^^^^^^^^
1455
1456
1457.. class:: OutputChecker()
1458
1459 A class used to check the whether the actual output from a doctest example
1460 matches the expected output. :class:`OutputChecker` defines two methods:
1461 :meth:`check_output`, which compares a given pair of outputs, and returns true
1462 if they match; and :meth:`output_difference`, which returns a string describing
1463 the differences between two outputs.
1464
Georg Brandl116aa622007-08-15 14:28:22 +00001465
Benjamin Petersone41251e2008-04-25 01:59:09 +00001466 :class:`OutputChecker` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001467
Benjamin Petersone41251e2008-04-25 01:59:09 +00001468 .. method:: check_output(want, got, optionflags)
Georg Brandl116aa622007-08-15 14:28:22 +00001469
Benjamin Petersone41251e2008-04-25 01:59:09 +00001470 Return ``True`` iff the actual output from an example (*got*) matches the
1471 expected output (*want*). These strings are always considered to match if
1472 they are identical; but depending on what option flags the test runner is
1473 using, several non-exact match types are also possible. See section
1474 :ref:`doctest-options` for more information about option flags.
Georg Brandl116aa622007-08-15 14:28:22 +00001475
1476
Benjamin Petersone41251e2008-04-25 01:59:09 +00001477 .. method:: output_difference(example, got, optionflags)
Georg Brandl116aa622007-08-15 14:28:22 +00001478
Benjamin Petersone41251e2008-04-25 01:59:09 +00001479 Return a string describing the differences between the expected output for a
1480 given example (*example*) and the actual output (*got*). *optionflags* is the
1481 set of option flags used to compare *want* and *got*.
Georg Brandl116aa622007-08-15 14:28:22 +00001482
1483
1484.. _doctest-debugging:
1485
1486Debugging
1487---------
1488
1489Doctest provides several mechanisms for debugging doctest examples:
1490
1491* Several functions convert doctests to executable Python programs, which can be
1492 run under the Python debugger, :mod:`pdb`.
1493
1494* The :class:`DebugRunner` class is a subclass of :class:`DocTestRunner` that
1495 raises an exception for the first failing example, containing information about
1496 that example. This information can be used to perform post-mortem debugging on
1497 the example.
1498
1499* The :mod:`unittest` cases generated by :func:`DocTestSuite` support the
1500 :meth:`debug` method defined by :class:`unittest.TestCase`.
1501
1502* You can add a call to :func:`pdb.set_trace` in a doctest example, and you'll
1503 drop into the Python debugger when that line is executed. Then you can inspect
1504 current values of variables, and so on. For example, suppose :file:`a.py`
1505 contains just this module docstring::
1506
1507 """
1508 >>> def f(x):
1509 ... g(x*2)
1510 >>> def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001511 ... print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001512 ... import pdb; pdb.set_trace()
1513 >>> f(3)
1514 9
1515 """
1516
1517 Then an interactive Python session may look like this::
1518
1519 >>> import a, doctest
1520 >>> doctest.testmod(a)
1521 --Return--
1522 > <doctest a[1]>(3)g()->None
1523 -> import pdb; pdb.set_trace()
1524 (Pdb) list
1525 1 def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001526 2 print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001527 3 -> import pdb; pdb.set_trace()
1528 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001529 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001530 6
1531 (Pdb) step
1532 --Return--
1533 > <doctest a[0]>(2)f()->None
1534 -> g(x*2)
1535 (Pdb) list
1536 1 def f(x):
1537 2 -> g(x*2)
1538 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001539 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001540 3
1541 (Pdb) step
1542 --Return--
1543 > <doctest a[2]>(1)?()->None
1544 -> f(3)
1545 (Pdb) cont
1546 (0, 3)
1547 >>>
1548
Georg Brandl116aa622007-08-15 14:28:22 +00001549
1550Functions that convert doctests to Python code, and possibly run the synthesized
1551code under the debugger:
1552
1553
1554.. function:: script_from_examples(s)
1555
1556 Convert text with examples to a script.
1557
1558 Argument *s* is a string containing doctest examples. The string is converted
1559 to a Python script, where doctest examples in *s* are converted to regular code,
1560 and everything else is converted to Python comments. The generated script is
1561 returned as a string. For example, ::
1562
1563 import doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001564 print(doctest.script_from_examples(r"""
Georg Brandl116aa622007-08-15 14:28:22 +00001565 Set x and y to 1 and 2.
1566 >>> x, y = 1, 2
1567
1568 Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001569 >>> print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001570 3
Georg Brandl6911e3c2007-09-04 07:15:32 +00001571 """))
Georg Brandl116aa622007-08-15 14:28:22 +00001572
1573 displays::
1574
1575 # 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 # Expected:
1581 ## 3
1582
1583 This function is used internally by other functions (see below), but can also be
1584 useful when you want to transform an interactive Python session into a Python
1585 script.
1586
Georg Brandl116aa622007-08-15 14:28:22 +00001587
1588.. function:: testsource(module, name)
1589
1590 Convert the doctest for an object to a script.
1591
1592 Argument *module* is a module object, or dotted name of a module, containing the
1593 object whose doctests are of interest. Argument *name* is the name (within the
1594 module) of the object with the doctests of interest. The result is a string,
1595 containing the object's docstring converted to a Python script, as described for
1596 :func:`script_from_examples` above. For example, if module :file:`a.py`
1597 contains a top-level function :func:`f`, then ::
1598
1599 import a, doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001600 print(doctest.testsource(a, "a.f"))
Georg Brandl116aa622007-08-15 14:28:22 +00001601
1602 prints a script version of function :func:`f`'s docstring, with doctests
1603 converted to code, and the rest placed in comments.
1604
Georg Brandl116aa622007-08-15 14:28:22 +00001605
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001606.. function:: debug(module, name, pm=False)
Georg Brandl116aa622007-08-15 14:28:22 +00001607
1608 Debug the doctests for an object.
1609
1610 The *module* and *name* arguments are the same as for function
1611 :func:`testsource` above. The synthesized Python script for the named object's
1612 docstring is written to a temporary file, and then that file is run under the
1613 control of the Python debugger, :mod:`pdb`.
1614
1615 A shallow copy of ``module.__dict__`` is used for both local and global
1616 execution context.
1617
1618 Optional argument *pm* controls whether post-mortem debugging is used. If *pm*
1619 has a true value, the script file is run directly, and the debugger gets
1620 involved only if the script terminates via raising an unhandled exception. If
1621 it does, then post-mortem debugging is invoked, via :func:`pdb.post_mortem`,
1622 passing the traceback object from the unhandled exception. If *pm* is not
1623 specified, or is false, the script is run under the debugger from the start, via
1624 passing an appropriate :func:`exec` call to :func:`pdb.run`.
1625
Georg Brandl116aa622007-08-15 14:28:22 +00001626
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001627.. function:: debug_src(src, pm=False, globs=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001628
1629 Debug the doctests in a string.
1630
1631 This is like function :func:`debug` above, except that a string containing
1632 doctest examples is specified directly, via the *src* argument.
1633
1634 Optional argument *pm* has the same meaning as in function :func:`debug` above.
1635
1636 Optional argument *globs* gives a dictionary to use as both local and global
1637 execution context. If not specified, or ``None``, an empty dictionary is used.
1638 If specified, a shallow copy of the dictionary is used.
1639
Georg Brandl116aa622007-08-15 14:28:22 +00001640
1641The :class:`DebugRunner` class, and the special exceptions it may raise, are of
1642most interest to testing framework authors, and will only be sketched here. See
1643the source code, and especially :class:`DebugRunner`'s docstring (which is a
1644doctest!) for more details:
1645
1646
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001647.. class:: DebugRunner(checker=None, verbose=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +00001648
1649 A subclass of :class:`DocTestRunner` that raises an exception as soon as a
1650 failure is encountered. If an unexpected exception occurs, an
1651 :exc:`UnexpectedException` exception is raised, containing the test, the
1652 example, and the original exception. If the output doesn't match, then a
1653 :exc:`DocTestFailure` exception is raised, containing the test, the example, and
1654 the actual output.
1655
1656 For information about the constructor parameters and methods, see the
1657 documentation for :class:`DocTestRunner` in section :ref:`doctest-advanced-api`.
1658
1659There are two exceptions that may be raised by :class:`DebugRunner` instances:
1660
1661
1662.. exception:: DocTestFailure(test, example, got)
1663
1664 An exception thrown by :class:`DocTestRunner` to signal that a doctest example's
1665 actual output did not match its expected output. The constructor arguments are
1666 used to initialize the member variables of the same names.
1667
1668:exc:`DocTestFailure` defines the following member variables:
1669
1670
1671.. attribute:: DocTestFailure.test
1672
1673 The :class:`DocTest` object that was being run when the example failed.
1674
1675
1676.. attribute:: DocTestFailure.example
1677
1678 The :class:`Example` that failed.
1679
1680
1681.. attribute:: DocTestFailure.got
1682
1683 The example's actual output.
1684
1685
1686.. exception:: UnexpectedException(test, example, exc_info)
1687
1688 An exception thrown by :class:`DocTestRunner` to signal that a doctest example
1689 raised an unexpected exception. The constructor arguments are used to
1690 initialize the member variables of the same names.
1691
1692:exc:`UnexpectedException` defines the following member variables:
1693
1694
1695.. attribute:: UnexpectedException.test
1696
1697 The :class:`DocTest` object that was being run when the example failed.
1698
1699
1700.. attribute:: UnexpectedException.example
1701
1702 The :class:`Example` that failed.
1703
1704
1705.. attribute:: UnexpectedException.exc_info
1706
1707 A tuple containing information about the unexpected exception, as returned by
1708 :func:`sys.exc_info`.
1709
1710
1711.. _doctest-soapbox:
1712
1713Soapbox
1714-------
1715
1716As mentioned in the introduction, :mod:`doctest` has grown to have three primary
1717uses:
1718
1719#. Checking examples in docstrings.
1720
1721#. Regression testing.
1722
1723#. Executable documentation / literate testing.
1724
1725These uses have different requirements, and it is important to distinguish them.
1726In particular, filling your docstrings with obscure test cases makes for bad
1727documentation.
1728
1729When writing a docstring, choose docstring examples with care. There's an art to
1730this that needs to be learned---it may not be natural at first. Examples should
1731add genuine value to the documentation. A good example can often be worth many
1732words. If done with care, the examples will be invaluable for your users, and
1733will pay back the time it takes to collect them many times over as the years go
1734by and things change. I'm still amazed at how often one of my :mod:`doctest`
1735examples stops working after a "harmless" change.
1736
1737Doctest also makes an excellent tool for regression testing, especially if you
1738don't skimp on explanatory text. By interleaving prose and examples, it becomes
1739much easier to keep track of what's actually being tested, and why. When a test
1740fails, good prose can make it much easier to figure out what the problem is, and
1741how it should be fixed. It's true that you could write extensive comments in
1742code-based testing, but few programmers do. Many have found that using doctest
1743approaches instead leads to much clearer tests. Perhaps this is simply because
1744doctest makes writing prose a little easier than writing code, while writing
1745comments in code is a little harder. I think it goes deeper than just that:
1746the natural attitude when writing a doctest-based test is that you want to
1747explain the fine points of your software, and illustrate them with examples.
1748This in turn naturally leads to test files that start with the simplest
1749features, and logically progress to complications and edge cases. A coherent
1750narrative is the result, instead of a collection of isolated functions that test
1751isolated bits of functionality seemingly at random. It's a different attitude,
1752and produces different results, blurring the distinction between testing and
1753explaining.
1754
1755Regression testing is best confined to dedicated objects or files. There are
1756several options for organizing tests:
1757
1758* Write text files containing test cases as interactive examples, and test the
1759 files using :func:`testfile` or :func:`DocFileSuite`. This is recommended,
1760 although is easiest to do for new projects, designed from the start to use
1761 doctest.
1762
1763* Define functions named ``_regrtest_topic`` that consist of single docstrings,
1764 containing test cases for the named topics. These functions can be included in
1765 the same file as the module, or separated out into a separate test file.
1766
1767* Define a ``__test__`` dictionary mapping from regression test topics to
1768 docstrings containing test cases.
1769
1770.. rubric:: Footnotes
1771
1772.. [#] Examples containing both expected output and an exception are not supported.
1773 Trying to guess where one ends and the other begins is too error-prone, and that
1774 also makes for a confusing test.