例子1

import unittest
from time import sleep
from selenium import webdriver


class TestBaidu(unittest.TestCase):
    """测试百度"""

    @classmethod
    def setUpClass(self):   # 实践证明 另外， 方法的参数 cls 与  self  并没有什么本质区别， 都只表示方法的一个参数
        self.driver = webdriver.Firefox()
        self.baseUrl = "http://www.baidu.com"

    @classmethod
    def tearDownClass(self):
        self.driver.quit()

    # def setUp(self):
    #     self.driver = webdriver.Firefox()
    #     self.baseUrl = "http://www.baidu.com"

    # def tearDown(self):
    #     self.driver.quit()

    def baiduSearch(self, searchKey):
        driver = self.driver
        driver.get(self.baseUrl)
        driver.find_element_by_id("kw").send_keys(searchKey)
        driver.find_element_by_id("su").click()
        sleep(2)

    def test_search_selenium(self):
        '''搜索selenium'''
        searchKey = "selenium"
        self.baiduSearch(searchKey)
        self.assertEqual(self.driver.title, searchKey+"百度搜索")

    def test_search_unittest(self):
        searchKey = "unittest"
        self.baiduSearch(searchKey)
        self.assertEqual(self.driver.title, searchKey+"_百度搜索")

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