お馬の写真 別館

「お馬の写真」管理者による徒然なるブログ

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」から始まる関数名にする必要あり。

 - プログラミング