packages/lang/python/python-pygments/test.sh

38 lines
1.1 KiB
Bash
Executable File

#!/bin/sh
[ "$1" = "python3-pygments" ] || exit 0
python3 - << EOF
import sys
import pygments
from pygments import highlight
from pygments.lexers import PythonLexer, get_lexer_by_name
from pygments.formatters import HtmlFormatter, NullFormatter
if pygments.__version__ != "$2":
print("Wrong version: " + pygments.__version__)
sys.exit(1)
code = "def hello(name):\n print('Hello, ' + name)\n"
# Test basic highlighting to HTML
formatter = HtmlFormatter()
result = highlight(code, PythonLexer(), formatter)
assert '<span' in result, "Expected HTML span tags in output"
assert 'hello' in result, "Expected function name in output"
# Test getting lexer by name
lexer = get_lexer_by_name("python")
assert lexer is not None, "Expected to get Python lexer by name"
# Test highlighting to plain text (NullFormatter strips markup)
plain = highlight(code, PythonLexer(), NullFormatter())
assert 'hello' in plain, "Expected function name in plain output"
# Test CSS generation
css = formatter.get_style_defs('.highlight')
assert '.highlight' in css, "Expected CSS class in style defs"
sys.exit(0)
EOF