class WidgetTestCase(unittest.TestCase): def setUp(self): self.widget = Widget("The widget") def tearDown(self): self.widget.dispose() self.widget = None def testDefaultSize(self): assert self.widget.getSize() == (50,50), "incorrect default size" def testResize(self): self.widget.resize(100,150) assert self.widget.getSize() == (100,150), 'wrong size after resize'
class WidgetTestCase(unittest.TestCase): # 对象的初始化工作可以在setUp()方法中完成 def setUp(self): self.widget = Widget() # 对象的资源的释放则可以在tearDown()方法中完成 def tearDown(self): self.widget = None # 对应widget类中的get_size函数测试 def test_size(self): self.assertEqual(self.widget.get_size(), (40, 40)) # 对应widget类中resize函数的测试 def test_Resize(self): self.widget.resize(100, 100) self.assertEqual(self.widget.get_size(), (10, 100), msg='不相等')