今天在做mock模块中的patch()方法只在运行测试的上下文中才替换对象时,使用了io.StringIO结果出现报错:

经确认是字符集的问题,考虑使用io.BytesIO解决了此问题
具体代码如下:
# -*- coding:utf-8 -*-
#
# from io import StringIO
from io import BytesIO
from unittest import TestCase
from mock import patch
import url
class TestUrlPrint(TestCase):
def test_url_gets_to_stdout(self):
protocol = 'http'
host = 'www'
domain = 'example.com'
expected_url = '{}://{}.{}\n'.format(protocol, host, domain)
with patch('sys.stdout', new=BytesIO()) as fake_out:
url.urlprint(protocol, host, domain)
self.assertEqual(fake_out.getvalue(), expected_url)
python2.7的字符转换问题,需要多加注意。