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()