Python菜譜8：支持簡單命令行

本任務(wù)最初的目的只是為了在測試過程中使用簡單的命令行運(yùn)行不同的函數(shù)，類似運(yùn)行 “python test_test.py” 運(yùn)行整個測試，運(yùn)行 “python test_test.py debug” 來運(yùn)行測試但是不收集運(yùn)行結(jié)果，請看如下的代碼:

import unittest
import sys

class Tests(unittest.TestCase):

    def testAddOnePlusOne(self):
        assert 1 == 2

def main():
    unittest.TextTestRunner().run(test_suite())

def test_suite():
    return unittest.makeSuite(Tests, 'test')

def debug():
    test_suite().debug()

if __name__ == '__main__':

    if len(sys.argv) > 1:
        globals()[sys.argv[1]]()
    else:
        main()

這里如果在命令行中直接運(yùn)行 “python cookbook_8.py” 就會執(zhí)行 “main()”；如果在命令行中運(yùn)行 “python cookbook_8.py debug” 會執(zhí)行 “debug()”。

“globals()” 返回的是當(dāng)前全局變量的引用。如果有其它的需求，可以充分利用本任務(wù)來延伸！