selenium与unittest结合使用

import unittest
from selenium import webdriver
from HTMLTestRunner import HTMLTestRunner
Class MyTest(unittest.TestCase):
    def setUp(self):
    	self.br = webdriver.Firefox()
    	self.br.get("https://www.baidu.com")
    def tearDown(self):
    	self.br.quit()
	def test_search(self):
        self.br.find_element_by_id("kw").send_keys("好好学习，天天向上")
        self.br.find_element_by_id("su").click()
        self.assertEqual(self.br.title,'好好学习，天天向上_百度搜索')
    
if __name__=='__main__':
    #1.使用main()执行测试用例
    unittest.main()	
    #2.使用测试套件执行测试用例,使用该方法可以生成HTML测试报告
    #生成测试套件
    test_suite = unittest.TestSuite()
    #添加测试用例
    test_suite.addTest(MyTest('test_search'))
    #定义HTML报告存放路径
    fp = open('./result.html','wb')	#wb表示二进制写入
    #生成测试执行器,返回一个执行器对象
    test_runner = HTMLTestRunner(stream=fp,title='百度搜索测试报告',description='用例执行情况')
    #使用测试执行器执行测试套件(最终执行了测试套件中的测试用例)
    test_runner.run(test_suite)
    fp.close()