Selenium WebDriver-通過斷言頁面是否存在某些關鍵字來確定頁面按照預期加載

#encoding=utf-8
import unittest
import time
import chardet
from selenium import webdriver
 
class VisitSogouByIE(unittest.TestCase):

    def setUp(self):
        #啟動IE瀏覽器
        #self.driver = webdriver.Firefox(executable_path = "e:\\geckodriver")
        self.driver = webdriver.Ie(executable_path = "e:\\IEDriverServer")
        
    def test_assertKeyWord(self):
        url = "http://www.baidu.com"
        # 訪問百度首頁
        self.driver.get(url)
        self.driver.find_element_by_id("kw").send_keys(u"光榮之路自動化測試")
        self.driver.find_element_by_id("su").click()
        time.sleep(4)
        # 通過斷言頁面是否存在某些關鍵字來確定頁面按照預期加載
        assert u"首頁--光榮之路" in self.driver.page_source, u"頁面源碼中不存在該關鍵字！"



    def tearDown(self):
        # 退出IE瀏覽器
        self.driver.quit()

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