python - 如何使用模拟在 python 中写入和读取模拟文件对象？

Question

我正在尝试将字符串写入 python 中的模拟文件对象 mock库，但事实证明，我写入模拟对象的方法似乎不会保留任何内容，即使在内存中也是如此。

我在 ipython shell 中这样尝试:

In [1]: import mock
In [2]: fnopen = mock.mock_open()
In [3]: filehandle = fnopen()
In [4]: filehandle.write('ABC')
In [5]: filehandle.read()
Out[5]: ''

如您所见，在执行 f.read() 时，我没有得到预期的 'ABC'，而是一个空字符串。

我在做什么或理解错了什么？

Answer 1

不需要模拟。

如果您想要测试的是内存中的文件，例如对象，您可以使用 StringIO :

>>> import StringIO
>>> file_like = StringIO.StringIO()
>>> file_like.write('ABC')
>>> file_like.seek(0)
>>> file_like.read()
'ABC'