Python Test Collector

I accidentally "discovered" the TestCollector idea independently in Python. My tests all end in "Test.py", and I set them up so I can run each of them independently from the command line, or all at once with this script. I have this in "runAllTests.py" and invoke it so that the current directory is the top of the program. (Unlike most Python programmers I actually have a lot of separate directories and modules.) It recursively searches the current directory for things to test and runs them all at once.

No guarantees, etc.

---

 #!/bin/python

 import sys
 import os
 import re
 import unittest
 import imp

 filesList = []

 def collector(arg, dirname, names):
     filesList.extend([dirname + os.sep + n for n in names])

 def returnTests():
     p = os.path.abspath(os.path.dirname(sys.argv[0]))
     os.path.walk(p, collector, None)
     fList = [x for x in filesList if x[-7:] == "Test.py" and '#' not
              in x] # '#' is for temp files in emacs, extend as needed
     fileList = [file(x, "r") for x in fList]
     modules = []
     for i in range(len(fList)):
         modules.append(imp.load_source(os.path.split(fList[i])[1][:-3],
                                        fList[i],
                                        fileList[i]))

     l = unittest.defaultTestLoader.loadTestsFromModule
     return unittest.TestSuite(map(l, modules))

 if __name__ == "__main__":
     unittest.main(defaultTest="returnTests")

---

There's about a billion ways to extend this, make it more robust, etc., but it's a start. -- JeremyBowers