diff options
Diffstat (limited to 'tests/test_basic_api.py')
| -rw-r--r-- | tests/test_basic_api.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/tests/test_basic_api.py b/tests/test_basic_api.py index e0df3447..32d675b6 100644 --- a/tests/test_basic_api.py +++ b/tests/test_basic_api.py @@ -16,7 +16,7 @@ from pygments import lexers, formatters, filters, format from pygments.token import _TokenType, Text from pygments.lexer import RegexLexer from pygments.formatters.img import FontNotFound -from pygments.util import text_type, StringIO, xrange, ClassNotFound +from pygments.util import text_type, StringIO, BytesIO, xrange, ClassNotFound import support @@ -154,7 +154,8 @@ def test_get_lexers(): def test_formatter_public_api(): # test that every formatter class has the correct public API ts = list(lexers.PythonLexer().get_tokens("def f(): pass")) - out = StringIO() + string_out = StringIO() + bytes_out = BytesIO() def verify(formatter): info = formatters.FORMATTERS[formatter.__name__] @@ -163,20 +164,21 @@ def test_formatter_public_api(): assert info[2], "missing formatter aliases" assert info[4], "missing formatter docstring" - if formatter.name == 'Raw tokens': - # will not work with Unicode output file - return - try: inst = formatter(opt1="val1") except (ImportError, FontNotFound): raise support.SkipTest + try: inst.get_style_defs() except NotImplementedError: # may be raised by formatters for which it doesn't make sense pass - inst.format(ts, out) + + if formatter.unicodeoutput: + inst.format(ts, string_out) + else: + inst.format(ts, bytes_out) for name in formatters.FORMATTERS: formatter = getattr(formatters, name) |