python3+selenium(十六)类/函数/模块编写百度搜索

命题

通过类/函数/模块实现百度搜索

思路

1. 导入模块
2. 定义类：百度搜索

• 定义实例：打开网页
• 定义实例：搜索信息，并进行断言

源码

import time
from selenium import webdriver

class TestBaidu(object):
    dr=webdriver.Chrome()
    dr.maximize_window()
    dr.implicitly_wait(2)

    def Open_Url(self):
        self.dr.get("http://www.baidu.com")
        time.sleep(1)

    def Test_Baidu(self):
        self.dr.find_element_by_id('kw').send_keys("selenium")
        time.sleep(1)
        print(self.dr.title)
        try:
            assert "selenium" in self.dr.title
            print("Test pass.")
        except Exception as e:
            print("Test fail.")
        self.dr.quit()

baidu=TestBaidu()
baidu.Open_Url()
baidu.Test_Baidu()