Python自动单元测试框架

http://www.ibm.com/developerworks/cn/linux/l-pyunit/

Widget.py

class Widget:
    def __init__(self, size = (40, 40)):
        self._size = size
    def getSize(self):
        return self._size
    def resize(self, width, height):
        if width < 0  or height < 0:
            raise ValueError, "illegal size"
        self._size = (width, height)
    def dispose(self):
        pass

 

 auto.py

#auto.py
from widget import Widget
import unittest

# Test Class widget
class WidgetTestCase(unittest.TestCase):
    def setUp(self):
        self.widget = Widget()
    def tearDown(self):
        self.widget = None
    def testSize(self):
        self.assertEqual(self.widget.getSize(), (40, 40))
        
# test construture
def suite():
    suite = unittest.TestSuite()
    suite.addTest(WidgetTestCase("testSize"))
    return suite
    
# test
if __name__ == "__main__":
    unittest.main(defaultTest = 'suite')