用例公共部分合并(将setup()和tearDown()提出来，放到单独的一个类中)

 

上一节课程内容中，每个测试类都有SetUp()和tearDown()方法，而且两个方法内容都是一样的，用于打印开始与结束提示语句，是否可以合并在一起呢？

用例公共部分合并
#/usr/bin python
#-*- coding:UTF-8 -*-
from calculator import Math
import unittest

class TestStartEnd(unittest.TestCase):
    def setUp(self):
        print("test start")

    def tearDown(self):
        print("test end")

class TestAdd(TestStartEnd):
    def test_add(self):
        sum = Math.add(10,2)
        self.assertEqual(sum,12)
        print("加-计算正确")

class TestSub(TestStartEnd):
    def test_sub(self):
        sub = Math.sub(10,2)
        self.assertEqual(sub,7)
        print("减-计算正确")

if __name__ == "__main__":
    unittest.main()

注意： unittest.main()  代表执行所有的案例

 

测试的python代码：

#/usr/bin python
#-*- coding:UTF-8 -*-
##计算模块##
class Math:
    #用于存放初始化数据
    def __int__(self,a,b):
        self.a = a
        self.b = b

    #待测的方法
    def add(a,b):
        return a+b

    def sub(a,b):
        return a-b