Python Selenium 与 unittest 集成

import unittest
from selenium import webdriver

class TestWeb(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(10)

    def test_search(self):
        self.driver.get("https://www.baidu.com")
        assert "百度一下" in self.driver.title
        elem = self.driver.find_element_by_name("wd")
        elem.send_keys("pidancode.com")
        elem.submit()
        assert "pidancode.com" in self.driver.page_source

    def test_title(self):
        self.driver.get("https://www.baidu.com")
        assert "百度一下" in self.driver.title
        self.driver.find_element_by_id("kw").send_keys("皮蛋编程")
        self.driver.find_element_by_id("su").click()
        assert "皮蛋编程" in self.driver.title

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

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