myFun.py

'''
單元測試作用：用來對一個函式、一個類或者一個模組進行正確性校驗
測試結果：
1.單元測試通過，說明我們測試的函式功能正常
2.單元測試不通過，說明函式功能有BUG，要麼測試條件輸入有誤


'''

def newSun(x,y):
    return x + y

def newSub(x,y):
    return x - y

myTest.py

#單元測試一般不在原檔案上測試，而是另寫一個測試檔案

#匯入相關的模組
import  unittest

#匯入要測試的函式，我的函式是寫在myFun.py檔案上，你應該匯入自己寫的被測函式
from myFun import newSun
from myFun import newSub

class Test(unittest.TestCase):
    def setUp(self):
        print("開始測試時自動呼叫")
    def tearDown(self):
        print("結束測試時自動呼叫")

    #測試函式newSun
    def test_newSun(self):
        self.assertEqual(newSun(1,2),3,"加法有誤")

    def test_newSun(self):
        self.assertEqual(newSub(2,1),1,"減法有誤")
if __name__ == '__main__':
    unittest.main()
 

測試最好寫兩個檔案，一個是被測的函式模組/本文中的myFun.py

一個是測試檔案，本文中的myTest.py

然後執行myTest.py檔案