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)
