unittest是python中的單元測試模組，類似於java中的junit包，下面就簡單介紹一下這個模組的使用方法

首先定義一個計算器類：calculator.py，其中有一個加法方法add

class Calculator:
    def __init__(self, a, b):
        self.a = int(a)
        self.b = int(b)
    
    def add(self):
        return self.a + self.b

接下來我們編寫另外一個檔案（單元測試程式碼與被測試程式碼分離開）：testadd.py

from calculator import Calculator
import unittest


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

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

    def test_add(self):
        s = Calculator(1, 2)
        self.assertEqual(s.add(), 3)

    def test_add2(self):
        s = Calculator(10, 20)
        self.assertEqual(s.add(), 30)


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

執行上述檔案就會自動執行TestAdd類中的test_add和test_add2方法，在Pycharm中執行的結果如下：

可以看到test case start和test case end分別被列印了兩次，這是由於在每個測試用例執行前，都會呼叫一次setUp方法用於初始化測試環境以及前置工作，每次測試用例執行後，自動呼叫tearDown方法恢復測試環境以及其他垃圾回收等工作。在本例中，在執行test_add和test_add2前後分別執行了這兩個方法，所以會有上圖所示的結果。

Pycharm自動識別出這是一個單元測試的程式，結果中顯示OK，ALL 2 tests passed，表示有兩個測試用例，且都通過。如果有不通過的測試用例，會有紅色展示。