unittest單元測試(簡單算法題)

• 建立測試用例case.py文件

 

#建立類
class test ():
    def __init__(self,a,c):
        self.a =int(a)
        self.c =int (c)
    #加
    def add (self):
        return self.a +self.c

    #減
    def sub (self):
        return self.a -self.c
    #乘
    def mul (self):
        return self.a *self.c
    #除
    def div (self):
        return self.a / self.c

 

• 建立unit.py建立兩個py文件這樣方便其餘文件引入，方便查看，用例必須以test開頭

#導入模塊
import unittest
#引入case.py文件中的測試test類
from case import test

class motest(unittest.TestCase):
#定義test中的a，c
#setUp測試用例開始，執行前的初始化工做（如：生成數據庫測試數據，打開瀏覽器）
    def setUp(self):
        self.test = test(8,4)
#用於測試用例執行以後（如：關閉數據庫，關閉瀏覽器）
    def tearDown(self):
#佔個位置
        pass
#測試用例必須以test開頭
    def test_add (self):
        result = self.test.add()
        self.assertEqual(result,12)

    def test_sub (self):
        result =self.test.sub()
        self.assertEqual(result,4)

    def test_mul (self):
        result =self.test.mul()
        self.assertEqual(result,32)

    def test_div (self):
        result =self.test.div()
        self.assertEqual(result,2)

#main()方法默認查找當前文件中繼承unittest.TestCase的測試類
#測試類下匹配「test」開頭的方法並執行
if __name__ =="__main__":
    unittest.main()
#調用unittest.TestSuite（）類中的addTest（）方法向測試嵌套用例
#測試集，一個對象集合因此單元測試
suite =unittest.TestSuite()
suite.addTest(motest("test_add"))
suite.addTest(motest("test_sub"))
suite.addTest(motest("test_mul"))
suite.addTest(motest("test_div"))
#執行用例出測試結果
runner =unittest.TextTestRunner()
runner.run(suite)