首页 > 代码库 > [PYTHON]一个简单的单元測试框架
[PYTHON]一个简单的单元測试框架
近期尝试了一下TDD(測试驱动)的模式。感觉效果不错。在此总结一下,同学们假设有更好的办法,一定要告诉我:)
1. 每一个功能模块(文件),配一个单元測试模块。
以手头这个项目为样例:有LogCat.py, LogModel.py, SceneBuilder.py 三个模块,那么就对应的新建LogCatTest.py, LogModelTest,SceneBuilderTest.py三个文件
2. 每一个函数都对应写一个单元測试例。
比方:在LogCat.py里有三个函数:
def parseDate(l):
p = "(?P<date>[0-9]+-[0-9]+) "
match = re.search(p, l)
s = ‘‘
if( match is not None):
s = match.group(‘date‘)
return s
def parsePath(l):
p = " (?P<path>(\S\S+\/)+\S+)"
match = re.search(p, l)
path = ‘‘
if( match is not None):
path = match.group(‘path‘)
if path.startswith("assets") is False:
path = "assets/" + path
return path
def parseTime(l):
p = " (?
P<time>[0-9]+:[0-9]+:[0-9]+.[0-9]+) "
match = re.search(p, l)
t = None
if( match is not None):
s = match.group(‘time‘)
t = datetime.datetime.strptime(s, ‘%H:%M:%S.%f‘)
return t
在LogCatTest.py里。写上对应的測试例:
#LogCatTest.py
from LogCat import *
logs = []
logs.append( ‘09-01 14:36:10.139 E/cocos2d-x debug info(23244): File loaded(4157 bytes): assets/UI/alpha/HVGA/card/card_turn_2.pvr.ccz‘)
logs.append( ‘09-04 15:41:26.895 D/Cocos2d-x debug(16852): File loaded: assets/lang/data/video.txt.dream‘)
logs.append( ‘09-04 15:41:26.895 D/Cocos2d-x debug(16852): File loaded: lang/data/video.start.mp3‘)
logs.append( ‘09-04 15:41:26.895 D/Cocos2d-x debug(16852): File loaded: assets/version.txt‘)
def unit_test_parseDate():
for l in logs:
print parseDate(l)
def unit_test_parseTime():
for l in logs:
print parseTime(l)
def unit_test_parsePath():
for l in logs:
print parsePath(l)
当然,这三个測试例測试的功能类似。能够整成一个。在这里仅仅是举个样例。
3. 在測试模块中,加入main函数, 其功能是通过命令行參数动态载入測试例:
#LogCatTest.py
import traceback
def main():
func = sys.argv[1]
unittest = "unit_test_" + func
globals()[unittest]()
if __name__ == "__main__":
try:
main()
except Exception as e:
traceback.print_exc()
sys.exit(1)
3. 在命令行执行:
python LogCatTest.py parsePath,就能够执行对应的測试例了。
这里要注意一个问题。 假设測试例也要从命令行读取參数,须要从sys.argv[2]開始,由于sys.argv[1]作为測试例的名字被占用了。
[PYTHON]一个简单的单元測试框架