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)
def test_get_background_style_defs_uses_multiple_css_prefixes(): formatter = HtmlFormatter() lines = formatter.get_background_style_defs([".foo", ".bar"]) assert lines[0].startswith(".foo .hll, .bar .hll {") assert lines[1].startswith(".foo , .bar {")