Python Unittest 单元测试案例

目录

一、calculator.py

二、start_end.py

三、test_add.py

四、test_sub.py

五、run_test.py

---

一、calculator.py

加减法的实现类

class Math:
    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

二、start_end.py

开始结束输出类

import unittest

class setUp_tearDown(unittest.TestCase):
    def setUp(self):
        print("test start")

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

三、test_add.py

加法测试用例类

from calculator import *
from start_end import *

class TestAdd(setUp_tearDown):
    def test_add_1(self):
        m=Math(10,5)
        self.assertEqual(m.add(),15)

    def test_add_2(self):
        m=Math(5,3)
        self.assertEqual(m.add(),8)

四、test_sub.py

减法测试用例类

from calculator import Math
from start_end import *

class TestSub(setUp_tearDown):
    def test_sub_1(self):
        m=Math(5,3)
        self.assertEqual(m.sub(),2)

    def test_sub_2(self):
        m=Math(6,2)
        self.assertEqual(m.sub(),4)

五、run_test.py

运行测试用例

# 使用 discover 可以一次调用多个脚本
# test_dir 被测试脚本的路径
# pattern 脚本名称匹配规则

import unittest
test_dir="./"

discover=unittest.defaultTestLoader.discover(test_dir,pattern="test*py")

if __name__ == '__main__':
    runner=unittest.TextTestRunner()
    runner.run(discover)

    # 运行单个用例
    # suite=unittest.TestSuite()
    # suite.addTest(TestAdd("test_add_1"))
    # runner=unittest.TextTestRunner()
    # runner.run(suite)