单元测试1

单元测试框架

python为我们提供了unittest包用来进行单元测试，还是以Count这个类为例。

class Count:
    def __init__(self,a,b):
        self.a = a
        self.b = b


    def add(self):
        return self.a+self.b

下面就对Count进行单元测试。

import unittest
from test1 import Count

# TestCount继承unittest.TestCase
class TestCount(unittest.TestCase):
    def setUp(self):
        print('测试开始')
        
	# 第一个测试用例
    def test_add(self):
    	# 创建实例
        count = Count(1,2)
        # 判断1+2是否等于3
        self.assertEqual(count.add(),3)
        
	# 第二个测试用例
    def test_add1(self):
        count = Count(2,2)
        self.assertEqual(count.add(),4)

    def tearDown(self):
        print('测试结束')

if __name__ == '__main__':
    # todo 构造测试集
    suite = unittest.TestSuite()
    suite.addTest(TestCount('test_add'))

    # todo 执行测试
    runner = unittest.TextTestRunner()
    runner.run(suite)

测试结果

Ran 1 test in 0.004s

OK
测试开始
测试结束