terminalwriter: fix crash trying to highlight empty source

For quick checking I don't know how we can reach here with an empty
source, so test just checks the function directly.

Fix #11758.
This commit is contained in:
Ran Benita
2024-01-02 19:31:37 +02:00
parent a53984a55b
commit cb5a42c836
3 changed files with 18 additions and 1 deletions

View File

@@ -306,3 +306,17 @@ def test_code_highlight(has_markup, code_highlight, expected, color_mapping):
match=re.escape("indents size (2) should have same size as lines (1)"),
):
tw._write_source(["assert 0"], [" ", " "])
def test_highlight_empty_source() -> None:
"""Don't crash trying to highlight empty source code.
Issue #11758.
"""
f = io.StringIO()
tw = terminalwriter.TerminalWriter(f)
tw.hasmarkup = True
tw.code_highlight = True
tw._write_source([])
assert f.getvalue() == ""