学习笔记之unittest单元测试框架2，内附Python的装饰器方法

当测试代码中含有两个测试类的时候，我们需要重新编写测试代码
被测代码，加了一个需要被测试的减法的功能：

#计算器类
from selenium import webdriver
class Count:
    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

测试代码为：

#两个类的测试执行方法
import unittest
from count import Count
class TestCountAdd(unittest.TestCase):
    def setUp(self):
        print("star add test")
    def test_add(self):
        t = Count(2,4)
        add = t.add()
        self.assertEqual(add,6)
        print("add test case1")

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

class TestCountSub(unittest.TestCase):
    def setUp(self):
        print("star sub test")
    def test_sub(self):
        t = Count(10,2)
        sub = t.sub()
        self