Question

我正在用Python进行单元测试，我试图检查两个列表中的元素是否在彼此的特定范围内。我正在考虑的两个列表是yields和list_of_yields，并且正在考虑执行self.assertEqual(round(yields-list_of_yields, 7), 0)。但是-是一个不受支持的列表类型，所以我的两个问题是如何检查元素是否在某个范围内以及如何对多个元素执行assert因为我被告知有多个asserts是不好的做法。我看到了this answer，但我的问题略有不同。

谢谢

Answer 1

如果要按照它们出现的确切顺序比较元素，您可以创建一个实用程序函数，该函数接受参数并检查它们是否满足某些条件：

def close_round(item1, item2, rounding_param):
    """Determines closeness for our test purposes"""
    return round(item1, rounding_param) == round(item2, rounding_param)

然后你可以在这样的测试用例中使用它：

assert len(yields1) == len(list_of_yields)
index = 0
for result, expected in zip(yields, list_of_yields):
    self.assertTrue(close_round(result, expected, 7),
                    msg="Test failed: got {0} expected {1} at index {2}".format(result, expected, index))
    index+=1

您可能会发现这种类型的模式很有用，在这种情况下您可以创建一个执行此操作的函数：

def test_lists_close(self, lst1, lst2, comp_function):
    """Tests if lst1 and lst2 are equal by applying comp_function
    to every element of both lists"""
    assert len(lst1) == len(lst2)
    index = 0
    for result, expected in zip(yields, list_of_yields):
        self.assertTrue(comp_function(result, expected),
                        msg="Test failed: got {0} expected {1} at index {2}".format(result, expected, index))
        index+=1

如果你经常使用它，你可能也想测试这个功能。

Answer 2

这是一种功能性方法

assert(0 == (reduce(lambda a,b:a+b, map(lambda c:round(c[0]-c[1], 7), zip(yields, list_of_yeilds))))

打破这种局面：获取zip和yields的{{1}}以获取对的列表：

list_of_yields

然后[(yields[0], list_of_yields[0]), (yields[1], list_of_yields[1]), ...]每对上的map函数得到lambda c:round(c[0]-c[1], 7)和yields的成对差异，四舍五入为7位小数。

list_of_yields

最后一步是检查此列表中的任何元素是否为非零（在这种情况下，列表不足够接近）。只需减少添加并检查0即可完成。

[round(yields[0] - list_of_yields[0], 7), round(yields[1] - list_of_yields[1], 7), ...]

Python中的单元测试涉及两个列表

2 个答案: