Pythonでユニットテストをする方法

import unittestして、unittest.TestCaseを継承したクラスを作るだけ。サンプルを参考にすると以下のようになる。

import unittest

class Calc:
    def add(self, a, b):
      return a + b
  
    def sub(self, a, b):
      return a - b

class CalcTest(unittest.TestCase):
    def test_add(self):
        calc = Calc()
        self.assertEqual(calc.add(1, 2), 3)

    def test_sub(self):
        calc = Calc()
        self.assertEqual(calc.sub(4, 2), 2)

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

実行すると以下のようになる。

python main.py
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK