Change the way unexpected output is reported: rather than stopping at

the first difference, let the test run till completion, then gather
all the output and compare it to the expected output using difflib.

XXX Still to do: produce diff output that only shows the sections that
differ; currently it produces ndiff-style output because that's the
easiest to produce with difflib, but this becomes a liability when the
output is voluminous and there are only a few differences.
This commit is contained in:
Guido van Rossum 2001-09-21 20:31:52 +00:00
parent 867a8d2e26
commit 0fcca4e815
2 changed files with 52 additions and 94 deletions

View File

@ -282,16 +282,10 @@ def runtest(test, generate, verbose, quiet, testdir = None):
if not testdir: testdir = findtestdir()
outputdir = os.path.join(testdir, "output")
outputfile = os.path.join(outputdir, test)
try:
if generate:
cfp = StringIO.StringIO()
elif verbose:
cfp = sys.stdout
else:
cfp = Compare(outputfile, sys.stdout)
except IOError:
if verbose or generate:
cfp = None
print "Warning: can't open", outputfile
else:
cfp = StringIO.StringIO()
try:
save_stdout = sys.stdout
try:
@ -306,10 +300,15 @@ def runtest(test, generate, verbose, quiet, testdir = None):
indirect_test = getattr(the_module, "test_main", None)
if indirect_test is not None:
indirect_test()
if cfp and not (generate or verbose):
cfp.close()
finally:
sys.stdout = save_stdout
if cfp and test_support.output_comparison_denied():
output = cfp.getvalue()
cfp = None
s = test + "\n"
if output.startswith(s):
output = output[len(s):]
sys.stdout.write(output)
except (ImportError, test_support.TestSkipped), msg:
if not quiet:
print "test", test, "skipped --", msg
@ -326,25 +325,44 @@ def runtest(test, generate, verbose, quiet, testdir = None):
traceback.print_exc(file=sys.stdout)
return 0
else:
if not cfp:
return 1
output = cfp.getvalue()
if generate:
output = cfp.getvalue()
if output == test + "\n":
if os.path.exists(outputfile):
# Write it since it already exists (and the contents
# may have changed), but let the user know it isn't
# needed:
fp = open(outputfile, "w")
fp.write(output)
fp.close()
print "output file", outputfile, \
"is no longer needed; consider removing it"
# else:
# We don't need it, so don't create it.
else:
fp = open(outputfile, "w")
fp.write(output)
fp.close()
return 1
else:
# We don't need it, so don't create it.
return 1
fp = open(outputfile, "w")
fp.write(output)
fp.close()
return 1
if os.path.exists(outputfile):
fp = open(outputfile, "r")
expected = fp.read()
fp.close()
else:
expected = test + "\n"
if output == expected:
return 1
print "test", test, "produced unexpected output:"
reportdiff(expected, output)
return 0
def reportdiff(expected, output):
print "*" * 70
import difflib
a = expected.splitlines(1)
b = output.splitlines(1)
diff = difflib.ndiff(a, b)
print ''.join(diff),
print "*" * 70
def findtestdir():
if __name__ == '__main__':
@ -384,62 +402,6 @@ def printlist(x, width=70, indent=4):
if len(line) > indent:
print line
class Compare:
def __init__(self, filename, origstdout):
self.origstdout = origstdout
if os.path.exists(filename):
self.fp = open(filename, 'r')
else:
self.fp = StringIO.StringIO(
os.path.basename(filename) + "\n")
self.stuffthatmatched = []
def write(self, data):
if test_support.suppress_output_comparison():
self.origstdout.write(data)
return
expected = self.fp.read(len(data))
if data == expected:
self.stuffthatmatched.append(expected)
else:
# This Compare instance is spoofing stdout, so we need to write
# to stderr instead.
from sys import stderr as e
print >> e, "The actual stdout doesn't match the expected stdout."
if self.stuffthatmatched:
print >> e, "This much did match (between asterisk lines):"
print >> e, "*" * 70
good = "".join(self.stuffthatmatched)
e.write(good)
if not good.endswith("\n"):
e.write("\n")
print >> e, "*" * 70
print >> e, "Then ..."
else:
print >> e, "The first write to stdout clashed:"
# Note that the prompts are the same length in next two lines.
# This is so what we expected and what we got line up.
print >> e, "We expected (repr):", `expected`
print >> e, "But instead we got:", `data`
raise test_support.TestFailed('Writing: ' + `data`+
', expected: ' + `expected`)
def writelines(self, listoflines):
map(self.write, listoflines)
def flush(self):
pass
def close(self):
leftover = self.fp.read()
if leftover:
raise test_support.TestFailed('Tail of expected stdout unseen: ' +
`leftover`)
self.fp.close()
def isatty(self):
return 0
class _Set:
def __init__(self, seq=[]):
data = self.data = {}

View File

@ -23,23 +23,22 @@ use_resources = None # Flag set to [] by regrtest.py
# _output_comparison controls whether regrtest will try to compare stdout
# with an expected-output file. For straight regrtests, it should.
# The doctest driver should set_output_comparison(0) for the duration, and
# restore the old value when it's done.
# The doctest driver resets this flag by calling deny_output_comparison().
# Note that this control is in addition to verbose mode: output will be
# compared if and only if _output_comparison is true and verbose mode is
# not in effect.
_output_comparison = 1
def set_output_comparison(newvalue):
def deny_output_comparison():
global _output_comparison
oldvalue = _output_comparison
_output_comparison = newvalue
return oldvalue
_output_comparison = 0
# regrtest's interface to _output_comparison.
def suppress_output_comparison():
return not _output_comparison
def output_comparison_denied():
global _output_comparison
denied = not _output_comparison
_output_comparison = 1
return denied
def unload(name):
try:
@ -199,10 +198,7 @@ def run_doctest(module, verbosity=None):
else:
verbosity = None
oldvalue = set_output_comparison(0)
try:
f, t = doctest.testmod(module, verbose=verbosity)
if f:
raise TestFailed("%d of %d doctests failed" % (f, t))
finally:
set_output_comparison(oldvalue)
deny_output_comparison()
f, t = doctest.testmod(module, verbose=verbosity)
if f:
raise TestFailed("%d of %d doctests failed" % (f, t))