欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

使用unittest和ddt进行数据驱动

程序员文章站 2022-06-23 23:17:22
1、安装ddt 2、卸载ddt ......

1、安装ddt

#pip install ddt

2、卸载ddt

 

 

# coding = utf-8
# encoding = utf-8
import ddt
import time
import unittest
import logging
import traceback
from selenium import webdriver
from selenium.common.exceptions import nosuchelementexception

logging.basicconfig(
    level=logging.info,
    format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',
    datefmt='%a %d %b %y %h: %m: %s',
    filename='d://pytest//test//report.log',
    filemode='w'
)


@ddt.ddt
class testdemo(unittest.testcase):
    def setup(self):
        self.driver = webdriver.chrome()
        self.driver.maximize_window()

    @ddt.data([u"神奇动物在哪里", u"叶茨"],
              [u"疯狂动物城", u"古德温"],
              [u"大话西游之月光宝盒", u"周星驰"])
    @ddt.unpack
    def test_datadrivenbyobj(self, testdata, expectdata):
        url = 'http://www.baidu.com'
        self.driver.get(url)
        self.driver.implicitly_wait(10)
        try:
            self.driver.find_element_by_id("kw").send_keys(testdata)
            self.driver.find_element_by_id("su").click()
            time.sleep(3)
            self.asserttrue(expectdata in self.driver.page_source)
        except nosuchelementexception:
            logging.error(u"查找的页面元素不存在:" + str(traceback.format_exc()))
        except assertionerror:
            logging.info(u"搜索:%s,期望:%s,失败" % (testdata, expectdata))
        else:
            logging.info(u"搜索:%s,期望:%s,通过" % (testdata, expectdata))

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


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