blob: c6677f737b815a24ea6b6f5138ba529eb29d4de5 [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',
Tim Peters1fbf9c52004-09-04 17:21:02 +000057 'IGNORE_EXCEPTION_DETAIL',
Tim Petersba601962004-09-04 15:04:06 +000058 'COMPARISON_FLAGS',
Edward Loper71f55af2004-08-26 01:41:51 +000059 'REPORT_UDIFF',
60 'REPORT_CDIFF',
61 'REPORT_NDIFF',
Jim Fultonf54bad42004-08-28 14:57:56 +000062 'REPORT_ONLY_FIRST_FAILURE',
Tim Petersba601962004-09-04 15:04:06 +000063 'REPORTING_FLAGS',
Edward Loperb7503ff2004-08-19 19:19:03 +000064 # 1. Utility Functions
Tim Peters8485b562004-08-04 18:46:34 +000065 'is_private',
Edward Loperb7503ff2004-08-19 19:19:03 +000066 # 2. Example & DocTest
Tim Peters8485b562004-08-04 18:46:34 +000067 'Example',
68 'DocTest',
Edward Loperb7503ff2004-08-19 19:19:03 +000069 # 3. Doctest Parser
70 'DocTestParser',
71 # 4. Doctest Finder
Tim Peters8485b562004-08-04 18:46:34 +000072 'DocTestFinder',
Edward Loperb7503ff2004-08-19 19:19:03 +000073 # 5. Doctest Runner
Tim Peters8485b562004-08-04 18:46:34 +000074 'DocTestRunner',
Edward Loperb7503ff2004-08-19 19:19:03 +000075 'OutputChecker',
76 'DocTestFailure',
77 'UnexpectedException',
78 'DebugRunner',
79 # 6. Test Functions
Tim Peters4fd9e2f2001-08-18 00:05:50 +000080 'testmod',
Edward Loper052d0cd2004-09-19 17:19:33 +000081 'testfile',
Tim Peters4fd9e2f2001-08-18 00:05:50 +000082 'run_docstring_examples',
Edward Loperb7503ff2004-08-19 19:19:03 +000083 # 7. Tester
Tim Peters4fd9e2f2001-08-18 00:05:50 +000084 'Tester',
Edward Loperb7503ff2004-08-19 19:19:03 +000085 # 8. Unittest Support
Tim Petersdb3756d2003-06-29 05:30:48 +000086 'DocTestSuite',
Edward Loperb7503ff2004-08-19 19:19:03 +000087 'DocFileSuite',
Tim Peters9d02a7c2004-09-26 01:50:24 +000088 'set_unittest_reportflags',
Edward Loperb7503ff2004-08-19 19:19:03 +000089 # 9. Debugging Support
90 'script_from_examples',
Tim Petersdb3756d2003-06-29 05:30:48 +000091 'testsource',
Edward Loperb7503ff2004-08-19 19:19:03 +000092 'debug_src',
Tim Petersdb3756d2003-06-29 05:30:48 +000093 'debug',
Tim Peters4fd9e2f2001-08-18 00:05:50 +000094]
Tim Peters8a7d2d52001-01-16 07:10:57 +000095
Tim Peters4fd9e2f2001-08-18 00:05:50 +000096import __future__
Tim Peters8a7d2d52001-01-16 07:10:57 +000097
Tim Peters19397e52004-08-06 22:02:59 +000098import sys, traceback, inspect, linecache, os, re, types
Jim Fulton356fd192004-08-09 11:34:47 +000099import unittest, difflib, pdb, tempfile
Tim Petersf727c6c2004-08-08 01:48:59 +0000100import warnings
Tim Peters8485b562004-08-04 18:46:34 +0000101from StringIO import StringIO
Tim Peters7402f792001-10-02 03:53:41 +0000102
Tim Petersdd50cb72004-08-23 22:42:55 +0000103# Don't whine about the deprecated is_private function in this
104# module's tests.
105warnings.filterwarnings("ignore", "is_private", DeprecationWarning,
106 __name__, 0)
107
Tim Peters19397e52004-08-06 22:02:59 +0000108# There are 4 basic classes:
109# - Example: a <source, want> pair, plus an intra-docstring line number.
110# - DocTest: a collection of examples, parsed from a docstring, plus
111# info about where the docstring came from (name, filename, lineno).
112# - DocTestFinder: extracts DocTests from a given object's docstring and
113# its contained objects' docstrings.
114# - DocTestRunner: runs DocTest cases, and accumulates statistics.
115#
116# So the basic picture is:
117#
118# list of:
119# +------+ +---------+ +-------+
120# |object| --DocTestFinder-> | DocTest | --DocTestRunner-> |results|
121# +------+ +---------+ +-------+
122# | Example |
123# | ... |
124# | Example |
125# +---------+
126
Edward Loperac20f572004-08-12 02:02:24 +0000127# Option constants.
Tim Peters38330fe2004-08-30 16:19:24 +0000128
Edward Loperac20f572004-08-12 02:02:24 +0000129OPTIONFLAGS_BY_NAME = {}
130def register_optionflag(name):
131 flag = 1 << len(OPTIONFLAGS_BY_NAME)
132 OPTIONFLAGS_BY_NAME[name] = flag
133 return flag
134
135DONT_ACCEPT_TRUE_FOR_1 = register_optionflag('DONT_ACCEPT_TRUE_FOR_1')
136DONT_ACCEPT_BLANKLINE = register_optionflag('DONT_ACCEPT_BLANKLINE')
137NORMALIZE_WHITESPACE = register_optionflag('NORMALIZE_WHITESPACE')
138ELLIPSIS = register_optionflag('ELLIPSIS')
Tim Peters1fbf9c52004-09-04 17:21:02 +0000139IGNORE_EXCEPTION_DETAIL = register_optionflag('IGNORE_EXCEPTION_DETAIL')
Tim Peters38330fe2004-08-30 16:19:24 +0000140
141COMPARISON_FLAGS = (DONT_ACCEPT_TRUE_FOR_1 |
142 DONT_ACCEPT_BLANKLINE |
143 NORMALIZE_WHITESPACE |
Tim Peters1fbf9c52004-09-04 17:21:02 +0000144 ELLIPSIS |
Edward Loper7d88a582004-09-28 05:50:57 +0000145 IGNORE_EXCEPTION_DETAIL)
Tim Peters38330fe2004-08-30 16:19:24 +0000146
Edward Loper71f55af2004-08-26 01:41:51 +0000147REPORT_UDIFF = register_optionflag('REPORT_UDIFF')
148REPORT_CDIFF = register_optionflag('REPORT_CDIFF')
149REPORT_NDIFF = register_optionflag('REPORT_NDIFF')
Edward Lopera89f88d2004-08-26 02:45:51 +0000150REPORT_ONLY_FIRST_FAILURE = register_optionflag('REPORT_ONLY_FIRST_FAILURE')
Edward Loperac20f572004-08-12 02:02:24 +0000151
Tim Peters38330fe2004-08-30 16:19:24 +0000152REPORTING_FLAGS = (REPORT_UDIFF |
153 REPORT_CDIFF |
154 REPORT_NDIFF |
155 REPORT_ONLY_FIRST_FAILURE)
156
Edward Loperac20f572004-08-12 02:02:24 +0000157# Special string markers for use in `want` strings:
158BLANKLINE_MARKER = '<BLANKLINE>'
159ELLIPSIS_MARKER = '...'
160
Tim Peters8485b562004-08-04 18:46:34 +0000161######################################################################
162## Table of Contents
163######################################################################
Edward Loper7c748462004-08-09 02:06:06 +0000164# 1. Utility Functions
165# 2. Example & DocTest -- store test cases
166# 3. DocTest Parser -- extracts examples from strings
167# 4. DocTest Finder -- extracts test cases from objects
168# 5. DocTest Runner -- runs test cases
169# 6. Test Functions -- convenient wrappers for testing
170# 7. Tester Class -- for backwards compatibility
171# 8. Unittest Support
172# 9. Debugging Support
173# 10. Example Usage
Tim Peters8a7d2d52001-01-16 07:10:57 +0000174
Tim Peters8485b562004-08-04 18:46:34 +0000175######################################################################
176## 1. Utility Functions
177######################################################################
Tim Peters8a7d2d52001-01-16 07:10:57 +0000178
179def is_private(prefix, base):
180 """prefix, base -> true iff name prefix + "." + base is "private".
181
182 Prefix may be an empty string, and base does not contain a period.
183 Prefix is ignored (although functions you write conforming to this
184 protocol may make use of it).
185 Return true iff base begins with an (at least one) underscore, but
186 does not both begin and end with (at least) two underscores.
187
188 >>> is_private("a.b", "my_func")
Guido van Rossum77f6a652002-04-03 22:41:51 +0000189 False
Tim Peters8a7d2d52001-01-16 07:10:57 +0000190 >>> is_private("____", "_my_func")
Guido van Rossum77f6a652002-04-03 22:41:51 +0000191 True
Tim Peters8a7d2d52001-01-16 07:10:57 +0000192 >>> is_private("someclass", "__init__")
Guido van Rossum77f6a652002-04-03 22:41:51 +0000193 False
Tim Peters8a7d2d52001-01-16 07:10:57 +0000194 >>> is_private("sometypo", "__init_")
Guido van Rossum77f6a652002-04-03 22:41:51 +0000195 True
Tim Peters8a7d2d52001-01-16 07:10:57 +0000196 >>> is_private("x.y.z", "_")
Guido van Rossum77f6a652002-04-03 22:41:51 +0000197 True
Tim Peters8a7d2d52001-01-16 07:10:57 +0000198 >>> is_private("_x.y.z", "__")
Guido van Rossum77f6a652002-04-03 22:41:51 +0000199 False
Tim Peters8a7d2d52001-01-16 07:10:57 +0000200 >>> is_private("", "") # senseless but consistent
Guido van Rossum77f6a652002-04-03 22:41:51 +0000201 False
Tim Peters8a7d2d52001-01-16 07:10:57 +0000202 """
Tim Petersbafb1fe2004-08-08 01:52:57 +0000203 warnings.warn("is_private is deprecated; it wasn't useful; "
204 "examine DocTestFinder.find() lists instead",
Tim Peters3ddd60a2004-08-08 02:43:33 +0000205 DeprecationWarning, stacklevel=2)
Tim Peters8a7d2d52001-01-16 07:10:57 +0000206 return base[:1] == "_" and not base[:2] == "__" == base[-2:]
207
Tim Peters8485b562004-08-04 18:46:34 +0000208def _extract_future_flags(globs):
209 """
210 Return the compiler-flags associated with the future features that
211 have been imported into the given namespace (globs).
212 """
213 flags = 0
214 for fname in __future__.all_feature_names:
215 feature = globs.get(fname, None)
216 if feature is getattr(__future__, fname):
217 flags |= feature.compiler_flag
218 return flags
Tim Peters7402f792001-10-02 03:53:41 +0000219
Tim Peters8485b562004-08-04 18:46:34 +0000220def _normalize_module(module, depth=2):
221 """
222 Return the module specified by `module`. In particular:
223 - If `module` is a module, then return module.
224 - If `module` is a string, then import and return the
225 module with that name.
226 - If `module` is None, then return the calling module.
227 The calling module is assumed to be the module of
228 the stack frame at the given depth in the call stack.
229 """
230 if inspect.ismodule(module):
231 return module
232 elif isinstance(module, (str, unicode)):
233 return __import__(module, globals(), locals(), ["*"])
234 elif module is None:
235 return sys.modules[sys._getframe(depth).f_globals['__name__']]
236 else:
237 raise TypeError("Expected a module, string, or None")
Tim Peters7402f792001-10-02 03:53:41 +0000238
Edward Loperaacf0832004-08-26 01:19:50 +0000239def _indent(s, indent=4):
Tim Peters8485b562004-08-04 18:46:34 +0000240 """
Edward Loperaacf0832004-08-26 01:19:50 +0000241 Add the given number of space characters to the beginning every
242 non-blank line in `s`, and return the result.
Tim Peters8485b562004-08-04 18:46:34 +0000243 """
Edward Loperaacf0832004-08-26 01:19:50 +0000244 # This regexp matches the start of non-blank lines:
245 return re.sub('(?m)^(?!$)', indent*' ', s)
Tim Peters8a7d2d52001-01-16 07:10:57 +0000246
Edward Loper8e4a34b2004-08-12 02:34:27 +0000247def _exception_traceback(exc_info):
248 """
249 Return a string containing a traceback message for the given
250 exc_info tuple (as returned by sys.exc_info()).
251 """
252 # Get a traceback message.
253 excout = StringIO()
254 exc_type, exc_val, exc_tb = exc_info
255 traceback.print_exception(exc_type, exc_val, exc_tb, file=excout)
256 return excout.getvalue()
257
Tim Peters8485b562004-08-04 18:46:34 +0000258# Override some StringIO methods.
259class _SpoofOut(StringIO):
260 def getvalue(self):
261 result = StringIO.getvalue(self)
262 # If anything at all was written, make sure there's a trailing
263 # newline. There's no way for the expected output to indicate
264 # that a trailing newline is missing.
265 if result and not result.endswith("\n"):
266 result += "\n"
267 # Prevent softspace from screwing up the next test case, in
268 # case they used print with a trailing comma in an example.
269 if hasattr(self, "softspace"):
270 del self.softspace
271 return result
Tim Peters8a7d2d52001-01-16 07:10:57 +0000272
Tim Peters8485b562004-08-04 18:46:34 +0000273 def truncate(self, size=None):
274 StringIO.truncate(self, size)
275 if hasattr(self, "softspace"):
276 del self.softspace
Tim Peters8a7d2d52001-01-16 07:10:57 +0000277
Tim Peters26b3ebb2004-08-19 08:10:08 +0000278# Worst-case linear-time ellipsis matching.
Tim Petersb0a04e12004-08-20 02:08:04 +0000279def _ellipsis_match(want, got):
Tim Petersdc5de3b2004-08-19 14:06:20 +0000280 """
281 Essentially the only subtle case:
Tim Petersb0a04e12004-08-20 02:08:04 +0000282 >>> _ellipsis_match('aa...aa', 'aaa')
Tim Petersdc5de3b2004-08-19 14:06:20 +0000283 False
284 """
Tim Peters26b3ebb2004-08-19 08:10:08 +0000285 if ELLIPSIS_MARKER not in want:
286 return want == got
Tim Petersdc5de3b2004-08-19 14:06:20 +0000287
Tim Peters26b3ebb2004-08-19 08:10:08 +0000288 # Find "the real" strings.
289 ws = want.split(ELLIPSIS_MARKER)
290 assert len(ws) >= 2
Tim Peters26b3ebb2004-08-19 08:10:08 +0000291
Tim Petersdc5de3b2004-08-19 14:06:20 +0000292 # Deal with exact matches possibly needed at one or both ends.
293 startpos, endpos = 0, len(got)
294 w = ws[0]
295 if w: # starts with exact match
296 if got.startswith(w):
297 startpos = len(w)
298 del ws[0]
299 else:
300 return False
301 w = ws[-1]
302 if w: # ends with exact match
303 if got.endswith(w):
304 endpos -= len(w)
305 del ws[-1]
306 else:
307 return False
308
309 if startpos > endpos:
310 # Exact end matches required more characters than we have, as in
Tim Petersb0a04e12004-08-20 02:08:04 +0000311 # _ellipsis_match('aa...aa', 'aaa')
Tim Petersdc5de3b2004-08-19 14:06:20 +0000312 return False
313
314 # For the rest, we only need to find the leftmost non-overlapping
315 # match for each piece. If there's no overall match that way alone,
316 # there's no overall match period.
Tim Peters26b3ebb2004-08-19 08:10:08 +0000317 for w in ws:
318 # w may be '' at times, if there are consecutive ellipses, or
319 # due to an ellipsis at the start or end of `want`. That's OK.
Tim Petersdc5de3b2004-08-19 14:06:20 +0000320 # Search for an empty string succeeds, and doesn't change startpos.
321 startpos = got.find(w, startpos, endpos)
322 if startpos < 0:
Tim Peters26b3ebb2004-08-19 08:10:08 +0000323 return False
Tim Petersdc5de3b2004-08-19 14:06:20 +0000324 startpos += len(w)
Tim Peters26b3ebb2004-08-19 08:10:08 +0000325
Tim Petersdc5de3b2004-08-19 14:06:20 +0000326 return True
Tim Peters26b3ebb2004-08-19 08:10:08 +0000327
Edward Loper00f8da72004-08-26 18:05:07 +0000328def _comment_line(line):
329 "Return a commented form of the given line"
330 line = line.rstrip()
331 if line:
332 return '# '+line
333 else:
334 return '#'
335
Edward Loper2de91ba2004-08-27 02:07:46 +0000336class _OutputRedirectingPdb(pdb.Pdb):
337 """
338 A specialized version of the python debugger that redirects stdout
339 to a given stream when interacting with the user. Stdout is *not*
340 redirected when traced code is executed.
341 """
342 def __init__(self, out):
343 self.__out = out
344 pdb.Pdb.__init__(self)
345
346 def trace_dispatch(self, *args):
347 # Redirect stdout to the given stream.
348 save_stdout = sys.stdout
349 sys.stdout = self.__out
350 # Call Pdb's trace dispatch method.
Tim Petersd7bbbbc2004-11-08 22:30:28 +0000351 try:
352 return pdb.Pdb.trace_dispatch(self, *args)
353 finally:
354 # Restore stdout.
355 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):
360 raise TypeError, 'Expected a module: %r' % module
361 if path.startswith('/'):
362 raise ValueError, 'Module-relative files may not have absolute paths'
363
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.
420 I.e., the number of space characters that preceed the
421 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
Tim Peters8485b562004-08-04 18:46:34 +0000447class DocTest:
448 """
449 A collection of doctest examples that should be run in a single
Edward Lopera1ef6112004-08-09 16:14:41 +0000450 namespace. Each `DocTest` defines the following attributes:
Tim Peters8a7d2d52001-01-16 07:10:57 +0000451
Tim Peters8485b562004-08-04 18:46:34 +0000452 - examples: the list of examples.
Tim Peters8a7d2d52001-01-16 07:10:57 +0000453
Tim Peters8485b562004-08-04 18:46:34 +0000454 - globs: The namespace (aka globals) that the examples should
455 be run in.
Tim Peters8a7d2d52001-01-16 07:10:57 +0000456
Tim Peters8485b562004-08-04 18:46:34 +0000457 - name: A name identifying the DocTest (typically, the name of
458 the object whose docstring this DocTest was extracted from).
Tim Peters8a7d2d52001-01-16 07:10:57 +0000459
Tim Peters8485b562004-08-04 18:46:34 +0000460 - filename: The name of the file that this DocTest was extracted
Edward Lopera1ef6112004-08-09 16:14:41 +0000461 from, or `None` if the filename is unknown.
Tim Peters8a7d2d52001-01-16 07:10:57 +0000462
Tim Peters8485b562004-08-04 18:46:34 +0000463 - lineno: The line number within filename where this DocTest
Edward Lopera1ef6112004-08-09 16:14:41 +0000464 begins, or `None` if the line number is unavailable. This
465 line number is zero-based, with respect to the beginning of
466 the file.
467
468 - docstring: The string that the examples were extracted from,
469 or `None` if the string is unavailable.
Tim Peters8485b562004-08-04 18:46:34 +0000470 """
Edward Lopera1ef6112004-08-09 16:14:41 +0000471 def __init__(self, examples, globs, name, filename, lineno, docstring):
Tim Peters8485b562004-08-04 18:46:34 +0000472 """
Edward Lopera1ef6112004-08-09 16:14:41 +0000473 Create a new DocTest containing the given examples. The
474 DocTest's globals are initialized with a copy of `globs`.
Tim Peters8485b562004-08-04 18:46:34 +0000475 """
Edward Lopera1ef6112004-08-09 16:14:41 +0000476 assert not isinstance(examples, basestring), \
477 "DocTest no longer accepts str; use DocTestParser instead"
478 self.examples = examples
479 self.docstring = docstring
Tim Peters8485b562004-08-04 18:46:34 +0000480 self.globs = globs.copy()
Tim Peters8485b562004-08-04 18:46:34 +0000481 self.name = name
482 self.filename = filename
483 self.lineno = lineno
Tim Peters8485b562004-08-04 18:46:34 +0000484
485 def __repr__(self):
486 if len(self.examples) == 0:
487 examples = 'no examples'
488 elif len(self.examples) == 1:
489 examples = '1 example'
490 else:
491 examples = '%d examples' % len(self.examples)
492 return ('<DocTest %s from %s:%s (%s)>' %
493 (self.name, self.filename, self.lineno, examples))
494
495
496 # This lets us sort tests by name:
497 def __cmp__(self, other):
498 if not isinstance(other, DocTest):
499 return -1
500 return cmp((self.name, self.filename, self.lineno, id(self)),
501 (other.name, other.filename, other.lineno, id(other)))
502
503######################################################################
Edward Loperb7503ff2004-08-19 19:19:03 +0000504## 3. DocTestParser
Edward Loper7c748462004-08-09 02:06:06 +0000505######################################################################
506
Edward Lopera1ef6112004-08-09 16:14:41 +0000507class DocTestParser:
Edward Loper7c748462004-08-09 02:06:06 +0000508 """
Edward Lopera1ef6112004-08-09 16:14:41 +0000509 A class used to parse strings containing doctest examples.
Edward Loper7c748462004-08-09 02:06:06 +0000510 """
Edward Loper8e4a34b2004-08-12 02:34:27 +0000511 # This regular expression is used to find doctest examples in a
512 # string. It defines three groups: `source` is the source code
513 # (including leading indentation and prompts); `indent` is the
514 # indentation of the first (PS1) line of the source code; and
515 # `want` is the expected output (including leading indentation).
Edward Loper7c748462004-08-09 02:06:06 +0000516 _EXAMPLE_RE = re.compile(r'''
Tim Petersd40a92b2004-08-09 03:28:45 +0000517 # Source consists of a PS1 line followed by zero or more PS2 lines.
518 (?P<source>
519 (?:^(?P<indent> [ ]*) >>> .*) # PS1 line
520 (?:\n [ ]* \.\.\. .*)*) # PS2 lines
521 \n?
522 # Want consists of any non-blank lines that do not start with PS1.
523 (?P<want> (?:(?![ ]*$) # Not a blank line
524 (?![ ]*>>>) # Not a line starting with PS1
525 .*$\n? # But any other line
526 )*)
527 ''', re.MULTILINE | re.VERBOSE)
Edward Loper8e4a34b2004-08-12 02:34:27 +0000528
Edward Lopera6b68322004-08-26 00:05:43 +0000529 # A regular expression for handling `want` strings that contain
530 # expected exceptions. It divides `want` into three pieces:
531 # - the traceback header line (`hdr`)
532 # - the traceback stack (`stack`)
533 # - the exception message (`msg`), as generated by
534 # traceback.format_exception_only()
535 # `msg` may have multiple lines. We assume/require that the
536 # exception message is the first non-indented line starting with a word
537 # character following the traceback header line.
538 _EXCEPTION_RE = re.compile(r"""
539 # Grab the traceback header. Different versions of Python have
540 # said different things on the first traceback line.
541 ^(?P<hdr> Traceback\ \(
542 (?: most\ recent\ call\ last
543 | innermost\ last
544 ) \) :
545 )
546 \s* $ # toss trailing whitespace on the header.
547 (?P<stack> .*?) # don't blink: absorb stuff until...
548 ^ (?P<msg> \w+ .*) # a line *starts* with alphanum.
549 """, re.VERBOSE | re.MULTILINE | re.DOTALL)
550
Tim Peters7ea48dd2004-08-13 01:52:59 +0000551 # A callable returning a true value iff its argument is a blank line
552 # or contains a single comment.
Edward Loper8e4a34b2004-08-12 02:34:27 +0000553 _IS_BLANK_OR_COMMENT = re.compile(r'^[ ]*(#.*)?$').match
Edward Loper7c748462004-08-09 02:06:06 +0000554
Edward Loper00f8da72004-08-26 18:05:07 +0000555 def parse(self, string, name='<string>'):
556 """
557 Divide the given string into examples and intervening text,
558 and return them as a list of alternating Examples and strings.
559 Line numbers for the Examples are 0-based. The optional
560 argument `name` is a name identifying this string, and is only
561 used for error messages.
562 """
563 string = string.expandtabs()
564 # If all lines begin with the same indentation, then strip it.
565 min_indent = self._min_indent(string)
566 if min_indent > 0:
567 string = '\n'.join([l[min_indent:] for l in string.split('\n')])
568
569 output = []
570 charno, lineno = 0, 0
571 # Find all doctest examples in the string:
Edward Loper2de91ba2004-08-27 02:07:46 +0000572 for m in self._EXAMPLE_RE.finditer(string):
Edward Loper00f8da72004-08-26 18:05:07 +0000573 # Add the pre-example text to `output`.
574 output.append(string[charno:m.start()])
575 # Update lineno (lines before this example)
576 lineno += string.count('\n', charno, m.start())
577 # Extract info from the regexp match.
578 (source, options, want, exc_msg) = \
579 self._parse_example(m, name, lineno)
580 # Create an Example, and add it to the list.
581 if not self._IS_BLANK_OR_COMMENT(source):
582 output.append( Example(source, want, exc_msg,
583 lineno=lineno,
584 indent=min_indent+len(m.group('indent')),
585 options=options) )
586 # Update lineno (lines inside this example)
587 lineno += string.count('\n', m.start(), m.end())
588 # Update charno.
589 charno = m.end()
590 # Add any remaining post-example text to `output`.
591 output.append(string[charno:])
592 return output
593
Edward Lopera1ef6112004-08-09 16:14:41 +0000594 def get_doctest(self, string, globs, name, filename, lineno):
Edward Loper7c748462004-08-09 02:06:06 +0000595 """
Edward Lopera1ef6112004-08-09 16:14:41 +0000596 Extract all doctest examples from the given string, and
597 collect them into a `DocTest` object.
598
599 `globs`, `name`, `filename`, and `lineno` are attributes for
600 the new `DocTest` object. See the documentation for `DocTest`
601 for more information.
602 """
603 return DocTest(self.get_examples(string, name), globs,
604 name, filename, lineno, string)
605
606 def get_examples(self, string, name='<string>'):
607 """
608 Extract all doctest examples from the given string, and return
609 them as a list of `Example` objects. Line numbers are
610 0-based, because it's most common in doctests that nothing
611 interesting appears on the same line as opening triple-quote,
612 and so the first interesting line is called \"line 1\" then.
613
614 The optional argument `name` is a name identifying this
615 string, and is only used for error messages.
Edward Loper7c748462004-08-09 02:06:06 +0000616 """
Edward Loper00f8da72004-08-26 18:05:07 +0000617 return [x for x in self.parse(string, name)
618 if isinstance(x, Example)]
Edward Loper7c748462004-08-09 02:06:06 +0000619
Edward Loper74bca7a2004-08-12 02:27:44 +0000620 def _parse_example(self, m, name, lineno):
621 """
622 Given a regular expression match from `_EXAMPLE_RE` (`m`),
623 return a pair `(source, want)`, where `source` is the matched
624 example's source code (with prompts and indentation stripped);
625 and `want` is the example's expected output (with indentation
626 stripped).
627
628 `name` is the string's name, and `lineno` is the line number
629 where the example starts; both are used for error messages.
630 """
Edward Loper7c748462004-08-09 02:06:06 +0000631 # Get the example's indentation level.
632 indent = len(m.group('indent'))
633
634 # Divide source into lines; check that they're properly
635 # indented; and then strip their indentation & prompts.
636 source_lines = m.group('source').split('\n')
Edward Lopera1ef6112004-08-09 16:14:41 +0000637 self._check_prompt_blank(source_lines, indent, name, lineno)
Tim Petersc5049152004-08-22 17:34:58 +0000638 self._check_prefix(source_lines[1:], ' '*indent + '.', name, lineno)
Edward Loper7c748462004-08-09 02:06:06 +0000639 source = '\n'.join([sl[indent+4:] for sl in source_lines])
Edward Loper7c748462004-08-09 02:06:06 +0000640
Tim Petersc5049152004-08-22 17:34:58 +0000641 # Divide want into lines; check that it's properly indented; and
642 # then strip the indentation. Spaces before the last newline should
643 # be preserved, so plain rstrip() isn't good enough.
Jim Fulton07a349c2004-08-22 14:10:00 +0000644 want = m.group('want')
Jim Fulton07a349c2004-08-22 14:10:00 +0000645 want_lines = want.split('\n')
Tim Petersc5049152004-08-22 17:34:58 +0000646 if len(want_lines) > 1 and re.match(r' *$', want_lines[-1]):
647 del want_lines[-1] # forget final newline & spaces after it
Edward Lopera1ef6112004-08-09 16:14:41 +0000648 self._check_prefix(want_lines, ' '*indent, name,
Tim Petersc5049152004-08-22 17:34:58 +0000649 lineno + len(source_lines))
Edward Loper7c748462004-08-09 02:06:06 +0000650 want = '\n'.join([wl[indent:] for wl in want_lines])
Edward Loper7c748462004-08-09 02:06:06 +0000651
Edward Lopera6b68322004-08-26 00:05:43 +0000652 # If `want` contains a traceback message, then extract it.
653 m = self._EXCEPTION_RE.match(want)
654 if m:
655 exc_msg = m.group('msg')
656 else:
657 exc_msg = None
658
Edward Loper00f8da72004-08-26 18:05:07 +0000659 # Extract options from the source.
660 options = self._find_options(source, name, lineno)
661
662 return source, options, want, exc_msg
Edward Loper7c748462004-08-09 02:06:06 +0000663
Edward Loper74bca7a2004-08-12 02:27:44 +0000664 # This regular expression looks for option directives in the
665 # source code of an example. Option directives are comments
666 # starting with "doctest:". Warning: this may give false
667 # positives for string-literals that contain the string
668 # "#doctest:". Eliminating these false positives would require
669 # actually parsing the string; but we limit them by ignoring any
670 # line containing "#doctest:" that is *followed* by a quote mark.
671 _OPTION_DIRECTIVE_RE = re.compile(r'#\s*doctest:\s*([^\n\'"]*)$',
672 re.MULTILINE)
673
674 def _find_options(self, source, name, lineno):
675 """
676 Return a dictionary containing option overrides extracted from
677 option directives in the given source string.
678
679 `name` is the string's name, and `lineno` is the line number
680 where the example starts; both are used for error messages.
681 """
682 options = {}
683 # (note: with the current regexp, this will match at most once:)
684 for m in self._OPTION_DIRECTIVE_RE.finditer(source):
685 option_strings = m.group(1).replace(',', ' ').split()
686 for option in option_strings:
687 if (option[0] not in '+-' or
688 option[1:] not in OPTIONFLAGS_BY_NAME):
689 raise ValueError('line %r of the doctest for %s '
690 'has an invalid option: %r' %
691 (lineno+1, name, option))
692 flag = OPTIONFLAGS_BY_NAME[option[1:]]
693 options[flag] = (option[0] == '+')
694 if options and self._IS_BLANK_OR_COMMENT(source):
695 raise ValueError('line %r of the doctest for %s has an option '
696 'directive on a line with no example: %r' %
697 (lineno, name, source))
698 return options
699
Edward Lopera5db6002004-08-12 02:41:30 +0000700 # This regular expression finds the indentation of every non-blank
701 # line in a string.
Edward Loper00f8da72004-08-26 18:05:07 +0000702 _INDENT_RE = re.compile('^([ ]*)(?=\S)', re.MULTILINE)
Edward Lopera5db6002004-08-12 02:41:30 +0000703
704 def _min_indent(self, s):
705 "Return the minimum indentation of any non-blank line in `s`"
Edward Loper00f8da72004-08-26 18:05:07 +0000706 indents = [len(indent) for indent in self._INDENT_RE.findall(s)]
707 if len(indents) > 0:
708 return min(indents)
Tim Petersdd0e4752004-08-09 03:31:56 +0000709 else:
Edward Loper00f8da72004-08-26 18:05:07 +0000710 return 0
Edward Loper7c748462004-08-09 02:06:06 +0000711
Edward Lopera1ef6112004-08-09 16:14:41 +0000712 def _check_prompt_blank(self, lines, indent, name, lineno):
Edward Loper74bca7a2004-08-12 02:27:44 +0000713 """
714 Given the lines of a source string (including prompts and
715 leading indentation), check to make sure that every prompt is
716 followed by a space character. If any line is not followed by
717 a space character, then raise ValueError.
718 """
Edward Loper7c748462004-08-09 02:06:06 +0000719 for i, line in enumerate(lines):
720 if len(line) >= indent+4 and line[indent+3] != ' ':
721 raise ValueError('line %r of the docstring for %s '
722 'lacks blank after %s: %r' %
Edward Lopera1ef6112004-08-09 16:14:41 +0000723 (lineno+i+1, name,
Edward Loper7c748462004-08-09 02:06:06 +0000724 line[indent:indent+3], line))
725
Edward Lopera1ef6112004-08-09 16:14:41 +0000726 def _check_prefix(self, lines, prefix, name, lineno):
Edward Loper74bca7a2004-08-12 02:27:44 +0000727 """
728 Check that every line in the given list starts with the given
729 prefix; if any line does not, then raise a ValueError.
730 """
Edward Loper7c748462004-08-09 02:06:06 +0000731 for i, line in enumerate(lines):
732 if line and not line.startswith(prefix):
733 raise ValueError('line %r of the docstring for %s has '
734 'inconsistent leading whitespace: %r' %
Edward Lopera1ef6112004-08-09 16:14:41 +0000735 (lineno+i+1, name, line))
Edward Loper7c748462004-08-09 02:06:06 +0000736
737
738######################################################################
739## 4. DocTest Finder
Tim Peters8485b562004-08-04 18:46:34 +0000740######################################################################
741
742class DocTestFinder:
743 """
744 A class used to extract the DocTests that are relevant to a given
745 object, from its docstring and the docstrings of its contained
746 objects. Doctests can currently be extracted from the following
747 object types: modules, functions, classes, methods, staticmethods,
748 classmethods, and properties.
Tim Peters8485b562004-08-04 18:46:34 +0000749 """
750
Edward Lopera1ef6112004-08-09 16:14:41 +0000751 def __init__(self, verbose=False, parser=DocTestParser(),
Tim Peters958cc892004-09-13 14:53:28 +0000752 recurse=True, _namefilter=None, exclude_empty=True):
Tim Peters8485b562004-08-04 18:46:34 +0000753 """
754 Create a new doctest finder.
755
Edward Lopera1ef6112004-08-09 16:14:41 +0000756 The optional argument `parser` specifies a class or
Tim Peters19397e52004-08-06 22:02:59 +0000757 function that should be used to create new DocTest objects (or
Tim Peters161c9632004-08-08 03:38:33 +0000758 objects that implement the same interface as DocTest). The
Tim Peters19397e52004-08-06 22:02:59 +0000759 signature for this factory function should match the signature
760 of the DocTest constructor.
761
Tim Peters8485b562004-08-04 18:46:34 +0000762 If the optional argument `recurse` is false, then `find` will
763 only examine the given object, and not any contained objects.
Edward Loper32ddbf72004-09-13 05:47:24 +0000764
Tim Peters958cc892004-09-13 14:53:28 +0000765 If the optional argument `exclude_empty` is false, then `find`
766 will include tests for objects with empty docstrings.
Tim Peters8485b562004-08-04 18:46:34 +0000767 """
Edward Lopera1ef6112004-08-09 16:14:41 +0000768 self._parser = parser
Tim Peters8485b562004-08-04 18:46:34 +0000769 self._verbose = verbose
Tim Peters8485b562004-08-04 18:46:34 +0000770 self._recurse = recurse
Edward Loper32ddbf72004-09-13 05:47:24 +0000771 self._exclude_empty = exclude_empty
Tim Petersf727c6c2004-08-08 01:48:59 +0000772 # _namefilter is undocumented, and exists only for temporary backward-
773 # compatibility support of testmod's deprecated isprivate mess.
774 self._namefilter = _namefilter
Tim Peters8485b562004-08-04 18:46:34 +0000775
776 def find(self, obj, name=None, module=None, globs=None,
Tim Petersf3f57472004-08-08 06:11:48 +0000777 extraglobs=None):
Tim Peters8485b562004-08-04 18:46:34 +0000778 """
779 Return a list of the DocTests that are defined by the given
780 object's docstring, or by any of its contained objects'
781 docstrings.
782
783 The optional parameter `module` is the module that contains
Tim Petersf3f57472004-08-08 06:11:48 +0000784 the given object. If the module is not specified or is None, then
785 the test finder will attempt to automatically determine the
Tim Peters8485b562004-08-04 18:46:34 +0000786 correct module. The object's module is used:
787
788 - As a default namespace, if `globs` is not specified.
789 - To prevent the DocTestFinder from extracting DocTests
Tim Petersf3f57472004-08-08 06:11:48 +0000790 from objects that are imported from other modules.
Tim Peters8485b562004-08-04 18:46:34 +0000791 - To find the name of the file containing the object.
792 - To help find the line number of the object within its
793 file.
794
Tim Petersf3f57472004-08-08 06:11:48 +0000795 Contained objects whose module does not match `module` are ignored.
796
797 If `module` is False, no attempt to find the module will be made.
798 This is obscure, of use mostly in tests: if `module` is False, or
799 is None but cannot be found automatically, then all objects are
800 considered to belong to the (non-existent) module, so all contained
801 objects will (recursively) be searched for doctests.
802
Tim Peters8485b562004-08-04 18:46:34 +0000803 The globals for each DocTest is formed by combining `globs`
804 and `extraglobs` (bindings in `extraglobs` override bindings
805 in `globs`). A new copy of the globals dictionary is created
806 for each DocTest. If `globs` is not specified, then it
807 defaults to the module's `__dict__`, if specified, or {}
808 otherwise. If `extraglobs` is not specified, then it defaults
809 to {}.
810
Tim Peters8485b562004-08-04 18:46:34 +0000811 """
812 # If name was not specified, then extract it from the object.
813 if name is None:
814 name = getattr(obj, '__name__', None)
815 if name is None:
816 raise ValueError("DocTestFinder.find: name must be given "
817 "when obj.__name__ doesn't exist: %r" %
818 (type(obj),))
819
820 # Find the module that contains the given object (if obj is
821 # a module, then module=obj.). Note: this may fail, in which
822 # case module will be None.
Tim Petersf3f57472004-08-08 06:11:48 +0000823 if module is False:
824 module = None
825 elif module is None:
Tim Peters8485b562004-08-04 18:46:34 +0000826 module = inspect.getmodule(obj)
827
828 # Read the module's source code. This is used by
829 # DocTestFinder._find_lineno to find the line number for a
830 # given object's docstring.
831 try:
832 file = inspect.getsourcefile(obj) or inspect.getfile(obj)
833 source_lines = linecache.getlines(file)
834 if not source_lines:
835 source_lines = None
836 except TypeError:
837 source_lines = None
838
839 # Initialize globals, and merge in extraglobs.
Tim Peters8a7d2d52001-01-16 07:10:57 +0000840 if globs is None:
Tim Peters8485b562004-08-04 18:46:34 +0000841 if module is None:
842 globs = {}
843 else:
844 globs = module.__dict__.copy()
845 else:
846 globs = globs.copy()
847 if extraglobs is not None:
848 globs.update(extraglobs)
Tim Peters8a7d2d52001-01-16 07:10:57 +0000849
Tim Peters8485b562004-08-04 18:46:34 +0000850 # Recursively expore `obj`, extracting DocTests.
851 tests = []
Tim Petersf3f57472004-08-08 06:11:48 +0000852 self._find(tests, obj, name, module, source_lines, globs, {})
Tim Peters8485b562004-08-04 18:46:34 +0000853 return tests
854
855 def _filter(self, obj, prefix, base):
856 """
857 Return true if the given object should not be examined.
858 """
Tim Petersf727c6c2004-08-08 01:48:59 +0000859 return (self._namefilter is not None and
860 self._namefilter(prefix, base))
Tim Peters8485b562004-08-04 18:46:34 +0000861
862 def _from_module(self, module, object):
863 """
864 Return true if the given object is defined in the given
865 module.
866 """
867 if module is None:
868 return True
869 elif inspect.isfunction(object):
870 return module.__dict__ is object.func_globals
871 elif inspect.isclass(object):
872 return module.__name__ == object.__module__
873 elif inspect.getmodule(object) is not None:
874 return module is inspect.getmodule(object)
875 elif hasattr(object, '__module__'):
876 return module.__name__ == object.__module__
877 elif isinstance(object, property):
878 return True # [XX] no way not be sure.
879 else:
880 raise ValueError("object must be a class or function")
881
Tim Petersf3f57472004-08-08 06:11:48 +0000882 def _find(self, tests, obj, name, module, source_lines, globs, seen):
Tim Peters8485b562004-08-04 18:46:34 +0000883 """
884 Find tests for the given object and any contained objects, and
885 add them to `tests`.
886 """
887 if self._verbose:
888 print 'Finding tests in %s' % name
889
890 # If we've already processed this object, then ignore it.
891 if id(obj) in seen:
892 return
893 seen[id(obj)] = 1
894
895 # Find a test for this object, and add it to the list of tests.
896 test = self._get_test(obj, name, module, globs, source_lines)
897 if test is not None:
898 tests.append(test)
899
900 # Look for tests in a module's contained objects.
901 if inspect.ismodule(obj) and self._recurse:
902 for valname, val in obj.__dict__.items():
903 # Check if this contained object should be ignored.
904 if self._filter(val, name, valname):
905 continue
906 valname = '%s.%s' % (name, valname)
907 # Recurse to functions & classes.
908 if ((inspect.isfunction(val) or inspect.isclass(val)) and
Tim Petersf3f57472004-08-08 06:11:48 +0000909 self._from_module(module, val)):
Tim Peters8485b562004-08-04 18:46:34 +0000910 self._find(tests, val, valname, module, source_lines,
Tim Petersf3f57472004-08-08 06:11:48 +0000911 globs, seen)
Tim Peters8485b562004-08-04 18:46:34 +0000912
913 # Look for tests in a module's __test__ dictionary.
914 if inspect.ismodule(obj) and self._recurse:
915 for valname, val in getattr(obj, '__test__', {}).items():
916 if not isinstance(valname, basestring):
917 raise ValueError("DocTestFinder.find: __test__ keys "
918 "must be strings: %r" %
919 (type(valname),))
920 if not (inspect.isfunction(val) or inspect.isclass(val) or
921 inspect.ismethod(val) or inspect.ismodule(val) or
922 isinstance(val, basestring)):
923 raise ValueError("DocTestFinder.find: __test__ values "
924 "must be strings, functions, methods, "
925 "classes, or modules: %r" %
926 (type(val),))
Tim Petersc5684782004-09-13 01:07:12 +0000927 valname = '%s.__test__.%s' % (name, valname)
Tim Peters8485b562004-08-04 18:46:34 +0000928 self._find(tests, val, valname, module, source_lines,
Tim Petersf3f57472004-08-08 06:11:48 +0000929 globs, seen)
Tim Peters8485b562004-08-04 18:46:34 +0000930
931 # Look for tests in a class's contained objects.
932 if inspect.isclass(obj) and self._recurse:
933 for valname, val in obj.__dict__.items():
934 # Check if this contained object should be ignored.
935 if self._filter(val, name, valname):
936 continue
937 # Special handling for staticmethod/classmethod.
938 if isinstance(val, staticmethod):
939 val = getattr(obj, valname)
940 if isinstance(val, classmethod):
941 val = getattr(obj, valname).im_func
942
943 # Recurse to methods, properties, and nested classes.
944 if ((inspect.isfunction(val) or inspect.isclass(val) or
Tim Petersf3f57472004-08-08 06:11:48 +0000945 isinstance(val, property)) and
946 self._from_module(module, val)):
Tim Peters8485b562004-08-04 18:46:34 +0000947 valname = '%s.%s' % (name, valname)
948 self._find(tests, val, valname, module, source_lines,
Tim Petersf3f57472004-08-08 06:11:48 +0000949 globs, seen)
Tim Peters8485b562004-08-04 18:46:34 +0000950
951 def _get_test(self, obj, name, module, globs, source_lines):
952 """
953 Return a DocTest for the given object, if it defines a docstring;
954 otherwise, return None.
955 """
956 # Extract the object's docstring. If it doesn't have one,
957 # then return None (no test for this object).
958 if isinstance(obj, basestring):
959 docstring = obj
960 else:
961 try:
962 if obj.__doc__ is None:
Edward Loper32ddbf72004-09-13 05:47:24 +0000963 docstring = ''
964 else:
Jim Fulton7d428782004-10-13 14:15:32 +0000965 docstring = obj.__doc__
966 if not isinstance(docstring, basestring):
967 docstring = str(docstring)
Tim Peters8485b562004-08-04 18:46:34 +0000968 except (TypeError, AttributeError):
Edward Loper32ddbf72004-09-13 05:47:24 +0000969 docstring = ''
Tim Peters8485b562004-08-04 18:46:34 +0000970
971 # Find the docstring's location in the file.
972 lineno = self._find_lineno(obj, source_lines)
973
Edward Loper32ddbf72004-09-13 05:47:24 +0000974 # Don't bother if the docstring is empty.
975 if self._exclude_empty and not docstring:
976 return None
977
Tim Peters8485b562004-08-04 18:46:34 +0000978 # Return a DocTest for this object.
979 if module is None:
980 filename = None
981 else:
982 filename = getattr(module, '__file__', module.__name__)
Jim Fulton07a349c2004-08-22 14:10:00 +0000983 if filename[-4:] in (".pyc", ".pyo"):
984 filename = filename[:-1]
Edward Lopera1ef6112004-08-09 16:14:41 +0000985 return self._parser.get_doctest(docstring, globs, name,
986 filename, lineno)
Tim Peters8485b562004-08-04 18:46:34 +0000987
988 def _find_lineno(self, obj, source_lines):
989 """
990 Return a line number of the given object's docstring. Note:
991 this method assumes that the object has a docstring.
992 """
993 lineno = None
994
995 # Find the line number for modules.
996 if inspect.ismodule(obj):
997 lineno = 0
998
999 # Find the line number for classes.
1000 # Note: this could be fooled if a class is defined multiple
1001 # times in a single file.
1002 if inspect.isclass(obj):
1003 if source_lines is None:
1004 return None
1005 pat = re.compile(r'^\s*class\s*%s\b' %
1006 getattr(obj, '__name__', '-'))
1007 for i, line in enumerate(source_lines):
1008 if pat.match(line):
1009 lineno = i
1010 break
1011
1012 # Find the line number for functions & methods.
1013 if inspect.ismethod(obj): obj = obj.im_func
1014 if inspect.isfunction(obj): obj = obj.func_code
1015 if inspect.istraceback(obj): obj = obj.tb_frame
1016 if inspect.isframe(obj): obj = obj.f_code
1017 if inspect.iscode(obj):
1018 lineno = getattr(obj, 'co_firstlineno', None)-1
1019
1020 # Find the line number where the docstring starts. Assume
1021 # that it's the first line that begins with a quote mark.
1022 # Note: this could be fooled by a multiline function
1023 # signature, where a continuation line begins with a quote
1024 # mark.
1025 if lineno is not None:
1026 if source_lines is None:
1027 return lineno+1
1028 pat = re.compile('(^|.*:)\s*\w*("|\')')
1029 for lineno in range(lineno, len(source_lines)):
1030 if pat.match(source_lines[lineno]):
1031 return lineno
1032
1033 # We couldn't find the line number.
1034 return None
1035
1036######################################################################
Edward Loper7c748462004-08-09 02:06:06 +00001037## 5. DocTest Runner
Tim Peters8485b562004-08-04 18:46:34 +00001038######################################################################
1039
Tim Peters8485b562004-08-04 18:46:34 +00001040class DocTestRunner:
1041 """
1042 A class used to run DocTest test cases, and accumulate statistics.
1043 The `run` method is used to process a single DocTest case. It
1044 returns a tuple `(f, t)`, where `t` is the number of test cases
1045 tried, and `f` is the number of test cases that failed.
1046
1047 >>> tests = DocTestFinder().find(_TestClass)
1048 >>> runner = DocTestRunner(verbose=False)
1049 >>> for test in tests:
1050 ... print runner.run(test)
1051 (0, 2)
1052 (0, 1)
1053 (0, 2)
1054 (0, 2)
1055
1056 The `summarize` method prints a summary of all the test cases that
1057 have been run by the runner, and returns an aggregated `(f, t)`
1058 tuple:
1059
1060 >>> runner.summarize(verbose=1)
1061 4 items passed all tests:
1062 2 tests in _TestClass
1063 2 tests in _TestClass.__init__
1064 2 tests in _TestClass.get
1065 1 tests in _TestClass.square
1066 7 tests in 4 items.
1067 7 passed and 0 failed.
1068 Test passed.
1069 (0, 7)
1070
1071 The aggregated number of tried examples and failed examples is
1072 also available via the `tries` and `failures` attributes:
1073
1074 >>> runner.tries
1075 7
1076 >>> runner.failures
1077 0
1078
1079 The comparison between expected outputs and actual outputs is done
Edward Loper34fcb142004-08-09 02:45:41 +00001080 by an `OutputChecker`. This comparison may be customized with a
1081 number of option flags; see the documentation for `testmod` for
1082 more information. If the option flags are insufficient, then the
1083 comparison may also be customized by passing a subclass of
1084 `OutputChecker` to the constructor.
Tim Peters8485b562004-08-04 18:46:34 +00001085
1086 The test runner's display output can be controlled in two ways.
1087 First, an output function (`out) can be passed to
1088 `TestRunner.run`; this function will be called with strings that
1089 should be displayed. It defaults to `sys.stdout.write`. If
1090 capturing the output is not sufficient, then the display output
1091 can be also customized by subclassing DocTestRunner, and
1092 overriding the methods `report_start`, `report_success`,
1093 `report_unexpected_exception`, and `report_failure`.
1094 """
1095 # This divider string is used to separate failure messages, and to
1096 # separate sections of the summary.
1097 DIVIDER = "*" * 70
1098
Edward Loper34fcb142004-08-09 02:45:41 +00001099 def __init__(self, checker=None, verbose=None, optionflags=0):
Tim Peters8485b562004-08-04 18:46:34 +00001100 """
1101 Create a new test runner.
1102
Edward Loper34fcb142004-08-09 02:45:41 +00001103 Optional keyword arg `checker` is the `OutputChecker` that
1104 should be used to compare the expected outputs and actual
1105 outputs of doctest examples.
1106
Tim Peters8485b562004-08-04 18:46:34 +00001107 Optional keyword arg 'verbose' prints lots of stuff if true,
1108 only failures if false; by default, it's true iff '-v' is in
1109 sys.argv.
1110
1111 Optional argument `optionflags` can be used to control how the
1112 test runner compares expected output to actual output, and how
1113 it displays failures. See the documentation for `testmod` for
1114 more information.
1115 """
Edward Loper34fcb142004-08-09 02:45:41 +00001116 self._checker = checker or OutputChecker()
Tim Peters8a7d2d52001-01-16 07:10:57 +00001117 if verbose is None:
Tim Peters8485b562004-08-04 18:46:34 +00001118 verbose = '-v' in sys.argv
1119 self._verbose = verbose
Tim Peters6ebe61f2003-06-27 20:48:05 +00001120 self.optionflags = optionflags
Jim Fulton07a349c2004-08-22 14:10:00 +00001121 self.original_optionflags = optionflags
Tim Peters6ebe61f2003-06-27 20:48:05 +00001122
Tim Peters8485b562004-08-04 18:46:34 +00001123 # Keep track of the examples we've run.
1124 self.tries = 0
1125 self.failures = 0
1126 self._name2ft = {}
Tim Peters8a7d2d52001-01-16 07:10:57 +00001127
Tim Peters8485b562004-08-04 18:46:34 +00001128 # Create a fake output target for capturing doctest output.
1129 self._fakeout = _SpoofOut()
Tim Peters4fd9e2f2001-08-18 00:05:50 +00001130
Tim Peters8485b562004-08-04 18:46:34 +00001131 #/////////////////////////////////////////////////////////////////
Tim Peters8485b562004-08-04 18:46:34 +00001132 # Reporting methods
1133 #/////////////////////////////////////////////////////////////////
Tim Peters17111f32001-10-03 04:08:26 +00001134
Tim Peters8485b562004-08-04 18:46:34 +00001135 def report_start(self, out, test, example):
Tim Peters8a7d2d52001-01-16 07:10:57 +00001136 """
Tim Peters8485b562004-08-04 18:46:34 +00001137 Report that the test runner is about to process the given
1138 example. (Only displays a message if verbose=True)
1139 """
1140 if self._verbose:
Edward Loperaacf0832004-08-26 01:19:50 +00001141 if example.want:
1142 out('Trying:\n' + _indent(example.source) +
1143 'Expecting:\n' + _indent(example.want))
1144 else:
1145 out('Trying:\n' + _indent(example.source) +
1146 'Expecting nothing\n')
Tim Peters8a7d2d52001-01-16 07:10:57 +00001147
Tim Peters8485b562004-08-04 18:46:34 +00001148 def report_success(self, out, test, example, got):
1149 """
1150 Report that the given example ran successfully. (Only
1151 displays a message if verbose=True)
1152 """
1153 if self._verbose:
1154 out("ok\n")
Tim Peters8a7d2d52001-01-16 07:10:57 +00001155
Tim Peters8485b562004-08-04 18:46:34 +00001156 def report_failure(self, out, test, example, got):
1157 """
1158 Report that the given example failed.
1159 """
Edward Loper8e4a34b2004-08-12 02:34:27 +00001160 out(self._failure_header(test, example) +
Edward Loperca9111e2004-08-26 03:00:24 +00001161 self._checker.output_difference(example, got, self.optionflags))
Tim Peters7402f792001-10-02 03:53:41 +00001162
Tim Peters8485b562004-08-04 18:46:34 +00001163 def report_unexpected_exception(self, out, test, example, exc_info):
1164 """
1165 Report that the given example raised an unexpected exception.
1166 """
Edward Loper8e4a34b2004-08-12 02:34:27 +00001167 out(self._failure_header(test, example) +
Edward Loperaacf0832004-08-26 01:19:50 +00001168 'Exception raised:\n' + _indent(_exception_traceback(exc_info)))
Tim Peters7402f792001-10-02 03:53:41 +00001169
Edward Loper8e4a34b2004-08-12 02:34:27 +00001170 def _failure_header(self, test, example):
Jim Fulton07a349c2004-08-22 14:10:00 +00001171 out = [self.DIVIDER]
1172 if test.filename:
1173 if test.lineno is not None and example.lineno is not None:
1174 lineno = test.lineno + example.lineno + 1
1175 else:
1176 lineno = '?'
1177 out.append('File "%s", line %s, in %s' %
1178 (test.filename, lineno, test.name))
Tim Peters8485b562004-08-04 18:46:34 +00001179 else:
Jim Fulton07a349c2004-08-22 14:10:00 +00001180 out.append('Line %s, in %s' % (example.lineno+1, test.name))
1181 out.append('Failed example:')
1182 source = example.source
Edward Loperaacf0832004-08-26 01:19:50 +00001183 out.append(_indent(source))
1184 return '\n'.join(out)
Tim Peters7402f792001-10-02 03:53:41 +00001185
Tim Peters8485b562004-08-04 18:46:34 +00001186 #/////////////////////////////////////////////////////////////////
1187 # DocTest Running
1188 #/////////////////////////////////////////////////////////////////
Tim Peters7402f792001-10-02 03:53:41 +00001189
Tim Peters8485b562004-08-04 18:46:34 +00001190 def __run(self, test, compileflags, out):
Tim Peters8a7d2d52001-01-16 07:10:57 +00001191 """
Tim Peters8485b562004-08-04 18:46:34 +00001192 Run the examples in `test`. Write the outcome of each example
1193 with one of the `DocTestRunner.report_*` methods, using the
1194 writer function `out`. `compileflags` is the set of compiler
1195 flags that should be used to execute examples. Return a tuple
1196 `(f, t)`, where `t` is the number of examples tried, and `f`
1197 is the number of examples that failed. The examples are run
1198 in the namespace `test.globs`.
1199 """
1200 # Keep track of the number of failures and tries.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001201 failures = tries = 0
Tim Peters8485b562004-08-04 18:46:34 +00001202
1203 # Save the option flags (since option directives can be used
1204 # to modify them).
1205 original_optionflags = self.optionflags
1206
Tim Peters1fbf9c52004-09-04 17:21:02 +00001207 SUCCESS, FAILURE, BOOM = range(3) # `outcome` state
1208
1209 check = self._checker.check_output
1210
Tim Peters8485b562004-08-04 18:46:34 +00001211 # Process each example.
Edward Loper2de91ba2004-08-27 02:07:46 +00001212 for examplenum, example in enumerate(test.examples):
1213
Edward Lopera89f88d2004-08-26 02:45:51 +00001214 # If REPORT_ONLY_FIRST_FAILURE is set, then supress
1215 # reporting after the first failure.
1216 quiet = (self.optionflags & REPORT_ONLY_FIRST_FAILURE and
1217 failures > 0)
1218
Edward Loper74bca7a2004-08-12 02:27:44 +00001219 # Merge in the example's options.
1220 self.optionflags = original_optionflags
1221 if example.options:
1222 for (optionflag, val) in example.options.items():
1223 if val:
1224 self.optionflags |= optionflag
1225 else:
1226 self.optionflags &= ~optionflag
Tim Peters8485b562004-08-04 18:46:34 +00001227
1228 # Record that we started this example.
1229 tries += 1
Edward Lopera89f88d2004-08-26 02:45:51 +00001230 if not quiet:
1231 self.report_start(out, test, example)
Tim Peters8485b562004-08-04 18:46:34 +00001232
Edward Loper2de91ba2004-08-27 02:07:46 +00001233 # Use a special filename for compile(), so we can retrieve
1234 # the source code during interactive debugging (see
1235 # __patched_linecache_getlines).
1236 filename = '<doctest %s[%d]>' % (test.name, examplenum)
1237
Tim Peters8485b562004-08-04 18:46:34 +00001238 # Run the example in the given context (globs), and record
1239 # any exception that gets raised. (But don't intercept
1240 # keyboard interrupts.)
1241 try:
Tim Peters208ca702004-08-09 04:12:36 +00001242 # Don't blink! This is where the user's code gets run.
Edward Loper2de91ba2004-08-27 02:07:46 +00001243 exec compile(example.source, filename, "single",
Tim Peters8485b562004-08-04 18:46:34 +00001244 compileflags, 1) in test.globs
Edward Loper2de91ba2004-08-27 02:07:46 +00001245 self.debugger.set_continue() # ==== Example Finished ====
Tim Peters8485b562004-08-04 18:46:34 +00001246 exception = None
1247 except KeyboardInterrupt:
1248 raise
1249 except:
1250 exception = sys.exc_info()
Edward Loper2de91ba2004-08-27 02:07:46 +00001251 self.debugger.set_continue() # ==== Example Finished ====
Tim Peters8485b562004-08-04 18:46:34 +00001252
Tim Peters208ca702004-08-09 04:12:36 +00001253 got = self._fakeout.getvalue() # the actual output
Tim Peters8485b562004-08-04 18:46:34 +00001254 self._fakeout.truncate(0)
Tim Peters1fbf9c52004-09-04 17:21:02 +00001255 outcome = FAILURE # guilty until proved innocent or insane
Tim Peters8485b562004-08-04 18:46:34 +00001256
1257 # If the example executed without raising any exceptions,
Tim Peters1fbf9c52004-09-04 17:21:02 +00001258 # verify its output.
Tim Peters8485b562004-08-04 18:46:34 +00001259 if exception is None:
Tim Peters1fbf9c52004-09-04 17:21:02 +00001260 if check(example.want, got, self.optionflags):
1261 outcome = SUCCESS
Tim Peters8485b562004-08-04 18:46:34 +00001262
Tim Peters1fbf9c52004-09-04 17:21:02 +00001263 # The example raised an exception: check if it was expected.
Tim Peters8485b562004-08-04 18:46:34 +00001264 else:
1265 exc_info = sys.exc_info()
1266 exc_msg = traceback.format_exception_only(*exc_info[:2])[-1]
Tim Peters1fbf9c52004-09-04 17:21:02 +00001267 if not quiet:
1268 got += _exception_traceback(exc_info)
Tim Peters8485b562004-08-04 18:46:34 +00001269
Tim Peters1fbf9c52004-09-04 17:21:02 +00001270 # If `example.exc_msg` is None, then we weren't expecting
1271 # an exception.
Edward Lopera6b68322004-08-26 00:05:43 +00001272 if example.exc_msg is None:
Tim Peters1fbf9c52004-09-04 17:21:02 +00001273 outcome = BOOM
1274
1275 # We expected an exception: see whether it matches.
1276 elif check(example.exc_msg, exc_msg, self.optionflags):
1277 outcome = SUCCESS
1278
1279 # Another chance if they didn't care about the detail.
1280 elif self.optionflags & IGNORE_EXCEPTION_DETAIL:
1281 m1 = re.match(r'[^:]*:', example.exc_msg)
1282 m2 = re.match(r'[^:]*:', exc_msg)
1283 if m1 and m2 and check(m1.group(0), m2.group(0),
1284 self.optionflags):
1285 outcome = SUCCESS
1286
1287 # Report the outcome.
1288 if outcome is SUCCESS:
1289 if not quiet:
1290 self.report_success(out, test, example, got)
1291 elif outcome is FAILURE:
1292 if not quiet:
1293 self.report_failure(out, test, example, got)
1294 failures += 1
1295 elif outcome is BOOM:
1296 if not quiet:
1297 self.report_unexpected_exception(out, test, example,
1298 exc_info)
1299 failures += 1
1300 else:
1301 assert False, ("unknown outcome", outcome)
Tim Peters8485b562004-08-04 18:46:34 +00001302
1303 # Restore the option flags (in case they were modified)
1304 self.optionflags = original_optionflags
1305
1306 # Record and return the number of failures and tries.
1307 self.__record_outcome(test, failures, tries)
Tim Peters8a7d2d52001-01-16 07:10:57 +00001308 return failures, tries
1309
Tim Peters8485b562004-08-04 18:46:34 +00001310 def __record_outcome(self, test, f, t):
1311 """
1312 Record the fact that the given DocTest (`test`) generated `f`
1313 failures out of `t` tried examples.
1314 """
1315 f2, t2 = self._name2ft.get(test.name, (0,0))
1316 self._name2ft[test.name] = (f+f2, t+t2)
1317 self.failures += f
1318 self.tries += t
1319
Edward Loper2de91ba2004-08-27 02:07:46 +00001320 __LINECACHE_FILENAME_RE = re.compile(r'<doctest '
1321 r'(?P<name>[\w\.]+)'
1322 r'\[(?P<examplenum>\d+)\]>$')
1323 def __patched_linecache_getlines(self, filename):
1324 m = self.__LINECACHE_FILENAME_RE.match(filename)
1325 if m and m.group('name') == self.test.name:
1326 example = self.test.examples[int(m.group('examplenum'))]
1327 return example.source.splitlines(True)
1328 else:
1329 return self.save_linecache_getlines(filename)
1330
Tim Peters8485b562004-08-04 18:46:34 +00001331 def run(self, test, compileflags=None, out=None, clear_globs=True):
1332 """
1333 Run the examples in `test`, and display the results using the
1334 writer function `out`.
1335
1336 The examples are run in the namespace `test.globs`. If
1337 `clear_globs` is true (the default), then this namespace will
1338 be cleared after the test runs, to help with garbage
1339 collection. If you would like to examine the namespace after
1340 the test completes, then use `clear_globs=False`.
1341
1342 `compileflags` gives the set of flags that should be used by
1343 the Python compiler when running the examples. If not
1344 specified, then it will default to the set of future-import
1345 flags that apply to `globs`.
1346
1347 The output of each example is checked using
1348 `DocTestRunner.check_output`, and the results are formatted by
1349 the `DocTestRunner.report_*` methods.
1350 """
Edward Loper2de91ba2004-08-27 02:07:46 +00001351 self.test = test
1352
Tim Peters8485b562004-08-04 18:46:34 +00001353 if compileflags is None:
1354 compileflags = _extract_future_flags(test.globs)
Jim Fulton356fd192004-08-09 11:34:47 +00001355
Tim Peters6c542b72004-08-09 16:43:36 +00001356 save_stdout = sys.stdout
Tim Peters8485b562004-08-04 18:46:34 +00001357 if out is None:
Tim Peters6c542b72004-08-09 16:43:36 +00001358 out = save_stdout.write
1359 sys.stdout = self._fakeout
Tim Peters8485b562004-08-04 18:46:34 +00001360
Edward Loper2de91ba2004-08-27 02:07:46 +00001361 # Patch pdb.set_trace to restore sys.stdout during interactive
1362 # debugging (so it's not still redirected to self._fakeout).
1363 # Note that the interactive output will go to *our*
1364 # save_stdout, even if that's not the real sys.stdout; this
1365 # allows us to write test cases for the set_trace behavior.
Tim Peters6c542b72004-08-09 16:43:36 +00001366 save_set_trace = pdb.set_trace
Edward Loper2de91ba2004-08-27 02:07:46 +00001367 self.debugger = _OutputRedirectingPdb(save_stdout)
1368 self.debugger.reset()
1369 pdb.set_trace = self.debugger.set_trace
1370
1371 # Patch linecache.getlines, so we can see the example's source
1372 # when we're inside the debugger.
1373 self.save_linecache_getlines = linecache.getlines
1374 linecache.getlines = self.__patched_linecache_getlines
1375
Tim Peters8485b562004-08-04 18:46:34 +00001376 try:
Tim Peters8485b562004-08-04 18:46:34 +00001377 return self.__run(test, compileflags, out)
1378 finally:
Tim Peters6c542b72004-08-09 16:43:36 +00001379 sys.stdout = save_stdout
1380 pdb.set_trace = save_set_trace
Edward Loper2de91ba2004-08-27 02:07:46 +00001381 linecache.getlines = self.save_linecache_getlines
Tim Peters8485b562004-08-04 18:46:34 +00001382 if clear_globs:
1383 test.globs.clear()
1384
1385 #/////////////////////////////////////////////////////////////////
1386 # Summarization
1387 #/////////////////////////////////////////////////////////////////
Tim Peters8a7d2d52001-01-16 07:10:57 +00001388 def summarize(self, verbose=None):
1389 """
Tim Peters8485b562004-08-04 18:46:34 +00001390 Print a summary of all the test cases that have been run by
1391 this DocTestRunner, and return a tuple `(f, t)`, where `f` is
1392 the total number of failed examples, and `t` is the total
1393 number of tried examples.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001394
Tim Peters8485b562004-08-04 18:46:34 +00001395 The optional `verbose` argument controls how detailed the
1396 summary is. If the verbosity is not specified, then the
1397 DocTestRunner's verbosity is used.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001398 """
Tim Peters8a7d2d52001-01-16 07:10:57 +00001399 if verbose is None:
Tim Peters8485b562004-08-04 18:46:34 +00001400 verbose = self._verbose
Tim Peters8a7d2d52001-01-16 07:10:57 +00001401 notests = []
1402 passed = []
1403 failed = []
1404 totalt = totalf = 0
Tim Peters8485b562004-08-04 18:46:34 +00001405 for x in self._name2ft.items():
Tim Peters8a7d2d52001-01-16 07:10:57 +00001406 name, (f, t) = x
1407 assert f <= t
Tim Peters8485b562004-08-04 18:46:34 +00001408 totalt += t
1409 totalf += f
Tim Peters8a7d2d52001-01-16 07:10:57 +00001410 if t == 0:
1411 notests.append(name)
1412 elif f == 0:
1413 passed.append( (name, t) )
1414 else:
1415 failed.append(x)
1416 if verbose:
1417 if notests:
1418 print len(notests), "items had no tests:"
1419 notests.sort()
1420 for thing in notests:
1421 print " ", thing
1422 if passed:
1423 print len(passed), "items passed all tests:"
1424 passed.sort()
1425 for thing, count in passed:
1426 print " %3d tests in %s" % (count, thing)
1427 if failed:
Tim Peters8485b562004-08-04 18:46:34 +00001428 print self.DIVIDER
Tim Peters8a7d2d52001-01-16 07:10:57 +00001429 print len(failed), "items had failures:"
1430 failed.sort()
1431 for thing, (f, t) in failed:
1432 print " %3d of %3d in %s" % (f, t, thing)
1433 if verbose:
Tim Peters8485b562004-08-04 18:46:34 +00001434 print totalt, "tests in", len(self._name2ft), "items."
Tim Peters8a7d2d52001-01-16 07:10:57 +00001435 print totalt - totalf, "passed and", totalf, "failed."
1436 if totalf:
1437 print "***Test Failed***", totalf, "failures."
1438 elif verbose:
1439 print "Test passed."
1440 return totalf, totalt
1441
Tim Peters82076ef2004-09-13 00:52:51 +00001442 #/////////////////////////////////////////////////////////////////
1443 # Backward compatibility cruft to maintain doctest.master.
1444 #/////////////////////////////////////////////////////////////////
1445 def merge(self, other):
1446 d = self._name2ft
1447 for name, (f, t) in other._name2ft.items():
1448 if name in d:
1449 print "*** DocTestRunner.merge: '" + name + "' in both" \
1450 " testers; summing outcomes."
1451 f2, t2 = d[name]
1452 f = f + f2
1453 t = t + t2
1454 d[name] = f, t
1455
Edward Loper34fcb142004-08-09 02:45:41 +00001456class OutputChecker:
1457 """
1458 A class used to check the whether the actual output from a doctest
1459 example matches the expected output. `OutputChecker` defines two
1460 methods: `check_output`, which compares a given pair of outputs,
1461 and returns true if they match; and `output_difference`, which
1462 returns a string describing the differences between two outputs.
1463 """
1464 def check_output(self, want, got, optionflags):
1465 """
Edward Loper74bca7a2004-08-12 02:27:44 +00001466 Return True iff the actual output from an example (`got`)
1467 matches the expected output (`want`). These strings are
1468 always considered to match if they are identical; but
1469 depending on what option flags the test runner is using,
1470 several non-exact match types are also possible. See the
1471 documentation for `TestRunner` for more information about
1472 option flags.
Edward Loper34fcb142004-08-09 02:45:41 +00001473 """
1474 # Handle the common case first, for efficiency:
1475 # if they're string-identical, always return true.
1476 if got == want:
1477 return True
1478
1479 # The values True and False replaced 1 and 0 as the return
1480 # value for boolean comparisons in Python 2.3.
1481 if not (optionflags & DONT_ACCEPT_TRUE_FOR_1):
1482 if (got,want) == ("True\n", "1\n"):
1483 return True
1484 if (got,want) == ("False\n", "0\n"):
1485 return True
1486
1487 # <BLANKLINE> can be used as a special sequence to signify a
1488 # blank line, unless the DONT_ACCEPT_BLANKLINE flag is used.
1489 if not (optionflags & DONT_ACCEPT_BLANKLINE):
1490 # Replace <BLANKLINE> in want with a blank line.
1491 want = re.sub('(?m)^%s\s*?$' % re.escape(BLANKLINE_MARKER),
1492 '', want)
1493 # If a line in got contains only spaces, then remove the
1494 # spaces.
1495 got = re.sub('(?m)^\s*?$', '', got)
1496 if got == want:
1497 return True
1498
1499 # This flag causes doctest to ignore any differences in the
1500 # contents of whitespace strings. Note that this can be used
Tim Peters3fa8c202004-08-23 21:43:39 +00001501 # in conjunction with the ELLIPSIS flag.
Tim Peters1cf3aa62004-08-19 06:49:33 +00001502 if optionflags & NORMALIZE_WHITESPACE:
Edward Loper34fcb142004-08-09 02:45:41 +00001503 got = ' '.join(got.split())
1504 want = ' '.join(want.split())
1505 if got == want:
1506 return True
1507
1508 # The ELLIPSIS flag says to let the sequence "..." in `want`
Tim Peters26b3ebb2004-08-19 08:10:08 +00001509 # match any substring in `got`.
Tim Peters1cf3aa62004-08-19 06:49:33 +00001510 if optionflags & ELLIPSIS:
Tim Petersb0a04e12004-08-20 02:08:04 +00001511 if _ellipsis_match(want, got):
Edward Loper34fcb142004-08-09 02:45:41 +00001512 return True
1513
1514 # We didn't find any match; return false.
1515 return False
1516
Tim Petersc6cbab02004-08-22 19:43:28 +00001517 # Should we do a fancy diff?
1518 def _do_a_fancy_diff(self, want, got, optionflags):
1519 # Not unless they asked for a fancy diff.
Edward Loper71f55af2004-08-26 01:41:51 +00001520 if not optionflags & (REPORT_UDIFF |
1521 REPORT_CDIFF |
1522 REPORT_NDIFF):
Tim Petersc6cbab02004-08-22 19:43:28 +00001523 return False
Tim Peters5b799c12004-08-26 05:21:59 +00001524
Tim Petersc6cbab02004-08-22 19:43:28 +00001525 # If expected output uses ellipsis, a meaningful fancy diff is
Tim Peters5b799c12004-08-26 05:21:59 +00001526 # too hard ... or maybe not. In two real-life failures Tim saw,
1527 # a diff was a major help anyway, so this is commented out.
1528 # [todo] _ellipsis_match() knows which pieces do and don't match,
1529 # and could be the basis for a kick-ass diff in this case.
1530 ##if optionflags & ELLIPSIS and ELLIPSIS_MARKER in want:
1531 ## return False
1532
Tim Petersc6cbab02004-08-22 19:43:28 +00001533 # ndiff does intraline difference marking, so can be useful even
Tim Peters5b799c12004-08-26 05:21:59 +00001534 # for 1-line differences.
Edward Loper71f55af2004-08-26 01:41:51 +00001535 if optionflags & REPORT_NDIFF:
Tim Petersc6cbab02004-08-22 19:43:28 +00001536 return True
Tim Peters5b799c12004-08-26 05:21:59 +00001537
Tim Petersc6cbab02004-08-22 19:43:28 +00001538 # The other diff types need at least a few lines to be helpful.
1539 return want.count('\n') > 2 and got.count('\n') > 2
1540
Edward Loperca9111e2004-08-26 03:00:24 +00001541 def output_difference(self, example, got, optionflags):
Edward Loper34fcb142004-08-09 02:45:41 +00001542 """
1543 Return a string describing the differences between the
Edward Loperca9111e2004-08-26 03:00:24 +00001544 expected output for a given example (`example`) and the actual
1545 output (`got`). `optionflags` is the set of option flags used
1546 to compare `want` and `got`.
Edward Loper34fcb142004-08-09 02:45:41 +00001547 """
Edward Loperca9111e2004-08-26 03:00:24 +00001548 want = example.want
Edward Loper68ba9a62004-08-12 02:43:49 +00001549 # If <BLANKLINE>s are being used, then replace blank lines
1550 # with <BLANKLINE> in the actual output string.
Edward Loper34fcb142004-08-09 02:45:41 +00001551 if not (optionflags & DONT_ACCEPT_BLANKLINE):
Edward Loper68ba9a62004-08-12 02:43:49 +00001552 got = re.sub('(?m)^[ ]*(?=\n)', BLANKLINE_MARKER, got)
Edward Loper34fcb142004-08-09 02:45:41 +00001553
Tim Peters5b799c12004-08-26 05:21:59 +00001554 # Check if we should use diff.
Tim Petersc6cbab02004-08-22 19:43:28 +00001555 if self._do_a_fancy_diff(want, got, optionflags):
Edward Loper34fcb142004-08-09 02:45:41 +00001556 # Split want & got into lines.
Tim Peterse7edcb82004-08-26 05:44:27 +00001557 want_lines = want.splitlines(True) # True == keep line ends
1558 got_lines = got.splitlines(True)
Edward Loper34fcb142004-08-09 02:45:41 +00001559 # Use difflib to find their differences.
Edward Loper71f55af2004-08-26 01:41:51 +00001560 if optionflags & REPORT_UDIFF:
Edward Loper56629292004-08-26 01:31:56 +00001561 diff = difflib.unified_diff(want_lines, got_lines, n=2)
1562 diff = list(diff)[2:] # strip the diff header
1563 kind = 'unified diff with -expected +actual'
Edward Loper71f55af2004-08-26 01:41:51 +00001564 elif optionflags & REPORT_CDIFF:
Edward Loper56629292004-08-26 01:31:56 +00001565 diff = difflib.context_diff(want_lines, got_lines, n=2)
1566 diff = list(diff)[2:] # strip the diff header
1567 kind = 'context diff with expected followed by actual'
Edward Loper71f55af2004-08-26 01:41:51 +00001568 elif optionflags & REPORT_NDIFF:
Tim Petersc6cbab02004-08-22 19:43:28 +00001569 engine = difflib.Differ(charjunk=difflib.IS_CHARACTER_JUNK)
1570 diff = list(engine.compare(want_lines, got_lines))
1571 kind = 'ndiff with -expected +actual'
Edward Loper34fcb142004-08-09 02:45:41 +00001572 else:
1573 assert 0, 'Bad diff option'
1574 # Remove trailing whitespace on diff output.
1575 diff = [line.rstrip() + '\n' for line in diff]
Edward Loperaacf0832004-08-26 01:19:50 +00001576 return 'Differences (%s):\n' % kind + _indent(''.join(diff))
Edward Loper34fcb142004-08-09 02:45:41 +00001577
1578 # If we're not using diff, then simply list the expected
1579 # output followed by the actual output.
Edward Loperaacf0832004-08-26 01:19:50 +00001580 if want and got:
1581 return 'Expected:\n%sGot:\n%s' % (_indent(want), _indent(got))
1582 elif want:
1583 return 'Expected:\n%sGot nothing\n' % _indent(want)
1584 elif got:
1585 return 'Expected nothing\nGot:\n%s' % _indent(got)
1586 else:
1587 return 'Expected nothing\nGot nothing\n'
Edward Loper34fcb142004-08-09 02:45:41 +00001588
Tim Peters19397e52004-08-06 22:02:59 +00001589class DocTestFailure(Exception):
1590 """A DocTest example has failed in debugging mode.
1591
1592 The exception instance has variables:
1593
1594 - test: the DocTest object being run
1595
1596 - excample: the Example object that failed
1597
1598 - got: the actual output
1599 """
1600 def __init__(self, test, example, got):
1601 self.test = test
1602 self.example = example
1603 self.got = got
1604
1605 def __str__(self):
1606 return str(self.test)
1607
1608class UnexpectedException(Exception):
1609 """A DocTest example has encountered an unexpected exception
1610
1611 The exception instance has variables:
1612
1613 - test: the DocTest object being run
1614
1615 - excample: the Example object that failed
1616
1617 - exc_info: the exception info
1618 """
1619 def __init__(self, test, example, exc_info):
1620 self.test = test
1621 self.example = example
1622 self.exc_info = exc_info
1623
1624 def __str__(self):
1625 return str(self.test)
Tim Petersd1b78272004-08-07 06:03:09 +00001626
Tim Peters19397e52004-08-06 22:02:59 +00001627class DebugRunner(DocTestRunner):
1628 r"""Run doc tests but raise an exception as soon as there is a failure.
1629
1630 If an unexpected exception occurs, an UnexpectedException is raised.
1631 It contains the test, the example, and the original exception:
1632
1633 >>> runner = DebugRunner(verbose=False)
Edward Lopera1ef6112004-08-09 16:14:41 +00001634 >>> test = DocTestParser().get_doctest('>>> raise KeyError\n42',
1635 ... {}, 'foo', 'foo.py', 0)
Tim Peters19397e52004-08-06 22:02:59 +00001636 >>> try:
1637 ... runner.run(test)
1638 ... except UnexpectedException, failure:
1639 ... pass
1640
1641 >>> failure.test is test
1642 True
1643
1644 >>> failure.example.want
1645 '42\n'
1646
1647 >>> exc_info = failure.exc_info
1648 >>> raise exc_info[0], exc_info[1], exc_info[2]
1649 Traceback (most recent call last):
1650 ...
1651 KeyError
1652
1653 We wrap the original exception to give the calling application
1654 access to the test and example information.
1655
1656 If the output doesn't match, then a DocTestFailure is raised:
1657
Edward Lopera1ef6112004-08-09 16:14:41 +00001658 >>> test = DocTestParser().get_doctest('''
Tim Peters19397e52004-08-06 22:02:59 +00001659 ... >>> x = 1
1660 ... >>> x
1661 ... 2
1662 ... ''', {}, 'foo', 'foo.py', 0)
1663
1664 >>> try:
1665 ... runner.run(test)
1666 ... except DocTestFailure, failure:
1667 ... pass
1668
1669 DocTestFailure objects provide access to the test:
1670
1671 >>> failure.test is test
1672 True
1673
1674 As well as to the example:
1675
1676 >>> failure.example.want
1677 '2\n'
1678
1679 and the actual output:
1680
1681 >>> failure.got
1682 '1\n'
1683
1684 If a failure or error occurs, the globals are left intact:
1685
1686 >>> del test.globs['__builtins__']
1687 >>> test.globs
1688 {'x': 1}
1689
Edward Lopera1ef6112004-08-09 16:14:41 +00001690 >>> test = DocTestParser().get_doctest('''
Tim Peters19397e52004-08-06 22:02:59 +00001691 ... >>> x = 2
1692 ... >>> raise KeyError
1693 ... ''', {}, 'foo', 'foo.py', 0)
1694
1695 >>> runner.run(test)
1696 Traceback (most recent call last):
1697 ...
1698 UnexpectedException: <DocTest foo from foo.py:0 (2 examples)>
Tim Petersd1b78272004-08-07 06:03:09 +00001699
Tim Peters19397e52004-08-06 22:02:59 +00001700 >>> del test.globs['__builtins__']
1701 >>> test.globs
1702 {'x': 2}
1703
1704 But the globals are cleared if there is no error:
1705
Edward Lopera1ef6112004-08-09 16:14:41 +00001706 >>> test = DocTestParser().get_doctest('''
Tim Peters19397e52004-08-06 22:02:59 +00001707 ... >>> x = 2
1708 ... ''', {}, 'foo', 'foo.py', 0)
1709
1710 >>> runner.run(test)
1711 (0, 1)
1712
1713 >>> test.globs
1714 {}
1715
1716 """
1717
1718 def run(self, test, compileflags=None, out=None, clear_globs=True):
1719 r = DocTestRunner.run(self, test, compileflags, out, False)
1720 if clear_globs:
1721 test.globs.clear()
1722 return r
1723
1724 def report_unexpected_exception(self, out, test, example, exc_info):
1725 raise UnexpectedException(test, example, exc_info)
1726
1727 def report_failure(self, out, test, example, got):
1728 raise DocTestFailure(test, example, got)
1729
Tim Peters8485b562004-08-04 18:46:34 +00001730######################################################################
Edward Loper7c748462004-08-09 02:06:06 +00001731## 6. Test Functions
Tim Peters8485b562004-08-04 18:46:34 +00001732######################################################################
1733# These should be backwards compatible.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001734
Tim Peters82076ef2004-09-13 00:52:51 +00001735# For backward compatibility, a global instance of a DocTestRunner
1736# class, updated by testmod.
1737master = None
1738
Martin v. Löwis4581cfa2002-11-22 08:23:09 +00001739def testmod(m=None, name=None, globs=None, verbose=None, isprivate=None,
Tim Peters19397e52004-08-06 22:02:59 +00001740 report=True, optionflags=0, extraglobs=None,
Tim Peters958cc892004-09-13 14:53:28 +00001741 raise_on_error=False, exclude_empty=False):
Tim Peters6ebe61f2003-06-27 20:48:05 +00001742 """m=None, name=None, globs=None, verbose=None, isprivate=None,
Tim Peters958cc892004-09-13 14:53:28 +00001743 report=True, optionflags=0, extraglobs=None, raise_on_error=False,
1744 exclude_empty=False
Tim Peters8a7d2d52001-01-16 07:10:57 +00001745
Martin v. Löwis4581cfa2002-11-22 08:23:09 +00001746 Test examples in docstrings in functions and classes reachable
1747 from module m (or the current module if m is not supplied), starting
Raymond Hettinger71adf7e2003-07-16 19:25:22 +00001748 with m.__doc__. Unless isprivate is specified, private names
1749 are not skipped.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001750
1751 Also test examples reachable from dict m.__test__ if it exists and is
Tim Petersc2388a22004-08-10 01:41:28 +00001752 not None. m.__test__ maps names to functions, classes and strings;
Tim Peters8a7d2d52001-01-16 07:10:57 +00001753 function and class docstrings are tested even if the name is private;
1754 strings are tested directly, as if they were docstrings.
1755
1756 Return (#failures, #tests).
1757
1758 See doctest.__doc__ for an overview.
1759
1760 Optional keyword arg "name" gives the name of the module; by default
1761 use m.__name__.
1762
1763 Optional keyword arg "globs" gives a dict to be used as the globals
1764 when executing examples; by default, use m.__dict__. A copy of this
1765 dict is actually used for each docstring, so that each docstring's
1766 examples start with a clean slate.
1767
Tim Peters8485b562004-08-04 18:46:34 +00001768 Optional keyword arg "extraglobs" gives a dictionary that should be
1769 merged into the globals that are used to execute examples. By
1770 default, no extra globals are used. This is new in 2.4.
1771
Tim Peters8a7d2d52001-01-16 07:10:57 +00001772 Optional keyword arg "verbose" prints lots of stuff if true, prints
1773 only failures if false; by default, it's true iff "-v" is in sys.argv.
1774
Tim Peters8a7d2d52001-01-16 07:10:57 +00001775 Optional keyword arg "report" prints a summary at the end when true,
1776 else prints nothing at the end. In verbose mode, the summary is
1777 detailed, else very brief (in fact, empty if all tests passed).
1778
Tim Peters6ebe61f2003-06-27 20:48:05 +00001779 Optional keyword arg "optionflags" or's together module constants,
Tim Petersf82a9de2004-08-22 20:51:53 +00001780 and defaults to 0. This is new in 2.3. Possible values (see the
1781 docs for details):
Tim Peters6ebe61f2003-06-27 20:48:05 +00001782
1783 DONT_ACCEPT_TRUE_FOR_1
Tim Peters8485b562004-08-04 18:46:34 +00001784 DONT_ACCEPT_BLANKLINE
Tim Peters8485b562004-08-04 18:46:34 +00001785 NORMALIZE_WHITESPACE
Tim Peters8485b562004-08-04 18:46:34 +00001786 ELLIPSIS
Edward Loper052d0cd2004-09-19 17:19:33 +00001787 IGNORE_EXCEPTION_DETAIL
Edward Loper71f55af2004-08-26 01:41:51 +00001788 REPORT_UDIFF
1789 REPORT_CDIFF
1790 REPORT_NDIFF
Edward Lopera89f88d2004-08-26 02:45:51 +00001791 REPORT_ONLY_FIRST_FAILURE
Tim Peters19397e52004-08-06 22:02:59 +00001792
1793 Optional keyword arg "raise_on_error" raises an exception on the
1794 first unexpected exception or failure. This allows failures to be
1795 post-mortem debugged.
1796
Tim Petersf727c6c2004-08-08 01:48:59 +00001797 Deprecated in Python 2.4:
1798 Optional keyword arg "isprivate" specifies a function used to
1799 determine whether a name is private. The default function is
1800 treat all functions as public. Optionally, "isprivate" can be
1801 set to doctest.is_private to skip over functions marked as private
1802 using the underscore naming convention; see its docs for details.
Tim Peters8485b562004-08-04 18:46:34 +00001803
Tim Peters8a7d2d52001-01-16 07:10:57 +00001804 Advanced tomfoolery: testmod runs methods of a local instance of
1805 class doctest.Tester, then merges the results into (or creates)
1806 global Tester instance doctest.master. Methods of doctest.master
1807 can be called directly too, if you want to do something unusual.
1808 Passing report=0 to testmod is especially useful then, to delay
1809 displaying a summary. Invoke doctest.master.summarize(verbose)
1810 when you're done fiddling.
1811 """
Tim Peters82076ef2004-09-13 00:52:51 +00001812 global master
1813
Tim Petersf727c6c2004-08-08 01:48:59 +00001814 if isprivate is not None:
1815 warnings.warn("the isprivate argument is deprecated; "
1816 "examine DocTestFinder.find() lists instead",
1817 DeprecationWarning)
1818
Tim Peters8485b562004-08-04 18:46:34 +00001819 # If no module was given, then use __main__.
Martin v. Löwis4581cfa2002-11-22 08:23:09 +00001820 if m is None:
Martin v. Löwis4581cfa2002-11-22 08:23:09 +00001821 # DWA - m will still be None if this wasn't invoked from the command
1822 # line, in which case the following TypeError is about as good an error
1823 # as we should expect
1824 m = sys.modules.get('__main__')
1825
Tim Peters8485b562004-08-04 18:46:34 +00001826 # Check that we were actually given a module.
1827 if not inspect.ismodule(m):
Walter Dörwald70a6b492004-02-12 17:35:32 +00001828 raise TypeError("testmod: module required; %r" % (m,))
Tim Peters8485b562004-08-04 18:46:34 +00001829
1830 # If no name was given, then use the module's name.
Tim Peters8a7d2d52001-01-16 07:10:57 +00001831 if name is None:
1832 name = m.__name__
Tim Peters8485b562004-08-04 18:46:34 +00001833
1834 # Find, parse, and run all tests in the given module.
Tim Peters958cc892004-09-13 14:53:28 +00001835 finder = DocTestFinder(_namefilter=isprivate, exclude_empty=exclude_empty)
Tim Peters19397e52004-08-06 22:02:59 +00001836
1837 if raise_on_error:
1838 runner = DebugRunner(verbose=verbose, optionflags=optionflags)
1839 else:
1840 runner = DocTestRunner(verbose=verbose, optionflags=optionflags)
1841
Tim Peters8485b562004-08-04 18:46:34 +00001842 for test in finder.find(m, name, globs=globs, extraglobs=extraglobs):
1843 runner.run(test)
1844
Tim Peters8a7d2d52001-01-16 07:10:57 +00001845 if report:
Tim Peters8485b562004-08-04 18:46:34 +00001846 runner.summarize()
Tim Peters8a7d2d52001-01-16 07:10:57 +00001847
Tim Peters82076ef2004-09-13 00:52:51 +00001848 if master is None:
1849 master = runner
1850 else:
1851 master.merge(runner)
1852
Tim Peters8485b562004-08-04 18:46:34 +00001853 return runner.failures, runner.tries
Tim Petersdb3756d2003-06-29 05:30:48 +00001854
Edward Loper052d0cd2004-09-19 17:19:33 +00001855def testfile(filename, module_relative=True, name=None, package=None,
1856 globs=None, verbose=None, report=True, optionflags=0,
Edward Loper498a1862004-09-27 03:42:58 +00001857 extraglobs=None, raise_on_error=False, parser=DocTestParser()):
Edward Loper052d0cd2004-09-19 17:19:33 +00001858 """
1859 Test examples in the given file. Return (#failures, #tests).
1860
1861 Optional keyword arg "module_relative" specifies how filenames
1862 should be interpreted:
1863
1864 - If "module_relative" is True (the default), then "filename"
1865 specifies a module-relative path. By default, this path is
1866 relative to the calling module's directory; but if the
1867 "package" argument is specified, then it is relative to that
1868 package. To ensure os-independence, "filename" should use
1869 "/" characters to separate path segments, and should not
1870 be an absolute path (i.e., it may not begin with "/").
1871
1872 - If "module_relative" is False, then "filename" specifies an
1873 os-specific path. The path may be absolute or relative (to
1874 the current working directory).
1875
Edward Lopera2fc7ec2004-09-21 03:24:24 +00001876 Optional keyword arg "name" gives the name of the test; by default
1877 use the file's basename.
Edward Loper052d0cd2004-09-19 17:19:33 +00001878
1879 Optional keyword argument "package" is a Python package or the
1880 name of a Python package whose directory should be used as the
1881 base directory for a module relative filename. If no package is
1882 specified, then the calling module's directory is used as the base
1883 directory for module relative filenames. It is an error to
1884 specify "package" if "module_relative" is False.
1885
1886 Optional keyword arg "globs" gives a dict to be used as the globals
1887 when executing examples; by default, use {}. A copy of this dict
1888 is actually used for each docstring, so that each docstring's
1889 examples start with a clean slate.
1890
1891 Optional keyword arg "extraglobs" gives a dictionary that should be
1892 merged into the globals that are used to execute examples. By
1893 default, no extra globals are used.
1894
1895 Optional keyword arg "verbose" prints lots of stuff if true, prints
1896 only failures if false; by default, it's true iff "-v" is in sys.argv.
1897
1898 Optional keyword arg "report" prints a summary at the end when true,
1899 else prints nothing at the end. In verbose mode, the summary is
1900 detailed, else very brief (in fact, empty if all tests passed).
1901
1902 Optional keyword arg "optionflags" or's together module constants,
1903 and defaults to 0. Possible values (see the docs for details):
1904
1905 DONT_ACCEPT_TRUE_FOR_1
1906 DONT_ACCEPT_BLANKLINE
1907 NORMALIZE_WHITESPACE
1908 ELLIPSIS
1909 IGNORE_EXCEPTION_DETAIL
1910 REPORT_UDIFF
1911 REPORT_CDIFF
1912 REPORT_NDIFF
1913 REPORT_ONLY_FIRST_FAILURE
1914
1915 Optional keyword arg "raise_on_error" raises an exception on the
1916 first unexpected exception or failure. This allows failures to be
1917 post-mortem debugged.
1918
Edward Loper498a1862004-09-27 03:42:58 +00001919 Optional keyword arg "parser" specifies a DocTestParser (or
1920 subclass) that should be used to extract tests from the files.
1921
Edward Loper052d0cd2004-09-19 17:19:33 +00001922 Advanced tomfoolery: testmod runs methods of a local instance of
1923 class doctest.Tester, then merges the results into (or creates)
1924 global Tester instance doctest.master. Methods of doctest.master
1925 can be called directly too, if you want to do something unusual.
1926 Passing report=0 to testmod is especially useful then, to delay
1927 displaying a summary. Invoke doctest.master.summarize(verbose)
1928 when you're done fiddling.
1929 """
1930 global master
1931
1932 if package and not module_relative:
1933 raise ValueError("Package may only be specified for module-"
1934 "relative paths.")
Tim Petersbab3e992004-09-20 19:52:34 +00001935
Edward Loper052d0cd2004-09-19 17:19:33 +00001936 # Relativize the path
1937 if module_relative:
1938 package = _normalize_module(package)
1939 filename = _module_relative_path(package, filename)
1940
1941 # If no name was given, then use the file's name.
1942 if name is None:
Edward Lopera2fc7ec2004-09-21 03:24:24 +00001943 name = os.path.basename(filename)
Edward Loper052d0cd2004-09-19 17:19:33 +00001944
1945 # Assemble the globals.
1946 if globs is None:
1947 globs = {}
1948 else:
1949 globs = globs.copy()
1950 if extraglobs is not None:
1951 globs.update(extraglobs)
1952
1953 if raise_on_error:
1954 runner = DebugRunner(verbose=verbose, optionflags=optionflags)
1955 else:
1956 runner = DocTestRunner(verbose=verbose, optionflags=optionflags)
1957
1958 # Read the file, convert it to a test, and run it.
1959 s = open(filename).read()
Edward Loper498a1862004-09-27 03:42:58 +00001960 test = parser.get_doctest(s, globs, name, filename, 0)
Edward Loper052d0cd2004-09-19 17:19:33 +00001961 runner.run(test)
1962
1963 if report:
1964 runner.summarize()
1965
1966 if master is None:
1967 master = runner
1968 else:
1969 master.merge(runner)
1970
1971 return runner.failures, runner.tries
1972
Tim Peters8485b562004-08-04 18:46:34 +00001973def run_docstring_examples(f, globs, verbose=False, name="NoName",
1974 compileflags=None, optionflags=0):
1975 """
1976 Test examples in the given object's docstring (`f`), using `globs`
1977 as globals. Optional argument `name` is used in failure messages.
1978 If the optional argument `verbose` is true, then generate output
1979 even if there are no failures.
Tim Petersdb3756d2003-06-29 05:30:48 +00001980
Tim Peters8485b562004-08-04 18:46:34 +00001981 `compileflags` gives the set of flags that should be used by the
1982 Python compiler when running the examples. If not specified, then
1983 it will default to the set of future-import flags that apply to
1984 `globs`.
Tim Petersdb3756d2003-06-29 05:30:48 +00001985
Tim Peters8485b562004-08-04 18:46:34 +00001986 Optional keyword arg `optionflags` specifies options for the
1987 testing and output. See the documentation for `testmod` for more
1988 information.
1989 """
1990 # Find, parse, and run all tests in the given module.
1991 finder = DocTestFinder(verbose=verbose, recurse=False)
1992 runner = DocTestRunner(verbose=verbose, optionflags=optionflags)
1993 for test in finder.find(f, name, globs=globs):
1994 runner.run(test, compileflags=compileflags)
Tim Petersdb3756d2003-06-29 05:30:48 +00001995
Tim Peters8485b562004-08-04 18:46:34 +00001996######################################################################
Edward Loper7c748462004-08-09 02:06:06 +00001997## 7. Tester
Tim Peters8485b562004-08-04 18:46:34 +00001998######################################################################
1999# This is provided only for backwards compatibility. It's not
2000# actually used in any way.
Tim Petersdb3756d2003-06-29 05:30:48 +00002001
Tim Peters8485b562004-08-04 18:46:34 +00002002class Tester:
2003 def __init__(self, mod=None, globs=None, verbose=None,
2004 isprivate=None, optionflags=0):
Tim Peters3ddd60a2004-08-08 02:43:33 +00002005
2006 warnings.warn("class Tester is deprecated; "
2007 "use class doctest.DocTestRunner instead",
2008 DeprecationWarning, stacklevel=2)
Tim Peters8485b562004-08-04 18:46:34 +00002009 if mod is None and globs is None:
2010 raise TypeError("Tester.__init__: must specify mod or globs")
Tim Peters4be7a922004-09-12 22:39:46 +00002011 if mod is not None and not inspect.ismodule(mod):
Tim Peters8485b562004-08-04 18:46:34 +00002012 raise TypeError("Tester.__init__: mod must be a module; %r" %
2013 (mod,))
2014 if globs is None:
2015 globs = mod.__dict__
2016 self.globs = globs
Tim Petersdb3756d2003-06-29 05:30:48 +00002017
Tim Peters8485b562004-08-04 18:46:34 +00002018 self.verbose = verbose
2019 self.isprivate = isprivate
2020 self.optionflags = optionflags
Tim Petersf727c6c2004-08-08 01:48:59 +00002021 self.testfinder = DocTestFinder(_namefilter=isprivate)
Tim Peters8485b562004-08-04 18:46:34 +00002022 self.testrunner = DocTestRunner(verbose=verbose,
2023 optionflags=optionflags)
Tim Petersdb3756d2003-06-29 05:30:48 +00002024
Tim Peters8485b562004-08-04 18:46:34 +00002025 def runstring(self, s, name):
Edward Lopera1ef6112004-08-09 16:14:41 +00002026 test = DocTestParser().get_doctest(s, self.globs, name, None, None)
Tim Peters8485b562004-08-04 18:46:34 +00002027 if self.verbose:
2028 print "Running string", name
2029 (f,t) = self.testrunner.run(test)
2030 if self.verbose:
2031 print f, "of", t, "examples failed in string", name
2032 return (f,t)
Tim Petersdb3756d2003-06-29 05:30:48 +00002033
Tim Petersf3f57472004-08-08 06:11:48 +00002034 def rundoc(self, object, name=None, module=None):
Tim Peters8485b562004-08-04 18:46:34 +00002035 f = t = 0
2036 tests = self.testfinder.find(object, name, module=module,
Tim Petersf3f57472004-08-08 06:11:48 +00002037 globs=self.globs)
Tim Peters8485b562004-08-04 18:46:34 +00002038 for test in tests:
2039 (f2, t2) = self.testrunner.run(test)
2040 (f,t) = (f+f2, t+t2)
2041 return (f,t)
Tim Petersdb3756d2003-06-29 05:30:48 +00002042
Tim Peters8485b562004-08-04 18:46:34 +00002043 def rundict(self, d, name, module=None):
2044 import new
2045 m = new.module(name)
2046 m.__dict__.update(d)
Tim Petersf3f57472004-08-08 06:11:48 +00002047 if module is None:
2048 module = False
2049 return self.rundoc(m, name, module)
Tim Petersdb3756d2003-06-29 05:30:48 +00002050
Tim Peters8485b562004-08-04 18:46:34 +00002051 def run__test__(self, d, name):
2052 import new
2053 m = new.module(name)
2054 m.__test__ = d
Tim Peters9661f9a2004-09-12 22:45:17 +00002055 return self.rundoc(m, name)
Tim Petersdb3756d2003-06-29 05:30:48 +00002056
Tim Peters8485b562004-08-04 18:46:34 +00002057 def summarize(self, verbose=None):
2058 return self.testrunner.summarize(verbose)
Tim Petersdb3756d2003-06-29 05:30:48 +00002059
Tim Peters8485b562004-08-04 18:46:34 +00002060 def merge(self, other):
Tim Peters82076ef2004-09-13 00:52:51 +00002061 self.testrunner.merge(other.testrunner)
Tim Petersdb3756d2003-06-29 05:30:48 +00002062
Tim Peters8485b562004-08-04 18:46:34 +00002063######################################################################
Edward Loper7c748462004-08-09 02:06:06 +00002064## 8. Unittest Support
Tim Peters8485b562004-08-04 18:46:34 +00002065######################################################################
Tim Petersdb3756d2003-06-29 05:30:48 +00002066
Jim Fultonf54bad42004-08-28 14:57:56 +00002067_unittest_reportflags = 0
Tim Peters38330fe2004-08-30 16:19:24 +00002068
Jim Fultonf54bad42004-08-28 14:57:56 +00002069def set_unittest_reportflags(flags):
Tim Peters38330fe2004-08-30 16:19:24 +00002070 """Sets the unittest option flags.
Jim Fultonf54bad42004-08-28 14:57:56 +00002071
2072 The old flag is returned so that a runner could restore the old
2073 value if it wished to:
2074
2075 >>> old = _unittest_reportflags
2076 >>> set_unittest_reportflags(REPORT_NDIFF |
2077 ... REPORT_ONLY_FIRST_FAILURE) == old
2078 True
2079
2080 >>> import doctest
2081 >>> doctest._unittest_reportflags == (REPORT_NDIFF |
2082 ... REPORT_ONLY_FIRST_FAILURE)
2083 True
Tim Petersdf7a2082004-08-29 00:38:17 +00002084
Jim Fultonf54bad42004-08-28 14:57:56 +00002085 Only reporting flags can be set:
2086
2087 >>> set_unittest_reportflags(ELLIPSIS)
2088 Traceback (most recent call last):
2089 ...
Tim Peters38330fe2004-08-30 16:19:24 +00002090 ValueError: ('Only reporting flags allowed', 8)
Jim Fultonf54bad42004-08-28 14:57:56 +00002091
2092 >>> set_unittest_reportflags(old) == (REPORT_NDIFF |
2093 ... REPORT_ONLY_FIRST_FAILURE)
2094 True
Jim Fultonf54bad42004-08-28 14:57:56 +00002095 """
Jim Fultonf54bad42004-08-28 14:57:56 +00002096 global _unittest_reportflags
Tim Peters38330fe2004-08-30 16:19:24 +00002097
2098 if (flags & REPORTING_FLAGS) != flags:
2099 raise ValueError("Only reporting flags allowed", flags)
Jim Fultonf54bad42004-08-28 14:57:56 +00002100 old = _unittest_reportflags
2101 _unittest_reportflags = flags
2102 return old
Tim Petersdf7a2082004-08-29 00:38:17 +00002103
Jim Fultonf54bad42004-08-28 14:57:56 +00002104
Tim Peters19397e52004-08-06 22:02:59 +00002105class DocTestCase(unittest.TestCase):
Tim Petersdb3756d2003-06-29 05:30:48 +00002106
Edward Loper34fcb142004-08-09 02:45:41 +00002107 def __init__(self, test, optionflags=0, setUp=None, tearDown=None,
2108 checker=None):
Jim Fulton07a349c2004-08-22 14:10:00 +00002109
Jim Fultona643b652004-07-14 19:06:50 +00002110 unittest.TestCase.__init__(self)
Tim Peters19397e52004-08-06 22:02:59 +00002111 self._dt_optionflags = optionflags
Edward Loper34fcb142004-08-09 02:45:41 +00002112 self._dt_checker = checker
Tim Peters19397e52004-08-06 22:02:59 +00002113 self._dt_test = test
2114 self._dt_setUp = setUp
2115 self._dt_tearDown = tearDown
Tim Petersdb3756d2003-06-29 05:30:48 +00002116
Jim Fultona643b652004-07-14 19:06:50 +00002117 def setUp(self):
Jim Fultonf54bad42004-08-28 14:57:56 +00002118 test = self._dt_test
Tim Petersdf7a2082004-08-29 00:38:17 +00002119
Tim Peters19397e52004-08-06 22:02:59 +00002120 if self._dt_setUp is not None:
Jim Fultonf54bad42004-08-28 14:57:56 +00002121 self._dt_setUp(test)
Jim Fultona643b652004-07-14 19:06:50 +00002122
2123 def tearDown(self):
Jim Fultonf54bad42004-08-28 14:57:56 +00002124 test = self._dt_test
2125
Tim Peters19397e52004-08-06 22:02:59 +00002126 if self._dt_tearDown is not None:
Jim Fultonf54bad42004-08-28 14:57:56 +00002127 self._dt_tearDown(test)
2128
2129 test.globs.clear()
Jim Fultona643b652004-07-14 19:06:50 +00002130
2131 def runTest(self):
Tim Peters19397e52004-08-06 22:02:59 +00002132 test = self._dt_test
Jim Fultona643b652004-07-14 19:06:50 +00002133 old = sys.stdout
2134 new = StringIO()
Jim Fultonf54bad42004-08-28 14:57:56 +00002135 optionflags = self._dt_optionflags
Tim Petersdf7a2082004-08-29 00:38:17 +00002136
Tim Peters38330fe2004-08-30 16:19:24 +00002137 if not (optionflags & REPORTING_FLAGS):
Jim Fultonf54bad42004-08-28 14:57:56 +00002138 # The option flags don't include any reporting flags,
2139 # so add the default reporting flags
2140 optionflags |= _unittest_reportflags
Tim Petersdf7a2082004-08-29 00:38:17 +00002141
Jim Fultonf54bad42004-08-28 14:57:56 +00002142 runner = DocTestRunner(optionflags=optionflags,
Edward Loper34fcb142004-08-09 02:45:41 +00002143 checker=self._dt_checker, verbose=False)
Tim Peters19397e52004-08-06 22:02:59 +00002144
Jim Fultona643b652004-07-14 19:06:50 +00002145 try:
Tim Peters19397e52004-08-06 22:02:59 +00002146 runner.DIVIDER = "-"*70
Jim Fultonf54bad42004-08-28 14:57:56 +00002147 failures, tries = runner.run(
2148 test, out=new.write, clear_globs=False)
Jim Fultona643b652004-07-14 19:06:50 +00002149 finally:
2150 sys.stdout = old
2151
2152 if failures:
Tim Peters19397e52004-08-06 22:02:59 +00002153 raise self.failureException(self.format_failure(new.getvalue()))
Tim Peters8485b562004-08-04 18:46:34 +00002154
Tim Peters19397e52004-08-06 22:02:59 +00002155 def format_failure(self, err):
2156 test = self._dt_test
2157 if test.lineno is None:
2158 lineno = 'unknown line number'
2159 else:
Jim Fulton07a349c2004-08-22 14:10:00 +00002160 lineno = '%s' % test.lineno
Tim Peters19397e52004-08-06 22:02:59 +00002161 lname = '.'.join(test.name.split('.')[-1:])
2162 return ('Failed doctest test for %s\n'
2163 ' File "%s", line %s, in %s\n\n%s'
2164 % (test.name, test.filename, lineno, lname, err)
2165 )
2166
2167 def debug(self):
2168 r"""Run the test case without results and without catching exceptions
2169
2170 The unit test framework includes a debug method on test cases
2171 and test suites to support post-mortem debugging. The test code
2172 is run in such a way that errors are not caught. This way a
2173 caller can catch the errors and initiate post-mortem debugging.
2174
2175 The DocTestCase provides a debug method that raises
2176 UnexpectedException errors if there is an unexepcted
2177 exception:
2178
Edward Lopera1ef6112004-08-09 16:14:41 +00002179 >>> test = DocTestParser().get_doctest('>>> raise KeyError\n42',
Tim Peters19397e52004-08-06 22:02:59 +00002180 ... {}, 'foo', 'foo.py', 0)
2181 >>> case = DocTestCase(test)
2182 >>> try:
2183 ... case.debug()
2184 ... except UnexpectedException, failure:
2185 ... pass
2186
2187 The UnexpectedException contains the test, the example, and
2188 the original exception:
2189
2190 >>> failure.test is test
2191 True
2192
2193 >>> failure.example.want
2194 '42\n'
2195
2196 >>> exc_info = failure.exc_info
2197 >>> raise exc_info[0], exc_info[1], exc_info[2]
2198 Traceback (most recent call last):
2199 ...
2200 KeyError
2201
2202 If the output doesn't match, then a DocTestFailure is raised:
2203
Edward Lopera1ef6112004-08-09 16:14:41 +00002204 >>> test = DocTestParser().get_doctest('''
Tim Peters19397e52004-08-06 22:02:59 +00002205 ... >>> x = 1
2206 ... >>> x
2207 ... 2
2208 ... ''', {}, 'foo', 'foo.py', 0)
2209 >>> case = DocTestCase(test)
2210
2211 >>> try:
2212 ... case.debug()
2213 ... except DocTestFailure, failure:
2214 ... pass
2215
2216 DocTestFailure objects provide access to the test:
2217
2218 >>> failure.test is test
2219 True
2220
2221 As well as to the example:
2222
2223 >>> failure.example.want
2224 '2\n'
2225
2226 and the actual output:
2227
2228 >>> failure.got
2229 '1\n'
2230
2231 """
2232
Jim Fultonf54bad42004-08-28 14:57:56 +00002233 self.setUp()
Edward Loper34fcb142004-08-09 02:45:41 +00002234 runner = DebugRunner(optionflags=self._dt_optionflags,
2235 checker=self._dt_checker, verbose=False)
Edward Loper3a3817f2004-08-19 19:26:06 +00002236 runner.run(self._dt_test)
Jim Fultonf54bad42004-08-28 14:57:56 +00002237 self.tearDown()
Jim Fultona643b652004-07-14 19:06:50 +00002238
2239 def id(self):
Tim Peters19397e52004-08-06 22:02:59 +00002240 return self._dt_test.name
Jim Fultona643b652004-07-14 19:06:50 +00002241
2242 def __repr__(self):
Tim Peters19397e52004-08-06 22:02:59 +00002243 name = self._dt_test.name.split('.')
Jim Fultona643b652004-07-14 19:06:50 +00002244 return "%s (%s)" % (name[-1], '.'.join(name[:-1]))
2245
2246 __str__ = __repr__
2247
2248 def shortDescription(self):
Tim Peters19397e52004-08-06 22:02:59 +00002249 return "Doctest: " + self._dt_test.name
Jim Fultona643b652004-07-14 19:06:50 +00002250
Jim Fultonf54bad42004-08-28 14:57:56 +00002251def DocTestSuite(module=None, globs=None, extraglobs=None, test_finder=None,
2252 **options):
Tim Peters8485b562004-08-04 18:46:34 +00002253 """
Tim Peters75dc5e12004-08-22 17:50:45 +00002254 Convert doctest tests for a module to a unittest test suite.
Jim Fultona643b652004-07-14 19:06:50 +00002255
Tim Peters19397e52004-08-06 22:02:59 +00002256 This converts each documentation string in a module that
2257 contains doctest tests to a unittest test case. If any of the
2258 tests in a doc string fail, then the test case fails. An exception
2259 is raised showing the name of the file containing the test and a
Jim Fultona643b652004-07-14 19:06:50 +00002260 (sometimes approximate) line number.
2261
Tim Peters19397e52004-08-06 22:02:59 +00002262 The `module` argument provides the module to be tested. The argument
Jim Fultona643b652004-07-14 19:06:50 +00002263 can be either a module or a module name.
2264
2265 If no argument is given, the calling module is used.
Jim Fultonf54bad42004-08-28 14:57:56 +00002266
2267 A number of options may be provided as keyword arguments:
2268
Jim Fultonf54bad42004-08-28 14:57:56 +00002269 setUp
Edward Lopera2fc7ec2004-09-21 03:24:24 +00002270 A set-up function. This is called before running the
Jim Fultonf54bad42004-08-28 14:57:56 +00002271 tests in each file. The setUp function will be passed a DocTest
2272 object. The setUp function can access the test globals as the
2273 globs attribute of the test passed.
2274
2275 tearDown
Edward Lopera2fc7ec2004-09-21 03:24:24 +00002276 A tear-down function. This is called after running the
Jim Fultonf54bad42004-08-28 14:57:56 +00002277 tests in each file. The tearDown function will be passed a DocTest
2278 object. The tearDown function can access the test globals as the
2279 globs attribute of the test passed.
2280
2281 globs
2282 A dictionary containing initial global variables for the tests.
2283
2284 optionflags
2285 A set of doctest option flags expressed as an integer.
Jim Fultona643b652004-07-14 19:06:50 +00002286 """
Jim Fultona643b652004-07-14 19:06:50 +00002287
Tim Peters8485b562004-08-04 18:46:34 +00002288 if test_finder is None:
2289 test_finder = DocTestFinder()
Tim Peters8485b562004-08-04 18:46:34 +00002290
Tim Peters19397e52004-08-06 22:02:59 +00002291 module = _normalize_module(module)
2292 tests = test_finder.find(module, globs=globs, extraglobs=extraglobs)
2293 if globs is None:
2294 globs = module.__dict__
Jim Fultonf54bad42004-08-28 14:57:56 +00002295 if not tests:
2296 # Why do we want to do this? Because it reveals a bug that might
2297 # otherwise be hidden.
Tim Peters19397e52004-08-06 22:02:59 +00002298 raise ValueError(module, "has no tests")
Tim Petersdb3756d2003-06-29 05:30:48 +00002299
2300 tests.sort()
2301 suite = unittest.TestSuite()
Tim Peters8485b562004-08-04 18:46:34 +00002302 for test in tests:
Tim Peters19397e52004-08-06 22:02:59 +00002303 if len(test.examples) == 0:
2304 continue
Tim Peters8485b562004-08-04 18:46:34 +00002305 if not test.filename:
Tim Petersdb3756d2003-06-29 05:30:48 +00002306 filename = module.__file__
Jim Fulton07a349c2004-08-22 14:10:00 +00002307 if filename[-4:] in (".pyc", ".pyo"):
Tim Petersdb3756d2003-06-29 05:30:48 +00002308 filename = filename[:-1]
Tim Peters8485b562004-08-04 18:46:34 +00002309 test.filename = filename
Jim Fultonf54bad42004-08-28 14:57:56 +00002310 suite.addTest(DocTestCase(test, **options))
Tim Peters19397e52004-08-06 22:02:59 +00002311
2312 return suite
2313
2314class DocFileCase(DocTestCase):
2315
2316 def id(self):
2317 return '_'.join(self._dt_test.name.split('.'))
2318
2319 def __repr__(self):
2320 return self._dt_test.filename
2321 __str__ = __repr__
2322
2323 def format_failure(self, err):
2324 return ('Failed doctest test for %s\n File "%s", line 0\n\n%s'
2325 % (self._dt_test.name, self._dt_test.filename, err)
2326 )
2327
Edward Loper052d0cd2004-09-19 17:19:33 +00002328def DocFileTest(path, module_relative=True, package=None,
Edward Loper498a1862004-09-27 03:42:58 +00002329 globs=None, parser=DocTestParser(), **options):
Tim Peters19397e52004-08-06 22:02:59 +00002330 if globs is None:
2331 globs = {}
2332
Edward Loper052d0cd2004-09-19 17:19:33 +00002333 if package and not module_relative:
2334 raise ValueError("Package may only be specified for module-"
2335 "relative paths.")
Tim Petersbab3e992004-09-20 19:52:34 +00002336
Edward Loper052d0cd2004-09-19 17:19:33 +00002337 # Relativize the path.
2338 if module_relative:
2339 package = _normalize_module(package)
2340 path = _module_relative_path(package, path)
Tim Peters19397e52004-08-06 22:02:59 +00002341
Edward Loper052d0cd2004-09-19 17:19:33 +00002342 # Find the file and read it.
Edward Lopera2fc7ec2004-09-21 03:24:24 +00002343 name = os.path.basename(path)
Edward Loper052d0cd2004-09-19 17:19:33 +00002344 doc = open(path).read()
2345
2346 # Convert it to a test, and wrap it in a DocFileCase.
Edward Loper498a1862004-09-27 03:42:58 +00002347 test = parser.get_doctest(doc, globs, name, path, 0)
Jim Fultonf54bad42004-08-28 14:57:56 +00002348 return DocFileCase(test, **options)
Tim Peters19397e52004-08-06 22:02:59 +00002349
2350def DocFileSuite(*paths, **kw):
Edward Loper052d0cd2004-09-19 17:19:33 +00002351 """A unittest suite for one or more doctest files.
Tim Petersbab3e992004-09-20 19:52:34 +00002352
Edward Loper052d0cd2004-09-19 17:19:33 +00002353 The path to each doctest file is given as a string; the
2354 interpretation of that string depends on the keyword argument
2355 "module_relative".
Tim Peters19397e52004-08-06 22:02:59 +00002356
2357 A number of options may be provided as keyword arguments:
2358
Edward Loper052d0cd2004-09-19 17:19:33 +00002359 module_relative
2360 If "module_relative" is True, then the given file paths are
2361 interpreted as os-independent module-relative paths. By
2362 default, these paths are relative to the calling module's
2363 directory; but if the "package" argument is specified, then
2364 they are relative to that package. To ensure os-independence,
2365 "filename" should use "/" characters to separate path
2366 segments, and may not be an absolute path (i.e., it may not
2367 begin with "/").
Tim Petersbab3e992004-09-20 19:52:34 +00002368
Edward Loper052d0cd2004-09-19 17:19:33 +00002369 If "module_relative" is False, then the given file paths are
2370 interpreted as os-specific paths. These paths may be absolute
2371 or relative (to the current working directory).
2372
Tim Peters19397e52004-08-06 22:02:59 +00002373 package
Edward Loper052d0cd2004-09-19 17:19:33 +00002374 A Python package or the name of a Python package whose directory
2375 should be used as the base directory for module relative paths.
2376 If "package" is not specified, then the calling module's
2377 directory is used as the base directory for module relative
2378 filenames. It is an error to specify "package" if
2379 "module_relative" is False.
Tim Peters19397e52004-08-06 22:02:59 +00002380
2381 setUp
Edward Lopera2fc7ec2004-09-21 03:24:24 +00002382 A set-up function. This is called before running the
Jim Fultonf54bad42004-08-28 14:57:56 +00002383 tests in each file. The setUp function will be passed a DocTest
2384 object. The setUp function can access the test globals as the
2385 globs attribute of the test passed.
Tim Peters19397e52004-08-06 22:02:59 +00002386
2387 tearDown
Edward Lopera2fc7ec2004-09-21 03:24:24 +00002388 A tear-down function. This is called after running the
Jim Fultonf54bad42004-08-28 14:57:56 +00002389 tests in each file. The tearDown function will be passed a DocTest
2390 object. The tearDown function can access the test globals as the
2391 globs attribute of the test passed.
Tim Peters19397e52004-08-06 22:02:59 +00002392
2393 globs
2394 A dictionary containing initial global variables for the tests.
Jim Fultonf54bad42004-08-28 14:57:56 +00002395
2396 optionflags
Edward Loper498a1862004-09-27 03:42:58 +00002397 A set of doctest option flags expressed as an integer.
2398
2399 parser
2400 A DocTestParser (or subclass) that should be used to extract
2401 tests from the files.
Tim Peters19397e52004-08-06 22:02:59 +00002402 """
2403 suite = unittest.TestSuite()
2404
2405 # We do this here so that _normalize_module is called at the right
2406 # level. If it were called in DocFileTest, then this function
2407 # would be the caller and we might guess the package incorrectly.
Edward Loper052d0cd2004-09-19 17:19:33 +00002408 if kw.get('module_relative', True):
2409 kw['package'] = _normalize_module(kw.get('package'))
Tim Peters19397e52004-08-06 22:02:59 +00002410
2411 for path in paths:
2412 suite.addTest(DocFileTest(path, **kw))
Jim Fultona643b652004-07-14 19:06:50 +00002413
Tim Petersdb3756d2003-06-29 05:30:48 +00002414 return suite
2415
Tim Peters8485b562004-08-04 18:46:34 +00002416######################################################################
Edward Loper7c748462004-08-09 02:06:06 +00002417## 9. Debugging Support
Tim Peters8485b562004-08-04 18:46:34 +00002418######################################################################
Jim Fultona643b652004-07-14 19:06:50 +00002419
Tim Peters19397e52004-08-06 22:02:59 +00002420def script_from_examples(s):
2421 r"""Extract script from text with examples.
2422
2423 Converts text with examples to a Python script. Example input is
2424 converted to regular code. Example output and all other words
2425 are converted to comments:
2426
2427 >>> text = '''
2428 ... Here are examples of simple math.
2429 ...
2430 ... Python has super accurate integer addition
2431 ...
2432 ... >>> 2 + 2
2433 ... 5
2434 ...
2435 ... And very friendly error messages:
2436 ...
2437 ... >>> 1/0
2438 ... To Infinity
2439 ... And
2440 ... Beyond
2441 ...
2442 ... You can use logic if you want:
2443 ...
2444 ... >>> if 0:
2445 ... ... blah
2446 ... ... blah
2447 ... ...
2448 ...
2449 ... Ho hum
2450 ... '''
2451
2452 >>> print script_from_examples(text)
Edward Lopera5db6002004-08-12 02:41:30 +00002453 # Here are examples of simple math.
Tim Peters19397e52004-08-06 22:02:59 +00002454 #
Edward Lopera5db6002004-08-12 02:41:30 +00002455 # Python has super accurate integer addition
Tim Peters19397e52004-08-06 22:02:59 +00002456 #
2457 2 + 2
2458 # Expected:
Edward Lopera5db6002004-08-12 02:41:30 +00002459 ## 5
Tim Peters19397e52004-08-06 22:02:59 +00002460 #
Edward Lopera5db6002004-08-12 02:41:30 +00002461 # And very friendly error messages:
Tim Peters19397e52004-08-06 22:02:59 +00002462 #
2463 1/0
2464 # Expected:
Edward Lopera5db6002004-08-12 02:41:30 +00002465 ## To Infinity
2466 ## And
2467 ## Beyond
Tim Peters19397e52004-08-06 22:02:59 +00002468 #
Edward Lopera5db6002004-08-12 02:41:30 +00002469 # You can use logic if you want:
Tim Peters19397e52004-08-06 22:02:59 +00002470 #
2471 if 0:
2472 blah
2473 blah
Tim Peters19397e52004-08-06 22:02:59 +00002474 #
Edward Lopera5db6002004-08-12 02:41:30 +00002475 # Ho hum
Tim Peters19397e52004-08-06 22:02:59 +00002476 """
Edward Loper00f8da72004-08-26 18:05:07 +00002477 output = []
2478 for piece in DocTestParser().parse(s):
2479 if isinstance(piece, Example):
2480 # Add the example's source code (strip trailing NL)
2481 output.append(piece.source[:-1])
2482 # Add the expected output:
2483 want = piece.want
2484 if want:
2485 output.append('# Expected:')
2486 output += ['## '+l for l in want.split('\n')[:-1]]
2487 else:
2488 # Add non-example text.
2489 output += [_comment_line(l)
2490 for l in piece.split('\n')[:-1]]
Tim Peters19397e52004-08-06 22:02:59 +00002491
Edward Loper00f8da72004-08-26 18:05:07 +00002492 # Trim junk on both ends.
2493 while output and output[-1] == '#':
2494 output.pop()
2495 while output and output[0] == '#':
2496 output.pop(0)
2497 # Combine the output, and return it.
2498 return '\n'.join(output)
Tim Petersdb3756d2003-06-29 05:30:48 +00002499
2500def testsource(module, name):
Tim Peters19397e52004-08-06 22:02:59 +00002501 """Extract the test sources from a doctest docstring as a script.
Tim Petersdb3756d2003-06-29 05:30:48 +00002502
2503 Provide the module (or dotted name of the module) containing the
Jim Fultona643b652004-07-14 19:06:50 +00002504 test to be debugged and the name (within the module) of the object
2505 with the doc string with tests to be debugged.
Tim Petersdb3756d2003-06-29 05:30:48 +00002506 """
Tim Peters8485b562004-08-04 18:46:34 +00002507 module = _normalize_module(module)
2508 tests = DocTestFinder().find(module)
2509 test = [t for t in tests if t.name == name]
Tim Petersdb3756d2003-06-29 05:30:48 +00002510 if not test:
2511 raise ValueError(name, "not found in tests")
2512 test = test[0]
Tim Peters19397e52004-08-06 22:02:59 +00002513 testsrc = script_from_examples(test.docstring)
Jim Fultona643b652004-07-14 19:06:50 +00002514 return testsrc
Tim Petersdb3756d2003-06-29 05:30:48 +00002515
Jim Fultona643b652004-07-14 19:06:50 +00002516def debug_src(src, pm=False, globs=None):
Tim Peters19397e52004-08-06 22:02:59 +00002517 """Debug a single doctest docstring, in argument `src`'"""
2518 testsrc = script_from_examples(src)
Tim Peters8485b562004-08-04 18:46:34 +00002519 debug_script(testsrc, pm, globs)
Tim Petersdb3756d2003-06-29 05:30:48 +00002520
Jim Fultona643b652004-07-14 19:06:50 +00002521def debug_script(src, pm=False, globs=None):
Tim Peters19397e52004-08-06 22:02:59 +00002522 "Debug a test script. `src` is the script, as a string."
Tim Petersdb3756d2003-06-29 05:30:48 +00002523 import pdb
Tim Petersdb3756d2003-06-29 05:30:48 +00002524
Tim Petersb6a04d62004-08-23 21:37:56 +00002525 # Note that tempfile.NameTemporaryFile() cannot be used. As the
2526 # docs say, a file so created cannot be opened by name a second time
2527 # on modern Windows boxes, and execfile() needs to open it.
2528 srcfilename = tempfile.mktemp(".py", "doctestdebug")
Tim Peters8485b562004-08-04 18:46:34 +00002529 f = open(srcfilename, 'w')
2530 f.write(src)
2531 f.close()
2532
Tim Petersb6a04d62004-08-23 21:37:56 +00002533 try:
2534 if globs:
2535 globs = globs.copy()
2536 else:
2537 globs = {}
Tim Petersdb3756d2003-06-29 05:30:48 +00002538
Tim Petersb6a04d62004-08-23 21:37:56 +00002539 if pm:
2540 try:
2541 execfile(srcfilename, globs, globs)
2542 except:
2543 print sys.exc_info()[1]
2544 pdb.post_mortem(sys.exc_info()[2])
2545 else:
2546 # Note that %r is vital here. '%s' instead can, e.g., cause
2547 # backslashes to get treated as metacharacters on Windows.
2548 pdb.run("execfile(%r)" % srcfilename, globs, globs)
2549
2550 finally:
2551 os.remove(srcfilename)
Tim Petersdb3756d2003-06-29 05:30:48 +00002552
Jim Fultona643b652004-07-14 19:06:50 +00002553def debug(module, name, pm=False):
Tim Peters19397e52004-08-06 22:02:59 +00002554 """Debug a single doctest docstring.
Jim Fultona643b652004-07-14 19:06:50 +00002555
2556 Provide the module (or dotted name of the module) containing the
2557 test to be debugged and the name (within the module) of the object
Tim Peters19397e52004-08-06 22:02:59 +00002558 with the docstring with tests to be debugged.
Jim Fultona643b652004-07-14 19:06:50 +00002559 """
Tim Peters8485b562004-08-04 18:46:34 +00002560 module = _normalize_module(module)
Jim Fultona643b652004-07-14 19:06:50 +00002561 testsrc = testsource(module, name)
2562 debug_script(testsrc, pm, module.__dict__)
2563
Tim Peters8485b562004-08-04 18:46:34 +00002564######################################################################
Edward Loper7c748462004-08-09 02:06:06 +00002565## 10. Example Usage
Tim Peters8485b562004-08-04 18:46:34 +00002566######################################################################
Tim Peters8a7d2d52001-01-16 07:10:57 +00002567class _TestClass:
2568 """
2569 A pointless class, for sanity-checking of docstring testing.
2570
2571 Methods:
2572 square()
2573 get()
2574
2575 >>> _TestClass(13).get() + _TestClass(-12).get()
2576 1
2577 >>> hex(_TestClass(13).square().get())
2578 '0xa9'
2579 """
2580
2581 def __init__(self, val):
2582 """val -> _TestClass object with associated value val.
2583
2584 >>> t = _TestClass(123)
2585 >>> print t.get()
2586 123
2587 """
2588
2589 self.val = val
2590
2591 def square(self):
2592 """square() -> square TestClass's associated value
2593
2594 >>> _TestClass(13).square().get()
2595 169
2596 """
2597
2598 self.val = self.val ** 2
2599 return self
2600
2601 def get(self):
2602 """get() -> return TestClass's associated value.
2603
2604 >>> x = _TestClass(-42)
2605 >>> print x.get()
2606 -42
2607 """
2608
2609 return self.val
2610
2611__test__ = {"_TestClass": _TestClass,
2612 "string": r"""
2613 Example of a string object, searched as-is.
2614 >>> x = 1; y = 2
2615 >>> x + y, x * y
2616 (3, 2)
Tim Peters6ebe61f2003-06-27 20:48:05 +00002617 """,
Tim Peters3fa8c202004-08-23 21:43:39 +00002618
Tim Peters6ebe61f2003-06-27 20:48:05 +00002619 "bool-int equivalence": r"""
2620 In 2.2, boolean expressions displayed
2621 0 or 1. By default, we still accept
2622 them. This can be disabled by passing
2623 DONT_ACCEPT_TRUE_FOR_1 to the new
2624 optionflags argument.
2625 >>> 4 == 4
2626 1
2627 >>> 4 == 4
2628 True
2629 >>> 4 > 4
2630 0
2631 >>> 4 > 4
2632 False
2633 """,
Tim Peters3fa8c202004-08-23 21:43:39 +00002634
Tim Peters8485b562004-08-04 18:46:34 +00002635 "blank lines": r"""
Tim Peters3fa8c202004-08-23 21:43:39 +00002636 Blank lines can be marked with <BLANKLINE>:
2637 >>> print 'foo\n\nbar\n'
2638 foo
2639 <BLANKLINE>
2640 bar
2641 <BLANKLINE>
Tim Peters8485b562004-08-04 18:46:34 +00002642 """,
Tim Peters3fa8c202004-08-23 21:43:39 +00002643
2644 "ellipsis": r"""
2645 If the ellipsis flag is used, then '...' can be used to
2646 elide substrings in the desired output:
2647 >>> print range(1000) #doctest: +ELLIPSIS
2648 [0, 1, 2, ..., 999]
2649 """,
2650
2651 "whitespace normalization": r"""
2652 If the whitespace normalization flag is used, then
2653 differences in whitespace are ignored.
2654 >>> print range(30) #doctest: +NORMALIZE_WHITESPACE
2655 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
2656 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26,
2657 27, 28, 29]
2658 """,
2659 }
Tim Peters8485b562004-08-04 18:46:34 +00002660
Tim Peters8a7d2d52001-01-16 07:10:57 +00002661def _test():
Tim Peters8485b562004-08-04 18:46:34 +00002662 r = unittest.TextTestRunner()
2663 r.run(DocTestSuite())
Tim Peters8a7d2d52001-01-16 07:10:57 +00002664
2665if __name__ == "__main__":
2666 _test()