My Portfolio

pythonのユニットテストのサンプル

登録日	：2024/09/15 10:27
カテゴリ	：Python基礎

pythonでのユニットテストの書き方サンプル

以下のクラスのテストをユニットテストで行う。
calcuration.py

class Cal(object):
    @staticmethod
    def add_num_and_double(x, y):
        if type(x) is not int or type(y) is not int:
            raise ValueError
        result = x + y
        result *= 2
        return result

calcuration.pyをユニットテストする。
test_calcuration.py

import unittest

import calcuration


release_name = 'prod'


class CalTest(unittest.TestCase):
    def setUp(self) -> None:
        print('setUp')
        self.cal = calcuration.Cal()

    def tearDown(self) -> None:
        print('cleanUp')
        del self.cal

    # @unittest.skip('skip!')
    @unittest.skipIf(release_name=='prod', 'skip_1')
    def test_add_num_and_double(self):
        self.assertEqual(self.cal.add_num_and_double(1, 1), 4)

    def test_add_num_and_double_raise(self):
        with self.assertRaises(ValueError):
            self.cal.add_num_and_double('1', '1')


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

実行は

python test_calcuration.py