unittest

ユニットテスト（標準ライブラリ）

意味プログラムの動作確認

unittestとは？

unittestは、Pythonのコードが正しく動作するかを確認するためのフレームワークです。プログラムの一部をテストケースとして記述し、自動で実行することで、コードの品質を保ちながら開発を進めることができます。

unittestの具体的な使い方

基本的な単体テスト

import unittest

def add(a, b):
    return a + b

class TestAddFunction(unittest.TestCase):
    def test_add_positive_numbers(self):
        self.assertEqual(add(2, 3), 5)
    
    def test_add_negative_numbers(self):
        self.assertEqual(add(-1, -1), -2)
    
    def test_add_zero(self):
        self.assertEqual(add(5, 0), 5)

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

👇出力結果

...
----------------------------------------------------------------------
Ran 3 tests in 0.001s

OK

このコードは、unittestモジュールを使用して基本的な単体テストを作成する方法を示しています。TestAddFunctionクラスがunittest.TestCaseを継承し、各テストケースをメソッドとして定義しています。assertEqual()メソッドを使用して、add()関数の結果が期待値と一致することを確認しています。

テストフィクスチャの使用

import unittest

class Calculator:
    def __init__(self):
        self.result = 0
    
    def add(self, x):
        self.result += x
    
    def subtract(self, x):
        self.result -= x

class TestCalculator(unittest.TestCase):
    def setUp(self):
        self.calc = Calculator()
    
    def test_add(self):
        self.calc.add(5)
        self.assertEqual(self.calc.result, 5)
        self.calc.add(3)
        self.assertEqual(self.calc.result, 8)
    
    def test_subtract(self):
        self.calc.add(10)
        self.calc.subtract(3)
        self.assertEqual(self.calc.result, 7)

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

👇出力結果

..
----------------------------------------------------------------------
Ran 2 tests in 0.001s

OK

このコードは、テストフィクスチャ（setUp()メソッド）を使用してテストを設定する方法を示しています。各テストメソッドの前にsetUp()メソッドが自動的に呼び出され、新しいCalculatorオブジェクトが作成されます。これにより、各テストが独立して実行され、テスト間で状態が干渉しないようになります。

unittestに関するよくある質問

Q. unittestの基本的な使い方は？

A. unittest.TestCaseを継承したクラスを作成し、その中にtest_で始まるメソッドを定義します。各メソッド内で、self.assertEqualなどのアサーションメソッドを使ってテストを記述します。

Q. テストを実行するには？

A. unittest.main()を呼び出すことで、定義したすべてのテストを実行できます。通常、if __name__ == '__main__':ブロック内に記述します。

Q. 複数のテストを書くには？

A. TestCaseクラス内に複数のtest_で始まるメソッドを定義することで、複数のテストケースを作成できます。各メソッドが1つのテストケースとして扱われます。

unittestが学べる書籍の紹介

「プログラミング単語帳」を使って、プログラミングの単語を英単語のように学習してみませんか？
プログラミング単語帳には、Pythonのunittestやthreading.Lock、dis、multiprocessing.Queue、などのような実務でよく使われる単語が数百以上収録されています。
この書籍には、プログラミングの単語の意味や読み方、単語の使い方がわかる例文などが掲載されており、いつでもどこでもプログラミングの学習ができます。

よく使われる単語にだけ絞って学習することができるので、効率的にプログラミングが学習できます。