分别基于和不基于unittest单元测试框架对一个加法做单元测试

不使用unittest单元测试框架做单元测试

被测代码，count.py

class Count:
    def add(self, a, b):
        return a+b
if __name__ == '__main__':
    a = Count()
    res = a.add(1, 2)
    print(res)

测试用例代码，test.py

from count import Count

def test_add_1_1():
    try:
        c = Count()
        result = c.add(1, 1)
        assert result==2, '1+1 fail'
    except AssertionError as e:
        print(e) # 如果执行，则输出1+1 fail
    else:
        print('1+1 pass') # 用例执行成功输出1+1 pass

def test_add_1_2():
    c = Count()
    result = c.add(1, 2)
    assert result==3

if __name__ == '__main__':
    test_add_1_1()
    test_add_1_2()

　　

使用unittest单元测试框架做单元测试

test_unittest.py

import unittest
from count import Count

class TestCount(unittest.TestCase):
    def setUp(self):
        print('测试用例开始执行...')
    def tearDown(self):
        print('测试用例执行完成...')

    def test_1_1(self):
        self.a = Count()
        ret = self.a.add(1, 1)
        self.assertEqual(ret, 2)

    def test_1_2(self):
        a = Count()
        ret = a.add(1, 2)
        self.assertEqual(ret, 3)

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

　　

import unittest
from count import Count

class TestCount(unittest.TestCase):
    def setUp(self):
        print('测试用例开始执行...')
        self.a = Count()
    def tearDown(self):
        print('测试用例执行完成...')

    def test_1_1(self):
        ret = self.a.add(1,1)
        self.assertEqual(ret, 2)

    def test_1_2(self):
        ret = self.a.add(1, 2)
        self.assertEqual(ret, 3)

if __name__ == '__nami__':
    unittest.main()