tests: adjust StdoutBuffer for python2/python3 compatibility

io.StringIO expects unicode writes, while string literal on python2 are
bytes. Act accordingly.
This commit is contained in:
Marek Marczykowski-Górecki 2017-03-09 00:17:37 +01:00
父節點 a575bc3422
當前提交 6b8bbd9c51
沒有發現已知的金鑰在資料庫的簽署中
GPG 金鑰 ID: 063938BA42CFA724

查看文件

@ -22,9 +22,13 @@
import io
import sys
class StdoutBuffer(object):
def __init__(self):
self.stdout = io.StringIO()
if sys.version_info[0] >= 3:
self.stdout = io.StringIO()
else:
self.stdout = io.BytesIO()
def __enter__(self):
sys.stdout = self.stdout