import inspect import random import re import unittest try: from StringIO import StringIO except ImportError: from io import StringIO from . import import_submodule class PygameTestLoader(unittest.TestLoader): def __init__( self, randomize_tests=False, include_incomplete=False, exclude=("interactive",) ): super(PygameTestLoader, self).__init__() self.randomize_tests = randomize_tests if exclude is None: self.exclude = set() else: self.exclude = set(exclude) if include_incomplete: self.testMethodPrefix = ("test", "todo_") def getTestCaseNames(self, testCaseClass): res = [] for name in super(PygameTestLoader, self).getTestCaseNames(testCaseClass): tags = get_tags(testCaseClass, getattr(testCaseClass, name)) if self.exclude.isdisjoint(tags): res.append(name) if self.randomize_tests: random.shuffle(res) return res # Exclude by tags: TAGS_RE = re.compile(r"\|[tT]ags:(-?[ a-zA-Z,0-9_\n]+)\|", re.M) class TestTags: def __init__(self): self.memoized = {} self.parent_modules = {} def get_parent_module(self, class_): if class_ not in self.parent_modules: self.parent_modules[class_] = import_submodule(class_.__module__) return self.parent_modules[class_] def __call__(self, parent_class, meth): key = (parent_class, meth.__name__) if key not in self.memoized: parent_module = self.get_parent_module(parent_class) module_tags = getattr(parent_module, "__tags__", []) class_tags = getattr(parent_class, "__tags__", []) tags = TAGS_RE.search(inspect.getdoc(meth) or "") if tags: test_tags = [t.strip() for t in tags.group(1).split(",")] else: test_tags = [] combined = set() for tags in (module_tags, class_tags, test_tags): if not tags: continue add = set([t for t in tags if not t.startswith("-")]) remove = set([t[1:] for t in tags if t not in add]) if add: combined.update(add) if remove: combined.difference_update(remove) self.memoized[key] = combined return self.memoized[key] get_tags = TestTags()