文檔測試

# Author：Luo
def add(a, b):
    """
    期望結果前面一定記得前面不要有空格，后面一定不要有其他無意義的數據
    :param a: 第一個數
    :param b: 第二個數
    :return: 返回a+b
>>> print(add(10,20))
30
    """
    return a+b

import doctest

# print(add(10,10))
doctest.testmod()

單元測試

# Author：Luo
def add(x, y):
    return x + y


def sub(x, y):
    return x - y


def mul(x, y):
    return x * y


# def sub(x,y):
#     return x/y

import unittest


class Test(unittest.TestCase):
    def setUp(self):  # 每個測試函數運行前運行
        pass

    def tearDown(self):  # 每個測試函數運行完后執行
        pass

    @classmethod
    def setUpClass(cls):  # 必須使用@classmethod 裝飾器,所有test運行前運行一次
        pass

    @classmethod
    def tearDownClass(cls):  # 必須使用@classmethod裝飾器,所有test運行完后運行一次
        pass

    def testAdd(self):
        self.assertEqual(first=add(1, 2), second=3, msg="你妹的加法都不會")

    def testSub(self):
        self.assertEqual(first=sub(2, 5), second=-3, msg="你妹的減法都不會")

    def testMul(self):
        self.assertEqual(first=mul(7, 8), second=56, msg="就你這樣還會乘法？")


if __name__ == "__main__":
    unittest.main() # 啟動測試

實例化測試類

class data:
    def __init__(self):
        pass
        
    def add(self,a,b):
        return a+b
        
    def sub(self, a, b):
        return a - b

import unittest
import myclass
class  mytest(unittest.TestCase):
    def setUp(self):
        self.tclass=myclass.data()#實例化測試的類
        pass
    def tearDown(self):
        pass
    def  testsum(self):
        self.assertEquals(self.tclass.add(1,2),31,"錯誤")
        pass
    def  testsub(self):
        self.assertEquals(self.tclass.sub(1, 2), -11, "錯誤")
        pass


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