python unittest单元测试

 # -*- coding: utf-8 -*-
class Count:
    def __init__(self,a,b):
        self.a=int(a)
        self.b=int(b)
    def add(self):
        return  self.a+self.b
#普通测试方法
class TestCount:
    def test_add(self):
        try:
            j=Count(3,3)
            add=j.add()
            assert (add==5),'Integer addition result error!'  # assert未通过
        except AssertionError as msg:
            print(msg)
        else:
            print('Test pass!')
import unittest

class TestCountU(unittest.TestCase):
    def setUp(self):
        print('test start')
    def test_add(self):
        j=Count(3,2)
        self.assertEqual(j.add(),25)   # 判断测试是否通过
    def test_add2(self):
        j = Count(5, 10)
        self.assertEqual(j.add(), 15)  # 判断测试是否通过
    def tearDown(self):
        print('test end')


if __name__=='__main__':
    suite=unittest.TestSuite()
    suite.addTest(TestCountU('test_add'))
    # 执行测试
    runner=unittest.TextTestRunner()
    runner.run(suite)