blob: b3ef98b879ee5a8ca80c8b128f6915203a519e8c [file] [log] [blame]
Eric S. Raymond630e69c2001-02-09 08:33:43 +00001# Module doctest version 0.9.7
Tim Peters8a7d2d52001-01-16 07:10:57 +00002# Released to the public domain 16-Jan-2001,
3# by Tim Peters (tim.one@home.com).
4
5# Provided as-is; use at your own risk; no warranty; no promises; enjoy!
6
7"""Module doctest -- a framework for running examples in docstrings.
8
9NORMAL USAGE
10
11In normal use, end each module M with:
12
13def _test():
14 import doctest, M # replace M with your module's name
15 return doctest.testmod(M) # ditto
16
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
37You can force verbose mode by passing "verbose=1" to testmod, or prohibit
38it by passing "verbose=0". In either of those cases, sys.argv is not
39examined by testmod.
40
41In any case, testmod returns a 2-tuple of ints (f, t), where f is the
42number of docstring examples that failed and t is the total number of
43docstring examples attempted.
44
45
46WHICH DOCSTRINGS ARE EXAMINED?
47
48+ M.__doc__.
49
50+ f.__doc__ for all functions f in M.__dict__.values(), except those
51 with private names.
52
53+ C.__doc__ for all classes C in M.__dict__.values(), except those with
54 private names.
55
56+ If M.__test__ exists and "is true", it must be a dict, and
57 each entry maps a (string) name to a function object, class object, or
58 string. Function and class object docstrings found from M.__test__
59 are searched even if the name is private, and strings are searched
60 directly as if they were docstrings. In output, a key K in M.__test__
61 appears with name
62 <name of M>.__test__.K
63
64Any classes found are recursively searched similarly, to test docstrings in
65their contained methods and nested classes. Private names reached from M's
66globals are skipped, but all names reached from M.__test__ are searched.
67
68By default, a name is considered to be private if it begins with an
69underscore (like "_my_func") but doesn't both begin and end with (at least)
70two underscores (like "__init__"). You can change the default by passing
71your own "isprivate" function to testmod.
72
73If you want to test docstrings in objects with private names too, stuff
74them into an M.__test__ dict, or see ADVANCED USAGE below (e.g., pass your
75own isprivate function to Tester's constructor, or call the rundoc method
76of a Tester instance).
77
78Warning: imports can cause trouble; e.g., if you do
79
80from XYZ import XYZclass
81
82then XYZclass is a name in M.__dict__ too, and doctest has no way to know
83that XYZclass wasn't *defined* in M. So it may try to execute the examples
84in XYZclass's docstring, and those in turn may require a different set of
85globals to work correctly. I prefer to do "import *"- friendly imports,
86a la
87
88import XYY
89_XYZclass = XYZ.XYZclass
90del XYZ
91
92or (Python 2.0)
93
94from XYZ import XYZclass as _XYZclass
95
96and then the leading underscore stops testmod from going nuts. You may
97prefer the method in the next section.
98
99
100WHAT'S THE EXECUTION CONTEXT?
101
102By default, each time testmod finds a docstring to test, it uses a *copy*
103of M's globals (so that running tests on a module doesn't change the
104module's real globals, and so that one test in M can't leave behind crumbs
105that accidentally allow another test to work). This means examples can
106freely use any names defined at top-level in M. It also means that sloppy
107imports (see above) can cause examples in external docstrings to use
108globals inappropriate for them.
109
110You can force use of your own dict as the execution context by passing
111"globs=your_dict" to testmod instead. Presumably this would be a copy of
112M.__dict__ merged with the globals from other imported modules.
113
114
115WHAT IF I WANT TO TEST A WHOLE PACKAGE?
116
117Piece o' cake, provided the modules do their testing from docstrings.
118Here's the test.py I use for the world's most elaborate Rational/
119floating-base-conversion pkg (which I'll distribute some day):
120
121from Rational import Cvt
122from Rational import Format
123from Rational import machprec
124from Rational import Rat
125from Rational import Round
126from Rational import utils
127
128modules = (Cvt,
129 Format,
130 machprec,
131 Rat,
132 Round,
133 utils)
134
135def _test():
136 import doctest
137 import sys
138 verbose = "-v" in sys.argv
139 for mod in modules:
140 doctest.testmod(mod, verbose=verbose, report=0)
141 doctest.master.summarize()
142
143if __name__ == "__main__":
144 _test()
145
146IOW, it just runs testmod on all the pkg modules. testmod remembers the
147names and outcomes (# of failures, # of tries) for each item it's seen, and
148passing "report=0" prevents it from printing a summary in verbose mode.
149Instead, the summary is delayed until all modules have been tested, and
150then "doctest.master.summarize()" forces the summary at the end.
151
152So this is very nice in practice: each module can be tested individually
153with almost no work beyond writing up docstring examples, and collections
154of modules can be tested too as a unit with no more work than the above.
155
156
157WHAT ABOUT EXCEPTIONS?
158
159No problem, as long as the only output generated by the example is the
160traceback itself. For example:
161
Tim Peters60e23f42001-02-14 00:43:21 +0000162 >>> [1, 2, 3].remove(42)
Tim Petersea4f9312001-02-13 20:54:42 +0000163 Traceback (most recent call last):
Tim Peters8a7d2d52001-01-16 07:10:57 +0000164 File "<stdin>", line 1, in ?
Tim Peters60e23f42001-02-14 00:43:21 +0000165 ValueError: list.remove(x): x not in list
Tim Peters8a7d2d52001-01-16 07:10:57 +0000166 >>>
167
168Note that only the exception type and value are compared (specifically,
169only the last line in the traceback).
170
171
172ADVANCED USAGE
173
174doctest.testmod() captures the testing policy I find most useful most
175often. You may want other policies.
176
177testmod() actually creates a local instance of class doctest.Tester, runs
178appropriate methods of that class, and merges the results into global
179Tester instance doctest.master.
180
181You can create your own instances of doctest.Tester, and so build your own
182policies, or even run methods of doctest.master directly. See
183doctest.Tester.__doc__ for details.
184
185
186SO WHAT DOES A DOCSTRING EXAMPLE LOOK LIKE ALREADY!?
187
188Oh ya. It's easy! In most cases a copy-and-paste of an interactive
189console session works fine -- just make sure the leading whitespace is
190rigidly consistent (you can mix tabs and spaces if you're too lazy to do it
191right, but doctest is not in the business of guessing what you think a tab
192means).
193
194 >>> # comments are ignored
195 >>> x = 12
196 >>> x
197 12
198 >>> if x == 13:
199 ... print "yes"
200 ... else:
201 ... print "no"
202 ... print "NO"
203 ... print "NO!!!"
204 ...
205 no
206 NO
207 NO!!!
208 >>>
209
210Any expected output must immediately follow the final ">>>" or "..." line
211containing the code, and the expected output (if any) extends to the next
212">>>" or all-whitespace line. That's it.
213
214Bummers:
215
216+ Expected output cannot contain an all-whitespace line, since such a line
217 is taken to signal the end of expected output.
218
219+ Output to stdout is captured, but not output to stderr (exception
220 tracebacks are captured via a different means).
221
222+ If you continue a line via backslashing in an interactive session, or for
223 any other reason use a backslash, you need to double the backslash in the
224 docstring version. This is simply because you're in a string, and so the
225 backslash must be escaped for it to survive intact. Like:
226
227>>> if "yes" == \\
228... "y" + \\
229... "es": # in the source code you'll see the doubled backslashes
230... print 'yes'
231yes
232
233The starting column doesn't matter:
234
235>>> assert "Easy!"
236 >>> import math
237 >>> math.floor(1.9)
238 1.0
239
240and as many leading whitespace characters are stripped from the expected
241output as appeared in the initial ">>>" line that triggered it.
242
243If you execute this very file, the examples above will be found and
244executed, leading to this output in verbose mode:
245
246Running doctest.__doc__
Tim Peters60e23f42001-02-14 00:43:21 +0000247Trying: [1, 2, 3].remove(42)
Tim Peters8a7d2d52001-01-16 07:10:57 +0000248Expecting:
Tim Petersea4f9312001-02-13 20:54:42 +0000249Traceback (most recent call last):
Tim Peters8a7d2d52001-01-16 07:10:57 +0000250 File "<stdin>", line 1, in ?
Tim Peters60e23f42001-02-14 00:43:21 +0000251ValueError: list.remove(x): x not in list
Tim Peters8a7d2d52001-01-16 07:10:57 +0000252ok
253Trying: x = 12
254Expecting: nothing
255ok
256Trying: x
257Expecting: 12
258ok
259Trying:
260if x == 13:
261 print "yes"
262else:
263 print "no"
264 print "NO"
265 print "NO!!!"
266Expecting:
267no
268NO
269NO!!!
270ok
271... and a bunch more like that, with this summary at the end:
272
2735 items had no tests:
274 doctest.Tester.__init__
275 doctest.Tester.run__test__
276 doctest.Tester.summarize
277 doctest.run_docstring_examples
278 doctest.testmod
27912 items passed all tests:
280 8 tests in doctest
281 6 tests in doctest.Tester
282 10 tests in doctest.Tester.merge
283 7 tests in doctest.Tester.rundict
284 3 tests in doctest.Tester.rundoc
285 3 tests in doctest.Tester.runstring
286 2 tests in doctest.__test__._TestClass
287 2 tests in doctest.__test__._TestClass.__init__
288 2 tests in doctest.__test__._TestClass.get
289 1 tests in doctest.__test__._TestClass.square
290 2 tests in doctest.__test__.string
291 7 tests in doctest.is_private
29253 tests in 17 items.
29353 passed and 0 failed.
294Test passed.
295"""
296
297# 0,0,1 06-Mar-1999
298# initial version posted
299# 0,0,2 06-Mar-1999
300# loosened parsing:
301# cater to stinkin' tabs
302# don't insist on a blank after PS2 prefix
303# so trailing "... " line from a compound stmt no longer
304# breaks if the file gets whitespace-trimmed
305# better error msgs for inconsistent leading whitespace
306# 0,9,1 08-Mar-1999
307# exposed the Tester class and added client methods
308# plus docstring examples of their use (eww - head-twisting!)
309# fixed logic error in reporting total # of tests & failures
310# added __test__ support to testmod (a pale reflection of Christian
311# Tismer's vision ...)
312# removed the "deep" argument; fiddle __test__ instead
313# simplified endcase logic for extracting tests, and running them.
314# before, if no output was expected but some was produced
315# anyway via an eval'ed result, the discrepancy wasn't caught
316# made TestClass private and used __test__ to get at it
317# many doc updates
318# speed _SpoofOut for long expected outputs
319# 0,9,2 09-Mar-1999
320# throw out comments from examples, enabling use of the much simpler
321# exec compile(... "single") ...
322# for simulating the runtime; that barfs on comment-only lines
323# used the traceback module to do a much better job of reporting
324# exceptions
325# run __doc__ values thru str(), "just in case"
326# privateness of names now determined by an overridable "isprivate"
327# function
328# by default a name now considered to be private iff it begins with
329# an underscore but doesn't both begin & end with two of 'em; so
330# e.g. Class.__init__ etc are searched now -- as they always
331# should have been
332# 0,9,3 18-Mar-1999
333# added .flush stub to _SpoofOut (JPython buglet diagnosed by
334# Hugh Emberson)
335# repaired ridiculous docs about backslashes in examples
336# minor internal changes
337# changed source to Unix line-end conventions
338# moved __test__ logic into new Tester.run__test__ method
339# 0,9,4 27-Mar-1999
340# report item name and line # in failing examples
341# 0,9,5 29-Jun-1999
342# allow straightforward exceptions in examples - thanks to Mark Hammond!
343# 0,9,6 16-Jan-2001
344# fiddling for changes in Python 2.0: some of the embedded docstring
345# examples no longer worked *exactly* as advertised, due to minor
346# language changes, and running doctest on itself pointed that out.
347# Hard to think of a better example of why this is useful <wink>.
Eric S. Raymond630e69c2001-02-09 08:33:43 +0000348# 0,9,7 9-Feb-2001
349# string method conversion
Tim Peters8a7d2d52001-01-16 07:10:57 +0000350
Tim Petersecb6fb92001-02-10 01:24:50 +0000351__version__ = 0, 9, 7
Tim Peters8a7d2d52001-01-16 07:10:57 +0000352
353import types
354_FunctionType = types.FunctionType
355_ClassType = types.ClassType
356_ModuleType = types.ModuleType
357_StringType = types.StringType
358del types
359
Tim Peters8a7d2d52001-01-16 07:10:57 +0000360import re
361PS1 = ">>>"
362PS2 = "..."
363_isPS1 = re.compile(r"(\s*)" + re.escape(PS1)).match
364_isPS2 = re.compile(r"(\s*)" + re.escape(PS2)).match
365_isEmpty = re.compile(r"\s*$").match
366_isComment = re.compile(r"\s*#").match
367del re
368
Skip Montanaroeccd02a2001-01-20 23:34:12 +0000369__all__ = []
370
Tim Peters8a7d2d52001-01-16 07:10:57 +0000371# Extract interactive examples from a string. Return a list of triples,
372# (source, outcome, lineno). "source" is the source code, and ends
373# with a newline iff the source spans more than one line. "outcome" is
374# the expected output if any, else an empty string. When not empty,
375# outcome always ends with a newline. "lineno" is the line number,
376# 0-based wrt the start of the string, of the first source line.
377
378def _extract_examples(s):
379 isPS1, isPS2 = _isPS1, _isPS2
380 isEmpty, isComment = _isEmpty, _isComment
381 examples = []
Eric S. Raymond630e69c2001-02-09 08:33:43 +0000382 lines = s.split("\n")
Tim Peters8a7d2d52001-01-16 07:10:57 +0000383 i, n = 0, len(lines)
384 while i < n:
385 line = lines[i]
386 i = i + 1
387 m = isPS1(line)
388 if m is None:
389 continue
390 j = m.end(0) # beyond the prompt
391 if isEmpty(line, j) or isComment(line, j):
392 # a bare prompt or comment -- not interesting
393 continue
394 lineno = i - 1
395 if line[j] != " ":
396 raise ValueError("line " + `lineno` + " of docstring lacks "
397 "blank after " + PS1 + ": " + line)
398 j = j + 1
399 blanks = m.group(1)
400 nblanks = len(blanks)
401 # suck up this and following PS2 lines
402 source = []
403 while 1:
404 source.append(line[j:])
405 line = lines[i]
406 m = isPS2(line)
407 if m:
408 if m.group(1) != blanks:
409 raise ValueError("inconsistent leading whitespace "
410 "in line " + `i` + " of docstring: " + line)
411 i = i + 1
412 else:
413 break
414 if len(source) == 1:
415 source = source[0]
416 else:
417 # get rid of useless null line from trailing empty "..."
418 if source[-1] == "":
419 del source[-1]
Eric S. Raymond630e69c2001-02-09 08:33:43 +0000420 source = "\n".join(source) + "\n"
Tim Peters8a7d2d52001-01-16 07:10:57 +0000421 # suck up response
422 if isPS1(line) or isEmpty(line):
423 expect = ""
424 else:
425 expect = []
426 while 1:
427 if line[:nblanks] != blanks:
428 raise ValueError("inconsistent leading whitespace "
429 "in line " + `i` + " of docstring: " + line)
430 expect.append(line[nblanks:])
431 i = i + 1
432 line = lines[i]
433 if isPS1(line) or isEmpty(line):
434 break
Eric S. Raymond630e69c2001-02-09 08:33:43 +0000435 expect = "\n".join(expect) + "\n"
Tim Peters8a7d2d52001-01-16 07:10:57 +0000436 examples.append( (source, expect, lineno) )
437 return examples
438
439# Capture stdout when running examples.
440
441class _SpoofOut:
442 def __init__(self):
443 self.clear()
444 def write(self, s):
445 self.buf.append(s)
446 def get(self):
Eric S. Raymond630e69c2001-02-09 08:33:43 +0000447 return "".join(self.buf)
Tim Peters8a7d2d52001-01-16 07:10:57 +0000448 def clear(self):
449 self.buf = []
450 def flush(self):
451 # JPython calls flush
452 pass
453
454# Display some tag-and-msg pairs nicely, keeping the tag and its msg
455# on the same line when that makes sense.
456
457def _tag_out(printer, *tag_msg_pairs):
458 for tag, msg in tag_msg_pairs:
459 printer(tag + ":")
460 msg_has_nl = msg[-1:] == "\n"
461 msg_has_two_nl = msg_has_nl and \
Eric S. Raymond630e69c2001-02-09 08:33:43 +0000462 msg.find("\n") < len(msg) - 1
Tim Peters8a7d2d52001-01-16 07:10:57 +0000463 if len(tag) + len(msg) < 76 and not msg_has_two_nl:
464 printer(" ")
465 else:
466 printer("\n")
467 printer(msg)
468 if not msg_has_nl:
469 printer("\n")
470
471# Run list of examples, in context globs. "out" can be used to display
472# stuff to "the real" stdout, and fakeout is an instance of _SpoofOut
473# that captures the examples' std output. Return (#failures, #tries).
474
475def _run_examples_inner(out, fakeout, examples, globs, verbose, name):
476 import sys, traceback
477 OK, BOOM, FAIL = range(3)
478 NADA = "nothing"
479 stderr = _SpoofOut()
480 failures = 0
481 for source, want, lineno in examples:
482 if verbose:
483 _tag_out(out, ("Trying", source),
484 ("Expecting", want or NADA))
485 fakeout.clear()
486 try:
487 exec compile(source, "<string>", "single") in globs
488 got = fakeout.get()
489 state = OK
490 except:
491 # See whether the exception was expected.
Tim Petersea4f9312001-02-13 20:54:42 +0000492 if want.find("Traceback (innermost last):\n") == 0 or \
493 want.find("Traceback (most recent call last):\n") == 0:
Tim Peters8a7d2d52001-01-16 07:10:57 +0000494 # Only compare exception type and value - the rest of
495 # the traceback isn't necessary.
Eric S. Raymond630e69c2001-02-09 08:33:43 +0000496 want = want.split('\n')[-2] + '\n'
Tim Peters8a7d2d52001-01-16 07:10:57 +0000497 exc_type, exc_val, exc_tb = sys.exc_info()
498 got = traceback.format_exception_only(exc_type, exc_val)[0]
499 state = OK
500 else:
501 # unexpected exception
502 stderr.clear()
503 traceback.print_exc(file=stderr)
504 state = BOOM
505
506 if state == OK:
507 if got == want:
508 if verbose:
509 out("ok\n")
510 continue
511 state = FAIL
512
513 assert state in (FAIL, BOOM)
514 failures = failures + 1
515 out("*" * 65 + "\n")
516 _tag_out(out, ("Failure in example", source))
517 out("from line #" + `lineno` + " of " + name + "\n")
518 if state == FAIL:
519 _tag_out(out, ("Expected", want or NADA), ("Got", got))
520 else:
521 assert state == BOOM
522 _tag_out(out, ("Exception raised", stderr.get()))
523
524 return failures, len(examples)
525
526# Run list of examples, in context globs. Return (#failures, #tries).
527
528def _run_examples(examples, globs, verbose, name):
529 import sys
530 saveout = sys.stdout
531 try:
532 sys.stdout = fakeout = _SpoofOut()
533 x = _run_examples_inner(saveout.write, fakeout, examples,
534 globs, verbose, name)
535 finally:
536 sys.stdout = saveout
537 return x
538
539def run_docstring_examples(f, globs, verbose=0, name="NoName"):
540 """f, globs, verbose=0, name="NoName" -> run examples from f.__doc__.
541
542 Use dict globs as the globals for execution.
543 Return (#failures, #tries).
544
545 If optional arg verbose is true, print stuff even if there are no
546 failures.
547 Use string name in failure msgs.
548 """
549
550 try:
551 doc = f.__doc__
552 if not doc:
553 # docstring empty or None
554 return 0, 0
555 # just in case CT invents a doc object that has to be forced
556 # to look like a string <0.9 wink>
557 doc = str(doc)
558 except:
559 return 0, 0
560
561 e = _extract_examples(doc)
562 if not e:
563 return 0, 0
564 return _run_examples(e, globs, verbose, name)
565
566def is_private(prefix, base):
567 """prefix, base -> true iff name prefix + "." + base is "private".
568
569 Prefix may be an empty string, and base does not contain a period.
570 Prefix is ignored (although functions you write conforming to this
571 protocol may make use of it).
572 Return true iff base begins with an (at least one) underscore, but
573 does not both begin and end with (at least) two underscores.
574
575 >>> is_private("a.b", "my_func")
576 0
577 >>> is_private("____", "_my_func")
578 1
579 >>> is_private("someclass", "__init__")
580 0
581 >>> is_private("sometypo", "__init_")
582 1
583 >>> is_private("x.y.z", "_")
584 1
585 >>> is_private("_x.y.z", "__")
586 0
587 >>> is_private("", "") # senseless but consistent
588 0
589 """
590
591 return base[:1] == "_" and not base[:2] == "__" == base[-2:]
592
593class Tester:
594 """Class Tester -- runs docstring examples and accumulates stats.
595
596In normal use, function doctest.testmod() hides all this from you,
597so use that if you can. Create your own instances of Tester to do
598fancier things.
599
600Methods:
601 runstring(s, name)
602 Search string s for examples to run; use name for logging.
603 Return (#failures, #tries).
604
605 rundoc(object, name=None)
606 Search object.__doc__ for examples to run; use name (or
607 object.__name__) for logging. Return (#failures, #tries).
608
609 rundict(d, name)
610 Search for examples in docstrings in all of d.values(); use name
611 for logging. Return (#failures, #tries).
612
613 run__test__(d, name)
614 Treat dict d like module.__test__. Return (#failures, #tries).
615
616 summarize(verbose=None)
617 Display summary of testing results, to stdout. Return
618 (#failures, #tries).
619
620 merge(other)
621 Merge in the test results from Tester instance "other".
622
623>>> from doctest import Tester
624>>> t = Tester(globs={'x': 42}, verbose=0)
625>>> t.runstring(r'''
626... >>> x = x * 2
627... >>> print x
628... 42
629... ''', 'XYZ')
630*****************************************************************
631Failure in example: print x
632from line #2 of XYZ
633Expected: 42
634Got: 84
635(1, 2)
636>>> t.runstring(">>> x = x * 2\\n>>> print x\\n84\\n", 'example2')
637(0, 2)
638>>> t.summarize()
6391 items had failures:
640 1 of 2 in XYZ
641***Test Failed*** 1 failures.
642(1, 4)
643>>> t.summarize(verbose=1)
6441 items passed all tests:
645 2 tests in example2
6461 items had failures:
647 1 of 2 in XYZ
6484 tests in 2 items.
6493 passed and 1 failed.
650***Test Failed*** 1 failures.
651(1, 4)
652>>>
653"""
654
655 def __init__(self, mod=None, globs=None, verbose=None,
656 isprivate=None):
657 """mod=None, globs=None, verbose=None, isprivate=None
658
659See doctest.__doc__ for an overview.
660
661Optional keyword arg "mod" is a module, whose globals are used for
662executing examples. If not specified, globs must be specified.
663
664Optional keyword arg "globs" gives a dict to be used as the globals
665when executing examples; if not specified, use the globals from
666module mod.
667
668In either case, a copy of the dict is used for each docstring
669examined.
670
671Optional keyword arg "verbose" prints lots of stuff if true, only
672failures if false; by default, it's true iff "-v" is in sys.argv.
673
674Optional keyword arg "isprivate" specifies a function used to determine
675whether a name is private. The default function is doctest.is_private;
676see its docs for details.
677"""
678
679 if mod is None and globs is None:
680 raise TypeError("Tester.__init__: must specify mod or globs")
681 if mod is not None and type(mod) is not _ModuleType:
682 raise TypeError("Tester.__init__: mod must be a module; " +
683 `mod`)
684 if globs is None:
685 globs = mod.__dict__
686 self.globs = globs
687
688 if verbose is None:
689 import sys
690 verbose = "-v" in sys.argv
691 self.verbose = verbose
692
693 if isprivate is None:
694 isprivate = is_private
695 self.isprivate = isprivate
696
697 self.name2ft = {} # map name to (#failures, #trials) pair
698
699 def runstring(self, s, name):
700 """
701 s, name -> search string s for examples to run, logging as name.
702
703 Use string name as the key for logging the outcome.
704 Return (#failures, #examples).
705
706 >>> t = Tester(globs={}, verbose=1)
707 >>> test = r'''
708 ... # just an example
709 ... >>> x = 1 + 2
710 ... >>> x
711 ... 3
712 ... '''
713 >>> t.runstring(test, "Example")
714 Running string Example
715 Trying: x = 1 + 2
716 Expecting: nothing
717 ok
718 Trying: x
719 Expecting: 3
720 ok
721 0 of 2 examples failed in string Example
722 (0, 2)
723 """
724
725 if self.verbose:
726 print "Running string", name
727 f = t = 0
728 e = _extract_examples(s)
729 if e:
730 f, t = _run_examples(e, self.globs.copy(), self.verbose, name)
731 if self.verbose:
732 print f, "of", t, "examples failed in string", name
733 self.__record_outcome(name, f, t)
734 return f, t
735
736 def rundoc(self, object, name=None):
737 """
738 object, name=None -> search object.__doc__ for examples to run.
739
740 Use optional string name as the key for logging the outcome;
741 by default use object.__name__.
742 Return (#failures, #examples).
743 If object is a class object, search recursively for method
744 docstrings too.
745 object.__doc__ is examined regardless of name, but if object is
746 a class, whether private names reached from object are searched
747 depends on the constructor's "isprivate" argument.
748
749 >>> t = Tester(globs={}, verbose=0)
750 >>> def _f():
751 ... '''Trivial docstring example.
752 ... >>> assert 2 == 2
753 ... '''
754 ... return 32
755 ...
756 >>> t.rundoc(_f) # expect 0 failures in 1 example
757 (0, 1)
758 """
759
760 if name is None:
761 try:
762 name = object.__name__
763 except AttributeError:
764 raise ValueError("Tester.rundoc: name must be given "
765 "when object.__name__ doesn't exist; " + `object`)
766 if self.verbose:
767 print "Running", name + ".__doc__"
768 f, t = run_docstring_examples(object, self.globs.copy(),
769 self.verbose, name)
770 if self.verbose:
771 print f, "of", t, "examples failed in", name + ".__doc__"
772 self.__record_outcome(name, f, t)
773 if type(object) is _ClassType:
774 f2, t2 = self.rundict(object.__dict__, name)
775 f = f + f2
776 t = t + t2
777 return f, t
778
779 def rundict(self, d, name):
780 """
781 d. name -> search for docstring examples in all of d.values().
782
783 For k, v in d.items() such that v is a function or class,
784 do self.rundoc(v, name + "." + k). Whether this includes
785 objects with private names depends on the constructor's
786 "isprivate" argument.
787 Return aggregate (#failures, #examples).
788
789 >>> def _f():
790 ... '''>>> assert 1 == 1
791 ... '''
792 >>> def g():
793 ... '''>>> assert 2 != 1
794 ... '''
795 >>> d = {"_f": _f, "g": g}
796 >>> t = Tester(globs={}, verbose=0)
797 >>> t.rundict(d, "rundict_test") # _f is skipped
798 (0, 1)
799 >>> t = Tester(globs={}, verbose=0, isprivate=lambda x,y: 0)
800 >>> t.rundict(d, "rundict_test_pvt") # both are searched
801 (0, 2)
802 """
803
804 if not hasattr(d, "items"):
805 raise TypeError("Tester.rundict: d must support .items(); " +
806 `d`)
807 f = t = 0
808 for thisname, value in d.items():
809 if type(value) in (_FunctionType, _ClassType):
810 f2, t2 = self.__runone(value, name + "." + thisname)
811 f = f + f2
812 t = t + t2
813 return f, t
814
815 def run__test__(self, d, name):
816 """d, name -> Treat dict d like module.__test__.
817
818 Return (#failures, #tries).
819 See testmod.__doc__ for details.
820 """
821
822 failures = tries = 0
823 prefix = name + "."
824 savepvt = self.isprivate
825 try:
826 self.isprivate = lambda *args: 0
827 for k, v in d.items():
828 thisname = prefix + k
829 if type(v) is _StringType:
830 f, t = self.runstring(v, thisname)
831 elif type(v) in (_FunctionType, _ClassType):
832 f, t = self.rundoc(v, thisname)
833 else:
834 raise TypeError("Tester.run__test__: values in "
835 "dict must be strings, functions "
836 "or classes; " + `v`)
837 failures = failures + f
838 tries = tries + t
839 finally:
840 self.isprivate = savepvt
841 return failures, tries
842
843 def summarize(self, verbose=None):
844 """
845 verbose=None -> summarize results, return (#failures, #tests).
846
847 Print summary of test results to stdout.
848 Optional arg 'verbose' controls how wordy this is. By
849 default, use the verbose setting established by the
850 constructor.
851 """
852
853 if verbose is None:
854 verbose = self.verbose
855 notests = []
856 passed = []
857 failed = []
858 totalt = totalf = 0
859 for x in self.name2ft.items():
860 name, (f, t) = x
861 assert f <= t
862 totalt = totalt + t
863 totalf = totalf + f
864 if t == 0:
865 notests.append(name)
866 elif f == 0:
867 passed.append( (name, t) )
868 else:
869 failed.append(x)
870 if verbose:
871 if notests:
872 print len(notests), "items had no tests:"
873 notests.sort()
874 for thing in notests:
875 print " ", thing
876 if passed:
877 print len(passed), "items passed all tests:"
878 passed.sort()
879 for thing, count in passed:
880 print " %3d tests in %s" % (count, thing)
881 if failed:
882 print len(failed), "items had failures:"
883 failed.sort()
884 for thing, (f, t) in failed:
885 print " %3d of %3d in %s" % (f, t, thing)
886 if verbose:
887 print totalt, "tests in", len(self.name2ft), "items."
888 print totalt - totalf, "passed and", totalf, "failed."
889 if totalf:
890 print "***Test Failed***", totalf, "failures."
891 elif verbose:
892 print "Test passed."
893 return totalf, totalt
894
895 def merge(self, other):
896 """
897 other -> merge in test results from the other Tester instance.
898
899 If self and other both have a test result for something
900 with the same name, the (#failures, #tests) results are
901 summed, and a warning is printed to stdout.
902
903 >>> from doctest import Tester
904 >>> t1 = Tester(globs={}, verbose=0)
905 >>> t1.runstring('''
906 ... >>> x = 12
907 ... >>> print x
908 ... 12
909 ... ''', "t1example")
910 (0, 2)
911 >>>
912 >>> t2 = Tester(globs={}, verbose=0)
913 >>> t2.runstring('''
914 ... >>> x = 13
915 ... >>> print x
916 ... 13
917 ... ''', "t2example")
918 (0, 2)
919 >>> common = ">>> assert 1 + 2 == 3\\n"
920 >>> t1.runstring(common, "common")
921 (0, 1)
922 >>> t2.runstring(common, "common")
923 (0, 1)
924 >>> t1.merge(t2)
925 *** Tester.merge: 'common' in both testers; summing outcomes.
926 >>> t1.summarize(1)
927 3 items passed all tests:
928 2 tests in common
929 2 tests in t1example
930 2 tests in t2example
931 6 tests in 3 items.
932 6 passed and 0 failed.
933 Test passed.
934 (0, 6)
935 >>>
936 """
937
938 d = self.name2ft
939 for name, (f, t) in other.name2ft.items():
940 if d.has_key(name):
941 print "*** Tester.merge: '" + name + "' in both" \
942 " testers; summing outcomes."
943 f2, t2 = d[name]
944 f = f + f2
945 t = t + t2
946 d[name] = f, t
947
948 def __record_outcome(self, name, f, t):
949 if self.name2ft.has_key(name):
950 print "*** Warning: '" + name + "' was tested before;", \
951 "summing outcomes."
952 f2, t2 = self.name2ft[name]
953 f = f + f2
954 t = t + t2
955 self.name2ft[name] = f, t
956
957 def __runone(self, target, name):
958 if "." in name:
Eric S. Raymond630e69c2001-02-09 08:33:43 +0000959 i = name.rindex(".")
Tim Peters8a7d2d52001-01-16 07:10:57 +0000960 prefix, base = name[:i], name[i+1:]
961 else:
962 prefix, base = "", base
963 if self.isprivate(prefix, base):
964 return 0, 0
965 return self.rundoc(target, name)
966
967master = None
968
969def testmod(m, name=None, globs=None, verbose=None, isprivate=None,
970 report=1):
971 """m, name=None, globs=None, verbose=None, isprivate=None, report=1
972
973 Test examples in docstrings in functions and classes reachable from
974 module m, starting with m.__doc__. Private names are skipped.
975
976 Also test examples reachable from dict m.__test__ if it exists and is
977 not None. m.__dict__ maps names to functions, classes and strings;
978 function and class docstrings are tested even if the name is private;
979 strings are tested directly, as if they were docstrings.
980
981 Return (#failures, #tests).
982
983 See doctest.__doc__ for an overview.
984
985 Optional keyword arg "name" gives the name of the module; by default
986 use m.__name__.
987
988 Optional keyword arg "globs" gives a dict to be used as the globals
989 when executing examples; by default, use m.__dict__. A copy of this
990 dict is actually used for each docstring, so that each docstring's
991 examples start with a clean slate.
992
993 Optional keyword arg "verbose" prints lots of stuff if true, prints
994 only failures if false; by default, it's true iff "-v" is in sys.argv.
995
996 Optional keyword arg "isprivate" specifies a function used to
997 determine whether a name is private. The default function is
998 doctest.is_private; see its docs for details.
999
1000 Optional keyword arg "report" prints a summary at the end when true,
1001 else prints nothing at the end. In verbose mode, the summary is
1002 detailed, else very brief (in fact, empty if all tests passed).
1003
1004 Advanced tomfoolery: testmod runs methods of a local instance of
1005 class doctest.Tester, then merges the results into (or creates)
1006 global Tester instance doctest.master. Methods of doctest.master
1007 can be called directly too, if you want to do something unusual.
1008 Passing report=0 to testmod is especially useful then, to delay
1009 displaying a summary. Invoke doctest.master.summarize(verbose)
1010 when you're done fiddling.
1011 """
1012
1013 global master
1014
1015 if type(m) is not _ModuleType:
1016 raise TypeError("testmod: module required; " + `m`)
1017 if name is None:
1018 name = m.__name__
1019 tester = Tester(m, globs=globs, verbose=verbose, isprivate=isprivate)
1020 failures, tries = tester.rundoc(m, name)
1021 f, t = tester.rundict(m.__dict__, name)
1022 failures = failures + f
1023 tries = tries + t
1024 if hasattr(m, "__test__"):
1025 testdict = m.__test__
1026 if testdict:
1027 if not hasattr(testdict, "items"):
1028 raise TypeError("testmod: module.__test__ must support "
1029 ".items(); " + `testdict`)
1030 f, t = tester.run__test__(testdict, name + ".__test__")
1031 failures = failures + f
1032 tries = tries + t
1033 if report:
1034 tester.summarize()
1035 if master is None:
1036 master = tester
1037 else:
1038 master.merge(tester)
1039 return failures, tries
1040
1041class _TestClass:
1042 """
1043 A pointless class, for sanity-checking of docstring testing.
1044
1045 Methods:
1046 square()
1047 get()
1048
1049 >>> _TestClass(13).get() + _TestClass(-12).get()
1050 1
1051 >>> hex(_TestClass(13).square().get())
1052 '0xa9'
1053 """
1054
1055 def __init__(self, val):
1056 """val -> _TestClass object with associated value val.
1057
1058 >>> t = _TestClass(123)
1059 >>> print t.get()
1060 123
1061 """
1062
1063 self.val = val
1064
1065 def square(self):
1066 """square() -> square TestClass's associated value
1067
1068 >>> _TestClass(13).square().get()
1069 169
1070 """
1071
1072 self.val = self.val ** 2
1073 return self
1074
1075 def get(self):
1076 """get() -> return TestClass's associated value.
1077
1078 >>> x = _TestClass(-42)
1079 >>> print x.get()
1080 -42
1081 """
1082
1083 return self.val
1084
1085__test__ = {"_TestClass": _TestClass,
1086 "string": r"""
1087 Example of a string object, searched as-is.
1088 >>> x = 1; y = 2
1089 >>> x + y, x * y
1090 (3, 2)
1091 """
1092 }
1093
1094def _test():
1095 import doctest
1096 return doctest.testmod(doctest)
1097
1098if __name__ == "__main__":
1099 _test()