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

时间:2013-09-28 20:58:11

标签: python python-unittest

我是Python的新手,并尝试在Ruby中经常做一些事情。即,迭代一组索引,使用它们作为函数的参数并将其结果与夹具输出数组进行比较。

所以我像往常一样在Ruby中编写它,但这导致只有一个测试用例。

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

我正在尝试为该范围内的每个项目进行测试。我怎么能这样做?

4 个答案:

答案 0 :(得分:12)

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

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)

Nose通过test generators使上述内容更容易。

import nose.tools

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

类似的问题:

答案 1 :(得分:8)

在python世界中,两个最常用的编写测试的选项是:

在pytest中,您可以非常轻松地参数化测试:

@pytest.mark.parametrize(('param1', 'param2'),[
                         (1, 'go'),
                         (2, 'do not go')])
def test_me(param1, param2):
    # write test

在运行测试时,这也会产生很好的输出:

go.py:2: test_me[1-go] PASSED
go.py:2: test_me[2-do not go] PASSED

我现在使用pytest两年了,这是非常好的工具。你有很多功能。除了参数化之外,还有一些固定装置,非常非常好的断言(你不需要编写assertEqual,只需要assert a==b,而pytest可以为它生成非常好的和有用的输出。)

答案 2 :(得分:8)

从python 3.4开始,你可以这样做:

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

https://docs.python.org/3.4/library/unittest.html?highlight=subtest#distinguishing-test-iterations-using-subtests

答案 3 :(得分:1)

如果您的问题是有关的,那么当您在诸如hackerrank或其他任何地方的竞争场所解决问题时。如果他们没有提供他们的环境来循环运行测试用例。

在本地,如果您正在运行python编译器的代码,这将很有用。

您可以简单地使用python的while循环或范围函数。

例如:

t = int(input("Enter Number of testcases"))
type(t)

while(t!=0):

    n = int(input("Enter number of data"))
    type(n)

    // Code logic or function Call

    t = t-1
相关问题