1import os
2import sys
3from unittest2.loader import defaultTestLoader
4
5
6def collector():
7    # import __main__ triggers code re-execution
8    __main__ = sys.modules['__main__']
9    setupDir = os.path.abspath(os.path.dirname(__main__.__file__))
10    return defaultTestLoader.discover(setupDir)
11