"""
Use this module to get and run all tk tests.

tkinter tests should live in a package inside the directory where this file
lives, like test_tkinter.
Extensions also should live in packages following the same rule as above.
"""

import os
import importlib
import test.support

this_dir_path = os.path.abspath(os.path.dirname(__file__))

def is_package(path):
 for name in os.listdir(path):
 if name in ('__init__.py', '__init__.pyc'):
 return True
 return False

def get_tests_modules(basepath=this_dir_path, gui=True, packages=None):
 """This will import and yield modules whose names start with test_
 and are inside packages found in the path starting at basepath.

 If packages is specified it should contain package names that
 want their tests collected.
 """
 py_ext = '.py'

 for dirpath, dirnames, filenames in os.walk(basepath):
 for dirname in list(dirnames):
 if dirname[0] == '.':
 dirnames.remove(dirname)

 if is_package(dirpath) and filenames:
 pkg_name = dirpath[len(basepath) + len(os.sep):].replace('/', '.')
 if packages and pkg_name not in packages:
 continue

 filenames = filter(
 lambda x: x.startswith('test_') and x.endswith(py_ext),
 filenames)

 for name in filenames:
 try:
 yield importlib.import_module(
 ".%s.%s" % (pkg_name, name[:-len(py_ext)]),
 "tkinter.test")
 except test.support.ResourceDenied:
 if gui:
 raise

def get_tests(text=True, gui=True, packages=None):
 """Yield all the tests in the modules found by get_tests_modules.

 If nogui is True, only tests that do not require a GUI will be
 returned."""
 attrs = []
 if text:
 attrs.append('tests_nogui')
 if gui:
 attrs.append('tests_gui')
 for module in get_tests_modules(gui=gui, packages=packages):
 for attr in attrs:
 for test in getattr(module, attr, ()):
 yield test

if __name__ == "__main__":
 test.support.run_unittest(*get_tests())