一、摘要
前边的文章我们看到执行测试用例的是通过调用unittest.mian()函数,它会将模块的测试用例收集起来并执行,然而当我们的测试用例增多了以后,这样的执行非常不灵活而且没有效率,我们更愿意看到自由组合的一组用例,只执行这组用例的方式,unittest.TestSuite能够满足我们这个需求,笔者将一random类作为被测试类来详细展示unittest.TestSuite
二、代码示例
- 以前的执行方式
# encoding = utf-8 import random import unittest class TestRandomFunction(unittest.TestCase): def setUp(self): self.str = "abcdef!@#$%" def tearDown(self): pass def test_randomchoice(self): var = random.choice(self.str) self.assertTrue(var in self.str) print(var) def test_randomsample(self): with self.assertRaises(ValueError): random.sample(self.str, 100) for var in random.sample(self.str, 6): self.assertTrue(var in self.str) print(var) class TestRandomShuffleFunction(unittest.TestCase): def setUp(self): self.list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 13] def tearDown(self): pass def test_randomshuffle(self): random.shuffle(self.list) print(self.list) self.list.sort() self.assertEqual(self.list, [1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 13]) if __name__ == '__main__': unittest.main()
- 方式一 使用unittest.TestLoader,它可以通过传给他的参数获取测试用例的测试方法,然后再组合成TestSuite,最后在将TestSuite传递给TestRunner 完成我们所期望的执行组合
# encoding = utf-8 import random import unittest class TestRandomFunction(unittest.TestCase): def setUp(self): self.str = "abcdef!@#$%" def tearDown(self): pass def test_randomchoice(self): var = random.choice(self.str) self.assertTrue(var in self.str) print(var) def test_randomsample(self): with self.assertRaises(ValueError): random.sample(self.str, 100) for var in random.sample(self.str, 6): self.assertTrue(var in self.str) print(var) class TestRandomShuffleFunction(unittest.TestCase): def setUp(self): self.list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 13] def tearDown(self): pass def test_randomshuffle(self): random.shuffle(self.list) print(self.list) self.list.sort() self.assertEqual(self.list, [1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 13]) if __name__ == '__main__': # unittest.main() testcase1 = unittest.TestLoader().loadTestsFromTestCase(TestRandomFunction) testcase2 = unittest.TestLoader().loadTestsFromTestCase(TestRandomShuffleFunction) suite = unittest.TestSuite([testcase1, testcase2]) unittest.TextTestRunner(verbosity=2).run(suite)
- 方式二 另创建一个.py文件,定义suite方法,使用unittest.TestSuite().addTest(测试类(测试方法))
# encoding = utf-8 import unittest from unittest3.TestSuiteDemo2 import * def suite(): suite = unittest.TestSuite() suite.addTest(TestRandomFunction("test_randomchoice")) suite.addTest(TestRandomShuffleFunction("test_randomshuffle")) return suite if __name__ == '__main__': runner = unittest.TextTestRunner() runner.run(suite())
- 方式三 另创建一个.py文件,使用unittest.TestLoader().discover(“路径”,“匹配文件名”)
# encoding = utf-8 import unittest if __name__ == '__main__': suite = unittest.TestLoader().discover('.', pattern='TestSuiteDemo1.py') unittest.TextTestRunner(verbosity=2).run(suite)
三、random实例
import random import string #随机获取0到10之间随机整数N,0 <= N <= 10, 语法randint(a, b),a <= N <= b random.randint(0,10) #等同于 random.randrange(0,11) #随机获取0到10之间的偶数M, 语法:randrange(start,stop,[step]) random.randrange(0,11,2) #随机获取浮点数 random.random() #随机后去0到10之间的浮点数O,语法:uniform(a, b)[a <= N <= b for a <= b and b <= N <= a for b < a] random.uniform(0,10) #获取字符串随机字符 random.choice('string') #获取特定数量的随机字符,返回list random.sample('string',3) #多个字符中选取特定数量的字符组成新字符串 list1 = ['a','b','c','d','e','f','g','h','i','j', 1, 2, 3] list2 = [str(i) for i in list1] str1 = ''.join(random.sample(list2, 5)) #获取随机字符串 random.choice(['apple', 'pear', 'peach', 'orange', 'lemon']) random.choice('abcdefghijklmn') #洗牌 string1 = 'a b c d e f g' string2 = string1.split() random.shuffle(string2) print(string2)
四、类说明
TestLoader类:它用于加载测试用例,然后返回一个测试用例集和
LoadTestsFromTestCase类:根据传给它的测试类获取其中以test开头的测试方法,然后返回一个测试集合
TestSuite类:用于组装测试用例的实例,然后传递给TestRunner进行测试执行
TextTestRunner类:用于执行测试用例,Text表示是以文本形式输出测试结果,它会返回一个TestResult实例对象,用于存储测试用例执行过程中的详细信息,可以使用Python的dir()进行查看
五、参数verbosity说明
verbosity =< 0时,输出结果中不提示执行成功的用例数
verbosity = 1 时,输出结果中仅仅以(.)的形式表示执行成功的用例数,成功几个就是几个点
verbosity >= 2 时,可以输出每个用例执行的详细信息,特别是在用例较多的情况下,此设置会比较有用