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'
def runTest(self): widget = Widget() self.assertEqual(widget.getSize(), (40, 40))