diff options
author | Georg Brandl <georg@python.org> | 2014-09-20 09:25:38 +0200 |
---|---|---|
committer | Georg Brandl <georg@python.org> | 2014-09-20 09:25:38 +0200 |
commit | f44f4394a82ca8d0e08c58850b4c08c67a981d61 (patch) | |
tree | e6c20c873b8e80551bfc39a6765483105baa292a /tests/test_basic_api.py | |
parent | b46db39223c23d199dad06d479bd2ac165dbed19 (diff) | |
download | pygments-f44f4394a82ca8d0e08c58850b4c08c67a981d61.tar.gz |
Refactored formatter mapping to work like the lexer mapping.
Thanks to Ilia Choly for the initial pull request.
Diffstat (limited to 'tests/test_basic_api.py')
-rw-r--r-- | tests/test_basic_api.py | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/tests/test_basic_api.py b/tests/test_basic_api.py index 893fa90c..e1f51d62 100644 --- a/tests/test_basic_api.py +++ b/tests/test_basic_api.py @@ -151,10 +151,10 @@ def test_formatter_public_api(): out = StringIO() # test that every formatter class has the correct public API def verify(formatter, info): - assert len(info) == 4 - assert info[0], "missing formatter name" - assert info[1], "missing formatter aliases" - assert info[3], "missing formatter docstring" + assert len(info) == 5 + assert info[1], "missing formatter name" + assert info[2], "missing formatter aliases" + assert info[4], "missing formatter docstring" if formatter.name == 'Raw tokens': # will not work with Unicode output file @@ -172,7 +172,9 @@ def test_formatter_public_api(): inst.format(ts, out) for formatter, info in formatters.FORMATTERS.items(): - yield verify, formatter, info + fmter = getattr(formatters, formatter) + yield verify, fmter, info + def test_formatter_encodings(): from pygments.formatters import HtmlFormatter @@ -223,7 +225,9 @@ def test_formatter_unicode_handling(): assert type(out) is bytes, '%s: %r' % (formatter, out) for formatter, info in formatters.FORMATTERS.items(): - yield verify, formatter + # this tests the automatic importing as well + fmter = getattr(formatters, formatter) + yield verify, fmter def test_get_formatters(): |