unittest - 对输入和输出列表运行相同的测试

Question

我有这个测试

import unittest

class TestName(unittest.TestCase):

        def setUp(self):
                self.name = "Bob"
                self.expected_name = "Bob"


        def test_name(self):
                # ... some operation over self.name
                print self.name
                self.assertEquals(self.name, self.expected_name)

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

我如何运行测试实例？

为输入和输出列表运行相同的测试(["Bob", "Alice", ...]),也许就像

TestName(name="Bob", expected_name="Bob")
TestName(name="Alice", expected_name="Alice")

Answer 1

查看DDT(数据驱动/装饰测试).

DDT允许您通过运行不同的测试数据来乘以测试用例,使其显示为多个测试用例.

考虑这个例子,使用DDT:

import unittest

from ddt import ddt, data, unpack


@ddt
class TestName(unittest.TestCase):

        # simple decorator usage:
        @data(1, 2)
        def test_greater_than_zero(self, value):
            self.assertGreater(value, 0)

        # passing data in tuples to achieve the 
        # scenarios from your given example:
        @data(('Bob', 'Bob'), ('Alice', 'Alice'))
        @unpack
        def test_name(self, first_value, second_value):
            name, expected_name = first_value, second_value
            self.assertEquals(name, expected_name)

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

我在上面的代码中定义了2个测试方法,但是将使用装饰器中提供的数据运行4个测试用例.

输出:

test_greater_than_zero_1 (__main__.TestName) ... ok
test_greater_than_zero_2 (__main__.TestName) ... ok
test_name_('Alice', 'Alice') (__main__.TestName) ... ok
test_name_('Bob', 'Bob') (__main__.TestName) ... ok

----------------------------------------------------------------------
Ran 4 tests in 0.000s

OK