ユニットテストのサンプル教えて

2024年6月8日

もちろんです。以下は、Pythonの unittest モジュールを使用した簡単なユニットテストのサンプルコードです。

まず、テスト対象のクラスとメソッドを持つモジュール calculator.py を考えます。

python
# calculator.py

class Calculator:
    def add(self, a, b):
        return a + b

    def subtract(self, a, b):
        return a - b

そして、それに対するユニットテストを行う test_calculator.py を作成します。

python
# test_calculator.py

import unittest
from calculator import Calculator

class TestCalculator(unittest.TestCase):
    def setUp(self):
        # テストケースごとに事前のセットアップが必要な場合に使用します
        pass

    def tearDown(self):
        # テストケースごとに後処理が必要な場合に使用します
        pass

    def test_add(self):
        calc = Calculator()
        result = calc.add(3, 5)
        self.assertEqual(result, 8)  # 3 + 5 = 8 が期待される結果

    def test_subtract(self):
        calc = Calculator()
        result = calc.subtract(7, 4)
        self.assertEqual(result, 3)  # 7 - 4 = 3 が期待される結果

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

このテストコードでは、Calculator クラスの add メソッドと subtract メソッドに対してそれぞれテストを行っています。各テストケースは unittest.TestCase を継承し、setUp メソッドと tearDown メソッドをオーバーライドすることができます。これらのメソッドは、それぞれテストケースの事前セットアップや後処理のために使われます。

テストの実行は、コマンドラインで python test_calculator.py を実行するか、IDEなどのツールを使用して行います。テストが正常に実行されると、各テストケースが成功したかどうかが表示されます。

これは非常に基本的な例ですが、実際のプロジェクトではもっと複雑なテストケースやモックを使用することが一般的です。