def suite(): suites = [ShapeGraphCase, BonusCase] suite = unittest.TestSuite() for s in suites: suite.addTest(unittest.makeSuite(s)) return suite