python 函数单元测试demo

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文件