Pythonでunittest

Unittestとは

unittest（単体テスト）は、作成した関数やプログラムが期待通り動作するか試行することを目的としています。
今回は覚書もかねて大まかな共有します。

環境

macOS 10.14.6

Unittestの基本的使い方

以下はtashizan関数をtestするときのコードです。
tashizanは引数２つに対し、その和を返す関数です。

Test.py

import unittest

def tashizan(val1,val2):
    ans = val1 + val2
    return ans

class Test(unittest.TestCase):
    
    def test_tashizan(self):
        actual = tashizan(1, 2)
        expected = 3
        self.assertEqual(actual, expected)
        
if __name__ = "__main__":
    unittest.main()

Test.py内のコードについて上から順に説明します。

Test.py

import unittest

def tashizan(val1,val2):
    ans = val1 + val2
    return ans

unittestをインポートし、関数tashizanを定義します。

Test.py

class Test(unittest.TestCase):
    
    def test_tashizan(self):
        actual = tashizan(1, 2)
        expected = 3
        self.assertEqual(actual, expected)

①Testクラスを定義し、unittest内のTestCaseクラスを継承します。
②変数actualに答えを格納します。
③変数expectedに期待される答えを格納します。この場合1+2=3となるため、expectedには3を格納しておきます。
④TestCase内のassertEqualメソッドは２つの引数が等しいかどうかを返します。