python自动化测试之DDT数据驱动的实现代码

下面是“python自动化测试之DDT数据驱动的实现代码”的完整攻略：

一、什么是DDT数据驱动？

DDT，即 Data-Driven Testing，数据驱动测试。它是一种基于数据的测试方法，它的主要思想是不同的输入数据可以得到不同的测试结果，因此我们可以通过不同的数据来验证系统的稳定性和可靠性。DDT可以通过将测试数据与测试脚本分离，实现更好的复用性和可维护性。

二、使用DDT进行数据驱动的实现

1. 安装DDT库

首先要安装DDT库，可以使用如下命令安装：

pip install ddt

2. 书写测试用例

在测试用例中需要使用@ddt装饰器，它能够将测试数据逐条传入测试用例中。

import unittest
from ddt import ddt, data, unpack

@ddt
class TestStringMethods(unittest.TestCase):

    @data(('ACB', 'ABC'), ('xyZ', 'xyz'))
    @unpack
    def test_lower(self, input_string, expected_output):
        self.assertEqual(input_string.lower(), expected_output)

上面的代码中使用了@data装饰器来指定测试数据，这里使用了一个元组来表示一条测试数据，元组中第一个元素为输入值，第二个元素为预期输出值。

使用@unpack来使元组中的元素能够被分开，分别传入测试用例中的不同参数中。这里input_string和expected_output就是元组中的两个元素。

3. 运行测试用例

最后一步是运行测试用例，使用unittest.TextTestRunner().run()方法来执行测试用例。

if __name__ == '__main__':
    unittest.TextTestRunner().run(unittest.TestLoader().loadTestsFromTestCase(TestStringMethods))

上面的代码中，使用TestLoader()来加载测试用例，并使用TextTestRunner()来运行测试用例。

三、示例说明

下面给出两个示例代码来说明DDT数据驱动的实现过程：

1. 字符串大小写转换

import unittest
from ddt import ddt, data, unpack

@ddt
class TestStringMethods(unittest.TestCase):

    @data(('ACB', 'ABC'), ('xyZ', 'xyz'))
    @unpack
    def test_lower(self, input_string, expected_output):
        self.assertEqual(input_string.lower(), expected_output)

    @data(('abc', 'ABC'), ('XYz', 'xyz'))
    @unpack
    def test_upper(self, input_string, expected_output):
        self.assertEqual(input_string.upper(), expected_output)

if __name__ == '__main__':
    unittest.TextTestRunner().run(unittest.TestLoader().loadTestsFromTestCase(TestStringMethods))

上面的代码中，test_lower和test_upper方法分别测试了lower()和upper()两个字符串大小写转换函数，使用了不同的测试数据。

2. 计数器测试

import unittest
from ddt import ddt, data, unpack

class Counter(object):

    def __init__(self):
        self.counter = 0

    def increment(self):
        self.counter += 1

    def decrement(self):
        self.counter -= 1

    def count(self):
        return self.counter

@ddt
class TestCounter(unittest.TestCase):

    def setUp(self):
        self.counter = Counter()

    @data((1, 2), (3, 4))
    @unpack
    def test_increment(self, initial_value, expected_value):
        self.counter.counter = initial_value
        self.counter.increment()
        self.assertEqual(self.counter.counter, expected_value)

    @data((5, 4), (3, 2))
    @unpack
    def test_decrement(self, initial_value, expected_value):
        self.counter.counter = initial_value
        self.counter.decrement()
        self.assertEqual(self.counter.counter, expected_value)

if __name__ == '__main__':
    unittest.TextTestRunner().run(unittest.TestLoader().loadTestsFromTestCase(TestCounter))

上面的代码中，Counter类实现了一个简单的计数器，包含了增加和减少计数器值的方法。在TestCounter测试类中，使用两个方法来测试增加和减少计数器的功能，并使用不同的测试数据。