from calculator import Count
import unittest


class TestCount(unittest.TestCase):
    #初始化工作
    def setUp(self):
        print("test start")

    def test_add(self):
        j=Count(2,3)
        self.assertEqual(j.add(),5)

    def test_add2(self):
        j=Count(41,76)
        self.assertEqual(j.add(),117)
    #收尾工作
    def tearDown(self):
        print("test end")

#__main__代表直接使用該test2.py
if __name__=='__main__':
    suite=unittest.TestSuite()
    suite.addTest(TestCount("test_add"))
    suite.addTest(TestCount("test_add2"))
    runner=unittest.TextTestRunner()
    runner.run(suite)
 

組織單元測試用例