Highlight lazy imports in the new REPL

This commit is contained in:
bswck 2025-10-04 01:02:12 +02:00
parent 5d6026a800
commit c0c0d80a75
No known key found for this signature in database
2 changed files with 5 additions and 0 deletions

View file

@ -270,6 +270,8 @@ def is_soft_keyword_used(*tokens: TI | None) -> bool:
TI(T.NAME, string=s)
):
return not keyword.iskeyword(s)
case (None | TI(T.NEWLINE) | TI(T.INDENT) | TI(T.DEDENT), TI(string="lazy"), TI(string="import")):
return True
case _:
return False

View file

@ -68,10 +68,13 @@ def test_gen_colors_keyword_highlighting(self):
("obj.list", [(".", "op")]),
("obj.match", [(".", "op")]),
("b. \\\n format", [(".", "op")]),
("lazy", []),
("lazy()", [('(', 'op'), (')', 'op')]),
# highlights
("set", [("set", "builtin")]),
("list", [("list", "builtin")]),
(" \n dict", [("dict", "builtin")]),
(" lazy import", [("lazy", "soft_keyword"), ("import", "keyword")]),
]
for code, expected_highlights in cases:
with self.subTest(code=code):