unittest 参数化1--数据驱动（虫师《selenium3自动化测试实战--基于Python语言笔记36》）...

以百度搜索为例：

baidu_data.csv:

 

 代码为：

import csv
import codecs
import unittest
from time import sleep
from selenium import webdriver
from itertools import islice


class TestBaidu(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
        cls.base_url = "http://www.baidu.com"
        cls.test_data = []
        with codecs.open('D:/Test1/data_csv/baidu_data.csv', 'r', 'utf_8_sig') as f:
            data = csv.reader(f)
            for line in islice(data, 1, None):
                cls.test_data.append(line)

    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()

    def baidu_search(self, search_key):
        self.driver.get(self.base_url)
        self.driver.find_element_by_id("kw").send_keys(search_key)
        self.driver.find_element_by_id("su").click()
        sleep(3)

    def test_search_selenium(self):
        self.baidu_search(self.test_data[0][1])

    def test_search_unittest(self):
        self.baidu_search(self.test_data[1][1])

    def test_search_parameterized(self):
        self.baidu_search(self.test_data[2][1])

if __name__ == '__main__':
    unittest.main(verbosity=2)

执行结果为：

 

转载于:https://www.cnblogs.com/kite123/p/11561634.html