python自动化测试_从命令行指定测试用例

class RomanNumeralConverter(object):
    def __init__(self, roman_numeral):
        self.roman_numeral = roman_numeral
        self.digit_map = {"M":1000, "D":500, "C":100, "L":50, "X":10, "V":5, "I":1}
        
    def convert_to_decimal(self):
        val = 0
        for char in self.roman_numeral:
            val += self.digit_map[char]
        return val
    
    
    
import unittest
class RomanNumeralConverterTest(unittest.TestCase):
    def test_parsing_millenia(self):
        value = RomanNumeralConverter("M")
        self.assertEquals(1000, value.convert_to_decimal())
        
    def test_parsing_century(self):
        value = RomanNumeralConverter("C")
        self.assertEquals(100, value.convert_to_decimal())
        
if __name__ == "__main__":
    import sys
    suite = unittest.TestSuite()
    if len(sys.argv) == 1:
        suite = unittest.TestLoader().loadTestsFromTestCase(RomanNumeralConverterTest)
    else:
        for test_name in sys.argv[1:]:
            suite.addTest(RomanNumeralConverterTest(test_name))
            
    unittest.TextTestRunner(verbosity=2).run(suite)

运行结果：

test_parsing_century (__main__.RomanNumeralConverterTest) ... ok
test_parsing_millenia (__main__.RomanNumeralConverterTest) ... ok


----------------------------------------------------------------------
Ran 2 tests in 0.000s


OK

如果在命令行指定要运行的测试用例，效果如下：

主要就是这几句代码：

 suite = unittest.TestSuite()
    if len(sys.argv) == 1:
        suite = unittest.TestLoader().loadTestsFromTestCase(RomanNumeralConverterTest)
    else:
        for test_name in sys.argv[1:]:
            suite.addTest(RomanNumeralConverterTest(test_name))