Python: Generate Test Case Stubs For All Classes In A Module

I had a module full of classes. I wanted to generate a unittest.TestCase stub for every class, such as this:

# TODO: Finish this
class Test_className(unittest.TestCase):
    def setUp(self):
        pass

So, I wrote the following Python script:

if __name__=="__main__":
    file = "c:/dev/sourceFile.py"
    f = open(file, "r")
    line = f.readline()
    count = 0
    thing = [ ]
 
    while line:
        if line[:5] == "class":
            count += 1
            parenpos = line.find("(")
            classname = line[6:parenpos]
            thing.append('# TODO: Finish this')
            thing.append('class Test_%s(unittest.TestCase):' % classname)
            thing.append('    def setUp(self):')
            thing.append('        pass')
            thing.append('')
 
 
        line = f.readline()
 
    f.close()
 
    for item in thing:
        print item

Just a quick way to save myself some mundane typing.

Posted on July 9, 2009 at 7:39 pm by David Andrzejewski · Permalink
In: Jigs · Tagged with: , ,

One Response

Subscribe to comments via RSS

  1. Written by Jason Felice
    on July 10, 2009 at 12:08 pm
    Permalink

    Now that’s what I’m talking about!

Subscribe to comments via RSS