python - sys.stdin のモックアップ?

Question

unittest次のような関数の sを書きたいとします。

def test_me(instream):
    out = ""
    for line in instream:
       out += foo(line)
    return out

通常は次のように使用されます。

test_me(sys.stdin)

さて、あなたunittestの s では使用したくないのですsys.stdinが、 の動作を模倣するオブジェクトでsys.stdinあり、そのオブジェクトはあなたが完全に制御する必要があります。このようにして、関数をそのままテストし、好きな偽の入力を挿入できます。そのためには、どのようなオブジェクトを作成する必要がありますか?

Answer 1

試してみてくださいStringIOおよび/またはcStringIOモジュール。

Answer 2

この場合、instream引数はジェネレーターとして使用されます。したがって、ジェネレータを実装するだけで十分です。おそらく、関数に行のリストを提供するだけで十分でしょう。