python_selenium_unittest基本框架

from selenium import webdriver
import unittest
import time

class BaiDu_test(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.url = "https://www.baidu.com/"
        self.driver.implicitly_wait(10)
        self.driver.maximize_window()
    def test_get(self):
        driver = self.driver
        driver.get(self.url)
        print("sleep")
        time.sleep(5)
        driver.find_element_by_id("kw").send_keys("selenium")
        time.sleep(2)
        driver.find_element_by_id("su").click()
        print("sarch")

    def tardown(self):
        self.driver.refresh()

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