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