run_tests.py

#!/usr/bin/env python3

import os
import re
import sys
import unittest

TEST_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'tests')

# Return a list of unittest.TestCase
def _collect_tests(suite, testCases = []):
    if (isinstance(suite, unittest.TestCase)):
        testCases.append(suite)
        return testCases

    if (not isinstance(suite, unittest.suite.TestSuite)):
        raise ValueError("Unknown test type: %s" % (str(type(suite))))

    for testObject in suite:
        _collect_tests(testObject, testCases)

    return testCases

def main(pattern = None):
    runner = unittest.TextTestRunner(verbosity = 3)
    discoveredSuite = unittest.TestLoader().discover(TEST_DIR)
    testCases = _collect_tests(discoveredSuite)

    tests = unittest.suite.TestSuite()

    fail = False
    for testCase in testCases:
        if (isinstance(testCase, unittest.loader._FailedTest)):
            print('Failed to load test: %s' % (testCase.id()))
            print(testCase._exception)
            fail = True
            continue

        if (pattern is None or re.search(pattern, testCase.id())):
            tests.addTest(testCase)
        else:
            print("Skipping %s because of match pattern." % (testCase.id()))

    if (not runner.run(tests).wasSuccessful() or fail):
        sys.exit(1)

def _load_args(args):
    executable = args.pop(0)
    if (len(args) >  1 or ({'h', 'help'} & {arg.lower().strip().replace('-', '') for arg in args})):
        print("USAGE: python3 %s [test pattern]" % (executable), file = sys.stderr)
        print('The test pattern will be used directly in re.search() to see if a test will be run.', file = sys.stderr)
        sys.exit(1)

    pattern = None
    if (len(args) > 0):
        pattern = args.pop(0)

    return (pattern, )

if __name__ == '__main__':
    main(*_load_args(sys.argv))