unittest单元测试Demo

import unittest
import HTMLTestRunner    #python3
import time
import sys
import os

class Test(unittest.TestCase):
    def setUp(self):
        print("----------SetUp -----\n")

    def tearDown(self):
        print("-----------TearDown----\n")

    def test_1(self):
        a=1
        b=2
        self.assertEqual(a,b)   #此项断言测试失败
    def test_2(self):
        a=1
        b=1
        self.assertEqual(a, b)  #此项断言测试通过



if __name__ == '__main__':

    suite = unittest.TestSuite()
    suite.addTest(Test('test_1'))
    suite.addTest(Test('test_2'))

    sv_report = 'report'  # 测试报告保存文件夹
    now = time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime())
    filename = sv_report + '/' + now + "_ReportCase.html"
    if sv_report not in os.listdir(os.getcwd()):  # 文件夹名称不存在才创建
        os.mkdir(os.getcwd() + '/' + sv_report + '/')
        fp = open(filename, 'wb')
    else:
        fp = open(filename, 'wb')
    print('run version {}'.format(sys.version))
    runner = HTMLTestRunner.HTMLTestRunner(stream=fp, title=u"自动化测试报告", description=u"TestCase Report")
    runner.run(suite)
    fp.close()