[PYTHON]一个简单的单元測试框架

Posted 2020-09-21 gccbuaa

近期尝试了一下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]作为測试例的名字被占用了。