#count_01.py
class Count:
def add(self,a,b):
return a + b
#testcase_01.py
import unittest
from count_01 import Count
#1、必须创建测试类,且必须继承unittest.TestCase
class TestAdd(unittest.TestCase):
def setUp(self):#setUp()方法用于测试用例执行前的初始化工作
print("test before")
#2、至少创建一个测试方法,且必须以“test”开头
def test_add1(self): #测试方法一般不要有传参
c = Count()
result = c.add(3,5)
self.assertEqual(result,8) #对add()的返回值进行断言,判断两者是否相等,assertEqual()方法由TestCase类继承而来
def test_add2(self):
c = Count()
result = c.add(‘hello‘,‘world‘)
self.assertEqual(result,‘helloworld‘)
def tearDown(self):
print("test end") #tearDown()方法与 setUp() 方法相呼应,用于测试用例执行之后的善后工作
if __name__ == ‘__main__‘:
#unittest.main() #默认执行当前文件下所有符合条件的testcase
suit = unittest.TestSuite() #3、创建测试套件类,采用构造测试集的方法来加载与运行测试用 例,实现了有选择地执行测试用例
suit.addTest(TestAdd("test_add1")) #格式:suit.addTest(类名("方法名"))
suit.addTest(TestAdd("test_add2"))
runner = unittest.TextTestRunner() #4、运行测试用例
runner.run(suit)
"""
调用 unittest 框架的 TestSuite()类来创建测试套件,
通过它所提供的 addTest()方法来添加测试用例,
接着调用 unittest 框架的 TextTestRunner()类,通过它下面的 run()方法来运行 suite 所组装的测试用 例
"""