__init__.py 673 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
"""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))