blob: ee4e0687844ad6e8f079d7c08a51e83b293e0db7 [file] [log] [blame]
Tim Peters4fd9e2f2001-08-18 00:05:50 +00001# Module doctest.
Tim Peters8485b562004-08-04 18:46:34 +00002# Released to the public domain 16-Jan-2001, by Tim Peters (tim@python.org).
Tim Peters19397e52004-08-06 22:02:59 +00003# Major enhancements and refactoring by:
Tim Peters8485b562004-08-04 18:46:34 +00004# Jim Fulton
5# Edward Loper
Tim Peters8a7d2d52001-01-16 07:10:57 +00006
7# Provided as-is; use at your own risk; no warranty; no promises; enjoy!
8
Martin v. Löwis92816de2004-05-31 19:01:00 +00009r"""Module doctest -- a framework for running examples in docstrings.
Tim Peters8a7d2d52001-01-16 07:10:57 +000010
Tim Peters80e53142004-08-09 04:34:45 +000011In simplest use, end each module M to be tested with:
Tim Peters8a7d2d52001-01-16 07:10:57 +000012
13def _test():
Tim Peters80e53142004-08-09 04:34:45 +000014 import doctest
Tim Peters48983fc2004-09-25 02:41:28 +000015 doctest.testmod()
Tim Peters8a7d2d52001-01-16 07:10:57 +000016
17if __name__ == "__main__":
18 _test()
19
20Then running the module as a script will cause the examples in the
21docstrings to get executed and verified:
22
23python M.py
24
25This won't display anything unless an example fails, in which case the
26failing example(s) and the cause(s) of the failure(s) are printed to stdout
27(why not stderr? because stderr is a lame hack <0.2 wink>), and the final
28line of output is "Test failed.".
29
30Run it with the -v switch instead:
31
32python M.py -v
33
34and a detailed report of all examples tried is printed to stdout, along
35with assorted summaries at the end.
36
Tim Peters80e53142004-08-09 04:34:45 +000037You can force verbose mode by passing "verbose=True" to testmod, or prohibit
38it by passing "verbose=False". In either of those cases, sys.argv is not
Tim Peters8a7d2d52001-01-16 07:10:57 +000039examined by testmod.
40
Tim Peters80e53142004-08-09 04:34:45 +000041There are a variety of other ways to run doctests, including integration
42with the unittest framework, and support for running non-Python text
43files containing doctests. There are also many ways to override parts
44of doctest's default behaviors. See the Library Reference Manual for
45details.
Tim Peters8a7d2d52001-01-16 07:10:57 +000046"""
Tim Peters48983fc2004-09-25 02:41:28 +000047
Edward Loper8e4a34b2004-08-12 02:34:27 +000048__docformat__ = 'reStructuredText en'
Tim Peters8a7d2d52001-01-16 07:10:57 +000049
Tim Peters4fd9e2f2001-08-18 00:05:50 +000050__all__ = [
Edward Loperb7503ff2004-08-19 19:19:03 +000051 # 0, Option Flags
52 'register_optionflag',
53 'DONT_ACCEPT_TRUE_FOR_1',
54 'DONT_ACCEPT_BLANKLINE',
55 'NORMALIZE_WHITESPACE',
56 'ELLIPSIS',
Thomas Wouters477c8d52006-05-27 19:21:47 +000057 'SKIP',
Tim Peters1fbf9c52004-09-04 17:21:02 +000058 'IGNORE_EXCEPTION_DETAIL',
Tim Petersba601962004-09-04 15:04:06 +000059 'COMPARISON_FLAGS',
Edward Loper71f55af2004-08-26 01:41:51 +000060 'REPORT_UDIFF',
61 'REPORT_CDIFF',
62 'REPORT_NDIFF',
Jim Fultonf54bad42004-08-28 14:57:56 +000063 'REPORT_ONLY_FIRST_FAILURE',
Tim Petersba601962004-09-04 15:04:06 +000064 'REPORTING_FLAGS',
R David Murray5a9d7062012-11-21 15:09:21 -050065 'FAIL_FAST',
Edward Loperb7503ff2004-08-19 19:19:03 +000066 # 1. Utility Functions
Edward Loperb7503ff2004-08-19 19:19:03 +000067 # 2. Example & DocTest
Tim Peters8485b562004-08-04 18:46:34 +000068 'Example',
69 'DocTest',
Edward Loperb7503ff2004-08-19 19:19:03 +000070 # 3. Doctest Parser
71 'DocTestParser',
72 # 4. Doctest Finder
Tim Peters8485b562004-08-04 18:46:34 +000073 'DocTestFinder',
Edward Loperb7503ff2004-08-19 19:19:03 +000074 # 5. Doctest Runner
Tim Peters8485b562004-08-04 18:46:34 +000075 'DocTestRunner',
Edward Loperb7503ff2004-08-19 19:19:03 +000076 'OutputChecker',
77 'DocTestFailure',
78 'UnexpectedException',
79 'DebugRunner',
80 # 6. Test Functions
Tim Peters4fd9e2f2001-08-18 00:05:50 +000081 'testmod',
Edward Loper052d0cd2004-09-19 17:19:33 +000082 'testfile',
Tim Peters4fd9e2f2001-08-18 00:05:50 +000083 'run_docstring_examples',
Georg Brandlcc80bfe2008-05-12 18:06:23 +000084 # 7. Unittest Support
Tim Petersdb3756d2003-06-29 05:30:48 +000085 'DocTestSuite',
Edward Loperb7503ff2004-08-19 19:19:03 +000086 'DocFileSuite',
Tim Peters9d02a7c2004-09-26 01:50:24 +000087 'set_unittest_reportflags',
Georg Brandlcc80bfe2008-05-12 18:06:23 +000088 # 8. Debugging Support
Edward Loperb7503ff2004-08-19 19:19:03 +000089 'script_from_examples',
Tim Petersdb3756d2003-06-29 05:30:48 +000090 'testsource',
Edward Loperb7503ff2004-08-19 19:19:03 +000091 'debug_src',
Tim Petersdb3756d2003-06-29 05:30:48 +000092 'debug',
Tim Peters4fd9e2f2001-08-18 00:05:50 +000093]
Tim Peters8a7d2d52001-01-16 07:10:57 +000094
Tim Peters4fd9e2f2001-08-18 00:05:50 +000095import __future__
R David Murray5707d502013-06-23 14:24:13 -040096import argparse
Victor Stinner12b8d142011-06-30 17:35:55 +020097import difflib
98import inspect
99import linecache
100import os
101import pdb
102import re
103import sys
104import traceback
105import unittest
Guido van Rossum34d19282007-08-09 01:03:29 +0000106from io import StringIO
Christian Heimes25bb7832008-01-11 16:17:00 +0000107from collections import namedtuple
108
109TestResults = namedtuple('TestResults', 'failed attempted')
Tim Peters7402f792001-10-02 03:53:41 +0000110
Tim Peters19397e52004-08-06 22:02:59 +0000111# There are 4 basic classes:
112# - Example: a <source, want> pair, plus an intra-docstring line number.
113# - DocTest: a collection of examples, parsed from a docstring, plus
114# info about where the docstring came from (name, filename, lineno).
115# - DocTestFinder: extracts DocTests from a given object's docstring and
116# its contained objects' docstrings.
117# - DocTestRunner: runs DocTest cases, and accumulates statistics.
118#
119# So the basic picture is:
120#
121# list of:
122# +------+ +---------+ +-------+
123# |object| --DocTestFinder-> | DocTest | --DocTestRunner-> |results|
124# +------+ +---------+ +-------+
125# | Example |
126# | ... |
127# | Example |
128# +---------+
129
Edward Loperac20f572004-08-12 02:02:24 +0000130# Option constants.
Tim Peters38330fe2004-08-30 16:19:24 +0000131
Edward Loperac20f572004-08-12 02:02:24 +0000132OPTIONFLAGS_BY_NAME = {}
133def register_optionflag(name):
Thomas Wouters477c8d52006-05-27 19:21:47 +0000134 # Create a new flag unless `name` is already known.
135 return OPTIONFLAGS_BY_NAME.setdefault(name, 1 << len(OPTIONFLAGS_BY_NAME))
Edward Loperac20f572004-08-12 02:02:24 +0000136
137DONT_ACCEPT_TRUE_FOR_1 = register_optionflag('DONT_ACCEPT_TRUE_FOR_1')
138DONT_ACCEPT_BLANKLINE = register_optionflag('DONT_ACCEPT_BLANKLINE')
139NORMALIZE_WHITESPACE = register_optionflag('NORMALIZE_WHITESPACE')
140ELLIPSIS = register_optionflag('ELLIPSIS')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000141SKIP = register_optionflag('SKIP')
Tim Peters1fbf9c52004-09-04 17:21:02 +0000142IGNORE_EXCEPTION_DETAIL = register_optionflag('IGNORE_EXCEPTION_DETAIL')
Tim Peters38330fe2004-08-30 16:19:24 +0000143
144COMPARISON_FLAGS = (DONT_ACCEPT_TRUE_FOR_1 |
145 DONT_ACCEPT_BLANKLINE |
146 NORMALIZE_WHITESPACE |
Tim Peters1fbf9c52004-09-04 17:21:02 +0000147 ELLIPSIS |
Thomas Wouters477c8d52006-05-27 19:21:47 +0000148 SKIP |
Edward Loper7d88a582004-09-28 05:50:57 +0000149 IGNORE_EXCEPTION_DETAIL)
Tim Peters38330fe2004-08-30 16:19:24 +0000150
Edward Loper71f55af2004-08-26 01:41:51 +0000151REPORT_UDIFF = register_optionflag('REPORT_UDIFF')
152REPORT_CDIFF = register_optionflag('REPORT_CDIFF')
153REPORT_NDIFF = register_optionflag('REPORT_NDIFF')
Edward Lopera89f88d2004-08-26 02:45:51 +0000154REPORT_ONLY_FIRST_FAILURE = register_optionflag('REPORT_ONLY_FIRST_FAILURE')
R David Murray5a9d7062012-11-21 15:09:21 -0500155FAIL_FAST = register_optionflag('FAIL_FAST')
Edward Loperac20f572004-08-12 02:02:24 +0000156
Tim Peters38330fe2004-08-30 16:19:24 +0000157REPORTING_FLAGS = (REPORT_UDIFF |
158 REPORT_CDIFF |
159 REPORT_NDIFF |
R David Murray5a9d7062012-11-21 15:09:21 -0500160 REPORT_ONLY_FIRST_FAILURE |
161 FAIL_FAST)
Tim Peters38330fe2004-08-30 16:19:24 +0000162
Edward Loperac20f572004-08-12 02:02:24 +0000163# Special string markers for use in `want` strings:
164BLANKLINE_MARKER = '<BLANKLINE>'
165ELLIPSIS_MARKER = '...'
166
Tim Peters8485b562004-08-04 18:46:34 +0000167######################################################################
168## Table of Contents
169######################################################################
Edward Loper7c748462004-08-09 02:06:06 +0000170# 1. Utility Functions
171# 2. Example & DocTest -- store test cases
172# 3. DocTest Parser -- extracts examples from strings
173# 4. DocTest Finder -- extracts test cases from objects
174# 5. DocTest Runner -- runs test cases
175# 6. Test Functions -- convenient wrappers for testing
Georg Brandl31835852008-05-12 17:38:56 +0000176# 7. Unittest Support
177# 8. Debugging Support
178# 9. Example Usage
Tim Peters8a7d2d52001-01-16 07:10:57 +0000179
Tim Peters8485b562004-08-04 18:46:34 +0000180######################################################################
181## 1. Utility Functions
182######################################################################
Tim Peters8a7d2d52001-01-16 07:10:57 +0000183
Tim Peters8485b562004-08-04 18:46:34 +0000184def _extract_future_flags(globs):
185 """
186 Return the compiler-flags associated with the future features that
187 have been imported into the given namespace (globs).
188 """
189 flags = 0
190 for fname in __future__.all_feature_names:
191 feature = globs.get(fname, None)
192 if feature is getattr(__future__, fname):
193 flags |= feature.compiler_flag
194 return flags
Tim Peters7402f792001-10-02 03:53:41 +0000195
Tim Peters8485b562004-08-04 18:46:34 +0000196def _normalize_module(module, depth=2):
197 """
198 Return the module specified by `module`. In particular:
199 - If `module` is a module, then return module.
200 - If `module` is a string, then import and return the
201 module with that name.
202 - If `module` is None, then return the calling module.
203 The calling module is assumed to be the module of
204 the stack frame at the given depth in the call stack.
205 """
206 if inspect.ismodule(module):
207 return module
Walter Dörwaldaa97f042007-05-03 21:05:51 +0000208 elif isinstance(module, str):
Tim Peters8485b562004-08-04 18:46:34 +0000209 return __import__(module, globals(), locals(), ["*"])
210 elif module is None:
211 return sys.modules[sys._getframe(depth).f_globals['__name__']]
212 else:
213 raise TypeError("Expected a module, string, or None")
Tim Peters7402f792001-10-02 03:53:41 +0000214
Guido van Rossum1b81e7b2007-08-29 03:53:53 +0000215def _load_testfile(filename, package, module_relative, encoding):
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000216 if module_relative:
217 package = _normalize_module(package, 3)
218 filename = _module_relative_path(package, filename)
Brett Cannon4c14b5d2013-05-04 13:56:58 -0400219 if getattr(package, '__loader__', None) is not None:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000220 if hasattr(package.__loader__, 'get_data'):
Guido van Rossumcd4d4522007-11-22 00:30:02 +0000221 file_contents = package.__loader__.get_data(filename)
222 file_contents = file_contents.decode(encoding)
223 # get_data() opens files as 'rb', so one must do the equivalent
224 # conversion as universal newlines would do.
225 return file_contents.replace(os.linesep, '\n'), filename
Antoine Pitrou92f60ed2010-10-14 22:11:44 +0000226 with open(filename, encoding=encoding) as f:
227 return f.read(), filename
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000228
Edward Loperaacf0832004-08-26 01:19:50 +0000229def _indent(s, indent=4):
Tim Peters8485b562004-08-04 18:46:34 +0000230 """
Florent Xicluna59250852010-02-27 14:21:57 +0000231 Add the given number of space characters to the beginning of
232 every non-blank line in `s`, and return the result.
Tim Peters8485b562004-08-04 18:46:34 +0000233 """
Edward Loperaacf0832004-08-26 01:19:50 +0000234 # This regexp matches the start of non-blank lines:
235 return re.sub('(?m)^(?!$)', indent*' ', s)
Tim Peters8a7d2d52001-01-16 07:10:57 +0000236
Edward Loper8e4a34b2004-08-12 02:34:27 +0000237def _exception_traceback(exc_info):
238 """
239 Return a string containing a traceback message for the given
240 exc_info tuple (as returned by sys.exc_info()).
241 """
242 # Get a traceback message.
243 excout = StringIO()
244 exc_type, exc_val, exc_tb = exc_info
245 traceback.print_exception(exc_type, exc_val, exc_tb, file=excout)
246 return excout.getvalue()
247
Tim Peters8485b562004-08-04 18:46:34 +0000248# Override some StringIO methods.
249class _SpoofOut(StringIO):
250 def getvalue(self):
251 result = StringIO.getvalue(self)
252 # If anything at all was written, make sure there's a trailing
253 # newline. There's no way for the expected output to indicate
254 # that a trailing newline is missing.
255 if result and not result.endswith("\n"):
256 result += "\n"
Tim Peters8485b562004-08-04 18:46:34 +0000257 return result
Tim Peters8a7d2d52001-01-16 07:10:57 +0000258
Guido van Rossum79139b22007-02-09 23:20:19 +0000259 def truncate(self, size=None):
Antoine Pitroub3d77002010-01-31 23:12:29 +0000260 self.seek(size)
261 StringIO.truncate(self)
Tim Peters8a7d2d52001-01-16 07:10:57 +0000262
Tim Peters26b3ebb2004-08-19 08:10:08 +0000263# Worst-case linear-time ellipsis matching.
Tim Petersb0a04e12004-08-20 02:08:04 +0000264def _ellipsis_match(want, got):
Tim Petersdc5de3b2004-08-19 14:06:20 +0000265 """
266 Essentially the only subtle case:
Tim Petersb0a04e12004-08-20 02:08:04 +0000267 >>> _ellipsis_match('aa...aa', 'aaa')
Tim Petersdc5de3b2004-08-19 14:06:20 +0000268 False
269 """
Tim Peters26b3ebb2004-08-19 08:10:08 +0000270 if ELLIPSIS_MARKER not in want:
271 return want == got
Tim Petersdc5de3b2004-08-19 14:06:20 +0000272
Tim Peters26b3ebb2004-08-19 08:10:08 +0000273 # Find "the real" strings.
274 ws = want.split(ELLIPSIS_MARKER)
275 assert len(ws) >= 2
Tim Peters26b3ebb2004-08-19 08:10:08 +0000276
Tim Petersdc5de3b2004-08-19 14:06:20 +0000277 # Deal with exact matches possibly needed at one or both ends.
278 startpos, endpos = 0, len(got)
279 w = ws[0]
280 if w: # starts with exact match
281 if got.startswith(w):
282 startpos = len(w)
283 del ws[0]
284 else:
285 return False
286 w = ws[-1]
287 if w: # ends with exact match
288 if got.endswith(w):
289 endpos -= len(w)
290 del ws[-1]
291 else:
292 return False
293
294 if startpos > endpos:
295 # Exact end matches required more characters than we have, as in
Tim Petersb0a04e12004-08-20 02:08:04 +0000296 # _ellipsis_match('aa...aa', 'aaa')
Tim Petersdc5de3b2004-08-19 14:06:20 +0000297 return False
298
299 # For the rest, we only need to find the leftmost non-overlapping
300 # match for each piece. If there's no overall match that way alone,
301 # there's no overall match period.
Tim Peters26b3ebb2004-08-19 08:10:08 +0000302 for w in ws:
303 # w may be '' at times, if there are consecutive ellipses, or
304 # due to an ellipsis at the start or end of `want`. That's OK.
Tim Petersdc5de3b2004-08-19 14:06:20 +0000305 # Search for an empty string succeeds, and doesn't change startpos.
306 startpos = got.find(w, startpos, endpos)
307 if startpos < 0:
Tim Peters26b3ebb2004-08-19 08:10:08 +0000308 return False
Tim Petersdc5de3b2004-08-19 14:06:20 +0000309 startpos += len(w)
Tim Peters26b3ebb2004-08-19 08:10:08 +0000310
Tim Petersdc5de3b2004-08-19 14:06:20 +0000311 return True
Tim Peters26b3ebb2004-08-19 08:10:08 +0000312
Edward Loper00f8da72004-08-26 18:05:07 +0000313def _comment_line(line):
314 "Return a commented form of the given line"
315 line = line.rstrip()
316 if line:
317 return '# '+line
318 else:
319 return '#'
320
Edward Loper2de91ba2004-08-27 02:07:46 +0000321class _OutputRedirectingPdb(pdb.Pdb):
322 """
323 A specialized version of the python debugger that redirects stdout
324 to a given stream when interacting with the user. Stdout is *not*
325 redirected when traced code is executed.
326 """
327 def __init__(self, out):
328 self.__out = out
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +0000329 self.__debugger_used = False
Georg Brandl34748cd2010-12-04 17:11:36 +0000330 # do not play signal games in the pdb
331 pdb.Pdb.__init__(self, stdout=out, nosigint=True)
Georg Brandld72e0432010-07-30 09:59:28 +0000332 # still use input() to get user input
333 self.use_rawinput = 1
Edward Loper2de91ba2004-08-27 02:07:46 +0000334
Guido van Rossum0d3fb8a2007-11-26 23:23:18 +0000335 def set_trace(self, frame=None):
336 self.__debugger_used = True
337 if frame is None:
338 frame = sys._getframe().f_back
339 pdb.Pdb.set_trace(self, frame)
340
341 def set_continue(self):
342 # Calling set_continue unconditionally would break unit test
343 # coverage reporting, as Bdb.set_continue calls sys.settrace(None).
344 if self.__debugger_used:
345 pdb.Pdb.set_continue(self)
346
Edward Loper2de91ba2004-08-27 02:07:46 +0000347 def trace_dispatch(self, *args):
348 # Redirect stdout to the given stream.
349 save_stdout = sys.stdout
350 sys.stdout = self.__out
351 # Call Pdb's trace dispatch method.
Tim Petersd7bbbbc2004-11-08 22:30:28 +0000352 try:
353 return pdb.Pdb.trace_dispatch(self, *args)
354 finally:
Tim Petersd7bbbbc2004-11-08 22:30:28 +0000355 sys.stdout = save_stdout
Edward Loper2de91ba2004-08-27 02:07:46 +0000356
Edward Lopera2fc7ec2004-09-21 03:24:24 +0000357# [XX] Normalize with respect to os.path.pardir?
Edward Loper052d0cd2004-09-19 17:19:33 +0000358def _module_relative_path(module, path):
359 if not inspect.ismodule(module):
Collin Winterce36ad82007-08-30 01:19:48 +0000360 raise TypeError('Expected a module: %r' % module)
Edward Loper052d0cd2004-09-19 17:19:33 +0000361 if path.startswith('/'):
Collin Winterce36ad82007-08-30 01:19:48 +0000362 raise ValueError('Module-relative files may not have absolute paths')
Edward Loper052d0cd2004-09-19 17:19:33 +0000363
364 # Find the base directory for the path.
365 if hasattr(module, '__file__'):
366 # A normal module/package
367 basedir = os.path.split(module.__file__)[0]
368 elif module.__name__ == '__main__':
369 # An interactive session.
370 if len(sys.argv)>0 and sys.argv[0] != '':
371 basedir = os.path.split(sys.argv[0])[0]
372 else:
373 basedir = os.curdir
374 else:
375 # A module w/o __file__ (this includes builtins)
376 raise ValueError("Can't resolve paths relative to the module " +
377 module + " (it has no __file__)")
378
379 # Combine the base directory and the path.
380 return os.path.join(basedir, *(path.split('/')))
381
Tim Peters8485b562004-08-04 18:46:34 +0000382######################################################################
383## 2. Example & DocTest
384######################################################################
385## - An "example" is a <source, want> pair, where "source" is a
386## fragment of source code, and "want" is the expected output for
387## "source." The Example class also includes information about
388## where the example was extracted from.
389##
Edward Lopera1ef6112004-08-09 16:14:41 +0000390## - A "doctest" is a collection of examples, typically extracted from
391## a string (such as an object's docstring). The DocTest class also
392## includes information about where the string was extracted from.
Tim Peters8a7d2d52001-01-16 07:10:57 +0000393
Tim Peters8485b562004-08-04 18:46:34 +0000394class Example:
395 """
396 A single doctest example, consisting of source code and expected
Edward Lopera1ef6112004-08-09 16:14:41 +0000397 output. `Example` defines the following attributes:
Tim Peters8a7d2d52001-01-16 07:10:57 +0000398
Edward Loper74bca7a2004-08-12 02:27:44 +0000399 - source: A single Python statement, always ending with a newline.
Tim Petersbb431472004-08-09 03:51:46 +0000400 The constructor adds a newline if needed.
Tim Peters8a7d2d52001-01-16 07:10:57 +0000401
Edward Loper74bca7a2004-08-12 02:27:44 +0000402 - want: The expected output from running the source code (either
Tim Petersbb431472004-08-09 03:51:46 +0000403 from stdout, or a traceback in case of exception). `want` ends
404 with a newline unless it's empty, in which case it's an empty
405 string. The constructor adds a newline if needed.
Tim Peters8a7d2d52001-01-16 07:10:57 +0000406
Edward Lopera6b68322004-08-26 00:05:43 +0000407 - exc_msg: The exception message generated by the example, if
408 the example is expected to generate an exception; or `None` if
409 it is not expected to generate an exception. This exception
410 message is compared against the return value of
411 `traceback.format_exception_only()`. `exc_msg` ends with a
412 newline unless it's `None`. The constructor adds a newline
413 if needed.
414
Edward Loper74bca7a2004-08-12 02:27:44 +0000415 - lineno: The line number within the DocTest string containing
Tim Peters8485b562004-08-04 18:46:34 +0000416 this Example where the Example begins. This line number is
417 zero-based, with respect to the beginning of the DocTest.
Edward Loper74bca7a2004-08-12 02:27:44 +0000418
419 - indent: The example's indentation in the DocTest string.
Ezio Melotti30b9d5d2013-08-17 15:50:46 +0300420 I.e., the number of space characters that precede the
Edward Loper74bca7a2004-08-12 02:27:44 +0000421 example's first prompt.
422
423 - options: A dictionary mapping from option flags to True or
424 False, which is used to override default options for this
425 example. Any option flags not contained in this dictionary
426 are left at their default value (as specified by the
427 DocTestRunner's optionflags). By default, no options are set.
Tim Peters8485b562004-08-04 18:46:34 +0000428 """
Edward Lopera6b68322004-08-26 00:05:43 +0000429 def __init__(self, source, want, exc_msg=None, lineno=0, indent=0,
430 options=None):
Tim Petersbb431472004-08-09 03:51:46 +0000431 # Normalize inputs.
432 if not source.endswith('\n'):
433 source += '\n'
434 if want and not want.endswith('\n'):
435 want += '\n'
Edward Lopera6b68322004-08-26 00:05:43 +0000436 if exc_msg is not None and not exc_msg.endswith('\n'):
437 exc_msg += '\n'
Tim Peters8485b562004-08-04 18:46:34 +0000438 # Store properties.
439 self.source = source
440 self.want = want
441 self.lineno = lineno
Edward Loper74bca7a2004-08-12 02:27:44 +0000442 self.indent = indent
443 if options is None: options = {}
444 self.options = options
Edward Lopera6b68322004-08-26 00:05:43 +0000445 self.exc_msg = exc_msg
Tim Peters8a7d2d52001-01-16 07:10:57 +0000446
Antoine Pitrou2bc801c2011-12-18 19:27:45 +0100447 def __eq__(self, other):
448 if type(self) is not type(other):
449 return NotImplemented
450
451 return self.source == other.source and \
452 self.want == other.want and \
453 self.lineno == other.lineno and \
454 self.indent == other.indent and \
455 self.options == other.options and \
456 self.exc_msg == other.exc_msg
457
458 def __ne__(self, other):
459 return not self == other
460
Antoine Pitrou165b1282011-12-18 20:20:17 +0100461 def __hash__(self):
462 return hash((self.source, self.want, self.lineno, self.indent,
463 self.exc_msg))
Antoine Pitrou2bc801c2011-12-18 19:27:45 +0100464
Tim Peters8485b562004-08-04 18:46:34 +0000465class DocTest:
466 """
467 A collection of doctest examples that should be run in a single
Edward Lopera1ef6112004-08-09 16:14:41 +0000468 namespace. Each `DocTest` defines the following attributes:
Tim Peters8a7d2d52001-01-16 07:10:57 +0000469
Tim Peters8485b562004-08-04 18:46:34 +0000470 - examples: the list of examples.
Tim Peters8a7d2d52001-01-16 07:10:57 +0000471
Tim Peters8485b562004-08-04 18:46:34 +0000472 - globs: The namespace (aka globals) that the examples should
473 be run in.
Tim Peters8a7d2d52001-01-16 07:10:57 +0000474
Tim Peters8485b562004-08-04 18:46:34 +0000475 - name: A name identifying the DocTest (typically, the name of
476 the object whose docstring this DocTest was extracted from).
Tim Peters8a7d2d52001-01-16 07:10:57 +0000477
Tim Peters8485b562004-08-04 18:46:34 +0000478 - filename: The name of the file that this DocTest was extracted
Edward Lopera1ef6112004-08-09 16:14:41 +0000479 from, or `None` if the filename is unknown.
Tim Peters8a7d2d52001-01-16 07:10:57 +0000480
Tim Peters8485b562004-08-04 18:46:34 +0000481 - lineno: The line number within filename where this DocTest
Edward Lopera1ef6112004-08-09 16:14:41 +0000482 begins, or `None` if the line number is unavailable. This
483 line number is zero-based, with respect to the beginning of
484 the file.
485
486 - docstring: The string that the examples were extracted from,
487 or `None` if the string is unavailable.
Tim Peters8485b562004-08-04 18:46:34 +0000488 """
Edward Lopera1ef6112004-08-09 16:14:41 +0000489 def __init__(self, examples, globs, name, filename, lineno, docstring):
Tim Peters8485b562004-08-04 18:46:34 +0000490 """
Edward Lopera1ef6112004-08-09 16:14:41 +0000491 Create a new DocTest containing the given examples. The
492 DocTest's globals are initialized with a copy of `globs`.
Tim Peters8485b562004-08-04 18:46:34 +0000493 """
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000494 assert not isinstance(examples, str), \
Edward Lopera1ef6112004-08-09 16:14:41 +0000495 "DocTest no longer accepts str; use DocTestParser instead"
496 self.examples = examples
497 self.docstring = docstring
Tim Peters8485b562004-08-04 18:46:34 +0000498 self.globs = globs.copy()
Tim Peters8485b562004-08-04 18:46:34 +0000499 self.name = name
500 self.filename = filename
501 self.lineno = lineno
Tim Peters8485b562004-08-04 18:46:34 +0000502
503 def __repr__(self):
504 if len(self.examples) == 0:
505 examples = 'no examples'
506 elif len(self.examples) == 1:
507 examples = '1 example'
508 else:
509 examples = '%d examples' % len(self.examples)
510 return ('<DocTest %s from %s:%s (%s)>' %
511 (self.name, self.filename, self.lineno, examples))
512
Antoine Pitrou2bc801c2011-12-18 19:27:45 +0100513 def __eq__(self, other):
514 if type(self) is not type(other):
515 return NotImplemented
516
517 return self.examples == other.examples and \
518 self.docstring == other.docstring and \
519 self.globs == other.globs and \
520 self.name == other.name and \
521 self.filename == other.filename and \
522 self.lineno == other.lineno
523
524 def __ne__(self, other):
525 return not self == other
Tim Peters8485b562004-08-04 18:46:34 +0000526
Antoine Pitrou165b1282011-12-18 20:20:17 +0100527 def __hash__(self):
528 return hash((self.docstring, self.name, self.filename, self.lineno))
529
Tim Peters8485b562004-08-04 18:46:34 +0000530 # This lets us sort tests by name:
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000531 def __lt__(self, other):
Tim Peters8485b562004-08-04 18:46:34 +0000532 if not isinstance(other, DocTest):
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000533 return NotImplemented
534 return ((self.name, self.filename, self.lineno, id(self))
535 <
536 (other.name, other.filename, other.lineno, id(other)))
Tim Peters8485b562004-08-04 18:46:34 +0000537
538######################################################################
Edward Loperb7503ff2004-08-19 19:19:03 +0000539## 3. DocTestParser
Edward Loper7c748462004-08-09 02:06:06 +0000540######################################################################
541
Edward Lopera1ef6112004-08-09 16:14:41 +0000542class DocTestParser:
Edward Loper7c748462004-08-09 02:06:06 +0000543 """
Edward Lopera1ef6112004-08-09 16:14:41 +0000544 A class used to parse strings containing doctest examples.
Edward Loper7c748462004-08-09 02:06:06 +0000545 """
Edward Loper8e4a34b2004-08-12 02:34:27 +0000546 # This regular expression is used to find doctest examples in a
547 # string. It defines three groups: `source` is the source code
548 # (including leading indentation and prompts); `indent` is the
549 # indentation of the first (PS1) line of the source code; and
550 # `want` is the expected output (including leading indentation).
Edward Loper7c748462004-08-09 02:06:06 +0000551 _EXAMPLE_RE = re.compile(r'''
Tim Petersd40a92b2004-08-09 03:28:45 +0000552 # Source consists of a PS1 line followed by zero or more PS2 lines.
553 (?P<source>
554 (?:^(?P<indent> [ ]*) >>> .*) # PS1 line
555 (?:\n [ ]* \.\.\. .*)*) # PS2 lines
556 \n?
557 # Want consists of any non-blank lines that do not start with PS1.
558 (?P<want> (?:(?![ ]*$) # Not a blank line
559 (?![ ]*>>>) # Not a line starting with PS1
Serhiy Storchaka1ca66ed2013-08-19 22:59:31 +0300560 .+$\n? # But any other line
Tim Petersd40a92b2004-08-09 03:28:45 +0000561 )*)
562 ''', re.MULTILINE | re.VERBOSE)
Edward Loper8e4a34b2004-08-12 02:34:27 +0000563
Edward Lopera6b68322004-08-26 00:05:43 +0000564 # A regular expression for handling `want` strings that contain
565 # expected exceptions. It divides `want` into three pieces:
566 # - the traceback header line (`hdr`)
567 # - the traceback stack (`stack`)
568 # - the exception message (`msg`), as generated by
569 # traceback.format_exception_only()
570 # `msg` may have multiple lines. We assume/require that the
571 # exception message is the first non-indented line starting with a word
572 # character following the traceback header line.
573 _EXCEPTION_RE = re.compile(r"""
574 # Grab the traceback header. Different versions of Python have
575 # said different things on the first traceback line.
576 ^(?P<hdr> Traceback\ \(
577 (?: most\ recent\ call\ last
578 | innermost\ last
579 ) \) :
580 )
581 \s* $ # toss trailing whitespace on the header.
582 (?P<stack> .*?) # don't blink: absorb stuff until...
583 ^ (?P<msg> \w+ .*) # a line *starts* with alphanum.
584 """, re.VERBOSE | re.MULTILINE | re.DOTALL)
585
Tim Peters7ea48dd2004-08-13 01:52:59 +0000586 # A callable returning a true value iff its argument is a blank line
587 # or contains a single comment.
Edward Loper8e4a34b2004-08-12 02:34:27 +0000588 _IS_BLANK_OR_COMMENT = re.compile(r'^[ ]*(#.*)?$').match
Edward Loper7c748462004-08-09 02:06:06 +0000589
Edward Loper00f8da72004-08-26 18:05:07 +0000590 def parse(self, string, name='<string>'):
591 """
592 Divide the given string into examples and intervening text,
593 and return them as a list of alternating Examples and strings.
594 Line numbers for the Examples are 0-based. The optional
595 argument `name` is a name identifying this string, and is only
596 used for error messages.
597 """
598 string = string.expandtabs()
599 # If all lines begin with the same indentation, then strip it.
600 min_indent = self._min_indent(string)
601 if min_indent > 0:
602 string = '\n'.join([l[min_indent:] for l in string.split('\n')])
603
604 output = []
605 charno, lineno = 0, 0
606 # Find all doctest examples in the string:
Edward Loper2de91ba2004-08-27 02:07:46 +0000607 for m in self._EXAMPLE_RE.finditer(string):
Edward Loper00f8da72004-08-26 18:05:07 +0000608 # Add the pre-example text to `output`.
609 output.append(string[charno:m.start()])
610 # Update lineno (lines before this example)
611 lineno += string.count('\n', charno, m.start())
612 # Extract info from the regexp match.
613 (source, options, want, exc_msg) = \
614 self._parse_example(m, name, lineno)
615 # Create an Example, and add it to the list.
616 if not self._IS_BLANK_OR_COMMENT(source):
617 output.append( Example(source, want, exc_msg,
618 lineno=lineno,
619 indent=min_indent+len(m.group('indent')),
620 options=options) )
621 # Update lineno (lines inside this example)
622 lineno += string.count('\n', m.start(), m.end())
623 # Update charno.
624 charno = m.end()
625 # Add any remaining post-example text to `output`.
626 output.append(string[charno:])
627 return output
628
Edward Lopera1ef6112004-08-09 16:14:41 +0000629 def get_doctest(self, string, globs, name, filename, lineno):
Edward Loper7c748462004-08-09 02:06:06 +0000630 """
Edward Lopera1ef6112004-08-09 16:14:41 +0000631 Extract all doctest examples from the given string, and
632 collect them into a `DocTest` object.
633
634 `globs`, `name`, `filename`, and `lineno` are attributes for
635 the new `DocTest` object. See the documentation for `DocTest`
636 for more information.
637 """
638 return DocTest(self.get_examples(string, name), globs,
639 name, filename, lineno, string)
640
641 def get_examples(self, string, name='<string>'):
642 """
643 Extract all doctest examples from the given string, and return
644 them as a list of `Example` objects. Line numbers are
645 0-based, because it's most common in doctests that nothing
646 interesting appears on the same line as opening triple-quote,
647 and so the first interesting line is called \"line 1\" then.
648
649 The optional argument `name` is a name identifying this
650 string, and is only used for error messages.
Edward Loper7c748462004-08-09 02:06:06 +0000651 """
Edward Loper00f8da72004-08-26 18:05:07 +0000652 return [x for x in self.parse(string, name)
653 if isinstance(x, Example)]
Edward Loper7c748462004-08-09 02:06:06 +0000654
Edward Loper74bca7a2004-08-12 02:27:44 +0000655 def _parse_example(self, m, name, lineno):
656 """
657 Given a regular expression match from `_EXAMPLE_RE` (`m`),
658 return a pair `(source, want)`, where `source` is the matched
659 example's source code (with prompts and indentation stripped);
660 and `want` is the example's expected output (with indentation
661 stripped).
662
663 `name` is the string's name, and `lineno` is the line number
664 where the example starts; both are used for error messages.
665 """
Edward Loper7c748462004-08-09 02:06:06 +0000666 # Get the example's indentation level.
667 indent = len(m.group('indent'))
668
669 # Divide source into lines; check that they're properly
670 # indented; and then strip their indentation & prompts.
671 source_lines = m.group('source').split('\n')
Edward Lopera1ef6112004-08-09 16:14:41 +0000672 self._check_prompt_blank(source_lines, indent, name, lineno)
Tim Petersc5049152004-08-22 17:34:58 +0000673 self._check_prefix(source_lines[1:], ' '*indent + '.', name, lineno)
Edward Loper7c748462004-08-09 02:06:06 +0000674 source = '\n'.join([sl[indent+4:] for sl in source_lines])
Edward Loper7c748462004-08-09 02:06:06 +0000675
Tim Petersc5049152004-08-22 17:34:58 +0000676 # Divide want into lines; check that it's properly indented; and
677 # then strip the indentation. Spaces before the last newline should
678 # be preserved, so plain rstrip() isn't good enough.
Jim Fulton07a349c2004-08-22 14:10:00 +0000679 want = m.group('want')
Jim Fulton07a349c2004-08-22 14:10:00 +0000680 want_lines = want.split('\n')
Tim Petersc5049152004-08-22 17:34:58 +0000681 if len(want_lines) > 1 and re.match(r' *$', want_lines[-1]):
682 del want_lines[-1] # forget final newline & spaces after it
Edward Lopera1ef6112004-08-09 16:14:41 +0000683 self._check_prefix(want_lines, ' '*indent, name,
Tim Petersc5049152004-08-22 17:34:58 +0000684 lineno + len(source_lines))
Edward Loper7c748462004-08-09 02:06:06 +0000685 want = '\n'.join([wl[indent:] for wl in want_lines])
Edward Loper7c748462004-08-09 02:06:06 +0000686
Edward Lopera6b68322004-08-26 00:05:43 +0000687 # If `want` contains a traceback message, then extract it.
688 m = self._EXCEPTION_RE.match(want)
689 if m:
690 exc_msg = m.group('msg')
691 else:
692 exc_msg = None
693
Edward Loper00f8da72004-08-26 18:05:07 +0000694 # Extract options from the source.
695 options = self._find_options(source, name, lineno)
696
697 return source, options, want, exc_msg
Edward Loper7c748462004-08-09 02:06:06 +0000698
Edward Loper74bca7a2004-08-12 02:27:44 +0000699 # This regular expression looks for option directives in the
700 # source code of an example. Option directives are comments
701 # starting with "doctest:". Warning: this may give false
702 # positives for string-literals that contain the string
703 # "#doctest:". Eliminating these false positives would require
704 # actually parsing the string; but we limit them by ignoring any
705 # line containing "#doctest:" that is *followed* by a quote mark.
706 _OPTION_DIRECTIVE_RE = re.compile(r'#\s*doctest:\s*([^\n\'"]*)$',
707 re.MULTILINE)
708
709 def _find_options(self, source, name, lineno):
710 """
711 Return a dictionary containing option overrides extracted from
712 option directives in the given source string.
713
714 `name` is the string's name, and `lineno` is the line number
715 where the example starts; both are used for error messages.
716 """
717 options = {}
718 # (note: with the current regexp, this will match at most once:)
719 for m in self._OPTION_DIRECTIVE_RE.finditer(source):
720 option_strings = m.group(1).replace(',', ' ').split()
721 for option in option_strings:
722 if (option[0] not in '+-' or
723 option[1:] not in OPTIONFLAGS_BY_NAME):
724 raise ValueError('line %r of the doctest for %s '
725 'has an invalid option: %r' %
726 (lineno+1, name, option))
727 flag = OPTIONFLAGS_BY_NAME[option[1:]]
728 options[flag] = (option[0] == '+')
729 if options and self._IS_BLANK_OR_COMMENT(source):
730 raise ValueError('line %r of the doctest for %s has an option '
731 'directive on a line with no example: %r' %
732 (lineno, name, source))
733 return options
734
Edward Lopera5db6002004-08-12 02:41:30 +0000735 # This regular expression finds the indentation of every non-blank
736 # line in a string.
Edward Loper00f8da72004-08-26 18:05:07 +0000737 _INDENT_RE = re.compile('^([ ]*)(?=\S)', re.MULTILINE)
Edward Lopera5db6002004-08-12 02:41:30 +0000738
739 def _min_indent(self, s):
740 "Return the minimum indentation of any non-blank line in `s`"
Edward Loper00f8da72004-08-26 18:05:07 +0000741 indents = [len(indent) for indent in self._INDENT_RE.findall(s)]
742 if len(indents) > 0:
743 return min(indents)
Tim Petersdd0e4752004-08-09 03:31:56 +0000744 else:
Edward Loper00f8da72004-08-26 18:05:07 +0000745 return 0
Edward Loper7c748462004-08-09 02:06:06 +0000746
Edward Lopera1ef6112004-08-09 16:14:41 +0000747 def _check_prompt_blank(self, lines, indent, name, lineno):
Edward Loper74bca7a2004-08-12 02:27:44 +0000748 """
749 Given the lines of a source string (including prompts and
750 leading indentation), check to make sure that every prompt is
751 followed by a space character. If any line is not followed by
752 a space character, then raise ValueError.
753 """
Edward Loper7c748462004-08-09 02:06:06 +0000754 for i, line in enumerate(lines):
755 if len(line) >= indent+4 and line[indent+3] != ' ':
756 raise ValueError('line %r of the docstring for %s '
757 'lacks blank after %s: %r' %
Edward Lopera1ef6112004-08-09 16:14:41 +0000758 (lineno+i+1, name,
Edward Loper7c748462004-08-09 02:06:06 +0000759 line[indent:indent+3], line))
760
Edward Lopera1ef6112004-08-09 16:14:41 +0000761 def _check_prefix(self, lines, prefix, name, lineno):
Edward Loper74bca7a2004-08-12 02:27:44 +0000762 """
763 Check that every line in the given list starts with the given
764 prefix; if any line does not, then raise a ValueError.
765 """
Edward Loper7c748462004-08-09 02:06:06 +0000766 for i, line in enumerate(lines):
767 if line and not line.startswith(prefix):
768 raise ValueError('line %r of the docstring for %s has '
769 'inconsistent leading whitespace: %r' %
Edward Lopera1ef6112004-08-09 16:14:41 +0000770 (lineno+i+1, name, line))
Edward Loper7c748462004-08-09 02:06:06 +0000771
772
773######################################################################
774## 4. DocTest Finder
Tim Peters8485b562004-08-04 18:46:34 +0000775######################################################################
776
777class DocTestFinder:
778 """
779 A class used to extract the DocTests that are relevant to a given
780 object, from its docstring and the docstrings of its contained
781 objects. Doctests can currently be extracted from the following
782 object types: modules, functions, classes, methods, staticmethods,
783 classmethods, and properties.
Tim Peters8485b562004-08-04 18:46:34 +0000784 """
785
Edward Lopera1ef6112004-08-09 16:14:41 +0000786 def __init__(self, verbose=False, parser=DocTestParser(),
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000787 recurse=True, exclude_empty=True):
Tim Peters8485b562004-08-04 18:46:34 +0000788 """
789 Create a new doctest finder.
790
Edward Lopera1ef6112004-08-09 16:14:41 +0000791 The optional argument `parser` specifies a class or
Tim Peters19397e52004-08-06 22:02:59 +0000792 function that should be used to create new DocTest objects (or
Tim Peters161c9632004-08-08 03:38:33 +0000793 objects that implement the same interface as DocTest). The
Tim Peters19397e52004-08-06 22:02:59 +0000794 signature for this factory function should match the signature
795 of the DocTest constructor.
796
Tim Peters8485b562004-08-04 18:46:34 +0000797 If the optional argument `recurse` is false, then `find` will
798 only examine the given object, and not any contained objects.
Edward Loper32ddbf72004-09-13 05:47:24 +0000799
Tim Peters958cc892004-09-13 14:53:28 +0000800 If the optional argument `exclude_empty` is false, then `find`
801 will include tests for objects with empty docstrings.
Tim Peters8485b562004-08-04 18:46:34 +0000802 """
Edward Lopera1ef6112004-08-09 16:14:41 +0000803 self._parser = parser
Tim Peters8485b562004-08-04 18:46:34 +0000804 self._verbose = verbose
Tim Peters8485b562004-08-04 18:46:34 +0000805 self._recurse = recurse
Edward Loper32ddbf72004-09-13 05:47:24 +0000806 self._exclude_empty = exclude_empty
Tim Peters8485b562004-08-04 18:46:34 +0000807
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000808 def find(self, obj, name=None, module=None, globs=None, extraglobs=None):
Tim Peters8485b562004-08-04 18:46:34 +0000809 """
810 Return a list of the DocTests that are defined by the given
811 object's docstring, or by any of its contained objects'
812 docstrings.
813
814 The optional parameter `module` is the module that contains
Tim Petersf3f57472004-08-08 06:11:48 +0000815 the given object. If the module is not specified or is None, then
816 the test finder will attempt to automatically determine the
Tim Peters8485b562004-08-04 18:46:34 +0000817 correct module. The object's module is used:
818
819 - As a default namespace, if `globs` is not specified.
820 - To prevent the DocTestFinder from extracting DocTests
Tim Petersf3f57472004-08-08 06:11:48 +0000821 from objects that are imported from other modules.
Tim Peters8485b562004-08-04 18:46:34 +0000822 - To find the name of the file containing the object.
823 - To help find the line number of the object within its
824 file.
825
Tim Petersf3f57472004-08-08 06:11:48 +0000826 Contained objects whose module does not match `module` are ignored.
827
828 If `module` is False, no attempt to find the module will be made.
829 This is obscure, of use mostly in tests: if `module` is False, or
830 is None but cannot be found automatically, then all objects are
831 considered to belong to the (non-existent) module, so all contained
832 objects will (recursively) be searched for doctests.
833
Tim Peters8485b562004-08-04 18:46:34 +0000834 The globals for each DocTest is formed by combining `globs`
835 and `extraglobs` (bindings in `extraglobs` override bindings
836 in `globs`). A new copy of the globals dictionary is created
837 for each DocTest. If `globs` is not specified, then it
838 defaults to the module's `__dict__`, if specified, or {}
839 otherwise. If `extraglobs` is not specified, then it defaults
840 to {}.
841
Tim Peters8485b562004-08-04 18:46:34 +0000842 """
843 # If name was not specified, then extract it from the object.
844 if name is None:
845 name = getattr(obj, '__name__', None)
846 if name is None:
847 raise ValueError("DocTestFinder.find: name must be given "
848 "when obj.__name__ doesn't exist: %r" %
849 (type(obj),))
850
851 # Find the module that contains the given object (if obj is
852 # a module, then module=obj.). Note: this may fail, in which
853 # case module will be None.
Tim Petersf3f57472004-08-08 06:11:48 +0000854 if module is False:
855 module = None
856 elif module is None:
Tim Peters8485b562004-08-04 18:46:34 +0000857 module = inspect.getmodule(obj)
858
859 # Read the module's source code. This is used by
860 # DocTestFinder._find_lineno to find the line number for a
861 # given object's docstring.
862 try:
R. David Murray58641de2009-06-12 15:33:19 +0000863 file = inspect.getsourcefile(obj)
Tim Peters8485b562004-08-04 18:46:34 +0000864 except TypeError:
865 source_lines = None
R. David Murray58641de2009-06-12 15:33:19 +0000866 else:
867 if not file:
868 # Check to see if it's one of our special internal "files"
869 # (see __patched_linecache_getlines).
870 file = inspect.getfile(obj)
871 if not file[0]+file[-2:] == '<]>': file = None
R. David Murray765b9762009-07-15 14:16:54 +0000872 if file is None:
873 source_lines = None
R. David Murray58641de2009-06-12 15:33:19 +0000874 else:
875 if module is not None:
876 # Supply the module globals in case the module was
877 # originally loaded via a PEP 302 loader and
878 # file is not a valid filesystem path
879 source_lines = linecache.getlines(file, module.__dict__)
880 else:
881 # No access to a loader, so assume it's a normal
882 # filesystem path
883 source_lines = linecache.getlines(file)
884 if not source_lines:
885 source_lines = None
Tim Peters8485b562004-08-04 18:46:34 +0000886
887 # Initialize globals, and merge in extraglobs.
Tim Peters8a7d2d52001-01-16 07:10:57 +0000888 if globs is None:
Tim Peters8485b562004-08-04 18:46:34 +0000889 if module is None:
890 globs = {}
891 else:
892 globs = module.__dict__.copy()
893 else:
894 globs = globs.copy()
895 if extraglobs is not None:
896 globs.update(extraglobs)
Raymond Hettinger0f055172009-01-27 10:06:09 +0000897 if '__name__' not in globs:
898 globs['__name__'] = '__main__' # provide a default module name
Tim Peters8a7d2d52001-01-16 07:10:57 +0000899
Ezio Melotti30b9d5d2013-08-17 15:50:46 +0300900 # Recursively explore `obj`, extracting DocTests.
Tim Peters8485b562004-08-04 18:46:34 +0000901 tests = []
Tim Petersf3f57472004-08-08 06:11:48 +0000902 self._find(tests, obj, name, module, source_lines, globs, {})
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000903 # Sort the tests by alpha order of names, for consistency in
904 # verbose-mode output. This was a feature of doctest in Pythons
905 # <= 2.3 that got lost by accident in 2.4. It was repaired in
906 # 2.4.4 and 2.5.
907 tests.sort()
Tim Peters8485b562004-08-04 18:46:34 +0000908 return tests
909
Tim Peters8485b562004-08-04 18:46:34 +0000910 def _from_module(self, module, object):
911 """
912 Return true if the given object is defined in the given
913 module.
914 """
915 if module is None:
916 return True
Benjamin Peterson6cadba72008-11-19 22:38:29 +0000917 elif inspect.getmodule(object) is not None:
918 return module is inspect.getmodule(object)
Tim Peters8485b562004-08-04 18:46:34 +0000919 elif inspect.isfunction(object):
Neal Norwitz221085d2007-02-25 20:55:47 +0000920 return module.__dict__ is object.__globals__
Zachary Warea4b7a752013-11-24 01:19:09 -0600921 elif inspect.ismethoddescriptor(object):
922 return module.__name__ == object.__objclass__.__module__
Tim Peters8485b562004-08-04 18:46:34 +0000923 elif inspect.isclass(object):
924 return module.__name__ == object.__module__
Tim Peters8485b562004-08-04 18:46:34 +0000925 elif hasattr(object, '__module__'):
926 return module.__name__ == object.__module__
927 elif isinstance(object, property):
928 return True # [XX] no way not be sure.
929 else:
930 raise ValueError("object must be a class or function")
931
Tim Petersf3f57472004-08-08 06:11:48 +0000932 def _find(self, tests, obj, name, module, source_lines, globs, seen):
Tim Peters8485b562004-08-04 18:46:34 +0000933 """
934 Find tests for the given object and any contained objects, and
935 add them to `tests`.
936 """
937 if self._verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000938 print('Finding tests in %s' % name)
Tim Peters8485b562004-08-04 18:46:34 +0000939
940 # If we've already processed this object, then ignore it.
941 if id(obj) in seen:
942 return
943 seen[id(obj)] = 1
944
945 # Find a test for this object, and add it to the list of tests.
946 test = self._get_test(obj, name, module, globs, source_lines)
947 if test is not None:
948 tests.append(test)
949
950 # Look for tests in a module's contained objects.
951 if inspect.ismodule(obj) and self._recurse:
952 for valname, val in obj.__dict__.items():
Tim Peters8485b562004-08-04 18:46:34 +0000953 valname = '%s.%s' % (name, valname)
954 # Recurse to functions & classes.
Zachary Warea4b7a752013-11-24 01:19:09 -0600955 if ((inspect.isroutine(val) or inspect.isclass(val)) and
Tim Petersf3f57472004-08-08 06:11:48 +0000956 self._from_module(module, val)):
Tim Peters8485b562004-08-04 18:46:34 +0000957 self._find(tests, val, valname, module, source_lines,
Tim Petersf3f57472004-08-08 06:11:48 +0000958 globs, seen)
Tim Peters8485b562004-08-04 18:46:34 +0000959
960 # Look for tests in a module's __test__ dictionary.
961 if inspect.ismodule(obj) and self._recurse:
962 for valname, val in getattr(obj, '__test__', {}).items():
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000963 if not isinstance(valname, str):
Tim Peters8485b562004-08-04 18:46:34 +0000964 raise ValueError("DocTestFinder.find: __test__ keys "
965 "must be strings: %r" %
966 (type(valname),))
Zachary Warea4b7a752013-11-24 01:19:09 -0600967 if not (inspect.isroutine(val) or inspect.isclass(val) or
968 inspect.ismodule(val) or isinstance(val, str)):
Tim Peters8485b562004-08-04 18:46:34 +0000969 raise ValueError("DocTestFinder.find: __test__ values "
970 "must be strings, functions, methods, "
971 "classes, or modules: %r" %
972 (type(val),))
Tim Petersc5684782004-09-13 01:07:12 +0000973 valname = '%s.__test__.%s' % (name, valname)
Tim Peters8485b562004-08-04 18:46:34 +0000974 self._find(tests, val, valname, module, source_lines,
Tim Petersf3f57472004-08-08 06:11:48 +0000975 globs, seen)
Tim Peters8485b562004-08-04 18:46:34 +0000976
977 # Look for tests in a class's contained objects.
978 if inspect.isclass(obj) and self._recurse:
979 for valname, val in obj.__dict__.items():
Tim Peters8485b562004-08-04 18:46:34 +0000980 # Special handling for staticmethod/classmethod.
981 if isinstance(val, staticmethod):
982 val = getattr(obj, valname)
983 if isinstance(val, classmethod):
Christian Heimesff737952007-11-27 10:40:20 +0000984 val = getattr(obj, valname).__func__
Tim Peters8485b562004-08-04 18:46:34 +0000985
986 # Recurse to methods, properties, and nested classes.
Zachary Warea4b7a752013-11-24 01:19:09 -0600987 if ((inspect.isroutine(val) or inspect.isclass(val) or
Tim Petersf3f57472004-08-08 06:11:48 +0000988 isinstance(val, property)) and
989 self._from_module(module, val)):
Tim Peters8485b562004-08-04 18:46:34 +0000990 valname = '%s.%s' % (name, valname)
991 self._find(tests, val, valname, module, source_lines,
Tim Petersf3f57472004-08-08 06:11:48 +0000992 globs, seen)
Tim Peters8485b562004-08-04 18:46:34 +0000993
994 def _get_test(self, obj, name, module, globs, source_lines):
995 """
996 Return a DocTest for the given object, if it defines a docstring;
997 otherwise, return None.
998 """
999 # Extract the object's docstring. If it doesn't have one,
1000 # then return None (no test for this object).
Guido van Rossum3172c5d2007-10-16 18:12:55 +00001001 if isinstance(obj, str):
Tim Peters8485b562004-08-04 18:46:34 +00001002 docstring = obj
1003 else:
1004 try:
1005 if obj.__doc__ is None:
Edward Loper32ddbf72004-09-13 05:47:24 +00001006 docstring = ''
1007 else:
Jim Fulton7d428782004-10-13 14:15:32 +00001008 docstring = obj.__doc__
Guido van Rossum3172c5d2007-10-16 18:12:55 +00001009 if not isinstance(docstring, str):
Jim Fulton7d428782004-10-13 14:15:32 +00001010 docstring = str(docstring)
Tim Peters8485b562004-08-04 18:46:34 +00001011 except (TypeError, AttributeError):
Edward Loper32ddbf72004-09-13 05:47:24 +00001012 docstring = ''
Tim Peters8485b562004-08-04 18:46:34 +00001013
1014 # Find the docstring's location in the file.
1015 lineno = self._find_lineno(obj, source_lines)
1016
Edward Loper32ddbf72004-09-13 05:47:24 +00001017 # Don't bother if the docstring is empty.
1018 if self._exclude_empty and not docstring:
1019 return None
1020
Tim Peters8485b562004-08-04 18:46:34 +00001021 # Return a DocTest for this object.
1022 if module is None:
1023 filename = None
1024 else:
1025 filename = getattr(module, '__file__', module.__name__)
Jim Fulton07a349c2004-08-22 14:10:00 +00001026 if filename[-4:] in (".pyc", ".pyo"):
1027 filename = filename[:-1]
Edward Lopera1ef6112004-08-09 16:14:41 +00001028 return self._parser.get_doctest(docstring, globs, name,
1029 filename, lineno)
Tim Peters8485b562004-08-04 18:46:34 +00001030
1031 def _find_lineno(self, obj, source_lines):
1032 """
1033 Return a line number of the given object's docstring. Note:
1034 this method assumes that the object has a docstring.
1035 """
1036 lineno = None
1037
1038 # Find the line number for modules.
1039 if inspect.ismodule(obj):
1040 lineno = 0
1041
1042 # Find the line number for classes.
1043 # Note: this could be fooled if a class is defined multiple
1044 # times in a single file.
1045 if inspect.isclass(obj):
1046 if source_lines is None:
1047 return None
1048 pat = re.compile(r'^\s*class\s*%s\b' %
1049 getattr(obj, '__name__', '-'))
1050 for i, line in enumerate(source_lines):
1051 if pat.match(line):
1052 lineno = i
1053 break
1054
1055 # Find the line number for functions & methods.
Christian Heimesff737952007-11-27 10:40:20 +00001056 if inspect.ismethod(obj): obj = obj.__func__
Neal Norwitz221085d2007-02-25 20:55:47 +00001057 if inspect.isfunction(obj): obj = obj.__code__
Tim Peters8485b562004-08-04 18:46:34 +00001058 if inspect.istraceback(obj): obj = obj.tb_frame
1059 if inspect.isframe(obj): obj = obj.f_code
1060 if inspect.iscode(obj):
1061 lineno = getattr(obj, 'co_firstlineno', None)-1
1062
1063 # Find the line number where the docstring starts. Assume
1064 # that it's the first line that begins with a quote mark.
1065 # Note: this could be fooled by a multiline function
1066 # signature, where a continuation line begins with a quote
1067 # mark.
1068 if lineno is not None:
1069 if source_lines is None:
1070 return lineno+1
1071 pat = re.compile('(^|.*:)\s*\w*("|\')')
1072 for lineno in range(lineno, len(source_lines)):
1073 if pat.match(source_lines[lineno]):
1074 return lineno
1075
1076 # We couldn't find the line number.
1077 return None
1078
1079######################################################################
Edward Loper7c748462004-08-09 02:06:06 +00001080## 5. DocTest Runner
Tim Peters8485b562004-08-04 18:46:34 +00001081######################################################################
1082
Tim Peters8485b562004-08-04 18:46:34 +00001083class DocTestRunner:
1084 """
1085 A class used to run DocTest test cases, and accumulate statistics.
1086 The `run` method is used to process a single DocTest case. It
1087 returns a tuple `(f, t)`, where `t` is the number of test cases
1088 tried, and `f` is the number of test cases that failed.
1089
1090 >>> tests = DocTestFinder().find(_TestClass)
1091 >>> runner = DocTestRunner(verbose=False)
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001092 >>> tests.sort(key = lambda test: test.name)
Tim Peters8485b562004-08-04 18:46:34 +00001093 >>> for test in tests:
Guido van Rossum7131f842007-02-09 20:13:25 +00001094 ... print(test.name, '->', runner.run(test))
Christian Heimes25bb7832008-01-11 16:17:00 +00001095 _TestClass -> TestResults(failed=0, attempted=2)
1096 _TestClass.__init__ -> TestResults(failed=0, attempted=2)
1097 _TestClass.get -> TestResults(failed=0, attempted=2)
1098 _TestClass.square -> TestResults(failed=0, attempted=1)
Tim Peters8485b562004-08-04 18:46:34 +00001099
1100 The `summarize` method prints a summary of all the test cases that
1101 have been run by the runner, and returns an aggregated `(f, t)`
1102 tuple:
1103
1104 >>> runner.summarize(verbose=1)
1105 4 items passed all tests:
1106 2 tests in _TestClass
1107 2 tests in _TestClass.__init__
1108 2 tests in _TestClass.get
1109 1 tests in _TestClass.square
1110 7 tests in 4 items.
1111 7 passed and 0 failed.
1112 Test passed.
Christian Heimes25bb7832008-01-11 16:17:00 +00001113 TestResults(failed=0, attempted=7)
Tim Peters8485b562004-08-04 18:46:34 +00001114
1115 The aggregated number of tried examples and failed examples is
1116 also available via the `tries` and `failures` attributes:
1117
1118 >>> runner.tries
1119 7
1120 >>> runner.failures
1121 0
1122
1123 The comparison between expected outputs and actual outputs is done
Edward Loper34fcb142004-08-09 02:45:41 +00001124 by an `OutputChecker`. This comparison may be customized with a
1125 number of option flags; see the documentation for `testmod` for
1126 more information. If the option flags are insufficient, then the
1127 comparison may also be customized by passing a subclass of
1128 `OutputChecker` to the constructor.
Tim Peters8485b562004-08-04 18:46:34 +00001129
1130 The test runner's display output can be controlled in two ways.
1131 First, an output function (`out) can be passed to
1132 `TestRunner.run`; this function will be called with strings that
1133 should be displayed. It defaults to `sys.stdout.write`. If
1134 capturing the output is not sufficient, then the display output
1135 can be also customized by subclassing DocTestRunner, and
1136 overriding the methods `report_start`, `report_success`,
1137 `report_unexpected_exception`, and `report_failure`.
1138 """
1139 # This divider string is used to separate failure messages, and to
1140 # separate sections of the summary.
1141 DIVIDER = "*" * 70
1142
Edward Loper34fcb142004-08-09 02:45:41 +00001143 def __init__(self, checker=None, verbose=None, optionflags=0):
Tim Peters8485b562004-08-04 18:46:34 +00001144 """
1145 Create a new test runner.
1146
Edward Loper34fcb142004-08-09 02:45:41 +00001147 Optional keyword arg `checker` is the `OutputChecker` that
1148 should be used to compare the expected outputs and actual
1149 outputs of doctest examples.
1150
Tim Peters8485b562004-08-04 18:46:34 +00001151 Optional keyword arg 'verbose' prints lots of stuff if true,
1152 only failures if false; by default, it's true iff '-v' is in
1153 sys.argv.
1154
1155 Optional argument `optionflags` can be used to control how the
1156 test runner compares expected output to actual output, and how
1157 it displays failures. See the documentation for `testmod` for
1158 more information.
1159 """
Edward Loper34fcb142004-08-09 02:45:41 +00001160 self._checker = checker or OutputChecker()
Tim Peters8a7d2d52001-01-16 07:10:57 +00001161 if verbose is None:
Tim Peters8485b562004-08-04 18:46:34 +00001162 verbose = '-v' in sys.argv
1163 self._verbose = verbose
Tim Peters6ebe61f2003-06-27 20:48:05 +00001164 self.optionflags = optionflags
Jim Fulton07a349c2004-08-22 14:10:00 +00001165 self.original_optionflags = optionflags
Tim Peters6ebe61f2003-06-27 20:48:05 +00001166
Tim Peters8485b562004-08-04 18:46:34 +00001167 # Keep track of the examples we've run.
1168 self.tries = 0
1169 self.failures = 0
1170 self._name2ft = {}
Tim Peters8a7d2d52001-01-16 07:10:57 +00001171
Tim Peters8485b562004-08-04 18:46:34 +00001172 # Create a fake output target for capturing doctest output.
1173 self._fakeout = _SpoofOut()
Tim Peters4fd9e2f2001-08-18 00:05:50 +00001174
Tim Peters8485b562004-08-04 18:46:34 +00001175 #/////////////////////////////////////////////////////////////////
Tim Peters8485b562004-08-04 18:46:34 +00001176 # Reporting methods
1177 #/////////////////////////////////////////////////////////////////
Tim Peters17111f32001-10-03 04:08:26 +00001178
Tim Peters8485b562004-08-04 18:46:34 +00001179 def report_start(self, out, test, example):
Tim Peters8a7d2d52001-01-16 07:10:57 +00001180 """
Tim Peters8485b562004-08-04 18:46:34 +00001181 Report that the test runner is about to process the given
1182 example. (Only displays a message if verbose=True)
1183 """
1184 if self._verbose:
Edward Loperaacf0832004-08-26 01:19:50 +00001185 if example.want:
1186 out('Trying:\n' + _indent(example.source) +
1187 'Expecting:\n' + _indent(example.want))
1188 else:
1189 out('Trying:\n' + _indent(example.source) +
1190 'Expecting nothing\n')
Tim Peters8a7d2d52001-01-16 07:10:57 +00001191
Tim Peters8485b562004-08-04 18:46:34 +00001192 def report_success(self, out, test, example, got):
1193 """
1194 Report that the given example ran successfully. (Only
1195 displays a message if verbose=True)
1196 """
1197 if self._verbose:
1198 out("ok\n")
Tim Peters8a7d2d52001-01-16 07:10:57 +00001199
Tim Peters8485b562004-08-04 18:46:34 +00001200 def report_failure(self, out, test, example, got):
1201 """
1202 Report that the given example failed.
1203 """
Edward Loper8e4a34b2004-08-12 02:34:27 +00001204 out(self._failure_header(test, example) +
Edward Loperca9111e2004-08-26 03:00:24 +00001205 self._checker.output_difference(example, got, self.optionflags))
Tim Peters7402f792001-10-02 03:53:41 +00001206
Tim Peters8485b562004-08-04 18:46:34 +00001207 def report_unexpected_exception(self, out, test, example, exc_info):
1208 """
1209 Report that the given example raised an unexpected exception.
1210 """
Edward Loper8e4a34b2004-08-12 02:34:27 +00001211 out(self._failure_header(test, example) +
Edward Loperaacf0832004-08-26 01:19:50 +00001212 'Exception raised:\n' + _indent(_exception_traceback(exc_info)))
Tim Peters7402f792001-10-02 03:53:41 +00001213
Edward Loper8e4a34b2004-08-12 02:34:27 +00001214 def _failure_header(self, test, example):
Jim Fulton07a349c2004-08-22 14:10:00 +00001215 out = [self.DIVIDER]
1216 if test.filename:
1217 if test.lineno is not None and example.lineno is not None:
1218 lineno = test.lineno + example.lineno + 1
1219 else:
1220 lineno = '?'
1221 out.append('File "%s", line %s, in %s' %
1222 (test.filename, lineno, test.name))
Tim Peters8485b562004-08-04 18:46:34 +00001223 else:
Jim Fulton07a349c2004-08-22 14:10:00 +00001224 out.append('Line %s, in %s' % (example.lineno+1, test.name))
1225 out.append('Failed example:')
1226 source = example.source
Edward Loperaacf0832004-08-26 01:19:50 +00001227 out.append(_indent(source))
1228 return '\n'.join(out)
Tim Peters7402f792001-10-02 03:53:41 +00001229
Tim Peters8485b562004-08-04 18:46:34 +00001230 #/////////////////////////////////////////////////////////////////
1231 # DocTest Running
1232 #/////////////////////////////////////////////////////////////////
Tim Peters7402f792001-10-02 03:53:41 +00001233
Tim Peters8485b562004-08-04 18:46:34 +00001234 def __run(self, test, compileflags, out):
Tim Peters8a7d2d52001-01-16 07:10:57 +00001235 """
Tim Peters8485b562004-08-04 18:46:34 +00001236 Run the examples in `test`. Write the outcome of each example
1237 with one of the `DocTestRunner.report_*` methods, using the
1238 writer function `out`. `compileflags` is the set of compiler
1239 flags that should be used to execute examples. Return a tuple
1240 `(f, t)`, where `t` is the number of examples tried, and `f`
1241 is the number of examples that failed. The examples are run
1242 in the namespace `test.globs`.
1243 """
1244 # Keep track of the number of failures and tries.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001245 failures = tries = 0
Tim Peters8485b562004-08-04 18:46:34 +00001246
1247 # Save the option flags (since option directives can be used
1248 # to modify them).
1249 original_optionflags = self.optionflags
1250
Tim Peters1fbf9c52004-09-04 17:21:02 +00001251 SUCCESS, FAILURE, BOOM = range(3) # `outcome` state
1252
1253 check = self._checker.check_output
1254
Tim Peters8485b562004-08-04 18:46:34 +00001255 # Process each example.
Edward Loper2de91ba2004-08-27 02:07:46 +00001256 for examplenum, example in enumerate(test.examples):
1257
Ezio Melotti13925002011-03-16 11:05:33 +02001258 # If REPORT_ONLY_FIRST_FAILURE is set, then suppress
Edward Lopera89f88d2004-08-26 02:45:51 +00001259 # reporting after the first failure.
1260 quiet = (self.optionflags & REPORT_ONLY_FIRST_FAILURE and
1261 failures > 0)
1262
Edward Loper74bca7a2004-08-12 02:27:44 +00001263 # Merge in the example's options.
1264 self.optionflags = original_optionflags
1265 if example.options:
1266 for (optionflag, val) in example.options.items():
1267 if val:
1268 self.optionflags |= optionflag
1269 else:
1270 self.optionflags &= ~optionflag
Tim Peters8485b562004-08-04 18:46:34 +00001271
Thomas Wouters477c8d52006-05-27 19:21:47 +00001272 # If 'SKIP' is set, then skip this example.
1273 if self.optionflags & SKIP:
1274 continue
1275
Tim Peters8485b562004-08-04 18:46:34 +00001276 # Record that we started this example.
1277 tries += 1
Edward Lopera89f88d2004-08-26 02:45:51 +00001278 if not quiet:
1279 self.report_start(out, test, example)
Tim Peters8485b562004-08-04 18:46:34 +00001280
Edward Loper2de91ba2004-08-27 02:07:46 +00001281 # Use a special filename for compile(), so we can retrieve
1282 # the source code during interactive debugging (see
1283 # __patched_linecache_getlines).
1284 filename = '<doctest %s[%d]>' % (test.name, examplenum)
1285
Tim Peters8485b562004-08-04 18:46:34 +00001286 # Run the example in the given context (globs), and record
1287 # any exception that gets raised. (But don't intercept
1288 # keyboard interrupts.)
1289 try:
Tim Peters208ca702004-08-09 04:12:36 +00001290 # Don't blink! This is where the user's code gets run.
Georg Brandl7cae87c2006-09-06 06:51:57 +00001291 exec(compile(example.source, filename, "single",
1292 compileflags, 1), test.globs)
Edward Loper2de91ba2004-08-27 02:07:46 +00001293 self.debugger.set_continue() # ==== Example Finished ====
Tim Peters8485b562004-08-04 18:46:34 +00001294 exception = None
1295 except KeyboardInterrupt:
1296 raise
1297 except:
1298 exception = sys.exc_info()
Edward Loper2de91ba2004-08-27 02:07:46 +00001299 self.debugger.set_continue() # ==== Example Finished ====
Tim Peters8485b562004-08-04 18:46:34 +00001300
Tim Peters208ca702004-08-09 04:12:36 +00001301 got = self._fakeout.getvalue() # the actual output
Tim Peters8485b562004-08-04 18:46:34 +00001302 self._fakeout.truncate(0)
Tim Peters1fbf9c52004-09-04 17:21:02 +00001303 outcome = FAILURE # guilty until proved innocent or insane
Tim Peters8485b562004-08-04 18:46:34 +00001304
1305 # If the example executed without raising any exceptions,
Tim Peters1fbf9c52004-09-04 17:21:02 +00001306 # verify its output.
Tim Peters8485b562004-08-04 18:46:34 +00001307 if exception is None:
Tim Peters1fbf9c52004-09-04 17:21:02 +00001308 if check(example.want, got, self.optionflags):
1309 outcome = SUCCESS
Tim Peters8485b562004-08-04 18:46:34 +00001310
Tim Peters1fbf9c52004-09-04 17:21:02 +00001311 # The example raised an exception: check if it was expected.
Tim Peters8485b562004-08-04 18:46:34 +00001312 else:
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001313 exc_msg = traceback.format_exception_only(*exception[:2])[-1]
Tim Peters1fbf9c52004-09-04 17:21:02 +00001314 if not quiet:
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001315 got += _exception_traceback(exception)
Tim Peters8485b562004-08-04 18:46:34 +00001316
Tim Peters1fbf9c52004-09-04 17:21:02 +00001317 # If `example.exc_msg` is None, then we weren't expecting
1318 # an exception.
Edward Lopera6b68322004-08-26 00:05:43 +00001319 if example.exc_msg is None:
Tim Peters1fbf9c52004-09-04 17:21:02 +00001320 outcome = BOOM
1321
1322 # We expected an exception: see whether it matches.
1323 elif check(example.exc_msg, exc_msg, self.optionflags):
1324 outcome = SUCCESS
1325
1326 # Another chance if they didn't care about the detail.
1327 elif self.optionflags & IGNORE_EXCEPTION_DETAIL:
Nick Coghlan5e76e942010-06-12 13:42:46 +00001328 m1 = re.match(r'(?:[^:]*\.)?([^:]*:)', example.exc_msg)
1329 m2 = re.match(r'(?:[^:]*\.)?([^:]*:)', exc_msg)
1330 if m1 and m2 and check(m1.group(1), m2.group(1),
Tim Peters1fbf9c52004-09-04 17:21:02 +00001331 self.optionflags):
1332 outcome = SUCCESS
1333
1334 # Report the outcome.
1335 if outcome is SUCCESS:
1336 if not quiet:
1337 self.report_success(out, test, example, got)
1338 elif outcome is FAILURE:
1339 if not quiet:
1340 self.report_failure(out, test, example, got)
1341 failures += 1
1342 elif outcome is BOOM:
1343 if not quiet:
1344 self.report_unexpected_exception(out, test, example,
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001345 exception)
Tim Peters1fbf9c52004-09-04 17:21:02 +00001346 failures += 1
1347 else:
1348 assert False, ("unknown outcome", outcome)
Tim Peters8485b562004-08-04 18:46:34 +00001349
R David Murray5a9d7062012-11-21 15:09:21 -05001350 if failures and self.optionflags & FAIL_FAST:
1351 break
1352
Tim Peters8485b562004-08-04 18:46:34 +00001353 # Restore the option flags (in case they were modified)
1354 self.optionflags = original_optionflags
1355
1356 # Record and return the number of failures and tries.
1357 self.__record_outcome(test, failures, tries)
Christian Heimes25bb7832008-01-11 16:17:00 +00001358 return TestResults(failures, tries)
Tim Peters8a7d2d52001-01-16 07:10:57 +00001359
Tim Peters8485b562004-08-04 18:46:34 +00001360 def __record_outcome(self, test, f, t):
1361 """
1362 Record the fact that the given DocTest (`test`) generated `f`
1363 failures out of `t` tried examples.
1364 """
1365 f2, t2 = self._name2ft.get(test.name, (0,0))
1366 self._name2ft[test.name] = (f+f2, t+t2)
1367 self.failures += f
1368 self.tries += t
1369
Edward Loper2de91ba2004-08-27 02:07:46 +00001370 __LINECACHE_FILENAME_RE = re.compile(r'<doctest '
Florent Xiclunad9f57632010-10-14 20:56:20 +00001371 r'(?P<name>.+)'
Edward Loper2de91ba2004-08-27 02:07:46 +00001372 r'\[(?P<examplenum>\d+)\]>$')
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001373 def __patched_linecache_getlines(self, filename, module_globals=None):
Edward Loper2de91ba2004-08-27 02:07:46 +00001374 m = self.__LINECACHE_FILENAME_RE.match(filename)
1375 if m and m.group('name') == self.test.name:
1376 example = self.test.examples[int(m.group('examplenum'))]
Ezio Melottid8b509b2011-09-28 17:37:55 +03001377 return example.source.splitlines(keepends=True)
Edward Loper2de91ba2004-08-27 02:07:46 +00001378 else:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001379 return self.save_linecache_getlines(filename, module_globals)
Edward Loper2de91ba2004-08-27 02:07:46 +00001380
Tim Peters8485b562004-08-04 18:46:34 +00001381 def run(self, test, compileflags=None, out=None, clear_globs=True):
1382 """
1383 Run the examples in `test`, and display the results using the
1384 writer function `out`.
1385
1386 The examples are run in the namespace `test.globs`. If
1387 `clear_globs` is true (the default), then this namespace will
1388 be cleared after the test runs, to help with garbage
1389 collection. If you would like to examine the namespace after
1390 the test completes, then use `clear_globs=False`.
1391
1392 `compileflags` gives the set of flags that should be used by
1393 the Python compiler when running the examples. If not
1394 specified, then it will default to the set of future-import
1395 flags that apply to `globs`.
1396
1397 The output of each example is checked using
1398 `DocTestRunner.check_output`, and the results are formatted by
1399 the `DocTestRunner.report_*` methods.
1400 """
Edward Loper2de91ba2004-08-27 02:07:46 +00001401 self.test = test
1402
Tim Peters8485b562004-08-04 18:46:34 +00001403 if compileflags is None:
1404 compileflags = _extract_future_flags(test.globs)
Jim Fulton356fd192004-08-09 11:34:47 +00001405
Tim Peters6c542b72004-08-09 16:43:36 +00001406 save_stdout = sys.stdout
Tim Peters8485b562004-08-04 18:46:34 +00001407 if out is None:
Florent Xicluna59250852010-02-27 14:21:57 +00001408 encoding = save_stdout.encoding
1409 if encoding is None or encoding.lower() == 'utf-8':
1410 out = save_stdout.write
1411 else:
1412 # Use backslashreplace error handling on write
1413 def out(s):
1414 s = str(s.encode(encoding, 'backslashreplace'), encoding)
1415 save_stdout.write(s)
Tim Peters6c542b72004-08-09 16:43:36 +00001416 sys.stdout = self._fakeout
Tim Peters8485b562004-08-04 18:46:34 +00001417
Edward Loper2de91ba2004-08-27 02:07:46 +00001418 # Patch pdb.set_trace to restore sys.stdout during interactive
1419 # debugging (so it's not still redirected to self._fakeout).
1420 # Note that the interactive output will go to *our*
1421 # save_stdout, even if that's not the real sys.stdout; this
1422 # allows us to write test cases for the set_trace behavior.
Brett Cannon31f59292011-02-21 19:29:56 +00001423 save_trace = sys.gettrace()
Tim Peters6c542b72004-08-09 16:43:36 +00001424 save_set_trace = pdb.set_trace
Edward Loper2de91ba2004-08-27 02:07:46 +00001425 self.debugger = _OutputRedirectingPdb(save_stdout)
1426 self.debugger.reset()
1427 pdb.set_trace = self.debugger.set_trace
1428
1429 # Patch linecache.getlines, so we can see the example's source
1430 # when we're inside the debugger.
1431 self.save_linecache_getlines = linecache.getlines
1432 linecache.getlines = self.__patched_linecache_getlines
1433
Georg Brandl25fbb892010-07-30 09:23:23 +00001434 # Make sure sys.displayhook just prints the value to stdout
1435 save_displayhook = sys.displayhook
1436 sys.displayhook = sys.__displayhook__
1437
Tim Peters8485b562004-08-04 18:46:34 +00001438 try:
Tim Peters8485b562004-08-04 18:46:34 +00001439 return self.__run(test, compileflags, out)
1440 finally:
Tim Peters6c542b72004-08-09 16:43:36 +00001441 sys.stdout = save_stdout
1442 pdb.set_trace = save_set_trace
Brett Cannon31f59292011-02-21 19:29:56 +00001443 sys.settrace(save_trace)
Edward Loper2de91ba2004-08-27 02:07:46 +00001444 linecache.getlines = self.save_linecache_getlines
Georg Brandl25fbb892010-07-30 09:23:23 +00001445 sys.displayhook = save_displayhook
Tim Peters8485b562004-08-04 18:46:34 +00001446 if clear_globs:
1447 test.globs.clear()
Benjamin Petersonfbf66bd2008-07-29 15:55:50 +00001448 import builtins
1449 builtins._ = None
Tim Peters8485b562004-08-04 18:46:34 +00001450
1451 #/////////////////////////////////////////////////////////////////
1452 # Summarization
1453 #/////////////////////////////////////////////////////////////////
Tim Peters8a7d2d52001-01-16 07:10:57 +00001454 def summarize(self, verbose=None):
1455 """
Tim Peters8485b562004-08-04 18:46:34 +00001456 Print a summary of all the test cases that have been run by
1457 this DocTestRunner, and return a tuple `(f, t)`, where `f` is
1458 the total number of failed examples, and `t` is the total
1459 number of tried examples.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001460
Tim Peters8485b562004-08-04 18:46:34 +00001461 The optional `verbose` argument controls how detailed the
1462 summary is. If the verbosity is not specified, then the
1463 DocTestRunner's verbosity is used.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001464 """
Tim Peters8a7d2d52001-01-16 07:10:57 +00001465 if verbose is None:
Tim Peters8485b562004-08-04 18:46:34 +00001466 verbose = self._verbose
Tim Peters8a7d2d52001-01-16 07:10:57 +00001467 notests = []
1468 passed = []
1469 failed = []
1470 totalt = totalf = 0
Tim Peters8485b562004-08-04 18:46:34 +00001471 for x in self._name2ft.items():
Tim Peters8a7d2d52001-01-16 07:10:57 +00001472 name, (f, t) = x
1473 assert f <= t
Tim Peters8485b562004-08-04 18:46:34 +00001474 totalt += t
1475 totalf += f
Tim Peters8a7d2d52001-01-16 07:10:57 +00001476 if t == 0:
1477 notests.append(name)
1478 elif f == 0:
1479 passed.append( (name, t) )
1480 else:
1481 failed.append(x)
1482 if verbose:
1483 if notests:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001484 print(len(notests), "items had no tests:")
Tim Peters8a7d2d52001-01-16 07:10:57 +00001485 notests.sort()
1486 for thing in notests:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001487 print(" ", thing)
Tim Peters8a7d2d52001-01-16 07:10:57 +00001488 if passed:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001489 print(len(passed), "items passed all tests:")
Tim Peters8a7d2d52001-01-16 07:10:57 +00001490 passed.sort()
1491 for thing, count in passed:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001492 print(" %3d tests in %s" % (count, thing))
Tim Peters8a7d2d52001-01-16 07:10:57 +00001493 if failed:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001494 print(self.DIVIDER)
1495 print(len(failed), "items had failures:")
Tim Peters8a7d2d52001-01-16 07:10:57 +00001496 failed.sort()
1497 for thing, (f, t) in failed:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001498 print(" %3d of %3d in %s" % (f, t, thing))
Tim Peters8a7d2d52001-01-16 07:10:57 +00001499 if verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001500 print(totalt, "tests in", len(self._name2ft), "items.")
1501 print(totalt - totalf, "passed and", totalf, "failed.")
Tim Peters8a7d2d52001-01-16 07:10:57 +00001502 if totalf:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001503 print("***Test Failed***", totalf, "failures.")
Tim Peters8a7d2d52001-01-16 07:10:57 +00001504 elif verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001505 print("Test passed.")
Christian Heimes25bb7832008-01-11 16:17:00 +00001506 return TestResults(totalf, totalt)
Tim Peters8a7d2d52001-01-16 07:10:57 +00001507
Tim Peters82076ef2004-09-13 00:52:51 +00001508 #/////////////////////////////////////////////////////////////////
1509 # Backward compatibility cruft to maintain doctest.master.
1510 #/////////////////////////////////////////////////////////////////
1511 def merge(self, other):
1512 d = self._name2ft
1513 for name, (f, t) in other._name2ft.items():
1514 if name in d:
Nick Coghlan38622002008-12-15 12:01:34 +00001515 # Don't print here by default, since doing
1516 # so breaks some of the buildbots
1517 #print("*** DocTestRunner.merge: '" + name + "' in both" \
1518 # " testers; summing outcomes.")
Tim Peters82076ef2004-09-13 00:52:51 +00001519 f2, t2 = d[name]
1520 f = f + f2
1521 t = t + t2
1522 d[name] = f, t
1523
Edward Loper34fcb142004-08-09 02:45:41 +00001524class OutputChecker:
1525 """
1526 A class used to check the whether the actual output from a doctest
1527 example matches the expected output. `OutputChecker` defines two
1528 methods: `check_output`, which compares a given pair of outputs,
1529 and returns true if they match; and `output_difference`, which
1530 returns a string describing the differences between two outputs.
1531 """
Georg Brandl559e5d72008-06-11 18:37:52 +00001532 def _toAscii(self, s):
1533 """
1534 Convert string to hex-escaped ASCII string.
1535 """
1536 return str(s.encode('ASCII', 'backslashreplace'), "ASCII")
1537
Edward Loper34fcb142004-08-09 02:45:41 +00001538 def check_output(self, want, got, optionflags):
1539 """
Edward Loper74bca7a2004-08-12 02:27:44 +00001540 Return True iff the actual output from an example (`got`)
1541 matches the expected output (`want`). These strings are
1542 always considered to match if they are identical; but
1543 depending on what option flags the test runner is using,
1544 several non-exact match types are also possible. See the
1545 documentation for `TestRunner` for more information about
1546 option flags.
Edward Loper34fcb142004-08-09 02:45:41 +00001547 """
Georg Brandl559e5d72008-06-11 18:37:52 +00001548
1549 # If `want` contains hex-escaped character such as "\u1234",
1550 # then `want` is a string of six characters(e.g. [\,u,1,2,3,4]).
1551 # On the other hand, `got` could be an another sequence of
1552 # characters such as [\u1234], so `want` and `got` should
1553 # be folded to hex-escaped ASCII string to compare.
1554 got = self._toAscii(got)
1555 want = self._toAscii(want)
1556
Edward Loper34fcb142004-08-09 02:45:41 +00001557 # Handle the common case first, for efficiency:
1558 # if they're string-identical, always return true.
1559 if got == want:
1560 return True
1561
1562 # The values True and False replaced 1 and 0 as the return
1563 # value for boolean comparisons in Python 2.3.
1564 if not (optionflags & DONT_ACCEPT_TRUE_FOR_1):
1565 if (got,want) == ("True\n", "1\n"):
1566 return True
1567 if (got,want) == ("False\n", "0\n"):
1568 return True
1569
1570 # <BLANKLINE> can be used as a special sequence to signify a
1571 # blank line, unless the DONT_ACCEPT_BLANKLINE flag is used.
1572 if not (optionflags & DONT_ACCEPT_BLANKLINE):
1573 # Replace <BLANKLINE> in want with a blank line.
1574 want = re.sub('(?m)^%s\s*?$' % re.escape(BLANKLINE_MARKER),
1575 '', want)
1576 # If a line in got contains only spaces, then remove the
1577 # spaces.
1578 got = re.sub('(?m)^\s*?$', '', got)
1579 if got == want:
1580 return True
1581
1582 # This flag causes doctest to ignore any differences in the
1583 # contents of whitespace strings. Note that this can be used
Tim Peters3fa8c202004-08-23 21:43:39 +00001584 # in conjunction with the ELLIPSIS flag.
Tim Peters1cf3aa62004-08-19 06:49:33 +00001585 if optionflags & NORMALIZE_WHITESPACE:
Edward Loper34fcb142004-08-09 02:45:41 +00001586 got = ' '.join(got.split())
1587 want = ' '.join(want.split())
1588 if got == want:
1589 return True
1590
1591 # The ELLIPSIS flag says to let the sequence "..." in `want`
Tim Peters26b3ebb2004-08-19 08:10:08 +00001592 # match any substring in `got`.
Tim Peters1cf3aa62004-08-19 06:49:33 +00001593 if optionflags & ELLIPSIS:
Tim Petersb0a04e12004-08-20 02:08:04 +00001594 if _ellipsis_match(want, got):
Edward Loper34fcb142004-08-09 02:45:41 +00001595 return True
1596
1597 # We didn't find any match; return false.
1598 return False
1599
Tim Petersc6cbab02004-08-22 19:43:28 +00001600 # Should we do a fancy diff?
1601 def _do_a_fancy_diff(self, want, got, optionflags):
1602 # Not unless they asked for a fancy diff.
Edward Loper71f55af2004-08-26 01:41:51 +00001603 if not optionflags & (REPORT_UDIFF |
1604 REPORT_CDIFF |
1605 REPORT_NDIFF):
Tim Petersc6cbab02004-08-22 19:43:28 +00001606 return False
Tim Peters5b799c12004-08-26 05:21:59 +00001607
Tim Petersc6cbab02004-08-22 19:43:28 +00001608 # If expected output uses ellipsis, a meaningful fancy diff is
Tim Peters5b799c12004-08-26 05:21:59 +00001609 # too hard ... or maybe not. In two real-life failures Tim saw,
1610 # a diff was a major help anyway, so this is commented out.
1611 # [todo] _ellipsis_match() knows which pieces do and don't match,
1612 # and could be the basis for a kick-ass diff in this case.
1613 ##if optionflags & ELLIPSIS and ELLIPSIS_MARKER in want:
1614 ## return False
1615
Tim Petersc6cbab02004-08-22 19:43:28 +00001616 # ndiff does intraline difference marking, so can be useful even
Tim Peters5b799c12004-08-26 05:21:59 +00001617 # for 1-line differences.
Edward Loper71f55af2004-08-26 01:41:51 +00001618 if optionflags & REPORT_NDIFF:
Tim Petersc6cbab02004-08-22 19:43:28 +00001619 return True
Tim Peters5b799c12004-08-26 05:21:59 +00001620
Tim Petersc6cbab02004-08-22 19:43:28 +00001621 # The other diff types need at least a few lines to be helpful.
1622 return want.count('\n') > 2 and got.count('\n') > 2
1623
Edward Loperca9111e2004-08-26 03:00:24 +00001624 def output_difference(self, example, got, optionflags):
Edward Loper34fcb142004-08-09 02:45:41 +00001625 """
1626 Return a string describing the differences between the
Edward Loperca9111e2004-08-26 03:00:24 +00001627 expected output for a given example (`example`) and the actual
1628 output (`got`). `optionflags` is the set of option flags used
1629 to compare `want` and `got`.
Edward Loper34fcb142004-08-09 02:45:41 +00001630 """
Edward Loperca9111e2004-08-26 03:00:24 +00001631 want = example.want
Edward Loper68ba9a62004-08-12 02:43:49 +00001632 # If <BLANKLINE>s are being used, then replace blank lines
1633 # with <BLANKLINE> in the actual output string.
Edward Loper34fcb142004-08-09 02:45:41 +00001634 if not (optionflags & DONT_ACCEPT_BLANKLINE):
Edward Loper68ba9a62004-08-12 02:43:49 +00001635 got = re.sub('(?m)^[ ]*(?=\n)', BLANKLINE_MARKER, got)
Edward Loper34fcb142004-08-09 02:45:41 +00001636
Tim Peters5b799c12004-08-26 05:21:59 +00001637 # Check if we should use diff.
Tim Petersc6cbab02004-08-22 19:43:28 +00001638 if self._do_a_fancy_diff(want, got, optionflags):
Edward Loper34fcb142004-08-09 02:45:41 +00001639 # Split want & got into lines.
Ezio Melottid8b509b2011-09-28 17:37:55 +03001640 want_lines = want.splitlines(keepends=True)
1641 got_lines = got.splitlines(keepends=True)
Edward Loper34fcb142004-08-09 02:45:41 +00001642 # Use difflib to find their differences.
Edward Loper71f55af2004-08-26 01:41:51 +00001643 if optionflags & REPORT_UDIFF:
Edward Loper56629292004-08-26 01:31:56 +00001644 diff = difflib.unified_diff(want_lines, got_lines, n=2)
1645 diff = list(diff)[2:] # strip the diff header
1646 kind = 'unified diff with -expected +actual'
Edward Loper71f55af2004-08-26 01:41:51 +00001647 elif optionflags & REPORT_CDIFF:
Edward Loper56629292004-08-26 01:31:56 +00001648 diff = difflib.context_diff(want_lines, got_lines, n=2)
1649 diff = list(diff)[2:] # strip the diff header
1650 kind = 'context diff with expected followed by actual'
Edward Loper71f55af2004-08-26 01:41:51 +00001651 elif optionflags & REPORT_NDIFF:
Tim Petersc6cbab02004-08-22 19:43:28 +00001652 engine = difflib.Differ(charjunk=difflib.IS_CHARACTER_JUNK)
1653 diff = list(engine.compare(want_lines, got_lines))
1654 kind = 'ndiff with -expected +actual'
Edward Loper34fcb142004-08-09 02:45:41 +00001655 else:
1656 assert 0, 'Bad diff option'
1657 # Remove trailing whitespace on diff output.
1658 diff = [line.rstrip() + '\n' for line in diff]
Edward Loperaacf0832004-08-26 01:19:50 +00001659 return 'Differences (%s):\n' % kind + _indent(''.join(diff))
Edward Loper34fcb142004-08-09 02:45:41 +00001660
1661 # If we're not using diff, then simply list the expected
1662 # output followed by the actual output.
Edward Loperaacf0832004-08-26 01:19:50 +00001663 if want and got:
1664 return 'Expected:\n%sGot:\n%s' % (_indent(want), _indent(got))
1665 elif want:
1666 return 'Expected:\n%sGot nothing\n' % _indent(want)
1667 elif got:
1668 return 'Expected nothing\nGot:\n%s' % _indent(got)
1669 else:
1670 return 'Expected nothing\nGot nothing\n'
Edward Loper34fcb142004-08-09 02:45:41 +00001671
Tim Peters19397e52004-08-06 22:02:59 +00001672class DocTestFailure(Exception):
1673 """A DocTest example has failed in debugging mode.
1674
1675 The exception instance has variables:
1676
1677 - test: the DocTest object being run
1678
Neal Norwitzc082cb72006-08-29 05:40:08 +00001679 - example: the Example object that failed
Tim Peters19397e52004-08-06 22:02:59 +00001680
1681 - got: the actual output
1682 """
1683 def __init__(self, test, example, got):
1684 self.test = test
1685 self.example = example
1686 self.got = got
1687
1688 def __str__(self):
1689 return str(self.test)
1690
1691class UnexpectedException(Exception):
1692 """A DocTest example has encountered an unexpected exception
1693
1694 The exception instance has variables:
1695
1696 - test: the DocTest object being run
1697
Guido van Rossum6a2a2a02006-08-26 20:37:44 +00001698 - example: the Example object that failed
Tim Peters19397e52004-08-06 22:02:59 +00001699
1700 - exc_info: the exception info
1701 """
1702 def __init__(self, test, example, exc_info):
1703 self.test = test
1704 self.example = example
1705 self.exc_info = exc_info
1706
1707 def __str__(self):
1708 return str(self.test)
Tim Petersd1b78272004-08-07 06:03:09 +00001709
Tim Peters19397e52004-08-06 22:02:59 +00001710class DebugRunner(DocTestRunner):
1711 r"""Run doc tests but raise an exception as soon as there is a failure.
1712
1713 If an unexpected exception occurs, an UnexpectedException is raised.
1714 It contains the test, the example, and the original exception:
1715
1716 >>> runner = DebugRunner(verbose=False)
Edward Lopera1ef6112004-08-09 16:14:41 +00001717 >>> test = DocTestParser().get_doctest('>>> raise KeyError\n42',
1718 ... {}, 'foo', 'foo.py', 0)
Tim Peters19397e52004-08-06 22:02:59 +00001719 >>> try:
1720 ... runner.run(test)
Guido van Rossumb940e112007-01-10 16:19:56 +00001721 ... except UnexpectedException as f:
1722 ... failure = f
Tim Peters19397e52004-08-06 22:02:59 +00001723
1724 >>> failure.test is test
1725 True
1726
1727 >>> failure.example.want
1728 '42\n'
1729
1730 >>> exc_info = failure.exc_info
Collin Winter828f04a2007-08-31 00:04:24 +00001731 >>> raise exc_info[1] # Already has the traceback
Tim Peters19397e52004-08-06 22:02:59 +00001732 Traceback (most recent call last):
1733 ...
1734 KeyError
1735
1736 We wrap the original exception to give the calling application
1737 access to the test and example information.
1738
1739 If the output doesn't match, then a DocTestFailure is raised:
1740
Edward Lopera1ef6112004-08-09 16:14:41 +00001741 >>> test = DocTestParser().get_doctest('''
Tim Peters19397e52004-08-06 22:02:59 +00001742 ... >>> x = 1
1743 ... >>> x
1744 ... 2
1745 ... ''', {}, 'foo', 'foo.py', 0)
1746
1747 >>> try:
1748 ... runner.run(test)
Guido van Rossumb940e112007-01-10 16:19:56 +00001749 ... except DocTestFailure as f:
1750 ... failure = f
Tim Peters19397e52004-08-06 22:02:59 +00001751
1752 DocTestFailure objects provide access to the test:
1753
1754 >>> failure.test is test
1755 True
1756
1757 As well as to the example:
1758
1759 >>> failure.example.want
1760 '2\n'
1761
1762 and the actual output:
1763
1764 >>> failure.got
1765 '1\n'
1766
1767 If a failure or error occurs, the globals are left intact:
1768
1769 >>> del test.globs['__builtins__']
1770 >>> test.globs
1771 {'x': 1}
1772
Edward Lopera1ef6112004-08-09 16:14:41 +00001773 >>> test = DocTestParser().get_doctest('''
Tim Peters19397e52004-08-06 22:02:59 +00001774 ... >>> x = 2
1775 ... >>> raise KeyError
1776 ... ''', {}, 'foo', 'foo.py', 0)
1777
1778 >>> runner.run(test)
1779 Traceback (most recent call last):
1780 ...
Guido van Rossum6a2a2a02006-08-26 20:37:44 +00001781 doctest.UnexpectedException: <DocTest foo from foo.py:0 (2 examples)>
Tim Petersd1b78272004-08-07 06:03:09 +00001782
Tim Peters19397e52004-08-06 22:02:59 +00001783 >>> del test.globs['__builtins__']
1784 >>> test.globs
1785 {'x': 2}
1786
1787 But the globals are cleared if there is no error:
1788
Edward Lopera1ef6112004-08-09 16:14:41 +00001789 >>> test = DocTestParser().get_doctest('''
Tim Peters19397e52004-08-06 22:02:59 +00001790 ... >>> x = 2
1791 ... ''', {}, 'foo', 'foo.py', 0)
1792
1793 >>> runner.run(test)
Christian Heimes25bb7832008-01-11 16:17:00 +00001794 TestResults(failed=0, attempted=1)
Tim Peters19397e52004-08-06 22:02:59 +00001795
1796 >>> test.globs
1797 {}
1798
1799 """
1800
1801 def run(self, test, compileflags=None, out=None, clear_globs=True):
1802 r = DocTestRunner.run(self, test, compileflags, out, False)
1803 if clear_globs:
1804 test.globs.clear()
1805 return r
1806
1807 def report_unexpected_exception(self, out, test, example, exc_info):
1808 raise UnexpectedException(test, example, exc_info)
1809
1810 def report_failure(self, out, test, example, got):
1811 raise DocTestFailure(test, example, got)
1812
Tim Peters8485b562004-08-04 18:46:34 +00001813######################################################################
Edward Loper7c748462004-08-09 02:06:06 +00001814## 6. Test Functions
Tim Peters8485b562004-08-04 18:46:34 +00001815######################################################################
1816# These should be backwards compatible.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001817
Tim Peters82076ef2004-09-13 00:52:51 +00001818# For backward compatibility, a global instance of a DocTestRunner
1819# class, updated by testmod.
1820master = None
1821
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001822def testmod(m=None, name=None, globs=None, verbose=None,
Tim Peters19397e52004-08-06 22:02:59 +00001823 report=True, optionflags=0, extraglobs=None,
Tim Peters958cc892004-09-13 14:53:28 +00001824 raise_on_error=False, exclude_empty=False):
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001825 """m=None, name=None, globs=None, verbose=None, report=True,
1826 optionflags=0, extraglobs=None, raise_on_error=False,
Tim Peters958cc892004-09-13 14:53:28 +00001827 exclude_empty=False
Tim Peters8a7d2d52001-01-16 07:10:57 +00001828
Martin v. Löwis4581cfa2002-11-22 08:23:09 +00001829 Test examples in docstrings in functions and classes reachable
1830 from module m (or the current module if m is not supplied), starting
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001831 with m.__doc__.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001832
1833 Also test examples reachable from dict m.__test__ if it exists and is
Tim Petersc2388a22004-08-10 01:41:28 +00001834 not None. m.__test__ maps names to functions, classes and strings;
Tim Peters8a7d2d52001-01-16 07:10:57 +00001835 function and class docstrings are tested even if the name is private;
1836 strings are tested directly, as if they were docstrings.
1837
1838 Return (#failures, #tests).
1839
Alexander Belopolsky977a6842010-08-16 20:17:07 +00001840 See help(doctest) for an overview.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001841
1842 Optional keyword arg "name" gives the name of the module; by default
1843 use m.__name__.
1844
1845 Optional keyword arg "globs" gives a dict to be used as the globals
1846 when executing examples; by default, use m.__dict__. A copy of this
1847 dict is actually used for each docstring, so that each docstring's
1848 examples start with a clean slate.
1849
Tim Peters8485b562004-08-04 18:46:34 +00001850 Optional keyword arg "extraglobs" gives a dictionary that should be
1851 merged into the globals that are used to execute examples. By
1852 default, no extra globals are used. This is new in 2.4.
1853
Tim Peters8a7d2d52001-01-16 07:10:57 +00001854 Optional keyword arg "verbose" prints lots of stuff if true, prints
1855 only failures if false; by default, it's true iff "-v" is in sys.argv.
1856
Tim Peters8a7d2d52001-01-16 07:10:57 +00001857 Optional keyword arg "report" prints a summary at the end when true,
1858 else prints nothing at the end. In verbose mode, the summary is
1859 detailed, else very brief (in fact, empty if all tests passed).
1860
Tim Peters6ebe61f2003-06-27 20:48:05 +00001861 Optional keyword arg "optionflags" or's together module constants,
Tim Petersf82a9de2004-08-22 20:51:53 +00001862 and defaults to 0. This is new in 2.3. Possible values (see the
1863 docs for details):
Tim Peters6ebe61f2003-06-27 20:48:05 +00001864
1865 DONT_ACCEPT_TRUE_FOR_1
Tim Peters8485b562004-08-04 18:46:34 +00001866 DONT_ACCEPT_BLANKLINE
Tim Peters8485b562004-08-04 18:46:34 +00001867 NORMALIZE_WHITESPACE
Tim Peters8485b562004-08-04 18:46:34 +00001868 ELLIPSIS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001869 SKIP
Edward Loper052d0cd2004-09-19 17:19:33 +00001870 IGNORE_EXCEPTION_DETAIL
Edward Loper71f55af2004-08-26 01:41:51 +00001871 REPORT_UDIFF
1872 REPORT_CDIFF
1873 REPORT_NDIFF
Edward Lopera89f88d2004-08-26 02:45:51 +00001874 REPORT_ONLY_FIRST_FAILURE
Tim Peters19397e52004-08-06 22:02:59 +00001875
1876 Optional keyword arg "raise_on_error" raises an exception on the
1877 first unexpected exception or failure. This allows failures to be
1878 post-mortem debugged.
1879
Tim Peters8a7d2d52001-01-16 07:10:57 +00001880 Advanced tomfoolery: testmod runs methods of a local instance of
1881 class doctest.Tester, then merges the results into (or creates)
1882 global Tester instance doctest.master. Methods of doctest.master
1883 can be called directly too, if you want to do something unusual.
1884 Passing report=0 to testmod is especially useful then, to delay
1885 displaying a summary. Invoke doctest.master.summarize(verbose)
1886 when you're done fiddling.
1887 """
Tim Peters82076ef2004-09-13 00:52:51 +00001888 global master
1889
Tim Peters8485b562004-08-04 18:46:34 +00001890 # If no module was given, then use __main__.
Martin v. Löwis4581cfa2002-11-22 08:23:09 +00001891 if m is None:
Martin v. Löwis4581cfa2002-11-22 08:23:09 +00001892 # DWA - m will still be None if this wasn't invoked from the command
1893 # line, in which case the following TypeError is about as good an error
1894 # as we should expect
1895 m = sys.modules.get('__main__')
1896
Tim Peters8485b562004-08-04 18:46:34 +00001897 # Check that we were actually given a module.
1898 if not inspect.ismodule(m):
Walter Dörwald70a6b492004-02-12 17:35:32 +00001899 raise TypeError("testmod: module required; %r" % (m,))
Tim Peters8485b562004-08-04 18:46:34 +00001900
1901 # If no name was given, then use the module's name.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001902 if name is None:
1903 name = m.__name__
Tim Peters8485b562004-08-04 18:46:34 +00001904
1905 # Find, parse, and run all tests in the given module.
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001906 finder = DocTestFinder(exclude_empty=exclude_empty)
Tim Peters19397e52004-08-06 22:02:59 +00001907
1908 if raise_on_error:
1909 runner = DebugRunner(verbose=verbose, optionflags=optionflags)
1910 else:
1911 runner = DocTestRunner(verbose=verbose, optionflags=optionflags)
1912
Tim Peters8485b562004-08-04 18:46:34 +00001913 for test in finder.find(m, name, globs=globs, extraglobs=extraglobs):
1914 runner.run(test)
1915
Tim Peters8a7d2d52001-01-16 07:10:57 +00001916 if report:
Tim Peters8485b562004-08-04 18:46:34 +00001917 runner.summarize()
Tim Peters8a7d2d52001-01-16 07:10:57 +00001918
Tim Peters82076ef2004-09-13 00:52:51 +00001919 if master is None:
1920 master = runner
1921 else:
1922 master.merge(runner)
1923
Christian Heimes25bb7832008-01-11 16:17:00 +00001924 return TestResults(runner.failures, runner.tries)
Tim Petersdb3756d2003-06-29 05:30:48 +00001925
Edward Loper052d0cd2004-09-19 17:19:33 +00001926def testfile(filename, module_relative=True, name=None, package=None,
1927 globs=None, verbose=None, report=True, optionflags=0,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001928 extraglobs=None, raise_on_error=False, parser=DocTestParser(),
1929 encoding=None):
Edward Loper052d0cd2004-09-19 17:19:33 +00001930 """
1931 Test examples in the given file. Return (#failures, #tests).
1932
1933 Optional keyword arg "module_relative" specifies how filenames
1934 should be interpreted:
1935
1936 - If "module_relative" is True (the default), then "filename"
1937 specifies a module-relative path. By default, this path is
1938 relative to the calling module's directory; but if the
1939 "package" argument is specified, then it is relative to that
1940 package. To ensure os-independence, "filename" should use
1941 "/" characters to separate path segments, and should not
1942 be an absolute path (i.e., it may not begin with "/").
1943
1944 - If "module_relative" is False, then "filename" specifies an
1945 os-specific path. The path may be absolute or relative (to
1946 the current working directory).
1947
Edward Lopera2fc7ec2004-09-21 03:24:24 +00001948 Optional keyword arg "name" gives the name of the test; by default
1949 use the file's basename.
Edward Loper052d0cd2004-09-19 17:19:33 +00001950
1951 Optional keyword argument "package" is a Python package or the
1952 name of a Python package whose directory should be used as the
1953 base directory for a module relative filename. If no package is
1954 specified, then the calling module's directory is used as the base
1955 directory for module relative filenames. It is an error to
1956 specify "package" if "module_relative" is False.
1957
1958 Optional keyword arg "globs" gives a dict to be used as the globals
1959 when executing examples; by default, use {}. A copy of this dict
1960 is actually used for each docstring, so that each docstring's
1961 examples start with a clean slate.
1962
1963 Optional keyword arg "extraglobs" gives a dictionary that should be
1964 merged into the globals that are used to execute examples. By
1965 default, no extra globals are used.
1966
1967 Optional keyword arg "verbose" prints lots of stuff if true, prints
1968 only failures if false; by default, it's true iff "-v" is in sys.argv.
1969
1970 Optional keyword arg "report" prints a summary at the end when true,
1971 else prints nothing at the end. In verbose mode, the summary is
1972 detailed, else very brief (in fact, empty if all tests passed).
1973
1974 Optional keyword arg "optionflags" or's together module constants,
1975 and defaults to 0. Possible values (see the docs for details):
1976
1977 DONT_ACCEPT_TRUE_FOR_1
1978 DONT_ACCEPT_BLANKLINE
1979 NORMALIZE_WHITESPACE
1980 ELLIPSIS
Thomas Wouters477c8d52006-05-27 19:21:47 +00001981 SKIP
Edward Loper052d0cd2004-09-19 17:19:33 +00001982 IGNORE_EXCEPTION_DETAIL
1983 REPORT_UDIFF
1984 REPORT_CDIFF
1985 REPORT_NDIFF
1986 REPORT_ONLY_FIRST_FAILURE
1987
1988 Optional keyword arg "raise_on_error" raises an exception on the
1989 first unexpected exception or failure. This allows failures to be
1990 post-mortem debugged.
1991
Edward Loper498a1862004-09-27 03:42:58 +00001992 Optional keyword arg "parser" specifies a DocTestParser (or
1993 subclass) that should be used to extract tests from the files.
1994
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001995 Optional keyword arg "encoding" specifies an encoding that should
1996 be used to convert the file to unicode.
1997
Edward Loper052d0cd2004-09-19 17:19:33 +00001998 Advanced tomfoolery: testmod runs methods of a local instance of
1999 class doctest.Tester, then merges the results into (or creates)
2000 global Tester instance doctest.master. Methods of doctest.master
2001 can be called directly too, if you want to do something unusual.
2002 Passing report=0 to testmod is especially useful then, to delay
2003 displaying a summary. Invoke doctest.master.summarize(verbose)
2004 when you're done fiddling.
2005 """
2006 global master
2007
2008 if package and not module_relative:
2009 raise ValueError("Package may only be specified for module-"
2010 "relative paths.")
Tim Petersbab3e992004-09-20 19:52:34 +00002011
Edward Loper052d0cd2004-09-19 17:19:33 +00002012 # Relativize the path
Guido van Rossum1b81e7b2007-08-29 03:53:53 +00002013 text, filename = _load_testfile(filename, package, module_relative,
2014 encoding or "utf-8")
Edward Loper052d0cd2004-09-19 17:19:33 +00002015
2016 # If no name was given, then use the file's name.
2017 if name is None:
Edward Lopera2fc7ec2004-09-21 03:24:24 +00002018 name = os.path.basename(filename)
Edward Loper052d0cd2004-09-19 17:19:33 +00002019
2020 # Assemble the globals.
2021 if globs is None:
2022 globs = {}
2023 else:
2024 globs = globs.copy()
2025 if extraglobs is not None:
2026 globs.update(extraglobs)
Raymond Hettinger0f055172009-01-27 10:06:09 +00002027 if '__name__' not in globs:
2028 globs['__name__'] = '__main__'
Edward Loper052d0cd2004-09-19 17:19:33 +00002029
2030 if raise_on_error:
2031 runner = DebugRunner(verbose=verbose, optionflags=optionflags)
2032 else:
2033 runner = DocTestRunner(verbose=verbose, optionflags=optionflags)
2034
2035 # Read the file, convert it to a test, and run it.
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002036 test = parser.get_doctest(text, globs, name, filename, 0)
Edward Loper052d0cd2004-09-19 17:19:33 +00002037 runner.run(test)
2038
2039 if report:
2040 runner.summarize()
2041
2042 if master is None:
2043 master = runner
2044 else:
2045 master.merge(runner)
2046
Christian Heimes25bb7832008-01-11 16:17:00 +00002047 return TestResults(runner.failures, runner.tries)
Edward Loper052d0cd2004-09-19 17:19:33 +00002048
Tim Peters8485b562004-08-04 18:46:34 +00002049def run_docstring_examples(f, globs, verbose=False, name="NoName",
2050 compileflags=None, optionflags=0):
2051 """
2052 Test examples in the given object's docstring (`f`), using `globs`
2053 as globals. Optional argument `name` is used in failure messages.
2054 If the optional argument `verbose` is true, then generate output
2055 even if there are no failures.
Tim Petersdb3756d2003-06-29 05:30:48 +00002056
Tim Peters8485b562004-08-04 18:46:34 +00002057 `compileflags` gives the set of flags that should be used by the
2058 Python compiler when running the examples. If not specified, then
2059 it will default to the set of future-import flags that apply to
2060 `globs`.
Tim Petersdb3756d2003-06-29 05:30:48 +00002061
Tim Peters8485b562004-08-04 18:46:34 +00002062 Optional keyword arg `optionflags` specifies options for the
2063 testing and output. See the documentation for `testmod` for more
2064 information.
2065 """
2066 # Find, parse, and run all tests in the given module.
2067 finder = DocTestFinder(verbose=verbose, recurse=False)
2068 runner = DocTestRunner(verbose=verbose, optionflags=optionflags)
2069 for test in finder.find(f, name, globs=globs):
2070 runner.run(test, compileflags=compileflags)
Tim Petersdb3756d2003-06-29 05:30:48 +00002071
Tim Peters8485b562004-08-04 18:46:34 +00002072######################################################################
Georg Brandl31835852008-05-12 17:38:56 +00002073## 7. Unittest Support
Tim Peters8485b562004-08-04 18:46:34 +00002074######################################################################
Tim Petersdb3756d2003-06-29 05:30:48 +00002075
Jim Fultonf54bad42004-08-28 14:57:56 +00002076_unittest_reportflags = 0
Tim Peters38330fe2004-08-30 16:19:24 +00002077
Jim Fultonf54bad42004-08-28 14:57:56 +00002078def set_unittest_reportflags(flags):
Tim Peters38330fe2004-08-30 16:19:24 +00002079 """Sets the unittest option flags.
Jim Fultonf54bad42004-08-28 14:57:56 +00002080
2081 The old flag is returned so that a runner could restore the old
2082 value if it wished to:
2083
Tim Petersb7e99b62005-03-28 23:50:54 +00002084 >>> import doctest
2085 >>> old = doctest._unittest_reportflags
2086 >>> doctest.set_unittest_reportflags(REPORT_NDIFF |
Jim Fultonf54bad42004-08-28 14:57:56 +00002087 ... REPORT_ONLY_FIRST_FAILURE) == old
2088 True
2089
Jim Fultonf54bad42004-08-28 14:57:56 +00002090 >>> doctest._unittest_reportflags == (REPORT_NDIFF |
2091 ... REPORT_ONLY_FIRST_FAILURE)
2092 True
Tim Petersdf7a2082004-08-29 00:38:17 +00002093
Jim Fultonf54bad42004-08-28 14:57:56 +00002094 Only reporting flags can be set:
2095
Tim Petersb7e99b62005-03-28 23:50:54 +00002096 >>> doctest.set_unittest_reportflags(ELLIPSIS)
Jim Fultonf54bad42004-08-28 14:57:56 +00002097 Traceback (most recent call last):
2098 ...
Tim Peters38330fe2004-08-30 16:19:24 +00002099 ValueError: ('Only reporting flags allowed', 8)
Jim Fultonf54bad42004-08-28 14:57:56 +00002100
Tim Petersb7e99b62005-03-28 23:50:54 +00002101 >>> doctest.set_unittest_reportflags(old) == (REPORT_NDIFF |
Jim Fultonf54bad42004-08-28 14:57:56 +00002102 ... REPORT_ONLY_FIRST_FAILURE)
2103 True
Jim Fultonf54bad42004-08-28 14:57:56 +00002104 """
Jim Fultonf54bad42004-08-28 14:57:56 +00002105 global _unittest_reportflags
Tim Peters38330fe2004-08-30 16:19:24 +00002106
2107 if (flags & REPORTING_FLAGS) != flags:
2108 raise ValueError("Only reporting flags allowed", flags)
Jim Fultonf54bad42004-08-28 14:57:56 +00002109 old = _unittest_reportflags
2110 _unittest_reportflags = flags
2111 return old
Tim Petersdf7a2082004-08-29 00:38:17 +00002112
Jim Fultonf54bad42004-08-28 14:57:56 +00002113
Tim Peters19397e52004-08-06 22:02:59 +00002114class DocTestCase(unittest.TestCase):
Tim Petersdb3756d2003-06-29 05:30:48 +00002115
Edward Loper34fcb142004-08-09 02:45:41 +00002116 def __init__(self, test, optionflags=0, setUp=None, tearDown=None,
2117 checker=None):
Jim Fulton07a349c2004-08-22 14:10:00 +00002118
Jim Fultona643b652004-07-14 19:06:50 +00002119 unittest.TestCase.__init__(self)
Tim Peters19397e52004-08-06 22:02:59 +00002120 self._dt_optionflags = optionflags
Edward Loper34fcb142004-08-09 02:45:41 +00002121 self._dt_checker = checker
Tim Peters19397e52004-08-06 22:02:59 +00002122 self._dt_test = test
2123 self._dt_setUp = setUp
2124 self._dt_tearDown = tearDown
Tim Petersdb3756d2003-06-29 05:30:48 +00002125
Jim Fultona643b652004-07-14 19:06:50 +00002126 def setUp(self):
Jim Fultonf54bad42004-08-28 14:57:56 +00002127 test = self._dt_test
Tim Petersdf7a2082004-08-29 00:38:17 +00002128
Tim Peters19397e52004-08-06 22:02:59 +00002129 if self._dt_setUp is not None:
Jim Fultonf54bad42004-08-28 14:57:56 +00002130 self._dt_setUp(test)
Jim Fultona643b652004-07-14 19:06:50 +00002131
2132 def tearDown(self):
Jim Fultonf54bad42004-08-28 14:57:56 +00002133 test = self._dt_test
2134
Tim Peters19397e52004-08-06 22:02:59 +00002135 if self._dt_tearDown is not None:
Jim Fultonf54bad42004-08-28 14:57:56 +00002136 self._dt_tearDown(test)
2137
2138 test.globs.clear()
Jim Fultona643b652004-07-14 19:06:50 +00002139
2140 def runTest(self):
Tim Peters19397e52004-08-06 22:02:59 +00002141 test = self._dt_test
Jim Fultona643b652004-07-14 19:06:50 +00002142 old = sys.stdout
2143 new = StringIO()
Jim Fultonf54bad42004-08-28 14:57:56 +00002144 optionflags = self._dt_optionflags
Tim Petersdf7a2082004-08-29 00:38:17 +00002145
Tim Peters38330fe2004-08-30 16:19:24 +00002146 if not (optionflags & REPORTING_FLAGS):
Jim Fultonf54bad42004-08-28 14:57:56 +00002147 # The option flags don't include any reporting flags,
2148 # so add the default reporting flags
2149 optionflags |= _unittest_reportflags
Tim Petersdf7a2082004-08-29 00:38:17 +00002150
Jim Fultonf54bad42004-08-28 14:57:56 +00002151 runner = DocTestRunner(optionflags=optionflags,
Edward Loper34fcb142004-08-09 02:45:41 +00002152 checker=self._dt_checker, verbose=False)
Tim Peters19397e52004-08-06 22:02:59 +00002153
Jim Fultona643b652004-07-14 19:06:50 +00002154 try:
Tim Peters19397e52004-08-06 22:02:59 +00002155 runner.DIVIDER = "-"*70
Jim Fultonf54bad42004-08-28 14:57:56 +00002156 failures, tries = runner.run(
2157 test, out=new.write, clear_globs=False)
Jim Fultona643b652004-07-14 19:06:50 +00002158 finally:
2159 sys.stdout = old
2160
2161 if failures:
Tim Peters19397e52004-08-06 22:02:59 +00002162 raise self.failureException(self.format_failure(new.getvalue()))
Tim Peters8485b562004-08-04 18:46:34 +00002163
Tim Peters19397e52004-08-06 22:02:59 +00002164 def format_failure(self, err):
2165 test = self._dt_test
2166 if test.lineno is None:
2167 lineno = 'unknown line number'
2168 else:
Jim Fulton07a349c2004-08-22 14:10:00 +00002169 lineno = '%s' % test.lineno
Tim Peters19397e52004-08-06 22:02:59 +00002170 lname = '.'.join(test.name.split('.')[-1:])
2171 return ('Failed doctest test for %s\n'
2172 ' File "%s", line %s, in %s\n\n%s'
2173 % (test.name, test.filename, lineno, lname, err)
2174 )
2175
2176 def debug(self):
2177 r"""Run the test case without results and without catching exceptions
2178
2179 The unit test framework includes a debug method on test cases
2180 and test suites to support post-mortem debugging. The test code
2181 is run in such a way that errors are not caught. This way a
2182 caller can catch the errors and initiate post-mortem debugging.
2183
2184 The DocTestCase provides a debug method that raises
Ezio Melotti13925002011-03-16 11:05:33 +02002185 UnexpectedException errors if there is an unexpected
Tim Peters19397e52004-08-06 22:02:59 +00002186 exception:
2187
Edward Lopera1ef6112004-08-09 16:14:41 +00002188 >>> test = DocTestParser().get_doctest('>>> raise KeyError\n42',
Tim Peters19397e52004-08-06 22:02:59 +00002189 ... {}, 'foo', 'foo.py', 0)
2190 >>> case = DocTestCase(test)
2191 >>> try:
2192 ... case.debug()
Guido van Rossumb940e112007-01-10 16:19:56 +00002193 ... except UnexpectedException as f:
2194 ... failure = f
Tim Peters19397e52004-08-06 22:02:59 +00002195
2196 The UnexpectedException contains the test, the example, and
2197 the original exception:
2198
2199 >>> failure.test is test
2200 True
2201
2202 >>> failure.example.want
2203 '42\n'
2204
2205 >>> exc_info = failure.exc_info
Collin Winter828f04a2007-08-31 00:04:24 +00002206 >>> raise exc_info[1] # Already has the traceback
Tim Peters19397e52004-08-06 22:02:59 +00002207 Traceback (most recent call last):
2208 ...
2209 KeyError
2210
2211 If the output doesn't match, then a DocTestFailure is raised:
2212
Edward Lopera1ef6112004-08-09 16:14:41 +00002213 >>> test = DocTestParser().get_doctest('''
Tim Peters19397e52004-08-06 22:02:59 +00002214 ... >>> x = 1
2215 ... >>> x
2216 ... 2
2217 ... ''', {}, 'foo', 'foo.py', 0)
2218 >>> case = DocTestCase(test)
2219
2220 >>> try:
2221 ... case.debug()
Guido van Rossumb940e112007-01-10 16:19:56 +00002222 ... except DocTestFailure as f:
2223 ... failure = f
Tim Peters19397e52004-08-06 22:02:59 +00002224
2225 DocTestFailure objects provide access to the test:
2226
2227 >>> failure.test is test
2228 True
2229
2230 As well as to the example:
2231
2232 >>> failure.example.want
2233 '2\n'
2234
2235 and the actual output:
2236
2237 >>> failure.got
2238 '1\n'
2239
2240 """
2241
Jim Fultonf54bad42004-08-28 14:57:56 +00002242 self.setUp()
Edward Loper34fcb142004-08-09 02:45:41 +00002243 runner = DebugRunner(optionflags=self._dt_optionflags,
2244 checker=self._dt_checker, verbose=False)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00002245 runner.run(self._dt_test, clear_globs=False)
Jim Fultonf54bad42004-08-28 14:57:56 +00002246 self.tearDown()
Jim Fultona643b652004-07-14 19:06:50 +00002247
2248 def id(self):
Tim Peters19397e52004-08-06 22:02:59 +00002249 return self._dt_test.name
Jim Fultona643b652004-07-14 19:06:50 +00002250
Antoine Pitrou2bc801c2011-12-18 19:27:45 +01002251 def __eq__(self, other):
2252 if type(self) is not type(other):
2253 return NotImplemented
2254
2255 return self._dt_test == other._dt_test and \
2256 self._dt_optionflags == other._dt_optionflags and \
2257 self._dt_setUp == other._dt_setUp and \
2258 self._dt_tearDown == other._dt_tearDown and \
2259 self._dt_checker == other._dt_checker
2260
2261 def __ne__(self, other):
2262 return not self == other
2263
Antoine Pitrou165b1282011-12-18 20:20:17 +01002264 def __hash__(self):
2265 return hash((self._dt_optionflags, self._dt_setUp, self._dt_tearDown,
2266 self._dt_checker))
2267
Jim Fultona643b652004-07-14 19:06:50 +00002268 def __repr__(self):
Tim Peters19397e52004-08-06 22:02:59 +00002269 name = self._dt_test.name.split('.')
Jim Fultona643b652004-07-14 19:06:50 +00002270 return "%s (%s)" % (name[-1], '.'.join(name[:-1]))
2271
2272 __str__ = __repr__
2273
2274 def shortDescription(self):
Tim Peters19397e52004-08-06 22:02:59 +00002275 return "Doctest: " + self._dt_test.name
Jim Fultona643b652004-07-14 19:06:50 +00002276
R. David Murray378c0cf2010-02-24 01:46:21 +00002277class SkipDocTestCase(DocTestCase):
R David Murraye1121532012-03-21 14:53:42 -04002278 def __init__(self, module):
2279 self.module = module
R. David Murray378c0cf2010-02-24 01:46:21 +00002280 DocTestCase.__init__(self, None)
2281
2282 def setUp(self):
2283 self.skipTest("DocTestSuite will not work with -O2 and above")
2284
2285 def test_skip(self):
2286 pass
2287
2288 def shortDescription(self):
R David Murraye1121532012-03-21 14:53:42 -04002289 return "Skipping tests from %s" % self.module.__name__
2290
2291 __str__ = shortDescription
2292
R. David Murray378c0cf2010-02-24 01:46:21 +00002293
Andrew Svetlov7c1017b2013-08-29 01:24:39 +03002294class _DocTestSuite(unittest.TestSuite):
2295
2296 def _removeTestAtIndex(self, index):
2297 pass
2298
2299
Jim Fultonf54bad42004-08-28 14:57:56 +00002300def DocTestSuite(module=None, globs=None, extraglobs=None, test_finder=None,
2301 **options):
Tim Peters8485b562004-08-04 18:46:34 +00002302 """
Tim Peters75dc5e12004-08-22 17:50:45 +00002303 Convert doctest tests for a module to a unittest test suite.
Jim Fultona643b652004-07-14 19:06:50 +00002304
Tim Peters19397e52004-08-06 22:02:59 +00002305 This converts each documentation string in a module that
2306 contains doctest tests to a unittest test case. If any of the
2307 tests in a doc string fail, then the test case fails. An exception
2308 is raised showing the name of the file containing the test and a
Jim Fultona643b652004-07-14 19:06:50 +00002309 (sometimes approximate) line number.
2310
Tim Peters19397e52004-08-06 22:02:59 +00002311 The `module` argument provides the module to be tested. The argument
Jim Fultona643b652004-07-14 19:06:50 +00002312 can be either a module or a module name.
2313
2314 If no argument is given, the calling module is used.
Jim Fultonf54bad42004-08-28 14:57:56 +00002315
2316 A number of options may be provided as keyword arguments:
2317
Jim Fultonf54bad42004-08-28 14:57:56 +00002318 setUp
Edward Lopera2fc7ec2004-09-21 03:24:24 +00002319 A set-up function. This is called before running the
Jim Fultonf54bad42004-08-28 14:57:56 +00002320 tests in each file. The setUp function will be passed a DocTest
2321 object. The setUp function can access the test globals as the
2322 globs attribute of the test passed.
2323
2324 tearDown
Edward Lopera2fc7ec2004-09-21 03:24:24 +00002325 A tear-down function. This is called after running the
Jim Fultonf54bad42004-08-28 14:57:56 +00002326 tests in each file. The tearDown function will be passed a DocTest
2327 object. The tearDown function can access the test globals as the
2328 globs attribute of the test passed.
2329
2330 globs
2331 A dictionary containing initial global variables for the tests.
2332
2333 optionflags
2334 A set of doctest option flags expressed as an integer.
Jim Fultona643b652004-07-14 19:06:50 +00002335 """
Jim Fultona643b652004-07-14 19:06:50 +00002336
Tim Peters8485b562004-08-04 18:46:34 +00002337 if test_finder is None:
2338 test_finder = DocTestFinder()
Tim Peters8485b562004-08-04 18:46:34 +00002339
Tim Peters19397e52004-08-06 22:02:59 +00002340 module = _normalize_module(module)
2341 tests = test_finder.find(module, globs=globs, extraglobs=extraglobs)
R. David Murray378c0cf2010-02-24 01:46:21 +00002342
2343 if not tests and sys.flags.optimize >=2:
2344 # Skip doctests when running with -O2
Andrew Svetlov7c1017b2013-08-29 01:24:39 +03002345 suite = _DocTestSuite()
R David Murraye1121532012-03-21 14:53:42 -04002346 suite.addTest(SkipDocTestCase(module))
R. David Murray378c0cf2010-02-24 01:46:21 +00002347 return suite
2348 elif not tests:
Jim Fultonf54bad42004-08-28 14:57:56 +00002349 # Why do we want to do this? Because it reveals a bug that might
2350 # otherwise be hidden.
R David Murray5abd76a2012-09-10 10:15:58 -04002351 # It is probably a bug that this exception is not also raised if the
2352 # number of doctest examples in tests is zero (i.e. if no doctest
2353 # examples were found). However, we should probably not be raising
2354 # an exception at all here, though it is too late to make this change
2355 # for a maintenance release. See also issue #14649.
2356 raise ValueError(module, "has no docstrings")
Tim Petersdb3756d2003-06-29 05:30:48 +00002357
2358 tests.sort()
Andrew Svetlov7c1017b2013-08-29 01:24:39 +03002359 suite = _DocTestSuite()
R. David Murray378c0cf2010-02-24 01:46:21 +00002360
Tim Peters8485b562004-08-04 18:46:34 +00002361 for test in tests:
Tim Peters19397e52004-08-06 22:02:59 +00002362 if len(test.examples) == 0:
2363 continue
Tim Peters8485b562004-08-04 18:46:34 +00002364 if not test.filename:
Tim Petersdb3756d2003-06-29 05:30:48 +00002365 filename = module.__file__
Jim Fulton07a349c2004-08-22 14:10:00 +00002366 if filename[-4:] in (".pyc", ".pyo"):
Tim Petersdb3756d2003-06-29 05:30:48 +00002367 filename = filename[:-1]
Tim Peters8485b562004-08-04 18:46:34 +00002368 test.filename = filename
Jim Fultonf54bad42004-08-28 14:57:56 +00002369 suite.addTest(DocTestCase(test, **options))
Tim Peters19397e52004-08-06 22:02:59 +00002370
2371 return suite
2372
2373class DocFileCase(DocTestCase):
2374
2375 def id(self):
2376 return '_'.join(self._dt_test.name.split('.'))
2377
2378 def __repr__(self):
2379 return self._dt_test.filename
2380 __str__ = __repr__
2381
2382 def format_failure(self, err):
2383 return ('Failed doctest test for %s\n File "%s", line 0\n\n%s'
2384 % (self._dt_test.name, self._dt_test.filename, err)
2385 )
2386
Edward Loper052d0cd2004-09-19 17:19:33 +00002387def DocFileTest(path, module_relative=True, package=None,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00002388 globs=None, parser=DocTestParser(),
2389 encoding=None, **options):
Tim Peters19397e52004-08-06 22:02:59 +00002390 if globs is None:
2391 globs = {}
Fred Drake7c404a42004-12-21 23:46:34 +00002392 else:
2393 globs = globs.copy()
Tim Peters19397e52004-08-06 22:02:59 +00002394
Edward Loper052d0cd2004-09-19 17:19:33 +00002395 if package and not module_relative:
2396 raise ValueError("Package may only be specified for module-"
2397 "relative paths.")
Tim Petersbab3e992004-09-20 19:52:34 +00002398
Edward Loper052d0cd2004-09-19 17:19:33 +00002399 # Relativize the path.
Guido van Rossum1b81e7b2007-08-29 03:53:53 +00002400 doc, path = _load_testfile(path, package, module_relative,
2401 encoding or "utf-8")
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002402
Fred Drake7c404a42004-12-21 23:46:34 +00002403 if "__file__" not in globs:
2404 globs["__file__"] = path
Tim Peters19397e52004-08-06 22:02:59 +00002405
Edward Loper052d0cd2004-09-19 17:19:33 +00002406 # Find the file and read it.
Edward Lopera2fc7ec2004-09-21 03:24:24 +00002407 name = os.path.basename(path)
Edward Loper052d0cd2004-09-19 17:19:33 +00002408
2409 # Convert it to a test, and wrap it in a DocFileCase.
Edward Loper498a1862004-09-27 03:42:58 +00002410 test = parser.get_doctest(doc, globs, name, path, 0)
Jim Fultonf54bad42004-08-28 14:57:56 +00002411 return DocFileCase(test, **options)
Tim Peters19397e52004-08-06 22:02:59 +00002412
2413def DocFileSuite(*paths, **kw):
Edward Loper052d0cd2004-09-19 17:19:33 +00002414 """A unittest suite for one or more doctest files.
Tim Petersbab3e992004-09-20 19:52:34 +00002415
Edward Loper052d0cd2004-09-19 17:19:33 +00002416 The path to each doctest file is given as a string; the
2417 interpretation of that string depends on the keyword argument
2418 "module_relative".
Tim Peters19397e52004-08-06 22:02:59 +00002419
2420 A number of options may be provided as keyword arguments:
2421
Edward Loper052d0cd2004-09-19 17:19:33 +00002422 module_relative
2423 If "module_relative" is True, then the given file paths are
2424 interpreted as os-independent module-relative paths. By
2425 default, these paths are relative to the calling module's
2426 directory; but if the "package" argument is specified, then
2427 they are relative to that package. To ensure os-independence,
2428 "filename" should use "/" characters to separate path
2429 segments, and may not be an absolute path (i.e., it may not
2430 begin with "/").
Tim Petersbab3e992004-09-20 19:52:34 +00002431
Edward Loper052d0cd2004-09-19 17:19:33 +00002432 If "module_relative" is False, then the given file paths are
2433 interpreted as os-specific paths. These paths may be absolute
2434 or relative (to the current working directory).
2435
Tim Peters19397e52004-08-06 22:02:59 +00002436 package
Edward Loper052d0cd2004-09-19 17:19:33 +00002437 A Python package or the name of a Python package whose directory
2438 should be used as the base directory for module relative paths.
2439 If "package" is not specified, then the calling module's
2440 directory is used as the base directory for module relative
2441 filenames. It is an error to specify "package" if
2442 "module_relative" is False.
Tim Peters19397e52004-08-06 22:02:59 +00002443
2444 setUp
Edward Lopera2fc7ec2004-09-21 03:24:24 +00002445 A set-up function. This is called before running the
Jim Fultonf54bad42004-08-28 14:57:56 +00002446 tests in each file. The setUp function will be passed a DocTest
2447 object. The setUp function can access the test globals as the
2448 globs attribute of the test passed.
Tim Peters19397e52004-08-06 22:02:59 +00002449
2450 tearDown
Edward Lopera2fc7ec2004-09-21 03:24:24 +00002451 A tear-down function. This is called after running the
Jim Fultonf54bad42004-08-28 14:57:56 +00002452 tests in each file. The tearDown function will be passed a DocTest
2453 object. The tearDown function can access the test globals as the
2454 globs attribute of the test passed.
Tim Peters19397e52004-08-06 22:02:59 +00002455
2456 globs
2457 A dictionary containing initial global variables for the tests.
Jim Fultonf54bad42004-08-28 14:57:56 +00002458
2459 optionflags
Edward Loper498a1862004-09-27 03:42:58 +00002460 A set of doctest option flags expressed as an integer.
2461
2462 parser
2463 A DocTestParser (or subclass) that should be used to extract
2464 tests from the files.
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00002465
2466 encoding
2467 An encoding that will be used to convert the files to unicode.
Tim Peters19397e52004-08-06 22:02:59 +00002468 """
Andrew Svetlov7c1017b2013-08-29 01:24:39 +03002469 suite = _DocTestSuite()
Tim Peters19397e52004-08-06 22:02:59 +00002470
2471 # We do this here so that _normalize_module is called at the right
2472 # level. If it were called in DocFileTest, then this function
2473 # would be the caller and we might guess the package incorrectly.
Edward Loper052d0cd2004-09-19 17:19:33 +00002474 if kw.get('module_relative', True):
2475 kw['package'] = _normalize_module(kw.get('package'))
Tim Peters19397e52004-08-06 22:02:59 +00002476
2477 for path in paths:
2478 suite.addTest(DocFileTest(path, **kw))
Jim Fultona643b652004-07-14 19:06:50 +00002479
Tim Petersdb3756d2003-06-29 05:30:48 +00002480 return suite
2481
Tim Peters8485b562004-08-04 18:46:34 +00002482######################################################################
Georg Brandl31835852008-05-12 17:38:56 +00002483## 8. Debugging Support
Tim Peters8485b562004-08-04 18:46:34 +00002484######################################################################
Jim Fultona643b652004-07-14 19:06:50 +00002485
Tim Peters19397e52004-08-06 22:02:59 +00002486def script_from_examples(s):
2487 r"""Extract script from text with examples.
2488
2489 Converts text with examples to a Python script. Example input is
2490 converted to regular code. Example output and all other words
2491 are converted to comments:
2492
2493 >>> text = '''
2494 ... Here are examples of simple math.
2495 ...
2496 ... Python has super accurate integer addition
2497 ...
2498 ... >>> 2 + 2
2499 ... 5
2500 ...
2501 ... And very friendly error messages:
2502 ...
2503 ... >>> 1/0
2504 ... To Infinity
2505 ... And
2506 ... Beyond
2507 ...
2508 ... You can use logic if you want:
2509 ...
2510 ... >>> if 0:
2511 ... ... blah
2512 ... ... blah
2513 ... ...
2514 ...
2515 ... Ho hum
2516 ... '''
2517
Guido van Rossum7131f842007-02-09 20:13:25 +00002518 >>> print(script_from_examples(text))
Edward Lopera5db6002004-08-12 02:41:30 +00002519 # Here are examples of simple math.
Tim Peters19397e52004-08-06 22:02:59 +00002520 #
Edward Lopera5db6002004-08-12 02:41:30 +00002521 # Python has super accurate integer addition
Tim Peters19397e52004-08-06 22:02:59 +00002522 #
2523 2 + 2
2524 # Expected:
Edward Lopera5db6002004-08-12 02:41:30 +00002525 ## 5
Tim Peters19397e52004-08-06 22:02:59 +00002526 #
Edward Lopera5db6002004-08-12 02:41:30 +00002527 # And very friendly error messages:
Tim Peters19397e52004-08-06 22:02:59 +00002528 #
2529 1/0
2530 # Expected:
Edward Lopera5db6002004-08-12 02:41:30 +00002531 ## To Infinity
2532 ## And
2533 ## Beyond
Tim Peters19397e52004-08-06 22:02:59 +00002534 #
Edward Lopera5db6002004-08-12 02:41:30 +00002535 # You can use logic if you want:
Tim Peters19397e52004-08-06 22:02:59 +00002536 #
2537 if 0:
2538 blah
2539 blah
Tim Peters19397e52004-08-06 22:02:59 +00002540 #
Edward Lopera5db6002004-08-12 02:41:30 +00002541 # Ho hum
Georg Brandlecf93c72005-06-26 23:09:51 +00002542 <BLANKLINE>
Tim Peters19397e52004-08-06 22:02:59 +00002543 """
Edward Loper00f8da72004-08-26 18:05:07 +00002544 output = []
2545 for piece in DocTestParser().parse(s):
2546 if isinstance(piece, Example):
2547 # Add the example's source code (strip trailing NL)
2548 output.append(piece.source[:-1])
2549 # Add the expected output:
2550 want = piece.want
2551 if want:
2552 output.append('# Expected:')
2553 output += ['## '+l for l in want.split('\n')[:-1]]
2554 else:
2555 # Add non-example text.
2556 output += [_comment_line(l)
2557 for l in piece.split('\n')[:-1]]
Tim Peters19397e52004-08-06 22:02:59 +00002558
Edward Loper00f8da72004-08-26 18:05:07 +00002559 # Trim junk on both ends.
2560 while output and output[-1] == '#':
2561 output.pop()
2562 while output and output[0] == '#':
2563 output.pop(0)
2564 # Combine the output, and return it.
Georg Brandl1f149642005-06-26 22:22:31 +00002565 # Add a courtesy newline to prevent exec from choking (see bug #1172785)
2566 return '\n'.join(output) + '\n'
Tim Petersdb3756d2003-06-29 05:30:48 +00002567
2568def testsource(module, name):
Tim Peters19397e52004-08-06 22:02:59 +00002569 """Extract the test sources from a doctest docstring as a script.
Tim Petersdb3756d2003-06-29 05:30:48 +00002570
2571 Provide the module (or dotted name of the module) containing the
Jim Fultona643b652004-07-14 19:06:50 +00002572 test to be debugged and the name (within the module) of the object
2573 with the doc string with tests to be debugged.
Tim Petersdb3756d2003-06-29 05:30:48 +00002574 """
Tim Peters8485b562004-08-04 18:46:34 +00002575 module = _normalize_module(module)
2576 tests = DocTestFinder().find(module)
2577 test = [t for t in tests if t.name == name]
Tim Petersdb3756d2003-06-29 05:30:48 +00002578 if not test:
2579 raise ValueError(name, "not found in tests")
2580 test = test[0]
Tim Peters19397e52004-08-06 22:02:59 +00002581 testsrc = script_from_examples(test.docstring)
Jim Fultona643b652004-07-14 19:06:50 +00002582 return testsrc
Tim Petersdb3756d2003-06-29 05:30:48 +00002583
Jim Fultona643b652004-07-14 19:06:50 +00002584def debug_src(src, pm=False, globs=None):
Tim Peters19397e52004-08-06 22:02:59 +00002585 """Debug a single doctest docstring, in argument `src`'"""
2586 testsrc = script_from_examples(src)
Tim Peters8485b562004-08-04 18:46:34 +00002587 debug_script(testsrc, pm, globs)
Tim Petersdb3756d2003-06-29 05:30:48 +00002588
Jim Fultona643b652004-07-14 19:06:50 +00002589def debug_script(src, pm=False, globs=None):
Tim Peters19397e52004-08-06 22:02:59 +00002590 "Debug a test script. `src` is the script, as a string."
Tim Petersdb3756d2003-06-29 05:30:48 +00002591 import pdb
Tim Petersdb3756d2003-06-29 05:30:48 +00002592
Victor Stinner12b8d142011-06-30 17:35:55 +02002593 if globs:
2594 globs = globs.copy()
2595 else:
2596 globs = {}
Tim Peters8485b562004-08-04 18:46:34 +00002597
Victor Stinner12b8d142011-06-30 17:35:55 +02002598 if pm:
2599 try:
2600 exec(src, globs, globs)
2601 except:
2602 print(sys.exc_info()[1])
2603 p = pdb.Pdb(nosigint=True)
2604 p.reset()
2605 p.interaction(None, sys.exc_info()[2])
2606 else:
2607 pdb.Pdb(nosigint=True).run("exec(%r)" % src, globs, globs)
Tim Petersdb3756d2003-06-29 05:30:48 +00002608
Jim Fultona643b652004-07-14 19:06:50 +00002609def debug(module, name, pm=False):
Tim Peters19397e52004-08-06 22:02:59 +00002610 """Debug a single doctest docstring.
Jim Fultona643b652004-07-14 19:06:50 +00002611
2612 Provide the module (or dotted name of the module) containing the
2613 test to be debugged and the name (within the module) of the object
Tim Peters19397e52004-08-06 22:02:59 +00002614 with the docstring with tests to be debugged.
Jim Fultona643b652004-07-14 19:06:50 +00002615 """
Tim Peters8485b562004-08-04 18:46:34 +00002616 module = _normalize_module(module)
Jim Fultona643b652004-07-14 19:06:50 +00002617 testsrc = testsource(module, name)
2618 debug_script(testsrc, pm, module.__dict__)
2619
Tim Peters8485b562004-08-04 18:46:34 +00002620######################################################################
Georg Brandl31835852008-05-12 17:38:56 +00002621## 9. Example Usage
Tim Peters8485b562004-08-04 18:46:34 +00002622######################################################################
Tim Peters8a7d2d52001-01-16 07:10:57 +00002623class _TestClass:
2624 """
2625 A pointless class, for sanity-checking of docstring testing.
2626
2627 Methods:
2628 square()
2629 get()
2630
2631 >>> _TestClass(13).get() + _TestClass(-12).get()
2632 1
2633 >>> hex(_TestClass(13).square().get())
2634 '0xa9'
2635 """
2636
2637 def __init__(self, val):
2638 """val -> _TestClass object with associated value val.
2639
2640 >>> t = _TestClass(123)
Guido van Rossum7131f842007-02-09 20:13:25 +00002641 >>> print(t.get())
Tim Peters8a7d2d52001-01-16 07:10:57 +00002642 123
2643 """
2644
2645 self.val = val
2646
2647 def square(self):
2648 """square() -> square TestClass's associated value
2649
2650 >>> _TestClass(13).square().get()
2651 169
2652 """
2653
2654 self.val = self.val ** 2
2655 return self
2656
2657 def get(self):
2658 """get() -> return TestClass's associated value.
2659
2660 >>> x = _TestClass(-42)
Guido van Rossum7131f842007-02-09 20:13:25 +00002661 >>> print(x.get())
Tim Peters8a7d2d52001-01-16 07:10:57 +00002662 -42
2663 """
2664
2665 return self.val
2666
2667__test__ = {"_TestClass": _TestClass,
2668 "string": r"""
2669 Example of a string object, searched as-is.
2670 >>> x = 1; y = 2
2671 >>> x + y, x * y
2672 (3, 2)
Tim Peters6ebe61f2003-06-27 20:48:05 +00002673 """,
Tim Peters3fa8c202004-08-23 21:43:39 +00002674
Tim Peters6ebe61f2003-06-27 20:48:05 +00002675 "bool-int equivalence": r"""
2676 In 2.2, boolean expressions displayed
2677 0 or 1. By default, we still accept
2678 them. This can be disabled by passing
2679 DONT_ACCEPT_TRUE_FOR_1 to the new
2680 optionflags argument.
2681 >>> 4 == 4
2682 1
2683 >>> 4 == 4
2684 True
2685 >>> 4 > 4
2686 0
2687 >>> 4 > 4
2688 False
2689 """,
Tim Peters3fa8c202004-08-23 21:43:39 +00002690
Tim Peters8485b562004-08-04 18:46:34 +00002691 "blank lines": r"""
Tim Peters3fa8c202004-08-23 21:43:39 +00002692 Blank lines can be marked with <BLANKLINE>:
Guido van Rossum7131f842007-02-09 20:13:25 +00002693 >>> print('foo\n\nbar\n')
Tim Peters3fa8c202004-08-23 21:43:39 +00002694 foo
2695 <BLANKLINE>
2696 bar
2697 <BLANKLINE>
Tim Peters8485b562004-08-04 18:46:34 +00002698 """,
Tim Peters3fa8c202004-08-23 21:43:39 +00002699
2700 "ellipsis": r"""
2701 If the ellipsis flag is used, then '...' can be used to
2702 elide substrings in the desired output:
Guido van Rossum805365e2007-05-07 22:24:25 +00002703 >>> print(list(range(1000))) #doctest: +ELLIPSIS
Tim Peters3fa8c202004-08-23 21:43:39 +00002704 [0, 1, 2, ..., 999]
2705 """,
2706
2707 "whitespace normalization": r"""
2708 If the whitespace normalization flag is used, then
2709 differences in whitespace are ignored.
Guido van Rossum805365e2007-05-07 22:24:25 +00002710 >>> print(list(range(30))) #doctest: +NORMALIZE_WHITESPACE
Tim Peters3fa8c202004-08-23 21:43:39 +00002711 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
2712 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26,
2713 27, 28, 29]
2714 """,
2715 }
Tim Peters8485b562004-08-04 18:46:34 +00002716
R. David Murray445448c2009-12-20 17:28:31 +00002717
Tim Peters8a7d2d52001-01-16 07:10:57 +00002718def _test():
R David Murray5707d502013-06-23 14:24:13 -04002719 parser = argparse.ArgumentParser(description="doctest runner")
2720 parser.add_argument('-v', '--verbose', action='store_true', default=False,
2721 help='print very verbose output for all tests')
2722 parser.add_argument('-o', '--option', action='append',
2723 choices=OPTIONFLAGS_BY_NAME.keys(), default=[],
2724 help=('specify a doctest option flag to apply'
2725 ' to the test run; may be specified more'
2726 ' than once to apply multiple options'))
2727 parser.add_argument('-f', '--fail-fast', action='store_true',
2728 help=('stop running tests after first failure (this'
2729 ' is a shorthand for -o FAIL_FAST, and is'
2730 ' in addition to any other -o options)'))
2731 parser.add_argument('file', nargs='+',
2732 help='file containing the tests to run')
2733 args = parser.parse_args()
2734 testfiles = args.file
2735 # Verbose used to be handled by the "inspect argv" magic in DocTestRunner,
2736 # but since we are using argparse we are passing it manually now.
2737 verbose = args.verbose
2738 options = 0
2739 for option in args.option:
2740 options |= OPTIONFLAGS_BY_NAME[option]
2741 if args.fail_fast:
2742 options |= FAIL_FAST
R. David Murray445448c2009-12-20 17:28:31 +00002743 for filename in testfiles:
2744 if filename.endswith(".py"):
2745 # It is a module -- insert its dir into sys.path and try to
2746 # import it. If it is part of a package, that possibly
2747 # won't work because of package imports.
2748 dirname, filename = os.path.split(filename)
2749 sys.path.insert(0, dirname)
2750 m = __import__(filename[:-3])
2751 del sys.path[0]
R David Murray5707d502013-06-23 14:24:13 -04002752 failures, _ = testmod(m, verbose=verbose, optionflags=options)
R. David Murray445448c2009-12-20 17:28:31 +00002753 else:
R David Murray5707d502013-06-23 14:24:13 -04002754 failures, _ = testfile(filename, module_relative=False,
2755 verbose=verbose, optionflags=options)
R. David Murray445448c2009-12-20 17:28:31 +00002756 if failures:
2757 return 1
Christian Heimes895627f2007-12-08 17:28:33 +00002758 return 0
Tim Peters8a7d2d52001-01-16 07:10:57 +00002759
R. David Murray445448c2009-12-20 17:28:31 +00002760
Tim Peters8a7d2d52001-01-16 07:10:57 +00002761if __name__ == "__main__":
Christian Heimes895627f2007-12-08 17:28:33 +00002762 sys.exit(_test())