#!/usr/bin/python
# -*- coding: utf-8 -*-

from unittest import TextTestRunner, TestCase
import tests

class AllTestCase(TestCase):
	def test_all(self):
		ttr = TextTestRunner(verbosity = 2)
		ttr.run(tests.get_all_test_suite())

if __name__ == "__main__":
	atc = AllTestCase()
	atc.test_all()

test.pyのポイント

• unittest.TestCaseを継承したクラスを含めることでPython2.7から追加されたdiscoverに対応する（python -m unittest discoverというコマンド1つで全体テストが実行出来る）
• if __name__ == “__main__”:でも実行するようにしてるのでpython test.pyでも全体テストが出来る

#!/usr/bin/python
# -*- coding: utf-8 -*-

import os, glob
from unittest import TestSuite, TestLoader, TextTestRunner

CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))

def get_all_test_mods():
	'''同階層のテストを全部インポートする'''
	test_mods = []
	pwd = os.getcwd()
	os.chdir(CURRENT_DIR)
	try:
		for file in glob.glob('*test.py'):
			test_mods.append(__import__(os.path.splitext(file.split('/')[-1])[0], globals(), locals(), []))
		return test_mods
	finally:
		os.chdir(pwd)

def get_all_test_pkgs():
	'''テストが定義されたパッケージをインポートする'''
	test_pkgs = []
	pwd = os.getcwd()
	os.chdir(CURRENT_DIR)
	try:
		for file in glob.glob('*/__init__.py'):
			pkg = __import__(file.split('/')[0], globals(), locals(), [])
			if getattr(pkg, 'get_all_test_suite', None):
				test_pkgs.append(pkg)
		return test_pkgs
	finally:
		os.chdir(pwd)

def get_all_test_suite():
	'''すべてのテストが含まれたTestSuiteを返す'''
	all_tests = TestSuite()

	for tm in get_all_test_mods():
		suite = TestLoader().loadTestsFromModule(tm)
		all_tests.addTest(suite)

	for tp in get_all_test_pkgs():
		suite = tp.get_all_test_suite()
		all_tests.addTest(suite)

	return all_tests

if __name__ == "__main__":
	ttr = TextTestRunner(verbosity = 2)
	ttr.run(get_all_test_suite())

__init__.pyのポイント

• tests配下の__init__.pyを全部上記の内容にすることで、同階層のテストとサブパッケージのテストを再帰的に行える
• if __name__ == ‘__main__’:でもテストを実行するようにしてるので、例えばpython tests/tests_libs/__init__.pyというコマンド1つでパッケージ毎のテストができる

#!/usr/bin/python
# -*- coding: utf-8 -*-

from libs import b
import unittest

class BTestCase(unittest.TestCase):
	'''Bのテストケース'''
	def setUp(self):
		'''テスト前処理'''
		pass

	def tearDown(self):
		'''テスト終了処理'''
		pass

	def test1_hoge(self):
		'''hogeテスト'''
		# ここにテストコードを書く
		pass

	def test2_fuga(self):
		'''fugaテスト'''
		# ここにテストコードを書く
		pass

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

tests_libs/btest.pyの例のポイント

• 内容が普通のunittest.TestCaseの使い方そのまんま（今回の構成には依存してない）
• if __name__ == ‘__main__’:でunittest.main()してるので、python tests/tests_libs/btest.pyというコマンド1つで単体テストできる