summaryrefslogtreecommitdiff
path: root/tests/test_basic_api.py
diff options
context:
space:
mode:
authorGeorg Brandl <georg@python.org>2014-09-20 09:25:38 +0200
committerGeorg Brandl <georg@python.org>2014-09-20 09:25:38 +0200
commitf44f4394a82ca8d0e08c58850b4c08c67a981d61 (patch)
treee6c20c873b8e80551bfc39a6765483105baa292a /tests/test_basic_api.py
parentb46db39223c23d199dad06d479bd2ac165dbed19 (diff)
downloadpygments-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.py16
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():