Home | History | Annotate | Download | only in test 
      1 """
      2 Use this module to get and run all tk tests.
      3 
      4 Tkinter tests should live in a package inside the directory where this file
      5 lives, like test_tkinter.
      6 Extensions also should live in packages following the same rule as above.
      7 """
      8 
      9 import os
     10 import sys
     11 import unittest
     12 import importlib
     13 import test.test_support
     14 
     15 this_dir_path = os.path.abspath(os.path.dirname(__file__))
     16 
     17 def is_package(path):
     18     for name in os.listdir(path):
     19         if name in ('__init__.py', '__init__.pyc', '__init.pyo'):
     20             return True
     21     return False
     22 
     23 def get_tests_modules(basepath=this_dir_path, gui=True, packages=None):
     24     """This will import and yield modules whose names start with test_
     25     and are inside packages found in the path starting at basepath.
     26 
     27     If packages is specified it should contain package names that want
     28     their tests collected.
     29     """
     30     py_ext = '.py'
     31 
     32     for dirpath, dirnames, filenames in os.walk(basepath):
     33         for dirname in list(dirnames):
     34             if dirname[0] == '.':
     35                 dirnames.remove(dirname)
     36 
     37         if is_package(dirpath) and filenames:
     38             pkg_name = dirpath[len(basepath) + len(os.sep):].replace('/', '.')
     39             if packages and pkg_name not in packages:
     40                 continue
     41 
     42             filenames = filter(
     43                     lambda x: x.startswith('test_') and x.endswith(py_ext),
     44                     filenames)
     45 
     46             for name in filenames:
     47                 try:
     48                     yield importlib.import_module(
     49                             ".%s" % name[:-len(py_ext)], pkg_name)
     50                 except test.test_support.ResourceDenied:
     51                     if gui:
     52                         raise
     53 
     54 def get_tests(text=True, gui=True, packages=None):
     55     """Yield all the tests in the modules found by get_tests_modules.
     56 
     57     If nogui is True, only tests that do not require a GUI will be
     58     returned."""
     59     attrs = []
     60     if text:
     61         attrs.append('tests_nogui')
     62     if gui:
     63         attrs.append('tests_gui')
     64     for module in get_tests_modules(gui=gui, packages=packages):
     65         for attr in attrs:
     66             for test in getattr(module, attr, ()):
     67                 yield test
     68 
     69 if __name__ == "__main__":
     70     test.test_support.run_unittest(*get_tests())
     71