Python_单元测试（unittest模板）

单元测试（unittest模板）

#encoding=utf-8
import unittest
import random
class TestSequenceFunctions(unittest.TestCase):
    def setUp(self):#初始化环境
        # 初始化一个递增序列
        self.seq = list(range(10))
        print ("setup completed!")

    def test_run(self):#必须以test开头，不能带其他参数
        # 从序列seq中随机选取一个元素
        element = random.choice(self.seq)
        # 验证随机元素确实属于列表中
        self.assertTrue(element in self.seq)

    def test_sth(self):
        assert 1==1

    def tearDown(self):#环境清理
        print ("tearDown completed")

class TestDictValueFormatFunctions(unittest.TestCase):
    def setUp(self):
        self.seq = list(range(10))

    def test_shuffle(self):
        # 随机打乱原seq的顺序
        random.shuffle(self.seq)
        self.seq.sort()
        self.assertEqual(self.seq, list(range(10)))
        # 验证执行函数时抛出了TypeError异常
        self.assertRaises(TypeError, random.shuffle, (1, 2, 3))

if __name__ == '__main__':
    unittest.main()

运行结果：

成功后，会打印一个.