diff options
Diffstat (limited to 'test/testbase.py')
| -rw-r--r-- | test/testbase.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/test/testbase.py b/test/testbase.py index 47ddaec8b..34c05b8e5 100644 --- a/test/testbase.py +++ b/test/testbase.py @@ -407,12 +407,13 @@ def cover(callable_): coverage_client.save() coverage_client.report(list(covered_files()), show_missing=False, ignore_errors=False) -def main(): +def main(suite=None): - if len(sys.argv[1:]): - suite =unittest.TestLoader().loadTestsFromNames(sys.argv[1:], __import__('__main__')) - else: - suite = unittest.TestLoader().loadTestsFromModule(__import__('__main__')) + if not suite: + if len(sys.argv[1:]): + suite =unittest.TestLoader().loadTestsFromNames(sys.argv[1:], __import__('__main__')) + else: + suite = unittest.TestLoader().loadTestsFromModule(__import__('__main__')) result = runTests(suite) sys.exit(not result.wasSuccessful()) |