how to unittest

转载自https://docs.python.org/3/library/unittest.html#unittest.TestCase.tearDown

备注：根据转载的内容，稍作修改，使得代码run起来

import unittest

class Widget:
    name = ''
    length = 100
    height = 200

    def __init__(self, n):
        self.name = n

    def size(self):
        print("size")
        return(self.length, self.height)

    def resize(self, x, y):
        self.length = x
        self.height = y
        print("resize")
        return (self.height, self.height)

    def dispose(self):
        name = ''
        length = 100
        height = 200
        print("dispose and exit")
        print()

    def speak(self):
        print("%s says begin..." % (self.name))


class WidgetTestCase(unittest.TestCase):
    def setUp(self):
        self.widget = Widget('Setup The widget')
        self.widget.speak()

    def tearDown(self):
        self.widget.dispose()

    def test_default_widget_size(self):
        print("test_default_widget_size")
        self.assertEqual(self.widget.size(), (50, 50),
                         'incorrect default size')

    def test_widget_resize(self):
        print("test_widget_resize")
        self.widget.resize(100, 150)
        self.assertEqual(self.widget.size(), (100, 150),
                         'wrong size after resize')

def suite():
    suite = unittest.TestSuite()
    suite.addTest(WidgetTestCase('test_default_widget_size'))
    suite.addTest(WidgetTestCase('test_widget_resize'))
    return suite

# Press the green button in the gutter to run the script.
if __name__ == '__main__':
#   unittest.main()
    runner = unittest.TextTestRunner()
    runner.run(suite())

测试结果