如何编写单元测试,其中每个测试用例具有不同的输入但是相同？

Question

我需要为一些python类创建一个单元测试.我有一个输入和预期结果的数据库,应由UUT为这些输入生成.

这是我想要做的伪代码:

for i=1 to NUM_TEST_CASES:
    Load input for test case i
    execute UUT on the input and save output of run
    Load expected result for test case i
    Compare output of run with the expected result

我可以使用unittest软件包实现这一目标,还是有更好的测试包用于此目的？

Answer 1

一般来说，您描述测试的方式与单元测试很奇怪。单元测试通常不会从外部文件加载测试数据或剩余结果。一般来说，它只是在单元测试中进行硬编码。

这并不是说你的计划行不通。只能说它不典型。

你有两个选择。

1. （我们所做的）。编写一个小脚本来执行“加载测试用例 i 的输入”和“加载测试用例 i 的预期结果”。使用它来生成所需的单元测试代码。（我们使用Jinja2模板从源文件编写 Python 代码。）

   然后删除源文件。是的，删除它们。他们只会让你感到困惑。

   您剩下的是“典型”形式的正确单元测试文件，其中包含测试用例的静态数据和预期结果。

2. 编写您的setUp方法来执行“加载测试用例 i 的输入”和“加载测试用例 i 的预期结果”。编写您的test方法来测试 UUT。

它可能看起来像这样。

class OurTest( unittest.TestCase ):
    def setUp( self ):
        self.load_data()
        self.load_results()
        self.uut = ... UUT ...
    def runTest( self ):
        ... exercise UUT with source data ...
        ... check results, using self.assertXXX methods ...

想要运行多次吗？一种方法是做这样的事情。

class Test1( OurTest ):
    source_file = 'this'
    result_file = 'that'

class Test2( OutTest ):
    source_file= 'foo'
    result_file= 'bar'

这将允许unittest主程序查找并运行您的测试。