Python unittestの基本と活用方法

Pythonのunittestは、Pythonの標準ライブラリの一部で、テストコードの実行に必要なクラスや関数がまとめられています。この記事では、その基本的な使い方と活用方法について説明します。

unittestの基本的な使い方

1. unittestをインポートします。
2. unittest.TestCaseを継承したクラスを作成します。
3. 作成したクラスの中に、テストケースを記述したメソッドを作成します。
4. unittest.main()でテストを実行します。

以下に、その基本的な使い方の例を示します。

import unittest

class TestSample(unittest.TestCase):
    def test_case1(self):
        self.assertEqual((10 + 5), 15)

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

このコードでは、TestSampleというunittest.TestCaseを継承したクラスを作成し、その中にtest_case1というテストケースを記述しています。unittest.main()を実行すると、このテストケースが実行されます。

unittestで例外をテストする

unittestでは、with self.assertRaisesを使用して例外の有無を確認することができます。以下にその使用例を示します。

import unittest

class TestSample(unittest.TestCase):
    def test_div(self):
        with self.assertRaises(ZeroDivisionError):
            return (5 / 0)

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

このコードでは、5 / 0がZeroDivisionErrorを引き起こすことをテストしています。

以上がPythonのunittestの基本的な使い方と活用方法です。これを活用することで、Pythonコードの品質を確保することができます。