如何在一个循环中运行多个Python测试用例？

问题描述

我是新手，正在尝试做一些我在Ruby中经常做的事情。也就是说，迭代一组索引，将它们用作函数的参数，并将其结果与一组装置输出进行比较。

所以我像在Ruby中通常所做的那样编写了它，但这只产生了一个测试用例。

  def test_output(self):
    for i in range(1,11):
      ....
      self.assertEqual(fn(i),output[i])

我正在尝试对范围中的每一项进行测试。我如何才能做到这一点？

---

解决方案

使用单元测试可以在一个测试用例中显示两个序列之间的差异。

seq1 = range(1, 11)
seq2 = (fn(j) for j in seq1)
assertSequenceEqual(seq1, seq2)

如果不够灵活，则使用unittest可以生成多个测试，但这有点棘手。

def fn(i): ...
output = ...

class TestSequence(unittest.TestCase):
    pass

for i in range(1,11):
    testmethodname = 'test_fn_{0}'.format(i)
    testmethod = lambda self: self.assertEqual(fn(i), output[i])
    setattr(TestSequence, testmethodname, testmethod)

通过test generators使上述操作更轻松。

import nose.tools

def test_fn():
    for i in range(1, 11):
        yield nose.tools.assert_equals, output[i], fn(i)

类似问题：

• Python unittest: Generate multiple tests programmatically?
• How to generate dynamic (parametrized) unit tests in python?