python使用ddt模塊對用例執行操作

import time

import unittest

import ddt

from selenium import webdriver

TEST_URL = "http://www.baidu.com"
TEST_DATA = ("python", "selenium", "unittest")

"""
測試類前加修飾@ddt.ddt
case前加修飾@ddt.data() 相同的測試用例
@ddt.unpack 分解data的參數
"""

@ddt.ddt
class MyTestCase(unittest.TestCase):

def setUp(self):
"""
執行測試用例前的初始化操作
"""
self.browser = webdriver.Chrome(executable_path="C:\Program Files (x86)\Google\Chrome\Application\chromedriver.exe")

@ddt.data(*TEST_DATA)
def testBaidu(self, keyword):
"""
具體的測試用例
"""
self.browser.get(TEST_URL)
self.browser.find_element_by_id("kw").send_keys(keyword)
self.browser.find_element_by_id("su").click()
time.sleep(5)
self.browser.quit()

def tearDown(self):
"""
執行測試用例后的操作
"""
print "Test Done..."

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