# encoding: utf-8 from __future__ import unicode_literals from io import StringIO, BytesIO from cinje import flatten, fragment tmpl = fragment("Zoë") class TestCinjeIO(object): def test_flatten_string_unicode(self): assert flatten(tmpl()) == 'Zoë\n' def test_flatten_string_binary(self): assert flatten(tmpl(), encoding='utf8') == 'Zoë\n'.encode('utf8') def test_flatten_filelike_unicode(self): container = StringIO() assert flatten(tmpl(), container) == 4 assert container.getvalue() == 'Zoë\n' def test_flatten_filelike_binary(self): container = BytesIO() assert flatten(tmpl(), container, encoding='utf8') == 5 assert container.getvalue() == 'Zoë\n'.encode('utf8')
def tmpl(): return fragment(CODE)