| Johnny Chen | 7325883 | 2010-08-05 23:42:46 +0000 | [diff] [blame] | 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) | ||||