单元测试之路（一）

使用UnitTest编写单元测试用例

1.测试文件中的加法函数add
1.1在UnitTest目录下建立test.py文件

def add(x, y, *args):
    x = x + y
    for val in args:
        x = x + val
    return x

1.2另外建立一个py文件，首先导入unittest及test文件下的add函数

import unittest
from UnitTest.test import add

1.3编写用例

class UnitTestAdd(unittest.TestCase):     
    def setUp(self) :             #准备环境
        pass

def tearDown(self):              #清理环境
    pass

def test_run(self):                      #测试用例，必须以test开头才能识别到用例
    print(add(1, 3, 6) == 10)            #断言

1.4执行用例

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

2.测试类中的add函数和sub函数
2.1在类中编写多个函数

class Count:
    def add(x, y, *args):
        x = x + y
        for val in args:
            x = x + val
        return x


    def sub(x, y, *args):
        x = x - y
        for val in args:
            x = x - val
        return x

2.2编写两条用例

class TestCount(unittest.TestCase):
    def setUp(self):
        pass

    def tearDown(self):
        pass

    def test_add(self):                          #测试加法函数
        print(Count.add(1, 3, 6) == 10)

    def test_sub(self):                          #测试减法函数
        print(Count.sub(1, 3, 6) == -8)