4

对于这个示例程序

N = int(raw_input());
n  = 0;
sum = 0;
while n<N:

    sum += int(raw_input());
    n+=1;

print sum;

我有一组测试用例，所以我需要一个调用上述 python 程序的 python 程序，提供输入并验证控制台中打印的输出。

4

4 回答 4

3

在 Unix shell 中，这可以通过以下方式实现：

$ python program.py < in > out  # Takes input from in and treats it as stdin.
                                # Your output goes to the out file.
$ diff -w out out_corr          # Validate with a correct output set

你可以像这样在 Python 中做同样的事情

from subprocess import Popen, PIPE, STDOUT

f = open('in','r')            # If you have multiple test-cases, store each 
                              # input/correct output file in a list and iterate
                              # over this snippet.
corr = open('out_corr', 'r')  # Correct outputs
cor_out = corr.read().split()
p = Popen(["python","program.py"], stdin=PIPE, stdout=PIPE)
out = p.communicate(input=f.read())[0]
out.split()
# Trivial - Validate by comparing the two lists element wise.
于 2013-01-19T07:27:13.797 回答
1

拿起分离的想法，我会考虑这个：

def input_ints():
    while True:
        yield int(raw_input())

def sum_up(it, N=None):
    sum = 0
    for n, value in enumerate(it):
        sum += int(raw_input());
        if N is not None and n >= N: break
    return sum

打印总和

要使用它，你可以这样做

inp = input_ints()
N = inp.next()
print sum_up(inp, N)

要对其进行测试，您可以执行类似的操作

inp = (1, 2, 3, 4, 5)
assert_equal(sum_up(inp), 15)
于 2013-01-19T14:37:11.863 回答
0

我写了一个测试框架（prego），可用于您的问题::

from hamcrest import contains_string
from prego import TestCase, Task

class SampleTests(TestCase):
    def test_output(self):
        task = Task()
        cmd = task.command('echo your_input | ./your_app.py')
        task.assert_that(cmd.stdout.content, 
                         contains_string('your_expected_output'))

当然，prego 提供的功能远不止这些。

于 2013-01-19T14:30:49.753 回答
-1

通常，您可能希望根据 Blender 在他的评论中建议的方式以不同的方式构建您的代码。但是，要回答您的问题，您可以使用subprocess模块编写一个脚本来调用此脚本，并将输出与预期值进行比较。

特别是看check_call方法。

于 2013-01-19T06:53:26.507 回答