unittest（封装用例）

from selenium import webdriver
from time import sleep
import unittest　　#导入unittest库
import HTMLTestRunner

#创建一个类，并且该类继承unittest.case类
class Test_baidu(unittest.TestCase):
    #初始环境,每一个用例执行时都会先执行这里的代码
    def setUp(self):
        self.dr=webdriver.Chrome()　　#调用指定的浏览器并赋值给dr
        self.dr.maximize_window()
        self.dr.implicitly_wait(10)　　#智能等待作用和sleep一样
        self.dr.get('https://www.baidu.com')
        sleep(2)

    #测试用例---执行搜索新方硕--测试用例的脚本
    def test_case2(self):
        self.dr.find_element_by_id('kw').send_keys(u'新方硕')
        sleep(3)

    #测试用例--执行搜索新东方
    def test_case1(self):
        self.dr.find_element_by_id('kw').send_keys(u'新东方')
        sleep(3)

   #还原环境（结束）---用例执行后关闭浏览器（扫尾工作）
    def tearDown(self):
        self.dr.close()

#方案一
if __name__=="__main__":
    unittest.main() 　　#他可以将一个单元测试的模块变为可以执行的脚本，所谓main（）的方法使用了
    # unittest中testloader类来搜索以test开头的测试用例，并自动执行
    # main（）执行顺序是以ascii用来执行顺序0-9，A-Z,a-z