python - 使用 io.StringIO 模拟文件时的 Unicode 问题

Question

我正在使用 io.StringIO 对象在类的单元测试中模拟文件。问题是此类似乎默认所有字符串都是 unicode，但内置 str 不返回 unicode 字符串:

>>> buffer = io.StringIO()
>>> buffer.write(str((1, 2)))
TypeError: can't write str to text stream

但是

>>> buffer.write(str((1, 2)) + u"")
6

有效。我认为这是因为与 unicode 字符串的连接也使结果成为 unicode。这个问题有更优雅的解决方案吗？

Answer 1

io 包提供 python3.x 兼容性。在 python 3 中，字符串默认是 unicode。

您的代码在标准 StringIO 包中运行良好，

>>> from StringIO import StringIO
>>> StringIO().write(str((1,2)))
>>>

如果您想以 Python 3 的方式进行操作，请使用 unicode() 而不是 str()。你必须在这里明确。

>>> io.StringIO().write(unicode((1,2)))
6