import unittest class FirstTestCase(unittest.TestCase): def runTest(self): """The first test.""" assert 1 == 1, 'One is one with itself' class SillyTestCase(unittest.TestCase): def setUp(self): self.value = 1 print("Setup") def tearDown(self): print("Teardown") def testPositive(self): assert self.value > 0, 'My value must be positive.' def testOdd(self): assert self.value %2 == 1, 'My value must be odd.' def suite(): """Build a test suite.""" sweet = unittest.TestSuite() sweet.addTest(FirstTestCase()) sweet.addTest(SillyTestCase('testPositive')) sweet.addTest(SillyTestCase('testOdd')) return sweet