新增用例管理(之前是对一个方法的单独测试add(),如果加进来一个方法sub(),该如何整理案例)

前面是针对单个add方法来进行单元测试，如果需要多个方法来进行测试，该如何处理？如新增一个Sub方法来进行单元测试验证。

案例脚本：

#/usr/bin python
#-*- coding:UTF-8 -*-
from calculator import Math
import unittest
class TestAdd(unittest.TestCase):
    def setUp(self):
        print("test start")

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

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

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

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

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

if __name__ == "__main__":
    suite = unittest.TestSuite()
    suite.addTest(TestAdd("test_add"))
    suite.addTest(TestSub("test_sub"))

    runner = unittest.TextTestRunner()
    runner.run(suite)

测试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