python - Python 中的单元测试是否有最小样式？

Question

我想知道人们使用什么技术来简化用于单元测试的代码的“大小”。例如，我试图编码该类的一个对象并测试编码的对象(但这假定编码工作正常)。

考虑类

import unittest
class Nums(object):
    def __init__(self, n1_, n2_, n3_):
        self.n1, self.n2, self.n3 = n1_, n2_, n3_
def marshal(self):
    return "n1 %g, n2 %g, n3 %g"%(self.n1,self.n2,self.n3)

然后是基于编码(marshal)处理、基于列表和正常测试

class NumsTests(unittest.TestCase):
    def setUp(self):
        self.nu = Nums(10,20,30)
    def test_init1(self):
        self.assertEquals(self.nu.marshal(),"n1 %g, n2 %g, n3 %g"%(10,20,30))
    def test_init2(self):
        self.assertEquals([self.nu.n1,self.nu.n2,self.nu.n3],[10,21,31])
    def test_init3(self):
        self.assertEquals(self.nu.n1,10)
        self.assertEquals(self.nu.n2,21)
        self.assertEquals(self.nu.n3,31)

出现以下错误(因为 20!=21 和 30!=31，我的测试初始化​​错误或测试条件错误)

AssertionError: 'n1 10, n2 20, n3 30' != 'n1 10, n2 21, n3 31'

AssertionError: [10, 20, 30] != [10, 21, 31]

AssertionError: 20 != 21

第一条和第二条错误消息很难理解(因为您必须解析字符串或列表)。然而，第三种技术在用于测试复杂对象的代码量方面迅速激增。

有没有更好的方法来简化单元测试而不产生困难的错误消息？而且，不依赖于编码函数的准确性？

[将 test_marshal 更改为 marshal]

Answer 1

我重复上面的评论，即您不应该在您正在测试的实际类上使用测试方法。像 test_marshal 这样的函数应该放在别处(假设它们确实存在用于测试而不是用于一般用途)，通常在您的单元测试文件中。但是，暂时搁置这一点，我会做这样的事情

import unittest

class Nums(object):
    FORMAT = "n1 %g, n2 %g, n3 %g"  # make this a variable for easy testing

    def __init__(self, n1, n2, n3):
        self.n1, self.n2, self.n3 = n1, n2, n3  # no underscores necessary

    def test_marshal(self):
        return self.FORMAT % (self.n1, self.n2, self.n3)


class NumsTests(unittest.TestCase):
    def setUp(self):
        self.nums = [10, 20, 30]    # make a param list variable to avoid duplication
        self.nu = Nums(*self.nums)  # Python "apply" syntax
        self.nu_nums = [self.nu.n1, self.nu.n2, self.nu.n3]  # we'll use this repeatedly

    def test_init1(self):
        self.assertEquals(self.nu.test_marshal(), self.nu.FORMAT % self.nums )

    def test_init2(self):
        self.assertEquals(self.nums, self.nu_nums)

    def test_init3(self):
        for reference, test in zip(self.nums, self.nu_nums):
            self.assertEquals(reference, test)

参见 http://docs.python.org/library/functions.html#apply上面使用的 apply 语法的解释。

通过将您要测试的内容放入变量中，您可以避免代码重复，这似乎是您最关心的问题。

至于错误消息是否令人困惑，我想这取决于您觉得需要多少细节。就个人而言，我的单元测试为我提供了预期但不存在的代码行和值这一事实往往使我很清楚出了什么问题。但是，如果您真的想要明确告诉您哪个字段不正确，而不仅仅是不匹配的值并且您希望避免代码重复，您可以这样写:

class NumsTests(unittest.TestCase):
    def setUp(self):
        self.nums = {"n1": 10, "n2": 20, "n3": 30}  # use a dict, not a list
        self.nu = Nums(**self.nums)                 # same Python "apply" syntax

    # test_init1 and test_init2 omitted for space

    def test_init3(self):
        for attr,val in self.nums.items():
            self.assertEqual([attr, val], [attr, getattr(self.nu, val)])

如果你曾经有过不匹配的值，你现在会得到看起来像这样的错误

AssertionError: ["n1", 10] != ["n1", 11]

因此您一眼就能知道哪个字段不匹配，而不必根据值是什么来推理。这种方法仍然存在代码扩展问题，因为无论您向 Nums 类中添加多少参数，test_init3 都将保持相同的大小。

请注意 getattr 的使用要求您的 __init__ 参数与 num 类中的字段同名，例如它们必须被命名为 n1 而不是 n1_ 等。另一种方法是使用 __dict__ attribute, as described here .