#8560: add progress indicator to regrtest.

This commit is contained in:
Georg Brandl 2010-08-02 18:59:52 +00:00
parent 6fcac0d6f6
commit dee7b8503d
1 changed files with 12 additions and 5 deletions

View File

@ -517,6 +517,9 @@ def main(tests=None, testdir=None, verbose=0, quiet=False,
else: else:
tests = iter(selected) tests = iter(selected)
tests = list(tests)
test_count = len(tests)
test_count_width = len(str(test_count))
if use_mp: if use_mp:
try: try:
from threading import Thread from threading import Thread
@ -559,8 +562,6 @@ def main(tests=None, testdir=None, verbose=0, quiet=False,
output.put((None, None, None, None)) output.put((None, None, None, None))
return return
result = json.loads(result) result = json.loads(result)
if not quiet:
stdout = test+'\n'+stdout
output.put((test, stdout.rstrip(), stderr.rstrip(), result)) output.put((test, stdout.rstrip(), stderr.rstrip(), result))
except BaseException: except BaseException:
output.put((None, None, None, None)) output.put((None, None, None, None))
@ -569,12 +570,16 @@ def main(tests=None, testdir=None, verbose=0, quiet=False,
for worker in workers: for worker in workers:
worker.start() worker.start()
finished = 0 finished = 0
test_index = 1
try: try:
while finished < use_mp: while finished < use_mp:
test, stdout, stderr, result = output.get() test, stdout, stderr, result = output.get()
if test is None: if test is None:
finished += 1 finished += 1
continue continue
if not quiet:
print("[{1:{0}}/{2:{0}}] {3}".format(
test_count_width, test_index, test_count, test))
if stdout: if stdout:
print(stdout) print(stdout)
if stderr: if stderr:
@ -583,15 +588,17 @@ def main(tests=None, testdir=None, verbose=0, quiet=False,
assert result[1] == 'KeyboardInterrupt' assert result[1] == 'KeyboardInterrupt'
raise KeyboardInterrupt # What else? raise KeyboardInterrupt # What else?
accumulate_result(test, result) accumulate_result(test, result)
test_index += 1
except KeyboardInterrupt: except KeyboardInterrupt:
interrupted = True interrupted = True
pending.close() pending.close()
for worker in workers: for worker in workers:
worker.join() worker.join()
else: else:
for test in tests: for test_index, test in enumerate(tests, 1):
if not quiet: if not quiet:
print(test) print("[{1:{0}}/{2:{0}}] {3}".format(
test_count_width, test_index, test_count, test))
sys.stdout.flush() sys.stdout.flush()
if trace: if trace:
# If we're tracing code coverage, then we don't exit with status # If we're tracing code coverage, then we don't exit with status