from __future__ import absolute_import, division, print_function import unittest import doctest import glob tests = ("correlation_test", "correlation_matrix_test", "marker_regression_test", "show_trait_js_test") def main(): tests = glob.glob("*_test.py") suite = unittest.TestSuite() for testname in tests: test = testname.rsplit(".", 1)[0] print("Test is:", test) mod = __import__(test) suite.addTest(doctest.DocTestSuite(mod)) runner = unittest.TextTestRunner() runner.run(suite) if __name__ == '__main__': main()