在 python 中编写类似单元测试的 DRY 方式

Question

我在 python 中有一些类似的单元测试。 如此相似，以至于只有一个论点在改变。

class TestFoo(TestCase):
    def test_typeA(self):
        self.assertTrue(foo(bar=TYPE_A))

    def test_typeB(self):
        self.assertTrue(foo(bar=TYPE_B))

    def test_typeC(self):
        self.assertTrue(foo(bar=TYPE_C))

    ...

显然这不是很干，如果你有 4-5 个不同的选项，代码将会非常重复

现在我可以做这样的事情了

class TestFoo(TestCase):
    BAR_TYPES = (
        TYPE_A,
        TYPE_B,
        TYPE_C,
        ...
    )

    def _foo_test(self, bar_type):
        self.assertTrue(foo(bar=bar_type))

    def test_foo_bar_type(self):
        for bar_type in BAR_TYPES:
            _foo_test(bar=bar_type))

这可行，但是当引发异常时，我如何知道 _foo_test 是否使用参数 TYPE_A、TYPE_B 或 TYPE_C 失败？

也许有更好的方法来构建这些非常相似的测试？

Answer 1

大多数 TestCase 断言方法，包括 assertTrue，都采用可选的 msg 参数。

如果您更改 BAR_TYPES 元组以包含变量名称，那么您可以将其包含在断言失败时显示的消息中。

class TestProcessCreateAgencyOfferAndDispatch(TestCase):
    BAR_TYPES = (
        ('TYPE_A', TYPE_A),
        ('TYPE_B', TYPE_B),
        ('TYPE_C', TYPE_C),
        ...
    )

    def _foo_test(self, var_name, bar_type):
        self.assertTrue(foo(bar=bar_type), var_name)

    def test_foo_bar_type(self):
        for (var_name, bar_type) in BAR_TYPES:
            _foo_test(bar=bar_type), var_name=var_name)

Answer 2

你想要做的本质上是一个参数化的测试。此功能不包含在标准 django 或 python unittest 模块中，但许多库提供它：nose-parameterized、py.test、ddt

到目前为止，我最喜欢的是 ddt：它最类似于 NUnit-JUnit 风格的参数化测试，非常轻量级，不会妨碍您，也不需要专门的测试运行器（如 nose-parameterized 那样）。它可以帮助您的方式是修改测试名称以包含所有参数，因此您可以通过查看测试名称清楚地看到哪个测试用例失败。

使用 ddt，您的示例将如下所示：

import ddt

@ddt.ddt
class TestProcessCreateAgencyOfferAndDispatch(TestCase):

    @ddt.data(TYPE_A, TYPE_B, TYPE_C)
    def test_foo_bar_type(self, type):
        self.assertTrue(foo(bar=type))

在这种情况下，名称将类似于test_foo_bar_type__TYPE_A（从技术上讲，它的构造类似于[test_name]__[repr(parameter_1)]__[repr(parameter_2)]）。

作为奖励，它更简洁（没有辅助方法），并且您获得了三种方法而不是一种。这里的好处是你可以在一个方法中测试各种代码路径，每个路径得到一个测试用例（但需要一定的思考，有时对某些代码路径进行专门的测试会更好）