def test_lineanchors_with_startnum(self):
optdict = dict(lineanchors="foo", linenostart=5)
outfile = StringIO()
fmt = CodeHtmlFormatter(instance_class=type, **optdict)
fmt.format(tokensource, outfile)
html = outfile.getvalue()
self.assertTrue(re.search("<pre><a name=\"foo-5\">", html))
def test_linenos_with_startnum(self):
optdict = dict(linenos=True, linenostart=5)
outfile = StringIO()
fmt = CodeHtmlFormatter(instance_class=type, **optdict)
fmt.format(tokensource, outfile)
html = outfile.getvalue()
self.assertTrue(re.search("<pre>\s+5\s+6\s+7", html))
def test_linenos(self):
optdict = dict(linenos=True)
outfile = StringIO()
fmt = CodeHtmlFormatter(instance_class=type, **optdict)
fmt.format(tokensource, outfile)
html = outfile.getvalue()
self.assertTrue(re.search("<pre>\s+1\s+2\s+3", html))
def test_incode_links(self):
# reference another method
self.noop()
this_token_source = list(PythonLexer().get_tokens(
inspect.getsource(CodeHtmlFormatterTest.test_incode_links)))
hfmt = CodeHtmlFormatter(instance_class=self.__class__, nowrap=True)
houtfile = StringIO()
hfmt.format(this_token_source, houtfile)
assert '<a href="#noop">noop</a>' in houtfile.getvalue()
def test_unicode_options(self):
fmt = CodeHtmlFormatter(title=u'Föö',
instance_class=type,
cssclass=u'bär',
cssstyles=u'div:before { content: \'bäz\' }',
encoding='utf-8')
handle, pathname = tempfile.mkstemp('.html')
tfile = os.fdopen(handle, 'w+b')
fmt.format(tokensource, tfile)
tfile.close()
def test_all_options(self):
for optdict in [
dict(nowrap=True),
dict(linenos=True),
dict(linenos=True, full=True),
dict(linenos=True, full=True, noclasses=True)
]:
outfile = StringIO()
fmt = CodeHtmlFormatter(instance_class=type, **optdict)
fmt.format(tokensource, outfile)
def test_correct_output(self):
hfmt = CodeHtmlFormatter(instance_class=type, nowrap=True)
houtfile = StringIO()
hfmt.format(tokensource, houtfile)
nfmt = NullFormatter()
noutfile = StringIO()
nfmt.format(tokensource, noutfile)
stripped_html = re.sub('<.*?>', '', houtfile.getvalue())
escaped_text = escape_html(noutfile.getvalue())
self.assertEqual(stripped_html, escaped_text)
def test_get_style_defs(self):
fmt = CodeHtmlFormatter(instance_class=type)
sd = fmt.get_style_defs()
self.assertTrue(sd.startswith('.'))
fmt = CodeHtmlFormatter(instance_class=type, cssclass='foo')
sd = fmt.get_style_defs()
self.assertTrue(sd.startswith('.foo'))
sd = fmt.get_style_defs('.bar')
self.assertTrue(sd.startswith('.bar'))
sd = fmt.get_style_defs(['.bar', '.baz'])
fl = sd.splitlines()[0]
self.assertTrue('.bar' in fl and '.baz' in fl)