| """Make tests/ into a package. This allows us to "import tests" and | |
| have tests.all_tests be a TestSuite representing all test cases | |
| from all test_*.py files in tests/.""" | |
| # Author: Collin Winter | |
| import os | |
| import os.path | |
| import unittest | |
| import types | |
| from . import support | |
| all_tests = unittest.TestSuite() | |
| tests_dir = os.path.join(os.path.dirname(__file__), '..', 'tests') | |
| tests = [t[0:-3] for t in os.listdir(tests_dir) | |
| if t.startswith('test_') and t.endswith('.py')] | |
| loader = unittest.TestLoader() | |
| for t in tests: | |
| __import__("",globals(),locals(),[t],level=1) | |
| mod = globals()[t] | |
| all_tests.addTests(loader.loadTestsFromModule(mod)) |