Python: unittest

a0002_009631
Pythonで少しだけ遊んでみています。
単体テストを使えるようにする上で少しだけハマったのでメモ代わりに記事にしておきます。

まずは簡単なテストコード。

import unittest
import MySQLdb
import sharedhorsexml


class shardhorsexmlTest(unittest.TestCase):

    def setUp(self):
        self.con = MySQLdb.connect(host='localhost', db='XXXX',
                                   user='XXXX', passwd='XXXX')

    def test_create_horseinfo_notnone(self):
        self.assertIsNotNone(sharedhorsexml.create_horseinfo(self.con.cursor(),
                                                             '2012104639'))

    def test_create_horseinfo_none(self):
        self.assertIsNone(sharedhorsexml.create_horseinfo(self.con.cursor(),
                                                          '2011106312'))


if __name__ == "__main__":
    unittest.main()
  • テスト用のクラスはunittest.TestCaseから継承。
  • 1つずつのテストケースには「test」から始まる関数名にする必要あり。
スポンサーリンク

フォローする

スポンサーリンク