Import UnitTest
From time import sleep
Import DDT
# import DDT module
From selenium import Webdriver
@ddt. DDT
# Declare the use of DDT
Class Mytestcase (UnitTest. TestCase):
def setUp (self):
Self.driver = Webdriver. Firefox ()
Self.base_url = "https://www.baidu.com/"
Self.driver.get (Self.base_url)
Self.driver.maximize_window ()
Sleep (2)
@ddt. Data (["China", "China _ Baidu Encyclopedia"],
["USA", "US _ Baidu Encyclopedia"],
["UK", "UK _ Baidu Encyclopedia"])
# test Data
@ddt. Unpack
# Unpack and pass the test data
def test_something (self, Searchterm, SearchResult):
Driver = Self.driver
Driver.find_element_by_xpath (".//*[@id = ' kw ']"). Send_keys (Searchterm)
Driver.find_element_by_xpath (".//*[@id = ' su ']"). Click ()
Sleep (2)
ResponseText = Driver.find_element_by_xpath (".//*[@id = ' 1 ']/h3/a"). Text
Self.assertequal (ResponseText, SearchResult)
def tearDown (self):
Self.driver.close ()
Self.driver.quit ()
if __name__ = = ' __main__ ':
Unittest.main ()
Python DDT Data Driven