Eric S. Raymond | 630e69c | 2001-02-09 08:33:43 +0000 | [diff] [blame] | 1 | # Module doctest version 0.9.7 |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 2 | # 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 | |
| 9 | NORMAL USAGE |
| 10 | |
| 11 | In normal use, end each module M with: |
| 12 | |
| 13 | def _test(): |
| 14 | import doctest, M # replace M with your module's name |
| 15 | return doctest.testmod(M) # ditto |
| 16 | |
| 17 | if __name__ == "__main__": |
| 18 | _test() |
| 19 | |
| 20 | Then running the module as a script will cause the examples in the |
| 21 | docstrings to get executed and verified: |
| 22 | |
| 23 | python M.py |
| 24 | |
| 25 | This won't display anything unless an example fails, in which case the |
| 26 | failing 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 |
| 28 | line of output is "Test failed.". |
| 29 | |
| 30 | Run it with the -v switch instead: |
| 31 | |
| 32 | python M.py -v |
| 33 | |
| 34 | and a detailed report of all examples tried is printed to stdout, along |
| 35 | with assorted summaries at the end. |
| 36 | |
| 37 | You can force verbose mode by passing "verbose=1" to testmod, or prohibit |
| 38 | it by passing "verbose=0". In either of those cases, sys.argv is not |
| 39 | examined by testmod. |
| 40 | |
| 41 | In any case, testmod returns a 2-tuple of ints (f, t), where f is the |
| 42 | number of docstring examples that failed and t is the total number of |
| 43 | docstring examples attempted. |
| 44 | |
| 45 | |
| 46 | WHICH 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 | |
| 64 | Any classes found are recursively searched similarly, to test docstrings in |
| 65 | their contained methods and nested classes. Private names reached from M's |
| 66 | globals are skipped, but all names reached from M.__test__ are searched. |
| 67 | |
| 68 | By default, a name is considered to be private if it begins with an |
| 69 | underscore (like "_my_func") but doesn't both begin and end with (at least) |
| 70 | two underscores (like "__init__"). You can change the default by passing |
| 71 | your own "isprivate" function to testmod. |
| 72 | |
| 73 | If you want to test docstrings in objects with private names too, stuff |
| 74 | them into an M.__test__ dict, or see ADVANCED USAGE below (e.g., pass your |
| 75 | own isprivate function to Tester's constructor, or call the rundoc method |
| 76 | of a Tester instance). |
| 77 | |
| 78 | Warning: imports can cause trouble; e.g., if you do |
| 79 | |
| 80 | from XYZ import XYZclass |
| 81 | |
| 82 | then XYZclass is a name in M.__dict__ too, and doctest has no way to know |
| 83 | that XYZclass wasn't *defined* in M. So it may try to execute the examples |
| 84 | in XYZclass's docstring, and those in turn may require a different set of |
| 85 | globals to work correctly. I prefer to do "import *"- friendly imports, |
| 86 | a la |
| 87 | |
| 88 | import XYY |
| 89 | _XYZclass = XYZ.XYZclass |
| 90 | del XYZ |
| 91 | |
| 92 | or (Python 2.0) |
| 93 | |
| 94 | from XYZ import XYZclass as _XYZclass |
| 95 | |
| 96 | and then the leading underscore stops testmod from going nuts. You may |
| 97 | prefer the method in the next section. |
| 98 | |
| 99 | |
| 100 | WHAT'S THE EXECUTION CONTEXT? |
| 101 | |
| 102 | By default, each time testmod finds a docstring to test, it uses a *copy* |
| 103 | of M's globals (so that running tests on a module doesn't change the |
| 104 | module's real globals, and so that one test in M can't leave behind crumbs |
| 105 | that accidentally allow another test to work). This means examples can |
| 106 | freely use any names defined at top-level in M. It also means that sloppy |
| 107 | imports (see above) can cause examples in external docstrings to use |
| 108 | globals inappropriate for them. |
| 109 | |
| 110 | You 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 |
| 112 | M.__dict__ merged with the globals from other imported modules. |
| 113 | |
| 114 | |
| 115 | WHAT IF I WANT TO TEST A WHOLE PACKAGE? |
| 116 | |
| 117 | Piece o' cake, provided the modules do their testing from docstrings. |
| 118 | Here's the test.py I use for the world's most elaborate Rational/ |
| 119 | floating-base-conversion pkg (which I'll distribute some day): |
| 120 | |
| 121 | from Rational import Cvt |
| 122 | from Rational import Format |
| 123 | from Rational import machprec |
| 124 | from Rational import Rat |
| 125 | from Rational import Round |
| 126 | from Rational import utils |
| 127 | |
| 128 | modules = (Cvt, |
| 129 | Format, |
| 130 | machprec, |
| 131 | Rat, |
| 132 | Round, |
| 133 | utils) |
| 134 | |
| 135 | def _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 | |
| 143 | if __name__ == "__main__": |
| 144 | _test() |
| 145 | |
| 146 | IOW, it just runs testmod on all the pkg modules. testmod remembers the |
| 147 | names and outcomes (# of failures, # of tries) for each item it's seen, and |
| 148 | passing "report=0" prevents it from printing a summary in verbose mode. |
| 149 | Instead, the summary is delayed until all modules have been tested, and |
| 150 | then "doctest.master.summarize()" forces the summary at the end. |
| 151 | |
| 152 | So this is very nice in practice: each module can be tested individually |
| 153 | with almost no work beyond writing up docstring examples, and collections |
| 154 | of modules can be tested too as a unit with no more work than the above. |
| 155 | |
| 156 | |
| 157 | WHAT ABOUT EXCEPTIONS? |
| 158 | |
| 159 | No problem, as long as the only output generated by the example is the |
| 160 | traceback itself. For example: |
| 161 | |
Tim Peters | 60e23f4 | 2001-02-14 00:43:21 +0000 | [diff] [blame] | 162 | >>> [1, 2, 3].remove(42) |
Tim Peters | ea4f931 | 2001-02-13 20:54:42 +0000 | [diff] [blame] | 163 | Traceback (most recent call last): |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 164 | File "<stdin>", line 1, in ? |
Tim Peters | 60e23f4 | 2001-02-14 00:43:21 +0000 | [diff] [blame] | 165 | ValueError: list.remove(x): x not in list |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 166 | >>> |
| 167 | |
| 168 | Note that only the exception type and value are compared (specifically, |
| 169 | only the last line in the traceback). |
| 170 | |
| 171 | |
| 172 | ADVANCED USAGE |
| 173 | |
| 174 | doctest.testmod() captures the testing policy I find most useful most |
| 175 | often. You may want other policies. |
| 176 | |
| 177 | testmod() actually creates a local instance of class doctest.Tester, runs |
| 178 | appropriate methods of that class, and merges the results into global |
| 179 | Tester instance doctest.master. |
| 180 | |
| 181 | You can create your own instances of doctest.Tester, and so build your own |
| 182 | policies, or even run methods of doctest.master directly. See |
| 183 | doctest.Tester.__doc__ for details. |
| 184 | |
| 185 | |
| 186 | SO WHAT DOES A DOCSTRING EXAMPLE LOOK LIKE ALREADY!? |
| 187 | |
| 188 | Oh ya. It's easy! In most cases a copy-and-paste of an interactive |
| 189 | console session works fine -- just make sure the leading whitespace is |
| 190 | rigidly consistent (you can mix tabs and spaces if you're too lazy to do it |
| 191 | right, but doctest is not in the business of guessing what you think a tab |
| 192 | means). |
| 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 | |
| 210 | Any expected output must immediately follow the final ">>>" or "..." line |
| 211 | containing the code, and the expected output (if any) extends to the next |
| 212 | ">>>" or all-whitespace line. That's it. |
| 213 | |
| 214 | Bummers: |
| 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' |
| 231 | yes |
| 232 | |
| 233 | The starting column doesn't matter: |
| 234 | |
| 235 | >>> assert "Easy!" |
| 236 | >>> import math |
| 237 | >>> math.floor(1.9) |
| 238 | 1.0 |
| 239 | |
| 240 | and as many leading whitespace characters are stripped from the expected |
| 241 | output as appeared in the initial ">>>" line that triggered it. |
| 242 | |
| 243 | If you execute this very file, the examples above will be found and |
| 244 | executed, leading to this output in verbose mode: |
| 245 | |
| 246 | Running doctest.__doc__ |
Tim Peters | 60e23f4 | 2001-02-14 00:43:21 +0000 | [diff] [blame] | 247 | Trying: [1, 2, 3].remove(42) |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 248 | Expecting: |
Tim Peters | ea4f931 | 2001-02-13 20:54:42 +0000 | [diff] [blame] | 249 | Traceback (most recent call last): |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 250 | File "<stdin>", line 1, in ? |
Tim Peters | 60e23f4 | 2001-02-14 00:43:21 +0000 | [diff] [blame] | 251 | ValueError: list.remove(x): x not in list |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 252 | ok |
| 253 | Trying: x = 12 |
| 254 | Expecting: nothing |
| 255 | ok |
| 256 | Trying: x |
| 257 | Expecting: 12 |
| 258 | ok |
| 259 | Trying: |
| 260 | if x == 13: |
| 261 | print "yes" |
| 262 | else: |
| 263 | print "no" |
| 264 | print "NO" |
| 265 | print "NO!!!" |
| 266 | Expecting: |
| 267 | no |
| 268 | NO |
| 269 | NO!!! |
| 270 | ok |
| 271 | ... and a bunch more like that, with this summary at the end: |
| 272 | |
| 273 | 5 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 |
| 279 | 12 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 |
| 292 | 53 tests in 17 items. |
| 293 | 53 passed and 0 failed. |
| 294 | Test 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. Raymond | 630e69c | 2001-02-09 08:33:43 +0000 | [diff] [blame] | 348 | # 0,9,7 9-Feb-2001 |
| 349 | # string method conversion |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 350 | |
Tim Peters | ecb6fb9 | 2001-02-10 01:24:50 +0000 | [diff] [blame] | 351 | __version__ = 0, 9, 7 |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 352 | |
| 353 | import types |
| 354 | _FunctionType = types.FunctionType |
| 355 | _ClassType = types.ClassType |
| 356 | _ModuleType = types.ModuleType |
| 357 | _StringType = types.StringType |
| 358 | del types |
| 359 | |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 360 | import re |
| 361 | PS1 = ">>>" |
| 362 | PS2 = "..." |
| 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 |
| 367 | del re |
| 368 | |
Skip Montanaro | eccd02a | 2001-01-20 23:34:12 +0000 | [diff] [blame] | 369 | __all__ = [] |
| 370 | |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 371 | # 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 | |
| 378 | def _extract_examples(s): |
| 379 | isPS1, isPS2 = _isPS1, _isPS2 |
| 380 | isEmpty, isComment = _isEmpty, _isComment |
| 381 | examples = [] |
Eric S. Raymond | 630e69c | 2001-02-09 08:33:43 +0000 | [diff] [blame] | 382 | lines = s.split("\n") |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 383 | 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. Raymond | 630e69c | 2001-02-09 08:33:43 +0000 | [diff] [blame] | 420 | source = "\n".join(source) + "\n" |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 421 | # 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. Raymond | 630e69c | 2001-02-09 08:33:43 +0000 | [diff] [blame] | 435 | expect = "\n".join(expect) + "\n" |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 436 | examples.append( (source, expect, lineno) ) |
| 437 | return examples |
| 438 | |
| 439 | # Capture stdout when running examples. |
| 440 | |
| 441 | class _SpoofOut: |
| 442 | def __init__(self): |
| 443 | self.clear() |
| 444 | def write(self, s): |
| 445 | self.buf.append(s) |
| 446 | def get(self): |
Eric S. Raymond | 630e69c | 2001-02-09 08:33:43 +0000 | [diff] [blame] | 447 | return "".join(self.buf) |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 448 | 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 | |
| 457 | def _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. Raymond | 630e69c | 2001-02-09 08:33:43 +0000 | [diff] [blame] | 462 | msg.find("\n") < len(msg) - 1 |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 463 | 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 | |
| 475 | def _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 Peters | ea4f931 | 2001-02-13 20:54:42 +0000 | [diff] [blame] | 492 | if want.find("Traceback (innermost last):\n") == 0 or \ |
| 493 | want.find("Traceback (most recent call last):\n") == 0: |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 494 | # Only compare exception type and value - the rest of |
| 495 | # the traceback isn't necessary. |
Eric S. Raymond | 630e69c | 2001-02-09 08:33:43 +0000 | [diff] [blame] | 496 | want = want.split('\n')[-2] + '\n' |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 497 | 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 | |
| 528 | def _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 | |
| 539 | def 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 | |
| 566 | def 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 | |
| 593 | class Tester: |
| 594 | """Class Tester -- runs docstring examples and accumulates stats. |
| 595 | |
| 596 | In normal use, function doctest.testmod() hides all this from you, |
| 597 | so use that if you can. Create your own instances of Tester to do |
| 598 | fancier things. |
| 599 | |
| 600 | Methods: |
| 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 | ***************************************************************** |
| 631 | Failure in example: print x |
| 632 | from line #2 of XYZ |
| 633 | Expected: 42 |
| 634 | Got: 84 |
| 635 | (1, 2) |
| 636 | >>> t.runstring(">>> x = x * 2\\n>>> print x\\n84\\n", 'example2') |
| 637 | (0, 2) |
| 638 | >>> t.summarize() |
| 639 | 1 items had failures: |
| 640 | 1 of 2 in XYZ |
| 641 | ***Test Failed*** 1 failures. |
| 642 | (1, 4) |
| 643 | >>> t.summarize(verbose=1) |
| 644 | 1 items passed all tests: |
| 645 | 2 tests in example2 |
| 646 | 1 items had failures: |
| 647 | 1 of 2 in XYZ |
| 648 | 4 tests in 2 items. |
| 649 | 3 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 | |
| 659 | See doctest.__doc__ for an overview. |
| 660 | |
| 661 | Optional keyword arg "mod" is a module, whose globals are used for |
| 662 | executing examples. If not specified, globs must be specified. |
| 663 | |
| 664 | Optional keyword arg "globs" gives a dict to be used as the globals |
| 665 | when executing examples; if not specified, use the globals from |
| 666 | module mod. |
| 667 | |
| 668 | In either case, a copy of the dict is used for each docstring |
| 669 | examined. |
| 670 | |
| 671 | Optional keyword arg "verbose" prints lots of stuff if true, only |
| 672 | failures if false; by default, it's true iff "-v" is in sys.argv. |
| 673 | |
| 674 | Optional keyword arg "isprivate" specifies a function used to determine |
| 675 | whether a name is private. The default function is doctest.is_private; |
| 676 | see 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. Raymond | 630e69c | 2001-02-09 08:33:43 +0000 | [diff] [blame] | 959 | i = name.rindex(".") |
Tim Peters | 8a7d2d5 | 2001-01-16 07:10:57 +0000 | [diff] [blame] | 960 | 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 | |
| 967 | master = None |
| 968 | |
| 969 | def 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 | |
| 1041 | class _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 | |
| 1094 | def _test(): |
| 1095 | import doctest |
| 1096 | return doctest.testmod(doctest) |
| 1097 | |
| 1098 | if __name__ == "__main__": |
| 1099 | _test() |