TypeError: unicode argument expected, got 'str'

今天在做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的字符转换问题，需要多加注意。

本文转自 许大树 51CTO博客，原文链接：http://blog.51cto.com/abelxu/1902730，如需转载请自行联系原作者