pandas拒绝低效数据驱动新手段海量资源.pdfVIP

本文由简悦SimpRead转码，原文地址

通过上一节课的学习，你应该解了数据驱动的基本用法，理解了DDT的原理，对于各种格式的数据，

都可以轻松地使用DDT进行数据驱动。

今天我将带领你深入pytest框架中的数据驱动。在pytest中，数据驱动是经由pytest自带的

pytest.mark.parametrize()来实现的。

pytest.mark.parametrize实现数据驱动

pytest.mark.parametrize是pytest的内置装饰器，它允许你在function或者class上定义多组参数和

fixture来实现数据驱动。

@pytest.mark.parametrize()

装饰器接收两个参数：第一个参数以字符串的形式存在，它代表能被被测

试函数所能接受的参数，如果被测试函数有多个参数，则以逗号分隔；第二个参数用于保存测试数据。

如果只有一组数据，以列表的形式存在，如果有多组数据，以列表嵌套元组的形式存在（例如[0,1]或

者[(0,1),(1,2)]）。

下面来看两组例子。

1.pytest.mark.parametrize单参数

importpytest

@pytest.mark.parametrize(number,[1,0])

deftest_equal(number):

assertnumber1

if__name____main__:

pytest.main([])

以上是单参数的一个例子，在这个例子中，test_equal函数接收一个参数number，这个参数有两组数

据，分别是1和0。

注意：装饰器pytest.mark.parametrize的第一个参数里的参数名称必须与测试函数中的参数称

保持一致。

在Pycharm中，使用右键选择运行这段代码，结果如下：

testsessionstarts

test_sample.py::test_equal[1]PASSED[50%]

test_sample.py::test_equal[0]FAILED[100%]

shorttestsummaryinfo

FAILED1.py::test_equal[0]-assert01

1failed,1passedin0.10s

可以看到，函数test_equal提供了两组参数1和0，所以它也执行了2次。

2.pytest.mark.parametrize多参数

pytest.mark.parametrize不仅支持单个参数，也可以支持多个参数，多个参数比较常见，因为在日常

工作中，我们提供测试数据，不仅仅包括用于测试的数据，还包括用于验证的数据，所以多参数还是比

较常见的。

pytest.mark.parametrize可以轻松支持多参数，请看下面的例子。

importtime

importpytest

fromseleniumimportwebdriver

@pytest.mark.baidu

classTestBaidu:

defsetup_method(self):

self.driver=webdriver.Chrome()

self.driver.implicitly_wait(30)

self.base_url=

@pytest.mark.parametrize(search_string,expect_string,[(iTesting,

iTesting),(,iTesting)])

deftest_baidu_search(self,search_string,expect_string):

driver=self.driver

driver.get(self.base_url+/)

driver.find_element_by_id(kw).send_keys(search_string)

driver.find_element_by_id(su).click()

time.sleep