5

我想测试一些直接使用printandinput函数的(python 3)代码。据我了解,最简单的方法是依赖注入:修改代码,使其将输入和输出流作为参数,默认情况下使用sys.stdinandsys.stdout并在测试期间传入模拟对象。很明显如何处理print调用:

print(text)
#replaced with...
print(text, file=output_stream)

但是,input没有输入和输出流的参数。以下代码是否正确再现了它的行为?

text = input(prompt)
#replaced with...
print(prompt, file=output_stream, end='')
text = input_stream.readline()[:-1]

我查看了 的实现input,它做了很多魔术,调用sys.stdin.fileno和检查sys.stdin.encodingsys.stdin.errors而不是调用任何read*方法——我不知道从哪里开始嘲笑这些。

4

2 回答 2

7

input()只有你提到的魔法才会发生stdin并且stdout不会被改变,因为只有这样它才能使用像 readline 库这样的东西。如果您将它们替换为其他内容(是否为真实文件),则归结为以下代码:

/* Fallback if we're not interactive */
if (promptarg != NULL) {
    if (PyFile_WriteObject(promptarg, fout, Py_PRINT_RAW) != 0)
         return NULL;
}
tmp = _PyObject_CallMethodId(fout, &PyId_flush, "");
if (tmp == NULL)
    PyErr_Clear();
else
    Py_DECREF(tmp);
return PyFile_GetLine(fin, -1);

在哪里PyFile_GetLine 调用readline方法。因此,嘲笑sys.std*将起作用。

建议您使用try: finally:上下文处理器或mock模块执行此操作,以便即使您正在测试的代码因异常而失败,也会恢复输出:

from unittest.mock import patch
from io import StringIO

with patch("sys.stdin", StringIO("FOO")), patch("sys.stdout", new_callable=StringIO) as mocked_out:
    x = input()
    print("Read:", x)

assert mocked_out.getvalue() == "Read: FOO\n"
于 2012-11-05T20:11:49.883 回答
6

如果您将类似文件的对象分配给sys.stdinPython 的input函数,将使用它而不是标准输入。但请记住sys.stdin在完成后重新分配回标准输入。同样的技巧适用于sys.stdout. 你可以这样做:

original_stdin = sys.stdin
sys.stdin = open('inputfile.txt', 'r')

original_stdout = sys.stdout
sys.stdout = open('outputfile.txt', 'w')

response = input('say hi: ')
print(response)

sys.stdin = original_stdin
sys.stdout = original_stdout

这两行

response = input('say hi: ')
print(response)

将使用指定的文件 ( inputfile.txtand outputfile.txt) 而不是标准输入和标准输出。

更新:如果您不想处理物理文件,请查看io模块。它提供了io.StringIO允许您执行内存中文本流操作的类。

original_stdin = sys.stdin
sys.stdin = io.StringIO('input string')

original_stdout = sys.stdout
sys.stdout = io.StringIO()

response = input('say hi: ')
print(response)

output = sys.stdout.getvalue()

sys.stdin = original_stdin
sys.stdout = original_stdout

print(output)
于 2012-11-05T19:27:27.387 回答