import unittest
from time import sleep
from selenium import webdriver

class TestBdai(unittest.TestCase):

    
    @classmethod#可減少瀏覽器關閉次數
    def setUp(cls):
        cls.driver = webdriver.Chrome()
        cls.base_url = "https://baidu.com/"

    def baidu_search(self, search_key):#封裝搜尋功能
        self.driver.get(self.base_url)
        self.driver.find_element_by_id(
 
"kw").send_keys(search_key)
        self.driver.find_element_by_id("su").click()
        sleep(2)

    def test_search_key_selenium(self):
        search_key = "selenium"
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title, search_key+"_百度搜索")

    def test_search_key_unittest(self):
        search_key 
 
= "unittest"
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title, search_key+"_百度搜索")

    @classmethod
    def tearDown(cls):
        cls.driver.quit()

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

注意事項：1、Unittest查詢和執行測試用例時，只會把以“test”開頭的方法當做測試用例。

2、把斷言寫在每一條測試用例裡。