diff --git a/src/calibre/db/tests/main.py b/src/calibre/db/tests/main.py index 85b5cf473f..7a41161572 100644 --- a/src/calibre/db/tests/main.py +++ b/src/calibre/db/tests/main.py @@ -61,11 +61,7 @@ class TestResult(unittest.TextTestResult): def find_tests(): return unittest.defaultTestLoader.discover(os.path.dirname(os.path.abspath(__file__)), pattern='*.py') -if __name__ == '__main__': - from calibre.utils.config_base import reset_tweaks_to_default - from calibre.ebooks.metadata.book.base import reset_field_metadata - reset_tweaks_to_default() - reset_field_metadata() +def run_tests(find_tests=find_tests): parser = argparse.ArgumentParser() parser.add_argument('name', nargs='?', default=None, help='The name of the test to run, for e.g. writing.WritingTest.many_many_basic or .many_many_basic for a shortcut') @@ -95,3 +91,10 @@ if __name__ == '__main__': r.resultclass = TestResult r(verbosity=4).run(tests) +if __name__ == '__main__': + from calibre.utils.config_base import reset_tweaks_to_default + from calibre.ebooks.metadata.book.base import reset_field_metadata + reset_tweaks_to_default() + reset_field_metadata() + run_tests() +