前面是针对单个add方法来进行单元测试,如果需要多个方法来进行测试,该如何处理?如新增一个Sub方法来进行单元测试验证。
class Math:
def __init__(self,a,b):
self.a=int(a)
self.b=int(b)
def add(self):
return self.a+self.b
def sub(self):
return self.a-self.b
同时对add和Sub方法进行单元测试验证
from calculator import *
import unittest
class Test_add(unittest.TestCase):
def setUp(self):
print("Test is start")
def test_add(self):
j=Math(5,5)
self.assertEqual(j.add(),10)
def test_add1(self):
j=Math(10,20)
self.assertEqual(j.add(),30)
def tearDown(self):
print("test is end!")
class Test_sub(unittest.TestCase):
def setUp(self):
print("Test is start")
def test_sub(self):
i=Math(8,8)
self.assertEqual(i.sub(),0)
def test_sub1(self):
i=Math(5,3)
self.assertEqual(i.sub(),2)
def tearDown(self):
print("test is end!")
if __name__ == '__main__':
suite=unittest.TestSuite()
suite.addTest(Test_add("test_add"))
suite.addTest(Test_add("test_add1"))
suite.addTest(Test_sub("test_sub"))
suite.addTest(Test_sub("test_sub1"))
runner=unittest.TextTestRunner()
runner.run(suite)