blob: cdd6c2624514da3e77e970048ba4327aad285fca [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
Éric Araujo713d3032010-11-18 16:38:46 +000091``-v`` to the script, and :mod:`doctest` prints a detailed log of what
Georg Brandl116aa622007-08-15 14:28:22 +000092it'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
Éric Araujo713d3032010-11-18 16:38:46 +0000154Run it with the ``-v`` switch instead::
Georg Brandl116aa622007-08-15 14:28:22 +0000155
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,
Éric Araujo713d3032010-11-18 16:38:46 +0000163``sys.argv`` is not examined by :func:`testmod` (so passing ``-v`` or not
Georg Brandl116aa622007-08-15 14:28:22 +0000164has 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
Éric Araujo713d3032010-11-18 16:38:46 +0000232``-v`` command-line switch or with the optional keyword argument
Georg Brandl116aa622007-08-15 14:28:22 +0000233*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 Murray9691e592010-06-15 23:46:40 +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 Murray9691e592010-06-15 23:46:40 +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
Georg Brandl116aa622007-08-15 14:28:22 +0000328* Output to stdout is captured, but not output to stderr (exception tracebacks
329 are captured via a different means).
330
331* If you continue a line via backslashing in an interactive session, or for any
332 other reason use a backslash, you should use a raw docstring, which will
333 preserve your backslashes exactly as you type them::
334
335 >>> def f(x):
336 ... r'''Backslashes in a raw docstring: m\n'''
Georg Brandl6911e3c2007-09-04 07:15:32 +0000337 >>> print(f.__doc__)
Georg Brandl116aa622007-08-15 14:28:22 +0000338 Backslashes in a raw docstring: m\n
339
340 Otherwise, the backslash will be interpreted as part of the string. For example,
341 the "\\" above would be interpreted as a newline character. Alternatively, you
342 can double each backslash in the doctest version (and not use a raw string)::
343
344 >>> def f(x):
345 ... '''Backslashes in a raw docstring: m\\n'''
Georg Brandl6911e3c2007-09-04 07:15:32 +0000346 >>> print(f.__doc__)
Georg Brandl116aa622007-08-15 14:28:22 +0000347 Backslashes in a raw docstring: m\n
348
349* The starting column doesn't matter::
350
351 >>> assert "Easy!"
352 >>> import math
353 >>> math.floor(1.9)
R. David Murray7c5714f2009-11-23 03:13:23 +0000354 1
Georg Brandl116aa622007-08-15 14:28:22 +0000355
356 and as many leading whitespace characters are stripped from the expected output
357 as appeared in the initial ``'>>> '`` line that started the example.
358
359
360.. _doctest-execution-context:
361
362What's the Execution Context?
363^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
364
365By default, each time :mod:`doctest` finds a docstring to test, it uses a
366*shallow copy* of :mod:`M`'s globals, so that running tests doesn't change the
367module's real globals, and so that one test in :mod:`M` can't leave behind
368crumbs that accidentally allow another test to work. This means examples can
369freely use any names defined at top-level in :mod:`M`, and names defined earlier
370in the docstring being run. Examples cannot see names defined in other
371docstrings.
372
373You can force use of your own dict as the execution context by passing
374``globs=your_dict`` to :func:`testmod` or :func:`testfile` instead.
375
376
377.. _doctest-exceptions:
378
379What About Exceptions?
380^^^^^^^^^^^^^^^^^^^^^^
381
382No problem, provided that the traceback is the only output produced by the
383example: just paste in the traceback. [#]_ Since tracebacks contain details
384that are likely to change rapidly (for example, exact file paths and line
385numbers), this is one case where doctest works hard to be flexible in what it
386accepts.
387
388Simple example::
389
390 >>> [1, 2, 3].remove(42)
391 Traceback (most recent call last):
392 File "<stdin>", line 1, in ?
393 ValueError: list.remove(x): x not in list
394
395That doctest succeeds if :exc:`ValueError` is raised, with the ``list.remove(x):
396x not in list`` detail as shown.
397
398The expected output for an exception must start with a traceback header, which
399may be either of the following two lines, indented the same as the first line of
400the example::
401
402 Traceback (most recent call last):
403 Traceback (innermost last):
404
405The traceback header is followed by an optional traceback stack, whose contents
406are ignored by doctest. The traceback stack is typically omitted, or copied
407verbatim from an interactive session.
408
409The traceback stack is followed by the most interesting part: the line(s)
410containing the exception type and detail. This is usually the last line of a
411traceback, but can extend across multiple lines if the exception has a
412multi-line detail::
413
414 >>> raise ValueError('multi\n line\ndetail')
415 Traceback (most recent call last):
416 File "<stdin>", line 1, in ?
417 ValueError: multi
418 line
419 detail
420
421The last three lines (starting with :exc:`ValueError`) are compared against the
422exception's type and detail, and the rest are ignored.
423
424Best practice is to omit the traceback stack, unless it adds significant
425documentation value to the example. So the last example is probably better as::
426
427 >>> raise ValueError('multi\n line\ndetail')
428 Traceback (most recent call last):
429 ...
430 ValueError: multi
431 line
432 detail
433
434Note that tracebacks are treated very specially. In particular, in the
435rewritten example, the use of ``...`` is independent of doctest's
436:const:`ELLIPSIS` option. The ellipsis in that example could be left out, or
437could just as well be three (or three hundred) commas or digits, or an indented
438transcript of a Monty Python skit.
439
440Some details you should read once, but won't need to remember:
441
442* Doctest can't guess whether your expected output came from an exception
443 traceback or from ordinary printing. So, e.g., an example that expects
444 ``ValueError: 42 is prime`` will pass whether :exc:`ValueError` is actually
445 raised or if the example merely prints that traceback text. In practice,
446 ordinary output rarely begins with a traceback header line, so this doesn't
447 create real problems.
448
449* Each line of the traceback stack (if present) must be indented further than
450 the first line of the example, *or* start with a non-alphanumeric character.
451 The first line following the traceback header indented the same and starting
452 with an alphanumeric is taken to be the start of the exception detail. Of
453 course this does the right thing for genuine tracebacks.
454
Nick Coghlan5e76e942010-06-12 13:42:46 +0000455* When the :const:`IGNORE_EXCEPTION_DETAIL` doctest option is specified,
456 everything following the leftmost colon and any module information in the
457 exception name is ignored.
Georg Brandl116aa622007-08-15 14:28:22 +0000458
459* The interactive shell omits the traceback header line for some
460 :exc:`SyntaxError`\ s. But doctest uses the traceback header line to
461 distinguish exceptions from non-exceptions. So in the rare case where you need
462 to test a :exc:`SyntaxError` that omits the traceback header, you will need to
463 manually add the traceback header line to your test example.
464
465* For some :exc:`SyntaxError`\ s, Python displays the character position of the
466 syntax error, using a ``^`` marker::
467
468 >>> 1 1
469 File "<stdin>", line 1
470 1 1
471 ^
472 SyntaxError: invalid syntax
473
474 Since the lines showing the position of the error come before the exception type
475 and detail, they are not checked by doctest. For example, the following test
476 would pass, even though it puts the ``^`` marker in the wrong location::
477
478 >>> 1 1
479 Traceback (most recent call last):
480 File "<stdin>", line 1
481 1 1
482 ^
483 SyntaxError: invalid syntax
484
Georg Brandl116aa622007-08-15 14:28:22 +0000485
486.. _doctest-options:
487
488Option Flags and Directives
489^^^^^^^^^^^^^^^^^^^^^^^^^^^
490
491A number of option flags control various aspects of doctest's behavior.
492Symbolic names for the flags are supplied as module constants, which can be
493or'ed together and passed to various functions. The names can also be used in
494doctest directives (see below).
495
496The first group of options define test semantics, controlling aspects of how
497doctest decides whether actual output matches an example's expected output:
498
499
500.. data:: DONT_ACCEPT_TRUE_FOR_1
501
502 By default, if an expected output block contains just ``1``, an actual output
503 block containing just ``1`` or just ``True`` is considered to be a match, and
504 similarly for ``0`` versus ``False``. When :const:`DONT_ACCEPT_TRUE_FOR_1` is
505 specified, neither substitution is allowed. The default behavior caters to that
506 Python changed the return type of many functions from integer to boolean;
507 doctests expecting "little integer" output still work in these cases. This
508 option will probably go away, but not for several years.
509
510
511.. data:: DONT_ACCEPT_BLANKLINE
512
513 By default, if an expected output block contains a line containing only the
514 string ``<BLANKLINE>``, then that line will match a blank line in the actual
515 output. Because a genuinely blank line delimits the expected output, this is
516 the only way to communicate that a blank line is expected. When
517 :const:`DONT_ACCEPT_BLANKLINE` is specified, this substitution is not allowed.
518
519
520.. data:: NORMALIZE_WHITESPACE
521
522 When specified, all sequences of whitespace (blanks and newlines) are treated as
523 equal. Any sequence of whitespace within the expected output will match any
524 sequence of whitespace within the actual output. By default, whitespace must
525 match exactly. :const:`NORMALIZE_WHITESPACE` is especially useful when a line of
526 expected output is very long, and you want to wrap it across multiple lines in
527 your source.
528
529
530.. data:: ELLIPSIS
531
532 When specified, an ellipsis marker (``...``) in the expected output can match
533 any substring in the actual output. This includes substrings that span line
534 boundaries, and empty substrings, so it's best to keep usage of this simple.
535 Complicated uses can lead to the same kinds of "oops, it matched too much!"
536 surprises that ``.*`` is prone to in regular expressions.
537
538
539.. data:: IGNORE_EXCEPTION_DETAIL
540
541 When specified, an example that expects an exception passes if an exception of
542 the expected type is raised, even if the exception detail does not match. For
543 example, an example expecting ``ValueError: 42`` will pass if the actual
544 exception raised is ``ValueError: 3*14``, but will fail, e.g., if
545 :exc:`TypeError` is raised.
546
Nick Coghlan5e76e942010-06-12 13:42:46 +0000547 It will also ignore the module name used in Python 3 doctest reports. Hence
548 both these variations will work regardless of whether the test is run under
549 Python 2.7 or Python 3.2 (or later versions):
550
551 >>> raise CustomError('message') #doctest: +IGNORE_EXCEPTION_DETAIL
552 Traceback (most recent call last):
553 CustomError: message
554
555 >>> raise CustomError('message') #doctest: +IGNORE_EXCEPTION_DETAIL
556 Traceback (most recent call last):
557 my_module.CustomError: message
558
559 Note that :const:`ELLIPSIS` can also be used to ignore the
560 details of the exception message, but such a test may still fail based
561 on whether or not the module details are printed as part of the
562 exception name. Using :const:`IGNORE_EXCEPTION_DETAIL` and the details
563 from Python 2.3 is also the only clear way to write a doctest that doesn't
564 care about the exception detail yet continues to pass under Python 2.3 or
565 earlier (those releases do not support doctest directives and ignore them
566 as irrelevant comments). For example, ::
Georg Brandl116aa622007-08-15 14:28:22 +0000567
568 >>> (1, 2)[3] = 'moo' #doctest: +IGNORE_EXCEPTION_DETAIL
569 Traceback (most recent call last):
570 File "<stdin>", line 1, in ?
571 TypeError: object doesn't support item assignment
572
Nick Coghlan5e76e942010-06-12 13:42:46 +0000573 passes under Python 2.3 and later Python versions, even though the detail
574 changed in Python 2.4 to say "does not" instead of "doesn't".
575
576 .. versionchanged:: 3.2
Georg Brandl67b21b72010-08-17 15:07:14 +0000577 :const:`IGNORE_EXCEPTION_DETAIL` now also ignores any information relating
578 to the module containing the exception under test.
Georg Brandl116aa622007-08-15 14:28:22 +0000579
580
581.. data:: SKIP
582
583 When specified, do not run the example at all. This can be useful in contexts
584 where doctest examples serve as both documentation and test cases, and an
585 example should be included for documentation purposes, but should not be
586 checked. E.g., the example's output might be random; or the example might
587 depend on resources which would be unavailable to the test driver.
588
589 The SKIP flag can also be used for temporarily "commenting out" examples.
590
591
592.. data:: COMPARISON_FLAGS
593
594 A bitmask or'ing together all the comparison flags above.
595
596The second group of options controls how test failures are reported:
597
598
599.. data:: REPORT_UDIFF
600
601 When specified, failures that involve multi-line expected and actual outputs are
602 displayed using a unified diff.
603
604
605.. data:: REPORT_CDIFF
606
607 When specified, failures that involve multi-line expected and actual outputs
608 will be displayed using a context diff.
609
610
611.. data:: REPORT_NDIFF
612
613 When specified, differences are computed by ``difflib.Differ``, using the same
614 algorithm as the popular :file:`ndiff.py` utility. This is the only method that
615 marks differences within lines as well as across lines. For example, if a line
616 of expected output contains digit ``1`` where actual output contains letter
617 ``l``, a line is inserted with a caret marking the mismatching column positions.
618
619
620.. data:: REPORT_ONLY_FIRST_FAILURE
621
622 When specified, display the first failing example in each doctest, but suppress
623 output for all remaining examples. This will prevent doctest from reporting
624 correct examples that break because of earlier failures; but it might also hide
625 incorrect examples that fail independently of the first failure. When
626 :const:`REPORT_ONLY_FIRST_FAILURE` is specified, the remaining examples are
627 still run, and still count towards the total number of failures reported; only
628 the output is suppressed.
629
630
631.. data:: REPORTING_FLAGS
632
633 A bitmask or'ing together all the reporting flags above.
634
635"Doctest directives" may be used to modify the option flags for individual
636examples. Doctest directives are expressed as a special Python comment
637following an example's source code:
638
639.. productionlist:: doctest
640 directive: "#" "doctest:" `directive_options`
641 directive_options: `directive_option` ("," `directive_option`)\*
642 directive_option: `on_or_off` `directive_option_name`
643 on_or_off: "+" \| "-"
644 directive_option_name: "DONT_ACCEPT_BLANKLINE" \| "NORMALIZE_WHITESPACE" \| ...
645
646Whitespace is not allowed between the ``+`` or ``-`` and the directive option
647name. The directive option name can be any of the option flag names explained
648above.
649
650An example's doctest directives modify doctest's behavior for that single
651example. Use ``+`` to enable the named behavior, or ``-`` to disable it.
652
653For example, this test passes::
654
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000655 >>> print(list(range(20))) #doctest: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000656 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9,
657 10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
658
659Without the directive it would fail, both because the actual output doesn't have
660two blanks before the single-digit list elements, and because the actual output
661is on a single line. This test also passes, and also requires a directive to do
662so::
663
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000664 >>> print(list(range(20))) # doctest: +ELLIPSIS
Georg Brandl116aa622007-08-15 14:28:22 +0000665 [0, 1, ..., 18, 19]
666
667Multiple directives can be used on a single physical line, separated by commas::
668
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000669 >>> print(list(range(20))) # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000670 [0, 1, ..., 18, 19]
671
672If multiple directive comments are used for a single example, then they are
673combined::
674
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000675 >>> print(list(range(20))) # doctest: +ELLIPSIS
676 ... # doctest: +NORMALIZE_WHITESPACE
Georg Brandl116aa622007-08-15 14:28:22 +0000677 [0, 1, ..., 18, 19]
678
679As the previous example shows, you can add ``...`` lines to your example
680containing only directives. This can be useful when an example is too long for
681a directive to comfortably fit on the same line::
682
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000683 >>> print(list(range(5)) + list(range(10, 20)) + list(range(30, 40)))
Georg Brandl116aa622007-08-15 14:28:22 +0000684 ... # doctest: +ELLIPSIS
Georg Brandl8f80a5b2010-03-21 09:25:54 +0000685 [0, ..., 4, 10, ..., 19, 30, ..., 39]
Georg Brandl116aa622007-08-15 14:28:22 +0000686
687Note that since all options are disabled by default, and directives apply only
688to the example they appear in, enabling options (via ``+`` in a directive) is
689usually the only meaningful choice. However, option flags can also be passed to
690functions that run doctests, establishing different defaults. In such cases,
691disabling an option via ``-`` in a directive can be useful.
692
Georg Brandl116aa622007-08-15 14:28:22 +0000693There's also a way to register new option flag names, although this isn't useful
694unless you intend to extend :mod:`doctest` internals via subclassing:
695
696
697.. function:: register_optionflag(name)
698
699 Create a new option flag with a given name, and return the new flag's integer
700 value. :func:`register_optionflag` can be used when subclassing
701 :class:`OutputChecker` or :class:`DocTestRunner` to create new options that are
702 supported by your subclasses. :func:`register_optionflag` should always be
703 called using the following idiom::
704
705 MY_FLAG = register_optionflag('MY_FLAG')
706
Georg Brandl116aa622007-08-15 14:28:22 +0000707
708.. _doctest-warnings:
709
710Warnings
711^^^^^^^^
712
713:mod:`doctest` is serious about requiring exact matches in expected output. If
714even a single character doesn't match, the test fails. This will probably
715surprise you a few times, as you learn exactly what Python does and doesn't
716guarantee about output. For example, when printing a dict, Python doesn't
717guarantee that the key-value pairs will be printed in any particular order, so a
Christian Heimes5b5e81c2007-12-31 16:14:33 +0000718test like ::
Georg Brandl116aa622007-08-15 14:28:22 +0000719
720 >>> foo()
721 {"Hermione": "hippogryph", "Harry": "broomstick"}
722
723is vulnerable! One workaround is to do ::
724
725 >>> foo() == {"Hermione": "hippogryph", "Harry": "broomstick"}
726 True
727
728instead. Another is to do ::
729
Ezio Melotti8f7649e2009-09-13 04:48:45 +0000730 >>> d = sorted(foo().items())
Georg Brandl116aa622007-08-15 14:28:22 +0000731 >>> d
732 [('Harry', 'broomstick'), ('Hermione', 'hippogryph')]
733
734There are others, but you get the idea.
735
736Another bad idea is to print things that embed an object address, like ::
737
738 >>> id(1.0) # certain to fail some of the time
739 7948648
740 >>> class C: pass
741 >>> C() # the default repr() for instances embeds an address
742 <__main__.C instance at 0x00AC18F0>
743
744The :const:`ELLIPSIS` directive gives a nice approach for the last example::
745
746 >>> C() #doctest: +ELLIPSIS
747 <__main__.C instance at 0x...>
748
749Floating-point numbers are also subject to small output variations across
750platforms, because Python defers to the platform C library for float formatting,
751and C libraries vary widely in quality here. ::
752
753 >>> 1./7 # risky
754 0.14285714285714285
Georg Brandl6911e3c2007-09-04 07:15:32 +0000755 >>> print(1./7) # safer
Georg Brandl116aa622007-08-15 14:28:22 +0000756 0.142857142857
Georg Brandl6911e3c2007-09-04 07:15:32 +0000757 >>> print(round(1./7, 6)) # much safer
Georg Brandl116aa622007-08-15 14:28:22 +0000758 0.142857
759
760Numbers of the form ``I/2.**J`` are safe across all platforms, and I often
761contrive doctest examples to produce numbers of that form::
762
763 >>> 3./4 # utterly safe
764 0.75
765
766Simple fractions are also easier for people to understand, and that makes for
767better documentation.
768
769
770.. _doctest-basic-api:
771
772Basic API
773---------
774
775The functions :func:`testmod` and :func:`testfile` provide a simple interface to
776doctest that should be sufficient for most basic uses. For a less formal
777introduction to these two functions, see sections :ref:`doctest-simple-testmod`
778and :ref:`doctest-simple-testfile`.
779
780
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000781.. 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 +0000782
783 All arguments except *filename* are optional, and should be specified in keyword
784 form.
785
786 Test examples in the file named *filename*. Return ``(failure_count,
787 test_count)``.
788
789 Optional argument *module_relative* specifies how the filename should be
790 interpreted:
791
792 * If *module_relative* is ``True`` (the default), then *filename* specifies an
793 OS-independent module-relative path. By default, this path is relative to the
794 calling module's directory; but if the *package* argument is specified, then it
795 is relative to that package. To ensure OS-independence, *filename* should use
796 ``/`` characters to separate path segments, and may not be an absolute path
797 (i.e., it may not begin with ``/``).
798
799 * If *module_relative* is ``False``, then *filename* specifies an OS-specific
800 path. The path may be absolute or relative; relative paths are resolved with
801 respect to the current working directory.
802
803 Optional argument *name* gives the name of the test; by default, or if ``None``,
804 ``os.path.basename(filename)`` is used.
805
806 Optional argument *package* is a Python package or the name of a Python package
807 whose directory should be used as the base directory for a module-relative
808 filename. If no package is specified, then the calling module's directory is
809 used as the base directory for module-relative filenames. It is an error to
810 specify *package* if *module_relative* is ``False``.
811
812 Optional argument *globs* gives a dict to be used as the globals when executing
813 examples. A new shallow copy of this dict is created for the doctest, so its
814 examples start with a clean slate. By default, or if ``None``, a new empty dict
815 is used.
816
817 Optional argument *extraglobs* gives a dict merged into the globals used to
818 execute examples. This works like :meth:`dict.update`: if *globs* and
819 *extraglobs* have a common key, the associated value in *extraglobs* appears in
820 the combined dict. By default, or if ``None``, no extra globals are used. This
821 is an advanced feature that allows parameterization of doctests. For example, a
822 doctest can be written for a base class, using a generic name for the class,
823 then reused to test any number of subclasses by passing an *extraglobs* dict
824 mapping the generic name to the subclass to be tested.
825
826 Optional argument *verbose* prints lots of stuff if true, and prints only
827 failures if false; by default, or if ``None``, it's true if and only if ``'-v'``
828 is in ``sys.argv``.
829
830 Optional argument *report* prints a summary at the end when true, else prints
831 nothing at the end. In verbose mode, the summary is detailed, else the summary
832 is very brief (in fact, empty if all tests passed).
833
834 Optional argument *optionflags* or's together option flags. See section
835 :ref:`doctest-options`.
836
837 Optional argument *raise_on_error* defaults to false. If true, an exception is
838 raised upon the first failure or unexpected exception in an example. This
839 allows failures to be post-mortem debugged. Default behavior is to continue
840 running examples.
841
842 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass) that
843 should be used to extract tests from the files. It defaults to a normal parser
844 (i.e., ``DocTestParser()``).
845
846 Optional argument *encoding* specifies an encoding that should be used to
847 convert the file to unicode.
848
Georg Brandl116aa622007-08-15 14:28:22 +0000849
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000850.. 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 +0000851
852 All arguments are optional, and all except for *m* should be specified in
853 keyword form.
854
855 Test examples in docstrings in functions and classes reachable from module *m*
856 (or module :mod:`__main__` if *m* is not supplied or is ``None``), starting with
857 ``m.__doc__``.
858
859 Also test examples reachable from dict ``m.__test__``, if it exists and is not
860 ``None``. ``m.__test__`` maps names (strings) to functions, classes and
861 strings; function and class docstrings are searched for examples; strings are
862 searched directly, as if they were docstrings.
863
864 Only docstrings attached to objects belonging to module *m* are searched.
865
866 Return ``(failure_count, test_count)``.
867
868 Optional argument *name* gives the name of the module; by default, or if
869 ``None``, ``m.__name__`` is used.
870
871 Optional argument *exclude_empty* defaults to false. If true, objects for which
872 no doctests are found are excluded from consideration. The default is a backward
873 compatibility hack, so that code still using :meth:`doctest.master.summarize` in
874 conjunction with :func:`testmod` continues to get output for objects with no
875 tests. The *exclude_empty* argument to the newer :class:`DocTestFinder`
876 constructor defaults to true.
877
878 Optional arguments *extraglobs*, *verbose*, *report*, *optionflags*,
879 *raise_on_error*, and *globs* are the same as for function :func:`testfile`
880 above, except that *globs* defaults to ``m.__dict__``.
881
Georg Brandl116aa622007-08-15 14:28:22 +0000882
883There's also a function to run the doctests associated with a single object.
884This function is provided for backward compatibility. There are no plans to
885deprecate it, but it's rarely useful:
886
887
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000888.. function:: run_docstring_examples(f, globs, verbose=False, name="NoName", compileflags=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +0000889
890 Test examples associated with object *f*; for example, *f* may be a module,
891 function, or class object.
892
893 A shallow copy of dictionary argument *globs* is used for the execution context.
894
895 Optional argument *name* is used in failure messages, and defaults to
896 ``"NoName"``.
897
898 If optional argument *verbose* is true, output is generated even if there are no
899 failures. By default, output is generated only in case of an example failure.
900
901 Optional argument *compileflags* gives the set of flags that should be used by
902 the Python compiler when running the examples. By default, or if ``None``,
903 flags are deduced corresponding to the set of future features found in *globs*.
904
905 Optional argument *optionflags* works as for function :func:`testfile` above.
906
907
908.. _doctest-unittest-api:
909
910Unittest API
911------------
912
913As your collection of doctest'ed modules grows, you'll want a way to run all
Georg Brandl31835852008-05-12 17:38:56 +0000914their doctests systematically. :mod:`doctest` provides two functions that can
915be used to create :mod:`unittest` test suites from modules and text files
Georg Brandla8514832010-07-10 12:20:38 +0000916containing doctests. To integrate with :mod:`unittest` test discovery, include
917a :func:`load_tests` function in your test module::
Georg Brandl116aa622007-08-15 14:28:22 +0000918
919 import unittest
920 import doctest
Georg Brandla8514832010-07-10 12:20:38 +0000921 import my_module_with_doctests
Georg Brandl116aa622007-08-15 14:28:22 +0000922
Georg Brandla8514832010-07-10 12:20:38 +0000923 def load_tests(loader, tests, ignore):
924 tests.addTests(doctest.DocTestSuite(my_module_with_doctests))
R. David Murray796343b2010-12-13 22:50:30 +0000925 return tests
Georg Brandl116aa622007-08-15 14:28:22 +0000926
927There are two main functions for creating :class:`unittest.TestSuite` instances
928from text files and modules with doctests:
929
930
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000931.. 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 +0000932
933 Convert doctest tests from one or more text files to a
934 :class:`unittest.TestSuite`.
935
936 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
937 and runs the interactive examples in each file. If an example in any file
938 fails, then the synthesized unit test fails, and a :exc:`failureException`
939 exception is raised showing the name of the file containing the test and a
940 (sometimes approximate) line number.
941
942 Pass one or more paths (as strings) to text files to be examined.
943
944 Options may be provided as keyword arguments:
945
946 Optional argument *module_relative* specifies how the filenames in *paths*
947 should be interpreted:
948
Benjamin Petersond23f8222009-04-05 19:13:16 +0000949 * If *module_relative* is ``True`` (the default), then each filename in
950 *paths* specifies an OS-independent module-relative path. By default, this
951 path is relative to the calling module's directory; but if the *package*
952 argument is specified, then it is relative to that package. To ensure
953 OS-independence, each filename should use ``/`` characters to separate path
954 segments, and may not be an absolute path (i.e., it may not begin with
955 ``/``).
Georg Brandl116aa622007-08-15 14:28:22 +0000956
Benjamin Petersond23f8222009-04-05 19:13:16 +0000957 * If *module_relative* is ``False``, then each filename in *paths* specifies
958 an OS-specific path. The path may be absolute or relative; relative paths
959 are resolved with respect to the current working directory.
Georg Brandl116aa622007-08-15 14:28:22 +0000960
Benjamin Petersond23f8222009-04-05 19:13:16 +0000961 Optional argument *package* is a Python package or the name of a Python
962 package whose directory should be used as the base directory for
963 module-relative filenames in *paths*. If no package is specified, then the
964 calling module's directory is used as the base directory for module-relative
965 filenames. It is an error to specify *package* if *module_relative* is
966 ``False``.
Georg Brandl116aa622007-08-15 14:28:22 +0000967
Benjamin Petersond23f8222009-04-05 19:13:16 +0000968 Optional argument *setUp* specifies a set-up function for the test suite.
969 This is called before running the tests in each file. The *setUp* function
Georg Brandl116aa622007-08-15 14:28:22 +0000970 will be passed a :class:`DocTest` object. The setUp function can access the
971 test globals as the *globs* attribute of the test passed.
972
Benjamin Petersond23f8222009-04-05 19:13:16 +0000973 Optional argument *tearDown* specifies a tear-down function for the test
974 suite. This is called after running the tests in each file. The *tearDown*
975 function will be passed a :class:`DocTest` object. The setUp function can
976 access the test globals as the *globs* attribute of the test passed.
977
Georg Brandl116aa622007-08-15 14:28:22 +0000978 Optional argument *globs* is a dictionary containing the initial global
979 variables for the tests. A new copy of this dictionary is created for each
980 test. By default, *globs* is a new empty dictionary.
981
982 Optional argument *optionflags* specifies the default doctest options for the
983 tests, created by or-ing together individual option flags. See section
Benjamin Petersond23f8222009-04-05 19:13:16 +0000984 :ref:`doctest-options`. See function :func:`set_unittest_reportflags` below
985 for a better way to set reporting options.
Georg Brandl116aa622007-08-15 14:28:22 +0000986
Benjamin Petersond23f8222009-04-05 19:13:16 +0000987 Optional argument *parser* specifies a :class:`DocTestParser` (or subclass)
988 that should be used to extract tests from the files. It defaults to a normal
989 parser (i.e., ``DocTestParser()``).
Georg Brandl116aa622007-08-15 14:28:22 +0000990
991 Optional argument *encoding* specifies an encoding that should be used to
992 convert the file to unicode.
993
Georg Brandl55ac8f02007-09-01 13:51:09 +0000994 The global ``__file__`` is added to the globals provided to doctests loaded
995 from a text file using :func:`DocFileSuite`.
Georg Brandl116aa622007-08-15 14:28:22 +0000996
997
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000998.. function:: DocTestSuite(module=None, globs=None, extraglobs=None, test_finder=None, setUp=None, tearDown=None, checker=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000999
1000 Convert doctest tests for a module to a :class:`unittest.TestSuite`.
1001
1002 The returned :class:`unittest.TestSuite` is to be run by the unittest framework
1003 and runs each doctest in the module. If any of the doctests fail, then the
1004 synthesized unit test fails, and a :exc:`failureException` exception is raised
1005 showing the name of the file containing the test and a (sometimes approximate)
1006 line number.
1007
1008 Optional argument *module* provides the module to be tested. It can be a module
1009 object or a (possibly dotted) module name. If not specified, the module calling
1010 this function is used.
1011
1012 Optional argument *globs* is a dictionary containing the initial global
1013 variables for the tests. A new copy of this dictionary is created for each
1014 test. By default, *globs* is a new empty dictionary.
1015
1016 Optional argument *extraglobs* specifies an extra set of global variables, which
1017 is merged into *globs*. By default, no extra globals are used.
1018
1019 Optional argument *test_finder* is the :class:`DocTestFinder` object (or a
1020 drop-in replacement) that is used to extract doctests from the module.
1021
1022 Optional arguments *setUp*, *tearDown*, and *optionflags* are the same as for
1023 function :func:`DocFileSuite` above.
1024
Georg Brandl55ac8f02007-09-01 13:51:09 +00001025 This function uses the same search technique as :func:`testmod`.
Georg Brandl116aa622007-08-15 14:28:22 +00001026
Georg Brandl116aa622007-08-15 14:28:22 +00001027
1028Under the covers, :func:`DocTestSuite` creates a :class:`unittest.TestSuite` out
1029of :class:`doctest.DocTestCase` instances, and :class:`DocTestCase` is a
1030subclass of :class:`unittest.TestCase`. :class:`DocTestCase` isn't documented
1031here (it's an internal detail), but studying its code can answer questions about
1032the exact details of :mod:`unittest` integration.
1033
1034Similarly, :func:`DocFileSuite` creates a :class:`unittest.TestSuite` out of
1035:class:`doctest.DocFileCase` instances, and :class:`DocFileCase` is a subclass
1036of :class:`DocTestCase`.
1037
1038So both ways of creating a :class:`unittest.TestSuite` run instances of
1039:class:`DocTestCase`. This is important for a subtle reason: when you run
1040:mod:`doctest` functions yourself, you can control the :mod:`doctest` options in
1041use directly, by passing option flags to :mod:`doctest` functions. However, if
1042you're writing a :mod:`unittest` framework, :mod:`unittest` ultimately controls
1043when and how tests get run. The framework author typically wants to control
1044:mod:`doctest` reporting options (perhaps, e.g., specified by command line
1045options), but there's no way to pass options through :mod:`unittest` to
1046:mod:`doctest` test runners.
1047
1048For this reason, :mod:`doctest` also supports a notion of :mod:`doctest`
1049reporting flags specific to :mod:`unittest` support, via this function:
1050
1051
1052.. function:: set_unittest_reportflags(flags)
1053
1054 Set the :mod:`doctest` reporting flags to use.
1055
1056 Argument *flags* or's together option flags. See section
1057 :ref:`doctest-options`. Only "reporting flags" can be used.
1058
1059 This is a module-global setting, and affects all future doctests run by module
1060 :mod:`unittest`: the :meth:`runTest` method of :class:`DocTestCase` looks at
1061 the option flags specified for the test case when the :class:`DocTestCase`
1062 instance was constructed. If no reporting flags were specified (which is the
1063 typical and expected case), :mod:`doctest`'s :mod:`unittest` reporting flags are
1064 or'ed into the option flags, and the option flags so augmented are passed to the
1065 :class:`DocTestRunner` instance created to run the doctest. If any reporting
1066 flags were specified when the :class:`DocTestCase` instance was constructed,
1067 :mod:`doctest`'s :mod:`unittest` reporting flags are ignored.
1068
1069 The value of the :mod:`unittest` reporting flags in effect before the function
1070 was called is returned by the function.
1071
Georg Brandl116aa622007-08-15 14:28:22 +00001072
1073.. _doctest-advanced-api:
1074
1075Advanced API
1076------------
1077
1078The basic API is a simple wrapper that's intended to make doctest easy to use.
1079It is fairly flexible, and should meet most users' needs; however, if you
1080require more fine-grained control over testing, or wish to extend doctest's
1081capabilities, then you should use the advanced API.
1082
1083The advanced API revolves around two container classes, which are used to store
1084the interactive examples extracted from doctest cases:
1085
Ezio Melotti0639d5a2009-12-19 23:26:38 +00001086* :class:`Example`: A single Python :term:`statement`, paired with its expected
Christian Heimesd8654cf2007-12-02 15:22:16 +00001087 output.
Georg Brandl116aa622007-08-15 14:28:22 +00001088
1089* :class:`DocTest`: A collection of :class:`Example`\ s, typically extracted
1090 from a single docstring or text file.
1091
1092Additional processing classes are defined to find, parse, and run, and check
1093doctest examples:
1094
1095* :class:`DocTestFinder`: Finds all docstrings in a given module, and uses a
1096 :class:`DocTestParser` to create a :class:`DocTest` from every docstring that
1097 contains interactive examples.
1098
1099* :class:`DocTestParser`: Creates a :class:`DocTest` object from a string (such
1100 as an object's docstring).
1101
1102* :class:`DocTestRunner`: Executes the examples in a :class:`DocTest`, and uses
1103 an :class:`OutputChecker` to verify their output.
1104
1105* :class:`OutputChecker`: Compares the actual output from a doctest example with
1106 the expected output, and decides whether they match.
1107
1108The relationships among these processing classes are summarized in the following
1109diagram::
1110
1111 list of:
1112 +------+ +---------+
1113 |module| --DocTestFinder-> | DocTest | --DocTestRunner-> results
1114 +------+ | ^ +---------+ | ^ (printed)
1115 | | | Example | | |
1116 v | | ... | v |
1117 DocTestParser | Example | OutputChecker
1118 +---------+
1119
1120
1121.. _doctest-doctest:
1122
1123DocTest Objects
1124^^^^^^^^^^^^^^^
1125
1126
1127.. class:: DocTest(examples, globs, name, filename, lineno, docstring)
1128
1129 A collection of doctest examples that should be run in a single namespace. The
Senthil Kumarana6bac952011-07-04 11:28:30 -07001130 constructor arguments are used to initialize the attributes of the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001131
Georg Brandl116aa622007-08-15 14:28:22 +00001132
Senthil Kumarana6bac952011-07-04 11:28:30 -07001133 :class:`DocTest` defines the following attributes. They are initialized by
Benjamin Petersone41251e2008-04-25 01:59:09 +00001134 the constructor, and should not be modified directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001135
1136
Benjamin Petersone41251e2008-04-25 01:59:09 +00001137 .. attribute:: examples
Georg Brandl116aa622007-08-15 14:28:22 +00001138
Benjamin Petersone41251e2008-04-25 01:59:09 +00001139 A list of :class:`Example` objects encoding the individual interactive Python
1140 examples that should be run by this test.
Georg Brandl116aa622007-08-15 14:28:22 +00001141
1142
Benjamin Petersone41251e2008-04-25 01:59:09 +00001143 .. attribute:: globs
Georg Brandl116aa622007-08-15 14:28:22 +00001144
Benjamin Petersone41251e2008-04-25 01:59:09 +00001145 The namespace (aka globals) that the examples should be run in. This is a
1146 dictionary mapping names to values. Any changes to the namespace made by the
1147 examples (such as binding new variables) will be reflected in :attr:`globs`
1148 after the test is run.
Georg Brandl116aa622007-08-15 14:28:22 +00001149
1150
Benjamin Petersone41251e2008-04-25 01:59:09 +00001151 .. attribute:: name
Georg Brandl116aa622007-08-15 14:28:22 +00001152
Benjamin Petersone41251e2008-04-25 01:59:09 +00001153 A string name identifying the :class:`DocTest`. Typically, this is the name
1154 of the object or file that the test was extracted from.
Georg Brandl116aa622007-08-15 14:28:22 +00001155
1156
Benjamin Petersone41251e2008-04-25 01:59:09 +00001157 .. attribute:: filename
Georg Brandl116aa622007-08-15 14:28:22 +00001158
Benjamin Petersone41251e2008-04-25 01:59:09 +00001159 The name of the file that this :class:`DocTest` was extracted from; or
1160 ``None`` if the filename is unknown, or if the :class:`DocTest` was not
1161 extracted from a file.
Georg Brandl116aa622007-08-15 14:28:22 +00001162
1163
Benjamin Petersone41251e2008-04-25 01:59:09 +00001164 .. attribute:: lineno
Georg Brandl116aa622007-08-15 14:28:22 +00001165
Benjamin Petersone41251e2008-04-25 01:59:09 +00001166 The line number within :attr:`filename` where this :class:`DocTest` begins, or
1167 ``None`` if the line number is unavailable. This line number is zero-based
1168 with respect to the beginning of the file.
Georg Brandl116aa622007-08-15 14:28:22 +00001169
1170
Benjamin Petersone41251e2008-04-25 01:59:09 +00001171 .. attribute:: docstring
Georg Brandl116aa622007-08-15 14:28:22 +00001172
Benjamin Petersone41251e2008-04-25 01:59:09 +00001173 The string that the test was extracted from, or 'None' if the string is
1174 unavailable, or if the test was not extracted from a string.
Georg Brandl116aa622007-08-15 14:28:22 +00001175
1176
1177.. _doctest-example:
1178
1179Example Objects
1180^^^^^^^^^^^^^^^
1181
1182
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001183.. class:: Example(source, want, exc_msg=None, lineno=0, indent=0, options=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001184
1185 A single interactive example, consisting of a Python statement and its expected
Senthil Kumarana6bac952011-07-04 11:28:30 -07001186 output. The constructor arguments are used to initialize the attributes of
1187 the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001188
Georg Brandl116aa622007-08-15 14:28:22 +00001189
Senthil Kumarana6bac952011-07-04 11:28:30 -07001190 :class:`Example` defines the following attributes. They are initialized by
Benjamin Petersone41251e2008-04-25 01:59:09 +00001191 the constructor, and should not be modified directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001192
1193
Benjamin Petersone41251e2008-04-25 01:59:09 +00001194 .. attribute:: source
Georg Brandl116aa622007-08-15 14:28:22 +00001195
Benjamin Petersone41251e2008-04-25 01:59:09 +00001196 A string containing the example's source code. This source code consists of a
1197 single Python statement, and always ends with a newline; the constructor adds
1198 a newline when necessary.
Georg Brandl116aa622007-08-15 14:28:22 +00001199
1200
Benjamin Petersone41251e2008-04-25 01:59:09 +00001201 .. attribute:: want
Georg Brandl116aa622007-08-15 14:28:22 +00001202
Benjamin Petersone41251e2008-04-25 01:59:09 +00001203 The expected output from running the example's source code (either from
1204 stdout, or a traceback in case of exception). :attr:`want` ends with a
1205 newline unless no output is expected, in which case it's an empty string. The
1206 constructor adds a newline when necessary.
Georg Brandl116aa622007-08-15 14:28:22 +00001207
1208
Benjamin Petersone41251e2008-04-25 01:59:09 +00001209 .. attribute:: exc_msg
Georg Brandl116aa622007-08-15 14:28:22 +00001210
Benjamin Petersone41251e2008-04-25 01:59:09 +00001211 The exception message generated by the example, if the example is expected to
1212 generate an exception; or ``None`` if it is not expected to generate an
1213 exception. This exception message is compared against the return value of
1214 :func:`traceback.format_exception_only`. :attr:`exc_msg` ends with a newline
1215 unless it's ``None``. The constructor adds a newline if needed.
Georg Brandl116aa622007-08-15 14:28:22 +00001216
1217
Benjamin Petersone41251e2008-04-25 01:59:09 +00001218 .. attribute:: lineno
Georg Brandl116aa622007-08-15 14:28:22 +00001219
Benjamin Petersone41251e2008-04-25 01:59:09 +00001220 The line number within the string containing this example where the example
1221 begins. This line number is zero-based with respect to the beginning of the
1222 containing string.
Georg Brandl116aa622007-08-15 14:28:22 +00001223
1224
Benjamin Petersone41251e2008-04-25 01:59:09 +00001225 .. attribute:: indent
Georg Brandl116aa622007-08-15 14:28:22 +00001226
Benjamin Petersone41251e2008-04-25 01:59:09 +00001227 The example's indentation in the containing string, i.e., the number of space
1228 characters that precede the example's first prompt.
Georg Brandl116aa622007-08-15 14:28:22 +00001229
1230
Benjamin Petersone41251e2008-04-25 01:59:09 +00001231 .. attribute:: options
Georg Brandl116aa622007-08-15 14:28:22 +00001232
Benjamin Petersone41251e2008-04-25 01:59:09 +00001233 A dictionary mapping from option flags to ``True`` or ``False``, which is used
1234 to override default options for this example. Any option flags not contained
1235 in this dictionary are left at their default value (as specified by the
1236 :class:`DocTestRunner`'s :attr:`optionflags`). By default, no options are set.
Georg Brandl116aa622007-08-15 14:28:22 +00001237
1238
1239.. _doctest-doctestfinder:
1240
1241DocTestFinder objects
1242^^^^^^^^^^^^^^^^^^^^^
1243
1244
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001245.. class:: DocTestFinder(verbose=False, parser=DocTestParser(), recurse=True, exclude_empty=True)
Georg Brandl116aa622007-08-15 14:28:22 +00001246
1247 A processing class used to extract the :class:`DocTest`\ s that are relevant to
1248 a given object, from its docstring and the docstrings of its contained objects.
1249 :class:`DocTest`\ s can currently be extracted from the following object types:
1250 modules, functions, classes, methods, staticmethods, classmethods, and
1251 properties.
1252
1253 The optional argument *verbose* can be used to display the objects searched by
1254 the finder. It defaults to ``False`` (no output).
1255
1256 The optional argument *parser* specifies the :class:`DocTestParser` object (or a
1257 drop-in replacement) that is used to extract doctests from docstrings.
1258
1259 If the optional argument *recurse* is false, then :meth:`DocTestFinder.find`
1260 will only examine the given object, and not any contained objects.
1261
1262 If the optional argument *exclude_empty* is false, then
1263 :meth:`DocTestFinder.find` will include tests for objects with empty docstrings.
1264
Georg Brandl116aa622007-08-15 14:28:22 +00001265
Benjamin Petersone41251e2008-04-25 01:59:09 +00001266 :class:`DocTestFinder` defines the following method:
Georg Brandl116aa622007-08-15 14:28:22 +00001267
1268
Benjamin Petersone41251e2008-04-25 01:59:09 +00001269 .. method:: find(obj[, name][, module][, globs][, extraglobs])
Georg Brandl116aa622007-08-15 14:28:22 +00001270
Benjamin Petersone41251e2008-04-25 01:59:09 +00001271 Return a list of the :class:`DocTest`\ s that are defined by *obj*'s
1272 docstring, or by any of its contained objects' docstrings.
Georg Brandl116aa622007-08-15 14:28:22 +00001273
Benjamin Petersone41251e2008-04-25 01:59:09 +00001274 The optional argument *name* specifies the object's name; this name will be
1275 used to construct names for the returned :class:`DocTest`\ s. If *name* is
1276 not specified, then ``obj.__name__`` is used.
Georg Brandl116aa622007-08-15 14:28:22 +00001277
Benjamin Petersone41251e2008-04-25 01:59:09 +00001278 The optional parameter *module* is the module that contains the given object.
1279 If the module is not specified or is None, then the test finder will attempt
1280 to automatically determine the correct module. The object's module is used:
Georg Brandl116aa622007-08-15 14:28:22 +00001281
Benjamin Petersone41251e2008-04-25 01:59:09 +00001282 * As a default namespace, if *globs* is not specified.
Georg Brandl116aa622007-08-15 14:28:22 +00001283
Benjamin Petersone41251e2008-04-25 01:59:09 +00001284 * To prevent the DocTestFinder from extracting DocTests from objects that are
1285 imported from other modules. (Contained objects with modules other than
1286 *module* are ignored.)
Georg Brandl116aa622007-08-15 14:28:22 +00001287
Benjamin Petersone41251e2008-04-25 01:59:09 +00001288 * To find the name of the file containing the object.
Georg Brandl116aa622007-08-15 14:28:22 +00001289
Benjamin Petersone41251e2008-04-25 01:59:09 +00001290 * To help find the line number of the object within its file.
Georg Brandl116aa622007-08-15 14:28:22 +00001291
Benjamin Petersone41251e2008-04-25 01:59:09 +00001292 If *module* is ``False``, no attempt to find the module will be made. This is
1293 obscure, of use mostly in testing doctest itself: if *module* is ``False``, or
1294 is ``None`` but cannot be found automatically, then all objects are considered
1295 to belong to the (non-existent) module, so all contained objects will
1296 (recursively) be searched for doctests.
Georg Brandl116aa622007-08-15 14:28:22 +00001297
Benjamin Petersone41251e2008-04-25 01:59:09 +00001298 The globals for each :class:`DocTest` is formed by combining *globs* and
1299 *extraglobs* (bindings in *extraglobs* override bindings in *globs*). A new
1300 shallow copy of the globals dictionary is created for each :class:`DocTest`.
1301 If *globs* is not specified, then it defaults to the module's *__dict__*, if
1302 specified, or ``{}`` otherwise. If *extraglobs* is not specified, then it
1303 defaults to ``{}``.
Georg Brandl116aa622007-08-15 14:28:22 +00001304
1305
1306.. _doctest-doctestparser:
1307
1308DocTestParser objects
1309^^^^^^^^^^^^^^^^^^^^^
1310
1311
1312.. class:: DocTestParser()
1313
1314 A processing class used to extract interactive examples from a string, and use
1315 them to create a :class:`DocTest` object.
1316
Georg Brandl116aa622007-08-15 14:28:22 +00001317
Benjamin Petersone41251e2008-04-25 01:59:09 +00001318 :class:`DocTestParser` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001319
1320
Benjamin Petersone41251e2008-04-25 01:59:09 +00001321 .. method:: get_doctest(string, globs, name, filename, lineno)
Georg Brandl116aa622007-08-15 14:28:22 +00001322
Benjamin Petersone41251e2008-04-25 01:59:09 +00001323 Extract all doctest examples from the given string, and collect them into a
1324 :class:`DocTest` object.
Georg Brandl116aa622007-08-15 14:28:22 +00001325
Benjamin Petersone41251e2008-04-25 01:59:09 +00001326 *globs*, *name*, *filename*, and *lineno* are attributes for the new
1327 :class:`DocTest` object. See the documentation for :class:`DocTest` for more
1328 information.
Georg Brandl116aa622007-08-15 14:28:22 +00001329
1330
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001331 .. method:: get_examples(string, name='<string>')
Georg Brandl116aa622007-08-15 14:28:22 +00001332
Benjamin Petersone41251e2008-04-25 01:59:09 +00001333 Extract all doctest examples from the given string, and return them as a list
1334 of :class:`Example` objects. Line numbers are 0-based. The optional argument
1335 *name* is a name identifying this string, and is only used for error messages.
Georg Brandl116aa622007-08-15 14:28:22 +00001336
1337
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001338 .. method:: parse(string, name='<string>')
Georg Brandl116aa622007-08-15 14:28:22 +00001339
Benjamin Petersone41251e2008-04-25 01:59:09 +00001340 Divide the given string into examples and intervening text, and return them as
1341 a list of alternating :class:`Example`\ s and strings. Line numbers for the
1342 :class:`Example`\ s are 0-based. The optional argument *name* is a name
1343 identifying this string, and is only used for error messages.
Georg Brandl116aa622007-08-15 14:28:22 +00001344
1345
1346.. _doctest-doctestrunner:
1347
1348DocTestRunner objects
1349^^^^^^^^^^^^^^^^^^^^^
1350
1351
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001352.. class:: DocTestRunner(checker=None, verbose=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +00001353
1354 A processing class used to execute and verify the interactive examples in a
1355 :class:`DocTest`.
1356
1357 The comparison between expected outputs and actual outputs is done by an
1358 :class:`OutputChecker`. This comparison may be customized with a number of
1359 option flags; see section :ref:`doctest-options` for more information. If the
1360 option flags are insufficient, then the comparison may also be customized by
1361 passing a subclass of :class:`OutputChecker` to the constructor.
1362
1363 The test runner's display output can be controlled in two ways. First, an output
1364 function can be passed to :meth:`TestRunner.run`; this function will be called
1365 with strings that should be displayed. It defaults to ``sys.stdout.write``. If
1366 capturing the output is not sufficient, then the display output can be also
1367 customized by subclassing DocTestRunner, and overriding the methods
1368 :meth:`report_start`, :meth:`report_success`,
1369 :meth:`report_unexpected_exception`, and :meth:`report_failure`.
1370
1371 The optional keyword argument *checker* specifies the :class:`OutputChecker`
1372 object (or drop-in replacement) that should be used to compare the expected
1373 outputs to the actual outputs of doctest examples.
1374
1375 The optional keyword argument *verbose* controls the :class:`DocTestRunner`'s
1376 verbosity. If *verbose* is ``True``, then information is printed about each
1377 example, as it is run. If *verbose* is ``False``, then only failures are
1378 printed. If *verbose* is unspecified, or ``None``, then verbose output is used
Éric Araujo713d3032010-11-18 16:38:46 +00001379 iff the command-line switch ``-v`` is used.
Georg Brandl116aa622007-08-15 14:28:22 +00001380
1381 The optional keyword argument *optionflags* can be used to control how the test
1382 runner compares expected output to actual output, and how it displays failures.
1383 For more information, see section :ref:`doctest-options`.
1384
Georg Brandl116aa622007-08-15 14:28:22 +00001385
Benjamin Petersone41251e2008-04-25 01:59:09 +00001386 :class:`DocTestParser` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001387
1388
Benjamin Petersone41251e2008-04-25 01:59:09 +00001389 .. method:: report_start(out, test, example)
Georg Brandl116aa622007-08-15 14:28:22 +00001390
Benjamin Petersone41251e2008-04-25 01:59:09 +00001391 Report that the test runner is about to process the given example. This method
1392 is provided to allow subclasses of :class:`DocTestRunner` to customize their
1393 output; it 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. *test* is the test
1396 *containing example*. *out* is the output function that was passed to
1397 :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001398
1399
Benjamin Petersone41251e2008-04-25 01:59:09 +00001400 .. method:: report_success(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 ran successfully. This method is provided to
1403 allow subclasses of :class:`DocTestRunner` to customize their output; it
1404 should not 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_failure(out, test, example, got)
Georg Brandl116aa622007-08-15 14:28:22 +00001412
Benjamin Petersone41251e2008-04-25 01:59:09 +00001413 Report that the given example failed. This method is provided to allow
1414 subclasses of :class:`DocTestRunner` to customize their output; it should not
1415 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. *got* is the actual output
1418 from the example. *test* is the test containing *example*. *out* is the
1419 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001420
1421
Benjamin Petersone41251e2008-04-25 01:59:09 +00001422 .. method:: report_unexpected_exception(out, test, example, exc_info)
Georg Brandl116aa622007-08-15 14:28:22 +00001423
Benjamin Petersone41251e2008-04-25 01:59:09 +00001424 Report that the given example raised an unexpected exception. This method is
1425 provided to allow subclasses of :class:`DocTestRunner` to customize their
1426 output; it should not be called directly.
Georg Brandl116aa622007-08-15 14:28:22 +00001427
Benjamin Petersone41251e2008-04-25 01:59:09 +00001428 *example* is the example about to be processed. *exc_info* is a tuple
1429 containing information about the unexpected exception (as returned by
1430 :func:`sys.exc_info`). *test* is the test containing *example*. *out* is the
1431 output function that was passed to :meth:`DocTestRunner.run`.
Georg Brandl116aa622007-08-15 14:28:22 +00001432
1433
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001434 .. method:: run(test, compileflags=None, out=None, clear_globs=True)
Georg Brandl116aa622007-08-15 14:28:22 +00001435
Benjamin Petersone41251e2008-04-25 01:59:09 +00001436 Run the examples in *test* (a :class:`DocTest` object), and display the
1437 results using the writer function *out*.
Georg Brandl116aa622007-08-15 14:28:22 +00001438
Benjamin Petersone41251e2008-04-25 01:59:09 +00001439 The examples are run in the namespace ``test.globs``. If *clear_globs* is
1440 true (the default), then this namespace will be cleared after the test runs,
1441 to help with garbage collection. If you would like to examine the namespace
1442 after the test completes, then use *clear_globs=False*.
Georg Brandl116aa622007-08-15 14:28:22 +00001443
Benjamin Petersone41251e2008-04-25 01:59:09 +00001444 *compileflags* gives the set of flags that should be used by the Python
1445 compiler when running the examples. If not specified, then it will default to
1446 the set of future-import flags that apply to *globs*.
Georg Brandl116aa622007-08-15 14:28:22 +00001447
Benjamin Petersone41251e2008-04-25 01:59:09 +00001448 The output of each example is checked using the :class:`DocTestRunner`'s
1449 output checker, and the results are formatted by the
1450 :meth:`DocTestRunner.report_\*` methods.
Georg Brandl116aa622007-08-15 14:28:22 +00001451
1452
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001453 .. method:: summarize(verbose=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001454
Benjamin Petersone41251e2008-04-25 01:59:09 +00001455 Print a summary of all the test cases that have been run by this DocTestRunner,
1456 and return a :term:`named tuple` ``TestResults(failed, attempted)``.
Georg Brandl116aa622007-08-15 14:28:22 +00001457
Benjamin Petersone41251e2008-04-25 01:59:09 +00001458 The optional *verbose* argument controls how detailed the summary is. If the
1459 verbosity is not specified, then the :class:`DocTestRunner`'s verbosity is
1460 used.
Georg Brandl116aa622007-08-15 14:28:22 +00001461
1462.. _doctest-outputchecker:
1463
1464OutputChecker objects
1465^^^^^^^^^^^^^^^^^^^^^
1466
1467
1468.. class:: OutputChecker()
1469
1470 A class used to check the whether the actual output from a doctest example
1471 matches the expected output. :class:`OutputChecker` defines two methods:
1472 :meth:`check_output`, which compares a given pair of outputs, and returns true
1473 if they match; and :meth:`output_difference`, which returns a string describing
1474 the differences between two outputs.
1475
Georg Brandl116aa622007-08-15 14:28:22 +00001476
Benjamin Petersone41251e2008-04-25 01:59:09 +00001477 :class:`OutputChecker` defines the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +00001478
Benjamin Petersone41251e2008-04-25 01:59:09 +00001479 .. method:: check_output(want, got, optionflags)
Georg Brandl116aa622007-08-15 14:28:22 +00001480
Benjamin Petersone41251e2008-04-25 01:59:09 +00001481 Return ``True`` iff the actual output from an example (*got*) matches the
1482 expected output (*want*). These strings are always considered to match if
1483 they are identical; but depending on what option flags the test runner is
1484 using, several non-exact match types are also possible. See section
1485 :ref:`doctest-options` for more information about option flags.
Georg Brandl116aa622007-08-15 14:28:22 +00001486
1487
Benjamin Petersone41251e2008-04-25 01:59:09 +00001488 .. method:: output_difference(example, got, optionflags)
Georg Brandl116aa622007-08-15 14:28:22 +00001489
Benjamin Petersone41251e2008-04-25 01:59:09 +00001490 Return a string describing the differences between the expected output for a
1491 given example (*example*) and the actual output (*got*). *optionflags* is the
1492 set of option flags used to compare *want* and *got*.
Georg Brandl116aa622007-08-15 14:28:22 +00001493
1494
1495.. _doctest-debugging:
1496
1497Debugging
1498---------
1499
1500Doctest provides several mechanisms for debugging doctest examples:
1501
1502* Several functions convert doctests to executable Python programs, which can be
1503 run under the Python debugger, :mod:`pdb`.
1504
1505* The :class:`DebugRunner` class is a subclass of :class:`DocTestRunner` that
1506 raises an exception for the first failing example, containing information about
1507 that example. This information can be used to perform post-mortem debugging on
1508 the example.
1509
1510* The :mod:`unittest` cases generated by :func:`DocTestSuite` support the
1511 :meth:`debug` method defined by :class:`unittest.TestCase`.
1512
1513* You can add a call to :func:`pdb.set_trace` in a doctest example, and you'll
1514 drop into the Python debugger when that line is executed. Then you can inspect
1515 current values of variables, and so on. For example, suppose :file:`a.py`
1516 contains just this module docstring::
1517
1518 """
1519 >>> def f(x):
1520 ... g(x*2)
1521 >>> def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001522 ... print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001523 ... import pdb; pdb.set_trace()
1524 >>> f(3)
1525 9
1526 """
1527
1528 Then an interactive Python session may look like this::
1529
1530 >>> import a, doctest
1531 >>> doctest.testmod(a)
1532 --Return--
1533 > <doctest a[1]>(3)g()->None
1534 -> import pdb; pdb.set_trace()
1535 (Pdb) list
1536 1 def g(x):
Georg Brandl6911e3c2007-09-04 07:15:32 +00001537 2 print(x+3)
Georg Brandl116aa622007-08-15 14:28:22 +00001538 3 -> import pdb; pdb.set_trace()
1539 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001540 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001541 6
1542 (Pdb) step
1543 --Return--
1544 > <doctest a[0]>(2)f()->None
1545 -> g(x*2)
1546 (Pdb) list
1547 1 def f(x):
1548 2 -> g(x*2)
1549 [EOF]
Georg Brandl6911e3c2007-09-04 07:15:32 +00001550 (Pdb) p x
Georg Brandl116aa622007-08-15 14:28:22 +00001551 3
1552 (Pdb) step
1553 --Return--
1554 > <doctest a[2]>(1)?()->None
1555 -> f(3)
1556 (Pdb) cont
1557 (0, 3)
1558 >>>
1559
Georg Brandl116aa622007-08-15 14:28:22 +00001560
1561Functions that convert doctests to Python code, and possibly run the synthesized
1562code under the debugger:
1563
1564
1565.. function:: script_from_examples(s)
1566
1567 Convert text with examples to a script.
1568
1569 Argument *s* is a string containing doctest examples. The string is converted
1570 to a Python script, where doctest examples in *s* are converted to regular code,
1571 and everything else is converted to Python comments. The generated script is
1572 returned as a string. For example, ::
1573
1574 import doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001575 print(doctest.script_from_examples(r"""
Georg Brandl116aa622007-08-15 14:28:22 +00001576 Set x and y to 1 and 2.
1577 >>> x, y = 1, 2
1578
1579 Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001580 >>> print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001581 3
Georg Brandl6911e3c2007-09-04 07:15:32 +00001582 """))
Georg Brandl116aa622007-08-15 14:28:22 +00001583
1584 displays::
1585
1586 # Set x and y to 1 and 2.
1587 x, y = 1, 2
1588 #
1589 # Print their sum:
Georg Brandl6911e3c2007-09-04 07:15:32 +00001590 print(x+y)
Georg Brandl116aa622007-08-15 14:28:22 +00001591 # Expected:
1592 ## 3
1593
1594 This function is used internally by other functions (see below), but can also be
1595 useful when you want to transform an interactive Python session into a Python
1596 script.
1597
Georg Brandl116aa622007-08-15 14:28:22 +00001598
1599.. function:: testsource(module, name)
1600
1601 Convert the doctest for an object to a script.
1602
1603 Argument *module* is a module object, or dotted name of a module, containing the
1604 object whose doctests are of interest. Argument *name* is the name (within the
1605 module) of the object with the doctests of interest. The result is a string,
1606 containing the object's docstring converted to a Python script, as described for
1607 :func:`script_from_examples` above. For example, if module :file:`a.py`
1608 contains a top-level function :func:`f`, then ::
1609
1610 import a, doctest
Georg Brandl6911e3c2007-09-04 07:15:32 +00001611 print(doctest.testsource(a, "a.f"))
Georg Brandl116aa622007-08-15 14:28:22 +00001612
1613 prints a script version of function :func:`f`'s docstring, with doctests
1614 converted to code, and the rest placed in comments.
1615
Georg Brandl116aa622007-08-15 14:28:22 +00001616
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001617.. function:: debug(module, name, pm=False)
Georg Brandl116aa622007-08-15 14:28:22 +00001618
1619 Debug the doctests for an object.
1620
1621 The *module* and *name* arguments are the same as for function
1622 :func:`testsource` above. The synthesized Python script for the named object's
1623 docstring is written to a temporary file, and then that file is run under the
1624 control of the Python debugger, :mod:`pdb`.
1625
1626 A shallow copy of ``module.__dict__`` is used for both local and global
1627 execution context.
1628
1629 Optional argument *pm* controls whether post-mortem debugging is used. If *pm*
1630 has a true value, the script file is run directly, and the debugger gets
1631 involved only if the script terminates via raising an unhandled exception. If
1632 it does, then post-mortem debugging is invoked, via :func:`pdb.post_mortem`,
1633 passing the traceback object from the unhandled exception. If *pm* is not
1634 specified, or is false, the script is run under the debugger from the start, via
1635 passing an appropriate :func:`exec` call to :func:`pdb.run`.
1636
Georg Brandl116aa622007-08-15 14:28:22 +00001637
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001638.. function:: debug_src(src, pm=False, globs=None)
Georg Brandl116aa622007-08-15 14:28:22 +00001639
1640 Debug the doctests in a string.
1641
1642 This is like function :func:`debug` above, except that a string containing
1643 doctest examples is specified directly, via the *src* argument.
1644
1645 Optional argument *pm* has the same meaning as in function :func:`debug` above.
1646
1647 Optional argument *globs* gives a dictionary to use as both local and global
1648 execution context. If not specified, or ``None``, an empty dictionary is used.
1649 If specified, a shallow copy of the dictionary is used.
1650
Georg Brandl116aa622007-08-15 14:28:22 +00001651
1652The :class:`DebugRunner` class, and the special exceptions it may raise, are of
1653most interest to testing framework authors, and will only be sketched here. See
1654the source code, and especially :class:`DebugRunner`'s docstring (which is a
1655doctest!) for more details:
1656
1657
Georg Brandlc2a4f4f2009-04-10 09:03:43 +00001658.. class:: DebugRunner(checker=None, verbose=None, optionflags=0)
Georg Brandl116aa622007-08-15 14:28:22 +00001659
1660 A subclass of :class:`DocTestRunner` that raises an exception as soon as a
1661 failure is encountered. If an unexpected exception occurs, an
1662 :exc:`UnexpectedException` exception is raised, containing the test, the
1663 example, and the original exception. If the output doesn't match, then a
1664 :exc:`DocTestFailure` exception is raised, containing the test, the example, and
1665 the actual output.
1666
1667 For information about the constructor parameters and methods, see the
1668 documentation for :class:`DocTestRunner` in section :ref:`doctest-advanced-api`.
1669
1670There are two exceptions that may be raised by :class:`DebugRunner` instances:
1671
1672
1673.. exception:: DocTestFailure(test, example, got)
1674
Georg Brandl7cb13192010-08-03 12:06:29 +00001675 An exception raised by :class:`DocTestRunner` to signal that a doctest example's
Georg Brandl116aa622007-08-15 14:28:22 +00001676 actual output did not match its expected output. The constructor arguments are
Senthil Kumarana6bac952011-07-04 11:28:30 -07001677 used to initialize the attributes of the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001678
Senthil Kumarana6bac952011-07-04 11:28:30 -07001679:exc:`DocTestFailure` defines the following attributes:
Georg Brandl116aa622007-08-15 14:28:22 +00001680
1681
1682.. attribute:: DocTestFailure.test
1683
1684 The :class:`DocTest` object that was being run when the example failed.
1685
1686
1687.. attribute:: DocTestFailure.example
1688
1689 The :class:`Example` that failed.
1690
1691
1692.. attribute:: DocTestFailure.got
1693
1694 The example's actual output.
1695
1696
1697.. exception:: UnexpectedException(test, example, exc_info)
1698
Georg Brandl7cb13192010-08-03 12:06:29 +00001699 An exception raised by :class:`DocTestRunner` to signal that a doctest
1700 example raised an unexpected exception. The constructor arguments are used
Senthil Kumarana6bac952011-07-04 11:28:30 -07001701 to initialize the attributes of the same names.
Georg Brandl116aa622007-08-15 14:28:22 +00001702
Senthil Kumarana6bac952011-07-04 11:28:30 -07001703:exc:`UnexpectedException` defines the following attributes:
Georg Brandl116aa622007-08-15 14:28:22 +00001704
1705
1706.. attribute:: UnexpectedException.test
1707
1708 The :class:`DocTest` object that was being run when the example failed.
1709
1710
1711.. attribute:: UnexpectedException.example
1712
1713 The :class:`Example` that failed.
1714
1715
1716.. attribute:: UnexpectedException.exc_info
1717
1718 A tuple containing information about the unexpected exception, as returned by
1719 :func:`sys.exc_info`.
1720
1721
1722.. _doctest-soapbox:
1723
1724Soapbox
1725-------
1726
1727As mentioned in the introduction, :mod:`doctest` has grown to have three primary
1728uses:
1729
1730#. Checking examples in docstrings.
1731
1732#. Regression testing.
1733
1734#. Executable documentation / literate testing.
1735
1736These uses have different requirements, and it is important to distinguish them.
1737In particular, filling your docstrings with obscure test cases makes for bad
1738documentation.
1739
1740When writing a docstring, choose docstring examples with care. There's an art to
1741this that needs to be learned---it may not be natural at first. Examples should
1742add genuine value to the documentation. A good example can often be worth many
1743words. If done with care, the examples will be invaluable for your users, and
1744will pay back the time it takes to collect them many times over as the years go
1745by and things change. I'm still amazed at how often one of my :mod:`doctest`
1746examples stops working after a "harmless" change.
1747
1748Doctest also makes an excellent tool for regression testing, especially if you
1749don't skimp on explanatory text. By interleaving prose and examples, it becomes
1750much easier to keep track of what's actually being tested, and why. When a test
1751fails, good prose can make it much easier to figure out what the problem is, and
1752how it should be fixed. It's true that you could write extensive comments in
1753code-based testing, but few programmers do. Many have found that using doctest
1754approaches instead leads to much clearer tests. Perhaps this is simply because
1755doctest makes writing prose a little easier than writing code, while writing
1756comments in code is a little harder. I think it goes deeper than just that:
1757the natural attitude when writing a doctest-based test is that you want to
1758explain the fine points of your software, and illustrate them with examples.
1759This in turn naturally leads to test files that start with the simplest
1760features, and logically progress to complications and edge cases. A coherent
1761narrative is the result, instead of a collection of isolated functions that test
1762isolated bits of functionality seemingly at random. It's a different attitude,
1763and produces different results, blurring the distinction between testing and
1764explaining.
1765
1766Regression testing is best confined to dedicated objects or files. There are
1767several options for organizing tests:
1768
1769* Write text files containing test cases as interactive examples, and test the
1770 files using :func:`testfile` or :func:`DocFileSuite`. This is recommended,
1771 although is easiest to do for new projects, designed from the start to use
1772 doctest.
1773
1774* Define functions named ``_regrtest_topic`` that consist of single docstrings,
1775 containing test cases for the named topics. These functions can be included in
1776 the same file as the module, or separated out into a separate test file.
1777
1778* Define a ``__test__`` dictionary mapping from regression test topics to
1779 docstrings containing test cases.
1780
1781.. rubric:: Footnotes
1782
1783.. [#] Examples containing both expected output and an exception are not supported.
1784 Trying to guess where one ends and the other begins is too error-prone, and that
1785 also makes for a confusing test.