1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
"""
Pygments terminal formatter tests
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
:copyright: Copyright 2006-2023 by the Pygments team, see AUTHORS.
:license: BSD, see LICENSE for details.
"""
import re
from io import StringIO
from pygments.lexers.sql import PlPgsqlLexer
from pygments.formatters import TerminalFormatter, Terminal256Formatter, \
HtmlFormatter, LatexFormatter
from pygments.style import Style
from pygments.token import Token
from pygments.lexers import Python3Lexer
from pygments import highlight
DEMO_TEXT = '''\
-- comment
select
* from bar;
'''
DEMO_LEXER = PlPgsqlLexer
DEMO_TOKENS = list(DEMO_LEXER().get_tokens(DEMO_TEXT))
ANSI_RE = re.compile(r'\x1b[\w\W]*?m')
def strip_ansi(x):
return ANSI_RE.sub('', x)
def test_reasonable_output():
out = StringIO()
TerminalFormatter().format(DEMO_TOKENS, out)
plain = strip_ansi(out.getvalue())
assert DEMO_TEXT.count('\n') == plain.count('\n')
print(repr(plain))
for a, b in zip(DEMO_TEXT.splitlines(), plain.splitlines()):
assert a == b
def test_reasonable_output_lineno():
out = StringIO()
TerminalFormatter(linenos=True).format(DEMO_TOKENS, out)
plain = strip_ansi(out.getvalue())
assert DEMO_TEXT.count('\n') + 1 == plain.count('\n')
print(repr(plain))
for a, b in zip(DEMO_TEXT.splitlines(), plain.splitlines()):
assert a in b
class MyStyle(Style):
styles = {
Token.Comment: 'ansibrightblack',
Token.String: 'ansibrightblue bg:ansired',
Token.Number: 'ansibrightgreen bg:ansigreen',
Token.Number.Hex: 'ansigreen bg:ansibrightred',
}
CODE = '''
# this should be a comment
print("Hello World")
async def function(a,b,c, *d, **kwarg:Bool)->Bool:
pass
return 123, 0xb3e3
'''
def test_style_html():
style = HtmlFormatter(style=MyStyle).get_style_defs()
assert '#555555' in style, "ansigray for comment not html css style"
def test_others_work():
"""Check other formatters don't crash."""
highlight(CODE, Python3Lexer(), LatexFormatter(style=MyStyle))
highlight(CODE, Python3Lexer(), HtmlFormatter(style=MyStyle))
def test_256esc_seq():
"""
Test that a few escape sequences are actually used when using ansi<> color
codes.
"""
def termtest(x):
return highlight(x, Python3Lexer(),
Terminal256Formatter(style=MyStyle))
assert '32;101' in termtest('0x123')
assert '92;42' in termtest('123')
assert '90' in termtest('#comment')
assert '94;41' in termtest('"String"')
|
