def test_get_token_style_defs_uses_css_prefix( formatter_kwargs, style_defs_args, assert_starts_with, assert_contains ): formatter = HtmlFormatter(**formatter_kwargs) for line in formatter.get_token_style_defs(*style_defs_args): assert line.startswith(assert_starts_with) for s in assert_contains: assert s in line
def _get_styles(formatter: HtmlFormatter, *, prefix: str) -> Iterator[str]: """Get styles out of a formatter, where everything has the correct prefix.""" for line in formatter.get_linenos_style_defs(): yield f"{prefix} {line}" yield from formatter.get_background_style_defs(prefix) yield from formatter.get_token_style_defs(prefix)