This commit is contained in:
Kovid Goyal 2016-06-22 08:06:36 +05:30
parent 55ca6cb33c
commit ea38ac5aaa

View File

@ -100,7 +100,7 @@ def filter_tests_by_module(suite, *names):
return m in names return m in names
return filter_tests(suite, q) return filter_tests(suite, q)
def run_tests(find_tests, verbosity=4, result_class=TestResult): def run_tests(find_tests, verbosity=4):
import argparse import argparse
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('name', nargs='?', default=None, parser.add_argument('name', nargs='?', default=None,
@ -114,10 +114,10 @@ def run_tests(find_tests, verbosity=4, result_class=TestResult):
tests = filter_tests_by_module(tests, args.name) tests = filter_tests_by_module(tests, args.name)
if not tests._tests: if not tests._tests:
raise SystemExit('No test named %s found' % args.name) raise SystemExit('No test named %s found' % args.name)
run_cli(tests, verbosity, result_class) run_cli(tests, verbosity)
def run_cli(suite, verbosity=4, result_class=TestResult): def run_cli(suite, verbosity=4):
r = unittest.TextTestRunner r = unittest.TextTestRunner
r.resultclass = result_class or unittest.TestResult r.resultclass = unittest.TextTestResult if verbosity < 2 else TestResult
init_env() init_env()
r(verbosity=verbosity).run(suite) r(verbosity=verbosity).run(suite)